Whamcloud - gitweb
LU-16831 lod: replace (__u16)-1 with LOV_ALL_STRIPES
[fs/lustre-release.git] / lustre / lod / lod_object.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,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License version 2 for more details.  A copy is
14  * included in the COPYING file that accompanied this code.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright  2009 Sun Microsystems, Inc. All rights reserved
24  * Use is subject to license terms.
25  *
26  * Copyright (c) 2012, 2017, Intel Corporation.
27  */
28 /*
29  * lustre/lod/lod_object.c
30  *
31  * This file contains implementations of methods for the OSD API
32  * for the Logical Object Device (LOD) layer, which provides a virtual
33  * local OSD object interface to the MDD layer, and abstracts the
34  * addressing of local (OSD) and remote (OSP) objects. The API is
35  * described in the file lustre/include/dt_object.h and in
36  * Documentation/osd-api.txt.
37  *
38  * Author: Alex Zhuravlev <alexey.zhuravlev@intel.com>
39  */
40
41 #define DEBUG_SUBSYSTEM S_MDS
42
43 #include <linux/random.h>
44
45 #include <obd.h>
46 #include <obd_class.h>
47 #include <obd_support.h>
48
49 #include <lustre_fid.h>
50 #include <lustre_linkea.h>
51 #include <lustre_lmv.h>
52 #include <uapi/linux/lustre/lustre_param.h>
53 #include <lustre_swab.h>
54 #include <uapi/linux/lustre/lustre_ver.h>
55 #include <lprocfs_status.h>
56 #include <md_object.h>
57
58 #include "lod_internal.h"
59
60 static const char dot[] = ".";
61 static const char dotdot[] = "..";
62
63 /**
64  * Implementation of dt_index_operations::dio_lookup
65  *
66  * Used with regular (non-striped) objects.
67  *
68  * \see dt_index_operations::dio_lookup() in the API description for details.
69  */
70 static int lod_lookup(const struct lu_env *env, struct dt_object *dt,
71                       struct dt_rec *rec, const struct dt_key *key)
72 {
73         struct dt_object *next = dt_object_child(dt);
74         return next->do_index_ops->dio_lookup(env, next, rec, key);
75 }
76
77 /**
78  * Implementation of dt_index_operations::dio_declare_insert.
79  *
80  * Used with regular (non-striped) objects.
81  *
82  * \see dt_index_operations::dio_declare_insert() in the API description
83  * for details.
84  */
85 static int lod_declare_insert(const struct lu_env *env, struct dt_object *dt,
86                               const struct dt_rec *rec,
87                               const struct dt_key *key, struct thandle *th)
88 {
89         return lod_sub_declare_insert(env, dt_object_child(dt), rec, key, th);
90 }
91
92 /**
93  * Implementation of dt_index_operations::dio_insert.
94  *
95  * Used with regular (non-striped) objects
96  *
97  * \see dt_index_operations::dio_insert() in the API description for details.
98  */
99 static int lod_insert(const struct lu_env *env, struct dt_object *dt,
100                       const struct dt_rec *rec, const struct dt_key *key,
101                       struct thandle *th)
102 {
103         return lod_sub_insert(env, dt_object_child(dt), rec, key, th);
104 }
105
106 /**
107  * Implementation of dt_index_operations::dio_declare_delete.
108  *
109  * Used with regular (non-striped) objects.
110  *
111  * \see dt_index_operations::dio_declare_delete() in the API description
112  * for details.
113  */
114 static int lod_declare_delete(const struct lu_env *env, struct dt_object *dt,
115                               const struct dt_key *key, struct thandle *th)
116 {
117         return lod_sub_declare_delete(env, dt_object_child(dt), key, th);
118 }
119
120 /**
121  * Implementation of dt_index_operations::dio_delete.
122  *
123  * Used with regular (non-striped) objects.
124  *
125  * \see dt_index_operations::dio_delete() in the API description for details.
126  */
127 static int lod_delete(const struct lu_env *env, struct dt_object *dt,
128                       const struct dt_key *key, struct thandle *th)
129 {
130         return lod_sub_delete(env, dt_object_child(dt), key, th);
131 }
132
133 /**
134  * Implementation of dt_it_ops::init.
135  *
136  * Used with regular (non-striped) objects.
137  *
138  * \see dt_it_ops::init() in the API description for details.
139  */
140 static struct dt_it *lod_it_init(const struct lu_env *env,
141                                  struct dt_object *dt, __u32 attr)
142 {
143         struct dt_object        *next = dt_object_child(dt);
144         struct lod_it           *it = &lod_env_info(env)->lti_it;
145         struct dt_it            *it_next;
146
147         it_next = next->do_index_ops->dio_it.init(env, next, attr);
148         if (IS_ERR(it_next))
149                 return it_next;
150
151         /* currently we do not use more than one iterator per thread
152          * so we store it in thread info. if at some point we need
153          * more active iterators in a single thread, we can allocate
154          * additional ones */
155         LASSERT(it->lit_obj == NULL);
156
157         it->lit_it = it_next;
158         it->lit_obj = next;
159
160         return (struct dt_it *)it;
161 }
162
163 #define LOD_CHECK_IT(env, it)                                   \
164 do {                                                            \
165         LASSERT((it)->lit_obj != NULL);                         \
166         LASSERT((it)->lit_it != NULL);                          \
167 } while (0)
168
169 /**
170  * Implementation of dt_index_operations::dio_it.fini.
171  *
172  * Used with regular (non-striped) objects.
173  *
174  * \see dt_index_operations::dio_it.fini() in the API description for details.
175  */
176 static void lod_it_fini(const struct lu_env *env, struct dt_it *di)
177 {
178         struct lod_it *it = (struct lod_it *)di;
179
180         LOD_CHECK_IT(env, it);
181         it->lit_obj->do_index_ops->dio_it.fini(env, it->lit_it);
182
183         /* the iterator not in use any more */
184         it->lit_obj = NULL;
185         it->lit_it = NULL;
186 }
187
188 /**
189  * Implementation of dt_it_ops::get.
190  *
191  * Used with regular (non-striped) objects.
192  *
193  * \see dt_it_ops::get() in the API description for details.
194  */
195 static int lod_it_get(const struct lu_env *env, struct dt_it *di,
196                       const struct dt_key *key)
197 {
198         const struct lod_it *it = (const struct lod_it *)di;
199
200         LOD_CHECK_IT(env, it);
201         return it->lit_obj->do_index_ops->dio_it.get(env, it->lit_it, key);
202 }
203
204 /**
205  * Implementation of dt_it_ops::put.
206  *
207  * Used with regular (non-striped) objects.
208  *
209  * \see dt_it_ops::put() in the API description for details.
210  */
211 static void lod_it_put(const struct lu_env *env, struct dt_it *di)
212 {
213         struct lod_it *it = (struct lod_it *)di;
214
215         LOD_CHECK_IT(env, it);
216         return it->lit_obj->do_index_ops->dio_it.put(env, it->lit_it);
217 }
218
219 /**
220  * Implementation of dt_it_ops::next.
221  *
222  * Used with regular (non-striped) objects
223  *
224  * \see dt_it_ops::next() in the API description for details.
225  */
226 static int lod_it_next(const struct lu_env *env, struct dt_it *di)
227 {
228         struct lod_it *it = (struct lod_it *)di;
229
230         LOD_CHECK_IT(env, it);
231         return it->lit_obj->do_index_ops->dio_it.next(env, it->lit_it);
232 }
233
234 /**
235  * Implementation of dt_it_ops::key.
236  *
237  * Used with regular (non-striped) objects.
238  *
239  * \see dt_it_ops::key() in the API description for details.
240  */
241 static struct dt_key *lod_it_key(const struct lu_env *env,
242                                  const struct dt_it *di)
243 {
244         const struct lod_it *it = (const struct lod_it *)di;
245
246         LOD_CHECK_IT(env, it);
247         return it->lit_obj->do_index_ops->dio_it.key(env, it->lit_it);
248 }
249
250 /**
251  * Implementation of dt_it_ops::key_size.
252  *
253  * Used with regular (non-striped) objects.
254  *
255  * \see dt_it_ops::key_size() in the API description for details.
256  */
257 static int lod_it_key_size(const struct lu_env *env, const struct dt_it *di)
258 {
259         struct lod_it *it = (struct lod_it *)di;
260
261         LOD_CHECK_IT(env, it);
262         return it->lit_obj->do_index_ops->dio_it.key_size(env, it->lit_it);
263 }
264
265 /**
266  * Implementation of dt_it_ops::rec.
267  *
268  * Used with regular (non-striped) objects.
269  *
270  * \see dt_it_ops::rec() in the API description for details.
271  */
272 static int lod_it_rec(const struct lu_env *env, const struct dt_it *di,
273                       struct dt_rec *rec, __u32 attr)
274 {
275         const struct lod_it *it = (const struct lod_it *)di;
276
277         LOD_CHECK_IT(env, it);
278         return it->lit_obj->do_index_ops->dio_it.rec(env, it->lit_it, rec,
279                                                      attr);
280 }
281
282 /**
283  * Implementation of dt_it_ops::rec_size.
284  *
285  * Used with regular (non-striped) objects.
286  *
287  * \see dt_it_ops::rec_size() in the API description for details.
288  */
289 static int lod_it_rec_size(const struct lu_env *env, const struct dt_it *di,
290                            __u32 attr)
291 {
292         const struct lod_it *it = (const struct lod_it *)di;
293
294         LOD_CHECK_IT(env, it);
295         return it->lit_obj->do_index_ops->dio_it.rec_size(env, it->lit_it,
296                                                           attr);
297 }
298
299 /**
300  * Implementation of dt_it_ops::store.
301  *
302  * Used with regular (non-striped) objects.
303  *
304  * \see dt_it_ops::store() in the API description for details.
305  */
306 static __u64 lod_it_store(const struct lu_env *env, const struct dt_it *di)
307 {
308         const struct lod_it *it = (const struct lod_it *)di;
309
310         LOD_CHECK_IT(env, it);
311         return it->lit_obj->do_index_ops->dio_it.store(env, it->lit_it);
312 }
313
314 /**
315  * Implementation of dt_it_ops::load.
316  *
317  * Used with regular (non-striped) objects.
318  *
319  * \see dt_it_ops::load() in the API description for details.
320  */
321 static int lod_it_load(const struct lu_env *env, const struct dt_it *di,
322                        __u64 hash)
323 {
324         const struct lod_it *it = (const struct lod_it *)di;
325
326         LOD_CHECK_IT(env, it);
327         return it->lit_obj->do_index_ops->dio_it.load(env, it->lit_it, hash);
328 }
329
330 /**
331  * Implementation of dt_it_ops::key_rec.
332  *
333  * Used with regular (non-striped) objects.
334  *
335  * \see dt_it_ops::rec() in the API description for details.
336  */
337 static int lod_it_key_rec(const struct lu_env *env, const struct dt_it *di,
338                           void *key_rec)
339 {
340         const struct lod_it *it = (const struct lod_it *)di;
341
342         LOD_CHECK_IT(env, it);
343         return it->lit_obj->do_index_ops->dio_it.key_rec(env, it->lit_it,
344                                                          key_rec);
345 }
346
347 static const struct dt_index_operations lod_index_ops = {
348         .dio_lookup             = lod_lookup,
349         .dio_declare_insert     = lod_declare_insert,
350         .dio_insert             = lod_insert,
351         .dio_declare_delete     = lod_declare_delete,
352         .dio_delete             = lod_delete,
353         .dio_it = {
354                 .init           = lod_it_init,
355                 .fini           = lod_it_fini,
356                 .get            = lod_it_get,
357                 .put            = lod_it_put,
358                 .next           = lod_it_next,
359                 .key            = lod_it_key,
360                 .key_size       = lod_it_key_size,
361                 .rec            = lod_it_rec,
362                 .rec_size       = lod_it_rec_size,
363                 .store          = lod_it_store,
364                 .load           = lod_it_load,
365                 .key_rec        = lod_it_key_rec,
366         }
367 };
368
369 /**
370  * Implementation of dt_index_operations::dio_lookup
371  *
372  * Used with striped directories.
373  *
374  * \see dt_index_operations::dio_lookup() in the API description for details.
375  */
376 static int lod_striped_lookup(const struct lu_env *env, struct dt_object *dt,
377                       struct dt_rec *rec, const struct dt_key *key)
378 {
379         struct lod_object *lo = lod_dt_obj(dt);
380         struct dt_object *next;
381         const char *name = (const char *)key;
382
383         LASSERT(lo->ldo_dir_stripe_count > 0);
384
385         if (strcmp(name, dot) == 0) {
386                 struct lu_fid *fid = (struct lu_fid *)rec;
387
388                 *fid = *lod_object_fid(lo);
389                 return 1;
390         }
391
392         if (strcmp(name, dotdot) == 0) {
393                 next = dt_object_child(dt);
394         } else {
395                 int index;
396
397                 index = __lmv_name_to_stripe_index(lo->ldo_dir_hash_type,
398                                                    lo->ldo_dir_stripe_count,
399                                                    lo->ldo_dir_migrate_hash,
400                                                    lo->ldo_dir_migrate_offset,
401                                                    name, strlen(name), true);
402                 if (index < 0)
403                         return index;
404
405                 next = lo->ldo_stripe[index];
406                 if (!next || !dt_object_exists(next))
407                         return -ENODEV;
408         }
409
410         return next->do_index_ops->dio_lookup(env, next, rec, key);
411 }
412
413 /**
414  * Implementation of dt_it_ops::init.
415  *
416  * Used with striped objects. Internally just initializes the iterator
417  * on the first stripe.
418  *
419  * \see dt_it_ops::init() in the API description for details.
420  */
421 static struct dt_it *lod_striped_it_init(const struct lu_env *env,
422                                          struct dt_object *dt, __u32 attr)
423 {
424         struct lod_object *lo = lod_dt_obj(dt);
425         struct dt_object *next;
426         struct lod_it *it = &lod_env_info(env)->lti_it;
427         struct dt_it *it_next;
428         __u16 index = 0;
429
430         LASSERT(lo->ldo_dir_stripe_count > 0);
431
432         do {
433                 next = lo->ldo_stripe[index];
434                 if (next && dt_object_exists(next))
435                         break;
436         } while (++index < lo->ldo_dir_stripe_count);
437
438         /* no valid stripe */
439         if (!next || !dt_object_exists(next))
440                 return ERR_PTR(-ENODEV);
441
442         LASSERT(next->do_index_ops != NULL);
443
444         it_next = next->do_index_ops->dio_it.init(env, next, attr);
445         if (IS_ERR(it_next))
446                 return it_next;
447
448         /* currently we do not use more than one iterator per thread
449          * so we store it in thread info. if at some point we need
450          * more active iterators in a single thread, we can allocate
451          * additional ones */
452         LASSERT(it->lit_obj == NULL);
453
454         it->lit_stripe_index = index;
455         it->lit_attr = attr;
456         it->lit_it = it_next;
457         it->lit_obj = dt;
458
459         return (struct dt_it *)it;
460 }
461
462 #define LOD_CHECK_STRIPED_IT(env, it, lo)                               \
463 do {                                                                    \
464         LASSERT((it)->lit_obj != NULL);                                 \
465         LASSERT((it)->lit_it != NULL);                                  \
466         LASSERT((lo)->ldo_dir_stripe_count > 0);                        \
467         LASSERT((it)->lit_stripe_index < (lo)->ldo_dir_stripe_count);   \
468 } while (0)
469
470 /**
471  * Implementation of dt_it_ops::fini.
472  *
473  * Used with striped objects.
474  *
475  * \see dt_it_ops::fini() in the API description for details.
476  */
477 static void lod_striped_it_fini(const struct lu_env *env, struct dt_it *di)
478 {
479         struct lod_it           *it = (struct lod_it *)di;
480         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
481         struct dt_object        *next;
482
483         /* If lit_it == NULL, then it means the sub_it has been finished,
484          * which only happens in failure cases, see lod_striped_it_next() */
485         if (it->lit_it != NULL) {
486                 LOD_CHECK_STRIPED_IT(env, it, lo);
487
488                 next = lo->ldo_stripe[it->lit_stripe_index];
489                 if (next) {
490                         LASSERT(next->do_index_ops != NULL);
491                         next->do_index_ops->dio_it.fini(env, it->lit_it);
492                 }
493         }
494
495         /* the iterator not in use any more */
496         it->lit_obj = NULL;
497         it->lit_it = NULL;
498         it->lit_stripe_index = 0;
499 }
500
501 /**
502  * Implementation of dt_it_ops::get.
503  *
504  * Right now it's not used widely, only to reset the iterator to the
505  * initial position. It should be possible to implement a full version
506  * which chooses a correct stripe to be able to position with any key.
507  *
508  * \see dt_it_ops::get() in the API description for details.
509  */
510 static int lod_striped_it_get(const struct lu_env *env, struct dt_it *di,
511                               const struct dt_key *key)
512 {
513         const struct lod_it *it = (const struct lod_it *)di;
514         struct lod_object *lo = lod_dt_obj(it->lit_obj);
515         struct dt_object *next;
516
517         LOD_CHECK_STRIPED_IT(env, it, lo);
518
519         next = lo->ldo_stripe[it->lit_stripe_index];
520         LASSERT(next != NULL);
521         LASSERT(dt_object_exists(next));
522         LASSERT(next->do_index_ops != NULL);
523
524         return next->do_index_ops->dio_it.get(env, it->lit_it, key);
525 }
526
527 /**
528  * Implementation of dt_it_ops::put.
529  *
530  * Used with striped objects.
531  *
532  * \see dt_it_ops::put() in the API description for details.
533  */
534 static void lod_striped_it_put(const struct lu_env *env, struct dt_it *di)
535 {
536         struct lod_it *it = (struct lod_it *)di;
537         struct lod_object *lo = lod_dt_obj(it->lit_obj);
538         struct dt_object *next;
539
540         /*
541          * If lit_it == NULL, then it means the sub_it has been finished,
542          * which only happens in failure cases, see lod_striped_it_next()
543          */
544         if (!it->lit_it)
545                 return;
546
547         LOD_CHECK_STRIPED_IT(env, it, lo);
548
549         next = lo->ldo_stripe[it->lit_stripe_index];
550         LASSERT(next != NULL);
551         LASSERT(next->do_index_ops != NULL);
552
553         return next->do_index_ops->dio_it.put(env, it->lit_it);
554 }
555
556 /**
557  * Implementation of dt_it_ops::next.
558  *
559  * Used with striped objects. When the end of the current stripe is
560  * reached, the method takes the next stripe's iterator.
561  *
562  * \see dt_it_ops::next() in the API description for details.
563  */
564 static int lod_striped_it_next(const struct lu_env *env, struct dt_it *di)
565 {
566         struct lod_it *it = (struct lod_it *)di;
567         struct lod_object *lo = lod_dt_obj(it->lit_obj);
568         struct dt_object *next;
569         struct dt_it *it_next;
570         __u32 index;
571         int rc;
572
573         ENTRY;
574
575         LOD_CHECK_STRIPED_IT(env, it, lo);
576
577         next = lo->ldo_stripe[it->lit_stripe_index];
578         LASSERT(next != NULL);
579         LASSERT(dt_object_exists(next));
580         LASSERT(next->do_index_ops != NULL);
581 again:
582         rc = next->do_index_ops->dio_it.next(env, it->lit_it);
583         if (rc < 0)
584                 RETURN(rc);
585
586         if (rc == 0 && it->lit_stripe_index == 0)
587                 RETURN(rc);
588
589         if (rc == 0 && it->lit_stripe_index > 0) {
590                 struct lu_dirent *ent;
591
592                 ent = (struct lu_dirent *)lod_env_info(env)->lti_key;
593
594                 rc = next->do_index_ops->dio_it.rec(env, it->lit_it,
595                                                     (struct dt_rec *)ent,
596                                                     it->lit_attr);
597                 if (rc != 0)
598                         RETURN(rc);
599
600                 /* skip . and .. for slave stripe */
601                 if ((strncmp(ent->lde_name, ".",
602                              le16_to_cpu(ent->lde_namelen)) == 0 &&
603                      le16_to_cpu(ent->lde_namelen) == 1) ||
604                     (strncmp(ent->lde_name, "..",
605                              le16_to_cpu(ent->lde_namelen)) == 0 &&
606                      le16_to_cpu(ent->lde_namelen) == 2))
607                         goto again;
608
609                 RETURN(rc);
610         }
611
612         next->do_index_ops->dio_it.put(env, it->lit_it);
613         next->do_index_ops->dio_it.fini(env, it->lit_it);
614         it->lit_it = NULL;
615
616         /* go to next stripe */
617         index = it->lit_stripe_index;
618         while (++index < lo->ldo_dir_stripe_count) {
619                 next = lo->ldo_stripe[index];
620                 if (!next)
621                         continue;
622
623                 if (!dt_object_exists(next))
624                         continue;
625
626                 rc = next->do_ops->do_index_try(env, next,
627                                                 &dt_directory_features);
628                 if (rc != 0)
629                         RETURN(rc);
630
631                 LASSERT(next->do_index_ops != NULL);
632
633                 it_next = next->do_index_ops->dio_it.init(env, next,
634                                                           it->lit_attr);
635                 if (IS_ERR(it_next))
636                         RETURN(PTR_ERR(it_next));
637
638                 rc = next->do_index_ops->dio_it.get(env, it_next,
639                                                     (const struct dt_key *)"");
640                 if (rc <= 0)
641                         RETURN(rc == 0 ? -EIO : rc);
642
643                 it->lit_it = it_next;
644                 it->lit_stripe_index = index;
645                 goto again;
646
647         }
648
649         RETURN(1);
650 }
651
652 /**
653  * Implementation of dt_it_ops::key.
654  *
655  * Used with striped objects.
656  *
657  * \see dt_it_ops::key() in the API description for details.
658  */
659 static struct dt_key *lod_striped_it_key(const struct lu_env *env,
660                                          const struct dt_it *di)
661 {
662         const struct lod_it     *it = (const struct lod_it *)di;
663         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
664         struct dt_object        *next;
665
666         LOD_CHECK_STRIPED_IT(env, it, lo);
667
668         next = lo->ldo_stripe[it->lit_stripe_index];
669         LASSERT(next != NULL);
670         LASSERT(next->do_index_ops != NULL);
671
672         return next->do_index_ops->dio_it.key(env, it->lit_it);
673 }
674
675 /**
676  * Implementation of dt_it_ops::key_size.
677  *
678  * Used with striped objects.
679  *
680  * \see dt_it_ops::size() in the API description for details.
681  */
682 static int lod_striped_it_key_size(const struct lu_env *env,
683                                    const struct dt_it *di)
684 {
685         struct lod_it           *it = (struct lod_it *)di;
686         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
687         struct dt_object        *next;
688
689         LOD_CHECK_STRIPED_IT(env, it, lo);
690
691         next = lo->ldo_stripe[it->lit_stripe_index];
692         LASSERT(next != NULL);
693         LASSERT(next->do_index_ops != NULL);
694
695         return next->do_index_ops->dio_it.key_size(env, it->lit_it);
696 }
697
698 /**
699  * Implementation of dt_it_ops::rec.
700  *
701  * Used with striped objects.
702  *
703  * \see dt_it_ops::rec() in the API description for details.
704  */
705 static int lod_striped_it_rec(const struct lu_env *env, const struct dt_it *di,
706                               struct dt_rec *rec, __u32 attr)
707 {
708         const struct lod_it     *it = (const struct lod_it *)di;
709         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
710         struct dt_object        *next;
711
712         LOD_CHECK_STRIPED_IT(env, it, lo);
713
714         next = lo->ldo_stripe[it->lit_stripe_index];
715         LASSERT(next != NULL);
716         LASSERT(next->do_index_ops != NULL);
717
718         return next->do_index_ops->dio_it.rec(env, it->lit_it, rec, attr);
719 }
720
721 /**
722  * Implementation of dt_it_ops::rec_size.
723  *
724  * Used with striped objects.
725  *
726  * \see dt_it_ops::rec_size() in the API description for details.
727  */
728 static int lod_striped_it_rec_size(const struct lu_env *env,
729                                    const struct dt_it *di, __u32 attr)
730 {
731         struct lod_it           *it = (struct lod_it *)di;
732         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
733         struct dt_object        *next;
734
735         LOD_CHECK_STRIPED_IT(env, it, lo);
736
737         next = lo->ldo_stripe[it->lit_stripe_index];
738         LASSERT(next != NULL);
739         LASSERT(next->do_index_ops != NULL);
740
741         return next->do_index_ops->dio_it.rec_size(env, it->lit_it, attr);
742 }
743
744 /**
745  * Implementation of dt_it_ops::store.
746  *
747  * Used with striped objects.
748  *
749  * \see dt_it_ops::store() in the API description for details.
750  */
751 static __u64 lod_striped_it_store(const struct lu_env *env,
752                                   const struct dt_it *di)
753 {
754         const struct lod_it     *it = (const struct lod_it *)di;
755         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
756         struct dt_object        *next;
757
758         LOD_CHECK_STRIPED_IT(env, it, lo);
759
760         next = lo->ldo_stripe[it->lit_stripe_index];
761         LASSERT(next != NULL);
762         LASSERT(next->do_index_ops != NULL);
763
764         return next->do_index_ops->dio_it.store(env, it->lit_it);
765 }
766
767 /**
768  * Implementation of dt_it_ops::load.
769  *
770  * Used with striped objects.
771  *
772  * \see dt_it_ops::load() in the API description for details.
773  */
774 static int lod_striped_it_load(const struct lu_env *env,
775                                const struct dt_it *di, __u64 hash)
776 {
777         const struct lod_it     *it = (const struct lod_it *)di;
778         struct lod_object       *lo = lod_dt_obj(it->lit_obj);
779         struct dt_object        *next;
780
781         LOD_CHECK_STRIPED_IT(env, it, lo);
782
783         next = lo->ldo_stripe[it->lit_stripe_index];
784         LASSERT(next != NULL);
785         LASSERT(next->do_index_ops != NULL);
786
787         return next->do_index_ops->dio_it.load(env, it->lit_it, hash);
788 }
789
790 static const struct dt_index_operations lod_striped_index_ops = {
791         .dio_lookup             = lod_striped_lookup,
792         .dio_declare_insert     = lod_declare_insert,
793         .dio_insert             = lod_insert,
794         .dio_declare_delete     = lod_declare_delete,
795         .dio_delete             = lod_delete,
796         .dio_it = {
797                 .init           = lod_striped_it_init,
798                 .fini           = lod_striped_it_fini,
799                 .get            = lod_striped_it_get,
800                 .put            = lod_striped_it_put,
801                 .next           = lod_striped_it_next,
802                 .key            = lod_striped_it_key,
803                 .key_size       = lod_striped_it_key_size,
804                 .rec            = lod_striped_it_rec,
805                 .rec_size       = lod_striped_it_rec_size,
806                 .store          = lod_striped_it_store,
807                 .load           = lod_striped_it_load,
808         }
809 };
810
811 /**
812  * Append the FID for each shard of the striped directory after the
813  * given LMV EA header.
814  *
815  * To simplify striped directory and the consistency verification,
816  * we only store the LMV EA header on disk, for both master object
817  * and slave objects. When someone wants to know the whole LMV EA,
818  * such as client readdir(), we can build the entrie LMV EA on the
819  * MDT side (in RAM) via iterating the sub-directory entries that
820  * are contained in the master object of the stripe directory.
821  *
822  * For the master object of the striped directroy, the valid name
823  * for each shard is composed of the ${shard_FID}:${shard_idx}.
824  *
825  * There may be holes in the LMV EA if some shards' name entries
826  * are corrupted or lost.
827  *
828  * \param[in] env       pointer to the thread context
829  * \param[in] lo        pointer to the master object of the striped directory
830  * \param[in] buf       pointer to the lu_buf which will hold the LMV EA
831  * \param[in] resize    whether re-allocate the buffer if it is not big enough
832  *
833  * \retval              positive size of the LMV EA
834  * \retval              0 for nothing to be loaded
835  * \retval              negative error number on failure
836  */
837 int lod_load_lmv_shards(const struct lu_env *env, struct lod_object *lo,
838                         struct lu_buf *buf, bool resize)
839 {
840         struct lu_dirent        *ent    =
841                         (struct lu_dirent *)lod_env_info(env)->lti_key;
842         struct lod_device       *lod    = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
843         struct dt_object        *obj    = dt_object_child(&lo->ldo_obj);
844         struct lmv_mds_md_v1    *lmv1   = buf->lb_buf;
845         struct dt_it            *it;
846         const struct dt_it_ops  *iops;
847         __u32                    stripes;
848         __u32                    magic  = le32_to_cpu(lmv1->lmv_magic);
849         size_t                   lmv1_size;
850         int                      rc;
851         ENTRY;
852
853         if (magic != LMV_MAGIC_V1)
854                 RETURN(0);
855
856         stripes = le32_to_cpu(lmv1->lmv_stripe_count);
857         if (stripes < 1)
858                 RETURN(0);
859
860         rc = lmv_mds_md_size(stripes, magic);
861         if (rc < 0)
862                 RETURN(rc);
863         lmv1_size = rc;
864         if (buf->lb_len < lmv1_size) {
865                 struct lu_buf tbuf;
866
867                 if (!resize)
868                         RETURN(-ERANGE);
869
870                 tbuf = *buf;
871                 buf->lb_buf = NULL;
872                 buf->lb_len = 0;
873                 lu_buf_alloc(buf, lmv1_size);
874                 lmv1 = buf->lb_buf;
875                 if (lmv1 == NULL)
876                         RETURN(-ENOMEM);
877
878                 memcpy(buf->lb_buf, tbuf.lb_buf, tbuf.lb_len);
879         }
880
881         if (unlikely(!dt_try_as_dir(env, obj, true)))
882                 RETURN(-ENOTDIR);
883
884         memset(&lmv1->lmv_stripe_fids[0], 0, stripes * sizeof(struct lu_fid));
885         iops = &obj->do_index_ops->dio_it;
886         it = iops->init(env, obj, LUDA_64BITHASH);
887         if (IS_ERR(it))
888                 RETURN(PTR_ERR(it));
889
890         rc = iops->load(env, it, 0);
891         if (rc == 0)
892                 rc = iops->next(env, it);
893         else if (rc > 0)
894                 rc = 0;
895
896         while (rc == 0) {
897                 char             name[FID_LEN + 2] = "";
898                 struct lu_fid    fid;
899                 __u32            index;
900                 int              len;
901
902                 rc = iops->rec(env, it, (struct dt_rec *)ent, LUDA_64BITHASH);
903                 if (rc != 0)
904                         break;
905
906                 rc = -EIO;
907
908                 fid_le_to_cpu(&fid, &ent->lde_fid);
909                 ent->lde_namelen = le16_to_cpu(ent->lde_namelen);
910                 if (ent->lde_name[0] == '.') {
911                         if (ent->lde_namelen == 1)
912                                 goto next;
913
914                         if (ent->lde_namelen == 2 && ent->lde_name[1] == '.')
915                                 goto next;
916                 }
917
918                 len = scnprintf(name, sizeof(name),
919                                 DFID":", PFID(&ent->lde_fid));
920                 /* The ent->lde_name is composed of ${FID}:${index} */
921                 if (ent->lde_namelen < len + 1 ||
922                     memcmp(ent->lde_name, name, len) != 0) {
923                         CDEBUG_LIMIT(lod->lod_lmv_failout ? D_ERROR : D_INFO,
924                                      "%s: invalid shard name %.*s with the FID "DFID" for the striped directory "DFID", %s\n",
925                                      lod2obd(lod)->obd_name, ent->lde_namelen,
926                                      ent->lde_name, PFID(&fid),
927                                      PFID(lu_object_fid(&obj->do_lu)),
928                                      lod->lod_lmv_failout ? "failout" : "skip");
929
930                         if (lod->lod_lmv_failout)
931                                 break;
932
933                         goto next;
934                 }
935
936                 index = 0;
937                 do {
938                         if (ent->lde_name[len] < '0' ||
939                             ent->lde_name[len] > '9') {
940                                 CDEBUG_LIMIT(lod->lod_lmv_failout ?
941                                              D_ERROR : D_INFO,
942                                              "%s: invalid shard name %.*s with the FID "DFID" for the striped directory "DFID", %s\n",
943                                              lod2obd(lod)->obd_name,
944                                              ent->lde_namelen,
945                                              ent->lde_name, PFID(&fid),
946                                              PFID(lu_object_fid(&obj->do_lu)),
947                                              lod->lod_lmv_failout ?
948                                              "failout" : "skip");
949
950                                 if (lod->lod_lmv_failout)
951                                         break;
952
953                                 goto next;
954                         }
955
956                         index = index * 10 + ent->lde_name[len++] - '0';
957                 } while (len < ent->lde_namelen);
958
959                 if (len == ent->lde_namelen) {
960                         /* Out of LMV EA range. */
961                         if (index >= stripes) {
962                                 CERROR("%s: the shard %.*s for the striped "
963                                        "directory "DFID" is out of the known "
964                                        "LMV EA range [0 - %u], failout\n",
965                                        lod2obd(lod)->obd_name, ent->lde_namelen,
966                                        ent->lde_name,
967                                        PFID(lu_object_fid(&obj->do_lu)),
968                                        stripes - 1);
969
970                                 break;
971                         }
972
973                         /* The slot has been occupied. */
974                         if (!fid_is_zero(&lmv1->lmv_stripe_fids[index])) {
975                                 struct lu_fid fid0;
976
977                                 fid_le_to_cpu(&fid0,
978                                         &lmv1->lmv_stripe_fids[index]);
979                                 CERROR("%s: both the shard "DFID" and "DFID
980                                        " for the striped directory "DFID
981                                        " claim the same LMV EA slot at the "
982                                        "index %d, failout\n",
983                                        lod2obd(lod)->obd_name,
984                                        PFID(&fid0), PFID(&fid),
985                                        PFID(lu_object_fid(&obj->do_lu)), index);
986
987                                 break;
988                         }
989
990                         /* stored as LE mode */
991                         lmv1->lmv_stripe_fids[index] = ent->lde_fid;
992
993 next:
994                         rc = iops->next(env, it);
995                 }
996         }
997
998         iops->put(env, it);
999         iops->fini(env, it);
1000
1001         RETURN(rc > 0 ? lmv_mds_md_size(stripes, magic) : rc);
1002 }
1003
1004 /**
1005  * Implementation of dt_object_operations::do_index_try.
1006  *
1007  * \see dt_object_operations::do_index_try() in the API description for details.
1008  */
1009 static int lod_index_try(const struct lu_env *env, struct dt_object *dt,
1010                          const struct dt_index_features *feat)
1011 {
1012         struct lod_object       *lo = lod_dt_obj(dt);
1013         struct dt_object        *next = dt_object_child(dt);
1014         int                     rc;
1015         ENTRY;
1016
1017         LASSERT(next->do_ops);
1018         LASSERT(next->do_ops->do_index_try);
1019
1020         rc = lod_striping_load(env, lo);
1021         if (rc != 0)
1022                 RETURN(rc);
1023
1024         rc = next->do_ops->do_index_try(env, next, feat);
1025         if (rc != 0)
1026                 RETURN(rc);
1027
1028         if (lo->ldo_dir_stripe_count > 0) {
1029                 int i;
1030
1031                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
1032                         if (!lo->ldo_stripe[i])
1033                                 continue;
1034                         if (!dt_object_exists(lo->ldo_stripe[i]))
1035                                 continue;
1036                         rc = lo->ldo_stripe[i]->do_ops->do_index_try(env,
1037                                                 lo->ldo_stripe[i], feat);
1038                         if (rc != 0)
1039                                 RETURN(rc);
1040                 }
1041                 dt->do_index_ops = &lod_striped_index_ops;
1042         } else {
1043                 dt->do_index_ops = &lod_index_ops;
1044         }
1045
1046         RETURN(rc);
1047 }
1048
1049 /**
1050  * Implementation of dt_object_operations::do_read_lock.
1051  *
1052  * \see dt_object_operations::do_read_lock() in the API description for details.
1053  */
1054 static void lod_read_lock(const struct lu_env *env, struct dt_object *dt,
1055                           unsigned role)
1056 {
1057         dt_read_lock(env, dt_object_child(dt), role);
1058 }
1059
1060 /**
1061  * Implementation of dt_object_operations::do_write_lock.
1062  *
1063  * \see dt_object_operations::do_write_lock() in the API description for
1064  * details.
1065  */
1066 static void lod_write_lock(const struct lu_env *env, struct dt_object *dt,
1067                            unsigned role)
1068 {
1069         dt_write_lock(env, dt_object_child(dt), role);
1070 }
1071
1072 /**
1073  * Implementation of dt_object_operations::do_read_unlock.
1074  *
1075  * \see dt_object_operations::do_read_unlock() in the API description for
1076  * details.
1077  */
1078 static void lod_read_unlock(const struct lu_env *env, struct dt_object *dt)
1079 {
1080         dt_read_unlock(env, dt_object_child(dt));
1081 }
1082
1083 /**
1084  * Implementation of dt_object_operations::do_write_unlock.
1085  *
1086  * \see dt_object_operations::do_write_unlock() in the API description for
1087  * details.
1088  */
1089 static void lod_write_unlock(const struct lu_env *env, struct dt_object *dt)
1090 {
1091         dt_write_unlock(env, dt_object_child(dt));
1092 }
1093
1094 /**
1095  * Implementation of dt_object_operations::do_write_locked.
1096  *
1097  * \see dt_object_operations::do_write_locked() in the API description for
1098  * details.
1099  */
1100 static int lod_write_locked(const struct lu_env *env, struct dt_object *dt)
1101 {
1102         return dt_write_locked(env, dt_object_child(dt));
1103 }
1104
1105 /**
1106  * Implementation of dt_object_operations::do_attr_get.
1107  *
1108  * \see dt_object_operations::do_attr_get() in the API description for details.
1109  */
1110 static int lod_attr_get(const struct lu_env *env,
1111                         struct dt_object *dt,
1112                         struct lu_attr *attr)
1113 {
1114         /* Note: for striped directory, client will merge attributes
1115          * from all of the sub-stripes see lmv_merge_attr(), and there
1116          * no MDD logic depend on directory nlink/size/time, so we can
1117          * always use master inode nlink and size for now. */
1118         return dt_attr_get(env, dt_object_child(dt), attr);
1119 }
1120
1121 void lod_adjust_stripe_size(struct lod_layout_component *comp,
1122                             __u32 def_stripe_size)
1123 {
1124         __u64 comp_end = comp->llc_extent.e_end;
1125
1126         /* Choose stripe size if not set. Note that default stripe size can't
1127          * be used as is, because it must be multiplier of given component end.
1128          *  - first check if default stripe size can be used
1129          *  - if not than select the lowest set bit from component end and use
1130          *    that value as stripe size
1131          */
1132         if (!comp->llc_stripe_size) {
1133                 if (comp_end == LUSTRE_EOF || !(comp_end % def_stripe_size))
1134                         comp->llc_stripe_size = def_stripe_size;
1135                 else
1136                         comp->llc_stripe_size = comp_end & ~(comp_end - 1);
1137         } else {
1138                 if (comp_end != LUSTRE_EOF &&
1139                     comp_end & (LOV_MIN_STRIPE_SIZE - 1)) {
1140                         CWARN("Component end %llu is not a multiple of min size %u\n",
1141                               comp_end, LOV_MIN_STRIPE_SIZE);
1142                         comp_end = round_up(comp_end, LOV_MIN_STRIPE_SIZE);
1143                 }
1144                 /* check stripe size is multiplier of comp_end */
1145                 if (comp_end != LUSTRE_EOF &&
1146                     comp_end != comp->llc_extent.e_start &&
1147                     comp_end % comp->llc_stripe_size) {
1148                         /* fix that even for defined stripe size but warn
1149                          * about the problem, that must not happen
1150                          */
1151                         CWARN("Component end %llu is not aligned by the stripe size %u\n",
1152                               comp_end, comp->llc_stripe_size);
1153                         comp->llc_stripe_size = comp_end & ~(comp_end - 1);
1154                 }
1155         }
1156 }
1157
1158 static inline void lod_adjust_stripe_info(struct lod_layout_component *comp,
1159                                           struct lov_desc *desc,
1160                                           int append_stripes)
1161 {
1162         if (!(comp->llc_pattern & LOV_PATTERN_MDT)) {
1163                 if (append_stripes) {
1164                         comp->llc_stripe_count = append_stripes;
1165                 } else if (!comp->llc_stripe_count) {
1166                         comp->llc_stripe_count =
1167                                 desc->ld_default_stripe_count;
1168                 }
1169         }
1170
1171         lod_adjust_stripe_size(comp, desc->ld_default_stripe_size);
1172 }
1173
1174 int lod_obj_for_each_stripe(const struct lu_env *env, struct lod_object *lo,
1175                             struct thandle *th,
1176                             struct lod_obj_stripe_cb_data *data)
1177 {
1178         struct lod_layout_component *lod_comp;
1179         int i, j, rc = 0;
1180         ENTRY;
1181
1182         mutex_lock(&lo->ldo_layout_mutex);
1183         for (i = 0; i < lo->ldo_comp_cnt; i++) {
1184                 lod_comp = &lo->ldo_comp_entries[i];
1185
1186                 if (lod_comp->llc_stripe == NULL)
1187                         continue;
1188
1189                 /* has stripe but not inited yet, this component has been
1190                  * declared to be created, but hasn't created yet.
1191                  */
1192                 if (!lod_comp_inited(lod_comp))
1193                         continue;
1194
1195                 if (data->locd_comp_skip_cb &&
1196                     data->locd_comp_skip_cb(env, lo, i, data))
1197                         continue;
1198
1199                 if (data->locd_comp_cb) {
1200                         rc = data->locd_comp_cb(env, lo, i, data);
1201                         if (rc)
1202                                 GOTO(unlock, rc);
1203                 }
1204
1205                 /* could used just to do sth about component, not each
1206                  * stripes
1207                  */
1208                 if (!data->locd_stripe_cb)
1209                         continue;
1210
1211                 LASSERT(lod_comp->llc_stripe_count > 0);
1212                 for (j = 0; j < lod_comp->llc_stripe_count; j++) {
1213                         struct dt_object *dt = lod_comp->llc_stripe[j];
1214
1215                         if (dt == NULL)
1216                                 continue;
1217                         rc = data->locd_stripe_cb(env, lo, dt, th, i, j, data);
1218                         if (rc != 0)
1219                                 GOTO(unlock, rc);
1220                 }
1221         }
1222 unlock:
1223         mutex_unlock(&lo->ldo_layout_mutex);
1224         RETURN(rc);
1225 }
1226
1227 static inline int
1228 lod_obj_stripe_attr_set_cb(const struct lu_env *env, struct lod_object *lo,
1229                            struct dt_object *dt, struct thandle *th,
1230                            int comp_idx, int stripe_idx,
1231                            struct lod_obj_stripe_cb_data *data)
1232 {
1233         if (data->locd_declare)
1234                 return lod_sub_declare_attr_set(env, dt, data->locd_attr, th);
1235
1236         if (data->locd_attr->la_valid & LA_LAYOUT_VERSION) {
1237                 CDEBUG(D_LAYOUT, DFID": set layout version: %u, comp_idx: %d\n",
1238                        PFID(lu_object_fid(&dt->do_lu)),
1239                        data->locd_attr->la_layout_version, comp_idx);
1240         }
1241
1242         return lod_sub_attr_set(env, dt, data->locd_attr, th);
1243 }
1244
1245 /**
1246  * Implementation of dt_object_operations::do_declare_attr_set.
1247  *
1248  * If the object is striped, then apply the changes to all the stripes.
1249  *
1250  * \see dt_object_operations::do_declare_attr_set() in the API description
1251  * for details.
1252  */
1253 static int lod_declare_attr_set(const struct lu_env *env,
1254                                 struct dt_object *dt,
1255                                 const struct lu_attr *attr,
1256                                 struct thandle *th)
1257 {
1258         struct dt_object  *next = dt_object_child(dt);
1259         struct lod_object *lo = lod_dt_obj(dt);
1260         int                rc, i;
1261         ENTRY;
1262
1263         /*
1264          * declare setattr on the local object
1265          */
1266         rc = lod_sub_declare_attr_set(env, next, attr, th);
1267         if (rc)
1268                 RETURN(rc);
1269
1270         /* osp_declare_attr_set() ignores all attributes other than
1271          * UID, GID, PROJID, and size, and osp_attr_set() ignores all
1272          * but UID, GID and PROJID. Declaration of size attr setting
1273          * happens through lod_declare_init_size(), and not through
1274          * this function. Therefore we need not load striping unless
1275          * ownership is changing.  This should save memory and (we hope)
1276          * speed up rename().
1277          */
1278         if (!S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
1279                 if (!(attr->la_valid & LA_REMOTE_ATTR_SET))
1280                         RETURN(rc);
1281
1282                 if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_OWNER))
1283                         RETURN(0);
1284         } else {
1285                 if (!(attr->la_valid & (LA_UID | LA_GID | LA_PROJID | LA_MODE |
1286                                         LA_ATIME | LA_MTIME | LA_CTIME |
1287                                         LA_FLAGS)))
1288                         RETURN(rc);
1289         }
1290         /*
1291          * load striping information, notice we don't do this when object
1292          * is being initialized as we don't need this information till
1293          * few specific cases like destroy, chown
1294          */
1295         rc = lod_striping_load(env, lo);
1296         if (rc)
1297                 RETURN(rc);
1298
1299         if (!lod_obj_is_striped(dt))
1300                 RETURN(0);
1301
1302         /*
1303          * if object is striped declare changes on the stripes
1304          */
1305         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
1306                 LASSERT(lo->ldo_stripe);
1307                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
1308                         if (lo->ldo_stripe[i] == NULL)
1309                                 continue;
1310                         if (!dt_object_exists(lo->ldo_stripe[i]))
1311                                 continue;
1312                         rc = lod_sub_declare_attr_set(env, lo->ldo_stripe[i],
1313                                                       attr, th);
1314                         if (rc != 0)
1315                                 RETURN(rc);
1316                 }
1317         } else {
1318                 struct lod_obj_stripe_cb_data data = { { 0 } };
1319
1320                 data.locd_attr = attr;
1321                 data.locd_declare = true;
1322                 data.locd_stripe_cb = lod_obj_stripe_attr_set_cb;
1323                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
1324         }
1325
1326         if (rc)
1327                 RETURN(rc);
1328
1329         if (!dt_object_exists(next) || dt_object_remote(next) ||
1330             !S_ISREG(attr->la_mode))
1331                 RETURN(0);
1332
1333         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_STRIPE)) {
1334                 rc = lod_sub_declare_xattr_del(env, next, XATTR_NAME_LOV, th);
1335                 RETURN(rc);
1336         }
1337
1338         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_CHANGE_STRIPE) ||
1339             CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_PFL_RANGE)) {
1340                 struct lod_thread_info *info = lod_env_info(env);
1341                 struct lu_buf *buf = &info->lti_buf;
1342
1343                 buf->lb_buf = info->lti_ea_store;
1344                 buf->lb_len = info->lti_ea_store_size;
1345                 rc = lod_sub_declare_xattr_set(env, next, buf, XATTR_NAME_LOV,
1346                                                LU_XATTR_REPLACE, th);
1347         }
1348
1349         RETURN(rc);
1350 }
1351
1352 /**
1353  * Implementation of dt_object_operations::do_attr_set.
1354  *
1355  * If the object is striped, then apply the changes to all or subset of
1356  * the stripes depending on the object type and specific attributes.
1357  *
1358  * \see dt_object_operations::do_attr_set() in the API description for details.
1359  */
1360 static int lod_attr_set(const struct lu_env *env,
1361                         struct dt_object *dt,
1362                         const struct lu_attr *attr,
1363                         struct thandle *th)
1364 {
1365         struct dt_object        *next = dt_object_child(dt);
1366         struct lod_object       *lo = lod_dt_obj(dt);
1367         int                     rc, i;
1368         ENTRY;
1369
1370         /*
1371          * apply changes to the local object
1372          */
1373         rc = lod_sub_attr_set(env, next, attr, th);
1374         if (rc)
1375                 RETURN(rc);
1376
1377         if (!S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
1378                 if (!(attr->la_valid & LA_REMOTE_ATTR_SET))
1379                         RETURN(rc);
1380
1381                 if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_OWNER))
1382                         RETURN(0);
1383         } else {
1384                 if (!(attr->la_valid & (LA_UID | LA_GID | LA_MODE | LA_PROJID |
1385                                         LA_ATIME | LA_MTIME | LA_CTIME |
1386                                         LA_FLAGS)))
1387                         RETURN(rc);
1388         }
1389
1390         /* FIXME: a tricky case in the code path of mdd_layout_change():
1391          * the in-memory striping information has been freed in lod_xattr_set()
1392          * due to layout change. It has to load stripe here again. It only
1393          * changes flags of layout so declare_attr_set() is still accurate */
1394         rc = lod_striping_load(env, lo);
1395         if (rc)
1396                 RETURN(rc);
1397
1398         if (!lod_obj_is_striped(dt))
1399                 RETURN(0);
1400
1401         /*
1402          * if object is striped, apply changes to all the stripes
1403          */
1404         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
1405                 LASSERT(lo->ldo_stripe);
1406                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
1407                         if (unlikely(lo->ldo_stripe[i] == NULL))
1408                                 continue;
1409
1410                         if ((dt_object_exists(lo->ldo_stripe[i]) == 0))
1411                                 continue;
1412
1413                         rc = lod_sub_attr_set(env, lo->ldo_stripe[i], attr, th);
1414                         if (rc != 0)
1415                                 break;
1416                 }
1417         } else {
1418                 struct lod_obj_stripe_cb_data data = { { 0 } };
1419
1420                 data.locd_attr = attr;
1421                 data.locd_declare = false;
1422                 data.locd_stripe_cb = lod_obj_stripe_attr_set_cb;
1423                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
1424         }
1425
1426         if (rc)
1427                 RETURN(rc);
1428
1429         if (!dt_object_exists(next) || dt_object_remote(next) ||
1430             !S_ISREG(attr->la_mode))
1431                 RETURN(0);
1432
1433         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_STRIPE)) {
1434                 rc = lod_sub_xattr_del(env, next, XATTR_NAME_LOV, th);
1435                 RETURN(rc);
1436         }
1437
1438         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_CHANGE_STRIPE)) {
1439                 struct lod_thread_info *info = lod_env_info(env);
1440                 struct lu_buf *buf = &info->lti_buf;
1441                 struct ost_id *oi = &info->lti_ostid;
1442                 struct lu_fid *fid = &info->lti_fid;
1443                 struct lov_mds_md_v1 *lmm;
1444                 struct lov_ost_data_v1 *objs;
1445                 __u32 magic;
1446
1447                 rc = lod_get_lov_ea(env, lo);
1448                 if (rc <= 0)
1449                         RETURN(rc);
1450
1451                 buf->lb_buf = info->lti_ea_store;
1452                 buf->lb_len = info->lti_ea_store_size;
1453                 lmm = info->lti_ea_store;
1454                 magic = le32_to_cpu(lmm->lmm_magic);
1455                 if (magic == LOV_MAGIC_COMP_V1 || magic == LOV_MAGIC_SEL) {
1456                         struct lov_comp_md_v1 *lcm = buf->lb_buf;
1457                         struct lov_comp_md_entry_v1 *lcme =
1458                                                 &lcm->lcm_entries[0];
1459
1460                         lmm = buf->lb_buf + le32_to_cpu(lcme->lcme_offset);
1461                         magic = le32_to_cpu(lmm->lmm_magic);
1462                 }
1463
1464                 if (magic == LOV_MAGIC_V1)
1465                         objs = &(lmm->lmm_objects[0]);
1466                 else
1467                         objs = &((struct lov_mds_md_v3 *)lmm)->lmm_objects[0];
1468                 ostid_le_to_cpu(&objs->l_ost_oi, oi);
1469                 ostid_to_fid(fid, oi, le32_to_cpu(objs->l_ost_idx));
1470                 fid->f_oid--;
1471                 fid_to_ostid(fid, oi);
1472                 ostid_cpu_to_le(oi, &objs->l_ost_oi);
1473
1474                 rc = lod_sub_xattr_set(env, next, buf, XATTR_NAME_LOV,
1475                                        LU_XATTR_REPLACE, th);
1476         } else if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_PFL_RANGE)) {
1477                 struct lod_thread_info *info = lod_env_info(env);
1478                 struct lu_buf *buf = &info->lti_buf;
1479                 struct lov_comp_md_v1 *lcm;
1480                 struct lov_comp_md_entry_v1 *lcme;
1481
1482                 rc = lod_get_lov_ea(env, lo);
1483                 if (rc <= 0)
1484                         RETURN(rc);
1485
1486                 buf->lb_buf = info->lti_ea_store;
1487                 buf->lb_len = info->lti_ea_store_size;
1488                 lcm = buf->lb_buf;
1489                 if (le32_to_cpu(lcm->lcm_magic) != LOV_MAGIC_COMP_V1 &&
1490                     le32_to_cpu(lcm->lcm_magic) != LOV_MAGIC_SEL)
1491                         RETURN(-EINVAL);
1492
1493                 le32_add_cpu(&lcm->lcm_layout_gen, 1);
1494                 lcme = &lcm->lcm_entries[0];
1495                 le64_add_cpu(&lcme->lcme_extent.e_start, 1);
1496                 le64_add_cpu(&lcme->lcme_extent.e_end, -1);
1497
1498                 rc = lod_sub_xattr_set(env, next, buf, XATTR_NAME_LOV,
1499                                        LU_XATTR_REPLACE, th);
1500         }
1501
1502         RETURN(rc);
1503 }
1504
1505 /**
1506  * Implementation of dt_object_operations::do_xattr_get.
1507  *
1508  * If LOV EA is requested from the root object and it's not
1509  * found, then return default striping for the filesystem.
1510  *
1511  * \see dt_object_operations::do_xattr_get() in the API description for details.
1512  */
1513 static int lod_xattr_get(const struct lu_env *env, struct dt_object *dt,
1514                          struct lu_buf *buf, const char *name)
1515 {
1516         struct lod_thread_info *info = lod_env_info(env);
1517         struct lod_device *dev = lu2lod_dev(dt->do_lu.lo_dev);
1518         int is_root;
1519         int rc;
1520         ENTRY;
1521
1522         rc = dt_xattr_get(env, dt_object_child(dt), buf, name);
1523         if (strcmp(name, XATTR_NAME_LMV) == 0) {
1524                 struct lmv_mds_md_v1    *lmv1;
1525                 struct lmv_foreign_md   *lfm;
1526                 int                      rc1 = 0;
1527
1528                 if (rc > (typeof(rc))sizeof(*lmv1))
1529                         RETURN(rc);
1530
1531                 /* short (<= sizeof(struct lmv_mds_md_v1)) foreign LMV case */
1532                 /* XXX empty foreign LMV is not allowed */
1533                 if (rc <= offsetof(typeof(*lfm), lfm_value))
1534                         RETURN(rc = rc > 0 ? -EINVAL : rc);
1535
1536                 if (buf->lb_buf == NULL || buf->lb_len == 0) {
1537                         BUILD_BUG_ON(sizeof(*lmv1) > sizeof(info->lti_key));
1538
1539                         /* lti_buf is large enough for *lmv1 or a short
1540                          * (<= sizeof(struct lmv_mds_md_v1)) foreign LMV
1541                          */
1542                         info->lti_buf.lb_buf = info->lti_key;
1543                         info->lti_buf.lb_len = sizeof(*lmv1);
1544                         rc = dt_xattr_get(env, dt_object_child(dt),
1545                                           &info->lti_buf, name);
1546                         if (unlikely(rc <= offsetof(typeof(*lfm),
1547                                                     lfm_value)))
1548                                 RETURN(rc = rc > 0 ? -EINVAL : rc);
1549
1550                         lfm = info->lti_buf.lb_buf;
1551                         if (le32_to_cpu(lfm->lfm_magic) == LMV_MAGIC_FOREIGN)
1552                                 RETURN(rc);
1553
1554                         if (unlikely(rc != sizeof(*lmv1)))
1555                                 RETURN(rc = rc > 0 ? -EINVAL : rc);
1556
1557                         lmv1 = info->lti_buf.lb_buf;
1558                         /* The on-disk LMV EA only contains header, but the
1559                          * returned LMV EA size should contain the space for
1560                          * the FIDs of all shards of the striped directory. */
1561                         if (le32_to_cpu(lmv1->lmv_magic) == LMV_MAGIC_V1)
1562                                 rc = lmv_mds_md_size(
1563                                         le32_to_cpu(lmv1->lmv_stripe_count),
1564                                         le32_to_cpu(lmv1->lmv_magic));
1565                 } else {
1566                         lmv1 = buf->lb_buf;
1567                         if (le32_to_cpu(lmv1->lmv_magic) != LMV_MAGIC_V1)
1568                                 RETURN(rc);
1569
1570                         if (rc != sizeof(*lmv1))
1571                                 RETURN(rc = rc > 0 ? -EINVAL : rc);
1572
1573                         rc1 = lod_load_lmv_shards(env, lod_dt_obj(dt),
1574                                                   buf, false);
1575                 }
1576
1577                 RETURN(rc = rc1 != 0 ? rc1 : rc);
1578         }
1579
1580         if ((rc > 0) && buf->lb_buf && strcmp(name, XATTR_NAME_LOV) == 0) {
1581                 struct lov_comp_md_v1 *lcm = buf->lb_buf;
1582
1583                 if (lcm->lcm_magic == cpu_to_le32(LOV_MAGIC_SEL))
1584                         lcm->lcm_magic = cpu_to_le32(LOV_MAGIC_COMP_V1);
1585         }
1586
1587         if (rc != -ENODATA || !S_ISDIR(dt->do_lu.lo_header->loh_attr & S_IFMT))
1588                 RETURN(rc);
1589
1590         /*
1591          * XXX: Only used by lfsck
1592          *
1593          * lod returns default striping on the real root of the device
1594          * this is like the root stores default striping for the whole
1595          * filesystem. historically we've been using a different approach
1596          * and store it in the config.
1597          */
1598         dt_root_get(env, dev->lod_child, &info->lti_fid);
1599         is_root = lu_fid_eq(&info->lti_fid, lu_object_fid(&dt->do_lu));
1600
1601         if (is_root && strcmp(XATTR_NAME_LOV, name) == 0) {
1602                 struct lov_user_md *lum = buf->lb_buf;
1603                 struct lov_desc *desc = &dev->lod_ost_descs.ltd_lov_desc;
1604
1605                 if (buf->lb_buf == NULL) {
1606                         rc = sizeof(*lum);
1607                 } else if (buf->lb_len >= sizeof(*lum)) {
1608                         lum->lmm_magic = cpu_to_le32(LOV_USER_MAGIC_V1);
1609                         lmm_oi_set_seq(&lum->lmm_oi, FID_SEQ_LOV_DEFAULT);
1610                         lmm_oi_set_id(&lum->lmm_oi, 0);
1611                         lmm_oi_cpu_to_le(&lum->lmm_oi, &lum->lmm_oi);
1612                         lum->lmm_pattern = cpu_to_le32(desc->ld_pattern);
1613                         lum->lmm_stripe_size = cpu_to_le32(
1614                                                 desc->ld_default_stripe_size);
1615                         lum->lmm_stripe_count = cpu_to_le16(
1616                                                 desc->ld_default_stripe_count);
1617                         lum->lmm_stripe_offset = cpu_to_le16(
1618                                                 desc->ld_default_stripe_offset);
1619                         rc = sizeof(*lum);
1620                 } else {
1621                         rc = -ERANGE;
1622                 }
1623         }
1624
1625         RETURN(rc);
1626 }
1627
1628 /**
1629  * Verify LVM EA.
1630  *
1631  * Checks that the magic of the stripe is sane.
1632  *
1633  * \param[in] lod       lod device
1634  * \param[in] lum       a buffer storing LMV EA to verify
1635  *
1636  * \retval              0 if the EA is sane
1637  * \retval              negative otherwise
1638  */
1639 static int lod_verify_md_striping(struct lod_device *lod,
1640                                   const struct lmv_user_md_v1 *lum)
1641 {
1642         if (unlikely(le32_to_cpu(lum->lum_magic) != LMV_USER_MAGIC)) {
1643                 CERROR("%s: invalid lmv_user_md: magic = %x, "
1644                        "stripe_offset = %d, stripe_count = %u: rc = %d\n",
1645                        lod2obd(lod)->obd_name, le32_to_cpu(lum->lum_magic),
1646                        (int)le32_to_cpu(lum->lum_stripe_offset),
1647                        le32_to_cpu(lum->lum_stripe_count), -EINVAL);
1648                 return -EINVAL;
1649         }
1650
1651         return 0;
1652 }
1653
1654 /**
1655  * Initialize LMV EA for a slave.
1656  *
1657  * Initialize slave's LMV EA from the master's LMV EA.
1658  *
1659  * \param[in] master_lmv        a buffer containing master's EA
1660  * \param[out] slave_lmv        a buffer where slave's EA will be stored
1661  *
1662  */
1663 static void lod_prep_slave_lmv_md(struct lmv_mds_md_v1 *slave_lmv,
1664                                   const struct lmv_mds_md_v1 *master_lmv)
1665 {
1666         *slave_lmv = *master_lmv;
1667         slave_lmv->lmv_magic = cpu_to_le32(LMV_MAGIC_STRIPE);
1668 }
1669
1670 /**
1671  * Generate LMV EA.
1672  *
1673  * Generate LMV EA from the object passed as \a dt. The object must have
1674  * the stripes created and initialized.
1675  *
1676  * \param[in] env       execution environment
1677  * \param[in] dt        object
1678  * \param[out] lmv_buf  buffer storing generated LMV EA
1679  *
1680  * \retval              0 on success
1681  * \retval              negative if failed
1682  */
1683 static int lod_prep_lmv_md(const struct lu_env *env, struct dt_object *dt,
1684                            struct lu_buf *lmv_buf)
1685 {
1686         struct lod_thread_info  *info = lod_env_info(env);
1687         struct lod_device       *lod = lu2lod_dev(dt->do_lu.lo_dev);
1688         struct lod_object       *lo = lod_dt_obj(dt);
1689         struct lmv_mds_md_v1    *lmm1;
1690         int                     stripe_count;
1691         int                     type = LU_SEQ_RANGE_ANY;
1692         int                     rc;
1693         __u32                   mdtidx;
1694         ENTRY;
1695
1696         LASSERT(lo->ldo_dir_striped != 0);
1697         LASSERT(lo->ldo_dir_stripe_count > 0);
1698         stripe_count = lo->ldo_dir_stripe_count;
1699         /* Only store the LMV EA heahder on the disk. */
1700         if (info->lti_ea_store_size < sizeof(*lmm1)) {
1701                 rc = lod_ea_store_resize(info, sizeof(*lmm1));
1702                 if (rc != 0)
1703                         RETURN(rc);
1704         } else {
1705                 memset(info->lti_ea_store, 0, sizeof(*lmm1));
1706         }
1707
1708         lmm1 = (struct lmv_mds_md_v1 *)info->lti_ea_store;
1709         memset(lmm1, 0, sizeof(*lmm1));
1710         lmm1->lmv_magic = cpu_to_le32(LMV_MAGIC);
1711         lmm1->lmv_stripe_count = cpu_to_le32(stripe_count);
1712         lmm1->lmv_hash_type = cpu_to_le32(lo->ldo_dir_hash_type);
1713         lmm1->lmv_layout_version = cpu_to_le32(lo->ldo_dir_layout_version);
1714         if (lod_is_layout_changing(lo)) {
1715                 lmm1->lmv_migrate_hash = cpu_to_le32(lo->ldo_dir_migrate_hash);
1716                 lmm1->lmv_migrate_offset =
1717                         cpu_to_le32(lo->ldo_dir_migrate_offset);
1718         }
1719         rc = lod_fld_lookup(env, lod, lu_object_fid(&dt->do_lu),
1720                             &mdtidx, &type);
1721         if (rc != 0)
1722                 RETURN(rc);
1723
1724         lmm1->lmv_master_mdt_index = cpu_to_le32(mdtidx);
1725         lmv_buf->lb_buf = info->lti_ea_store;
1726         lmv_buf->lb_len = sizeof(*lmm1);
1727
1728         RETURN(rc);
1729 }
1730
1731 /**
1732  * Create in-core represenation for a striped directory.
1733  *
1734  * Parse the buffer containing LMV EA and instantiate LU objects
1735  * representing the stripe objects. The pointers to the objects are
1736  * stored in ldo_stripe field of \a lo. This function is used when
1737  * we need to access an already created object (i.e. load from a disk).
1738  *
1739  * \param[in] env       execution environment
1740  * \param[in] lo        lod object
1741  * \param[in] buf       buffer containing LMV EA
1742  *
1743  * \retval              0 on success
1744  * \retval              negative if failed
1745  */
1746 int lod_parse_dir_striping(const struct lu_env *env, struct lod_object *lo,
1747                            const struct lu_buf *buf)
1748 {
1749         struct lod_thread_info  *info = lod_env_info(env);
1750         struct lod_device       *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
1751         struct lod_tgt_descs    *ltd = &lod->lod_mdt_descs;
1752         struct dt_object        **stripe;
1753         union lmv_mds_md        *lmm = buf->lb_buf;
1754         struct lmv_mds_md_v1    *lmv1 = &lmm->lmv_md_v1;
1755         struct lu_fid           *fid = &info->lti_fid;
1756         unsigned int            i;
1757         int                     rc = 0;
1758         ENTRY;
1759
1760         LASSERT(mutex_is_locked(&lo->ldo_layout_mutex));
1761
1762         /* XXX may be useless as not called for foreign LMV ?? */
1763         if (le32_to_cpu(lmv1->lmv_magic) == LMV_MAGIC_FOREIGN)
1764                 RETURN(0);
1765
1766         if (le32_to_cpu(lmv1->lmv_magic) == LMV_MAGIC_STRIPE) {
1767                 lo->ldo_dir_slave_stripe = 1;
1768                 RETURN(0);
1769         }
1770
1771         if (!lmv_is_sane(lmv1))
1772                 RETURN(-EINVAL);
1773
1774         LASSERT(lo->ldo_stripe == NULL);
1775         OBD_ALLOC_PTR_ARRAY(stripe, le32_to_cpu(lmv1->lmv_stripe_count));
1776         if (stripe == NULL)
1777                 RETURN(-ENOMEM);
1778
1779         for (i = 0; i < le32_to_cpu(lmv1->lmv_stripe_count); i++) {
1780                 struct dt_device        *tgt_dt;
1781                 struct dt_object        *dto;
1782                 int                     type = LU_SEQ_RANGE_ANY;
1783                 __u32                   idx;
1784
1785                 fid_le_to_cpu(fid, &lmv1->lmv_stripe_fids[i]);
1786                 if (!fid_is_sane(fid)) {
1787                         stripe[i] = NULL;
1788                         continue;
1789                 }
1790
1791                 rc = lod_fld_lookup(env, lod, fid, &idx, &type);
1792                 if (rc != 0)
1793                         GOTO(out, rc);
1794
1795                 if (idx == lod2lu_dev(lod)->ld_site->ld_seq_site->ss_node_id) {
1796                         tgt_dt = lod->lod_child;
1797                 } else {
1798                         struct lod_tgt_desc     *tgt;
1799
1800                         tgt = LTD_TGT(ltd, idx);
1801                         if (tgt == NULL)
1802                                 GOTO(out, rc = -ESTALE);
1803                         tgt_dt = tgt->ltd_tgt;
1804                 }
1805
1806                 dto = dt_locate_at(env, tgt_dt, fid,
1807                                   lo->ldo_obj.do_lu.lo_dev->ld_site->ls_top_dev,
1808                                   NULL);
1809                 if (IS_ERR(dto))
1810                         GOTO(out, rc = PTR_ERR(dto));
1811
1812                 stripe[i] = dto;
1813         }
1814 out:
1815         lo->ldo_stripe = stripe;
1816         lo->ldo_is_foreign = 0;
1817         lo->ldo_dir_stripe_count = le32_to_cpu(lmv1->lmv_stripe_count);
1818         lo->ldo_dir_stripes_allocated = le32_to_cpu(lmv1->lmv_stripe_count);
1819         lo->ldo_dir_layout_version = le32_to_cpu(lmv1->lmv_layout_version);
1820         lo->ldo_dir_migrate_offset = le32_to_cpu(lmv1->lmv_migrate_offset);
1821         lo->ldo_dir_migrate_hash = le32_to_cpu(lmv1->lmv_migrate_hash);
1822         lo->ldo_dir_hash_type = le32_to_cpu(lmv1->lmv_hash_type);
1823         if (rc != 0)
1824                 lod_striping_free_nolock(env, lo);
1825
1826         RETURN(rc);
1827 }
1828
1829 /**
1830  * Declare create a striped directory.
1831  *
1832  * Declare creating a striped directory with a given stripe pattern on the
1833  * specified MDTs. A striped directory is represented as a regular directory
1834  * - an index listing all the stripes. The stripes point back to the master
1835  * object with ".." and LinkEA. The master object gets LMV EA which
1836  * identifies it as a striped directory. The function allocates FIDs
1837  * for all stripes.
1838  *
1839  * \param[in] env       execution environment
1840  * \param[in] dt        object
1841  * \param[in] attr      attributes to initialize the objects with
1842  * \param[in] dof       type of objects to be created
1843  * \param[in] th        transaction handle
1844  *
1845  * \retval              0 on success
1846  * \retval              negative if failed
1847  */
1848 static int lod_dir_declare_create_stripes(const struct lu_env *env,
1849                                           struct dt_object *dt,
1850                                           struct lu_attr *attr,
1851                                           struct dt_object_format *dof,
1852                                           struct thandle *th)
1853 {
1854         struct lod_thread_info  *info = lod_env_info(env);
1855         struct lu_buf           lmv_buf;
1856         struct lu_buf           slave_lmv_buf;
1857         struct lmv_mds_md_v1    *lmm;
1858         struct lmv_mds_md_v1    *slave_lmm = NULL;
1859         struct dt_insert_rec    *rec = &info->lti_dt_rec;
1860         struct lod_object       *lo = lod_dt_obj(dt);
1861         int                     rc;
1862         __u32                   i;
1863         ENTRY;
1864
1865         rc = lod_prep_lmv_md(env, dt, &lmv_buf);
1866         if (rc != 0)
1867                 GOTO(out, rc);
1868         lmm = lmv_buf.lb_buf;
1869
1870         OBD_ALLOC_PTR(slave_lmm);
1871         if (slave_lmm == NULL)
1872                 GOTO(out, rc = -ENOMEM);
1873
1874         lod_prep_slave_lmv_md(slave_lmm, lmm);
1875         slave_lmv_buf.lb_buf = slave_lmm;
1876         slave_lmv_buf.lb_len = sizeof(*slave_lmm);
1877
1878         if (!dt_try_as_dir(env, dt_object_child(dt), false))
1879                 GOTO(out, rc = -EINVAL);
1880
1881         rec->rec_type = S_IFDIR;
1882         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
1883                 struct dt_object        *dto = lo->ldo_stripe[i];
1884                 char                    *stripe_name = info->lti_key;
1885                 struct lu_name          *sname;
1886                 struct linkea_data       ldata          = { NULL };
1887                 struct lu_buf           linkea_buf;
1888
1889                 /* OBD_FAIL_MDS_STRIPE_FID may leave stripe uninitialized */
1890                 if (!dto)
1891                         continue;
1892
1893                 /* directory split skip create for existing stripes */
1894                 if (!(lod_is_splitting(lo) && i < lo->ldo_dir_split_offset)) {
1895                         rc = lod_sub_declare_create(env, dto, attr, NULL, dof,
1896                                                     th);
1897                         if (rc != 0)
1898                                 GOTO(out, rc);
1899
1900                         if (!dt_try_as_dir(env, dto, false))
1901                                 GOTO(out, rc = -EINVAL);
1902
1903                         rc = lod_sub_declare_ref_add(env, dto, th);
1904                         if (rc != 0)
1905                                 GOTO(out, rc);
1906
1907                         rec->rec_fid = lu_object_fid(&dto->do_lu);
1908                         rc = lod_sub_declare_insert(env, dto,
1909                                                     (const struct dt_rec *)rec,
1910                                                     (const struct dt_key *)dot,
1911                                                     th);
1912                         if (rc != 0)
1913                                 GOTO(out, rc);
1914
1915                         /* master stripe FID will be put to .. */
1916                         rec->rec_fid = lu_object_fid(&dt->do_lu);
1917                         rc = lod_sub_declare_insert(env, dto,
1918                                                   (const struct dt_rec *)rec,
1919                                                   (const struct dt_key *)dotdot,
1920                                                   th);
1921                         if (rc != 0)
1922                                 GOTO(out, rc);
1923
1924                         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_SLAVE_NAME) &&
1925                             cfs_fail_val == i)
1926                                 snprintf(stripe_name, sizeof(info->lti_key),
1927                                          DFID":%u",
1928                                          PFID(lu_object_fid(&dto->do_lu)),
1929                                          i + 1);
1930                         else
1931                                 snprintf(stripe_name, sizeof(info->lti_key),
1932                                          DFID":%u",
1933                                          PFID(lu_object_fid(&dto->do_lu)), i);
1934
1935                         sname = lod_name_get(env, stripe_name,
1936                                              strlen(stripe_name));
1937                         rc = linkea_links_new(&ldata, &info->lti_linkea_buf,
1938                                               sname, lu_object_fid(&dt->do_lu));
1939                         if (rc != 0)
1940                                 GOTO(out, rc);
1941
1942                         linkea_buf.lb_buf = ldata.ld_buf->lb_buf;
1943                         linkea_buf.lb_len = ldata.ld_leh->leh_len;
1944                         rc = lod_sub_declare_xattr_set(env, dto, &linkea_buf,
1945                                                        XATTR_NAME_LINK, 0, th);
1946                         if (rc != 0)
1947                                 GOTO(out, rc);
1948
1949                         rec->rec_fid = lu_object_fid(&dto->do_lu);
1950                         rc = lod_sub_declare_insert(env, dt_object_child(dt),
1951                                         (const struct dt_rec *)rec,
1952                                         (const struct dt_key *)stripe_name, th);
1953                         if (rc != 0)
1954                                 GOTO(out, rc);
1955
1956                         rc = lod_sub_declare_ref_add(env, dt_object_child(dt),
1957                                                      th);
1958                         if (rc != 0)
1959                                 GOTO(out, rc);
1960                 }
1961
1962                 if (!CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_SLAVE_LMV) ||
1963                     cfs_fail_val != i) {
1964                         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_SLAVE_LMV) &&
1965                             cfs_fail_val == i)
1966                                 slave_lmm->lmv_master_mdt_index =
1967                                                         cpu_to_le32(i + 1);
1968                         else
1969                                 slave_lmm->lmv_master_mdt_index =
1970                                                         cpu_to_le32(i);
1971                         rc = lod_sub_declare_xattr_set(env, dto, &slave_lmv_buf,
1972                                                        XATTR_NAME_LMV, 0, th);
1973                         if (rc != 0)
1974                                 GOTO(out, rc);
1975                 }
1976         }
1977
1978         rc = lod_sub_declare_xattr_set(env, dt_object_child(dt),
1979                                        &lmv_buf, XATTR_NAME_LMV, 0, th);
1980         if (rc != 0)
1981                 GOTO(out, rc);
1982 out:
1983         if (slave_lmm != NULL)
1984                 OBD_FREE_PTR(slave_lmm);
1985
1986         RETURN(rc);
1987 }
1988
1989 /**
1990  * Allocate a striping on a predefined set of MDTs.
1991  *
1992  * Allocates new striping using the MDT index range provided by the data from
1993  * the lum_obejcts contained in the lmv_user_md passed to this method if
1994  * \a is_specific is true; or allocates new layout starting from MDT index in
1995  * lo->ldo_dir_stripe_offset. The exact order of MDTs is not important and
1996  * varies depending on MDT status. The number of stripes needed and stripe
1997  * offset are taken from the object. If that number cannot be met, then the
1998  * function returns an error and then it's the caller's responsibility to
1999  * release the stripes allocated. All the internal structures are protected,
2000  * but no concurrent allocation is allowed on the same objects.
2001  *
2002  * \param[in] env               execution environment for this thread
2003  * \param[in] lo                LOD object
2004  * \param[out] stripes          striping created
2005  * \param[out] mdt_indices      MDT indices of striping created
2006  * \param[in] is_specific       true if the MDTs are provided by lum; false if
2007  *                              only the starting MDT index is provided
2008  *
2009  * \retval positive     stripes allocated, including the first stripe allocated
2010  *                      outside
2011  * \retval negative     errno on failure
2012  */
2013 static int lod_mdt_alloc_specific(const struct lu_env *env,
2014                                   struct lod_object *lo,
2015                                   struct dt_object **stripes,
2016                                   __u32 *mdt_indices, bool is_specific)
2017 {
2018         struct lod_device *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
2019         struct lu_tgt_descs *ltd = &lod->lod_mdt_descs;
2020         struct lu_tgt_desc *tgt = NULL;
2021         struct lu_object_conf conf = { .loc_flags = LOC_F_NEW };
2022         struct dt_device *tgt_dt = NULL;
2023         struct lu_fid fid = { 0 };
2024         struct dt_object *dto;
2025         u32 master_index;
2026         u32 stripe_count = lo->ldo_dir_stripe_count;
2027         int stripe_idx = 1;
2028         int j;
2029         int idx;
2030         int rc;
2031
2032         master_index = lu_site2seq(lod2lu_dev(lod)->ld_site)->ss_node_id;
2033         if (!is_specific && stripe_count > 1)
2034                 /* Set the start index for the 2nd stripe allocation */
2035                 mdt_indices[1] = (mdt_indices[0] + 1) %
2036                                         (lod->lod_remote_mdt_count + 1);
2037
2038         for (; stripe_idx < stripe_count; stripe_idx++) {
2039                 /* Try to find next avaible target */
2040                 idx = mdt_indices[stripe_idx];
2041                 for (j = 0; j < lod->lod_remote_mdt_count;
2042                      j++, idx = (idx + 1) % (lod->lod_remote_mdt_count + 1)) {
2043                         bool already_allocated = false;
2044                         __u32 k;
2045
2046                         CDEBUG(D_INFO, "try idx %d, mdt cnt %u, allocated %u\n",
2047                                idx, lod->lod_remote_mdt_count + 1, stripe_idx);
2048
2049                         if (likely(!is_specific &&
2050                                    !CFS_FAIL_CHECK(OBD_FAIL_LARGE_STRIPE))) {
2051                                 /* check whether the idx already exists
2052                                  * in current allocated array */
2053                                 for (k = 0; k < stripe_idx; k++) {
2054                                         if (mdt_indices[k] == idx) {
2055                                                 already_allocated = true;
2056                                                 break;
2057                                         }
2058                                 }
2059
2060                                 if (already_allocated)
2061                                         continue;
2062                         }
2063
2064                         /* Sigh, this index is not in the bitmap, let's check
2065                          * next available target */
2066                         if (!test_bit(idx, ltd->ltd_tgt_bitmap) &&
2067                             idx != master_index)
2068                                 continue;
2069
2070                         if (idx == master_index) {
2071                                 /* Allocate the FID locally */
2072                                 tgt_dt = lod->lod_child;
2073                                 rc = dt_fid_alloc(env, tgt_dt, &fid, NULL,
2074                                                   NULL);
2075                                 if (rc < 0)
2076                                         continue;
2077                                 break;
2078                         }
2079
2080                         /* check the status of the OSP */
2081                         tgt = LTD_TGT(ltd, idx);
2082                         if (!tgt)
2083                                 continue;
2084
2085                         tgt_dt = tgt->ltd_tgt;
2086                         if (!tgt->ltd_active)
2087                                 /* this OSP doesn't feel well */
2088                                 continue;
2089
2090                         rc = dt_fid_alloc(env, tgt_dt, &fid, NULL, NULL);
2091                         if (rc < 0)
2092                                 continue;
2093
2094                         break;
2095                 }
2096
2097                 /* Can not allocate more stripes */
2098                 if (j == lod->lod_remote_mdt_count) {
2099                         CDEBUG(D_INFO, "%s: require stripes %u only get %d\n",
2100                                lod2obd(lod)->obd_name, stripe_count,
2101                                stripe_idx);
2102                         break;
2103                 }
2104
2105                 CDEBUG(D_INFO, "Get idx %d, for stripe %d "DFID"\n",
2106                        idx, stripe_idx, PFID(&fid));
2107                 mdt_indices[stripe_idx] = idx;
2108                 /* Set the start index for next stripe allocation */
2109                 if (!is_specific && stripe_idx < stripe_count - 1) {
2110                         /*
2111                          * for large dir test, put all other slaves on one
2112                          * remote MDT, otherwise we may save too many local
2113                          * slave locks which will exceed RS_MAX_LOCKS.
2114                          */
2115                         if (unlikely(CFS_FAIL_CHECK(OBD_FAIL_LARGE_STRIPE)))
2116                                 idx = master_index;
2117                         mdt_indices[stripe_idx + 1] = (idx + 1) %
2118                                            (lod->lod_remote_mdt_count + 1);
2119                 }
2120                 /* tgt_dt and fid must be ready after search avaible OSP
2121                  * in the above loop */
2122                 LASSERT(tgt_dt != NULL);
2123                 LASSERT(fid_is_sane(&fid));
2124
2125                 /* fail a remote stripe FID allocation */
2126                 if (stripe_idx && CFS_FAIL_CHECK(OBD_FAIL_MDS_STRIPE_FID))
2127                         continue;
2128
2129                 dto = dt_locate_at(env, tgt_dt, &fid,
2130                                   lo->ldo_obj.do_lu.lo_dev->ld_site->ls_top_dev,
2131                                   &conf);
2132                 if (IS_ERR(dto)) {
2133                         rc = PTR_ERR(dto);
2134                         goto error;
2135                 }
2136
2137                 stripes[stripe_idx] = dto;
2138         }
2139
2140         return stripe_idx;
2141
2142 error:
2143         for (j = 1; j < stripe_idx; j++) {
2144                 LASSERT(stripes[j] != NULL);
2145                 dt_object_put(env, stripes[j]);
2146                 stripes[j] = NULL;
2147         }
2148         return rc;
2149 }
2150
2151 static int lod_prep_md_striped_create(const struct lu_env *env,
2152                                       struct dt_object *dt,
2153                                       struct lu_attr *attr,
2154                                       const struct lmv_user_md_v1 *lum,
2155                                       struct dt_object_format *dof,
2156                                       struct thandle *th)
2157 {
2158         struct lod_device *lod = lu2lod_dev(dt->do_lu.lo_dev);
2159         struct lod_object *lo = lod_dt_obj(dt);
2160         struct dt_object **stripes;
2161         struct lu_object_conf conf = { .loc_flags = LOC_F_NEW };
2162         struct lu_fid fid = { 0 };
2163         __u32 stripe_count;
2164         int i;
2165         int rc = 0;
2166
2167         ENTRY;
2168
2169         /* The lum has been verifed in lod_verify_md_striping */
2170         LASSERT(le32_to_cpu(lum->lum_magic) == LMV_USER_MAGIC ||
2171                 le32_to_cpu(lum->lum_magic) == LMV_USER_MAGIC_SPECIFIC);
2172
2173         stripe_count = lo->ldo_dir_stripe_count;
2174
2175         OBD_ALLOC_PTR_ARRAY(stripes, stripe_count);
2176         if (!stripes)
2177                 RETURN(-ENOMEM);
2178
2179         /* Allocate the first stripe locally */
2180         rc = dt_fid_alloc(env, lod->lod_child, &fid, NULL, NULL);
2181         if (rc < 0)
2182                 GOTO(out, rc);
2183
2184         stripes[0] = dt_locate_at(env, lod->lod_child, &fid,
2185                                   dt->do_lu.lo_dev->ld_site->ls_top_dev, &conf);
2186         if (IS_ERR(stripes[0]))
2187                 GOTO(out, rc = PTR_ERR(stripes[0]));
2188
2189         if (lo->ldo_dir_stripe_offset == LMV_OFFSET_DEFAULT) {
2190                 lod_qos_statfs_update(env, lod, &lod->lod_mdt_descs);
2191                 rc = lod_mdt_alloc_qos(env, lo, stripes, 1, stripe_count);
2192                 if (rc == -EAGAIN)
2193                         rc = lod_mdt_alloc_rr(env, lo, stripes, 1,
2194                                               stripe_count);
2195         } else {
2196                 int *idx_array;
2197                 bool is_specific = false;
2198
2199                 OBD_ALLOC_PTR_ARRAY(idx_array, stripe_count);
2200                 if (!idx_array)
2201                         GOTO(out, rc = -ENOMEM);
2202
2203                 if (le32_to_cpu(lum->lum_magic) == LMV_USER_MAGIC_SPECIFIC) {
2204                         is_specific = true;
2205                         for (i = 0; i < stripe_count; i++)
2206                                 idx_array[i] =
2207                                        le32_to_cpu(lum->lum_objects[i].lum_mds);
2208                 }
2209
2210                 /* stripe 0 is local */
2211                 idx_array[0] =
2212                         lu_site2seq(lod2lu_dev(lod)->ld_site)->ss_node_id;
2213                 rc = lod_mdt_alloc_specific(env, lo, stripes, idx_array,
2214                                             is_specific);
2215                 OBD_FREE_PTR_ARRAY(idx_array, stripe_count);
2216         }
2217
2218         if (rc < 0)
2219                 GOTO(out, rc);
2220
2221         LASSERT(rc > 0);
2222
2223         lo->ldo_dir_striped = 1;
2224         lo->ldo_stripe = stripes;
2225         lo->ldo_dir_stripe_count = rc;
2226         lo->ldo_dir_stripes_allocated = stripe_count;
2227         smp_mb();
2228         lo->ldo_dir_stripe_loaded = 1;
2229
2230         rc = lod_dir_declare_create_stripes(env, dt, attr, dof, th);
2231         if (rc < 0)
2232                 lod_striping_free(env, lo);
2233
2234         RETURN(rc);
2235
2236 out:
2237         LASSERT(rc < 0);
2238         if (!IS_ERR_OR_NULL(stripes[0]))
2239                 dt_object_put(env, stripes[0]);
2240         for (i = 1; i < stripe_count; i++)
2241                 LASSERT(!stripes[i]);
2242         OBD_FREE_PTR_ARRAY(stripes, stripe_count);
2243
2244         return rc;
2245 }
2246
2247 /**
2248  *
2249  * Alloc cached foreign LOV
2250  *
2251  * \param[in] lo        object
2252  * \param[in] size      size of foreign LOV
2253  *
2254  * \retval              0 on success
2255  * \retval              negative if failed
2256  */
2257 int lod_alloc_foreign_lov(struct lod_object *lo, size_t size)
2258 {
2259         OBD_ALLOC_LARGE(lo->ldo_foreign_lov, size);
2260         if (lo->ldo_foreign_lov == NULL)
2261                 return -ENOMEM;
2262         lo->ldo_foreign_lov_size = size;
2263         lo->ldo_is_foreign = 1;
2264         return 0;
2265 }
2266
2267 /**
2268  *
2269  * Free cached foreign LOV
2270  *
2271  * \param[in] lo        object
2272  */
2273 void lod_free_foreign_lov(struct lod_object *lo)
2274 {
2275         if (lo->ldo_foreign_lov != NULL)
2276                 OBD_FREE_LARGE(lo->ldo_foreign_lov, lo->ldo_foreign_lov_size);
2277         lo->ldo_foreign_lov = NULL;
2278         lo->ldo_foreign_lov_size = 0;
2279         lo->ldo_is_foreign = 0;
2280 }
2281
2282 /**
2283  *
2284  * Alloc cached foreign LMV
2285  *
2286  * \param[in] lo        object
2287  * \param[in] size      size of foreign LMV
2288  *
2289  * \retval              0 on success
2290  * \retval              negative if failed
2291  */
2292 int lod_alloc_foreign_lmv(struct lod_object *lo, size_t size)
2293 {
2294         OBD_ALLOC_LARGE(lo->ldo_foreign_lmv, size);
2295         if (lo->ldo_foreign_lmv == NULL)
2296                 return -ENOMEM;
2297         lo->ldo_foreign_lmv_size = size;
2298         lo->ldo_is_foreign = 1;
2299
2300         return 0;
2301 }
2302
2303 /**
2304  *
2305  * Free cached foreign LMV
2306  *
2307  * \param[in] lo        object
2308  */
2309 void lod_free_foreign_lmv(struct lod_object *lo)
2310 {
2311         if (lo->ldo_foreign_lmv != NULL)
2312                 OBD_FREE_LARGE(lo->ldo_foreign_lmv, lo->ldo_foreign_lmv_size);
2313         lo->ldo_foreign_lmv = NULL;
2314         lo->ldo_foreign_lmv_size = 0;
2315         lo->ldo_is_foreign = 0;
2316 }
2317
2318 /**
2319  * Declare create striped md object.
2320  *
2321  * The function declares intention to create a striped directory. This is a
2322  * wrapper for lod_prep_md_striped_create(). The only additional functionality
2323  * is to verify pattern \a lum_buf is good. Check that function for the details.
2324  *
2325  * \param[in] env       execution environment
2326  * \param[in] dt        object
2327  * \param[in] attr      attributes to initialize the objects with
2328  * \param[in] lum_buf   a pattern specifying the number of stripes and
2329  *                      MDT to start from
2330  * \param[in] dof       type of objects to be created
2331  * \param[in] th        transaction handle
2332  *
2333  * \retval              0 on success
2334  * \retval              negative if failed
2335  *
2336  */
2337 static int lod_declare_xattr_set_lmv(const struct lu_env *env,
2338                                      struct dt_object *dt,
2339                                      struct lu_attr *attr,
2340                                      const struct lu_buf *lum_buf,
2341                                      struct dt_object_format *dof,
2342                                      struct thandle *th)
2343 {
2344         struct lod_object *lo = lod_dt_obj(dt);
2345         struct lmv_user_md_v1 *lum = lum_buf->lb_buf;
2346         int rc;
2347
2348         ENTRY;
2349         LASSERT(lum != NULL);
2350
2351         CDEBUG(D_INFO,
2352                "lum magic=%x hash=%x count=%u offset=%d inherit=%u rr=%u\n",
2353                le32_to_cpu(lum->lum_magic), le32_to_cpu(lum->lum_hash_type),
2354                le32_to_cpu(lum->lum_stripe_count),
2355                (int)le32_to_cpu(lum->lum_stripe_offset),
2356                lum->lum_max_inherit, lum->lum_max_inherit_rr);
2357
2358         if (lo->ldo_dir_stripe_count == 0) {
2359                 if (lo->ldo_is_foreign) {
2360                         rc = lod_alloc_foreign_lmv(lo, lum_buf->lb_len);
2361                         if (rc != 0)
2362                                 GOTO(out, rc);
2363                         memcpy(lo->ldo_foreign_lmv, lum, lum_buf->lb_len);
2364                         lo->ldo_dir_stripe_loaded = 1;
2365                 }
2366                 GOTO(out, rc = 0);
2367         }
2368
2369         /* prepare dir striped objects */
2370         rc = lod_prep_md_striped_create(env, dt, attr, lum, dof, th);
2371         if (rc != 0) {
2372                 /* failed to create striping, let's reset
2373                  * config so that others don't get confused */
2374                 lod_striping_free(env, lo);
2375                 GOTO(out, rc);
2376         }
2377 out:
2378         RETURN(rc);
2379 }
2380
2381 /**
2382  * Set or replace striped directory layout, and LFSCK may set layout on a plain
2383  * directory, so don't check stripe count.
2384  *
2385  * \param[in] env       execution environment
2386  * \param[in] dt        target object
2387  * \param[in] lmv_buf   LMV buf which contains source stripe FIDs
2388  * \param[in] fl        set or replace
2389  * \param[in] th        transaction handle
2390  *
2391  * \retval              0 on success
2392  * \retval              negative if failed
2393  */
2394 static int lod_dir_layout_set(const struct lu_env *env,
2395                               struct dt_object *dt,
2396                               const struct lu_buf *lmv_buf,
2397                               int fl,
2398                               struct thandle *th)
2399 {
2400         struct dt_object *next = dt_object_child(dt);
2401         struct lod_object *lo = lod_dt_obj(dt);
2402         struct lod_device *lod = lu2lod_dev(lod2lu_obj(lo)->lo_dev);
2403         struct lmv_mds_md_v1 *lmv = lmv_buf->lb_buf;
2404         struct lmv_mds_md_v1 *slave_lmv;
2405         struct lu_buf slave_buf;
2406         int i;
2407         int rc;
2408
2409         ENTRY;
2410
2411         if (!lmv_is_sane2(lmv))
2412                 RETURN(-EINVAL);
2413
2414         /* adjust hash for dir merge, which may not be set in user command */
2415         if (lmv_is_merging(lmv) &&
2416             !(lmv->lmv_migrate_hash & LMV_HASH_TYPE_MASK))
2417                 lmv->lmv_merge_hash |=
2418                         lod->lod_mdt_descs.ltd_lmv_desc.ld_pattern &
2419                         LMV_HASH_TYPE_MASK;
2420
2421         LMV_DEBUG(D_INFO, lmv, "set");
2422
2423         rc = lod_sub_xattr_set(env, next, lmv_buf, XATTR_NAME_LMV, fl, th);
2424         if (rc)
2425                 RETURN(rc);
2426
2427         /* directory restripe may update stripe LMV directly */
2428         if (!lo->ldo_dir_stripe_count)
2429                 RETURN(0);
2430
2431         lo->ldo_dir_hash_type = le32_to_cpu(lmv->lmv_hash_type);
2432         lo->ldo_dir_migrate_offset = le32_to_cpu(lmv->lmv_migrate_offset);
2433         lo->ldo_dir_migrate_hash = le32_to_cpu(lmv->lmv_migrate_hash);
2434         lo->ldo_dir_layout_version = le32_to_cpu(lmv->lmv_layout_version);
2435
2436         OBD_ALLOC_PTR(slave_lmv);
2437         if (!slave_lmv)
2438                 RETURN(-ENOMEM);
2439
2440         lod_prep_slave_lmv_md(slave_lmv, lmv);
2441         slave_buf.lb_buf = slave_lmv;
2442         slave_buf.lb_len = sizeof(*slave_lmv);
2443
2444         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
2445                 if (!lo->ldo_stripe[i])
2446                         continue;
2447
2448                 if (!dt_object_exists(lo->ldo_stripe[i]))
2449                         continue;
2450
2451                 rc = lod_sub_xattr_set(env, lo->ldo_stripe[i], &slave_buf,
2452                                        XATTR_NAME_LMV, fl, th);
2453                 if (rc)
2454                         break;
2455         }
2456
2457         OBD_FREE_PTR(slave_lmv);
2458
2459         RETURN(rc);
2460 }
2461
2462 /**
2463  * Implementation of dt_object_operations::do_declare_xattr_set.
2464  *
2465  * Used with regular (non-striped) objects. Basically it
2466  * initializes the striping information and applies the
2467  * change to all the stripes.
2468  *
2469  * \see dt_object_operations::do_declare_xattr_set() in the API description
2470  * for details.
2471  */
2472 static int lod_dir_declare_xattr_set(const struct lu_env *env,
2473                                      struct dt_object *dt,
2474                                      const struct lu_buf *buf,
2475                                      const char *name, int fl,
2476                                      struct thandle *th)
2477 {
2478         struct dt_object        *next = dt_object_child(dt);
2479         struct lod_device       *d = lu2lod_dev(dt->do_lu.lo_dev);
2480         struct lod_object       *lo = lod_dt_obj(dt);
2481         int                     i;
2482         int                     rc;
2483         ENTRY;
2484
2485         if (strcmp(name, XATTR_NAME_DEFAULT_LMV) == 0) {
2486                 struct lmv_user_md_v1 *lum;
2487
2488                 LASSERT(buf != NULL && buf->lb_buf != NULL);
2489                 lum = buf->lb_buf;
2490                 rc = lod_verify_md_striping(d, lum);
2491                 if (rc != 0)
2492                         RETURN(rc);
2493         } else if (strcmp(name, XATTR_NAME_LOV) == 0) {
2494                 rc = lod_verify_striping(env, d, lo, buf, false);
2495                 if (rc != 0)
2496                         RETURN(rc);
2497         }
2498
2499         rc = lod_sub_declare_xattr_set(env, next, buf, name, fl, th);
2500         if (rc != 0)
2501                 RETURN(rc);
2502
2503         /* Note: Do not set LinkEA on sub-stripes, otherwise
2504          * it will confuse the fid2path process(see mdt_path_current()).
2505          * The linkEA between master and sub-stripes is set in
2506          * lod_xattr_set_lmv(). */
2507         if (strcmp(name, XATTR_NAME_LINK) == 0)
2508                 RETURN(0);
2509
2510         /* set xattr to each stripes, if needed */
2511         rc = lod_striping_load(env, lo);
2512         if (rc != 0)
2513                 RETURN(rc);
2514
2515         if (lo->ldo_dir_stripe_count == 0)
2516                 RETURN(0);
2517
2518         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
2519                 if (!lo->ldo_stripe[i])
2520                         continue;
2521
2522                 if (!dt_object_exists(lo->ldo_stripe[i]))
2523                         continue;
2524
2525                 rc = lod_sub_declare_xattr_set(env, lo->ldo_stripe[i],
2526                                                buf, name, fl, th);
2527                 if (rc != 0)
2528                         break;
2529         }
2530
2531         RETURN(rc);
2532 }
2533
2534 static int
2535 lod_obj_stripe_replace_parent_fid_cb(const struct lu_env *env,
2536                                      struct lod_object *lo,
2537                                      struct dt_object *dt, struct thandle *th,
2538                                      int comp_idx, int stripe_idx,
2539                                      struct lod_obj_stripe_cb_data *data)
2540 {
2541         struct lod_thread_info *info = lod_env_info(env);
2542         struct lod_layout_component *comp = &lo->ldo_comp_entries[comp_idx];
2543         struct filter_fid *ff = &info->lti_ff;
2544         struct lu_buf *buf = &info->lti_buf;
2545         int rc;
2546
2547         buf->lb_buf = ff;
2548         buf->lb_len = sizeof(*ff);
2549         rc = dt_xattr_get(env, dt, buf, XATTR_NAME_FID);
2550         if (rc < 0) {
2551                 if (rc == -ENODATA)
2552                         return 0;
2553                 return rc;
2554         }
2555
2556         /*
2557          * locd_buf is set if it's called by dir migration, which doesn't check
2558          * pfid and comp id.
2559          */
2560         if (data->locd_buf) {
2561                 memset(ff, 0, sizeof(*ff));
2562                 ff->ff_parent = *(struct lu_fid *)data->locd_buf->lb_buf;
2563         } else {
2564                 filter_fid_le_to_cpu(ff, ff, sizeof(*ff));
2565
2566                 if (lu_fid_eq(lod_object_fid(lo), &ff->ff_parent) &&
2567                     ff->ff_layout.ol_comp_id == comp->llc_id)
2568                         return 0;
2569
2570                 memset(ff, 0, sizeof(*ff));
2571                 ff->ff_parent = *lu_object_fid(&lo->ldo_obj.do_lu);
2572         }
2573
2574         /* rewrite filter_fid */
2575         ff->ff_parent.f_ver = stripe_idx;
2576         ff->ff_layout.ol_stripe_size = comp->llc_stripe_size;
2577         ff->ff_layout.ol_stripe_count = comp->llc_stripe_count;
2578         ff->ff_layout.ol_comp_id = comp->llc_id;
2579         ff->ff_layout.ol_comp_start = comp->llc_extent.e_start;
2580         ff->ff_layout.ol_comp_end = comp->llc_extent.e_end;
2581         filter_fid_cpu_to_le(ff, ff, sizeof(*ff));
2582
2583         if (data->locd_declare)
2584                 rc = lod_sub_declare_xattr_set(env, dt, buf, XATTR_NAME_FID,
2585                                                LU_XATTR_REPLACE, th);
2586         else
2587                 rc = lod_sub_xattr_set(env, dt, buf, XATTR_NAME_FID,
2588                                        LU_XATTR_REPLACE, th);
2589
2590         return rc;
2591 }
2592
2593 /**
2594  * Reset parent FID on OST object
2595  *
2596  * Replace parent FID with @dt object FID, which is only called during migration
2597  * to reset the parent FID after the MDT object is migrated to the new MDT, i.e.
2598  * the FID is changed.
2599  *
2600  * \param[in] env execution environment
2601  * \param[in] dt dt_object whose stripes's parent FID will be reset
2602  * \parem[in] th thandle
2603  * \param[in] declare if it is declare
2604  *
2605  * \retval      0 if reset succeeds
2606  * \retval      negative errno if reset fails
2607  */
2608 static int lod_replace_parent_fid(const struct lu_env *env,
2609                                   struct dt_object *dt,
2610                                   const struct lu_buf *buf,
2611                                   struct thandle *th, bool declare)
2612 {
2613         struct lod_object *lo = lod_dt_obj(dt);
2614         struct lod_thread_info  *info = lod_env_info(env);
2615         struct filter_fid *ff;
2616         struct lod_obj_stripe_cb_data data = { { 0 } };
2617         int rc;
2618         ENTRY;
2619
2620         LASSERT(S_ISREG(dt->do_lu.lo_header->loh_attr));
2621
2622         /* set xattr to each stripes, if needed */
2623         rc = lod_striping_load(env, lo);
2624         if (rc != 0)
2625                 RETURN(rc);
2626
2627         if (!lod_obj_is_striped(dt))
2628                 RETURN(0);
2629
2630         if (info->lti_ea_store_size < sizeof(*ff)) {
2631                 rc = lod_ea_store_resize(info, sizeof(*ff));
2632                 if (rc != 0)
2633                         RETURN(rc);
2634         }
2635
2636         data.locd_declare = declare;
2637         data.locd_stripe_cb = lod_obj_stripe_replace_parent_fid_cb;
2638         data.locd_buf = buf;
2639         rc = lod_obj_for_each_stripe(env, lo, th, &data);
2640
2641         RETURN(rc);
2642 }
2643
2644 __u16 lod_comp_entry_stripe_count(struct lod_object *lo, int comp_idx,
2645                                   bool is_dir)
2646 {
2647         struct lod_device *lod = lu2lod_dev(lod2lu_obj(lo)->lo_dev);
2648         struct lod_layout_component *entry;
2649         enum lod_uses_hint flags = LOD_USES_ASSIGNED_STRIPE;
2650
2651         if (is_dir)
2652                 return  0;
2653
2654         entry = &lo->ldo_comp_entries[comp_idx];
2655         if (lod_comp_inited(entry))
2656                 return entry->llc_stripe_count;
2657         if (entry->llc_stripe_count == LOV_ALL_STRIPES)
2658                 return lod_get_stripe_count_plain(lod, lo,
2659                                                   entry->llc_stripe_count,
2660                                                   entry->llc_pattern &
2661                                                       LOV_PATTERN_OVERSTRIPING,
2662                                                   &flags);
2663
2664         return lod_get_stripe_count(lod, lo, comp_idx, entry->llc_stripe_count,
2665                                  entry->llc_pattern & LOV_PATTERN_OVERSTRIPING,
2666                                  &flags);
2667 }
2668
2669 static int lod_comp_md_size(struct lod_object *lo, bool is_dir)
2670 {
2671         int magic, size = 0, i;
2672         struct lod_layout_component *comp_entries;
2673         __u16 comp_cnt;
2674         bool is_composite, is_foreign = false;
2675
2676         if (is_dir) {
2677                 comp_cnt = lo->ldo_def_striping->lds_def_comp_cnt;
2678                 comp_entries = lo->ldo_def_striping->lds_def_comp_entries;
2679                 is_composite =
2680                         lo->ldo_def_striping->lds_def_striping_is_composite;
2681         } else {
2682                 comp_cnt = lo->ldo_comp_cnt;
2683                 comp_entries = lo->ldo_comp_entries;
2684                 is_composite = lo->ldo_is_composite;
2685                 is_foreign = lo->ldo_is_foreign;
2686         }
2687
2688         if (is_foreign)
2689                 return lo->ldo_foreign_lov_size;
2690
2691         LASSERT(comp_cnt != 0 && comp_entries != NULL);
2692         if (is_composite) {
2693                 size = sizeof(struct lov_comp_md_v1) +
2694                        sizeof(struct lov_comp_md_entry_v1) * comp_cnt;
2695                 LASSERT(size % sizeof(__u64) == 0);
2696         }
2697
2698         for (i = 0; i < comp_cnt; i++) {
2699                 __u16 stripe_count;
2700
2701                 magic = comp_entries[i].llc_pool ? LOV_MAGIC_V3 : LOV_MAGIC_V1;
2702                 stripe_count = lod_comp_entry_stripe_count(lo, i, is_dir);
2703                 if (!is_dir && is_composite)
2704                         lod_comp_shrink_stripe_count(&comp_entries[i],
2705                                                      &stripe_count);
2706
2707                 size += lov_user_md_size(stripe_count, magic);
2708                 LASSERT(size % sizeof(__u64) == 0);
2709         }
2710         return size;
2711 }
2712
2713 /**
2714  * Declare component add. The xattr name is XATTR_LUSTRE_LOV.add, and
2715  * the xattr value is binary lov_comp_md_v1 which contains component(s)
2716  * to be added.
2717   *
2718  * \param[in] env       execution environment
2719  * \param[in] dt        dt_object to add components on
2720  * \param[in] buf       buffer contains components to be added
2721  * \parem[in] th        thandle
2722  *
2723  * \retval      0 on success
2724  * \retval      negative errno on failure
2725  */
2726 static int lod_declare_layout_add(const struct lu_env *env,
2727                                   struct dt_object *dt,
2728                                   const struct lu_buf *buf,
2729                                   struct thandle *th)
2730 {
2731         struct lod_thread_info  *info = lod_env_info(env);
2732         struct lod_layout_component *comp_array, *lod_comp, *old_array;
2733         struct lod_device *d = lu2lod_dev(dt->do_lu.lo_dev);
2734         struct dt_object *next = dt_object_child(dt);
2735         struct lov_desc *desc = &d->lod_ost_descs.ltd_lov_desc;
2736         struct lod_object *lo = lod_dt_obj(dt);
2737         struct lov_comp_md_v1 *comp_v1 = buf->lb_buf;
2738         __u32 magic;
2739         int i, rc, array_cnt, old_array_cnt;
2740         ENTRY;
2741
2742         LASSERT(lo->ldo_is_composite);
2743
2744         if (lo->ldo_flr_state != LCM_FL_NONE)
2745                 RETURN(-EBUSY);
2746
2747         rc = lod_verify_striping(env, d, lo, buf, false);
2748         if (rc != 0)
2749                 RETURN(rc);
2750
2751         magic = comp_v1->lcm_magic;
2752         if (magic == __swab32(LOV_USER_MAGIC_COMP_V1)) {
2753                 lustre_swab_lov_comp_md_v1(comp_v1);
2754                 magic = comp_v1->lcm_magic;
2755         }
2756
2757         if (magic != LOV_USER_MAGIC_COMP_V1)
2758                 RETURN(-EINVAL);
2759
2760         mutex_lock(&lo->ldo_layout_mutex);
2761
2762         array_cnt = lo->ldo_comp_cnt + comp_v1->lcm_entry_count;
2763         OBD_ALLOC_PTR_ARRAY(comp_array, array_cnt);
2764         if (comp_array == NULL) {
2765                 mutex_unlock(&lo->ldo_layout_mutex);
2766                 RETURN(-ENOMEM);
2767         }
2768
2769
2770         memcpy(comp_array, lo->ldo_comp_entries,
2771                sizeof(*comp_array) * lo->ldo_comp_cnt);
2772
2773         for (i = 0; i < comp_v1->lcm_entry_count; i++) {
2774                 struct lov_user_md_v1 *v1;
2775                 struct lu_extent *ext;
2776
2777                 v1 = (struct lov_user_md *)((char *)comp_v1 +
2778                                 comp_v1->lcm_entries[i].lcme_offset);
2779                 ext = &comp_v1->lcm_entries[i].lcme_extent;
2780
2781                 lod_comp = &comp_array[lo->ldo_comp_cnt + i];
2782                 lod_comp->llc_extent.e_start = ext->e_start;
2783                 lod_comp->llc_extent.e_end = ext->e_end;
2784                 lod_comp->llc_stripe_offset = v1->lmm_stripe_offset;
2785                 lod_comp->llc_flags = comp_v1->lcm_entries[i].lcme_flags;
2786
2787                 lod_comp->llc_stripe_size = v1->lmm_stripe_size;
2788                 lod_comp->llc_stripe_count = v1->lmm_stripe_count;
2789                 /**
2790                  * limit stripe count so that it's less than/equal to
2791                  * extent_size / stripe_size.
2792                  *
2793                  * Note: extension size reused llc_stripe_size field and
2794                  * uninstantiated component could be defined with
2795                  * extent_start == extent_end as extension component will
2796                  * expand it later.
2797                  */
2798                 if (!(lod_comp->llc_flags & LCME_FL_EXTENSION) &&
2799                     (lod_comp_inited(lod_comp) ||
2800                      lod_comp->llc_extent.e_start <
2801                      lod_comp->llc_extent.e_end) &&
2802                     lod_comp->llc_stripe_count != LOV_ALL_STRIPES &&
2803                     ext->e_end != OBD_OBJECT_EOF &&
2804                     (__u64)(lod_comp->llc_stripe_count *
2805                             lod_comp->llc_stripe_size) >
2806                     (ext->e_end - ext->e_start))
2807                         lod_comp->llc_stripe_count =
2808                                 DIV_ROUND_UP(ext->e_end - ext->e_start,
2809                                              lod_comp->llc_stripe_size);
2810                 lod_adjust_stripe_info(lod_comp, desc, 0);
2811
2812                 if (v1->lmm_magic == LOV_USER_MAGIC_V3) {
2813                         struct lov_user_md_v3 *v3 = (typeof(*v3) *) v1;
2814
2815                         if (v3->lmm_pool_name[0] != '\0' &&
2816                             !lov_pool_is_ignored(v3->lmm_pool_name)) {
2817                                 rc = lod_set_pool(&lod_comp->llc_pool,
2818                                                   v3->lmm_pool_name);
2819                                 if (rc)
2820                                         GOTO(error, rc);
2821                         }
2822                 }
2823         }
2824
2825         old_array = lo->ldo_comp_entries;
2826         old_array_cnt = lo->ldo_comp_cnt;
2827
2828         lo->ldo_comp_entries = comp_array;
2829         lo->ldo_comp_cnt = array_cnt;
2830
2831         /* No need to increase layout generation here, it will be increased
2832          * later when generating component ID for the new components */
2833
2834         info->lti_buf.lb_len = lod_comp_md_size(lo, false);
2835         rc = lod_sub_declare_xattr_set(env, next, &info->lti_buf,
2836                                               XATTR_NAME_LOV, 0, th);
2837         if (rc) {
2838                 lo->ldo_comp_entries = old_array;
2839                 lo->ldo_comp_cnt = old_array_cnt;
2840                 GOTO(error, rc);
2841         }
2842
2843         OBD_FREE_PTR_ARRAY(old_array, old_array_cnt);
2844
2845         LASSERT(lo->ldo_mirror_count == 1);
2846         lo->ldo_mirrors[0].lme_end = array_cnt - 1;
2847
2848         mutex_unlock(&lo->ldo_layout_mutex);
2849
2850         RETURN(0);
2851
2852 error:
2853         for (i = lo->ldo_comp_cnt; i < array_cnt; i++) {
2854                 lod_comp = &comp_array[i];
2855                 if (lod_comp->llc_pool != NULL) {
2856                         OBD_FREE(lod_comp->llc_pool,
2857                                  strlen(lod_comp->llc_pool) + 1);
2858                         lod_comp->llc_pool = NULL;
2859                 }
2860         }
2861         OBD_FREE_PTR_ARRAY(comp_array, array_cnt);
2862         mutex_unlock(&lo->ldo_layout_mutex);
2863
2864         RETURN(rc);
2865 }
2866
2867 /**
2868  * lod_last_non_stale_mirror() - Check if a mirror is the last non-stale mirror.
2869  * @mirror_id: Mirror id to be checked.
2870  * @lo:        LOD object.
2871  *
2872  * This function checks if a mirror with specified @mirror_id is the last
2873  * non-stale mirror of a LOD object @lo.
2874  *
2875  * Return: true or false.
2876  */
2877 static inline
2878 bool lod_last_non_stale_mirror(__u16 mirror_id, struct lod_object *lo)
2879 {
2880         struct lod_layout_component *lod_comp;
2881         bool has_stale_flag;
2882         int i;
2883
2884         for (i = 0; i < lo->ldo_mirror_count; i++) {
2885                 if (lo->ldo_mirrors[i].lme_id == mirror_id ||
2886                     lo->ldo_mirrors[i].lme_stale)
2887                         continue;
2888
2889                 has_stale_flag = false;
2890                 lod_foreach_mirror_comp(lod_comp, lo, i) {
2891                         if (lod_comp->llc_flags & LCME_FL_STALE) {
2892                                 has_stale_flag = true;
2893                                 break;
2894                         }
2895                 }
2896                 if (!has_stale_flag)
2897                         return false;
2898         }
2899
2900         return true;
2901 }
2902
2903 /**
2904  * Declare component set. The xattr is name XATTR_LUSTRE_LOV.set.$field,
2905  * the '$field' can only be 'flags' now. The xattr value is binary
2906  * lov_comp_md_v1 which contains the component ID(s) and the value of
2907  * the field to be modified.
2908  * Please update allowed_lustre_lov macro if $field groks more values
2909  * in the future.
2910  *
2911  * \param[in] env       execution environment
2912  * \param[in] dt        dt_object to be modified
2913  * \param[in] op        operation string, like "set.flags"
2914  * \param[in] buf       buffer contains components to be set
2915  * \parem[in] th        thandle
2916  *
2917  * \retval      0 on success
2918  * \retval      negative errno on failure
2919  */
2920 static int lod_declare_layout_set(const struct lu_env *env,
2921                                   struct dt_object *dt,
2922                                   char *op, const struct lu_buf *buf,
2923                                   struct thandle *th)
2924 {
2925         struct lod_layout_component     *lod_comp;
2926         struct lod_thread_info  *info = lod_env_info(env);
2927         struct lod_device       *d = lu2lod_dev(dt->do_lu.lo_dev);
2928         struct lod_object       *lo = lod_dt_obj(dt);
2929         struct lov_comp_md_v1   *comp_v1 = buf->lb_buf;
2930         __u32   magic;
2931         int     i, j, rc;
2932         bool    changed = false;
2933         ENTRY;
2934
2935         /* Please update allowed_lustre_lov macro if op
2936          * groks more values in the future
2937          */
2938         if (strcmp(op, "set.flags") != 0) {
2939                 CDEBUG(D_LAYOUT, "%s: operation (%s) not supported.\n",
2940                        lod2obd(d)->obd_name, op);
2941                 RETURN(-ENOTSUPP);
2942         }
2943
2944         magic = comp_v1->lcm_magic;
2945         if (magic == __swab32(LOV_USER_MAGIC_COMP_V1)) {
2946                 lustre_swab_lov_comp_md_v1(comp_v1);
2947                 magic = comp_v1->lcm_magic;
2948         }
2949
2950         if (magic != LOV_USER_MAGIC_COMP_V1)
2951                 RETURN(-EINVAL);
2952
2953         if (comp_v1->lcm_entry_count == 0) {
2954                 CDEBUG(D_LAYOUT, "%s: entry count is zero.\n",
2955                        lod2obd(d)->obd_name);
2956                 RETURN(-EINVAL);
2957         }
2958
2959         mutex_lock(&lo->ldo_layout_mutex);
2960         for (i = 0; i < comp_v1->lcm_entry_count; i++) {
2961                 __u32 id = comp_v1->lcm_entries[i].lcme_id;
2962                 __u32 flags = comp_v1->lcm_entries[i].lcme_flags;
2963                 __u32 mirror_flag = flags & LCME_MIRROR_FLAGS;
2964                 __u16 mirror_id = mirror_id_of(id);
2965                 bool neg = flags & LCME_FL_NEG;
2966
2967                 if (flags & LCME_FL_INIT) {
2968                         if (changed)
2969                                 lod_striping_free_nolock(env, lo);
2970                         mutex_unlock(&lo->ldo_layout_mutex);
2971                         RETURN(-EINVAL);
2972                 }
2973
2974                 flags &= ~(LCME_MIRROR_FLAGS | LCME_FL_NEG);
2975                 for (j = 0; j < lo->ldo_comp_cnt; j++) {
2976                         lod_comp = &lo->ldo_comp_entries[j];
2977
2978                         /* lfs only put one flag in each entry */
2979                         if ((flags && id != lod_comp->llc_id) ||
2980                             (mirror_flag && mirror_id !=
2981                                             mirror_id_of(lod_comp->llc_id)))
2982                                 continue;
2983
2984                         if (neg) {
2985                                 if (flags)
2986                                         lod_comp->llc_flags &= ~flags;
2987                                 if (mirror_flag)
2988                                         lod_comp->llc_flags &= ~mirror_flag;
2989                         } else {
2990                                 if (flags) {
2991                                         if ((flags & LCME_FL_STALE) &&
2992                                             lod_last_non_stale_mirror(mirror_id,
2993                                                                       lo)) {
2994                                                 mutex_unlock(
2995                                                         &lo->ldo_layout_mutex);
2996                                                 RETURN(-EUCLEAN);
2997                                         }
2998                                         lod_comp->llc_flags |= flags;
2999                                 }
3000                                 if (mirror_flag) {
3001                                         lod_comp->llc_flags |= mirror_flag;
3002                                         if (mirror_flag & LCME_FL_NOSYNC)
3003                                                 lod_comp->llc_timestamp =
3004                                                        ktime_get_real_seconds();
3005                                 }
3006                         }
3007                         changed = true;
3008                 }
3009         }
3010         mutex_unlock(&lo->ldo_layout_mutex);
3011
3012         if (!changed) {
3013                 CDEBUG(D_LAYOUT, "%s: requested component(s) not found.\n",
3014                        lod2obd(d)->obd_name);
3015                 RETURN(-EINVAL);
3016         }
3017
3018         lod_obj_inc_layout_gen(lo);
3019
3020         info->lti_buf.lb_len = lod_comp_md_size(lo, false);
3021         rc = lod_sub_declare_xattr_set(env, dt_object_child(dt), &info->lti_buf,
3022                                        XATTR_NAME_LOV, LU_XATTR_REPLACE, th);
3023         RETURN(rc);
3024 }
3025
3026 /**
3027  * Declare component deletion. The xattr name is XATTR_LUSTRE_LOV.del,
3028  * and the xattr value is a unique component ID or a special lcme_id.
3029  *
3030  * \param[in] env       execution environment
3031  * \param[in] dt        dt_object to be operated on
3032  * \param[in] buf       buffer contains component ID or lcme_id
3033  * \parem[in] th        thandle
3034  *
3035  * \retval      0 on success
3036  * \retval      negative errno on failure
3037  */
3038 static int lod_declare_layout_del(const struct lu_env *env,
3039                                   struct dt_object *dt,
3040                                   const struct lu_buf *buf,
3041                                   struct thandle *th)
3042 {
3043         struct lod_thread_info  *info = lod_env_info(env);
3044         struct dt_object *next = dt_object_child(dt);
3045         struct lod_device *d = lu2lod_dev(dt->do_lu.lo_dev);
3046         struct lod_object *lo = lod_dt_obj(dt);
3047         struct lu_attr *attr = &lod_env_info(env)->lti_attr;
3048         struct lov_comp_md_v1 *comp_v1 = buf->lb_buf;
3049         __u32 magic, id, flags, neg_flags = 0;
3050         int rc, i, j, left;
3051         ENTRY;
3052
3053         LASSERT(lo->ldo_is_composite);
3054
3055         if (lo->ldo_flr_state != LCM_FL_NONE)
3056                 RETURN(-EBUSY);
3057
3058         magic = comp_v1->lcm_magic;
3059         if (magic == __swab32(LOV_USER_MAGIC_COMP_V1)) {
3060                 lustre_swab_lov_comp_md_v1(comp_v1);
3061                 magic = comp_v1->lcm_magic;
3062         }
3063
3064         if (magic != LOV_USER_MAGIC_COMP_V1)
3065                 RETURN(-EINVAL);
3066
3067         id = comp_v1->lcm_entries[0].lcme_id;
3068         flags = comp_v1->lcm_entries[0].lcme_flags;
3069
3070         if (id > LCME_ID_MAX || (flags & ~LCME_KNOWN_FLAGS)) {
3071                 CDEBUG(D_LAYOUT, "%s: invalid component id %#x, flags %#x\n",
3072                        lod2obd(d)->obd_name, id, flags);
3073                 RETURN(-EINVAL);
3074         }
3075
3076         if (id != LCME_ID_INVAL && flags != 0) {
3077                 CDEBUG(D_LAYOUT, "%s: specified both id and flags.\n",
3078                        lod2obd(d)->obd_name);
3079                 RETURN(-EINVAL);
3080         }
3081
3082         if (id == LCME_ID_INVAL && !flags) {
3083                 CDEBUG(D_LAYOUT, "%s: no id or flags specified.\n",
3084                        lod2obd(d)->obd_name);
3085                 RETURN(-EINVAL);
3086         }
3087
3088         if (flags & LCME_FL_NEG) {
3089                 neg_flags = flags & ~LCME_FL_NEG;
3090                 flags = 0;
3091         }
3092
3093         mutex_lock(&lo->ldo_layout_mutex);
3094
3095         left = lo->ldo_comp_cnt;
3096         if (left <= 0) {
3097                 mutex_unlock(&lo->ldo_layout_mutex);
3098                 RETURN(-EINVAL);
3099         }
3100
3101         for (i = (lo->ldo_comp_cnt - 1); i >= 0; i--) {
3102                 struct lod_layout_component *lod_comp;
3103
3104                 lod_comp = &lo->ldo_comp_entries[i];
3105
3106                 if (id != LCME_ID_INVAL && id != lod_comp->llc_id)
3107                         continue;
3108                 else if (flags && !(flags & lod_comp->llc_flags))
3109                         continue;
3110                 else if (neg_flags && (neg_flags & lod_comp->llc_flags))
3111                         continue;
3112
3113                 if (left != (i + 1)) {
3114                         CDEBUG(D_LAYOUT, "%s: this deletion will create "
3115                                "a hole.\n", lod2obd(d)->obd_name);
3116                         mutex_unlock(&lo->ldo_layout_mutex);
3117                         RETURN(-EINVAL);
3118                 }
3119                 left--;
3120
3121                 /* Mark the component as deleted */
3122                 lod_comp->llc_id = LCME_ID_INVAL;
3123
3124                 /* Not instantiated component */
3125                 if (lod_comp->llc_stripe == NULL)
3126                         continue;
3127
3128                 LASSERT(lod_comp->llc_stripe_count > 0);
3129                 for (j = 0; j < lod_comp->llc_stripe_count; j++) {
3130                         struct dt_object *obj = lod_comp->llc_stripe[j];
3131
3132                         if (obj == NULL)
3133                                 continue;
3134                         rc = lod_sub_declare_destroy(env, obj, th);
3135                         if (rc) {
3136                                 mutex_unlock(&lo->ldo_layout_mutex);
3137                                 RETURN(rc);
3138                         }
3139                 }
3140         }
3141
3142         LASSERTF(left >= 0, "left = %d\n", left);
3143         if (left == lo->ldo_comp_cnt) {
3144                 CDEBUG(D_LAYOUT, "%s: requested component id:%#x not found\n",
3145                        lod2obd(d)->obd_name, id);
3146                 mutex_unlock(&lo->ldo_layout_mutex);
3147                 RETURN(-EINVAL);
3148         }
3149
3150         mutex_unlock(&lo->ldo_layout_mutex);
3151
3152         memset(attr, 0, sizeof(*attr));
3153         attr->la_valid = LA_SIZE;
3154         rc = lod_sub_declare_attr_set(env, next, attr, th);
3155         if (rc)
3156                 RETURN(rc);
3157
3158         if (left > 0) {
3159                 info->lti_buf.lb_len = lod_comp_md_size(lo, false);
3160                 rc = lod_sub_declare_xattr_set(env, next, &info->lti_buf,
3161                                                XATTR_NAME_LOV, 0, th);
3162         } else {
3163                 rc = lod_sub_declare_xattr_del(env, next, XATTR_NAME_LOV, th);
3164         }
3165
3166         RETURN(rc);
3167 }
3168
3169 /**
3170  * Declare layout add/set/del operations issued by special xattr names:
3171  *
3172  * XATTR_LUSTRE_LOV.add         add component(s) to existing file
3173  * XATTR_LUSTRE_LOV.del         delete component(s) from existing file
3174  * XATTR_LUSTRE_LOV.set.$field  set specified field of certain component(s)
3175  *
3176  * \param[in] env       execution environment
3177  * \param[in] dt        object
3178  * \param[in] name      name of xattr
3179  * \param[in] buf       lu_buf contains xattr value
3180  * \param[in] th        transaction handle
3181  *
3182  * \retval              0 on success
3183  * \retval              negative if failed
3184  */
3185 static int lod_declare_modify_layout(const struct lu_env *env,
3186                                      struct dt_object *dt,
3187                                      const char *name,
3188                                      const struct lu_buf *buf,
3189                                      struct thandle *th)
3190 {
3191         struct lod_device *d = lu2lod_dev(dt->do_lu.lo_dev);
3192         struct lod_object *lo = lod_dt_obj(dt);
3193         char *op;
3194         int rc, len = strlen(XATTR_LUSTRE_LOV);
3195         ENTRY;
3196
3197         LASSERT(dt_object_exists(dt));
3198
3199         if (strlen(name) <= len || name[len] != '.') {
3200                 CDEBUG(D_LAYOUT, "%s: invalid xattr name: %s\n",
3201                        lod2obd(d)->obd_name, name);
3202                 RETURN(-EINVAL);
3203         }
3204         len++;
3205
3206         rc = lod_striping_load(env, lo);
3207         if (rc)
3208                 GOTO(unlock, rc);
3209
3210         /* the layout to be modified must be a composite layout */
3211         if (!lo->ldo_is_composite) {
3212                 CDEBUG(D_LAYOUT, "%s: object "DFID" isn't a composite file.\n",
3213                        lod2obd(d)->obd_name, PFID(lu_object_fid(&dt->do_lu)));
3214                 GOTO(unlock, rc = -EINVAL);
3215         }
3216
3217         op = (char *)name + len;
3218         if (strcmp(op, "add") == 0) {
3219                 rc = lod_declare_layout_add(env, dt, buf, th);
3220         } else if (strcmp(op, "del") == 0) {
3221                 rc = lod_declare_layout_del(env, dt, buf, th);
3222         } else if (strncmp(op, "set", strlen("set")) == 0) {
3223                 rc = lod_declare_layout_set(env, dt, op, buf, th);
3224         } else  {
3225                 CDEBUG(D_LAYOUT, "%s: unsupported xattr name:%s\n",
3226                        lod2obd(d)->obd_name, name);
3227                 GOTO(unlock, rc = -ENOTSUPP);
3228         }
3229 unlock:
3230         if (rc)
3231                 lod_striping_free(env, lo);
3232
3233         RETURN(rc);
3234 }
3235
3236 /**
3237  * Convert a plain file lov_mds_md to a composite layout.
3238  *
3239  * \param[in,out] info  the thread info::lti_ea_store buffer contains little
3240  *                      endian plain file layout
3241  *
3242  * \retval              0 on success, <0 on failure
3243  */
3244 static int lod_layout_convert(struct lod_thread_info *info)
3245 {
3246         struct lov_mds_md *lmm = info->lti_ea_store;
3247         struct lov_mds_md *lmm_save;
3248         struct lov_comp_md_v1 *lcm;
3249         struct lov_comp_md_entry_v1 *lcme;
3250         size_t size;
3251         __u32 blob_size;
3252         int rc = 0;
3253         ENTRY;
3254
3255         /* realloc buffer to a composite layout which contains one component */
3256         blob_size = lov_mds_md_size(le16_to_cpu(lmm->lmm_stripe_count),
3257                                     le32_to_cpu(lmm->lmm_magic));
3258         size = sizeof(*lcm) + sizeof(*lcme) + blob_size;
3259
3260         OBD_ALLOC_LARGE(lmm_save, blob_size);
3261         if (!lmm_save)
3262                 GOTO(out, rc = -ENOMEM);
3263
3264         memcpy(lmm_save, lmm, blob_size);
3265
3266         if (info->lti_ea_store_size < size) {
3267                 rc = lod_ea_store_resize(info, size);
3268                 if (rc)
3269                         GOTO(out, rc);
3270         }
3271
3272         lcm = info->lti_ea_store;
3273         memset(lcm, 0, sizeof(*lcm) + sizeof(*lcme));
3274         lcm->lcm_magic = cpu_to_le32(LOV_MAGIC_COMP_V1);
3275         lcm->lcm_size = cpu_to_le32(size);
3276         lcm->lcm_layout_gen = cpu_to_le32(le16_to_cpu(
3277                                                 lmm_save->lmm_layout_gen));
3278         lcm->lcm_flags = cpu_to_le16(LCM_FL_NONE);
3279         lcm->lcm_entry_count = cpu_to_le16(1);
3280
3281         lcme = &lcm->lcm_entries[0];
3282         lcme->lcme_flags = cpu_to_le32(LCME_FL_INIT);
3283         lcme->lcme_extent.e_start = 0;
3284         lcme->lcme_extent.e_end = cpu_to_le64(OBD_OBJECT_EOF);
3285         lcme->lcme_offset = cpu_to_le32(sizeof(*lcm) + sizeof(*lcme));
3286         lcme->lcme_size = cpu_to_le32(blob_size);
3287
3288         memcpy((char *)lcm + lcme->lcme_offset, (char *)lmm_save, blob_size);
3289
3290         EXIT;
3291 out:
3292         if (lmm_save)
3293                 OBD_FREE_LARGE(lmm_save, blob_size);
3294         return rc;
3295 }
3296
3297 /**
3298  * Merge layouts to form a mirrored file.
3299  */
3300 static int lod_declare_layout_merge(const struct lu_env *env,
3301                 struct dt_object *dt, const struct lu_buf *mbuf,
3302                 struct thandle *th)
3303 {
3304         struct lod_thread_info *info = lod_env_info(env);
3305         struct lu_attr *layout_attr = &info->lti_layout_attr;
3306         struct lu_buf *buf = &info->lti_buf;
3307         struct lod_object *lo = lod_dt_obj(dt);
3308         struct lov_comp_md_v1 *lcm;
3309         struct lov_comp_md_v1 *cur_lcm;
3310         struct lov_comp_md_v1 *merge_lcm;
3311         struct lov_comp_md_entry_v1 *lcme;
3312         struct lov_mds_md_v1 *lmm;
3313         size_t size = 0;
3314         size_t offset;
3315         __u16 cur_entry_count;
3316         __u16 merge_entry_count;
3317         __u32 id = 0;
3318         __u16 mirror_id = 0;
3319         __u32 mirror_count;
3320         int rc, i;
3321         bool merge_has_dom;
3322
3323         ENTRY;
3324
3325         merge_lcm = mbuf->lb_buf;
3326         if (mbuf->lb_len < sizeof(*merge_lcm))
3327                 RETURN(-EINVAL);
3328
3329         /* must be an existing layout from disk */
3330         if (le32_to_cpu(merge_lcm->lcm_magic) != LOV_MAGIC_COMP_V1)
3331                 RETURN(-EINVAL);
3332
3333         merge_entry_count = le16_to_cpu(merge_lcm->lcm_entry_count);
3334
3335         /* do not allow to merge two mirrored files */
3336         if (le16_to_cpu(merge_lcm->lcm_mirror_count))
3337                 RETURN(-EBUSY);
3338
3339         /* verify the target buffer */
3340         rc = lod_get_lov_ea(env, lo);
3341         if (rc <= 0)
3342                 RETURN(rc ? : -ENODATA);
3343
3344         cur_lcm = info->lti_ea_store;
3345         switch (le32_to_cpu(cur_lcm->lcm_magic)) {
3346         case LOV_MAGIC_V1:
3347         case LOV_MAGIC_V3:
3348                 rc = lod_layout_convert(info);
3349                 break;
3350         case LOV_MAGIC_COMP_V1:
3351         case LOV_MAGIC_SEL:
3352                 rc = 0;
3353                 break;
3354         default:
3355                 rc = -EINVAL;
3356         }
3357         if (rc)
3358                 RETURN(rc);
3359
3360         /* info->lti_ea_store could be reallocated in lod_layout_convert() */
3361         cur_lcm = info->lti_ea_store;
3362         cur_entry_count = le16_to_cpu(cur_lcm->lcm_entry_count);
3363
3364         /* 'lcm_mirror_count + 1' is the current # of mirrors the file has */
3365         mirror_count = le16_to_cpu(cur_lcm->lcm_mirror_count) + 1;
3366         if (mirror_count + 1 > LUSTRE_MIRROR_COUNT_MAX)
3367                 RETURN(-ERANGE);
3368
3369         /* size of new layout */
3370         size = le32_to_cpu(cur_lcm->lcm_size) +
3371                le32_to_cpu(merge_lcm->lcm_size) - sizeof(*cur_lcm);
3372
3373         memset(buf, 0, sizeof(*buf));
3374         lu_buf_alloc(buf, size);
3375         if (buf->lb_buf == NULL)
3376                 RETURN(-ENOMEM);
3377
3378         lcm = buf->lb_buf;
3379         memcpy(lcm, cur_lcm, sizeof(*lcm) + cur_entry_count * sizeof(*lcme));
3380
3381         offset = sizeof(*lcm) +
3382                  sizeof(*lcme) * (cur_entry_count + merge_entry_count);
3383         for (i = 0; i < cur_entry_count; i++) {
3384                 struct lov_comp_md_entry_v1 *cur_lcme;
3385
3386                 lcme = &lcm->lcm_entries[i];
3387                 cur_lcme = &cur_lcm->lcm_entries[i];
3388
3389                 lcme->lcme_offset = cpu_to_le32(offset);
3390                 memcpy((char *)lcm + offset,
3391                        (char *)cur_lcm + le32_to_cpu(cur_lcme->lcme_offset),
3392                        le32_to_cpu(lcme->lcme_size));
3393
3394                 offset += le32_to_cpu(lcme->lcme_size);
3395
3396                 if (mirror_count == 1 &&
3397                     mirror_id_of(le32_to_cpu(lcme->lcme_id)) == 0) {
3398                         /* Add mirror from a non-flr file, create new mirror ID.
3399                          * Otherwise, keep existing mirror's component ID, used
3400                          * for mirror extension.
3401                          */
3402                         id = pflr_id(1, i + 1);
3403                         lcme->lcme_id = cpu_to_le32(id);
3404                 }
3405
3406                 id = max(le32_to_cpu(lcme->lcme_id), id);
3407         }
3408
3409         mirror_id = mirror_id_of(id) + 1;
3410
3411         /* check if first entry in new layout is DOM */
3412         lmm = (struct lov_mds_md_v1 *)((char *)merge_lcm +
3413                                         merge_lcm->lcm_entries[0].lcme_offset);
3414         merge_has_dom = lov_pattern(le32_to_cpu(lmm->lmm_pattern)) &
3415                         LOV_PATTERN_MDT;
3416
3417         for (i = 0; i < merge_entry_count; i++) {
3418                 struct lov_comp_md_entry_v1 *merge_lcme;
3419
3420                 merge_lcme = &merge_lcm->lcm_entries[i];
3421                 lcme = &lcm->lcm_entries[cur_entry_count + i];
3422
3423                 *lcme = *merge_lcme;
3424                 lcme->lcme_offset = cpu_to_le32(offset);
3425                 if (merge_has_dom && i == 0)
3426                         lcme->lcme_flags |= cpu_to_le32(LCME_FL_STALE);
3427
3428                 id = pflr_id(mirror_id, i + 1);
3429                 lcme->lcme_id = cpu_to_le32(id);
3430
3431                 memcpy((char *)lcm + offset,
3432                        (char *)merge_lcm + le32_to_cpu(merge_lcme->lcme_offset),
3433                        le32_to_cpu(lcme->lcme_size));
3434
3435                 offset += le32_to_cpu(lcme->lcme_size);
3436         }
3437
3438         /* fixup layout information */
3439         lcm->lcm_size = cpu_to_le32(size);
3440         lcm->lcm_entry_count = cpu_to_le16(cur_entry_count + merge_entry_count);
3441         lcm->lcm_mirror_count = cpu_to_le16(mirror_count);
3442         if ((le16_to_cpu(lcm->lcm_flags) & LCM_FL_FLR_MASK) == LCM_FL_NONE)
3443                 lcm->lcm_flags = cpu_to_le32(LCM_FL_RDONLY);
3444
3445         rc = lod_striping_reload(env, lo, buf, 0);
3446         if (rc)
3447                 GOTO(out, rc);
3448
3449         lod_obj_inc_layout_gen(lo);
3450         lcm->lcm_layout_gen = cpu_to_le32(lo->ldo_layout_gen);
3451
3452         /* transfer layout version to OST objects. */
3453         if (lo->ldo_mirror_count > 1) {
3454                 struct lod_obj_stripe_cb_data data = { {0} };
3455
3456                 layout_attr->la_valid = LA_LAYOUT_VERSION;
3457                 layout_attr->la_layout_version = 0;
3458                 data.locd_attr = layout_attr;
3459                 data.locd_declare = true;
3460                 data.locd_stripe_cb = lod_obj_stripe_attr_set_cb;
3461                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
3462                 if (rc)
3463                         GOTO(out, rc);
3464         }
3465
3466         rc = lod_sub_declare_xattr_set(env, dt_object_child(dt), buf,
3467                                         XATTR_NAME_LOV, LU_XATTR_REPLACE, th);
3468
3469 out:
3470         lu_buf_free(buf);
3471         RETURN(rc);
3472 }
3473
3474 /**
3475  * Split layouts, just set the LOVEA with the layout from mbuf.
3476  */
3477 static int lod_declare_layout_split(const struct lu_env *env,
3478                 struct dt_object *dt, const struct lu_buf *mbuf,
3479                 struct thandle *th)
3480 {
3481         struct lod_thread_info *info = lod_env_info(env);
3482         struct lu_attr *layout_attr = &info->lti_layout_attr;
3483         struct lod_object *lo = lod_dt_obj(dt);
3484         struct lov_comp_md_v1 *lcm = mbuf->lb_buf;
3485         int rc;
3486         ENTRY;
3487
3488         rc = lod_striping_reload(env, lo, mbuf, LVF_ALL_STALE);
3489         if (rc)
3490                 RETURN(rc);
3491
3492         lod_obj_inc_layout_gen(lo);
3493         /* fix on-disk layout gen */
3494         lcm->lcm_layout_gen = cpu_to_le32(lo->ldo_layout_gen);
3495
3496         /* transfer layout version to OST objects. */
3497         if (lo->ldo_mirror_count > 1) {
3498                 struct lod_obj_stripe_cb_data data = { {0} };
3499
3500                 layout_attr->la_valid = LA_LAYOUT_VERSION;
3501                 layout_attr->la_layout_version = 0;
3502                 data.locd_attr = layout_attr;
3503                 data.locd_declare = true;
3504                 data.locd_stripe_cb = lod_obj_stripe_attr_set_cb;
3505                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
3506                 if (rc)
3507                         RETURN(rc);
3508         }
3509
3510         rc = lod_sub_declare_xattr_set(env, dt_object_child(dt), mbuf,
3511                                        XATTR_NAME_LOV, LU_XATTR_REPLACE, th);
3512         RETURN(rc);
3513 }
3514
3515 static int lod_layout_declare_or_purge_mirror(const struct lu_env *env,
3516                         struct dt_object *dt, const struct lu_buf *buf,
3517                         struct thandle *th, bool declare)
3518 {
3519         struct lod_thread_info *info = lod_env_info(env);
3520         struct lod_device *d = lu2lod_dev(dt->do_lu.lo_dev);
3521         struct lod_object *lo = lod_dt_obj(dt);
3522         struct lov_comp_md_v1 *comp_v1 = buf->lb_buf;
3523         struct lov_comp_md_entry_v1 *entry;
3524         struct lov_mds_md_v1 *lmm;
3525         struct dt_object **sub_objs = NULL;
3526         int rc = 0, i, k, array_count = 0;
3527
3528         ENTRY;
3529
3530         /**
3531          * other ops (like lod_declare_destroy) could destroying sub objects
3532          * as well.
3533          */
3534         mutex_lock(&lo->ldo_layout_mutex);
3535
3536         if (!declare) {
3537                 /* prepare sub-objects array */
3538                 for (i = 0; i < comp_v1->lcm_entry_count; i++) {
3539                         entry = &comp_v1->lcm_entries[i];
3540
3541                         if (!(entry->lcme_flags & LCME_FL_INIT))
3542                                 continue;
3543
3544                         lmm = (struct lov_mds_md_v1 *)
3545                                         ((char *)comp_v1 + entry->lcme_offset);
3546                         array_count += lmm->lmm_stripe_count;
3547                 }
3548                 OBD_ALLOC_PTR_ARRAY(sub_objs, array_count);
3549                 if (sub_objs == NULL) {
3550                         mutex_unlock(&lo->ldo_layout_mutex);
3551                         RETURN(-ENOMEM);
3552                 }
3553         }
3554
3555         k = 0;  /* sub_objs index */
3556         for (i = 0; i < comp_v1->lcm_entry_count; i++) {
3557                 struct lov_ost_data_v1 *objs;
3558                 struct lu_object *o, *n;
3559                 struct dt_object *dto;
3560                 struct lu_device *nd;
3561                 struct lov_mds_md_v3 *v3;
3562                 __u32 idx;
3563                 int j;
3564
3565                 entry = &comp_v1->lcm_entries[i];
3566
3567                 if (!(entry->lcme_flags & LCME_FL_INIT))
3568                         continue;
3569
3570                 lmm = (struct lov_mds_md_v1 *)
3571                                 ((char *)comp_v1 + entry->lcme_offset);
3572                 v3 = (struct lov_mds_md_v3 *)lmm;
3573                 if (lmm->lmm_magic == LOV_MAGIC_V3)
3574                         objs = &v3->lmm_objects[0];
3575                 else
3576                         objs = &lmm->lmm_objects[0];
3577
3578                 for (j = 0; j < lmm->lmm_stripe_count; j++) {
3579                         idx = objs[j].l_ost_idx;
3580                         rc = ostid_to_fid(&info->lti_fid, &objs[j].l_ost_oi,
3581                                           idx);
3582                         if (rc)
3583                                 GOTO(out, rc);
3584
3585                         if (!fid_is_sane(&info->lti_fid)) {
3586                                 CERROR("%s: sub-object insane fid "DFID"\n",
3587                                        lod2obd(d)->obd_name,
3588                                        PFID(&info->lti_fid));
3589                                 GOTO(out, rc = -EINVAL);
3590                         }
3591
3592                         lod_getref(&d->lod_ost_descs);
3593
3594                         rc = validate_lod_and_idx(d, idx);
3595                         if (unlikely(rc)) {
3596                                 lod_putref(d, &d->lod_ost_descs);
3597                                 GOTO(out, rc);
3598                         }
3599
3600                         nd = &OST_TGT(d, idx)->ltd_tgt->dd_lu_dev;
3601                         lod_putref(d, &d->lod_ost_descs);
3602
3603                         o = lu_object_find_at(env, nd, &info->lti_fid, NULL);
3604                         if (IS_ERR(o))
3605                                 GOTO(out, rc = PTR_ERR(o));
3606
3607                         n = lu_object_locate(o->lo_header, nd->ld_type);
3608                         if (unlikely(!n)) {
3609                                 lu_object_put(env, n);
3610                                 GOTO(out, rc = -ENOENT);
3611                         }
3612
3613                         dto = container_of(n, struct dt_object, do_lu);
3614
3615                         if (declare) {
3616                                 rc = lod_sub_declare_destroy(env, dto, th);
3617                                 dt_object_put(env, dto);
3618                                 if (rc)
3619                                         GOTO(out, rc);
3620                         } else {
3621                                 /**
3622                                  * collect to-be-destroyed sub objects, the
3623                                  * reference would be released after actual
3624                                  * deletion.
3625                                  */
3626                                 sub_objs[k] = dto;
3627                                 k++;
3628                         }
3629                 } /* for each stripe */
3630         } /* for each component in the mirror */
3631 out:
3632         if (!declare) {
3633                 i = 0;
3634                 if (!rc) {
3635                         /* destroy the sub objects */
3636                         for (; i < k; i++) {
3637                                 rc = lod_sub_destroy(env, sub_objs[i], th);
3638                                 if (rc)
3639                                         break;
3640                                 dt_object_put(env, sub_objs[i]);
3641                         }
3642                 }
3643                 /**
3644                  * if a sub object destroy failed, we'd release sub objects
3645                  * reference get from above sub_objs collection.
3646                  */
3647                 for (; i < k; i++)
3648                         dt_object_put(env, sub_objs[i]);
3649
3650                 OBD_FREE_PTR_ARRAY(sub_objs, array_count);
3651         }
3652         mutex_unlock(&lo->ldo_layout_mutex);
3653
3654         RETURN(rc);
3655 }
3656
3657 /**
3658  * Purge layouts, delete sub objects in the mirror stored in the vic_buf,
3659  * and set the LOVEA with the layout from mbuf.
3660  */
3661 static int lod_declare_layout_purge(const struct lu_env *env,
3662                 struct dt_object *dt, const struct lu_buf *buf,
3663                 struct thandle *th)
3664 {
3665         struct lod_device *d = lu2lod_dev(dt->do_lu.lo_dev);
3666         struct lov_comp_md_v1 *comp_v1 = buf->lb_buf;
3667         int rc;
3668
3669         ENTRY;
3670
3671         if (le32_to_cpu(comp_v1->lcm_magic) != LOV_MAGIC_COMP_V1) {
3672                 CERROR("%s: invalid layout magic %#x != %#x\n",
3673                        lod2obd(d)->obd_name, le32_to_cpu(comp_v1->lcm_magic),
3674                        LOV_MAGIC_COMP_V1);
3675                 RETURN(-EINVAL);
3676         }
3677
3678         if (cpu_to_le32(LOV_MAGIC_COMP_V1) != LOV_MAGIC_COMP_V1)
3679                 lustre_swab_lov_comp_md_v1(comp_v1);
3680
3681         /* from now on, @buf contains cpu endian data */
3682
3683         if (comp_v1->lcm_mirror_count != 0) {
3684                 CERROR("%s: can only purge one mirror from "DFID"\n",
3685                        lod2obd(d)->obd_name, PFID(lu_object_fid(&dt->do_lu)));
3686                 RETURN(-EINVAL);
3687         }
3688
3689         /* delcare sub objects deletion in the mirror stored in @buf */
3690         rc = lod_layout_declare_or_purge_mirror(env, dt, buf, th, true);
3691         RETURN(rc);
3692 }
3693
3694 /* delete sub objects from the mirror stored in @buf */
3695 static int lod_layout_purge(const struct lu_env *env, struct dt_object *dt,
3696                             const struct lu_buf *buf, struct thandle *th)
3697 {
3698         int rc;
3699
3700         ENTRY;
3701         rc = lod_layout_declare_or_purge_mirror(env, dt, buf, th, false);
3702         RETURN(rc);
3703 }
3704
3705 /**
3706  * Implementation of dt_object_operations::do_declare_xattr_set.
3707  *
3708  * \see dt_object_operations::do_declare_xattr_set() in the API description
3709  * for details.
3710  *
3711  * the extension to the API:
3712  *   - declaring LOVEA requests striping creation
3713  *   - LU_XATTR_REPLACE means layout swap
3714  */
3715 static int lod_declare_xattr_set(const struct lu_env *env,
3716                                  struct dt_object *dt,
3717                                  const struct lu_buf *buf,
3718                                  const char *name, int fl,
3719                                  struct thandle *th)
3720 {
3721         struct dt_object *next = dt_object_child(dt);
3722         struct lu_attr   *attr = &lod_env_info(env)->lti_attr;
3723         __u32             mode;
3724         int               rc;
3725         ENTRY;
3726
3727         mode = dt->do_lu.lo_header->loh_attr & S_IFMT;
3728         if ((S_ISREG(mode) || mode == 0) &&
3729             !(fl & (LU_XATTR_REPLACE | LU_XATTR_MERGE | LU_XATTR_SPLIT |
3730                     LU_XATTR_PURGE)) &&
3731             (strcmp(name, XATTR_NAME_LOV) == 0 ||
3732              strcmp(name, XATTR_LUSTRE_LOV) == 0)) {
3733                 /*
3734                  * this is a request to create object's striping.
3735                  *
3736                  * allow to declare predefined striping on a new (!mode) object
3737                  * which is supposed to be replay of regular file creation
3738                  * (when LOV setting is declared)
3739                  *
3740                  * LU_XATTR_REPLACE is set to indicate a layout swap
3741                  */
3742                 if (dt_object_exists(dt)) {
3743                         rc = dt_attr_get(env, next, attr);
3744                         if (rc)
3745                                 RETURN(rc);
3746                 } else {
3747                         memset(attr, 0, sizeof(*attr));
3748                         attr->la_valid = LA_TYPE | LA_MODE;
3749                         attr->la_mode = S_IFREG;
3750                 }
3751                 rc = lod_declare_striped_create(env, dt, attr, buf, th);
3752         } else if (fl & LU_XATTR_MERGE) {
3753                 LASSERT(strcmp(name, XATTR_NAME_LOV) == 0 ||
3754                         strcmp(name, XATTR_LUSTRE_LOV) == 0);
3755                 rc = lod_declare_layout_merge(env, dt, buf, th);
3756         } else if (fl & LU_XATTR_SPLIT) {
3757                 LASSERT(strcmp(name, XATTR_NAME_LOV) == 0 ||
3758                         strcmp(name, XATTR_LUSTRE_LOV) == 0);
3759                 rc = lod_declare_layout_split(env, dt, buf, th);
3760         } else if (fl & LU_XATTR_PURGE) {
3761                 LASSERT(strcmp(name, XATTR_NAME_LOV) == 0 ||
3762                         strcmp(name, XATTR_LUSTRE_LOV) == 0);
3763                 rc = lod_declare_layout_purge(env, dt, buf, th);
3764         } else if (S_ISREG(mode) &&
3765                    strlen(name) >= sizeof(XATTR_LUSTRE_LOV) + 3 &&
3766                    allowed_lustre_lov(name)) {
3767                 /*
3768                  * this is a request to modify object's striping.
3769                  * add/set/del component(s).
3770                  */
3771                 if (!dt_object_exists(dt))
3772                         RETURN(-ENOENT);
3773
3774                 rc = lod_declare_modify_layout(env, dt, name, buf, th);
3775         } else if (S_ISDIR(mode)) {
3776                 rc = lod_dir_declare_xattr_set(env, dt, buf, name, fl, th);
3777         } else if (strcmp(name, XATTR_NAME_FID) == 0) {
3778                 rc = lod_replace_parent_fid(env, dt, buf, th, true);
3779         } else {
3780                 rc = lod_sub_declare_xattr_set(env, next, buf, name, fl, th);
3781         }
3782
3783         RETURN(rc);
3784 }
3785
3786 /**
3787  * Apply xattr changes to the object.
3788  *
3789  * Applies xattr changes to the object and the stripes if the latter exist.
3790  *
3791  * \param[in] env       execution environment
3792  * \param[in] dt        object
3793  * \param[in] buf       buffer pointing to the new value of xattr
3794  * \param[in] name      name of xattr
3795  * \param[in] fl        flags
3796  * \param[in] th        transaction handle
3797  *
3798  * \retval              0 on success
3799  * \retval              negative if failed
3800  */
3801 static int lod_xattr_set_internal(const struct lu_env *env,
3802                                   struct dt_object *dt,
3803                                   const struct lu_buf *buf,
3804                                   const char *name, int fl,
3805                                   struct thandle *th)
3806 {
3807         struct dt_object        *next = dt_object_child(dt);
3808         struct lod_object       *lo = lod_dt_obj(dt);
3809         int                     rc;
3810         int                     i;
3811         ENTRY;
3812
3813         rc = lod_sub_xattr_set(env, next, buf, name, fl, th);
3814         if (rc != 0 || !S_ISDIR(dt->do_lu.lo_header->loh_attr))
3815                 RETURN(rc);
3816
3817         /* Note: Do not set LinkEA on sub-stripes, otherwise
3818          * it will confuse the fid2path process(see mdt_path_current()).
3819          * The linkEA between master and sub-stripes is set in
3820          * lod_xattr_set_lmv(). */
3821         if (lo->ldo_dir_stripe_count == 0 || strcmp(name, XATTR_NAME_LINK) == 0)
3822                 RETURN(0);
3823
3824         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
3825                 if (!lo->ldo_stripe[i])
3826                         continue;
3827
3828                 if (!dt_object_exists(lo->ldo_stripe[i]))
3829                         continue;
3830
3831                 rc = lod_sub_xattr_set(env, lo->ldo_stripe[i], buf, name,
3832                                        fl, th);
3833                 if (rc != 0)
3834                         break;
3835         }
3836
3837         RETURN(rc);
3838 }
3839
3840 /**
3841  * Delete an extended attribute.
3842  *
3843  * Deletes specified xattr from the object and the stripes if the latter exist.
3844  *
3845  * \param[in] env       execution environment
3846  * \param[in] dt        object
3847  * \param[in] name      name of xattr
3848  * \param[in] th        transaction handle
3849  *
3850  * \retval              0 on success
3851  * \retval              negative if failed
3852  */
3853 static int lod_xattr_del_internal(const struct lu_env *env,
3854                                   struct dt_object *dt,
3855                                   const char *name, struct thandle *th)
3856 {
3857         struct dt_object *next = dt_object_child(dt);
3858         struct lod_object *lo = lod_dt_obj(dt);
3859         int i;
3860         int rc;
3861
3862         ENTRY;
3863
3864         rc = lod_sub_xattr_del(env, next, name, th);
3865         if (rc != 0 || !S_ISDIR(dt->do_lu.lo_header->loh_attr))
3866                 RETURN(rc);
3867
3868         if (lo->ldo_dir_stripe_count == 0)
3869                 RETURN(rc);
3870
3871         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
3872                 if (!lo->ldo_stripe[i])
3873                         continue;
3874
3875                 if (!dt_object_exists(lo->ldo_stripe[i]))
3876                         continue;
3877
3878                 rc = lod_sub_xattr_del(env, lo->ldo_stripe[i], name, th);
3879                 if (rc != 0)
3880                         break;
3881         }
3882
3883         RETURN(rc);
3884 }
3885
3886 /**
3887  * Set default striping on a directory.
3888  *
3889  * Sets specified striping on a directory object unless it matches the default
3890  * striping (LOVEA_DELETE_VALUES() macro). In the latter case remove existing
3891  * EA. This striping will be used when regular file is being created in this
3892  * directory.
3893  *
3894  * \param[in] env       execution environment
3895  * \param[in] dt        the striped object
3896  * \param[in] buf       buffer with the striping
3897  * \param[in] name      name of EA
3898  * \param[in] fl        xattr flag (see OSD API description)
3899  * \param[in] th        transaction handle
3900  *
3901  * \retval              0 on success
3902  * \retval              negative if failed
3903  */
3904 static int lod_xattr_set_lov_on_dir(const struct lu_env *env,
3905                                     struct dt_object *dt,
3906                                     const struct lu_buf *buf,
3907                                     const char *name, int fl,
3908                                     struct thandle *th)
3909 {
3910         struct lov_user_md_v1   *lum;
3911         struct lov_user_md_v3   *v3 = NULL;
3912         const char              *pool_name = NULL;
3913         int                      rc;
3914         bool                     is_del;
3915         ENTRY;
3916
3917         LASSERT(buf != NULL && buf->lb_buf != NULL);
3918         lum = buf->lb_buf;
3919
3920         switch (lum->lmm_magic) {
3921         case LOV_USER_MAGIC_SPECIFIC:
3922         case LOV_USER_MAGIC_V3:
3923                 v3 = buf->lb_buf;
3924                 if (lov_pool_is_reserved(v3->lmm_pool_name))
3925                         memset(v3->lmm_pool_name, 0, sizeof(v3->lmm_pool_name));
3926                 else if (v3->lmm_pool_name[0] != '\0')
3927                         pool_name = v3->lmm_pool_name;
3928                 fallthrough;
3929         case LOV_USER_MAGIC_V1:
3930                 /* if { size, offset, count } = { 0, -1, 0 } and no pool
3931                  * (i.e. all default values specified) then delete default
3932                  * striping from dir. */
3933                 CDEBUG(D_LAYOUT,
3934                        "set default striping: sz %u # %u offset %d %s %s\n",
3935                        (unsigned)lum->lmm_stripe_size,
3936                        (unsigned)lum->lmm_stripe_count,
3937                        (int)lum->lmm_stripe_offset,
3938                        v3 ? "from" : "", v3 ? v3->lmm_pool_name : "");
3939
3940                 is_del = LOVEA_DELETE_VALUES(lum->lmm_stripe_size,
3941                                              lum->lmm_stripe_count,
3942                                              lum->lmm_stripe_offset,
3943                                              pool_name);
3944                 break;
3945         case LOV_USER_MAGIC_COMP_V1:
3946         {
3947                 struct lov_comp_md_v1 *lcm = (struct lov_comp_md_v1 *)lum;
3948                 struct lov_comp_md_entry_v1 *lcme;
3949                 int i, comp_cnt;
3950
3951                 comp_cnt = le16_to_cpu(lcm->lcm_entry_count);
3952                 for (i = 0; i < comp_cnt; i++) {
3953                         lcme = &lcm->lcm_entries[i];
3954                         if (lcme->lcme_flags & cpu_to_le32(LCME_FL_EXTENSION)) {
3955                                 lcm->lcm_magic = cpu_to_le32(LOV_MAGIC_SEL);
3956                                 break;
3957                         }
3958                 }
3959
3960                 is_del = false;
3961                 break;
3962         }
3963         default:
3964                 CERROR("Invalid magic %x\n", lum->lmm_magic);
3965                 RETURN(-EINVAL);
3966         }
3967
3968         if (is_del) {
3969                 rc = lod_xattr_del_internal(env, dt, name, th);
3970                 if (rc == -ENODATA)
3971                         rc = 0;
3972         } else {
3973                 rc = lod_xattr_set_internal(env, dt, buf, name, fl, th);
3974         }
3975
3976         RETURN(rc);
3977 }
3978
3979 static int lod_get_default_lov_striping(const struct lu_env *env,
3980                                        struct lod_object *lo,
3981                                        struct lod_default_striping *lds,
3982                                        struct dt_allocation_hint *ah);
3983
3984 /**
3985  * Helper function to convert compound layout to compound layout with
3986  * pool
3987  *
3988  * Copy lcm_entries array of \a src to \a tgt. Replace lov_user_md_v1
3989  * components of \a src with lov_user_md_v3 using \a pool.
3990  *
3991  * \param[in] src       source layout
3992  * \param[in] pool      pool to use in \a tgt
3993  * \param[out] tgt      target layout
3994  */
3995 static void embed_pool_to_comp_v1(const struct lov_comp_md_v1 *src,
3996                                   const char *pool,
3997                                   struct lov_comp_md_v1 *tgt)
3998 {
3999         size_t shift;
4000         struct lov_user_md_v1 *lum;
4001         struct lov_user_md_v3 *lum3;
4002         struct lov_comp_md_entry_v1 *entry;
4003         int i;
4004         __u32 offset;
4005
4006         entry = tgt->lcm_entries;
4007         shift = 0;
4008         for (i = 0; i < le16_to_cpu(src->lcm_entry_count); i++, entry++) {
4009                 *entry = src->lcm_entries[i];
4010                 offset = le32_to_cpu(src->lcm_entries[i].lcme_offset);
4011                 entry->lcme_offset = cpu_to_le32(offset + shift);
4012
4013                 lum = (struct lov_user_md_v1 *)((char *)src + offset);
4014                 lum3 = (struct lov_user_md_v3 *)((char *)tgt + offset + shift);
4015                 *(struct lov_user_md_v1 *)lum3 = *lum;
4016                 if (lum->lmm_pattern & cpu_to_le32(LOV_PATTERN_MDT)) {
4017                         lum3->lmm_magic = cpu_to_le32(LOV_USER_MAGIC_V1);
4018                 } else {
4019                         lum3->lmm_magic = cpu_to_le32(LOV_USER_MAGIC_V3);
4020                         entry->lcme_size = cpu_to_le32(sizeof(*lum3));
4021                         strlcpy(lum3->lmm_pool_name, pool,
4022                                 sizeof(lum3->lmm_pool_name));
4023                         shift += sizeof(*lum3) - sizeof(*lum);
4024                 }
4025         }
4026 }
4027
4028 /**
4029  * Set default striping on a directory.
4030  *
4031  * Sets specified striping on a directory object unless it matches the default
4032  * striping (LOVEA_DELETE_VALUES() macro). In the latter case remove existing
4033  * EA. This striping will be used when regular file is being created in this
4034  * directory.
4035  * If current default striping includes a pool but specifed striping
4036  * does not - retain the pool if it exists.
4037  *
4038  * \param[in] env       execution environment
4039  * \param[in] dt        the striped object
4040  * \param[in] buf       buffer with the striping
4041  * \param[in] name      name of EA
4042  * \param[in] fl        xattr flag (see OSD API description)
4043  * \param[in] th        transaction handle
4044  *
4045  * \retval              0 on success
4046  * \retval              negative if failed
4047  */
4048 static int lod_xattr_set_default_lov_on_dir(const struct lu_env *env,
4049                                             struct dt_object *dt,
4050                                             const struct lu_buf *buf,
4051                                             const char *name, int fl,
4052                                             struct thandle *th)
4053 {
4054         struct lod_default_striping     *lds = lod_lds_buf_get(env);
4055         struct lov_user_md_v1           *v1 = buf->lb_buf;
4056         char                             pool[LOV_MAXPOOLNAME + 1];
4057         bool                             is_del;
4058         int                              rc;
4059
4060         ENTRY;
4061
4062         /* get existing striping config */
4063         rc = lod_get_default_lov_striping(env, lod_dt_obj(dt), lds, NULL);
4064         if (rc)
4065                 RETURN(rc);
4066
4067         memset(pool, 0, sizeof(pool));
4068         if (lds->lds_def_striping_set == 1)
4069                 lod_layout_get_pool(lds->lds_def_comp_entries,
4070                                     lds->lds_def_comp_cnt, pool,
4071                                     sizeof(pool));
4072
4073         is_del = LOVEA_DELETE_VALUES(v1->lmm_stripe_size,
4074                                      v1->lmm_stripe_count,
4075                                      v1->lmm_stripe_offset,
4076                                      NULL);
4077
4078         /* Retain the pool name if it is not given */
4079         if (v1->lmm_magic == LOV_USER_MAGIC_V1 && pool[0] != '\0' &&
4080             !is_del) {
4081                 struct lod_thread_info *info = lod_env_info(env);
4082                 struct lov_user_md_v3 *v3  = info->lti_ea_store;
4083
4084                 memset(v3, 0, sizeof(*v3));
4085                 v3->lmm_magic = cpu_to_le32(LOV_USER_MAGIC_V3);
4086                 v3->lmm_pattern = cpu_to_le32(v1->lmm_pattern);
4087                 v3->lmm_stripe_count = cpu_to_le32(v1->lmm_stripe_count);
4088                 v3->lmm_stripe_offset = cpu_to_le32(v1->lmm_stripe_offset);
4089                 v3->lmm_stripe_size = cpu_to_le32(v1->lmm_stripe_size);
4090
4091                 strlcpy(v3->lmm_pool_name, pool, sizeof(v3->lmm_pool_name));
4092
4093                 info->lti_buf.lb_buf = v3;
4094                 info->lti_buf.lb_len = sizeof(*v3);
4095                 rc = lod_xattr_set_lov_on_dir(env, dt, &info->lti_buf,
4096                                               name, fl, th);
4097         } else if (v1->lmm_magic == LOV_USER_MAGIC_COMP_V1 &&
4098                    pool[0] != '\0' && !is_del) {
4099                 /*
4100                  * try to retain the pool from default layout if the
4101                  * specified component layout does not provide pool
4102                  * info explicitly
4103                  */
4104                 struct lod_thread_info *info = lod_env_info(env);
4105                 struct lov_comp_md_v1 *comp_v1 = buf->lb_buf;
4106                 struct lov_comp_md_v1 *comp_v1p;
4107                 struct lov_user_md_v1 *lum;
4108                 int entry_count;
4109                 int i;
4110                 __u32 offset;
4111                 struct lov_comp_md_entry_v1 *entry;
4112                 int size;
4113
4114                 entry_count = le16_to_cpu(comp_v1->lcm_entry_count);
4115                 size = sizeof(*comp_v1) +
4116                         entry_count * sizeof(comp_v1->lcm_entries[0]);
4117                 entry = comp_v1->lcm_entries;
4118                 for (i = 0; i < entry_count; i++, entry++) {
4119                         offset = le32_to_cpu(entry->lcme_offset);
4120                         lum = (struct lov_user_md_v1 *)((char *)comp_v1 +
4121                                                         offset);
4122                         if (le32_to_cpu(lum->lmm_magic) != LOV_USER_MAGIC_V1)
4123                                 /* the i-th component includes pool info */
4124                                 break;
4125                         if (lum->lmm_pattern & cpu_to_le32(LOV_PATTERN_MDT))
4126                                 size += sizeof(struct lov_user_md_v1);
4127                         else
4128                                 size += sizeof(struct lov_user_md_v3);
4129                 }
4130
4131                 if (i == entry_count) {
4132                         /*
4133                          * re-compose the layout to include the pool for
4134                          * each component
4135                          */
4136                         if (info->lti_ea_store_size < size)
4137                                 rc = lod_ea_store_resize(info, size);
4138
4139                         if (rc == 0) {
4140                                 comp_v1p = info->lti_ea_store;
4141                                 *comp_v1p = *comp_v1;
4142                                 comp_v1p->lcm_size = cpu_to_le32(size);
4143                                 embed_pool_to_comp_v1(comp_v1, pool, comp_v1p);
4144
4145                                 info->lti_buf.lb_buf = comp_v1p;
4146                                 info->lti_buf.lb_len = size;
4147                                 rc = lod_xattr_set_lov_on_dir(env, dt,
4148                                                               &info->lti_buf,
4149                                                               name, fl, th);
4150                         }
4151                 } else {
4152                         rc = lod_xattr_set_lov_on_dir(env, dt, buf, name, fl,
4153                                                       th);
4154                 }
4155         } else {
4156                 rc = lod_xattr_set_lov_on_dir(env, dt, buf, name, fl, th);
4157         }
4158
4159         if (lds->lds_def_striping_set == 1 && lds->lds_def_comp_entries != NULL)
4160                 lod_free_def_comp_entries(lds);
4161
4162         RETURN(rc);
4163 }
4164
4165 /**
4166  * Set default striping on a directory object.
4167  *
4168  * Sets specified striping on a directory object unless it matches the default
4169  * striping (LOVEA_DELETE_VALUES() macro). In the latter case remove existing
4170  * EA. This striping will be used when a new directory is being created in the
4171  * directory.
4172  *
4173  * \param[in] env       execution environment
4174  * \param[in] dt        the striped object
4175  * \param[in] buf       buffer with the striping
4176  * \param[in] name      name of EA
4177  * \param[in] fl        xattr flag (see OSD API description)
4178  * \param[in] th        transaction handle
4179  *
4180  * \retval              0 on success
4181  * \retval              negative if failed
4182  */
4183 static int lod_xattr_set_default_lmv_on_dir(const struct lu_env *env,
4184                                             struct dt_object *dt,
4185                                             const struct lu_buf *buf,
4186                                             const char *name, int fl,
4187                                             struct thandle *th)
4188 {
4189         struct lmv_user_md_v1 *lum;
4190         int rc;
4191
4192         ENTRY;
4193
4194         LASSERT(buf != NULL && buf->lb_buf != NULL);
4195         lum = buf->lb_buf;
4196
4197         CDEBUG(D_INFO,
4198                "set default stripe_count # %u stripe_offset %d hash %u\n",
4199               le32_to_cpu(lum->lum_stripe_count),
4200               (int)le32_to_cpu(lum->lum_stripe_offset),
4201               le32_to_cpu(lum->lum_hash_type));
4202
4203         if (LMVEA_DELETE_VALUES((le32_to_cpu(lum->lum_stripe_count)),
4204                                  le32_to_cpu(lum->lum_stripe_offset)) &&
4205             le32_to_cpu(lum->lum_magic) == LMV_USER_MAGIC) {
4206                 rc = lod_xattr_del_internal(env, dt, name, th);
4207                 if (rc == -ENODATA)
4208                         rc = 0;
4209         } else {
4210                 rc = lod_xattr_set_internal(env, dt, buf, name, fl, th);
4211                 if (rc != 0)
4212                         RETURN(rc);
4213         }
4214
4215         RETURN(rc);
4216 }
4217
4218 /**
4219  * Turn directory into a striped directory.
4220  *
4221  * During replay the client sends the striping created before MDT
4222  * failure, then the layer above LOD sends this defined striping
4223  * using ->do_xattr_set(), so LOD uses this method to replay creation
4224  * of the stripes. Notice the original information for the striping
4225  * (#stripes, FIDs, etc) was transferred in declare path.
4226  *
4227  * \param[in] env       execution environment
4228  * \param[in] dt        the striped object
4229  * \param[in] buf       not used currently
4230  * \param[in] name      not used currently
4231  * \param[in] fl        xattr flag (see OSD API description)
4232  * \param[in] th        transaction handle
4233  *
4234  * \retval              0 on success
4235  * \retval              negative if failed
4236  */
4237 static int lod_xattr_set_lmv(const struct lu_env *env, struct dt_object *dt,
4238                              const struct lu_buf *buf, const char *name,
4239                              int fl, struct thandle *th)
4240 {
4241         struct lod_object       *lo = lod_dt_obj(dt);
4242         struct lod_thread_info  *info = lod_env_info(env);
4243         struct lu_attr          *attr = &info->lti_attr;
4244         struct dt_object_format *dof = &info->lti_format;
4245         struct lu_buf           lmv_buf;
4246         struct lu_buf           slave_lmv_buf;
4247         struct lmv_mds_md_v1    *lmm;
4248         struct lmv_mds_md_v1    *slave_lmm = NULL;
4249         struct dt_insert_rec    *rec = &info->lti_dt_rec;
4250         int                     i;
4251         int                     rc;
4252         ENTRY;
4253
4254         if (!S_ISDIR(dt->do_lu.lo_header->loh_attr))
4255                 RETURN(-ENOTDIR);
4256
4257         /* The stripes are supposed to be allocated in declare phase,
4258          * if there are no stripes being allocated, it will skip */
4259         if (lo->ldo_dir_stripe_count == 0) {
4260                 if (lo->ldo_is_foreign) {
4261                         rc = lod_sub_xattr_set(env, dt_object_child(dt), buf,
4262                                                XATTR_NAME_LMV, fl, th);
4263                         if (rc != 0)
4264                                 RETURN(rc);
4265                 }
4266                 RETURN(0);
4267         }
4268
4269         rc = dt_attr_get(env, dt_object_child(dt), attr);
4270         if (rc != 0)
4271                 RETURN(rc);
4272
4273         attr->la_valid = LA_ATIME | LA_MTIME | LA_CTIME | LA_FLAGS |
4274                          LA_MODE | LA_UID | LA_GID | LA_TYPE | LA_PROJID;
4275         dof->dof_type = DFT_DIR;
4276
4277         rc = lod_prep_lmv_md(env, dt, &lmv_buf);
4278         if (rc != 0)
4279                 RETURN(rc);
4280         lmm = lmv_buf.lb_buf;
4281
4282         OBD_ALLOC_PTR(slave_lmm);
4283         if (slave_lmm == NULL)
4284                 RETURN(-ENOMEM);
4285
4286         lod_prep_slave_lmv_md(slave_lmm, lmm);
4287         slave_lmv_buf.lb_buf = slave_lmm;
4288         slave_lmv_buf.lb_len = sizeof(*slave_lmm);
4289
4290         rec->rec_type = S_IFDIR;
4291         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
4292                 struct dt_object *dto = lo->ldo_stripe[i];
4293                 char *stripe_name = info->lti_key;
4294                 struct lu_name *sname;
4295                 struct linkea_data ldata = { NULL };
4296                 struct lu_buf linkea_buf;
4297
4298                 /* OBD_FAIL_MDS_STRIPE_FID may leave stripe uninitialized */
4299                 if (!dto)
4300                         continue;
4301
4302                 /* fail a remote stripe creation */
4303                 if (i && CFS_FAIL_CHECK(OBD_FAIL_MDS_STRIPE_CREATE))
4304                         continue;
4305
4306                 /* don't create stripe if:
4307                  * 1. it's source stripe of migrating directory
4308                  * 2. it's existed stripe of splitting directory
4309                  */
4310                 if ((lod_is_migrating(lo) && i >= lo->ldo_dir_migrate_offset) ||
4311                     (lod_is_splitting(lo) && i < lo->ldo_dir_split_offset)) {
4312                         if (!dt_object_exists(dto))
4313                                 GOTO(out, rc = -EINVAL);
4314                 } else {
4315                         dt_write_lock(env, dto, DT_TGT_CHILD);
4316                         rc = lod_sub_create(env, dto, attr, NULL, dof, th);
4317                         if (rc != 0) {
4318                                 dt_write_unlock(env, dto);
4319                                 GOTO(out, rc);
4320                         }
4321
4322                         rc = lod_sub_ref_add(env, dto, th);
4323                         dt_write_unlock(env, dto);
4324                         if (rc != 0)
4325                                 GOTO(out, rc);
4326
4327                         rec->rec_fid = lu_object_fid(&dto->do_lu);
4328                         rc = lod_sub_insert(env, dto,
4329                                             (const struct dt_rec *)rec,
4330                                             (const struct dt_key *)dot, th);
4331                         if (rc != 0)
4332                                 GOTO(out, rc);
4333                 }
4334
4335                 if (!CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_SLAVE_LMV) ||
4336                     cfs_fail_val != i) {
4337                         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_SLAVE_LMV) &&
4338                             cfs_fail_val == i)
4339                                 slave_lmm->lmv_master_mdt_index =
4340                                                         cpu_to_le32(i + 1);
4341                         else
4342                                 slave_lmm->lmv_master_mdt_index =
4343                                                         cpu_to_le32(i);
4344
4345                         rc = lod_sub_xattr_set(env, dto, &slave_lmv_buf,
4346                                                XATTR_NAME_LMV, 0, th);
4347                         if (rc != 0)
4348                                 GOTO(out, rc);
4349                 }
4350
4351                 /* don't insert stripe if it's existed stripe of splitting
4352                  * directory (this directory is striped).
4353                  * NB, plain directory will insert itself as the first
4354                  * stripe in target.
4355                  */
4356                 if (lod_is_splitting(lo) && lo->ldo_dir_split_offset > 1 &&
4357                     lo->ldo_dir_split_offset > i)
4358                         continue;
4359
4360                 rec->rec_fid = lu_object_fid(&dt->do_lu);
4361                 rc = lod_sub_insert(env, dto, (struct dt_rec *)rec,
4362                                     (const struct dt_key *)dotdot, th);
4363                 if (rc != 0)
4364                         GOTO(out, rc);
4365
4366                 if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_SLAVE_NAME) &&
4367                     cfs_fail_val == i)
4368                         snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
4369                                  PFID(lu_object_fid(&dto->do_lu)), i + 1);
4370                 else
4371                         snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
4372                                  PFID(lu_object_fid(&dto->do_lu)), i);
4373
4374                 sname = lod_name_get(env, stripe_name, strlen(stripe_name));
4375                 rc = linkea_links_new(&ldata, &info->lti_linkea_buf,
4376                                       sname, lu_object_fid(&dt->do_lu));
4377                 if (rc != 0)
4378                         GOTO(out, rc);
4379
4380                 linkea_buf.lb_buf = ldata.ld_buf->lb_buf;
4381                 linkea_buf.lb_len = ldata.ld_leh->leh_len;
4382                 rc = lod_sub_xattr_set(env, dto, &linkea_buf,
4383                                        XATTR_NAME_LINK, 0, th);
4384                 if (rc != 0)
4385                         GOTO(out, rc);
4386
4387                 rec->rec_fid = lu_object_fid(&dto->do_lu);
4388                 rc = lod_sub_insert(env, dt_object_child(dt),
4389                                     (const struct dt_rec *)rec,
4390                                     (const struct dt_key *)stripe_name, th);
4391                 if (rc != 0)
4392                         GOTO(out, rc);
4393
4394                 rc = lod_sub_ref_add(env, dt_object_child(dt), th);
4395                 if (rc != 0)
4396                         GOTO(out, rc);
4397         }
4398
4399         if (!CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_MASTER_LMV))
4400                 rc = lod_sub_xattr_set(env, dt_object_child(dt),
4401                                        &lmv_buf, XATTR_NAME_LMV, fl, th);
4402 out:
4403         if (slave_lmm != NULL)
4404                 OBD_FREE_PTR(slave_lmm);
4405
4406         RETURN(rc);
4407 }
4408
4409 /**
4410  * Helper function to declare/execute creation of a striped directory
4411  *
4412  * Called in declare/create object path, prepare striping for a directory
4413  * and prepare defaults data striping for the objects to be created in
4414  * that directory. Notice the function calls "declaration" or "execution"
4415  * methods depending on \a declare param. This is a consequence of the
4416  * current approach while we don't have natural distributed transactions:
4417  * we basically execute non-local updates in the declare phase. So, the
4418  * arguments for the both phases are the same and this is the reason for
4419  * this function to exist.
4420  *
4421  * \param[in] env       execution environment
4422  * \param[in] dt        object
4423  * \param[in] attr      attributes the stripes will be created with
4424  * \param[in] lmu       lmv_user_md if MDT indices are specified
4425  * \param[in] dof       format of stripes (see OSD API description)
4426  * \param[in] th        transaction handle
4427  * \param[in] declare   where to call "declare" or "execute" methods
4428  *
4429  * \retval              0 on success
4430  * \retval              negative if failed
4431  */
4432 static int lod_dir_striping_create_internal(const struct lu_env *env,
4433                                             struct dt_object *dt,
4434                                             struct lu_attr *attr,
4435                                             const struct lu_buf *lmu,
4436                                             struct dt_object_format *dof,
4437                                             struct thandle *th,
4438                                             bool declare)
4439 {
4440         struct lod_thread_info *info = lod_env_info(env);
4441         struct lod_object *lo = lod_dt_obj(dt);
4442         const struct lod_default_striping *lds = lo->ldo_def_striping;
4443         int rc;
4444         ENTRY;
4445
4446         LASSERT(ergo(lds != NULL,
4447                      lds->lds_def_striping_set ||
4448                      lds->lds_dir_def_striping_set));
4449
4450         if (!LMVEA_DELETE_VALUES(lo->ldo_dir_stripe_count,
4451                                  lo->ldo_dir_stripe_offset)) {
4452                 if (!lmu) {
4453                         struct lmv_user_md_v1 *v1 = info->lti_ea_store;
4454                         int stripe_count = lo->ldo_dir_stripe_count;
4455
4456                         if (info->lti_ea_store_size < sizeof(*v1)) {
4457                                 rc = lod_ea_store_resize(info, sizeof(*v1));
4458                                 if (rc != 0)
4459                                         RETURN(rc);
4460                                 v1 = info->lti_ea_store;
4461                         }
4462
4463                         memset(v1, 0, sizeof(*v1));
4464                         v1->lum_magic = cpu_to_le32(LMV_USER_MAGIC);
4465                         v1->lum_stripe_count = cpu_to_le32(stripe_count);
4466                         v1->lum_stripe_offset =
4467                                         cpu_to_le32(lo->ldo_dir_stripe_offset);
4468
4469                         info->lti_buf.lb_buf = v1;
4470                         info->lti_buf.lb_len = sizeof(*v1);
4471                         lmu = &info->lti_buf;
4472                 }
4473
4474                 if (declare)
4475                         rc = lod_declare_xattr_set_lmv(env, dt, attr, lmu, dof,
4476                                                        th);
4477                 else
4478                         rc = lod_xattr_set_lmv(env, dt, lmu, XATTR_NAME_LMV, 0,
4479                                                th);
4480                 if (rc != 0)
4481                         RETURN(rc);
4482         } else {
4483                 /* foreign LMV EA case */
4484                 if (lmu) {
4485                         struct lmv_foreign_md *lfm = lmu->lb_buf;
4486
4487                         if (lfm->lfm_magic == LMV_MAGIC_FOREIGN) {
4488                                 rc = lod_declare_xattr_set_lmv(env, dt, attr,
4489                                                                lmu, dof, th);
4490                         }
4491                 } else {
4492                         if (lo->ldo_is_foreign) {
4493                                 LASSERT(lo->ldo_foreign_lmv != NULL &&
4494                                         lo->ldo_foreign_lmv_size > 0);
4495                                 info->lti_buf.lb_buf = lo->ldo_foreign_lmv;
4496                                 info->lti_buf.lb_len = lo->ldo_foreign_lmv_size;
4497                                 lmu = &info->lti_buf;
4498                                 rc = lod_xattr_set_lmv(env, dt, lmu,
4499                                                        XATTR_NAME_LMV, 0, th);
4500                         }
4501                 }
4502         }
4503
4504         /* Transfer default LMV striping from the parent */
4505         if (lds != NULL && lds->lds_dir_def_striping_set &&
4506             lds->lds_dir_def_max_inherit != LMV_INHERIT_END &&
4507             lds->lds_dir_def_max_inherit != LMV_INHERIT_NONE &&
4508             !(LMVEA_DELETE_VALUES(lds->lds_dir_def_stripe_count,
4509                                  lds->lds_dir_def_stripe_offset) &&
4510               le32_to_cpu(lds->lds_dir_def_hash_type) !=
4511               LMV_HASH_TYPE_UNKNOWN)) {
4512                 struct lmv_user_md_v1 *v1 = info->lti_ea_store;
4513
4514                 if (info->lti_ea_store_size < sizeof(*v1)) {
4515                         rc = lod_ea_store_resize(info, sizeof(*v1));
4516                         if (rc != 0)
4517                                 RETURN(rc);
4518                         v1 = info->lti_ea_store;
4519                 }
4520
4521                 memset(v1, 0, sizeof(*v1));
4522                 v1->lum_magic = cpu_to_le32(LMV_USER_MAGIC);
4523                 v1->lum_stripe_count =
4524                         cpu_to_le32(lds->lds_dir_def_stripe_count);
4525                 v1->lum_stripe_offset =
4526                         cpu_to_le32(lds->lds_dir_def_stripe_offset);
4527                 v1->lum_hash_type =
4528                         cpu_to_le32(lds->lds_dir_def_hash_type);
4529                 v1->lum_max_inherit =
4530                         lmv_inherit_next(lds->lds_dir_def_max_inherit);
4531                 v1->lum_max_inherit_rr =
4532                         lmv_inherit_rr_next(lds->lds_dir_def_max_inherit_rr);
4533
4534                 info->lti_buf.lb_buf = v1;
4535                 info->lti_buf.lb_len = sizeof(*v1);
4536                 if (declare)
4537                         rc = lod_dir_declare_xattr_set(env, dt, &info->lti_buf,
4538                                                        XATTR_NAME_DEFAULT_LMV,
4539                                                        0, th);
4540                 else
4541                         rc = lod_xattr_set_default_lmv_on_dir(env, dt,
4542                                                   &info->lti_buf,
4543                                                   XATTR_NAME_DEFAULT_LMV, 0,
4544                                                   th);
4545                 if (rc != 0)
4546                         RETURN(rc);
4547         }
4548
4549         /* Transfer default LOV striping from the parent */
4550         if (lds != NULL && lds->lds_def_striping_set &&
4551             lds->lds_def_comp_cnt != 0) {
4552                 struct lov_mds_md *lmm;
4553                 int lmm_size = lod_comp_md_size(lo, true);
4554
4555                 if (info->lti_ea_store_size < lmm_size) {
4556                         rc = lod_ea_store_resize(info, lmm_size);
4557                         if (rc != 0)
4558                                 RETURN(rc);
4559                 }
4560                 lmm = info->lti_ea_store;
4561
4562                 rc = lod_generate_lovea(env, lo, lmm, &lmm_size, true);
4563                 if (rc != 0)
4564                         RETURN(rc);
4565
4566                 info->lti_buf.lb_buf = lmm;
4567                 info->lti_buf.lb_len = lmm_size;
4568
4569                 if (declare)
4570                         rc = lod_dir_declare_xattr_set(env, dt, &info->lti_buf,
4571                                                        XATTR_NAME_LOV, 0, th);
4572                 else
4573                         rc = lod_xattr_set_lov_on_dir(env, dt, &info->lti_buf,
4574                                                       XATTR_NAME_LOV, 0, th);
4575                 if (rc != 0)
4576                         RETURN(rc);
4577         }
4578
4579         /* ldo_def_striping is not allocated, clear after use, in case directory
4580          * layout is changed later.
4581          */
4582         if (!declare)
4583                 lo->ldo_def_striping = NULL;
4584
4585         RETURN(0);
4586 }
4587
4588 static int lod_declare_dir_striping_create(const struct lu_env *env,
4589                                            struct dt_object *dt,
4590                                            struct lu_attr *attr,
4591                                            struct lu_buf *lmu,
4592                                            struct dt_object_format *dof,
4593                                            struct thandle *th)
4594 {
4595         return lod_dir_striping_create_internal(env, dt, attr, lmu, dof, th,
4596                                                 true);
4597 }
4598
4599 static int lod_dir_striping_create(const struct lu_env *env,
4600                                    struct dt_object *dt,
4601                                    struct lu_attr *attr,
4602                                    struct dt_object_format *dof,
4603                                    struct thandle *th)
4604 {
4605         return lod_dir_striping_create_internal(env, dt, attr, NULL, dof, th,
4606                                                 false);
4607 }
4608
4609 /**
4610  * Make LOV EA for striped object.
4611  *
4612  * Generate striping information and store it in the LOV EA of the given
4613  * object. The caller must ensure nobody else is calling the function
4614  * against the object concurrently. The transaction must be started.
4615  * FLDB service must be running as well; it's used to map FID to the target,
4616  * which is stored in LOV EA.
4617  *
4618  * \param[in] env               execution environment for this thread
4619  * \param[in] lo                LOD object
4620  * \param[in] th                transaction handle
4621  *
4622  * \retval                      0 if LOV EA is stored successfully
4623  * \retval                      negative error number on failure
4624  */
4625 static int lod_generate_and_set_lovea(const struct lu_env *env,
4626                                       struct lod_object *lo,
4627                                       struct thandle *th)
4628 {
4629         struct lod_thread_info  *info = lod_env_info(env);
4630         struct dt_object        *next = dt_object_child(&lo->ldo_obj);
4631         struct lov_mds_md_v1    *lmm;
4632         int                      rc, lmm_size;
4633         ENTRY;
4634
4635         LASSERT(lo);
4636
4637         if (lo->ldo_comp_cnt == 0 && !lo->ldo_is_foreign) {
4638                 lod_striping_free_nolock(env, lo);
4639                 rc = lod_sub_xattr_del(env, next, XATTR_NAME_LOV, th);
4640                 RETURN(rc);
4641         }
4642
4643         lmm_size = lod_comp_md_size(lo, false);
4644         if (info->lti_ea_store_size < lmm_size) {
4645                 rc = lod_ea_store_resize(info, lmm_size);
4646                 if (rc)
4647                         RETURN(rc);
4648         }
4649         lmm = info->lti_ea_store;
4650
4651         rc = lod_generate_lovea(env, lo, lmm, &lmm_size, false);
4652         if (rc)
4653                 RETURN(rc);
4654
4655         info->lti_buf.lb_buf = lmm;
4656         info->lti_buf.lb_len = lmm_size;
4657         rc = lod_sub_xattr_set(env, next, &info->lti_buf,
4658                                XATTR_NAME_LOV, 0, th);
4659         RETURN(rc);
4660 }
4661
4662 static __u32 lod_gen_component_id(struct lod_object *lo,
4663                                   int mirror_id, int comp_idx);
4664
4665 /**
4666  * Repeat an existing component
4667  *
4668  * Creates a new layout by replicating an existing component.  Uses striping
4669  * policy from previous component as a template for the striping for the new
4670  * new component.
4671  *
4672  * New component starts with zero length, will be extended (or removed) before
4673  * returning layout to client.
4674  *
4675  * NB: Reallocates layout components array (lo->ldo_comp_entries), invalidating
4676  * any pre-existing pointers to components.  Handle with care.
4677  *
4678  * \param[in] env       execution environment for this thread
4679  * \param[in,out] lo    object to update the layout of
4680  * \param[in] index     index of component to copy
4681  *
4682  * \retval      0 on success
4683  * \retval      negative errno on error
4684  */
4685 static int lod_layout_repeat_comp(const struct lu_env *env,
4686                                   struct lod_object *lo, int index)
4687 {
4688         struct lod_layout_component *lod_comp;
4689         struct lod_layout_component *new_comp = NULL;
4690         struct lod_layout_component *comp_array;
4691         int rc = 0, i, new_cnt = lo->ldo_comp_cnt + 1;
4692         __u16 mirror_id;
4693         int offset = 0;
4694         ENTRY;
4695
4696         lod_comp = &lo->ldo_comp_entries[index];
4697         LASSERT(lod_comp_inited(lod_comp) && lod_comp->llc_id != LCME_ID_INVAL);
4698
4699         CDEBUG(D_LAYOUT, "repeating component %d\n", index);
4700
4701         OBD_ALLOC_PTR_ARRAY(comp_array, new_cnt);
4702         if (comp_array == NULL)
4703                 GOTO(out, rc = -ENOMEM);
4704
4705         for (i = 0; i < lo->ldo_comp_cnt; i++) {
4706                 memcpy(&comp_array[i + offset], &lo->ldo_comp_entries[i],
4707                        sizeof(*comp_array));
4708
4709                 /* Duplicate this component in to the next slot */
4710                 if (i == index) {
4711                         new_comp = &comp_array[i + 1];
4712                         memcpy(&comp_array[i + 1], &lo->ldo_comp_entries[i],
4713                                sizeof(*comp_array));
4714                         /* We must now skip this new component when copying */
4715                         offset = 1;
4716                 }
4717         }
4718
4719         /* Set up copied component */
4720         new_comp->llc_flags &= ~LCME_FL_INIT;
4721         new_comp->llc_stripe = NULL;
4722         new_comp->llc_stripes_allocated = 0;
4723         new_comp->llc_ost_indices = NULL;
4724         new_comp->llc_stripe_offset = LOV_OFFSET_DEFAULT;
4725         /* for uninstantiated components, layout gen stores default stripe
4726          * offset */
4727         new_comp->llc_layout_gen = lod_comp->llc_stripe_offset;
4728         /* This makes the repeated component zero-length, placed at the end of
4729          * the preceding component */
4730         new_comp->llc_extent.e_start = new_comp->llc_extent.e_end;
4731         new_comp->llc_timestamp = lod_comp->llc_timestamp;
4732         new_comp->llc_pool = NULL;
4733
4734         rc = lod_set_pool(&new_comp->llc_pool, lod_comp->llc_pool);
4735         if (rc)
4736                 GOTO(out, rc);
4737
4738         if (new_comp->llc_ostlist.op_array) {
4739                 __u32 *op_array = NULL;
4740
4741                 OBD_ALLOC(op_array, new_comp->llc_ostlist.op_size);
4742                 if (!op_array)
4743                         GOTO(out, rc = -ENOMEM);
4744                 memcpy(op_array, &new_comp->llc_ostlist.op_array,
4745                        new_comp->llc_ostlist.op_size);
4746                 new_comp->llc_ostlist.op_array = op_array;
4747         }
4748
4749         OBD_FREE_PTR_ARRAY(lo->ldo_comp_entries, lo->ldo_comp_cnt);
4750         lo->ldo_comp_entries = comp_array;
4751         lo->ldo_comp_cnt = new_cnt;
4752
4753         /* Generate an id for the new component */
4754         mirror_id = mirror_id_of(new_comp->llc_id);
4755         new_comp->llc_id = LCME_ID_INVAL;
4756         new_comp->llc_id = lod_gen_component_id(lo, mirror_id, index + 1);
4757         if (new_comp->llc_id == LCME_ID_INVAL)
4758                 GOTO(out, rc = -ERANGE);
4759
4760         EXIT;
4761 out:
4762         if (rc)
4763                 OBD_FREE_PTR_ARRAY(comp_array, new_cnt);
4764
4765         return rc;
4766 }
4767
4768 static int lod_layout_data_init(struct lod_thread_info *info, __u32 comp_cnt)
4769 {
4770         ENTRY;
4771
4772         /* clear memory region that will be used for layout change */
4773         memset(&info->lti_layout_attr, 0, sizeof(struct lu_attr));
4774         info->lti_count = 0;
4775
4776         if (info->lti_comp_size >= comp_cnt)
4777                 RETURN(0);
4778
4779         if (info->lti_comp_size > 0) {
4780                 OBD_FREE_PTR_ARRAY(info->lti_comp_idx, info->lti_comp_size);
4781                 info->lti_comp_size = 0;
4782         }
4783
4784         OBD_ALLOC_PTR_ARRAY(info->lti_comp_idx, comp_cnt);
4785         if (!info->lti_comp_idx)
4786                 RETURN(-ENOMEM);
4787
4788         info->lti_comp_size = comp_cnt;
4789         RETURN(0);
4790 }
4791
4792 /**
4793  * Prepare new layout minus deleted components
4794  *
4795  * Removes components marked for deletion (LCME_ID_INVAL) by copying to a new
4796  * layout and skipping those components.  Removes stripe objects if any exist.
4797  *
4798  * NB:
4799  * Reallocates layout components array (lo->ldo_comp_entries), invalidating
4800  * any pre-existing pointers to components.
4801  *
4802  * Caller is responsible for updating mirror end (ldo_mirror[].lme_end).
4803  *
4804  * \param[in] env       execution environment for this thread
4805  * \param[in,out] lo    object to update the layout of
4806  * \param[in] th        transaction handle for this operation
4807  *
4808  * \retval      # of components deleted
4809  * \retval      negative errno on error
4810  */
4811 static int lod_layout_del_prep_layout(const struct lu_env *env,
4812                                       struct lod_object *lo,
4813                                       struct thandle *th)
4814 {
4815         struct lod_layout_component     *lod_comp;
4816         struct lod_thread_info  *info = lod_env_info(env);
4817         int rc = 0, i, j, deleted = 0;
4818
4819         ENTRY;
4820
4821         LASSERT(lo->ldo_is_composite);
4822         LASSERT(lo->ldo_comp_cnt > 0 && lo->ldo_comp_entries != NULL);
4823
4824         rc = lod_layout_data_init(info, lo->ldo_comp_cnt);
4825         if (rc)
4826                 RETURN(rc);
4827
4828         for (i = 0; i < lo->ldo_comp_cnt; i++) {
4829                 lod_comp = &lo->ldo_comp_entries[i];
4830
4831                 if (lod_comp->llc_id != LCME_ID_INVAL) {
4832                         /* Build array of things to keep */
4833                         info->lti_comp_idx[info->lti_count++] = i;
4834                         continue;
4835                 }
4836
4837                 lod_obj_set_pool(lo, i, NULL);
4838                 if (lod_comp->llc_ostlist.op_array) {
4839                         OBD_FREE(lod_comp->llc_ostlist.op_array,
4840                                  lod_comp->llc_ostlist.op_size);
4841                         lod_comp->llc_ostlist.op_array = NULL;
4842                         lod_comp->llc_ostlist.op_size = 0;
4843                 }
4844
4845                 deleted++;
4846                 CDEBUG(D_LAYOUT, "deleting comp %d, left %d\n", i,
4847                        lo->ldo_comp_cnt - deleted);
4848
4849                 /* No striping info for this component */
4850                 if (lod_comp->llc_stripe == NULL)
4851                         continue;
4852
4853                 LASSERT(lod_comp->llc_stripe_count > 0);
4854                 for (j = 0; j < lod_comp->llc_stripe_count; j++) {
4855                         struct dt_object *obj = lod_comp->llc_stripe[j];
4856
4857                         if (obj == NULL)
4858                                 continue;
4859
4860                         /* components which are not init have no sub objects
4861                          * to destroy */
4862                         if (lod_comp_inited(lod_comp)) {
4863                                 rc = lod_sub_destroy(env, obj, th);
4864                                 if (rc)
4865                                         GOTO(out, rc);
4866                         }
4867
4868                         lu_object_put(env, &obj->do_lu);
4869                         lod_comp->llc_stripe[j] = NULL;
4870                 }
4871                 OBD_FREE_PTR_ARRAY(lod_comp->llc_stripe,
4872                                    lod_comp->llc_stripes_allocated);
4873                 lod_comp->llc_stripe = NULL;
4874                 OBD_FREE_PTR_ARRAY(lod_comp->llc_ost_indices,
4875                                    lod_comp->llc_stripes_allocated);
4876                 lod_comp->llc_ost_indices = NULL;
4877                 lod_comp->llc_stripes_allocated = 0;
4878         }
4879
4880         /* info->lti_count has the amount of left components */
4881         LASSERTF(info->lti_count >= 0 && info->lti_count < lo->ldo_comp_cnt,
4882                  "left = %d, lo->ldo_comp_cnt %d\n", (int)info->lti_count,
4883                  (int)lo->ldo_comp_cnt);
4884
4885         if (info->lti_count > 0) {
4886                 struct lod_layout_component *comp_array;
4887
4888                 OBD_ALLOC_PTR_ARRAY(comp_array, info->lti_count);
4889                 if (comp_array == NULL)
4890                         GOTO(out, rc = -ENOMEM);
4891
4892                 for (i = 0; i < info->lti_count; i++) {
4893                         memcpy(&comp_array[i],
4894                                &lo->ldo_comp_entries[info->lti_comp_idx[i]],
4895                                sizeof(*comp_array));
4896                 }
4897
4898                 OBD_FREE_PTR_ARRAY(lo->ldo_comp_entries, lo->ldo_comp_cnt);
4899                 lo->ldo_comp_entries = comp_array;
4900                 lo->ldo_comp_cnt = info->lti_count;
4901         } else {
4902                 lod_free_comp_entries(lo);
4903         }
4904
4905         EXIT;
4906 out:
4907         return rc ? rc : deleted;
4908 }
4909
4910 /**
4911  * Delete layout component(s)
4912  *
4913  * This function sets up the layout data in the env and does the setattrs
4914  * required to write out the new layout.  The layout itself is modified in
4915  * lod_layout_del_prep_layout.
4916  *
4917  * \param[in] env       execution environment for this thread
4918  * \param[in] dt        object
4919  * \param[in] th        transaction handle
4920  *
4921  * \retval      0 on success
4922  * \retval      negative error number on failure
4923  */
4924 static int lod_layout_del(const struct lu_env *env, struct dt_object *dt,
4925                           struct thandle *th)
4926 {
4927         struct lod_object *lo = lod_dt_obj(dt);
4928         struct dt_object *next = dt_object_child(dt);
4929         struct lu_attr *attr = &lod_env_info(env)->lti_attr;
4930         int rc;
4931
4932         LASSERT(lo->ldo_mirror_count == 1);
4933
4934         mutex_lock(&lo->ldo_layout_mutex);
4935
4936         rc = lod_layout_del_prep_layout(env, lo, th);
4937         if (rc < 0)
4938                 GOTO(out, rc);
4939
4940         /* Only do this if we didn't delete all components */
4941         if (lo->ldo_comp_cnt > 0) {
4942                 lo->ldo_mirrors[0].lme_end = lo->ldo_comp_cnt - 1;
4943                 lod_obj_inc_layout_gen(lo);
4944         }
4945
4946         LASSERT(dt_object_exists(dt));
4947         rc = dt_attr_get(env, next, attr);
4948         if (rc)
4949                 GOTO(out, rc);
4950
4951         if (attr->la_size > 0) {
4952                 attr->la_size = 0;
4953                 attr->la_valid = LA_SIZE;
4954                 rc = lod_sub_attr_set(env, next, attr, th);
4955                 if (rc)
4956                         GOTO(out, rc);
4957         }
4958
4959         rc = lod_generate_and_set_lovea(env, lo, th);
4960         EXIT;
4961 out:
4962         if (rc)
4963                 lod_striping_free_nolock(env, lo);
4964
4965         mutex_unlock(&lo->ldo_layout_mutex);
4966
4967         return rc;
4968 }
4969
4970
4971 /**
4972  * Implementation of dt_object_operations::do_xattr_set.
4973  *
4974  * Sets specified extended attribute on the object. Three types of EAs are
4975  * special:
4976  *   LOV EA - stores striping for a regular file or default striping (when set
4977  *            on a directory)
4978  *   LMV EA - stores a marker for the striped directories
4979  *   DMV EA - stores default directory striping
4980  *
4981  * When striping is applied to a non-striped existing object (this is called
4982  * late striping), then LOD notices the caller wants to turn the object into a
4983  * striped one. The stripe objects are created and appropriate EA is set:
4984  * LOV EA storing all the stripes directly or LMV EA storing just a small header
4985  * with striping configuration.
4986  *
4987  * \see dt_object_operations::do_xattr_set() in the API description for details.
4988  */
4989 static int lod_xattr_set(const struct lu_env *env,
4990                          struct dt_object *dt, const struct lu_buf *buf,
4991                          const char *name, int fl, struct thandle *th)
4992 {
4993         struct dt_object *next = dt_object_child(dt);
4994         struct lu_attr *layout_attr = &lod_env_info(env)->lti_layout_attr;
4995         struct lod_object *lo = lod_dt_obj(dt);
4996         struct lod_obj_stripe_cb_data data = { {0} };
4997         int rc = 0;
4998
4999         ENTRY;
5000
5001         if (S_ISDIR(dt->do_lu.lo_header->loh_attr) &&
5002             !strcmp(name, XATTR_NAME_LMV)) {
5003                 switch (fl) {
5004                 case LU_XATTR_CREATE:
5005                         rc = lod_dir_striping_create(env, dt, NULL, NULL, th);
5006                         break;
5007                 case 0:
5008                 case LU_XATTR_REPLACE:
5009                         rc = lod_dir_layout_set(env, dt, buf, fl, th);
5010                         break;
5011                 default:
5012                         LBUG();
5013                 }
5014
5015                 RETURN(rc);
5016         } else if (S_ISDIR(dt->do_lu.lo_header->loh_attr) &&
5017                    strcmp(name, XATTR_NAME_LOV) == 0) {
5018                 rc = lod_xattr_set_default_lov_on_dir(env, dt, buf, name, fl,
5019                                                       th);
5020                 RETURN(rc);
5021         } else if (S_ISDIR(dt->do_lu.lo_header->loh_attr) &&
5022                    strcmp(name, XATTR_NAME_DEFAULT_LMV) == 0) {
5023                 /* default LMVEA */
5024                 rc = lod_xattr_set_default_lmv_on_dir(env, dt, buf, name, fl,
5025                                                       th);
5026                 RETURN(rc);
5027         } else if (S_ISREG(dt->do_lu.lo_header->loh_attr) &&
5028                    (strcmp(name, XATTR_NAME_LOV) == 0 ||
5029                     strcmp(name, XATTR_LUSTRE_LOV) == 0 ||
5030                     allowed_lustre_lov(name))) {
5031                 /* in case of lov EA swap, just set it
5032                  * if not, it is a replay so check striping match what we
5033                  * already have during req replay, declare_xattr_set()
5034                  * defines striping, then create() does the work */
5035                 if (fl & LU_XATTR_REPLACE) {
5036                         /* free stripes, then update disk */
5037                         lod_striping_free(env, lod_dt_obj(dt));
5038
5039                         rc = lod_sub_xattr_set(env, next, buf, name, fl, th);
5040                 } else if (fl & LU_XATTR_SPLIT) {
5041                         rc = lod_sub_xattr_set(env, next, buf, name, fl, th);
5042                         if (rc)
5043                                 RETURN(rc);
5044
5045                         rc = lod_striping_reload(env, lo, buf, LVF_ALL_STALE);
5046                         if (rc)
5047                                 RETURN(rc);
5048
5049                         if (lo->ldo_mirror_count > 1 &&
5050                             layout_attr->la_valid & LA_LAYOUT_VERSION) {
5051                                 /* mirror split */
5052                                 layout_attr->la_layout_version =
5053                                                 lo->ldo_layout_gen;
5054                                 data.locd_attr = layout_attr;
5055                                 data.locd_declare = false;
5056                                 data.locd_stripe_cb =
5057                                                 lod_obj_stripe_attr_set_cb;
5058                                 rc = lod_obj_for_each_stripe(env, lo, th,
5059                                                              &data);
5060                                 if (rc)
5061                                         RETURN(rc);
5062                         }
5063                 } else if (fl & LU_XATTR_PURGE) {
5064                         rc = lod_layout_purge(env, dt, buf, th);
5065                 } else if (dt_object_remote(dt)) {
5066                         /* This only happens during migration, see
5067                          * mdd_migrate_create(), in which Master MDT will
5068                          * create a remote target object, and only set
5069                          * (migrating) stripe EA on the remote object,
5070                          * and does not need creating each stripes. */
5071                         rc = lod_sub_xattr_set(env, next, buf, name,
5072                                                       fl, th);
5073                 } else if (strcmp(name, XATTR_LUSTRE_LOV".del") == 0) {
5074                         /* delete component(s) */
5075                         LASSERT(lod_dt_obj(dt)->ldo_comp_cached);
5076                         rc = lod_layout_del(env, dt, th);
5077                 } else {
5078                         /*
5079                          * When 'name' is XATTR_LUSTRE_LOV or XATTR_NAME_LOV,
5080                          * it's going to create create file with specified
5081                          * component(s), the striping must have not being
5082                          * cached in this case;
5083                          *
5084                          * Otherwise, it's going to add/change component(s) to
5085                          * an existing file, the striping must have been cached
5086                          * in this case.
5087                          */
5088                         LASSERT(equi(!strcmp(name, XATTR_LUSTRE_LOV) ||
5089                                      !strcmp(name, XATTR_NAME_LOV),
5090                                 !lod_dt_obj(dt)->ldo_comp_cached));
5091
5092                         rc = lod_striped_create(env, dt, NULL, NULL, th);
5093                         if (rc)
5094                                 RETURN(rc);
5095
5096                         if (fl & LU_XATTR_MERGE && lo->ldo_mirror_count > 1 &&
5097                             layout_attr->la_valid & LA_LAYOUT_VERSION) {
5098                                 /* mirror merge exec phase */
5099                                 layout_attr->la_layout_version =
5100                                                 lo->ldo_layout_gen;
5101                                 data.locd_attr = layout_attr;
5102                                 data.locd_declare = false;
5103                                 data.locd_stripe_cb =
5104                                                 lod_obj_stripe_attr_set_cb;
5105                                 rc = lod_obj_for_each_stripe(env, lo, th,
5106                                                              &data);
5107                                 if (rc)
5108                                         RETURN(rc);
5109                         }
5110                 }
5111                 RETURN(rc);
5112         } else if (strcmp(name, XATTR_NAME_FID) == 0) {
5113                 rc = lod_replace_parent_fid(env, dt, buf, th, false);
5114
5115                 RETURN(rc);
5116         }
5117
5118         /* then all other xattr */
5119         rc = lod_xattr_set_internal(env, dt, buf, name, fl, th);
5120
5121         RETURN(rc);
5122 }
5123
5124 /**
5125  * Implementation of dt_object_operations::do_declare_xattr_del.
5126  *
5127  * \see dt_object_operations::do_declare_xattr_del() in the API description
5128  * for details.
5129  */
5130 static int lod_declare_xattr_del(const struct lu_env *env,
5131                                  struct dt_object *dt, const char *name,
5132                                  struct thandle *th)
5133 {
5134         struct lod_object *lo = lod_dt_obj(dt);
5135         struct dt_object *next = dt_object_child(dt);
5136         int i;
5137         int rc;
5138         ENTRY;
5139
5140         rc = lod_sub_declare_xattr_del(env, next, name, th);
5141         if (rc != 0)
5142                 RETURN(rc);
5143
5144         if (!S_ISDIR(dt->do_lu.lo_header->loh_attr))
5145                 RETURN(0);
5146
5147         /* NB: don't delete stripe LMV, because when we do this, normally we
5148          * will remove stripes, besides, if directory LMV is corrupt, this will
5149          * prevent deleting its LMV and fixing it (via LFSCK).
5150          */
5151         if (!strcmp(name, XATTR_NAME_LMV))
5152                 RETURN(0);
5153
5154         rc = lod_striping_load(env, lo);
5155         if (rc != 0)
5156                 RETURN(rc);
5157
5158         if (lo->ldo_dir_stripe_count == 0)
5159                 RETURN(0);
5160
5161         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
5162                 struct dt_object *dto = lo->ldo_stripe[i];
5163
5164                 if (!dto)
5165                         continue;
5166
5167                 if (!dt_object_exists(dto))
5168                         continue;
5169
5170                 rc = lod_sub_declare_xattr_del(env, dto, name, th);
5171                 if (rc != 0)
5172                         break;
5173         }
5174
5175         RETURN(rc);
5176 }
5177
5178 /**
5179  * Implementation of dt_object_operations::do_xattr_del.
5180  *
5181  * If EA storing a regular striping is being deleted, then release
5182  * all the references to the stripe objects in core.
5183  *
5184  * \see dt_object_operations::do_xattr_del() in the API description for details.
5185  */
5186 static int lod_xattr_del(const struct lu_env *env, struct dt_object *dt,
5187                          const char *name, struct thandle *th)
5188 {
5189         int rc;
5190
5191         ENTRY;
5192
5193         if (!strcmp(name, XATTR_NAME_LOV) || !strcmp(name, XATTR_NAME_LMV))
5194                 lod_striping_free(env, lod_dt_obj(dt));
5195
5196         rc = lod_xattr_del_internal(env, dt, name, th);
5197
5198         RETURN(rc);
5199 }
5200
5201 /**
5202  * Implementation of dt_object_operations::do_xattr_list.
5203  *
5204  * \see dt_object_operations::do_xattr_list() in the API description
5205  * for details.
5206  */
5207 static int lod_xattr_list(const struct lu_env *env,
5208                           struct dt_object *dt, const struct lu_buf *buf)
5209 {
5210         return dt_xattr_list(env, dt_object_child(dt), buf);
5211 }
5212
5213 static inline int lod_object_will_be_striped(int is_reg, const struct lu_fid *fid)
5214 {
5215         return (is_reg && fid_seq(fid) != FID_SEQ_LOCAL_FILE);
5216 }
5217
5218 /**
5219  * Copy OST list from layout provided by user.
5220  *
5221  * \param[in] lod_comp          layout_component to be filled
5222  * \param[in] v3                LOV EA V3 user data
5223  *
5224  * \retval              0 on success
5225  * \retval              negative if failed
5226  */
5227 int lod_comp_copy_ost_lists(struct lod_layout_component *lod_comp,
5228                             struct lov_user_md_v3 *v3)
5229 {
5230         int j;
5231
5232         ENTRY;
5233
5234         if (v3->lmm_stripe_offset == LOV_OFFSET_DEFAULT)
5235                 v3->lmm_stripe_offset = v3->lmm_objects[0].l_ost_idx;
5236
5237         if (lod_comp->llc_ostlist.op_array) {
5238                 if (lod_comp->llc_ostlist.op_size >=
5239                     v3->lmm_stripe_count * sizeof(__u32))  {
5240                         lod_comp->llc_ostlist.op_count =
5241                                         v3->lmm_stripe_count;
5242                         goto skip;
5243                 }
5244                 OBD_FREE(lod_comp->llc_ostlist.op_array,
5245                          lod_comp->llc_ostlist.op_size);
5246         }
5247
5248         /* copy ost list from lmm */
5249         lod_comp->llc_ostlist.op_count = v3->lmm_stripe_count;
5250         lod_comp->llc_ostlist.op_size = v3->lmm_stripe_count * sizeof(__u32);
5251         OBD_ALLOC(lod_comp->llc_ostlist.op_array,
5252                   lod_comp->llc_ostlist.op_size);
5253         if (!lod_comp->llc_ostlist.op_array)
5254                 RETURN(-ENOMEM);
5255 skip:
5256         for (j = 0; j < v3->lmm_stripe_count; j++) {
5257                 lod_comp->llc_ostlist.op_array[j] =
5258                         v3->lmm_objects[j].l_ost_idx;
5259         }
5260
5261         RETURN(0);
5262 }
5263
5264
5265 /**
5266  * Get default striping.
5267  *
5268  * \param[in] env               execution environment
5269  * \param[in] lo                object
5270  * \param[out] lds              default striping
5271  *
5272  * \retval              0 on success
5273  * \retval              negative if failed
5274  */
5275 static int lod_get_default_lov_striping(const struct lu_env *env,
5276                                         struct lod_object *lo,
5277                                         struct lod_default_striping *lds,
5278                                         struct dt_allocation_hint *dah)
5279 {
5280         struct lod_thread_info *info = lod_env_info(env);
5281         struct lov_user_md_v1 *v1 = NULL;
5282         struct lov_user_md_v3 *v3 = NULL;
5283         struct lov_comp_md_v1 *lcm = NULL;
5284         __u32 magic;
5285         int append_stripe_count = dah != NULL ? dah->dah_append_stripe_count : 0;
5286         const char *append_pool = (dah != NULL &&
5287                                    dah->dah_append_pool != NULL &&
5288                                    dah->dah_append_pool[0] != '\0') ?
5289                                   dah->dah_append_pool : NULL;
5290         __u16 entry_count = 1;
5291         __u16 mirror_count = 0;
5292         bool want_composite = false;
5293         int rc, i, j;
5294
5295         ENTRY;
5296
5297         lds->lds_def_striping_set = 0;
5298
5299         rc = lod_get_lov_ea(env, lo);
5300         if (rc < 0)
5301                 RETURN(rc);
5302
5303         if (rc < (typeof(rc))sizeof(struct lov_user_md))
5304                 RETURN(0);
5305
5306         magic = *(__u32 *)info->lti_ea_store;
5307         if (magic == __swab32(LOV_USER_MAGIC_V1)) {
5308                 lustre_swab_lov_user_md_v1(info->lti_ea_store);
5309         } else if (magic == __swab32(LOV_USER_MAGIC_V3)) {
5310                 lustre_swab_lov_user_md_v3(info->lti_ea_store);
5311         } else if (magic == __swab32(LOV_USER_MAGIC_SPECIFIC)) {
5312                 v3 = (struct lov_user_md_v3 *)info->lti_ea_store;
5313                 lustre_swab_lov_user_md_v3(v3);
5314                 lustre_swab_lov_user_md_objects(v3->lmm_objects,
5315                                                 v3->lmm_stripe_count);
5316         } else if (magic == __swab32(LOV_USER_MAGIC_COMP_V1) ||
5317                    magic == __swab32(LOV_USER_MAGIC_SEL)) {
5318                 lustre_swab_lov_comp_md_v1(info->lti_ea_store);
5319         }
5320
5321         switch (magic) {
5322         case LOV_MAGIC_V1:
5323         case LOV_MAGIC_V3:
5324         case LOV_USER_MAGIC_SPECIFIC:
5325                 v1 = info->lti_ea_store;
5326                 break;
5327         case LOV_MAGIC_COMP_V1:
5328         case LOV_MAGIC_SEL:
5329                 lcm = info->lti_ea_store;
5330                 entry_count = lcm->lcm_entry_count;
5331                 if (entry_count == 0)
5332                         RETURN(-EINVAL);
5333
5334                 mirror_count = lcm->lcm_mirror_count + 1;
5335                 want_composite = true;
5336                 break;
5337         default:
5338                 RETURN(-ENOTSUPP);
5339         }
5340
5341         if (append_stripe_count != 0 || append_pool != NULL) {
5342                 entry_count = 1;
5343                 mirror_count = 0;
5344                 want_composite = false;
5345         }
5346
5347         /* realloc default comp entries if necessary */
5348         rc = lod_def_striping_comp_resize(lds, entry_count);
5349         if (rc < 0)
5350                 RETURN(rc);
5351
5352         lds->lds_def_comp_cnt = entry_count;
5353         lds->lds_def_striping_is_composite = want_composite;
5354         lds->lds_def_mirror_cnt = mirror_count;
5355
5356         for (i = 0; i < entry_count; i++) {
5357                 struct lod_layout_component *llc = &lds->lds_def_comp_entries[i];
5358                 const char *pool;
5359
5360                 /*
5361                  * reset llc values, llc_stripes is always NULL in the
5362                  * default striping template, llc_pool will be reset
5363                  * later below using lod_set_pool().
5364                  *
5365                  * XXX At this point llc_pool may point to valid (!)
5366                  * kmalloced strings from previous RPCs.
5367                  */
5368                 memset(llc, 0, offsetof(typeof(*llc), llc_pool));
5369
5370                 if (lcm != NULL) {
5371                         v1 = (struct lov_user_md *)((char *)lcm +
5372                                                     lcm->lcm_entries[i].lcme_offset);
5373
5374                         if (want_composite) {
5375                                 llc->llc_extent = lcm->lcm_entries[i].lcme_extent;
5376                                 /* We only inherit certain flags from the layout */
5377                                 llc->llc_flags = lcm->lcm_entries[i].lcme_flags &
5378                                         LCME_TEMPLATE_FLAGS;
5379                         }
5380                 }
5381
5382                 CDEBUG(D_LAYOUT, DFID" magic = %#08x, pattern = %#x, stripe_count = %hu, stripe_size = %u, stripe_offset = %hu, append_pool = '%s', append_stripe_count = %d\n",
5383                        PFID(lu_object_fid(&lo->ldo_obj.do_lu)),
5384                        v1->lmm_magic,
5385                        v1->lmm_pattern,
5386                        v1->lmm_stripe_count,
5387                        v1->lmm_stripe_size,
5388                        v1->lmm_stripe_offset,
5389                        append_pool ?: "",
5390                        append_stripe_count);
5391
5392                 if (!lov_pattern_supported(v1->lmm_pattern) &&
5393                     !(v1->lmm_pattern & LOV_PATTERN_F_RELEASED)) {
5394                         lod_free_def_comp_entries(lds);
5395                         RETURN(-EINVAL);
5396                 }
5397
5398                 llc->llc_stripe_count = v1->lmm_stripe_count;
5399                 llc->llc_stripe_size = v1->lmm_stripe_size;
5400                 llc->llc_stripe_offset = v1->lmm_stripe_offset;
5401                 llc->llc_pattern = v1->lmm_pattern;
5402
5403                 if (append_stripe_count != 0 || append_pool != NULL)
5404                         llc->llc_pattern = LOV_PATTERN_RAID0;
5405
5406                 if (append_stripe_count != 0)
5407                         llc->llc_stripe_count = append_stripe_count;
5408
5409                 pool = NULL;
5410                 if (append_pool != NULL) {
5411                         pool = append_pool;
5412                 } else if (v1->lmm_magic == LOV_USER_MAGIC_V3) {
5413                         /* XXX: sanity check here */
5414                         v3 = (struct lov_user_md_v3 *)v1;
5415                         if (v3->lmm_pool_name[0] != '\0')
5416                                 pool = v3->lmm_pool_name;
5417                 }
5418
5419                 lod_set_pool(&llc->llc_pool, pool);
5420
5421                 if (v1->lmm_magic == LOV_USER_MAGIC_SPECIFIC &&
5422                     append_stripe_count == 0 &&
5423                     append_pool == NULL) {
5424                         v3 = (struct lov_user_md_v3 *)v1;
5425                         rc = lod_comp_copy_ost_lists(llc, v3);
5426                         if (rc)
5427                                 RETURN(rc);
5428                 } else if (llc->llc_ostlist.op_array &&
5429                            llc->llc_ostlist.op_count) {
5430                         for (j = 0; j < llc->llc_ostlist.op_count; j++)
5431                                 llc->llc_ostlist.op_array[j] = -1;
5432                         llc->llc_ostlist.op_count = 0;
5433                 }
5434         }
5435
5436         lds->lds_def_striping_set = 1;
5437         RETURN(rc);
5438 }
5439
5440 static inline void lod_lum2lds(struct lod_default_striping *lds,
5441                                const struct lmv_user_md *lum)
5442 {
5443         lds->lds_dir_def_stripe_count = le32_to_cpu(lum->lum_stripe_count);
5444         lds->lds_dir_def_stripe_offset = le32_to_cpu(lum->lum_stripe_offset);
5445         lds->lds_dir_def_hash_type = le32_to_cpu(lum->lum_hash_type);
5446         lds->lds_dir_def_max_inherit = lum->lum_max_inherit;
5447         lds->lds_dir_def_max_inherit_rr = lum->lum_max_inherit_rr;
5448         lds->lds_dir_def_striping_set = 1;
5449 }
5450
5451 /**
5452  * Get default directory striping.
5453  *
5454  * \param[in] env               execution environment
5455  * \param[in] lo                object
5456  * \param[out] lds              default striping
5457  *
5458  * \retval              0 on success
5459  * \retval              negative if failed
5460  */
5461 static int lod_get_default_lmv_striping(const struct lu_env *env,
5462                                         struct lod_object *lo,
5463                                         struct lod_default_striping *lds)
5464 {
5465         struct lmv_user_md *lmu;
5466         int rc;
5467
5468         lds->lds_dir_def_striping_set = 0;
5469
5470         rc = lod_get_default_lmv_ea(env, lo);
5471         if (rc < 0)
5472                 return rc;
5473
5474         if (rc >= (int)sizeof(*lmu)) {
5475                 struct lod_thread_info *info = lod_env_info(env);
5476
5477                 lmu = info->lti_ea_store;
5478                 lod_lum2lds(lds, lmu);
5479         }
5480
5481         return 0;
5482 }
5483
5484 /**
5485  * Get default striping in the object.
5486  *
5487  * Get object default striping and default directory striping.
5488  *
5489  * \param[in] env               execution environment
5490  * \param[in] lo                object
5491  * \param[out] lds              default striping
5492  *
5493  * \retval              0 on success
5494  * \retval              negative if failed
5495  */
5496 static int lod_get_default_striping(const struct lu_env *env,
5497                                     struct lod_object *lo,
5498                                     struct dt_allocation_hint *ah,
5499                                     struct lod_default_striping *lds)
5500 {
5501         int rc, rc1;
5502
5503         rc = lod_get_default_lov_striping(env, lo, lds, NULL);
5504         if (lds->lds_def_striping_set) {
5505                 struct lod_thread_info *info = lod_env_info(env);
5506                 struct lod_device *d = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
5507
5508                 rc = lod_verify_striping(env, d, lo, &info->lti_buf, false);
5509                 if (rc)
5510                         lds->lds_def_striping_set = 0;
5511         }
5512
5513         if (ah->dah_eadata_is_dmv) {
5514                 lod_lum2lds(lds, ah->dah_eadata);
5515         } else if (ah->dah_dmv_imp_inherit) {
5516                 lds->lds_dir_def_striping_set = 0;
5517         } else {
5518                 rc1 = lod_get_default_lmv_striping(env, lo, lds);
5519                 if (rc == 0 && rc1 < 0)
5520                         rc = rc1;
5521         }
5522
5523         return rc;
5524 }
5525
5526 /**
5527  * Apply default striping on object.
5528  *
5529  * If object striping pattern is not set, set to the one in default striping.
5530  * The default striping is from parent or fs.
5531  *
5532  * \param[in] lo                new object
5533  * \param[in] lds               default striping
5534  * \param[in] mode              new object's mode
5535  */
5536 static void lod_striping_from_default(struct lod_object *lo,
5537                                       const struct lod_default_striping *lds,
5538                                       umode_t mode)
5539 {
5540         struct lod_device *d = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
5541         int i, rc;
5542
5543         if (lds->lds_def_striping_set && S_ISREG(mode)) {
5544                 struct lov_desc *desc = &d->lod_ost_descs.ltd_lov_desc;
5545
5546                 rc = lod_alloc_comp_entries(lo, lds->lds_def_mirror_cnt,
5547                                             lds->lds_def_comp_cnt);
5548                 if (rc != 0)
5549                         return;
5550
5551                 lo->ldo_is_composite = lds->lds_def_striping_is_composite;
5552                 if (lds->lds_def_mirror_cnt > 1)
5553                         lo->ldo_flr_state = LCM_FL_RDONLY;
5554
5555                 for (i = 0; i < lo->ldo_comp_cnt; i++) {
5556                         struct lod_layout_component *obj_comp =
5557                                                 &lo->ldo_comp_entries[i];
5558                         struct lod_layout_component *def_comp =
5559                                                 &lds->lds_def_comp_entries[i];
5560
5561                         CDEBUG(D_LAYOUT,
5562                                "inherit "DFID" file layout from default: flags=%#x size=%u nr=%u offset=%u pattern=%#x pool=%s\n",
5563                                PFID(lu_object_fid(&lo->ldo_obj.do_lu)),
5564                                def_comp->llc_flags,
5565                                def_comp->llc_stripe_size,
5566                                def_comp->llc_stripe_count,
5567                                def_comp->llc_stripe_offset,
5568                                def_comp->llc_pattern,
5569                                def_comp->llc_pool ?: "");
5570
5571                         *obj_comp = *def_comp;
5572                         if (def_comp->llc_pool != NULL) {
5573                                 /* pointer was copied from def_comp */
5574                                 obj_comp->llc_pool = NULL;
5575                                 lod_obj_set_pool(lo, i, def_comp->llc_pool);
5576                         }
5577
5578                         /* copy ost list */
5579                         if (def_comp->llc_ostlist.op_array &&
5580                             def_comp->llc_ostlist.op_count) {
5581                                 OBD_ALLOC(obj_comp->llc_ostlist.op_array,
5582                                           obj_comp->llc_ostlist.op_size);
5583                                 if (!obj_comp->llc_ostlist.op_array)
5584                                         return;
5585                                 memcpy(obj_comp->llc_ostlist.op_array,
5586                                        def_comp->llc_ostlist.op_array,
5587                                        obj_comp->llc_ostlist.op_size);
5588                         } else if (def_comp->llc_ostlist.op_array) {
5589                                 obj_comp->llc_ostlist.op_array = NULL;
5590                         }
5591
5592                         /*
5593                          * Don't initialize these fields for plain layout
5594                          * (v1/v3) here, they are inherited in the order of
5595                          * 'parent' -> 'fs default (root)' -> 'global default
5596                          * values for stripe_count & stripe_size'.
5597                          *
5598                          * see lod_ah_init().
5599                          */
5600                         if (!lo->ldo_is_composite)
5601                                 continue;
5602
5603                         lod_adjust_stripe_info(obj_comp, desc, 0);
5604                 }
5605         } else if (lds->lds_dir_def_striping_set && S_ISDIR(mode)) {
5606                 if (lo->ldo_dir_stripe_count == 0)
5607                         lo->ldo_dir_stripe_count =
5608                                 lds->lds_dir_def_stripe_count;
5609                 if (lo->ldo_dir_stripe_offset == -1)
5610                         lo->ldo_dir_stripe_offset =
5611                                 lds->lds_dir_def_stripe_offset;
5612                 if (lo->ldo_dir_hash_type == LMV_HASH_TYPE_UNKNOWN)
5613                         lo->ldo_dir_hash_type = lds->lds_dir_def_hash_type;
5614
5615                 CDEBUG(D_LAYOUT,
5616                        "inherit "DFID" dir layout from default: count=%hu offset=%u hash_type=%x\n",
5617                        PFID(lu_object_fid(&lo->ldo_obj.do_lu)),
5618                        lo->ldo_dir_stripe_count, lo->ldo_dir_stripe_offset,
5619                        lo->ldo_dir_hash_type);
5620         }
5621 }
5622
5623 static inline bool lod_need_inherit_more(struct lod_object *lo, bool from_root,
5624                                          const char *append_pool)
5625 {
5626         struct lod_layout_component *lod_comp;
5627
5628         if (lo->ldo_comp_cnt == 0)
5629                 return true;
5630
5631         if (lo->ldo_is_composite)
5632                 return false;
5633
5634         lod_comp = &lo->ldo_comp_entries[0];
5635
5636         if (lod_comp->llc_stripe_count <= 0 ||
5637             lod_comp->llc_stripe_size <= 0)
5638                 return true;
5639
5640         if (from_root && (lod_comp->llc_pool == NULL ||
5641                           lod_comp->llc_stripe_offset == LOV_OFFSET_DEFAULT))
5642                 return true;
5643
5644         if (append_pool && append_pool[0])
5645                 return true;
5646
5647         return false;
5648 }
5649
5650 /**
5651  * Implementation of dt_object_operations::do_ah_init.
5652  *
5653  * This method is used to make a decision on the striping configuration for the
5654  * object being created. It can be taken from the \a parent object if it exists,
5655  * or filesystem's default. The resulting configuration (number of stripes,
5656  * stripe size/offset, pool name, hash_type, etc.) is stored in the object
5657  * itself and will be used by the methods like ->doo_declare_create().
5658  *
5659  * \see dt_object_operations::do_ah_init() in the API description for details.
5660  */
5661 static void lod_ah_init(const struct lu_env *env,
5662                         struct dt_allocation_hint *ah,
5663                         struct dt_object *parent,
5664                         struct dt_object *child,
5665                         umode_t child_mode)
5666 {
5667         struct lod_device *d = lu2lod_dev(child->do_lu.lo_dev);
5668         struct lod_thread_info *info = lod_env_info(env);
5669         struct lod_default_striping *lds = lod_lds_buf_get(env);
5670         struct dt_object *nextp = NULL;
5671         struct dt_object *nextc;
5672         struct lod_object *lp = NULL;
5673         struct lod_object *lc;
5674         struct lov_desc *desc;
5675         struct lod_layout_component *lod_comp;
5676         int rc;
5677         ENTRY;
5678
5679         LASSERT(child);
5680
5681         if (ah->dah_append_stripe_count == -1)
5682                 ah->dah_append_stripe_count =
5683                         d->lod_ost_descs.ltd_lov_desc.ld_tgt_count;
5684
5685         if (likely(parent)) {
5686                 nextp = dt_object_child(parent);
5687                 lp = lod_dt_obj(parent);
5688         }
5689
5690         nextc = dt_object_child(child);
5691         lc = lod_dt_obj(child);
5692
5693         LASSERT(!lod_obj_is_striped(child));
5694         /* default layout template may have been set on the regular file
5695          * when this is called from mdd_create_data() */
5696         if (S_ISREG(child_mode))
5697                 lod_free_comp_entries(lc);
5698
5699         if (!dt_object_exists(nextc))
5700                 nextc->do_ops->do_ah_init(env, ah, nextp, nextc, child_mode);
5701
5702         if (S_ISDIR(child_mode)) {
5703                 const struct lmv_user_md_v1 *lum1 = ah->dah_eadata;
5704
5705                 /* other default values are 0 */
5706                 lc->ldo_dir_stripe_offset = -1;
5707
5708                 /* no default striping configuration is needed for
5709                  * foreign dirs
5710                  */
5711                 if (ah->dah_eadata != NULL && ah->dah_eadata_len != 0 &&
5712                     le32_to_cpu(lum1->lum_magic) == LMV_MAGIC_FOREIGN) {
5713                         lc->ldo_is_foreign = true;
5714                         /* keep stripe_count 0 and stripe_offset -1 */
5715                         CDEBUG(D_INFO, "no default striping for foreign dir\n");
5716                         RETURN_EXIT;
5717                 }
5718
5719                 if (likely(lp != NULL))
5720                         lod_get_default_striping(env, lp, ah, lds);
5721
5722                 /* It should always honour the specified stripes */
5723                 /* Note: old client (< 2.7)might also do lfs mkdir, whose EA
5724                  * will have old magic. In this case, we should ignore the
5725                  * stripe count and try to create dir by default stripe.
5726                  */
5727                 if (ah->dah_eadata && ah->dah_eadata_len &&
5728                     !ah->dah_eadata_is_dmv &&
5729                     (le32_to_cpu(lum1->lum_magic) == LMV_USER_MAGIC ||
5730                      le32_to_cpu(lum1->lum_magic) == LMV_USER_MAGIC_SPECIFIC)) {
5731                         lc->ldo_dir_stripe_count =
5732                                 le32_to_cpu(lum1->lum_stripe_count);
5733                         lc->ldo_dir_stripe_offset =
5734                                 le32_to_cpu(lum1->lum_stripe_offset);
5735                         lc->ldo_dir_hash_type =
5736                                 le32_to_cpu(lum1->lum_hash_type);
5737                         CDEBUG(D_INFO,
5738                                "set dirstripe: count %hu, offset %d, hash %x\n",
5739                                 lc->ldo_dir_stripe_count,
5740                                 (int)lc->ldo_dir_stripe_offset,
5741                                 lc->ldo_dir_hash_type);
5742
5743                         if (d->lod_mdt_descs.ltd_lmv_desc.ld_active_tgt_count &&
5744                             lc->ldo_dir_stripe_count < 2 &&
5745                             lum1->lum_max_inherit != LMV_INHERIT_NONE) {
5746                                 /* when filesystem-wide default LMV is set, dirs
5747                                  * will be created on MDT by space usage, but if
5748                                  * dir is created with "lfs mkdir -c 1 ...", its
5749                                  * subdirs should be kept on the same MDT. To
5750                                  * guarantee this, set default LMV for such dir.
5751                                  */
5752                                 lds->lds_dir_def_stripe_count =
5753                                         le32_to_cpu(lum1->lum_stripe_count);
5754                                 /* if "-1" stripe offset is set, save current
5755                                  * MDT index in default LMV.
5756                                  */
5757                                 if (le32_to_cpu(lum1->lum_stripe_offset) ==
5758                                     LMV_OFFSET_DEFAULT)
5759                                         lds->lds_dir_def_stripe_offset =
5760                                                 lod2lu_dev(d)->ld_site->ld_seq_site->ss_node_id;
5761                                 else
5762                                         lds->lds_dir_def_stripe_offset =
5763                                                 le32_to_cpu(lum1->lum_stripe_offset);
5764                                 lds->lds_dir_def_hash_type =
5765                                         le32_to_cpu(lum1->lum_hash_type);
5766                                 lds->lds_dir_def_max_inherit =
5767                                         lum1->lum_max_inherit;
5768                                 /* it will be decreased by 1 later in setting */
5769                                 if (lum1->lum_max_inherit >= LMV_INHERIT_END &&
5770                                     lum1->lum_max_inherit < LMV_INHERIT_MAX)
5771                                         lds->lds_dir_def_max_inherit++;
5772                                 lds->lds_dir_def_max_inherit_rr =
5773                                         lum1->lum_max_inherit_rr;
5774                                 lds->lds_dir_def_striping_set = 1;
5775                                 /* don't inherit LOV from ROOT */
5776                                 if (lds->lds_def_striping_set &&
5777                                     fid_is_root(lod_object_fid(lp)))
5778                                         lds->lds_def_striping_set = 0;
5779                                 lc->ldo_def_striping = lds;
5780                         } else if (lds->lds_def_striping_set &&
5781                                    !fid_is_root(lod_object_fid(lp))) {
5782                                 /* don't inherit default LMV for "lfs mkdir" */
5783                                 lds->lds_dir_def_striping_set = 0;
5784                                 lc->ldo_def_striping = lds;
5785                         }
5786                 } else {
5787                         /* inherit default striping except ROOT */
5788                         if ((lds->lds_def_striping_set ||
5789                              lds->lds_dir_def_striping_set) &&
5790                             !fid_is_root(lod_object_fid(lp)))
5791                                 lc->ldo_def_striping = lds;
5792
5793                         /* transfer defaults LMV to new directory */
5794                         lod_striping_from_default(lc, lds, child_mode);
5795
5796                         /* set count 0 to create normal directory */
5797                         if (lc->ldo_dir_stripe_count == 1)
5798                                 lc->ldo_dir_stripe_count = 0;
5799
5800                         /* do not save default LMV on server */
5801                         if (ah->dah_dmv_imp_inherit) {
5802                                 lds->lds_dir_def_striping_set = 0;
5803                                 if (!lds->lds_def_striping_set)
5804                                         lc->ldo_def_striping = NULL;
5805                         }
5806                 }
5807
5808                 /* shrink the stripe count to max_mdt_stripecount if it is -1
5809                  * and max_mdt_stripecount is not 0
5810                  */
5811                 if (lc->ldo_dir_stripe_count == (__u16)(-1) &&
5812                     d->lod_max_mdt_stripecount)
5813                         lc->ldo_dir_stripe_count = d->lod_max_mdt_stripecount;
5814
5815                 /* shrink the stripe_count to the avaible MDT count */
5816                 if (lc->ldo_dir_stripe_count > d->lod_remote_mdt_count + 1 &&
5817                     !CFS_FAIL_CHECK(OBD_FAIL_LARGE_STRIPE)) {
5818                         lc->ldo_dir_stripe_count = d->lod_remote_mdt_count + 1;
5819                         if (lc->ldo_dir_stripe_count == 1)
5820                                 lc->ldo_dir_stripe_count = 0;
5821                 }
5822
5823                 if (!lmv_is_known_hash_type(lc->ldo_dir_hash_type))
5824                         lc->ldo_dir_hash_type =
5825                                 (lc->ldo_dir_hash_type & LMV_HASH_FLAG_KNOWN) |
5826                                 d->lod_mdt_descs.ltd_lmv_desc.ld_pattern;
5827
5828                 /* make sure all fscrypt metadata stays on same mdt */
5829                 if (child->do_lu.lo_header->loh_attr & LOHA_FSCRYPT_MD) {
5830                         lc->ldo_dir_stripe_count = 0;
5831                         lds->lds_dir_def_stripe_offset =
5832                                 lod2lu_dev(d)->ld_site->ld_seq_site->ss_node_id;
5833                         lds->lds_dir_def_striping_set = 1;
5834                         lc->ldo_def_striping = lds;
5835                 }
5836
5837                 CDEBUG(D_INFO, "final dir stripe_count=%hu offset=%d hash=%u\n",
5838                        lc->ldo_dir_stripe_count,
5839                        (int)lc->ldo_dir_stripe_offset, lc->ldo_dir_hash_type);
5840
5841                 RETURN_EXIT;
5842         }
5843
5844         /* child object regular file*/
5845
5846         if (!lod_object_will_be_striped(S_ISREG(child_mode),
5847                                         lu_object_fid(&child->do_lu)))
5848                 RETURN_EXIT;
5849
5850         /* If object is going to be striped over OSTs, transfer default
5851          * striping information to the child, so that we can use it
5852          * during declaration and creation.
5853          *
5854          * Try from the parent first.
5855          */
5856         if (likely(lp != NULL)) {
5857                 rc = lod_get_default_lov_striping(env, lp, lds, ah);
5858                 if (rc == 0 && lds->lds_def_striping_set) {
5859                         rc = lod_verify_striping(env, d, lp, &info->lti_buf,
5860                                                  false);
5861                         if (rc == 0)
5862                                 lod_striping_from_default(lc, lds, child_mode);
5863                 }
5864         }
5865
5866         /* Initialize lod_device::lod_md_root object reference */
5867         if (d->lod_md_root == NULL) {
5868                 struct dt_object *root;
5869                 struct lod_object *lroot;
5870
5871                 lu_root_fid(&info->lti_fid);
5872                 root = dt_locate(env, &d->lod_dt_dev, &info->lti_fid);
5873                 if (!IS_ERR(root)) {
5874                         lroot = lod_dt_obj(root);
5875
5876                         spin_lock(&d->lod_lock);
5877                         if (d->lod_md_root != NULL)
5878                                 dt_object_put(env, &d->lod_md_root->ldo_obj);
5879                         d->lod_md_root = lroot;
5880                         spin_unlock(&d->lod_lock);
5881                 }
5882         }
5883
5884         /* try inherit layout from the root object (fs default) when:
5885          *  - parent does not have default layout; or
5886          *  - parent has plain(v1/v3) default layout, and some attributes
5887          *    are not specified in the default layout;
5888          */
5889         if (d->lod_md_root != NULL &&
5890             lod_need_inherit_more(lc, true, ah->dah_append_pool)) {
5891                 rc = lod_get_default_lov_striping(env, d->lod_md_root, lds,
5892                                                   ah);
5893                 if (rc || !lds->lds_def_striping_set)
5894                         goto out;
5895
5896                 rc = lod_verify_striping(env, d, d->lod_md_root, &info->lti_buf,
5897                                          false);
5898                 if (rc)
5899                         goto out;
5900
5901                 if (lc->ldo_comp_cnt == 0) {
5902                         lod_striping_from_default(lc, lds, child_mode);
5903                 } else if (!lds->lds_def_striping_is_composite) {
5904                         struct lod_layout_component *def_comp;
5905
5906                         LASSERT(!lc->ldo_is_composite);
5907                         lod_comp = &lc->ldo_comp_entries[0];
5908                         def_comp = &lds->lds_def_comp_entries[0];
5909
5910                         if (lod_comp->llc_stripe_count <= 0)
5911                                 lod_comp->llc_stripe_count =
5912                                         def_comp->llc_stripe_count;
5913                         if (lod_comp->llc_stripe_size <= 0)
5914                                 lod_comp->llc_stripe_size =
5915                                         def_comp->llc_stripe_size;
5916                         if (lod_comp->llc_stripe_offset == LOV_OFFSET_DEFAULT &&
5917                             (!lod_comp->llc_pool || !lod_comp->llc_pool[0]))
5918                                 lod_comp->llc_stripe_offset =
5919                                         def_comp->llc_stripe_offset;
5920                         if (lod_comp->llc_pool == NULL)
5921                                 lod_obj_set_pool(lc, 0, def_comp->llc_pool);
5922                 }
5923         }
5924 out:
5925         /*
5926          * fs default striping may not be explicitly set, or historically set
5927          * in config log, use them.
5928          */
5929         if (lod_need_inherit_more(lc, false, ah->dah_append_pool)) {
5930                 if (lc->ldo_comp_cnt == 0) {
5931                         rc = lod_alloc_comp_entries(lc, 0, 1);
5932                         if (rc)
5933                                 /* fail to allocate memory, will create a
5934                                  * non-striped file. */
5935                                 RETURN_EXIT;
5936                         lc->ldo_is_composite = 0;
5937                         lod_comp = &lc->ldo_comp_entries[0];
5938                         lod_comp->llc_stripe_offset = LOV_OFFSET_DEFAULT;
5939                 }
5940                 LASSERT(!lc->ldo_is_composite);
5941                 lod_comp = &lc->ldo_comp_entries[0];
5942                 desc = &d->lod_ost_descs.ltd_lov_desc;
5943                 lod_adjust_stripe_info(lod_comp, desc,
5944                                        ah->dah_append_stripe_count);
5945                 if (ah->dah_append_pool && ah->dah_append_pool[0])
5946                         lod_obj_set_pool(lc, 0, ah->dah_append_pool);
5947         }
5948
5949         EXIT;
5950 }
5951
5952 /**
5953  * Size initialization on late striping.
5954  *
5955  * Propagate the size of a truncated object to a deferred striping.
5956  * This function handles a special case when truncate was done on a
5957  * non-striped object and now while the striping is being created
5958  * we can't lose that size, so we have to propagate it to the stripes
5959  * being created.
5960  *
5961  * \param[in] env       execution environment
5962  * \param[in] dt        object
5963  * \param[in] th        transaction handle
5964  *
5965  * \retval              0 on success
5966  * \retval              negative if failed
5967  */
5968 static int lod_declare_init_size(const struct lu_env *env,
5969                                  struct dt_object *dt, struct thandle *th)
5970 {
5971         struct dt_object        *next = dt_object_child(dt);
5972         struct lod_object       *lo = lod_dt_obj(dt);
5973         struct dt_object        **objects = NULL;
5974         struct lu_attr  *attr = &lod_env_info(env)->lti_attr;
5975         uint64_t        size, offs;
5976         int     i, rc, stripe, stripe_count = 0, stripe_size = 0;
5977         struct lu_extent size_ext;
5978         ENTRY;
5979
5980         if (!lod_obj_is_striped(dt))
5981                 RETURN(0);
5982
5983         rc = dt_attr_get(env, next, attr);
5984         LASSERT(attr->la_valid & LA_SIZE);
5985         if (rc)
5986                 RETURN(rc);
5987
5988         size = attr->la_size;
5989         if (size == 0)
5990                 RETURN(0);
5991
5992         size_ext = (typeof(size_ext)){ .e_start = size - 1, .e_end = size };
5993         for (i = 0; i < lo->ldo_comp_cnt; i++) {
5994                 struct lod_layout_component *lod_comp;
5995                 struct lu_extent *extent;
5996
5997                 lod_comp = &lo->ldo_comp_entries[i];
5998
5999                 if (lod_comp->llc_stripe == NULL)
6000                         continue;
6001
6002                 extent = &lod_comp->llc_extent;
6003                 CDEBUG(D_INFO, "%lld "DEXT"\n", size, PEXT(extent));
6004                 if (!lo->ldo_is_composite ||
6005                     lu_extent_is_overlapped(extent, &size_ext)) {
6006                         objects = lod_comp->llc_stripe;
6007                         stripe_count = lod_comp->llc_stripe_count;
6008                         stripe_size = lod_comp->llc_stripe_size;
6009
6010                         /* next mirror */
6011                         if (stripe_count == 0)
6012                                 continue;
6013
6014                         LASSERT(objects != NULL && stripe_size != 0);
6015                         do_div(size, stripe_size);
6016                         stripe = do_div(size, stripe_count);
6017                         LASSERT(objects[stripe] != NULL);
6018
6019                         size = size * stripe_size;
6020                         offs = attr->la_size;
6021                         size += do_div(offs, stripe_size);
6022
6023                         attr->la_valid = LA_SIZE;
6024                         attr->la_size = size;
6025
6026                         rc = lod_sub_declare_attr_set(env, objects[stripe],
6027                                                       attr, th);
6028                 }
6029         }
6030
6031         RETURN(rc);
6032 }
6033
6034 /**
6035  * Declare creation of striped object.
6036  *
6037  * The function declares creation stripes for a regular object. The function
6038  * also declares whether the stripes will be created with non-zero size if
6039  * previously size was set non-zero on the master object. If object \a dt is
6040  * not local, then only fully defined striping can be applied in \a lovea.
6041  * Otherwise \a lovea can be in the form of pattern, see lod_qos_parse_config()
6042  * for the details.
6043  *
6044  * \param[in] env       execution environment
6045  * \param[in] dt        object
6046  * \param[in] attr      attributes the stripes will be created with
6047  * \param[in] lovea     a buffer containing striping description
6048  * \param[in] th        transaction handle
6049  *
6050  * \retval              0 on success
6051  * \retval              negative if failed
6052  */
6053 int lod_declare_striped_create(const struct lu_env *env, struct dt_object *dt,
6054                                struct lu_attr *attr,
6055                                const struct lu_buf *lovea, struct thandle *th)
6056 {
6057         struct lod_thread_info  *info = lod_env_info(env);
6058         struct dt_object        *next = dt_object_child(dt);
6059         struct lod_object       *lo = lod_dt_obj(dt);
6060         int                      rc;
6061         ENTRY;
6062
6063         if (CFS_FAIL_CHECK(OBD_FAIL_MDS_ALLOC_OBDO))
6064                 GOTO(out, rc = -ENOMEM);
6065
6066         if (!dt_object_remote(next)) {
6067                 /* choose OST and generate appropriate objects */
6068                 rc = lod_prepare_create(env, lo, attr, lovea, th);
6069                 if (rc)
6070                         GOTO(out, rc);
6071
6072                 /*
6073                  * declare storage for striping data
6074                  */
6075                 info->lti_buf.lb_len = lod_comp_md_size(lo, false);
6076         } else {
6077                 /* LOD can not choose OST objects for remote objects, i.e.
6078                  * stripes must be ready before that. Right now, it can only
6079                  * happen during migrate, i.e. migrate process needs to create
6080                  * remote regular file (mdd_migrate_create), then the migrate
6081                  * process will provide stripeEA. */
6082                 LASSERT(lovea != NULL);
6083                 info->lti_buf = *lovea;
6084         }
6085
6086         rc = lod_sub_declare_xattr_set(env, next, &info->lti_buf,
6087                                        XATTR_NAME_LOV, 0, th);
6088         if (rc)
6089                 GOTO(out, rc);
6090
6091         /*
6092          * if striping is created with local object's size > 0,
6093          * we have to propagate this size to specific object
6094          * the case is possible only when local object was created previously
6095          */
6096         if (dt_object_exists(next))
6097                 rc = lod_declare_init_size(env, dt, th);
6098
6099 out:
6100         /* failed to create striping or to set initial size, let's reset
6101          * config so that others don't get confused */
6102         if (rc)
6103                 lod_striping_free(env, lo);
6104
6105         RETURN(rc);
6106 }
6107
6108 /*
6109  * Whether subdirectories under \a dt should be created on MDTs by space QoS
6110  *
6111  * If LMV_HASH_FLAG_SPACE is set on directory default layout, its subdirectories
6112  * should be created on MDT by space QoS.
6113  *
6114  * \param[in] env       execution environment
6115  * \param[in] dev       lu device
6116  * \param[in] dt        object
6117  *
6118  * \retval              1 if directory should create subdir by space usage
6119  * \retval              0 if not
6120  * \retval              -ev if failed
6121  */
6122 static inline int dt_object_qos_mkdir(const struct lu_env *env,
6123                                       struct lu_device *dev,
6124                                       struct dt_object *dt)
6125 {
6126         struct lod_thread_info *info = lod_env_info(env);
6127         struct lu_object *obj;
6128         struct lod_object *lo;
6129         struct lmv_user_md *lmu;
6130         int rc;
6131
6132         obj = lu_object_find_slice(env, dev, lu_object_fid(&dt->do_lu), NULL);
6133         if (IS_ERR(obj))
6134                 return PTR_ERR(obj);
6135
6136         lo = lu2lod_obj(obj);
6137
6138         rc = lod_get_default_lmv_ea(env, lo);
6139         dt_object_put(env, dt);
6140         if (rc <= 0)
6141                 return rc;
6142
6143         if (rc < (int)sizeof(*lmu))
6144                 return -EINVAL;
6145
6146         lmu = info->lti_ea_store;
6147         return le32_to_cpu(lmu->lum_stripe_offset) == LMV_OFFSET_DEFAULT;
6148 }
6149
6150 /**
6151  * Implementation of dt_object_operations::do_declare_create.
6152  *
6153  * The method declares creation of a new object. If the object will be striped,
6154  * then helper functions are called to find FIDs for the stripes, declare
6155  * creation of the stripes and declare initialization of the striping
6156  * information to be stored in the master object.
6157  *
6158  * \see dt_object_operations::do_declare_create() in the API description
6159  * for details.
6160  */
6161 static int lod_declare_create(const struct lu_env *env, struct dt_object *dt,
6162                               struct lu_attr *attr,
6163                               struct dt_allocation_hint *hint,
6164                               struct dt_object_format *dof, struct thandle *th)
6165 {
6166         struct dt_object   *next = dt_object_child(dt);
6167         struct lod_object  *lo = lod_dt_obj(dt);
6168         int                 rc;
6169         ENTRY;
6170
6171         LASSERT(dof);
6172         LASSERT(attr);
6173         LASSERT(th);
6174
6175         /*
6176          * first of all, we declare creation of local object
6177          */
6178         rc = lod_sub_declare_create(env, next, attr, hint, dof, th);
6179         if (rc != 0)
6180                 GOTO(out, rc);
6181
6182         /*
6183          * it's lod_ah_init() that has decided the object will be striped
6184          */
6185         if (dof->dof_type == DFT_REGULAR) {
6186                 /* callers don't want stripes */
6187                 /* XXX: all tricky interactions with ->ah_make_hint() decided
6188                  * to use striping, then ->declare_create() behaving differently
6189                  * should be cleaned */
6190                 if (dof->u.dof_reg.striped != 0)
6191                         rc = lod_declare_striped_create(env, dt, attr,
6192                                                         NULL, th);
6193         } else if (dof->dof_type == DFT_DIR) {
6194                 struct seq_server_site *ss;
6195                 struct lu_buf buf = { NULL };
6196                 struct lu_buf *lmu = NULL;
6197
6198                 ss = lu_site2seq(dt->do_lu.lo_dev->ld_site);
6199
6200                 /* If the parent has default stripeEA, and client
6201                  * did not find it before sending create request,
6202                  * then MDT will return -EREMOTE, and client will
6203                  * retrieve the default stripeEA and re-create the
6204                  * sub directory.
6205                  *
6206                  * Note: if dah_eadata != NULL, it means creating the
6207                  * striped directory with specified stripeEA, then it
6208                  * should ignore the default stripeEA */
6209                 if (hint != NULL && hint->dah_eadata == NULL) {
6210                         if (CFS_FAIL_CHECK(OBD_FAIL_MDS_STALE_DIR_LAYOUT))
6211                                 GOTO(out, rc = -EREMOTE);
6212
6213                         if (lo->ldo_dir_stripe_offset != LMV_OFFSET_DEFAULT &&
6214                             lo->ldo_dir_stripe_offset != ss->ss_node_id) {
6215                                 struct lod_device *lod;
6216                                 struct lu_tgt_desc *mdt = NULL;
6217                                 bool found_mdt = false;
6218
6219                                 lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
6220                                 lod_foreach_mdt(lod, mdt) {
6221                                         if (mdt->ltd_index ==
6222                                                 lo->ldo_dir_stripe_offset) {
6223                                                 found_mdt = true;
6224                                                 break;
6225                                         }
6226                                 }
6227
6228                                 /* If the MDT indicated by stripe_offset can be
6229                                  * found, then tell client to resend the create
6230                                  * request to the correct MDT, otherwise return
6231                                  * error to client */
6232                                 if (found_mdt)
6233                                         GOTO(out, rc = -EREMOTE);
6234                                 else
6235                                         GOTO(out, rc = -EINVAL);
6236                         }
6237                 } else if (hint && hint->dah_eadata) {
6238                         lmu = &buf;
6239                         lmu->lb_buf = (void *)hint->dah_eadata;
6240                         lmu->lb_len = hint->dah_eadata_len;
6241                 }
6242
6243                 rc = lod_declare_dir_striping_create(env, dt, attr, lmu, dof,
6244                                                      th);
6245         }
6246 out:
6247         /* failed to create striping or to set initial size, let's reset
6248          * config so that others don't get confused */
6249         if (rc)
6250                 lod_striping_free(env, lo);
6251         RETURN(rc);
6252 }
6253
6254 /**
6255  * Generate component ID for new created component.
6256  *
6257  * \param[in] lo                LOD object
6258  * \param[in] comp_idx          index of ldo_comp_entries
6259  *
6260  * \retval                      component ID on success
6261  * \retval                      LCME_ID_INVAL on failure
6262  */
6263 static __u32 lod_gen_component_id(struct lod_object *lo,
6264                                   int mirror_id, int comp_idx)
6265 {
6266         struct lod_layout_component *lod_comp;
6267         __u32   id, start, end;
6268         int     i;
6269
6270         LASSERT(lo->ldo_comp_entries[comp_idx].llc_id == LCME_ID_INVAL);
6271
6272         lod_obj_inc_layout_gen(lo);
6273         id = lo->ldo_layout_gen;
6274         if (likely(id <= SEQ_ID_MAX))
6275                 RETURN(pflr_id(mirror_id, id & SEQ_ID_MASK));
6276
6277         /* Layout generation wraps, need to check collisions. */
6278         start = id & SEQ_ID_MASK;
6279         end = SEQ_ID_MAX;
6280 again:
6281         for (id = start; id <= end; id++) {
6282                 for (i = 0; i < lo->ldo_comp_cnt; i++) {
6283                         lod_comp = &lo->ldo_comp_entries[i];
6284                         if (pflr_id(mirror_id, id) == lod_comp->llc_id)
6285                                 break;
6286                 }
6287                 /* Found the ununsed ID */
6288                 if (i == lo->ldo_comp_cnt)
6289                         RETURN(pflr_id(mirror_id, id));
6290         }
6291
6292         if (end == SEQ_ID_MAX) {
6293                 end = min_t(__u32, start, SEQ_ID_MAX) - 1;
6294                 start = 1;
6295                 goto again;
6296         }
6297
6298         RETURN(LCME_ID_INVAL);
6299 }
6300
6301 /**
6302  * Creation of a striped regular object.
6303  *
6304  * The function is called to create the stripe objects for a regular
6305  * striped file. This can happen at the initial object creation or
6306  * when the caller asks LOD to do so using ->do_xattr_set() method
6307  * (so called late striping). Notice all the information are already
6308  * prepared in the form of the list of objects (ldo_stripe field).
6309  * This is done during declare phase.
6310  *
6311  * \param[in] env       execution environment
6312  * \param[in] dt        object
6313  * \param[in] attr      attributes the stripes will be created with
6314  * \param[in] dof       format of stripes (see OSD API description)
6315  * \param[in] th        transaction handle
6316  *
6317  * \retval              0 on success
6318  * \retval              negative if failed
6319  */
6320 int lod_striped_create(const struct lu_env *env, struct dt_object *dt,
6321                        struct lu_attr *attr, struct dt_object_format *dof,
6322                        struct thandle *th)
6323 {
6324         struct lod_layout_component     *lod_comp;
6325         struct lod_object       *lo = lod_dt_obj(dt);
6326         __u16   mirror_id;
6327         int     rc = 0, i, j;
6328         ENTRY;
6329
6330         mutex_lock(&lo->ldo_layout_mutex);
6331
6332         LASSERT((lo->ldo_comp_cnt != 0 && lo->ldo_comp_entries != NULL) ||
6333                 lo->ldo_is_foreign);
6334
6335         mirror_id = 0; /* non-flr file's mirror_id is 0 */
6336         if (lo->ldo_mirror_count > 1) {
6337                 for (i = 0; i < lo->ldo_comp_cnt; i++) {
6338                         lod_comp = &lo->ldo_comp_entries[i];
6339                         if (lod_comp->llc_id != LCME_ID_INVAL &&
6340                             mirror_id_of(lod_comp->llc_id) > mirror_id)
6341                                 mirror_id = mirror_id_of(lod_comp->llc_id);
6342                 }
6343         }
6344
6345         /* create all underlying objects */
6346         for (i = 0; i < lo->ldo_comp_cnt; i++) {
6347                 lod_comp = &lo->ldo_comp_entries[i];
6348
6349                 if (lod_comp->llc_id == LCME_ID_INVAL) {
6350                         /* only the component of FLR layout with more than 1
6351                          * mirror has mirror ID in its component ID.
6352                          */
6353                         if (lod_comp->llc_extent.e_start == 0 &&
6354                             lo->ldo_mirror_count > 1)
6355                                 ++mirror_id;
6356
6357                         lod_comp->llc_id = lod_gen_component_id(lo,
6358                                                                 mirror_id, i);
6359                         if (lod_comp->llc_id == LCME_ID_INVAL)
6360                                 GOTO(out, rc = -ERANGE);
6361                 }
6362
6363                 if (lod_comp_inited(lod_comp))
6364                         continue;
6365
6366                 if (lod_comp->llc_pattern & LOV_PATTERN_F_RELEASED)
6367                         lod_comp_set_init(lod_comp);
6368
6369                 if (lov_pattern(lod_comp->llc_pattern) & LOV_PATTERN_MDT)
6370                         lod_comp_set_init(lod_comp);
6371
6372                 if (lod_comp->llc_stripe == NULL)
6373                         continue;
6374
6375                 LASSERT(lod_comp->llc_stripe_count);
6376                 for (j = 0; j < lod_comp->llc_stripe_count; j++) {
6377                         struct dt_object *object = lod_comp->llc_stripe[j];
6378                         LASSERT(object != NULL);
6379                         rc = lod_sub_create(env, object, attr, NULL, dof, th);
6380                         if (rc)
6381                                 GOTO(out, rc);
6382                 }
6383                 lod_comp_set_init(lod_comp);
6384         }
6385
6386         rc = lod_fill_mirrors(lo);
6387         if (rc)
6388                 GOTO(out, rc);
6389
6390         lo->ldo_comp_cached = 1;
6391
6392         rc = lod_generate_and_set_lovea(env, lo, th);
6393         if (rc)
6394                 GOTO(out, rc);
6395
6396         mutex_unlock(&lo->ldo_layout_mutex);
6397
6398         RETURN(0);
6399
6400 out:
6401         lod_striping_free_nolock(env, lo);
6402         mutex_unlock(&lo->ldo_layout_mutex);
6403
6404         RETURN(rc);
6405 }
6406
6407 static inline bool lod_obj_is_dom(struct dt_object *dt)
6408 {
6409         struct lod_object *lo = lod_dt_obj(dt);
6410
6411         if (!dt_object_exists(dt_object_child(dt)))
6412                 return false;
6413
6414         if (S_ISDIR(dt->do_lu.lo_header->loh_attr))
6415                 return false;
6416
6417         if (!lo->ldo_comp_cnt)
6418                 return false;
6419
6420         return (lov_pattern(lo->ldo_comp_entries[0].llc_pattern) &
6421                 LOV_PATTERN_MDT);
6422 }
6423
6424 /**
6425  * Implementation of dt_object_operations::do_create.
6426  *
6427  * If any of preceeding methods (like ->do_declare_create(),
6428  * ->do_ah_init(), etc) chose to create a striped object,
6429  * then this method will create the master and the stripes.
6430  *
6431  * \see dt_object_operations::do_create() in the API description for details.
6432  */
6433 static int lod_create(const struct lu_env *env, struct dt_object *dt,
6434                       struct lu_attr *attr, struct dt_allocation_hint *hint,
6435                       struct dt_object_format *dof, struct thandle *th)
6436 {
6437         int                 rc;
6438         ENTRY;
6439
6440         /* create local object */
6441         rc = lod_sub_create(env, dt_object_child(dt), attr, hint, dof, th);
6442         if (rc != 0)
6443                 RETURN(rc);
6444
6445         if (S_ISREG(dt->do_lu.lo_header->loh_attr) &&
6446             (lod_obj_is_striped(dt) || lod_obj_is_dom(dt)) &&
6447             dof->u.dof_reg.striped != 0) {
6448                 LASSERT(lod_dt_obj(dt)->ldo_comp_cached == 0);
6449                 rc = lod_striped_create(env, dt, attr, dof, th);
6450         }
6451
6452         RETURN(rc);
6453 }
6454
6455 static inline int
6456 lod_obj_stripe_destroy_cb(const struct lu_env *env, struct lod_object *lo,
6457                           struct dt_object *dt, struct thandle *th,
6458                           int comp_idx, int stripe_idx,
6459                           struct lod_obj_stripe_cb_data *data)
6460 {
6461         if (data->locd_declare)
6462                 return lod_sub_declare_destroy(env, dt, th);
6463
6464         if (!CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_SPEOBJ) ||
6465             stripe_idx == cfs_fail_val)
6466                 return lod_sub_destroy(env, dt, th);
6467
6468         return 0;
6469 }
6470
6471 /**
6472  * Implementation of dt_object_operations::do_declare_destroy.
6473  *
6474  * If the object is a striped directory, then the function declares reference
6475  * removal from the master object (this is an index) to the stripes and declares
6476  * destroy of all the stripes. In all the cases, it declares an intention to
6477  * destroy the object itself.
6478  *
6479  * \see dt_object_operations::do_declare_destroy() in the API description
6480  * for details.
6481  */
6482 static int lod_declare_destroy(const struct lu_env *env, struct dt_object *dt,
6483                                struct thandle *th)
6484 {
6485         struct dt_object *next = dt_object_child(dt);
6486         struct lod_object *lo = lod_dt_obj(dt);
6487         struct lod_thread_info *info = lod_env_info(env);
6488         struct dt_object *stripe;
6489         char *stripe_name = info->lti_key;
6490         int rc, i;
6491
6492         ENTRY;
6493
6494         /*
6495          * load striping information, notice we don't do this when object
6496          * is being initialized as we don't need this information till
6497          * few specific cases like destroy, chown
6498          */
6499         rc = lod_striping_load(env, lo);
6500         if (rc)
6501                 RETURN(rc);
6502
6503         /* declare destroy for all underlying objects */
6504         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
6505                 rc = next->do_ops->do_index_try(env, next,
6506                                                 &dt_directory_features);
6507                 if (rc != 0)
6508                         RETURN(rc);
6509
6510                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6511                         stripe = lo->ldo_stripe[i];
6512                         if (!stripe)
6513                                 continue;
6514
6515                         rc = lod_sub_declare_ref_del(env, next, th);
6516                         if (rc != 0)
6517                                 RETURN(rc);
6518
6519                         snprintf(stripe_name, sizeof(info->lti_key),
6520                                  DFID":%d",
6521                                  PFID(lu_object_fid(&stripe->do_lu)), i);
6522                         rc = lod_sub_declare_delete(env, next,
6523                                         (const struct dt_key *)stripe_name, th);
6524                         if (rc != 0)
6525                                 RETURN(rc);
6526                 }
6527         }
6528
6529         /*
6530          * we declare destroy for the local object
6531          */
6532         rc = lod_sub_declare_destroy(env, next, th);
6533         if (rc)
6534                 RETURN(rc);
6535
6536         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_MDTOBJ) ||
6537             CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_MDTOBJ2))
6538                 RETURN(0);
6539
6540         if (!lod_obj_is_striped(dt))
6541                 RETURN(0);
6542
6543         /* declare destroy all striped objects */
6544         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
6545                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6546                         stripe = lo->ldo_stripe[i];
6547                         if (!stripe)
6548                                 continue;
6549
6550                         if (!dt_object_exists(stripe))
6551                                 continue;
6552
6553                         rc = lod_sub_declare_ref_del(env, stripe, th);
6554                         if (rc != 0)
6555                                 break;
6556
6557                         rc = lod_sub_declare_destroy(env, stripe, th);
6558                         if (rc != 0)
6559                                 break;
6560                 }
6561         } else {
6562                 struct lod_obj_stripe_cb_data data = { { 0 } };
6563
6564                 data.locd_declare = true;
6565                 data.locd_stripe_cb = lod_obj_stripe_destroy_cb;
6566                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
6567         }
6568
6569         RETURN(rc);
6570 }
6571
6572 /**
6573  * Implementation of dt_object_operations::do_destroy.
6574  *
6575  * If the object is a striped directory, then the function removes references
6576  * from the master object (this is an index) to the stripes and destroys all
6577  * the stripes. In all the cases, the function destroys the object itself.
6578  *
6579  * \see dt_object_operations::do_destroy() in the API description for details.
6580  */
6581 static int lod_destroy(const struct lu_env *env, struct dt_object *dt,
6582                        struct thandle *th)
6583 {
6584         struct dt_object  *next = dt_object_child(dt);
6585         struct lod_object *lo = lod_dt_obj(dt);
6586         struct lod_thread_info *info = lod_env_info(env);
6587         char *stripe_name = info->lti_key;
6588         struct dt_object *stripe;
6589         unsigned int i;
6590         int rc;
6591
6592         ENTRY;
6593
6594         /* destroy sub-stripe of master object */
6595         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
6596                 rc = next->do_ops->do_index_try(env, next,
6597                                                 &dt_directory_features);
6598                 if (rc != 0)
6599                         RETURN(rc);
6600
6601                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6602                         stripe = lo->ldo_stripe[i];
6603                         if (!stripe)
6604                                 continue;
6605
6606                         rc = lod_sub_ref_del(env, next, th);
6607                         if (rc != 0)
6608                                 RETURN(rc);
6609
6610                         snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
6611                                 PFID(lu_object_fid(&stripe->do_lu)), i);
6612
6613                         CDEBUG(D_INFO, DFID" delete stripe %s "DFID"\n",
6614                                PFID(lu_object_fid(&dt->do_lu)), stripe_name,
6615                                PFID(lu_object_fid(&stripe->do_lu)));
6616
6617                         rc = lod_sub_delete(env, next,
6618                                        (const struct dt_key *)stripe_name, th);
6619                         if (rc != 0)
6620                                 RETURN(rc);
6621                 }
6622         }
6623
6624         rc = lod_sub_destroy(env, next, th);
6625         if (rc != 0)
6626                 RETURN(rc);
6627
6628         if (CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_MDTOBJ) ||
6629             CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_MDTOBJ2))
6630                 RETURN(0);
6631
6632         if (!lod_obj_is_striped(dt))
6633                 RETURN(0);
6634
6635         /* destroy all striped objects */
6636         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
6637                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6638                         stripe = lo->ldo_stripe[i];
6639                         if (!stripe)
6640                                 continue;
6641
6642                         if (!dt_object_exists(stripe))
6643                                 continue;
6644
6645                         if (!CFS_FAIL_CHECK(OBD_FAIL_LFSCK_LOST_SPEOBJ) ||
6646                             i == cfs_fail_val) {
6647                                 dt_write_lock(env, stripe, DT_TGT_CHILD);
6648                                 rc = lod_sub_ref_del(env, stripe, th);
6649                                 dt_write_unlock(env, stripe);
6650                                 if (rc != 0)
6651                                         break;
6652
6653                                 rc = lod_sub_destroy(env, stripe, th);
6654                                 if (rc != 0)
6655                                         break;
6656                         }
6657                 }
6658         } else {
6659                 struct lod_obj_stripe_cb_data data = { { 0 } };
6660
6661                 data.locd_declare = false;
6662                 data.locd_stripe_cb = lod_obj_stripe_destroy_cb;
6663                 rc = lod_obj_for_each_stripe(env, lo, th, &data);
6664         }
6665
6666         RETURN(rc);
6667 }
6668
6669 /**
6670  * Implementation of dt_object_operations::do_declare_ref_add.
6671  *
6672  * \see dt_object_operations::do_declare_ref_add() in the API description
6673  * for details.
6674  */
6675 static int lod_declare_ref_add(const struct lu_env *env,
6676                                struct dt_object *dt, struct thandle *th)
6677 {
6678         return lod_sub_declare_ref_add(env, dt_object_child(dt), th);
6679 }
6680
6681 /**
6682  * Implementation of dt_object_operations::do_ref_add.
6683  *
6684  * \see dt_object_operations::do_ref_add() in the API description for details.
6685  */
6686 static int lod_ref_add(const struct lu_env *env,
6687                        struct dt_object *dt, struct thandle *th)
6688 {
6689         return lod_sub_ref_add(env, dt_object_child(dt), th);
6690 }
6691
6692 /**
6693  * Implementation of dt_object_operations::do_declare_ref_del.
6694  *
6695  * \see dt_object_operations::do_declare_ref_del() in the API description
6696  * for details.
6697  */
6698 static int lod_declare_ref_del(const struct lu_env *env,
6699                                struct dt_object *dt, struct thandle *th)
6700 {
6701         return lod_sub_declare_ref_del(env, dt_object_child(dt), th);
6702 }
6703
6704 /**
6705  * Implementation of dt_object_operations::do_ref_del
6706  *
6707  * \see dt_object_operations::do_ref_del() in the API description for details.
6708  */
6709 static int lod_ref_del(const struct lu_env *env,
6710                        struct dt_object *dt, struct thandle *th)
6711 {
6712         return lod_sub_ref_del(env, dt_object_child(dt), th);
6713 }
6714
6715 /**
6716  * Implementation of dt_object_operations::do_object_sync.
6717  *
6718  * \see dt_object_operations::do_object_sync() in the API description
6719  * for details.
6720  */
6721 static int lod_object_sync(const struct lu_env *env, struct dt_object *dt,
6722                            __u64 start, __u64 end)
6723 {
6724         return dt_object_sync(env, dt_object_child(dt), start, end);
6725 }
6726
6727 /**
6728  * Implementation of dt_object_operations::do_object_unlock.
6729  *
6730  * Used to release LDLM lock(s).
6731  *
6732  * \see dt_object_operations::do_object_unlock() in the API description
6733  * for details.
6734  */
6735 static int lod_object_unlock(const struct lu_env *env, struct dt_object *dt,
6736                              struct ldlm_enqueue_info *einfo,
6737                              union ldlm_policy_data *policy)
6738 {
6739         struct lod_object *lo = lod_dt_obj(dt);
6740         struct lustre_handle_array *slave_locks = einfo->ei_cbdata;
6741         int slave_locks_size;
6742         int i;
6743         ENTRY;
6744
6745         if (slave_locks == NULL)
6746                 RETURN(0);
6747
6748         LASSERT(S_ISDIR(dt->do_lu.lo_header->loh_attr));
6749         /* Note: for remote lock for single stripe dir, MDT will cancel
6750          * the lock by lockh directly */
6751         LASSERT(!dt_object_remote(dt_object_child(dt)));
6752
6753         /* locks were unlocked in MDT layer */
6754         for (i = 0; i < slave_locks->ha_count; i++)
6755                 LASSERT(!lustre_handle_is_used(&slave_locks->ha_handles[i]));
6756
6757         /*
6758          * NB, ha_count may not equal to ldo_dir_stripe_count, because dir
6759          * layout may change, e.g., shrink dir layout after migration.
6760          */
6761         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6762                 if (lo->ldo_stripe[i])
6763                         dt_invalidate(env, lo->ldo_stripe[i]);
6764         }
6765
6766         slave_locks_size = offsetof(typeof(*slave_locks),
6767                                     ha_handles[slave_locks->ha_count]);
6768         OBD_FREE(slave_locks, slave_locks_size);
6769         einfo->ei_cbdata = NULL;
6770
6771         RETURN(0);
6772 }
6773
6774 /**
6775  * Implementation of dt_object_operations::do_object_lock.
6776  *
6777  * Used to get LDLM lock on the non-striped and striped objects.
6778  *
6779  * \see dt_object_operations::do_object_lock() in the API description
6780  * for details.
6781  */
6782 static int lod_object_lock(const struct lu_env *env,
6783                            struct dt_object *dt,
6784                            struct lustre_handle *lh,
6785                            struct ldlm_enqueue_info *einfo,
6786                            union ldlm_policy_data *policy)
6787 {
6788         struct lod_object *lo = lod_dt_obj(dt);
6789         int slave_locks_size;
6790         struct lustre_handle_array *slave_locks = NULL;
6791         int i;
6792         int rc;
6793         ENTRY;
6794
6795         /* remote object lock */
6796         if (!einfo->ei_enq_slave) {
6797                 LASSERT(dt_object_remote(dt));
6798                 return dt_object_lock(env, dt_object_child(dt), lh, einfo,
6799                                       policy);
6800         }
6801
6802         if (!S_ISDIR(dt->do_lu.lo_header->loh_attr))
6803                 RETURN(-ENOTDIR);
6804
6805         rc = lod_striping_load(env, lo);
6806         if (rc != 0)
6807                 RETURN(rc);
6808
6809         /* No stripes */
6810         if (lo->ldo_dir_stripe_count <= 1)
6811                 RETURN(0);
6812
6813         slave_locks_size = offsetof(typeof(*slave_locks),
6814                                     ha_handles[lo->ldo_dir_stripe_count]);
6815         /* Freed in lod_object_unlock */
6816         OBD_ALLOC(slave_locks, slave_locks_size);
6817         if (!slave_locks)
6818                 RETURN(-ENOMEM);
6819         slave_locks->ha_count = lo->ldo_dir_stripe_count;
6820
6821         /* striped directory lock */
6822         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
6823                 struct lustre_handle lockh;
6824                 struct ldlm_res_id *res_id;
6825                 struct dt_object *stripe;
6826
6827                 stripe = lo->ldo_stripe[i];
6828                 if (!stripe)
6829                         continue;
6830
6831                 res_id = &lod_env_info(env)->lti_res_id;
6832                 fid_build_reg_res_name(lu_object_fid(&stripe->do_lu), res_id);
6833                 einfo->ei_res_id = res_id;
6834
6835                 if (dt_object_remote(stripe)) {
6836                         set_bit(i, (void *)slave_locks->ha_map);
6837                         rc = dt_object_lock(env, stripe, &lockh, einfo, policy);
6838                 } else {
6839                         struct ldlm_namespace *ns = einfo->ei_namespace;
6840                         ldlm_blocking_callback blocking = einfo->ei_cb_local_bl;
6841                         ldlm_completion_callback completion = einfo->ei_cb_cp;
6842                         __u64 dlmflags = LDLM_FL_ATOMIC_CB;
6843
6844                         LASSERT(ns != NULL);
6845                         rc = ldlm_cli_enqueue_local(env, ns, res_id, LDLM_IBITS,
6846                                                     policy, einfo->ei_mode,
6847                                                     &dlmflags, blocking,
6848                                                     completion, NULL,
6849                                                     NULL, 0, LVB_T_NONE,
6850                                                     NULL, &lockh);
6851                 }
6852                 if (rc) {
6853                         while (i--)
6854                                 ldlm_lock_decref_and_cancel(
6855                                                 &slave_locks->ha_handles[i],
6856                                                 einfo->ei_mode);
6857                         OBD_FREE(slave_locks, slave_locks_size);
6858                         RETURN(rc);
6859                 }
6860                 slave_locks->ha_handles[i] = lockh;
6861         }
6862         einfo->ei_cbdata = slave_locks;
6863
6864         RETURN(0);
6865 }
6866
6867 /**
6868  * Implementation of dt_object_operations::do_invalidate.
6869  *
6870  * \see dt_object_operations::do_invalidate() in the API description for details
6871  */
6872 static int lod_invalidate(const struct lu_env *env, struct dt_object *dt)
6873 {
6874         return dt_invalidate(env, dt_object_child(dt));
6875 }
6876
6877 static int lod_declare_instantiate_components(const struct lu_env *env,
6878                                               struct lod_object *lo,
6879                                               struct thandle *th,
6880                                               __u64 reserve)
6881 {
6882         struct lod_thread_info *info = lod_env_info(env);
6883         int i;
6884         int rc = 0;
6885         ENTRY;
6886
6887         LASSERT(info->lti_count < lo->ldo_comp_cnt);
6888
6889         for (i = 0; i < info->lti_count; i++) {
6890                 rc = lod_qos_prep_create(env, lo, NULL, th,
6891                                          info->lti_comp_idx[i], reserve);
6892                 if (rc)
6893                         break;
6894         }
6895
6896         if (!rc) {
6897                 info->lti_buf.lb_len = lod_comp_md_size(lo, false);
6898                 rc = lod_sub_declare_xattr_set(env, lod_object_child(lo),
6899                                 &info->lti_buf, XATTR_NAME_LOV, 0, th);
6900         }
6901
6902         RETURN(rc);
6903 }
6904
6905 /**
6906  * Check OSTs for an existing component for further extension
6907  *
6908  * Checks if OSTs are still healthy and not out of space.  Gets free space
6909  * on OSTs (relative to allocation watermark rmb_low) and compares to
6910  * the proposed new_end for this component.
6911  *
6912  * Decides whether or not to extend a component on its current OSTs.
6913  *
6914  * \param[in] env               execution environment for this thread
6915  * \param[in] lo                object we're checking
6916  * \param[in] index             index of this component
6917  * \param[in] extension_size    extension size for this component
6918  * \param[in] extent            layout extent for requested operation
6919  * \param[in] comp_extent       extension component extent
6920  * \param[in] write             if this is write operation
6921  *
6922  * \retval      true - OK to extend on current OSTs
6923  * \retval      false - do not extend on current OSTs
6924  */
6925 static bool lod_sel_osts_allowed(const struct lu_env *env,
6926                                  struct lod_object *lo,
6927                                  int index, __u64 reserve,
6928                                  struct lu_extent *extent,
6929                                  struct lu_extent *comp_extent, int write)
6930 {
6931         struct lod_layout_component *lod_comp = &lo->ldo_comp_entries[index];
6932         struct lod_device *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
6933         struct lod_thread_info *tinfo = lod_env_info(env);
6934         struct obd_statfs *sfs = &tinfo->lti_osfs;
6935         __u64 available = 0;
6936         bool ret = true;
6937         int i, rc;
6938
6939         ENTRY;
6940
6941         LASSERT(lod_comp->llc_stripe_count != 0);
6942
6943         lod_getref(&lod->lod_ost_descs);
6944         for (i = 0; i < lod_comp->llc_stripe_count; i++) {
6945                 int index = lod_comp->llc_ost_indices[i];
6946                 struct lod_tgt_desc *ost = OST_TGT(lod, index);
6947                 struct obd_statfs_info info = { 0 };
6948                 int j, repeated = 0;
6949
6950                 LASSERT(ost);
6951
6952                 /* Get the number of times this OST repeats in this component.
6953                  * Note: inter-component repeats are not counted as this is
6954                  * considered as a rare case: we try to not repeat OST in other
6955                  * components if possible. */
6956                 for (j = 0; j < lod_comp->llc_stripe_count; j++) {
6957                         if (index != lod_comp->llc_ost_indices[j])
6958                                 continue;
6959
6960                         /* already handled */
6961                         if (j < i)
6962                                 break;
6963
6964                         repeated++;
6965                 }
6966                 if (j < lod_comp->llc_stripe_count)
6967                         continue;
6968
6969                 if (!test_bit(index, lod->lod_ost_bitmap)) {
6970                         CDEBUG(D_LAYOUT, "ost %d no longer present\n", index);
6971                         ret = false;
6972                         break;
6973                 }
6974
6975                 rc = dt_statfs_info(env, ost->ltd_tgt, sfs, &info);
6976                 if (rc) {
6977                         CDEBUG(D_LAYOUT, "statfs failed for ost %d, error %d\n",
6978                                index, rc);
6979                         ret = false;
6980                         break;
6981                 }
6982
6983                 if (sfs->os_state & OS_STATFS_ENOSPC ||
6984                     sfs->os_state & OS_STATFS_READONLY ||
6985                     sfs->os_state & OS_STATFS_DEGRADED) {
6986                         CDEBUG(D_LAYOUT, "ost %d is not availble for SEL "
6987                                "extension, state %u\n", index, sfs->os_state);
6988                         ret = false;
6989                         break;
6990                 }
6991
6992                 /* In bytes */
6993                 available = sfs->os_bavail * sfs->os_bsize;
6994                 /* 'available' is relative to the allocation threshold */
6995                 available -= (__u64) info.os_reserved_mb_low << 20;
6996
6997                 CDEBUG(D_LAYOUT, "ost %d lowwm: %d highwm: %d, "
6998                        "%llu %% blocks available, %llu %% blocks free\n",
6999                        index, info.os_reserved_mb_low, info.os_reserved_mb_high,
7000                        (100ull * sfs->os_bavail) / sfs->os_blocks,
7001                        (100ull * sfs->os_bfree) / sfs->os_blocks);
7002
7003                 if (reserve * repeated > available) {
7004                         ret = false;
7005                         CDEBUG(D_LAYOUT, "low space on ost %d, available %llu "
7006                                "< extension size %llu repeated %d\n", index,
7007                                available, reserve, repeated);
7008                         break;
7009                 }
7010         }
7011         lod_putref(lod, &lod->lod_ost_descs);
7012
7013         RETURN(ret);
7014 }
7015
7016 /**
7017  * Adjust extents after component removal
7018  *
7019  * When we remove an extension component, we move the start of the next
7020  * component to match the start of the extension component, so no space is left
7021  * without layout.
7022  *
7023  * \param[in] env       execution environment for this thread
7024  * \param[in] lo        object
7025  * \param[in] max_comp  layout component
7026  * \param[in] index     index of this component
7027  *
7028  * \retval              0 on success
7029  * \retval              negative errno on error
7030  */
7031 static void lod_sel_adjust_extents(const struct lu_env *env,
7032                                    struct lod_object *lo,
7033                                    int max_comp, int index)
7034 {
7035         struct lod_layout_component *lod_comp = NULL;
7036         struct lod_layout_component *next = NULL;
7037         struct lod_layout_component *prev = NULL;
7038         __u64 new_start = 0;
7039         __u64 start;
7040         int i;
7041
7042         /* Extension space component */
7043         lod_comp = &lo->ldo_comp_entries[index];
7044         next = &lo->ldo_comp_entries[index + 1];
7045         prev = &lo->ldo_comp_entries[index - 1];
7046
7047         LASSERT(lod_comp != NULL && prev != NULL && next != NULL);
7048         LASSERT(lod_comp->llc_flags & LCME_FL_EXTENSION);
7049
7050         /* Previous is being removed */
7051         if (prev && prev->llc_id == LCME_ID_INVAL)
7052                 new_start = prev->llc_extent.e_start;
7053         else
7054                 new_start = lod_comp->llc_extent.e_start;
7055
7056         for (i = index + 1; i < max_comp; i++) {
7057                 lod_comp = &lo->ldo_comp_entries[i];
7058
7059                 start = lod_comp->llc_extent.e_start;
7060                 lod_comp->llc_extent.e_start = new_start;
7061
7062                 /* We only move zero length extendable components */
7063                 if (!(start == lod_comp->llc_extent.e_end))
7064                         break;
7065
7066                 LASSERT(!(lod_comp->llc_flags & LCME_FL_INIT));
7067
7068                 lod_comp->llc_extent.e_end = new_start;
7069         }
7070 }
7071
7072 /* Calculate the proposed 'new end' for a component we're extending */
7073 static __u64 lod_extension_new_end(__u64 extension_size, __u64 extent_end,
7074                                    __u32 stripe_size, __u64 component_end,
7075                                    __u64 extension_end)
7076 {
7077         __u64 new_end;
7078
7079         LASSERT(extension_size != 0 && stripe_size != 0);
7080
7081         /* Round up to extension size */
7082         if (extent_end == OBD_OBJECT_EOF) {
7083                 new_end = OBD_OBJECT_EOF;
7084         } else {
7085                 /* Add at least extension_size to the previous component_end,
7086                  * covering the req layout extent */
7087                 new_end = max(extent_end - component_end, extension_size);
7088                 new_end = roundup(new_end, extension_size);
7089                 new_end += component_end;
7090
7091                 /* Component end must be min stripe size aligned */
7092                 if (new_end % stripe_size) {
7093                         CDEBUG(D_LAYOUT, "new component end is not aligned "
7094                                "by the stripe size %u: [%llu, %llu) ext size "
7095                                "%llu new end %llu, aligning\n",
7096                                stripe_size, component_end, extent_end,
7097                                extension_size, new_end);
7098                         new_end = roundup(new_end, stripe_size);
7099                 }
7100
7101                 /* Overflow */
7102                 if (new_end < extent_end)
7103                         new_end = OBD_OBJECT_EOF;
7104         }
7105
7106         /* Don't extend past the end of the extension component */
7107         if (new_end > extension_end)
7108                 new_end = extension_end;
7109
7110         return new_end;
7111 }
7112
7113 /**
7114  * Calculate the exact reservation (per-OST extension_size) on the OSTs being
7115  * instantiated. It needs to be calculated in advance and taken into account at
7116  * the instantiation time, because otherwise lod_statfs_and_check() may consider
7117  * an OST as OK, but SEL needs its extension_size to fit the free space and the
7118  * OST may turn out to be low-on-space, thus inappropriate OST may be used and
7119  * ENOSPC occurs.
7120  *
7121  * \param[in] lod_comp          lod component we are checking
7122  *
7123  * \retval      size to reserved on each OST of lod_comp's stripe.
7124  */
7125 static __u64 lod_sel_stripe_reserved(struct lod_layout_component *lod_comp)
7126 {
7127         /* extension_size is file level, so we must divide by stripe count to
7128          * compare it to available space on a single OST */
7129         return  lod_comp->llc_stripe_size * SEL_UNIT_SIZE /
7130                 lod_comp->llc_stripe_count;
7131 }
7132
7133 /* As lod_sel_handler() could be re-entered for the same component several
7134  * times, this is the data for the next call. Fields could be changed to
7135  * component indexes when needed, (e.g. if there is no need to instantiate
7136  * all the previous components up to the current position) to tell the caller
7137  * where to start over from. */
7138 struct sel_data {
7139         int sd_force;
7140         int sd_repeat;
7141 };
7142
7143 /**
7144  * Process extent updates for a particular layout component
7145  *
7146  * Handle layout updates for a particular extension space component touched by
7147  * a layout update operation.  Core function of self-extending PFL feature.
7148  *
7149  * In general, this function processes exactly *one* stage of an extension
7150  * operation, modifying the layout accordingly, then returns to the caller.
7151  * The caller is responsible for restarting processing with the new layout,
7152  * which may repeatedly return to this function until the extension updates
7153  * are complete.
7154  *
7155  * This function does one of a few things to the layout:
7156  * 1. Extends the component before the current extension space component to
7157  * allow it to accomodate the requested operation (if space/policy permit that
7158  * component to continue on its current OSTs)
7159  *
7160  * 2. If extension of the existing component fails, we do one of two things:
7161  *    a. If there is a component after the extension space, we remove the
7162  *       extension space component, move the start of the next component down
7163  *       accordingly, then notify the caller to restart processing w/the new
7164  *       layout.
7165  *    b. If there is no following component, we try repeating the current
7166  *       component, creating a new component using the current one as a
7167  *       template (keeping its stripe properties but not specific striping),
7168  *       and try assigning striping for this component.  If there is sufficient
7169  *       free space on the OSTs chosen for this component, it is instantiated
7170  *       and i/o continues there.
7171  *
7172  *       If there is not sufficient space on the new OSTs, we remove this new
7173  *       component & extend the current component.
7174  *
7175  * Note further that uninited components followed by extension space can be zero
7176  * length meaning that we will try to extend them before initializing them, and
7177  * if that fails, they will be removed without initialization.
7178  *
7179  * 3. If we extend to/beyond the end of an extension space component, that
7180  * component is exhausted (all of its range has been given to real components),
7181  * so we remove it and restart processing.
7182  *
7183  * \param[in] env               execution environment for this thread
7184  * \param[in,out] lo            object to update the layout of
7185  * \param[in] extent            layout extent for requested operation, update
7186  *                              layout to fit this operation
7187  * \param[in] th                transaction handle for this operation
7188  * \param[in,out] max_comp      the highest comp for the portion of the layout
7189  *                              we are operating on (For FLR, the chosen
7190  *                              replica).  Updated because we may remove
7191  *                              components.
7192  * \param[in] index             index of the extension space component we're
7193  *                              working on
7194  * \param[in] write             if this is write op
7195  * \param[in,out] force         if the extension is to be forced; set here
7196                                 to force it on the 2nd call for the same
7197                                 extension component
7198  *
7199  * \retval      0 on success
7200  * \retval      negative errno on error
7201  */
7202 static int lod_sel_handler(const struct lu_env *env,
7203                           struct lod_object *lo,
7204                           struct lu_extent *extent,
7205                           struct thandle *th, int *max_comp,
7206                           int index, int write,
7207                           struct sel_data *sd)
7208 {
7209         struct lod_device *d = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
7210         struct lod_thread_info *info = lod_env_info(env);
7211         struct lod_layout_component *lod_comp;
7212         struct lod_layout_component *prev;
7213         struct lod_layout_component *next = NULL;
7214         __u64 extension_size, reserve;
7215         __u64 new_end = 0;
7216         bool repeated;
7217         int change = 0;
7218         int rc = 0;
7219         ENTRY;
7220
7221         /* First component cannot be extension space */
7222         if (index == 0) {
7223                 CERROR("%s: "DFID" first component cannot be extension space\n",
7224                        lod2obd(d)->obd_name, PFID(lod_object_fid(lo)));
7225                 RETURN(-EINVAL);
7226         }
7227
7228         lod_comp = &lo->ldo_comp_entries[index];
7229         prev = &lo->ldo_comp_entries[index - 1];
7230         if ((index + 1) < *max_comp)
7231                 next = &lo->ldo_comp_entries[index + 1];
7232
7233         /* extension size uses the stripe size field as KiB */
7234         extension_size = lod_comp->llc_stripe_size * SEL_UNIT_SIZE;
7235
7236         CDEBUG(D_LAYOUT, "prev start %llu, extension start %llu, extension end"
7237                " %llu, extension size %llu\n", prev->llc_extent.e_start,
7238                lod_comp->llc_extent.e_start, lod_comp->llc_extent.e_end,
7239                extension_size);
7240
7241         /* Two extension space components cannot be adjacent & extension space
7242          * components cannot be init */
7243         if ((prev->llc_flags & LCME_FL_EXTENSION) ||
7244             !(ergo(next, !(next->llc_flags & LCME_FL_EXTENSION))) ||
7245              lod_comp_inited(lod_comp)) {
7246                 CERROR("%s: "DFID" invalid extension space components\n",
7247                        lod2obd(d)->obd_name, PFID(lod_object_fid(lo)));
7248                 RETURN(-EINVAL);
7249         }
7250
7251         reserve = lod_sel_stripe_reserved(lod_comp);
7252
7253         if (!prev->llc_stripe) {
7254                 CDEBUG(D_LAYOUT, "Previous component not inited\n");
7255                 info->lti_count = 1;
7256                 info->lti_comp_idx[0] = index - 1;
7257                 rc = lod_declare_instantiate_components(env, lo, th, reserve);
7258                 /* ENOSPC tells us we can't use this component.  If there is
7259                  * a next or we are repeating, we either spill over (next) or
7260                  * extend the original comp (repeat).  Otherwise, return the
7261                  * error to the user. */
7262                 if (rc == -ENOSPC && (next || sd->sd_repeat))
7263                         rc = 1;
7264                 if (rc < 0)
7265                         RETURN(rc);
7266         }
7267
7268         if (sd->sd_force == 0 && rc == 0)
7269                 rc = !lod_sel_osts_allowed(env, lo, index - 1, reserve, extent,
7270                                            &lod_comp->llc_extent, write);
7271
7272         repeated = !!(sd->sd_repeat);
7273         sd->sd_repeat = 0;
7274         sd->sd_force = 0;
7275
7276         /* Extend previous component */
7277         if (rc == 0) {
7278                 new_end = lod_extension_new_end(extension_size, extent->e_end,
7279                                                 prev->llc_stripe_size,
7280                                                 prev->llc_extent.e_end,
7281                                                 lod_comp->llc_extent.e_end);
7282
7283                 CDEBUG(D_LAYOUT, "new end %llu\n", new_end);
7284                 lod_comp->llc_extent.e_start = new_end;
7285                 prev->llc_extent.e_end = new_end;
7286
7287                 if (prev->llc_extent.e_end == lod_comp->llc_extent.e_end) {
7288                         CDEBUG(D_LAYOUT, "Extension component exhausted\n");
7289                         lod_comp->llc_id = LCME_ID_INVAL;
7290                         change--;
7291                 }
7292         } else {
7293                 /* rc == 1, failed to extend current component */
7294                 LASSERT(rc == 1);
7295                 if (next) {
7296                         /* Normal 'spillover' case - Remove the extension
7297                          * space component & bring down the start of the next
7298                          * component. */
7299                         lod_comp->llc_id = LCME_ID_INVAL;
7300                         change--;
7301                         if (!(prev->llc_flags & LCME_FL_INIT)) {
7302                                 prev->llc_id = LCME_ID_INVAL;
7303                                 change--;
7304                         }
7305                         lod_sel_adjust_extents(env, lo, *max_comp, index);
7306                 } else if (lod_comp_inited(prev)) {
7307                         /* If there is no next, and the previous component is
7308                          * INIT'ed, try repeating the previous component. */
7309                         LASSERT(repeated == 0);
7310                         rc = lod_layout_repeat_comp(env, lo, index - 1);
7311                         if (rc < 0)
7312                                 RETURN(rc);
7313                         change++;
7314                         /* The previous component is a repeated component.
7315                          * Record this so we don't keep trying to repeat it. */
7316                         sd->sd_repeat = 1;
7317                 } else {
7318                         /* If the previous component is not INIT'ed, this may
7319                          * be a component we have just instantiated but failed
7320                          * to extend. Or even a repeated component we failed
7321                          * to prepare a striping for. Do not repeat but instead
7322                          * remove the repeated component & force the extention
7323                          * of the original one */
7324                         sd->sd_force = 1;
7325                         if (repeated) {
7326                                 prev->llc_id = LCME_ID_INVAL;
7327                                 change--;
7328                         }
7329                 }
7330         }
7331
7332         if (change < 0) {
7333                 rc = lod_layout_del_prep_layout(env, lo, NULL);
7334                 if (rc < 0)
7335                         RETURN(rc);
7336                 LASSERTF(-rc == change,
7337                          "number deleted %d != requested %d\n", -rc,
7338                          change);
7339         }
7340         *max_comp = *max_comp + change;
7341
7342         /* lod_del_prep_layout reallocates ldo_comp_entries, so we must
7343          * refresh these pointers before using them */
7344         lod_comp = &lo->ldo_comp_entries[index];
7345         prev = &lo->ldo_comp_entries[index - 1];
7346         CDEBUG(D_LAYOUT, "After extent updates: prev start %llu, current start "
7347                "%llu, current end %llu max_comp %d ldo_comp_cnt %d\n",
7348                prev->llc_extent.e_start, lod_comp->llc_extent.e_start,
7349                lod_comp->llc_extent.e_end, *max_comp, lo->ldo_comp_cnt);
7350
7351         /* Layout changed successfully */
7352         RETURN(0);
7353 }
7354
7355 /**
7356  * Declare layout extent updates
7357  *
7358  * Handles extensions.  Identifies extension components touched by current
7359  * operation and passes them to processing function.
7360  *
7361  * Restarts with updated layouts from the processing function until the current
7362  * operation no longer touches an extension space component.
7363  *
7364  * \param[in] env       execution environment for this thread
7365  * \param[in,out] lo    object to update the layout of
7366  * \param[in] extent    layout extent for requested operation, update layout to
7367  *                      fit this operation
7368  * \param[in] th        transaction handle for this operation
7369  * \param[in] pick      identifies chosen mirror for FLR layouts
7370  * \param[in] write     if this is write op
7371  *
7372  * \retval      1 on layout changed, 0 on no change
7373  * \retval      negative errno on error
7374  */
7375 static int lod_declare_update_extents(const struct lu_env *env,
7376                 struct lod_object *lo, struct lu_extent *extent,
7377                 struct thandle *th, int pick, int write)
7378 {
7379         struct lod_thread_info *info = lod_env_info(env);
7380         struct lod_layout_component *lod_comp;
7381         bool layout_changed = false;
7382         struct sel_data sd = { 0 };
7383         int start_index;
7384         int i = 0;
7385         int max_comp = 0;
7386         int rc = 0, rc2;
7387         int change = 0;
7388         ENTRY;
7389
7390         /* This makes us work on the components of the chosen mirror */
7391         start_index = lo->ldo_mirrors[pick].lme_start;
7392         max_comp = lo->ldo_mirrors[pick].lme_end + 1;
7393         if (lo->ldo_flr_state == LCM_FL_NONE)
7394                 LASSERT(start_index == 0 && max_comp == lo->ldo_comp_cnt);
7395
7396         CDEBUG(D_LAYOUT, "extent->e_start %llu, extent->e_end %llu\n",
7397                extent->e_start, extent->e_end);
7398         for (i = start_index; i < max_comp; i++) {
7399                 lod_comp = &lo->ldo_comp_entries[i];
7400
7401                 /* We've passed all components of interest */
7402                 if (lod_comp->llc_extent.e_start >= extent->e_end)
7403                         break;
7404
7405                 if (lod_comp->llc_flags & LCME_FL_EXTENSION) {
7406                         layout_changed = true;
7407                         rc = lod_sel_handler(env, lo, extent, th, &max_comp,
7408                                              i, write, &sd);
7409                         if (rc < 0)
7410                                 GOTO(out, rc);
7411
7412                         /* Nothing has changed behind the prev one */
7413                         i -= 2;
7414                         continue;
7415                 }
7416         }
7417
7418         /* We may have added or removed components.  If so, we must update the
7419          * start & ends of all the mirrors after the current one, and the end
7420          * of the current mirror. */
7421         change = max_comp - 1 - lo->ldo_mirrors[pick].lme_end;
7422         if (change) {
7423                 lo->ldo_mirrors[pick].lme_end += change;
7424                 for (i = pick + 1; i < lo->ldo_mirror_count; i++) {
7425                         lo->ldo_mirrors[i].lme_start += change;
7426                         lo->ldo_mirrors[i].lme_end += change;
7427                 }
7428         }
7429
7430         EXIT;
7431 out:
7432         /* The amount of components has changed, adjust the lti_comp_idx */
7433         rc2 = lod_layout_data_init(info, lo->ldo_comp_cnt);
7434
7435         return rc < 0 ? rc : rc2 < 0 ? rc2 : layout_changed;
7436 }
7437
7438 /* If striping is already instantiated or INIT'ed DOM? */
7439 static bool lod_is_instantiation_needed(struct lod_layout_component *comp)
7440 {
7441         return !(((lov_pattern(comp->llc_pattern) & LOV_PATTERN_MDT) &&
7442                   lod_comp_inited(comp)) || comp->llc_stripe);
7443 }
7444
7445 /**
7446  * Declare layout update for a non-FLR layout.
7447  *
7448  * \param[in] env       execution environment for this thread
7449  * \param[in,out] lo    object to update the layout of
7450  * \param[in] layout    layout intent for requested operation, "update" is
7451  *                      a process of reacting to this
7452  * \param[in] buf       buffer containing lov ea (see comment on usage inline)
7453  * \param[in] th        transaction handle for this operation
7454  *
7455  * \retval      0 on success
7456  * \retval      negative errno on error
7457  */
7458 static int lod_declare_update_plain(const struct lu_env *env,
7459                 struct lod_object *lo, struct layout_intent *layout,
7460                 const struct lu_buf *buf, struct thandle *th)
7461 {
7462         struct lod_thread_info *info = lod_env_info(env);
7463         struct lod_device *d = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
7464         struct lod_layout_component *lod_comp;
7465         struct lov_comp_md_v1 *comp_v1 = NULL;
7466         bool layout_changed = false;
7467         bool replay = false;
7468         int i, rc;
7469         ENTRY;
7470
7471         LASSERT(lo->ldo_flr_state == LCM_FL_NONE);
7472
7473         /*
7474          * In case the client is passing lovea, which only happens during
7475          * the replay of layout intent write RPC for now, we may need to
7476          * parse the lovea and apply new layout configuration.
7477          */
7478         if (buf && buf->lb_len)  {
7479                 struct lov_user_md_v1 *v1 = buf->lb_buf;
7480
7481                 if (v1->lmm_magic != (LOV_MAGIC_DEFINED | LOV_MAGIC_COMP_V1) &&
7482                     v1->lmm_magic != __swab32(LOV_MAGIC_DEFINED |
7483                                               LOV_MAGIC_COMP_V1)) {
7484                         CERROR("%s: the replay buffer of layout extend "
7485                                "(magic %#x) does not contain expected "
7486                                "composite layout.\n",
7487                                lod2obd(d)->obd_name, v1->lmm_magic);
7488                         GOTO(out, rc = -EINVAL);
7489                 }
7490
7491                 rc = lod_use_defined_striping(env, lo, buf);
7492                 if (rc)
7493                         GOTO(out, rc);
7494                 lo->ldo_comp_cached = 1;
7495
7496                 rc = lod_get_lov_ea(env, lo);
7497                 if (rc <= 0)
7498                         GOTO(out, rc);
7499                 /* old on-disk EA is stored in info->lti_buf */
7500                 comp_v1 = (struct lov_comp_md_v1 *)info->lti_buf.lb_buf;
7501                 replay = true;
7502                 layout_changed = true;
7503
7504                 rc = lod_layout_data_init(info, lo->ldo_comp_cnt);
7505                 if (rc)
7506                         GOTO(out, rc);
7507         } else {
7508                 /* non replay path */
7509                 rc = lod_striping_load(env, lo);
7510                 if (rc)
7511                         GOTO(out, rc);
7512         }
7513
7514         /* Make sure defined layout covers the requested write range. */
7515         lod_comp = &lo->ldo_comp_entries[lo->ldo_comp_cnt - 1];
7516         if (lo->ldo_comp_cnt > 1 &&
7517             lod_comp->llc_extent.e_end != OBD_OBJECT_EOF &&
7518             lod_comp->llc_extent.e_end < layout->li_extent.e_end) {
7519                 CDEBUG_LIMIT(replay ? D_ERROR : D_LAYOUT,
7520                              "%s: the defined layout [0, %#llx) does not "
7521                              "covers the write range "DEXT"\n",
7522                              lod2obd(d)->obd_name, lod_comp->llc_extent.e_end,
7523                              PEXT(&layout->li_extent));
7524                 GOTO(out, rc = -EINVAL);
7525         }
7526
7527         CDEBUG(D_LAYOUT, "%s: "DFID": update components "DEXT"\n",
7528                lod2obd(d)->obd_name, PFID(lod_object_fid(lo)),
7529                PEXT(&layout->li_extent));
7530
7531         if (!replay) {
7532                 rc = lod_declare_update_extents(env, lo, &layout->li_extent,
7533                                 th, 0, layout->li_opc == LAYOUT_INTENT_WRITE);
7534                 if (rc < 0)
7535                         GOTO(out, rc);
7536                 else if (rc)
7537                         layout_changed = true;
7538         }
7539
7540         /*
7541          * Iterate ld->ldo_comp_entries, find the component whose extent under
7542          * the write range and not instantianted.
7543          */
7544         for (i = 0; i < lo->ldo_comp_cnt; i++) {
7545                 lod_comp = &lo->ldo_comp_entries[i];
7546
7547                 if (lod_comp->llc_extent.e_start >= layout->li_extent.e_end)
7548                         break;
7549
7550                 if (!replay) {
7551                         /* If striping is instantiated or INIT'ed DOM skip */
7552                         if (!lod_is_instantiation_needed(lod_comp))
7553                                 continue;
7554                 } else {
7555                         /**
7556                          * In replay path, lod_comp is the EA passed by
7557                          * client replay buffer,  comp_v1 is the pre-recovery
7558                          * on-disk EA, we'd sift out those components which
7559                          * were init-ed in the on-disk EA.
7560                          */
7561                         if (le32_to_cpu(comp_v1->lcm_entries[i].lcme_flags) &
7562                             LCME_FL_INIT)
7563                                 continue;
7564                 }
7565                 /*
7566                  * this component hasn't instantiated in normal path, or during
7567                  * replay it needs replay the instantiation.
7568                  */
7569
7570                 /* A released component is being extended */
7571                 if (lod_comp->llc_pattern & LOV_PATTERN_F_RELEASED)
7572                         GOTO(out, rc = -EINVAL);
7573
7574                 LASSERT(info->lti_comp_idx != NULL);
7575                 info->lti_comp_idx[info->lti_count++] = i;
7576                 layout_changed = true;
7577         }
7578
7579         if (!layout_changed)
7580                 RETURN(-EALREADY);
7581
7582         lod_obj_inc_layout_gen(lo);
7583         rc = lod_declare_instantiate_components(env, lo, th, 0);
7584         EXIT;
7585 out:
7586         if (rc)
7587                 lod_striping_free(env, lo);
7588         return rc;
7589 }
7590
7591 static inline int lod_comp_index(struct lod_object *lo,
7592                                  struct lod_layout_component *lod_comp)
7593 {
7594         LASSERT(lod_comp >= lo->ldo_comp_entries &&
7595                 lod_comp <= &lo->ldo_comp_entries[lo->ldo_comp_cnt - 1]);
7596
7597         return lod_comp - lo->ldo_comp_entries;
7598 }
7599
7600 /**
7601  * Stale other mirrors by writing extent.
7602  */
7603 static int lod_stale_components(const struct lu_env *env, struct lod_object *lo,
7604                                 int primary, struct lu_extent *extent,
7605                                 struct thandle *th)
7606 {
7607         struct lod_layout_component *pri_comp, *lod_comp;
7608         struct lu_extent pri_extent;
7609         int rc = 0;
7610         int i;
7611         ENTRY;
7612
7613         /* The writing extent decides which components in the primary
7614          * are affected... */
7615         CDEBUG(D_LAYOUT, "primary mirror %d, "DEXT"\n", primary, PEXT(extent));
7616
7617 restart:
7618         lod_foreach_mirror_comp(pri_comp, lo, primary) {
7619                 if (!lu_extent_is_overlapped(extent, &pri_comp->llc_extent))
7620                         continue;
7621
7622                 CDEBUG(D_LAYOUT, "primary comp %u "DEXT"\n",
7623                        lod_comp_index(lo, pri_comp),
7624                        PEXT(&pri_comp->llc_extent));
7625
7626                 pri_extent.e_start = pri_comp->llc_extent.e_start;
7627                 pri_extent.e_end = pri_comp->llc_extent.e_end;
7628
7629                 for (i = 0; i < lo->ldo_mirror_count; i++) {
7630                         if (i == primary)
7631                                 continue;
7632                         rc = lod_declare_update_extents(env, lo, &pri_extent,
7633                                                         th, i, 0);
7634                         /* if update_extents changed the layout, it may have
7635                          * reallocated the component array, so start over to
7636                          * avoid using stale pointers */
7637                         if (rc == 1)
7638                                 goto restart;
7639                         if (rc < 0)
7640                                 RETURN(rc);
7641
7642                         /* ... and then stale other components that are
7643                          * overlapping with primary components */
7644                         lod_foreach_mirror_comp(lod_comp, lo, i) {
7645                                 if (!lu_extent_is_overlapped(
7646                                                         &pri_extent,
7647                                                         &lod_comp->llc_extent))
7648                                         continue;
7649
7650                                 CDEBUG(D_LAYOUT, "stale: %u / %u\n",
7651                                       i, lod_comp_index(lo, lod_comp));
7652
7653                                 lod_comp->llc_flags |= LCME_FL_STALE;
7654                                 lo->ldo_mirrors[i].lme_stale = 1;
7655                         }
7656                 }
7657         }
7658
7659         RETURN(rc);
7660 }
7661
7662 /**
7663  * check an OST's availability
7664  * \param[in] env       execution environment
7665  * \param[in] lo        lod object
7666  * \param[in] dt        dt object
7667  * \param[in] index     mirror index
7668  *
7669  * \retval      negative if failed
7670  * \retval      1 if \a dt is available
7671  * \retval      0 if \a dt is not available
7672  */
7673 static inline int lod_check_ost_avail(const struct lu_env *env,
7674                                       struct lod_object *lo,
7675                                       struct dt_object *dt, int index)
7676 {
7677         struct lod_device *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
7678         struct lod_tgt_desc *ost;
7679         __u32 idx;
7680         int type = LU_SEQ_RANGE_OST;
7681         int rc;
7682
7683         rc = lod_fld_lookup(env, lod, lu_object_fid(&dt->do_lu), &idx, &type);
7684         if (rc < 0) {
7685                 CERROR("%s: can't locate "DFID":rc = %d\n",
7686                        lod2obd(lod)->obd_name, PFID(lu_object_fid(&dt->do_lu)),
7687                        rc);
7688                 return rc;
7689         }
7690
7691         ost = OST_TGT(lod, idx);
7692         if (ost->ltd_active == 0) {
7693                 CDEBUG(D_LAYOUT, DFID ": mirror %d OST%d unavail\n",
7694                        PFID(lod_object_fid(lo)), index, idx);
7695                 return 0;
7696         }
7697
7698         return 1;
7699 }
7700
7701 /**
7702  * Pick primary mirror for write
7703  * \param[in] env       execution environment
7704  * \param[in] lo        object
7705  * \param[in] extent    write range
7706  */
7707 static int lod_primary_pick(const struct lu_env *env, struct lod_object *lo,
7708                             struct lu_extent *extent)
7709 {
7710         struct lod_device *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
7711         unsigned int seq = 0;
7712         struct lod_layout_component *lod_comp;
7713         int i, j, rc;
7714         int picked = -1, second_pick = -1, third_pick = -1;
7715         ENTRY;
7716
7717         if (CFS_FAIL_CHECK(OBD_FAIL_FLR_RANDOM_PICK_MIRROR)) {
7718                 get_random_bytes(&seq, sizeof(seq));
7719                 seq %= lo->ldo_mirror_count;
7720         }
7721
7722         /**
7723          * Pick a mirror as the primary, and check the availability of OSTs.
7724          *
7725          * This algo can be revised later after knowing the topology of
7726          * cluster.
7727          */
7728         lod_qos_statfs_update(env, lod, &lod->lod_ost_descs);
7729
7730         rc = lod_fill_mirrors(lo);
7731         if (rc)
7732                 RETURN(rc);
7733
7734         for (i = 0; i < lo->ldo_mirror_count; i++) {
7735                 bool ost_avail = true;
7736                 int index = (i + seq) % lo->ldo_mirror_count;
7737
7738                 if (lo->ldo_mirrors[index].lme_stale) {
7739                         CDEBUG(D_LAYOUT, DFID": mirror %d stale\n",
7740                                PFID(lod_object_fid(lo)), index);
7741                         continue;
7742                 }
7743
7744                 /* 2nd pick is for the primary mirror containing unavail OST */
7745                 if (lo->ldo_mirrors[index].lme_prefer && second_pick < 0)
7746                         second_pick = index;
7747
7748                 /* 3rd pick is for non-primary mirror containing unavail OST */
7749                 if (second_pick < 0 && third_pick < 0)
7750                         third_pick = index;
7751
7752                 /**
7753                  * we found a non-primary 1st pick, we'd like to find a
7754                  * potential pirmary mirror.
7755                  */
7756                 if (picked >= 0 && !lo->ldo_mirrors[index].lme_prefer)
7757                         continue;
7758
7759                 /* check the availability of OSTs */
7760                 lod_foreach_mirror_comp(lod_comp, lo, index) {
7761                         if (!lod_comp_inited(lod_comp) || !lod_comp->llc_stripe)
7762                                 continue;
7763
7764                         for (j = 0; j < lod_comp->llc_stripe_count; j++) {
7765                                 struct dt_object *dt = lod_comp->llc_stripe[j];
7766
7767                                 rc = lod_check_ost_avail(env, lo, dt, index);
7768                                 if (rc < 0)
7769                                         RETURN(rc);
7770
7771                                 ost_avail = !!rc;
7772                                 if (!ost_avail)
7773                                         break;
7774                         } /* for all dt object in one component */
7775                         if (!ost_avail)
7776                                 break;
7777                 } /* for all components in a mirror */
7778
7779                 /**
7780                  * the OSTs where allocated objects locates in the components
7781                  * of the mirror are available.
7782                  */
7783                 if (!ost_avail)
7784                         continue;
7785
7786                 /* this mirror has all OSTs available */
7787                 picked = index;
7788
7789                 /**
7790                  * primary with all OSTs are available, this is the perfect
7791                  * 1st pick.
7792                  */
7793                 if (lo->ldo_mirrors[index].lme_prefer)
7794                         break;
7795         } /* for all mirrors */
7796
7797         /* failed to pick a sound mirror, lower our expectation */
7798         if (picked < 0)
7799                 picked = second_pick;
7800         if (picked < 0)
7801                 picked = third_pick;
7802         if (picked < 0)
7803                 RETURN(-ENODATA);
7804
7805         RETURN(picked);
7806 }
7807
7808 static int lod_prepare_resync_mirror(const struct lu_env *env,
7809                                      struct lod_object *lo,
7810                                      __u16 mirror_id)
7811 {
7812         struct lod_thread_info *info = lod_env_info(env);
7813         struct lod_layout_component *lod_comp;
7814         bool neg = !!(MIRROR_ID_NEG & mirror_id);
7815         int i;
7816
7817         mirror_id &= ~MIRROR_ID_NEG;
7818
7819         for (i = 0; i < lo->ldo_mirror_count; i++) {
7820                 if ((!neg && lo->ldo_mirrors[i].lme_id != mirror_id) ||
7821                     (neg && lo->ldo_mirrors[i].lme_id == mirror_id))
7822                         continue;
7823
7824                 lod_foreach_mirror_comp(lod_comp, lo, i) {
7825                         if (lod_comp_inited(lod_comp))
7826                                 continue;
7827
7828                         info->lti_comp_idx[info->lti_count++] =
7829                                 lod_comp_index(lo, lod_comp);
7830                 }
7831         }
7832
7833         return 0;
7834 }
7835
7836 /**
7837  * figure out the components should be instantiated for resync.
7838  */
7839 static int lod_prepare_resync(const struct lu_env *env, struct lod_object *lo,
7840                               struct lu_extent *extent)
7841 {
7842         struct lod_thread_info *info = lod_env_info(env);
7843         struct lod_layout_component *lod_comp;
7844         unsigned int need_sync = 0;
7845         int i;
7846
7847         CDEBUG(D_LAYOUT,
7848                DFID": instantiate all stale components in "DEXT"\n",
7849                PFID(lod_object_fid(lo)), PEXT(extent));
7850
7851         /**
7852          * instantiate all components within this extent, even non-stale
7853          * components.
7854          */
7855         for (i = 0; i < lo->ldo_mirror_count; i++) {
7856                 if (!lo->ldo_mirrors[i].lme_stale)
7857                         continue;
7858
7859                 lod_foreach_mirror_comp(lod_comp, lo, i) {
7860                         if (!lu_extent_is_overlapped(extent,
7861                                                 &lod_comp->llc_extent))
7862                                 break;
7863
7864                         need_sync++;
7865
7866                         if (lod_comp_inited(lod_comp))
7867                                 continue;
7868
7869                         CDEBUG(D_LAYOUT, "resync instantiate %d / %d\n",
7870                                i, lod_comp_index(lo, lod_comp));
7871                         info->lti_comp_idx[info->lti_count++] =
7872                                         lod_comp_index(lo, lod_comp);
7873                 }
7874         }
7875
7876         return need_sync ? 0 : -EALREADY;
7877 }
7878
7879 static int lod_declare_update_rdonly(const struct lu_env *env,
7880                 struct lod_object *lo, struct md_layout_change *mlc,
7881                 struct thandle *th)
7882 {
7883         struct lod_thread_info *info = lod_env_info(env);
7884         struct lu_attr *layout_attr = &info->lti_layout_attr;
7885         struct lod_layout_component *lod_comp;
7886         struct lu_extent extent = { 0 };
7887         int rc;
7888         ENTRY;
7889
7890         LASSERT(lo->ldo_flr_state == LCM_FL_RDONLY);
7891         LASSERT(mlc->mlc_opc == MD_LAYOUT_WRITE ||
7892                 mlc->mlc_opc == MD_LAYOUT_RESYNC);
7893         LASSERT(lo->ldo_mirror_count > 0);
7894
7895         if (mlc->mlc_opc == MD_LAYOUT_WRITE) {
7896                 struct layout_intent *layout = mlc->mlc_intent;
7897                 int write = layout->li_opc == LAYOUT_INTENT_WRITE;
7898                 int picked;
7899
7900                 extent = layout->li_extent;
7901                 CDEBUG(D_LAYOUT, DFID": trying to write :"DEXT"\n",
7902                        PFID(lod_object_fid(lo)), PEXT(&extent));
7903
7904                 picked = lod_primary_pick(env, lo, &extent);
7905                 if (picked < 0)
7906                         RETURN(picked);
7907
7908                 CDEBUG(D_LAYOUT, DFID": picked mirror id %u as primary\n",
7909                        PFID(lod_object_fid(lo)),
7910                        lo->ldo_mirrors[picked].lme_id);
7911
7912                 /* Update extents of primary before staling */
7913                 rc = lod_declare_update_extents(env, lo, &extent, th, picked,
7914                                                 write);
7915                 if (rc < 0)
7916                         GOTO(out, rc);
7917
7918                 if (layout->li_opc == LAYOUT_INTENT_TRUNC) {
7919                         /**
7920                          * trunc transfers [0, size) in the intent extent, we'd
7921                          * stale components overlapping [size, eof).
7922                          */
7923                         extent.e_start = extent.e_end;
7924                         extent.e_end = OBD_OBJECT_EOF;
7925                 }
7926
7927                 /* stale overlapping components from other mirrors */
7928                 rc = lod_stale_components(env, lo, picked, &extent, th);
7929                 if (rc < 0)
7930                         GOTO(out, rc);
7931
7932                 /* restore truncate intent extent */
7933                 if (layout->li_opc == LAYOUT_INTENT_TRUNC)
7934                         extent.e_end = extent.e_start;
7935
7936                 /* instantiate components for the picked mirror, start from 0 */
7937                 extent.e_start = 0;
7938
7939                 lod_foreach_mirror_comp(lod_comp, lo, picked) {
7940                         if (!lu_extent_is_overlapped(&extent,
7941                                                      &lod_comp->llc_extent))
7942                                 break;
7943
7944                         if (!lod_is_instantiation_needed(lod_comp))
7945                                 continue;
7946
7947                         info->lti_comp_idx[info->lti_count++] =
7948                                                 lod_comp_index(lo, lod_comp);
7949                 }
7950
7951                 lo->ldo_flr_state = LCM_FL_WRITE_PENDING;
7952         } else { /* MD_LAYOUT_RESYNC */
7953                 int i;
7954
7955                 /**
7956                  * could contain multiple non-stale mirrors, so we need to
7957                  * prep uninited all components assuming any non-stale mirror
7958                  * could be picked as the primary mirror.
7959                  */
7960                 if (mlc->mlc_mirror_id == 0) {
7961                         /* normal resync */
7962                         for (i = 0; i < lo->ldo_mirror_count; i++) {
7963                                 if (lo->ldo_mirrors[i].lme_stale)
7964                                         continue;
7965
7966                                 lod_foreach_mirror_comp(lod_comp, lo, i) {
7967                                         if (!lod_comp_inited(lod_comp))
7968                                                 break;
7969
7970                                         if (extent.e_end <
7971                                                 lod_comp->llc_extent.e_end)
7972                                                 extent.e_end =
7973                                                      lod_comp->llc_extent.e_end;
7974                                 }
7975                         }
7976                         rc = lod_prepare_resync(env, lo, &extent);
7977                         if (rc)
7978                                 GOTO(out, rc);
7979                 } else {
7980                         /* mirror write, try to init its all components */
7981                         rc = lod_prepare_resync_mirror(env, lo,
7982                                                        mlc->mlc_mirror_id);
7983                         if (rc)
7984                                 GOTO(out, rc);
7985                 }
7986
7987                 /* change the file state to SYNC_PENDING */
7988                 lo->ldo_flr_state = LCM_FL_SYNC_PENDING;
7989         }
7990
7991         /* Reset the layout version once it's becoming too large.
7992          * This way it can make sure that the layout version is
7993          * monotonously increased in this writing era. */
7994         lod_obj_inc_layout_gen(lo);
7995
7996         rc = lod_declare_instantiate_components(env, lo, th, 0);
7997         if (rc)
7998                 GOTO(out, rc);
7999
8000         layout_attr->la_valid = LA_LAYOUT_VERSION;
8001         layout_attr->la_layout_version = 0;
8002         if (mlc->mlc_opc == MD_LAYOUT_RESYNC)
8003                 layout_attr->la_layout_version = LU_LAYOUT_RESYNC;
8004         rc = lod_declare_attr_set(env, &lo->ldo_obj, layout_attr, th);
8005         if (rc)
8006                 GOTO(out, rc);
8007
8008 out:
8009         if (rc)
8010                 lod_striping_free(env, lo);
8011         RETURN(rc);
8012 }
8013
8014 static int lod_declare_update_write_pending(const struct lu_env *env,
8015                 struct lod_object *lo, struct md_layout_change *mlc,
8016                 struct thandle *th)
8017 {
8018         struct lod_thread_info *info = lod_env_info(env);
8019         struct lu_attr *layout_attr = &info->lti_layout_attr;
8020         struct lod_layout_component *lod_comp;
8021         struct lu_extent extent = { 0 };
8022         int primary = -1;
8023         int i;
8024         int rc;
8025         ENTRY;
8026
8027         LASSERT(lo->ldo_flr_state == LCM_FL_WRITE_PENDING);
8028         LASSERT(mlc->mlc_opc == MD_LAYOUT_WRITE ||
8029                 mlc->mlc_opc == MD_LAYOUT_RESYNC);
8030
8031         /* look for the first preferred mirror */
8032         for (i = 0; i < lo->ldo_mirror_count; i++) {
8033                 if (lo->ldo_mirrors[i].lme_stale)
8034                         continue;
8035                 if (lo->ldo_mirrors[i].lme_prefer == 0)
8036                         continue;
8037
8038                 primary = i;
8039                 break;
8040         }
8041         if (primary < 0) {
8042                 /* no primary, use any in-sync */
8043                 for (i = 0; i < lo->ldo_mirror_count; i++) {
8044                         if (lo->ldo_mirrors[i].lme_stale)
8045                                 continue;
8046                         primary = i;
8047                         break;
8048                 }
8049                 if (primary < 0) {
8050                         CERROR(DFID ": doesn't have a primary mirror\n",
8051                                PFID(lod_object_fid(lo)));
8052                         GOTO(out, rc = -ENODATA);
8053                 }
8054         }
8055
8056         CDEBUG(D_LAYOUT, DFID": found primary %u\n",
8057                PFID(lod_object_fid(lo)), lo->ldo_mirrors[primary].lme_id);
8058
8059         LASSERT(!lo->ldo_mirrors[primary].lme_stale);
8060
8061         /* for LAYOUT_WRITE opc, it has to do the following operations:
8062          * 1. stale overlapping componets from stale mirrors;
8063          * 2. instantiate components of the primary mirror;
8064          * 3. transfter layout version to all objects of the primary;
8065          *
8066          * for LAYOUT_RESYNC opc, it will do:
8067          * 1. instantiate components of all stale mirrors;
8068          * 2. transfer layout version to all objects to close write era. */
8069
8070         if (mlc->mlc_opc == MD_LAYOUT_WRITE) {
8071                 struct layout_intent *layout = mlc->mlc_intent;
8072                 int write = layout->li_opc == LAYOUT_INTENT_WRITE;
8073
8074                 LASSERT(mlc->mlc_intent != NULL);
8075
8076                 extent = mlc->mlc_intent->li_extent;
8077
8078                 CDEBUG(D_LAYOUT, DFID": intent to write: "DEXT"\n",
8079                        PFID(lod_object_fid(lo)), PEXT(&extent));
8080
8081                 /* 1. Update extents of primary before staling */
8082                 rc = lod_declare_update_extents(env, lo, &extent, th, primary,
8083                                                 write);
8084                 if (rc < 0)
8085                         GOTO(out, rc);
8086
8087                 if (mlc->mlc_intent->li_opc == LAYOUT_INTENT_TRUNC) {
8088                         /**
8089                          * trunc transfers [0, size) in the intent extent, we'd
8090                          * stale components overlapping [size, eof).
8091                          */
8092                         extent.e_start = extent.e_end;
8093                         extent.e_end = OBD_OBJECT_EOF;
8094                 }
8095
8096                 /* 2. stale overlapping components */
8097                 rc = lod_stale_components(env, lo, primary, &extent, th);
8098                 if (rc < 0)
8099                         GOTO(out, rc);
8100
8101                 /* 3. find the components which need instantiating.
8102                  * instantiate [0, mlc->mlc_intent->e_end) */
8103
8104                 /* restore truncate intent extent */
8105                 if (mlc->mlc_intent->li_opc == LAYOUT_INTENT_TRUNC)
8106                         extent.e_end = extent.e_start;
8107                 extent.e_start = 0;
8108
8109                 lod_foreach_mirror_comp(lod_comp, lo, primary) {
8110                         if (!lu_extent_is_overlapped(&extent,
8111                                                      &lod_comp->llc_extent))
8112                                 break;
8113
8114                         if (!lod_is_instantiation_needed(lod_comp))
8115                                 continue;
8116
8117                         CDEBUG(D_LAYOUT, "write instantiate %d / %d\n",
8118                                primary, lod_comp_index(lo, lod_comp));
8119                         info->lti_comp_idx[info->lti_count++] =
8120                                                 lod_comp_index(lo, lod_comp);
8121                 }
8122         } else { /* MD_LAYOUT_RESYNC */
8123                 if (mlc->mlc_mirror_id == 0) {
8124                         /* normal resync */
8125                         lod_foreach_mirror_comp(lod_comp, lo, primary) {
8126                                 if (!lod_comp_inited(lod_comp))
8127                                         break;
8128
8129                                 extent.e_end = lod_comp->llc_extent.e_end;
8130                         }
8131
8132                         rc = lod_prepare_resync(env, lo, &extent);
8133                         if (rc)
8134                                 GOTO(out, rc);
8135                 } else {
8136                         /* mirror write, try to init its all components */
8137                         rc = lod_prepare_resync_mirror(env, lo,
8138                                                        mlc->mlc_mirror_id);
8139                         if (rc)
8140                                 GOTO(out, rc);
8141                 }
8142
8143                 /* change the file state to SYNC_PENDING */
8144                 lo->ldo_flr_state = LCM_FL_SYNC_PENDING;
8145         }
8146
8147         rc = lod_declare_instantiate_components(env, lo, th, 0);
8148         if (rc)
8149                 GOTO(out, rc);
8150
8151         lod_obj_inc_layout_gen(lo);
8152
8153         /* 3. transfer layout version to OST objects.
8154          * transfer new layout version to OST objects so that stale writes
8155          * can be denied. It also ends an era of writing by setting
8156          * LU_LAYOUT_RESYNC. Normal client can never use this bit to
8157          * send write RPC; only resync RPCs could do it. */
8158         layout_attr->la_valid = LA_LAYOUT_VERSION;
8159         layout_attr->la_layout_version = 0;
8160         if (mlc->mlc_opc == MD_LAYOUT_RESYNC)
8161                 layout_attr->la_layout_version = LU_LAYOUT_RESYNC;
8162         rc = lod_declare_attr_set(env, &lo->ldo_obj, layout_attr, th);
8163         if (rc)
8164                 GOTO(out, rc);
8165 out:
8166         if (rc)
8167                 lod_striping_free(env, lo);
8168         RETURN(rc);
8169 }
8170
8171 static int lod_declare_update_sync_pending(const struct lu_env *env,
8172                 struct lod_object *lo, struct md_layout_change *mlc,
8173                 struct thandle *th)
8174 {
8175         struct lod_thread_info  *info = lod_env_info(env);
8176         struct lu_attr *layout_attr = &info->lti_layout_attr;
8177         unsigned sync_components = 0;
8178         unsigned resync_components = 0;
8179         int i;
8180         int rc;
8181         ENTRY;
8182
8183         LASSERT(lo->ldo_flr_state == LCM_FL_SYNC_PENDING);
8184         LASSERT(mlc->mlc_opc == MD_LAYOUT_RESYNC_DONE ||
8185                 mlc->mlc_opc == MD_LAYOUT_WRITE);
8186
8187         CDEBUG(D_LAYOUT, DFID ": received op %d in sync pending\n",
8188                PFID(lod_object_fid(lo)), mlc->mlc_opc);
8189
8190         if (mlc->mlc_opc == MD_LAYOUT_WRITE) {
8191                 CDEBUG(D_LAYOUT, DFID": cocurrent write to sync pending\n",
8192                        PFID(lod_object_fid(lo)));
8193
8194                 lo->ldo_flr_state = LCM_FL_WRITE_PENDING;
8195                 return lod_declare_update_write_pending(env, lo, mlc, th);
8196         }
8197
8198         /* MD_LAYOUT_RESYNC_DONE */
8199
8200         for (i = 0; i < lo->ldo_comp_cnt; i++) {
8201                 struct lod_layout_component *lod_comp;
8202                 int j;
8203
8204                 lod_comp = &lo->ldo_comp_entries[i];
8205
8206                 if (!(lod_comp->llc_flags & LCME_FL_STALE)) {
8207                         sync_components++;
8208                         continue;
8209                 }
8210
8211                 for (j = 0; j < mlc->mlc_resync_count; j++) {
8212                         if (lod_comp->llc_id != mlc->mlc_resync_ids[j])
8213                                 continue;
8214
8215                         mlc->mlc_resync_ids[j] = LCME_ID_INVAL;
8216                         lod_comp->llc_flags &= ~LCME_FL_STALE;
8217                         resync_components++;
8218                         break;
8219                 }
8220         }
8221
8222         /* valid check */
8223         for (i = 0; i < mlc->mlc_resync_count; i++) {
8224                 if (mlc->mlc_resync_ids[i] == LCME_ID_INVAL)
8225                         continue;
8226
8227                 CDEBUG(D_LAYOUT, DFID": lcme id %u (%d / %zd) not exist "
8228                        "or already synced\n", PFID(lod_object_fid(lo)),
8229                        mlc->mlc_resync_ids[i], i, mlc->mlc_resync_count);
8230                 GOTO(out, rc = -EINVAL);
8231         }
8232
8233         if (!sync_components || (mlc->mlc_resync_count && !resync_components)) {
8234                 CDEBUG(D_LAYOUT, DFID": no mirror in sync\n",
8235                        PFID(lod_object_fid(lo)));
8236
8237                 /* tend to return an error code here to prevent
8238                  * the MDT from setting SoM attribute */
8239                 GOTO(out, rc = -EINVAL);
8240         }
8241
8242         CDEBUG(D_LAYOUT, DFID": synced %u resynced %u/%zu components\n",
8243                PFID(lod_object_fid(lo)),
8244                sync_components, resync_components, mlc->mlc_resync_count);
8245
8246         lo->ldo_flr_state = LCM_FL_RDONLY;
8247         lod_obj_inc_layout_gen(lo);
8248
8249         layout_attr->la_valid = LA_LAYOUT_VERSION;
8250         layout_attr->la_layout_version = 0;
8251         rc = lod_declare_attr_set(env, &lo->ldo_obj, layout_attr, th);
8252         if (rc)
8253                 GOTO(out, rc);
8254
8255         info->lti_buf.lb_len = lod_comp_md_size(lo, false);
8256         rc = lod_sub_declare_xattr_set(env, lod_object_child(lo),
8257                                        &info->lti_buf, XATTR_NAME_LOV, 0, th);
8258         EXIT;
8259
8260 out:
8261         if (rc)
8262                 lod_striping_free(env, lo);
8263         RETURN(rc);
8264 }
8265
8266 typedef int (*mlc_handler)(const struct lu_env *env, struct dt_object *dt,
8267                            const struct md_layout_change *mlc,
8268                            struct thandle *th);
8269
8270 /**
8271  * Attach stripes after target's for migrating directory. NB, we
8272  * only need to declare this, the actual work is done inside
8273  * lod_xattr_set_lmv().
8274  *
8275  * \param[in] env       execution environment
8276  * \param[in] dt        target object
8277  * \param[in] mlc       layout change data
8278  * \param[in] th        transaction handle
8279  *
8280  * \retval              0 on success
8281  * \retval              negative if failed
8282  */
8283 static int lod_dir_declare_layout_attach(const struct lu_env *env,
8284                                          struct dt_object *dt,
8285                                          const struct md_layout_change *mlc,
8286                                          struct thandle *th)
8287 {
8288         struct lod_thread_info *info = lod_env_info(env);
8289         struct lod_device *lod = lu2lod_dev(dt->do_lu.lo_dev);
8290         struct lod_tgt_descs *ltd = &lod->lod_mdt_descs;
8291         struct lod_object *lo = lod_dt_obj(dt);
8292         struct dt_object *next = dt_object_child(dt);
8293         struct dt_object_format *dof = &info->lti_format;
8294         struct lmv_mds_md_v1 *lmv = mlc->mlc_buf.lb_buf;
8295         struct dt_object **stripes;
8296         __u32 stripe_count = le32_to_cpu(lmv->lmv_stripe_count);
8297         struct lu_fid *fid = &info->lti_fid;
8298         struct lod_tgt_desc *tgt;
8299         struct dt_object *dto;
8300         struct dt_device *tgt_dt;
8301         int type = LU_SEQ_RANGE_ANY;
8302         struct dt_insert_rec *rec = &info->lti_dt_rec;
8303         char *stripe_name = info->lti_key;
8304         struct lu_name *sname;
8305         struct linkea_data ldata = { NULL };
8306         struct lu_buf linkea_buf;
8307         __u32 idx;
8308         int i;
8309         int rc;
8310
8311         ENTRY;
8312
8313         if (!lmv_is_sane(lmv))
8314                 RETURN(-EINVAL);
8315
8316         if (!dt_try_as_dir(env, dt, false))
8317                 RETURN(-ENOTDIR);
8318
8319         dof->dof_type = DFT_DIR;
8320
8321         OBD_ALLOC_PTR_ARRAY(stripes, (lo->ldo_dir_stripe_count + stripe_count));
8322         if (!stripes)
8323                 RETURN(-ENOMEM);
8324
8325         for (i = 0; i < lo->ldo_dir_stripe_count; i++)
8326                 stripes[i] = lo->ldo_stripe[i];
8327
8328         rec->rec_type = S_IFDIR;
8329
8330         for (i = 0; i < stripe_count; i++) {
8331                 fid_le_to_cpu(fid,
8332                         &lmv->lmv_stripe_fids[i]);
8333                 if (!fid_is_sane(fid))
8334                         continue;
8335
8336                 rc = lod_fld_lookup(env, lod, fid, &idx, &type);
8337                 if (rc)
8338                         GOTO(out, rc);
8339
8340                 if (idx == lod2lu_dev(lod)->ld_site->ld_seq_site->ss_node_id) {
8341                         tgt_dt = lod->lod_child;
8342                 } else {
8343                         tgt = LTD_TGT(ltd, idx);
8344                         if (tgt == NULL)
8345                                 GOTO(out, rc = -ESTALE);
8346                         tgt_dt = tgt->ltd_tgt;
8347                 }
8348
8349                 dto = dt_locate_at(env, tgt_dt, fid,
8350                                   lo->ldo_obj.do_lu.lo_dev->ld_site->ls_top_dev,
8351                                   NULL);
8352                 if (IS_ERR(dto))
8353                         GOTO(out, rc = PTR_ERR(dto));
8354
8355                 stripes[i + lo->ldo_dir_stripe_count] = dto;
8356
8357                 if (!dt_try_as_dir(env, dto, true))
8358                         GOTO(out, rc = -ENOTDIR);
8359
8360                 rc = lod_sub_declare_ref_add(env, dto, th);
8361                 if (rc)
8362                         GOTO(out, rc);
8363
8364                 rec->rec_fid = lu_object_fid(&dto->do_lu);
8365                 rc = lod_sub_declare_insert(env, dto,
8366                                             (const struct dt_rec *)rec,
8367                                             (const struct dt_key *)dot, th);
8368                 if (rc)
8369                         GOTO(out, rc);
8370
8371                 rc = lod_sub_declare_insert(env, dto,
8372                                             (const struct dt_rec *)rec,
8373                                             (const struct dt_key *)dotdot, th);
8374                 if (rc)
8375                         GOTO(out, rc);
8376
8377                 rc = lod_sub_declare_xattr_set(env, dto, &mlc->mlc_buf,
8378                                                 XATTR_NAME_LMV, 0, th);
8379                 if (rc)
8380                         GOTO(out, rc);
8381
8382                 snprintf(stripe_name, sizeof(info->lti_key), DFID":%u",
8383                          PFID(lu_object_fid(&dto->do_lu)),
8384                          i + lo->ldo_dir_stripe_count);
8385
8386                 sname = lod_name_get(env, stripe_name, strlen(stripe_name));
8387                 rc = linkea_links_new(&ldata, &info->lti_linkea_buf,
8388                                       sname, lu_object_fid(&dt->do_lu));
8389                 if (rc)
8390                         GOTO(out, rc);
8391
8392                 linkea_buf.lb_buf = ldata.ld_buf->lb_buf;
8393                 linkea_buf.lb_len = ldata.ld_leh->leh_len;
8394                 rc = lod_sub_declare_xattr_set(env, dto, &linkea_buf,
8395                                                XATTR_NAME_LINK, 0, th);
8396                 if (rc)
8397                         GOTO(out, rc);
8398
8399                 rc = lod_sub_declare_insert(env, next,
8400                                             (const struct dt_rec *)rec,
8401                                             (const struct dt_key *)stripe_name,
8402                                             th);
8403                 if (rc)
8404                         GOTO(out, rc);
8405
8406                 rc = lod_sub_declare_ref_add(env, next, th);
8407                 if (rc)
8408                         GOTO(out, rc);
8409         }
8410
8411         if (lo->ldo_stripe)
8412                 OBD_FREE_PTR_ARRAY(lo->ldo_stripe,
8413                                    lo->ldo_dir_stripes_allocated);
8414         lo->ldo_stripe = stripes;
8415         lo->ldo_is_foreign = 0;
8416         lo->ldo_dir_migrate_offset = lo->ldo_dir_stripe_count;
8417         lo->ldo_dir_migrate_hash = le32_to_cpu(lmv->lmv_hash_type);
8418         lo->ldo_dir_stripe_count += stripe_count;
8419         lo->ldo_dir_stripes_allocated += stripe_count;
8420
8421         /* plain directory split creates target as a plain directory, while
8422          * after source attached as the first stripe, it becomes a striped
8423          * directory, set correct do_index_ops, otherwise it can't be unlinked.
8424          */
8425         dt->do_index_ops = &lod_striped_index_ops;
8426
8427         RETURN(0);
8428 out:
8429         i = lo->ldo_dir_stripe_count;
8430         while (i < lo->ldo_dir_stripe_count + stripe_count && stripes[i])
8431                 dt_object_put(env, stripes[i++]);
8432
8433         OBD_FREE_PTR_ARRAY(stripes, stripe_count + lo->ldo_dir_stripe_count);
8434         return rc;
8435 }
8436
8437 static int lod_dir_declare_layout_detach(const struct lu_env *env,
8438                                          struct dt_object *dt,
8439                                          const struct md_layout_change *unused,
8440                                          struct thandle *th)
8441 {
8442         struct lod_thread_info *info = lod_env_info(env);
8443         struct lod_object *lo = lod_dt_obj(dt);
8444         struct dt_object *next = dt_object_child(dt);
8445         char *stripe_name = info->lti_key;
8446         struct dt_object *dto;
8447         int i;
8448         int rc = 0;
8449
8450         if (!dt_try_as_dir(env, dt, true))
8451                 return -ENOTDIR;
8452
8453         if (!lo->ldo_dir_stripe_count)
8454                 return lod_sub_declare_delete(env, next,
8455                                         (const struct dt_key *)dotdot, th);
8456
8457         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
8458                 dto = lo->ldo_stripe[i];
8459                 if (!dto)
8460                         continue;
8461
8462                 if (!dt_try_as_dir(env, dto, true))
8463                         return -ENOTDIR;
8464
8465                 rc = lod_sub_declare_delete(env, dto,
8466                                         (const struct dt_key *)dotdot, th);
8467                 if (rc)
8468                         return rc;
8469
8470                 snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
8471                          PFID(lu_object_fid(&dto->do_lu)), i);
8472
8473                 rc = lod_sub_declare_delete(env, next,
8474                                         (const struct dt_key *)stripe_name, th);
8475                 if (rc)
8476                         return rc;
8477
8478                 rc = lod_sub_declare_ref_del(env, next, th);
8479                 if (rc)
8480                         return rc;
8481         }
8482
8483         return 0;
8484 }
8485
8486 static int dt_dir_is_empty(const struct lu_env *env,
8487                            struct dt_object *obj)
8488 {
8489         struct dt_it *it;
8490         const struct dt_it_ops *iops;
8491         int rc;
8492
8493         ENTRY;
8494
8495         if (!dt_try_as_dir(env, obj, true))
8496                 RETURN(-ENOTDIR);
8497
8498         iops = &obj->do_index_ops->dio_it;
8499         it = iops->init(env, obj, LUDA_64BITHASH);
8500         if (IS_ERR(it))
8501                 RETURN(PTR_ERR(it));
8502
8503         rc = iops->get(env, it, (const struct dt_key *)"");
8504         if (rc > 0) {
8505                 int i;
8506
8507                 for (rc = 0, i = 0; rc == 0 && i < 3; ++i)
8508                         rc = iops->next(env, it);
8509                 if (!rc)
8510                         rc = -ENOTEMPTY;
8511                 else if (rc == 1)
8512                         rc = 0;
8513         } else if (!rc) {
8514                 /* Huh? Index contains no zero key? */
8515                 rc = -EIO;
8516         }
8517
8518         iops->put(env, it);
8519         iops->fini(env, it);
8520
8521         RETURN(rc);
8522 }
8523
8524 static int lod_dir_declare_layout_shrink(const struct lu_env *env,
8525                                          struct dt_object *dt,
8526                                          const struct md_layout_change *mlc,
8527                                          struct thandle *th)
8528 {
8529         struct lod_thread_info *info = lod_env_info(env);
8530         struct lod_object *lo = lod_dt_obj(dt);
8531         struct dt_object *next = dt_object_child(dt);
8532         struct lmv_user_md *lmu = mlc->mlc_buf.lb_buf;
8533         char *stripe_name = info->lti_key;
8534         struct lu_buf *lmv_buf = &info->lti_buf;
8535         __u32 final_stripe_count;
8536         struct dt_object *dto;
8537         int i;
8538         int rc;
8539
8540         LASSERT(lmu);
8541
8542         if (!dt_try_as_dir(env, dt, true))
8543                 return -ENOTDIR;
8544
8545         /* shouldn't be called on plain directory */
8546         LASSERT(lo->ldo_dir_stripe_count);
8547
8548         lmv_buf->lb_buf = &info->lti_lmv.lmv_md_v1;
8549         lmv_buf->lb_len = sizeof(info->lti_lmv.lmv_md_v1);
8550
8551         final_stripe_count = le32_to_cpu(lmu->lum_stripe_count);
8552         LASSERT(final_stripe_count &&
8553                 final_stripe_count < lo->ldo_dir_stripe_count);
8554
8555         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
8556                 dto = lo->ldo_stripe[i];
8557                 if (!dto)
8558                         continue;
8559
8560                 if (i < final_stripe_count) {
8561                         rc = lod_sub_declare_xattr_set(env, dto, lmv_buf,
8562                                                        XATTR_NAME_LMV,
8563                                                        LU_XATTR_REPLACE, th);
8564                         if (rc)
8565                                 return rc;
8566
8567                         continue;
8568                 }
8569
8570                 rc = dt_dir_is_empty(env, dto);
8571                 if (rc < 0)
8572                         return rc;
8573
8574                 rc = lod_sub_declare_ref_del(env, dto, th);
8575                 if (rc)
8576                         return rc;
8577
8578                 rc = lod_sub_declare_destroy(env, dto, th);
8579                 if (rc)
8580                         return rc;
8581
8582                 snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
8583                          PFID(lu_object_fid(&dto->do_lu)), i);
8584
8585                 rc = lod_sub_declare_delete(env, next,
8586                                         (const struct dt_key *)stripe_name, th);
8587                 if (rc)
8588                         return rc;
8589
8590                 rc = lod_sub_declare_ref_del(env, next, th);
8591                 if (rc)
8592                         return rc;
8593         }
8594
8595         rc = lod_sub_declare_xattr_set(env, next, lmv_buf, XATTR_NAME_LMV,
8596                                        LU_XATTR_REPLACE, th);
8597         return rc;
8598 }
8599
8600 /**
8601  * Allocate stripes for split directory.
8602  *
8603  * \param[in] env       execution environment
8604  * \param[in] dt        target object
8605  * \param[in] mlc       layout change data
8606  * \param[in] th        transaction handle
8607  *
8608  * \retval              0 on success
8609  * \retval              negative if failed
8610  */
8611 static int lod_dir_declare_layout_split(const struct lu_env *env,
8612                                         struct dt_object *dt,
8613                                         const struct md_layout_change *mlc,
8614                                         struct thandle *th)
8615 {
8616         struct lod_thread_info *info = lod_env_info(env);
8617         struct lod_device *lod = lu2lod_dev(dt->do_lu.lo_dev);
8618         struct lod_object *lo = lod_dt_obj(dt);
8619         struct dt_object_format *dof = &info->lti_format;
8620         struct lmv_user_md_v1 *lum = mlc->mlc_spec->u.sp_ea.eadata;
8621         struct dt_object **stripes;
8622         u32 stripe_count;
8623         u32 saved_count;
8624         int i;
8625         int rc;
8626
8627         ENTRY;
8628
8629         LASSERT(le32_to_cpu(lum->lum_magic) == LMV_USER_MAGIC);
8630         LASSERT(le32_to_cpu(lum->lum_stripe_offset) == LMV_OFFSET_DEFAULT);
8631
8632         saved_count = lo->ldo_dir_stripes_allocated;
8633         stripe_count = le32_to_cpu(lum->lum_stripe_count);
8634         if (stripe_count <= saved_count)
8635                 RETURN(-EINVAL);
8636
8637         dof->dof_type = DFT_DIR;
8638
8639         OBD_ALLOC(stripes, sizeof(*stripes) * stripe_count);
8640         if (!stripes)
8641                 RETURN(-ENOMEM);
8642
8643         for (i = 0; i < lo->ldo_dir_stripes_allocated; i++)
8644                 stripes[i] = lo->ldo_stripe[i];
8645
8646         lod_qos_statfs_update(env, lod, &lod->lod_mdt_descs);
8647         rc = lod_mdt_alloc_qos(env, lo, stripes, saved_count, stripe_count);
8648         if (rc == -EAGAIN)
8649                 rc = lod_mdt_alloc_rr(env, lo, stripes, saved_count,
8650                                       stripe_count);
8651         if (rc < 0) {
8652                 OBD_FREE(stripes, sizeof(*stripes) * stripe_count);
8653                 RETURN(rc);
8654         }
8655
8656         LASSERT(rc > saved_count);
8657         OBD_FREE(lo->ldo_stripe,
8658                  sizeof(*stripes) * lo->ldo_dir_stripes_allocated);
8659         lo->ldo_stripe = stripes;
8660         lo->ldo_is_foreign = 0;
8661         lo->ldo_dir_striped = 1;
8662         lo->ldo_dir_stripe_count = rc;
8663         lo->ldo_dir_stripes_allocated = stripe_count;
8664         lo->ldo_dir_split_hash = lo->ldo_dir_hash_type;
8665         lo->ldo_dir_hash_type = le32_to_cpu(lum->lum_hash_type);
8666         if (!lmv_is_known_hash_type(lo->ldo_dir_hash_type))
8667                 lo->ldo_dir_hash_type =
8668                         lod->lod_mdt_descs.ltd_lmv_desc.ld_pattern;
8669         lo->ldo_dir_hash_type |= LMV_HASH_FLAG_SPLIT | LMV_HASH_FLAG_MIGRATION;
8670         lo->ldo_dir_split_offset = saved_count;
8671         lo->ldo_dir_layout_version++;
8672         lo->ldo_dir_stripe_loaded = 1;
8673
8674         rc = lod_dir_declare_create_stripes(env, dt, mlc->mlc_attr, dof, th);
8675         if (rc)
8676                 lod_striping_free(env, lo);
8677
8678         RETURN(rc);
8679 }
8680
8681 /*
8682  * detach all stripes from dir master object, NB, stripes are not destroyed, but
8683  * deleted from it's parent namespace, this function is called in two places:
8684  * 1. mdd_migrate_mdt() detach stripes from source, and attach them to
8685  *    target.
8686  * 2. mdd_dir_layout_update() detach stripe before turning 1-stripe directory to
8687  *    a plain directory.
8688  *
8689  * \param[in] env       execution environment
8690  * \param[in] dt        target object
8691  * \param[in] mlc       layout change data
8692  * \param[in] th        transaction handle
8693  *
8694  * \retval              0 on success
8695  * \retval              negative if failed
8696  */
8697 static int lod_dir_layout_detach(const struct lu_env *env,
8698                                  struct dt_object *dt,
8699                                  const struct md_layout_change *mlc,
8700                                  struct thandle *th)
8701 {
8702         struct lod_thread_info *info = lod_env_info(env);
8703         struct lod_object *lo = lod_dt_obj(dt);
8704         struct dt_object *next = dt_object_child(dt);
8705         char *stripe_name = info->lti_key;
8706         struct dt_object *dto;
8707         int i;
8708         int rc = 0;
8709
8710         ENTRY;
8711
8712         if (!lo->ldo_dir_stripe_count) {
8713                 /* plain directory delete .. */
8714                 rc = lod_sub_delete(env, next,
8715                                     (const struct dt_key *)dotdot, th);
8716                 RETURN(rc);
8717         }
8718
8719         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
8720                 dto = lo->ldo_stripe[i];
8721                 if (!dto)
8722                         continue;
8723
8724                 rc = lod_sub_delete(env, dto,
8725                                     (const struct dt_key *)dotdot, th);
8726                 if (rc)
8727                         break;
8728
8729                 snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
8730                          PFID(lu_object_fid(&dto->do_lu)), i);
8731
8732                 rc = lod_sub_delete(env, next,
8733                                     (const struct dt_key *)stripe_name, th);
8734                 if (rc)
8735                         break;
8736
8737                 rc = lod_sub_ref_del(env, next, th);
8738                 if (rc)
8739                         break;
8740         }
8741
8742         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
8743                 dto = lo->ldo_stripe[i];
8744                 if (dto)
8745                         dt_object_put(env, dto);
8746         }
8747         OBD_FREE_PTR_ARRAY(lo->ldo_stripe, lo->ldo_dir_stripes_allocated);
8748         lo->ldo_stripe = NULL;
8749         lo->ldo_dir_stripes_allocated = 0;
8750         lo->ldo_dir_stripe_count = 0;
8751         dt->do_index_ops = &lod_index_ops;
8752
8753         RETURN(rc);
8754 }
8755
8756 static int lod_dir_layout_shrink(const struct lu_env *env,
8757                                  struct dt_object *dt,
8758                                  const struct md_layout_change *mlc,
8759                                  struct thandle *th)
8760 {
8761         struct lod_thread_info *info = lod_env_info(env);
8762         struct lod_object *lo = lod_dt_obj(dt);
8763         struct lod_device *lod = lu2lod_dev(lo->ldo_obj.do_lu.lo_dev);
8764         struct dt_object *next = dt_object_child(dt);
8765         struct lmv_user_md *lmu = mlc->mlc_buf.lb_buf;
8766         __u32 final_stripe_count;
8767         char *stripe_name = info->lti_key;
8768         struct dt_object *dto;
8769         struct lu_buf *lmv_buf = &info->lti_buf;
8770         struct lmv_mds_md_v1 *lmv = &info->lti_lmv.lmv_md_v1;
8771         u32 mdtidx;
8772         int type = LU_SEQ_RANGE_ANY;
8773         int i;
8774         int rc;
8775
8776         ENTRY;
8777
8778         final_stripe_count = le32_to_cpu(lmu->lum_stripe_count);
8779
8780         lmv_buf->lb_buf = lmv;
8781         lmv_buf->lb_len = sizeof(*lmv);
8782         lmv->lmv_magic = cpu_to_le32(LMV_MAGIC_STRIPE);
8783         lmv->lmv_stripe_count = cpu_to_le32(final_stripe_count);
8784         lmv->lmv_hash_type = cpu_to_le32(lo->ldo_dir_hash_type) &
8785                              cpu_to_le32(LMV_HASH_TYPE_MASK |
8786                                          LMV_HASH_FLAG_FIXED);
8787         lmv->lmv_layout_version =
8788                         cpu_to_le32(lo->ldo_dir_layout_version + 1);
8789         lmv->lmv_migrate_offset = 0;
8790         lmv->lmv_migrate_hash = 0;
8791
8792         for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
8793                 dto = lo->ldo_stripe[i];
8794                 if (!dto)
8795                         continue;
8796
8797                 if (i < final_stripe_count) {
8798                         rc = lod_fld_lookup(env, lod,
8799                                             lu_object_fid(&dto->do_lu),
8800                                             &mdtidx, &type);
8801                         if (rc)
8802                                 RETURN(rc);
8803
8804                         lmv->lmv_master_mdt_index = cpu_to_le32(mdtidx);
8805                         rc = lod_sub_xattr_set(env, dto, lmv_buf,
8806                                                XATTR_NAME_LMV,
8807                                                LU_XATTR_REPLACE, th);
8808                         if (rc)
8809                                 RETURN(rc);
8810
8811                         continue;
8812                 }
8813
8814                 dt_write_lock(env, dto, DT_TGT_CHILD);
8815                 rc = lod_sub_ref_del(env, dto, th);
8816                 dt_write_unlock(env, dto);
8817                 if (rc)
8818                         RETURN(rc);
8819
8820                 rc = lod_sub_destroy(env, dto, th);
8821                 if (rc)
8822                         RETURN(rc);
8823
8824                 snprintf(stripe_name, sizeof(info->lti_key), DFID":%d",
8825                          PFID(lu_object_fid(&dto->do_lu)), i);
8826
8827                 rc = lod_sub_delete(env, next,
8828                                     (const struct dt_key *)stripe_name, th);
8829                 if (rc)
8830                         RETURN(rc);
8831
8832                 rc = lod_sub_ref_del(env, next, th);
8833                 if (rc)
8834                         RETURN(rc);
8835         }
8836
8837         rc = lod_fld_lookup(env, lod, lu_object_fid(&dt->do_lu), &mdtidx,
8838                             &type);
8839         if (rc)
8840                 RETURN(rc);
8841
8842         lmv->lmv_magic = cpu_to_le32(LMV_MAGIC_V1);
8843         lmv->lmv_master_mdt_index = cpu_to_le32(mdtidx);
8844         rc = lod_sub_xattr_set(env, next, lmv_buf, XATTR_NAME_LMV,
8845                                LU_XATTR_REPLACE, th);
8846         if (rc)
8847                 RETURN(rc);
8848
8849         for (i = final_stripe_count; i < lo->ldo_dir_stripe_count; i++) {
8850                 dto = lo->ldo_stripe[i];
8851                 if (dto)
8852                         dt_object_put(env, dto);
8853         }
8854         lo->ldo_dir_stripe_count = final_stripe_count;
8855
8856         RETURN(rc);
8857 }
8858
8859 static mlc_handler dir_mlc_declare_ops[MD_LAYOUT_MAX] = {
8860         [MD_LAYOUT_ATTACH] = lod_dir_declare_layout_attach,
8861         [MD_LAYOUT_DETACH] = lod_dir_declare_layout_detach,
8862         [MD_LAYOUT_SHRINK] = lod_dir_declare_layout_shrink,
8863         [MD_LAYOUT_SPLIT]  = lod_dir_declare_layout_split,
8864 };
8865
8866 static mlc_handler dir_mlc_ops[MD_LAYOUT_MAX] = {
8867         [MD_LAYOUT_DETACH] = lod_dir_layout_detach,
8868         [MD_LAYOUT_SHRINK] = lod_dir_layout_shrink,
8869 };
8870
8871 static int lod_declare_layout_change(const struct lu_env *env,
8872                 struct dt_object *dt, struct md_layout_change *mlc,
8873                 struct thandle *th)
8874 {
8875         struct lod_thread_info  *info = lod_env_info(env);
8876         struct lod_object *lo = lod_dt_obj(dt);
8877         int rc;
8878
8879         ENTRY;
8880
8881         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
8882                 LASSERT(dir_mlc_declare_ops[mlc->mlc_opc]);
8883                 rc = dir_mlc_declare_ops[mlc->mlc_opc](env, dt, mlc, th);
8884                 RETURN(rc);
8885         }
8886
8887         if (!S_ISREG(dt->do_lu.lo_header->loh_attr) || !dt_object_exists(dt) ||
8888             dt_object_remote(dt_object_child(dt)))
8889                 RETURN(-EINVAL);
8890
8891         rc = lod_striping_load(env, lo);
8892         if (rc)
8893                 GOTO(out, rc);
8894
8895         LASSERT(lo->ldo_comp_cnt > 0);
8896
8897         rc = lod_layout_data_init(info, lo->ldo_comp_cnt);
8898         if (rc)
8899                 GOTO(out, rc);
8900
8901         switch (lo->ldo_flr_state) {
8902         case LCM_FL_NONE:
8903                 rc = lod_declare_update_plain(env, lo, mlc->mlc_intent,
8904                                               &mlc->mlc_buf, th);
8905                 break;
8906         case LCM_FL_RDONLY:
8907                 rc = lod_declare_update_rdonly(env, lo, mlc, th);
8908                 break;
8909         case LCM_FL_WRITE_PENDING:
8910                 rc = lod_declare_update_write_pending(env, lo, mlc, th);
8911                 break;
8912         case LCM_FL_SYNC_PENDING:
8913                 rc = lod_declare_update_sync_pending(env, lo, mlc, th);
8914                 break;
8915         default:
8916                 rc = -ENOTSUPP;
8917                 break;
8918         }
8919 out:
8920         RETURN(rc);
8921 }
8922
8923 /**
8924  * Instantiate layout component objects which covers the intent write offset.
8925  */
8926 static int lod_layout_change(const struct lu_env *env, struct dt_object *dt,
8927                              struct md_layout_change *mlc, struct thandle *th)
8928 {
8929         struct lu_attr *attr = &lod_env_info(env)->lti_attr;
8930         struct lu_attr *layout_attr = &lod_env_info(env)->lti_layout_attr;
8931         struct lod_object *lo = lod_dt_obj(dt);
8932         int rc;
8933
8934         ENTRY;
8935
8936         if (S_ISDIR(dt->do_lu.lo_header->loh_attr)) {
8937                 LASSERT(dir_mlc_ops[mlc->mlc_opc]);
8938                 rc = dir_mlc_ops[mlc->mlc_opc](env, dt, mlc, th);
8939                 RETURN(rc);
8940         }
8941
8942         rc = lod_striped_create(env, dt, attr, NULL, th);
8943         if (!rc && layout_attr->la_valid & LA_LAYOUT_VERSION) {
8944                 layout_attr->la_layout_version |= lo->ldo_layout_gen;
8945                 rc = lod_attr_set(env, dt, layout_attr, th);
8946         }
8947
8948         RETURN(rc);
8949 }
8950
8951 const struct dt_object_operations lod_obj_ops = {
8952         .do_read_lock           = lod_read_lock,
8953         .do_write_lock          = lod_write_lock,
8954         .do_read_unlock         = lod_read_unlock,
8955         .do_write_unlock        = lod_write_unlock,
8956         .do_write_locked        = lod_write_locked,
8957         .do_attr_get            = lod_attr_get,
8958         .do_declare_attr_set    = lod_declare_attr_set,
8959         .do_attr_set            = lod_attr_set,
8960         .do_xattr_get           = lod_xattr_get,
8961         .do_declare_xattr_set   = lod_declare_xattr_set,
8962         .do_xattr_set           = lod_xattr_set,
8963         .do_declare_xattr_del   = lod_declare_xattr_del,
8964         .do_xattr_del           = lod_xattr_del,
8965         .do_xattr_list          = lod_xattr_list,
8966         .do_ah_init             = lod_ah_init,
8967         .do_declare_create      = lod_declare_create,
8968         .do_create              = lod_create,
8969         .do_declare_destroy     = lod_declare_destroy,
8970         .do_destroy             = lod_destroy,
8971         .do_index_try           = lod_index_try,
8972         .do_declare_ref_add     = lod_declare_ref_add,
8973         .do_ref_add             = lod_ref_add,
8974         .do_declare_ref_del     = lod_declare_ref_del,
8975         .do_ref_del             = lod_ref_del,
8976         .do_object_sync         = lod_object_sync,
8977         .do_object_lock         = lod_object_lock,
8978         .do_object_unlock       = lod_object_unlock,
8979         .do_invalidate          = lod_invalidate,
8980         .do_declare_layout_change = lod_declare_layout_change,
8981         .do_layout_change       = lod_layout_change,
8982 };
8983
8984 /**
8985  * Implementation of dt_body_operations::dbo_read.
8986  *
8987  * \see dt_body_operations::dbo_read() in the API description for details.
8988  */
8989 static ssize_t lod_read(const struct lu_env *env, struct dt_object *dt,
8990                         struct lu_buf *buf, loff_t *pos)
8991 {
8992         struct dt_object *next = dt_object_child(dt);
8993
8994         LASSERT(S_ISREG(dt->do_lu.lo_header->loh_attr) ||
8995                 S_ISLNK(dt->do_lu.lo_header->loh_attr));
8996         return next->do_body_ops->dbo_read(env, next, buf, pos);
8997 }
8998
8999 /**
9000  * Implementation of dt_body_operations::dbo_declare_write.
9001  *
9002  * \see dt_body_operations::dbo_declare_write() in the API description
9003  * for details.
9004  */
9005 static ssize_t lod_declare_write(const struct lu_env *env,
9006                                  struct dt_object *dt,
9007                                  const struct lu_buf *buf, loff_t pos,
9008                                  struct thandle *th)
9009 {
9010         return lod_sub_declare_write(env, dt_object_child(dt), buf, pos, th);
9011 }
9012
9013 /**
9014  * Implementation of dt_body_operations::dbo_write.
9015  *
9016  * \see dt_body_operations::dbo_write() in the API description for details.
9017  */
9018 static ssize_t lod_write(const struct lu_env *env, struct dt_object *dt,
9019                          const struct lu_buf *buf, loff_t *pos,
9020                          struct thandle *th)
9021 {
9022         LASSERT(S_ISREG(dt->do_lu.lo_header->loh_attr) ||
9023                 S_ISLNK(dt->do_lu.lo_header->loh_attr));
9024         return lod_sub_write(env, dt_object_child(dt), buf, pos, th);
9025 }
9026
9027 static int lod_declare_punch(const struct lu_env *env, struct dt_object *dt,
9028                              __u64 start, __u64 end, struct thandle *th)
9029 {
9030         if (dt_object_remote(dt))
9031                 return -ENOTSUPP;
9032
9033         return lod_sub_declare_punch(env, dt_object_child(dt), start, end, th);
9034 }
9035
9036 static int lod_punch(const struct lu_env *env, struct dt_object *dt,
9037                      __u64 start, __u64 end, struct thandle *th)
9038 {
9039         if (dt_object_remote(dt))
9040                 return -ENOTSUPP;
9041
9042         LASSERT(S_ISREG(dt->do_lu.lo_header->loh_attr));
9043         return lod_sub_punch(env, dt_object_child(dt), start, end, th);
9044 }
9045
9046 /*
9047  * different type of files use the same body_ops because object may be created
9048  * in OUT, where there is no chance to set correct body_ops for each type, so
9049  * body_ops themselves will check file type inside, see lod_read/write/punch for
9050  * details.
9051  */
9052 static const struct dt_body_operations lod_body_ops = {
9053         .dbo_read               = lod_read,
9054         .dbo_declare_write      = lod_declare_write,
9055         .dbo_write              = lod_write,
9056         .dbo_declare_punch      = lod_declare_punch,
9057         .dbo_punch              = lod_punch,
9058 };
9059
9060 /**
9061  * Implementation of lu_object_operations::loo_object_init.
9062  *
9063  * The function determines the type and the index of the target device using
9064  * sequence of the object's FID. Then passes control down to the
9065  * corresponding device:
9066  *  OSD for the local objects, OSP for remote
9067  *
9068  * \see lu_object_operations::loo_object_init() in the API description
9069  * for details.
9070  */
9071 static int lod_object_init(const struct lu_env *env, struct lu_object *lo,
9072                            const struct lu_object_conf *conf)
9073 {
9074         struct lod_device       *lod    = lu2lod_dev(lo->lo_dev);
9075         struct lu_device        *cdev   = NULL;
9076         struct lu_object        *cobj;
9077         struct lod_tgt_descs    *ltd    = NULL;
9078         struct lod_tgt_desc     *tgt;
9079         u32                      idx    = 0;
9080         int                      type   = LU_SEQ_RANGE_ANY;
9081         int                      rc;
9082         ENTRY;
9083
9084         rc = lod_fld_lookup(env, lod, lu_object_fid(lo), &idx, &type);
9085         if (rc != 0)
9086                 RETURN(rc);
9087
9088         if (type == LU_SEQ_RANGE_MDT &&
9089             idx == lu_site2seq(lo->lo_dev->ld_site)->ss_node_id) {
9090                 cdev = &lod->lod_child->dd_lu_dev;
9091         } else if (type == LU_SEQ_RANGE_MDT) {
9092                 ltd = &lod->lod_mdt_descs;
9093                 lod_getref(ltd);
9094         } else if (type == LU_SEQ_RANGE_OST) {
9095                 ltd = &lod->lod_ost_descs;
9096                 lod_getref(ltd);
9097         } else {
9098                 LBUG();
9099         }
9100
9101         if (ltd != NULL) {
9102                 if (ltd->ltd_tgts_size > idx &&
9103                     test_bit(idx, ltd->ltd_tgt_bitmap)) {
9104                         tgt = LTD_TGT(ltd, idx);
9105
9106                         LASSERT(tgt != NULL);
9107                         LASSERT(tgt->ltd_tgt != NULL);
9108
9109                         cdev = &(tgt->ltd_tgt->dd_lu_dev);
9110                 }
9111                 lod_putref(lod, ltd);
9112         }
9113
9114         if (unlikely(cdev == NULL))
9115                 RETURN(-ENOENT);
9116
9117         cobj = cdev->ld_ops->ldo_object_alloc(env, lo->lo_header, cdev);
9118         if (unlikely(cobj == NULL))
9119                 RETURN(-ENOMEM);
9120
9121         lu2lod_obj(lo)->ldo_obj.do_body_ops = &lod_body_ops;
9122
9123         lu_object_add(lo, cobj);
9124
9125         RETURN(0);
9126 }
9127
9128 /**
9129  *
9130  * Release resources associated with striping.
9131  *
9132  * If the object is striped (regular or directory), then release
9133  * the stripe objects references and free the ldo_stripe array.
9134  *
9135  * \param[in] env       execution environment
9136  * \param[in] lo        object
9137  */
9138 void lod_striping_free_nolock(const struct lu_env *env, struct lod_object *lo)
9139 {
9140         struct lod_layout_component *lod_comp;
9141         __u32 obj_attr = lo->ldo_obj.do_lu.lo_header->loh_attr;
9142         int i, j;
9143
9144         if (unlikely(lo->ldo_is_foreign)) {
9145                 if (S_ISREG(obj_attr)) {
9146                         lod_free_foreign_lov(lo);
9147                         lo->ldo_comp_cached = 0;
9148                 } else if (S_ISDIR(obj_attr)) {
9149                         lod_free_foreign_lmv(lo);
9150                         lo->ldo_dir_stripe_loaded = 0;
9151                 }
9152         } else if (lo->ldo_stripe != NULL) {
9153                 LASSERT(lo->ldo_comp_entries == NULL);
9154                 LASSERT(lo->ldo_dir_stripes_allocated > 0);
9155
9156                 for (i = 0; i < lo->ldo_dir_stripe_count; i++) {
9157                         if (lo->ldo_stripe[i])
9158                                 dt_object_put(env, lo->ldo_stripe[i]);
9159                 }
9160
9161                 j = sizeof(struct dt_object *) * lo->ldo_dir_stripes_allocated;
9162                 OBD_FREE(lo->ldo_stripe, j);
9163                 lo->ldo_stripe = NULL;
9164                 lo->ldo_dir_stripes_allocated = 0;
9165                 lo->ldo_dir_stripe_loaded = 0;
9166                 lo->ldo_dir_stripe_count = 0;
9167                 lo->ldo_obj.do_index_ops = NULL;
9168         } else if (lo->ldo_comp_entries != NULL) {
9169                 for (i = 0; i < lo->ldo_comp_cnt; i++) {
9170                         /* free lod_layout_component::llc_stripe array */
9171                         lod_comp = &lo->ldo_comp_entries[i];
9172
9173                         if (lod_comp->llc_stripe == NULL)
9174                                 continue;
9175                         LASSERT(lod_comp->llc_stripes_allocated != 0);
9176                         for (j = 0; j < lod_comp->llc_stripes_allocated; j++) {
9177                                 if (lod_comp->llc_stripe[j] != NULL)
9178                                         lu_object_put(env,
9179                                                &lod_comp->llc_stripe[j]->do_lu);
9180                         }
9181                         OBD_FREE_PTR_ARRAY(lod_comp->llc_stripe,
9182                                            lod_comp->llc_stripes_allocated);
9183                         lod_comp->llc_stripe = NULL;
9184                         OBD_FREE_PTR_ARRAY(lod_comp->llc_ost_indices,
9185                                            lod_comp->llc_stripes_allocated);
9186                         lod_comp->llc_ost_indices = NULL;
9187                         lod_comp->llc_stripes_allocated = 0;
9188                 }
9189                 lod_free_comp_entries(lo);
9190                 lo->ldo_comp_cached = 0;
9191         }
9192 }
9193
9194 void lod_striping_free(const struct lu_env *env, struct lod_object *lo)
9195 {
9196         mutex_lock(&lo->ldo_layout_mutex);
9197         lod_striping_free_nolock(env, lo);
9198         mutex_unlock(&lo->ldo_layout_mutex);
9199 }
9200
9201 /**
9202  * Implementation of lu_object_operations::loo_object_free.
9203  *
9204  * \see lu_object_operations::loo_object_free() in the API description
9205  * for details.
9206  */
9207 static void lod_object_free(const struct lu_env *env, struct lu_object *o)
9208 {
9209         struct lod_object *lo = lu2lod_obj(o);
9210
9211         /* release all underlying object pinned */
9212         lod_striping_free(env, lo);
9213         lu_object_fini(o);
9214         /* lo doesn't contain a lu_object_header, so we don't need call_rcu */
9215         OBD_SLAB_FREE_PTR(lo, lod_object_kmem);
9216 }
9217
9218 /**
9219  * Implementation of lu_object_operations::loo_object_release.
9220  *
9221  * \see lu_object_operations::loo_object_release() in the API description
9222  * for details.
9223  */
9224 static void lod_object_release(const struct lu_env *env, struct lu_object *o)
9225 {
9226         /* XXX: shouldn't we release everything here in case if object
9227          * creation failed before? */
9228 }
9229
9230 /**
9231  * Implementation of lu_object_operations::loo_object_print.
9232  *
9233  * \see lu_object_operations::loo_object_print() in the API description
9234  * for details.
9235  */
9236 static int lod_object_print(const struct lu_env *env, void *cookie,
9237                             lu_printer_t p, const struct lu_object *l)
9238 {
9239         struct lod_object *o = lu2lod_obj((struct lu_object *) l);
9240
9241         return (*p)(env, cookie, LUSTRE_LOD_NAME"-object@%p", o);
9242 }
9243
9244 const struct lu_object_operations lod_lu_obj_ops = {
9245         .loo_object_init        = lod_object_init,
9246         .loo_object_free        = lod_object_free,
9247         .loo_object_release     = lod_object_release,
9248         .loo_object_print       = lod_object_print,
9249 };