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