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