Whamcloud - gitweb
fdfab4a50e3a3d2efc8a87407b86322fda21a5f5
[fs/lustre-release.git] / lustre / lfsck / lfsck_lib.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 (c) 2013, 2017, Intel Corporation.
24  */
25 /*
26  * lustre/lfsck/lfsck_lib.c
27  *
28  * Author: Fan, Yong <fan.yong@intel.com>
29  */
30
31 #define DEBUG_SUBSYSTEM S_LFSCK
32
33 #include <linux/kthread.h>
34 #include <linux/sched.h>
35 #include <linux/list.h>
36 #include <linux/delay.h>
37 #include <lu_object.h>
38 #include <dt_object.h>
39 #include <md_object.h>
40 #include <lustre_fld.h>
41 #include <lustre_lib.h>
42 #include <lustre_net.h>
43 #include <lustre_lfsck.h>
44
45 #include "lfsck_internal.h"
46
47 #define LFSCK_CHECKPOINT_SKIP   1
48
49 /* define lfsck thread key */
50 LU_KEY_INIT(lfsck, struct lfsck_thread_info);
51
52 static void lfsck_key_fini(const struct lu_context *ctx,
53                            struct lu_context_key *key, void *data)
54 {
55         struct lfsck_thread_info *info = data;
56
57         lu_buf_free(&info->lti_linkea_buf);
58         lu_buf_free(&info->lti_linkea_buf2);
59         lu_buf_free(&info->lti_big_buf);
60         OBD_FREE_PTR(info);
61 }
62
63 LU_CONTEXT_KEY_DEFINE(lfsck, LCT_MD_THREAD | LCT_DT_THREAD);
64 LU_KEY_INIT_GENERIC(lfsck);
65
66 static LIST_HEAD(lfsck_instance_list);
67 static LIST_HEAD(lfsck_ost_orphan_list);
68 static LIST_HEAD(lfsck_mdt_orphan_list);
69 static DEFINE_SPINLOCK(lfsck_instance_lock);
70
71 const char *const lfsck_flags_names[] = {
72         "scanned-once",
73         "inconsistent",
74         "upgrade",
75         "incomplete",
76         "crashed_lastid",
77         NULL
78 };
79
80 const char *const lfsck_param_names[] = {
81         NULL,
82         "failout",
83         "dryrun",
84         "all_targets",
85         "broadcast",
86         "orphan",
87         "create_ostobj",
88         "create_mdtobj",
89         NULL,
90         "delay_create_ostobj",
91         NULL
92 };
93
94 enum lfsck_verify_lpf_types {
95         LVLT_BY_BOOKMARK        = 0,
96         LVLT_BY_NAMEENTRY       = 1,
97 };
98
99 static inline void
100 lfsck_reset_ltd_status(struct lfsck_tgt_desc *ltd, enum lfsck_type type)
101 {
102         if (type == LFSCK_TYPE_LAYOUT) {
103                 ltd->ltd_layout_status = LS_MAX;
104                 ltd->ltd_layout_repaired = 0;
105         } else {
106                 ltd->ltd_namespace_status = LS_MAX;
107                 ltd->ltd_namespace_repaired = 0;
108         }
109 }
110
111 static int lfsck_tgt_descs_init(struct lfsck_tgt_descs *ltds)
112 {
113         spin_lock_init(&ltds->ltd_lock);
114         init_rwsem(&ltds->ltd_rw_sem);
115         INIT_LIST_HEAD(&ltds->ltd_orphan);
116         ltds->ltd_tgts_bitmap = CFS_ALLOCATE_BITMAP(BITS_PER_LONG);
117         if (ltds->ltd_tgts_bitmap == NULL)
118                 return -ENOMEM;
119
120         return 0;
121 }
122
123 static void lfsck_tgt_descs_fini(struct lfsck_tgt_descs *ltds)
124 {
125         struct lfsck_tgt_desc   *ltd;
126         struct lfsck_tgt_desc   *next;
127         int                      idx;
128
129         down_write(&ltds->ltd_rw_sem);
130
131         list_for_each_entry_safe(ltd, next, &ltds->ltd_orphan,
132                                  ltd_orphan_list) {
133                 list_del_init(&ltd->ltd_orphan_list);
134                 lfsck_tgt_put(ltd);
135         }
136
137         if (unlikely(ltds->ltd_tgts_bitmap == NULL)) {
138                 up_write(&ltds->ltd_rw_sem);
139
140                 return;
141         }
142
143         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
144                 ltd = lfsck_ltd2tgt(ltds, idx);
145                 if (likely(ltd != NULL)) {
146                         LASSERT(list_empty(&ltd->ltd_layout_list));
147                         LASSERT(list_empty(&ltd->ltd_layout_phase_list));
148                         LASSERT(list_empty(&ltd->ltd_namespace_list));
149                         LASSERT(list_empty(&ltd->ltd_namespace_phase_list));
150
151                         ltds->ltd_tgtnr--;
152                         cfs_bitmap_clear(ltds->ltd_tgts_bitmap, idx);
153                         lfsck_assign_tgt(ltds, NULL, idx);
154                         lfsck_tgt_put(ltd);
155                 }
156         }
157
158         LASSERTF(ltds->ltd_tgtnr == 0, "tgt count unmatched: %d\n",
159                  ltds->ltd_tgtnr);
160
161         for (idx = 0; idx < ARRAY_SIZE(ltds->ltd_tgts_idx); idx++) {
162                 if (ltds->ltd_tgts_idx[idx] != NULL) {
163                         OBD_FREE_PTR(ltds->ltd_tgts_idx[idx]);
164                         ltds->ltd_tgts_idx[idx] = NULL;
165                 }
166         }
167
168         CFS_FREE_BITMAP(ltds->ltd_tgts_bitmap);
169         ltds->ltd_tgts_bitmap = NULL;
170         up_write(&ltds->ltd_rw_sem);
171 }
172
173 static int __lfsck_add_target(const struct lu_env *env,
174                               struct lfsck_instance *lfsck,
175                               struct lfsck_tgt_desc *ltd,
176                               bool for_ost, bool locked)
177 {
178         struct lfsck_tgt_descs *ltds;
179         __u32                   index = ltd->ltd_index;
180         int                     rc    = 0;
181         ENTRY;
182
183         if (for_ost)
184                 ltds = &lfsck->li_ost_descs;
185         else
186                 ltds = &lfsck->li_mdt_descs;
187
188         if (!locked)
189                 down_write(&ltds->ltd_rw_sem);
190
191         LASSERT(ltds->ltd_tgts_bitmap != NULL);
192
193         if (index >= ltds->ltd_tgts_bitmap->size) {
194                 __u32 newsize = max((__u32)ltds->ltd_tgts_bitmap->size,
195                                     (__u32)BITS_PER_LONG);
196                 struct cfs_bitmap *old_bitmap = ltds->ltd_tgts_bitmap;
197                 struct cfs_bitmap *new_bitmap;
198
199                 while (newsize < index + 1)
200                         newsize <<= 1;
201
202                 new_bitmap = CFS_ALLOCATE_BITMAP(newsize);
203                 if (new_bitmap == NULL)
204                         GOTO(unlock, rc = -ENOMEM);
205
206                 if (ltds->ltd_tgtnr > 0)
207                         cfs_bitmap_copy(new_bitmap, old_bitmap);
208                 ltds->ltd_tgts_bitmap = new_bitmap;
209                 CFS_FREE_BITMAP(old_bitmap);
210         }
211
212         if (cfs_bitmap_check(ltds->ltd_tgts_bitmap, index)) {
213                 CERROR("%s: the device %s (%u) is registered already\n",
214                        lfsck_lfsck2name(lfsck),
215                        ltd->ltd_tgt->dd_lu_dev.ld_obd->obd_name, index);
216                 GOTO(unlock, rc = -EEXIST);
217         }
218
219         if (ltds->ltd_tgts_idx[index / TGT_PTRS_PER_BLOCK] == NULL) {
220                 OBD_ALLOC_PTR(ltds->ltd_tgts_idx[index / TGT_PTRS_PER_BLOCK]);
221                 if (ltds->ltd_tgts_idx[index / TGT_PTRS_PER_BLOCK] == NULL)
222                         GOTO(unlock, rc = -ENOMEM);
223         }
224
225         lfsck_assign_tgt(ltds, ltd, index);
226         cfs_bitmap_set(ltds->ltd_tgts_bitmap, index);
227         ltds->ltd_tgtnr++;
228
229         GOTO(unlock, rc = 0);
230
231 unlock:
232         if (!locked)
233                 up_write(&ltds->ltd_rw_sem);
234
235         return rc;
236 }
237
238 static int lfsck_add_target_from_orphan(const struct lu_env *env,
239                                         struct lfsck_instance *lfsck)
240 {
241         struct lfsck_tgt_descs  *ltds    = &lfsck->li_ost_descs;
242         struct lfsck_tgt_desc   *ltd;
243         struct lfsck_tgt_desc   *next;
244         struct list_head        *head    = &lfsck_ost_orphan_list;
245         int                      rc;
246         bool                     for_ost = true;
247
248 again:
249         spin_lock(&lfsck_instance_lock);
250         list_for_each_entry_safe(ltd, next, head, ltd_orphan_list) {
251                 if (ltd->ltd_key == lfsck->li_bottom)
252                         list_move_tail(&ltd->ltd_orphan_list,
253                                        &ltds->ltd_orphan);
254         }
255         spin_unlock(&lfsck_instance_lock);
256
257         down_write(&ltds->ltd_rw_sem);
258         while (!list_empty(&ltds->ltd_orphan)) {
259                 ltd = list_entry(ltds->ltd_orphan.next,
260                                  struct lfsck_tgt_desc,
261                                  ltd_orphan_list);
262                 list_del_init(&ltd->ltd_orphan_list);
263                 rc = __lfsck_add_target(env, lfsck, ltd, for_ost, true);
264                 /* Do not hold the semaphore for too long time. */
265                 up_write(&ltds->ltd_rw_sem);
266                 if (rc != 0)
267                         return rc;
268
269                 down_write(&ltds->ltd_rw_sem);
270         }
271         up_write(&ltds->ltd_rw_sem);
272
273         if (for_ost) {
274                 ltds = &lfsck->li_mdt_descs;
275                 head = &lfsck_mdt_orphan_list;
276                 for_ost = false;
277                 goto again;
278         }
279
280         return 0;
281 }
282
283 static inline struct lfsck_component *
284 __lfsck_component_find(struct lfsck_instance *lfsck, __u16 type,
285                        struct list_head *list)
286 {
287         struct lfsck_component *com;
288
289         list_for_each_entry(com, list, lc_link) {
290                 if (com->lc_type == type)
291                         return com;
292         }
293         return NULL;
294 }
295
296 struct lfsck_component *
297 lfsck_component_find(struct lfsck_instance *lfsck, __u16 type)
298 {
299         struct lfsck_component *com;
300
301         spin_lock(&lfsck->li_lock);
302         com = __lfsck_component_find(lfsck, type, &lfsck->li_list_scan);
303         if (com != NULL)
304                 goto unlock;
305
306         com = __lfsck_component_find(lfsck, type,
307                                      &lfsck->li_list_double_scan);
308         if (com != NULL)
309                 goto unlock;
310
311         com = __lfsck_component_find(lfsck, type, &lfsck->li_list_idle);
312
313 unlock:
314         if (com != NULL)
315                 lfsck_component_get(com);
316         spin_unlock(&lfsck->li_lock);
317         return com;
318 }
319
320 void lfsck_component_cleanup(const struct lu_env *env,
321                              struct lfsck_component *com)
322 {
323         if (!list_empty(&com->lc_link))
324                 list_del_init(&com->lc_link);
325         if (!list_empty(&com->lc_link_dir))
326                 list_del_init(&com->lc_link_dir);
327
328         lfsck_component_put(env, com);
329 }
330
331 int lfsck_fid_alloc(const struct lu_env *env, struct lfsck_instance *lfsck,
332                     struct lu_fid *fid, bool locked)
333 {
334         struct lfsck_bookmark   *bk = &lfsck->li_bookmark_ram;
335         int                      rc = 0;
336         ENTRY;
337
338         if (!locked)
339                 mutex_lock(&lfsck->li_mutex);
340
341         rc = seq_client_alloc_fid(env, lfsck->li_seq, fid);
342         if (rc >= 0) {
343                 bk->lb_last_fid = *fid;
344                 /* We do not care about whether the subsequent sub-operations
345                  * failed or not. The worst case is that one FID is lost that
346                  * is not a big issue for the LFSCK since it is relative rare
347                  * for LFSCK create. */
348                 rc = lfsck_bookmark_store(env, lfsck);
349         }
350
351         if (!locked)
352                 mutex_unlock(&lfsck->li_mutex);
353
354         RETURN(rc);
355 }
356
357 static int __lfsck_ibits_lock(const struct lu_env *env,
358                               struct lfsck_instance *lfsck,
359                               struct dt_object *obj, struct ldlm_res_id *resid,
360                               struct lustre_handle *lh, __u64 bits,
361                               enum ldlm_mode mode)
362 {
363         struct lfsck_thread_info        *info   = lfsck_env_info(env);
364         union ldlm_policy_data          *policy = &info->lti_policy;
365         __u64                            flags  = LDLM_FL_ATOMIC_CB;
366         int                              rc;
367
368         LASSERT(lfsck->li_namespace != NULL);
369
370         memset(policy, 0, sizeof(*policy));
371         policy->l_inodebits.bits = bits;
372         if (dt_object_remote(obj)) {
373                 struct ldlm_enqueue_info *einfo = &info->lti_einfo;
374
375                 memset(einfo, 0, sizeof(*einfo));
376                 einfo->ei_type = LDLM_IBITS;
377                 einfo->ei_mode = mode;
378                 einfo->ei_cb_bl = ldlm_blocking_ast;
379                 einfo->ei_cb_cp = ldlm_completion_ast;
380                 einfo->ei_res_id = resid;
381                 einfo->ei_req_slot = 1;
382
383                 rc = dt_object_lock(env, obj, lh, einfo, policy);
384                 /* for regular checks LFSCK doesn't use LDLM locking,
385                  * so the state isn't coherent. here we just took LDLM
386                  * lock for coherency and it's time to invalidate
387                  * previous state */
388                 if (rc == ELDLM_OK)
389                         dt_invalidate(env, obj);
390         } else {
391                 rc = ldlm_cli_enqueue_local(env, lfsck->li_namespace, resid,
392                                             LDLM_IBITS, policy, mode,
393                                             &flags, ldlm_blocking_ast,
394                                             ldlm_completion_ast, NULL, NULL,
395                                             0, LVB_T_NONE, NULL, lh);
396         }
397
398         if (rc == ELDLM_OK) {
399                 rc = 0;
400         } else {
401                 memset(lh, 0, sizeof(*lh));
402                 rc = -EIO;
403         }
404
405         return rc;
406 }
407
408 /**
409  * Request the specified ibits lock for the given object.
410  *
411  * Before the LFSCK modifying on the namespace visible object,
412  * it needs to acquire related ibits ldlm lock.
413  *
414  * \param[in] env       pointer to the thread context
415  * \param[in] lfsck     pointer to the lfsck instance
416  * \param[in] obj       pointer to the dt_object to be locked
417  * \param[out] lh       pointer to the lock handle
418  * \param[in] bits      the bits for the ldlm lock to be acquired
419  * \param[in] mode      the mode for the ldlm lock to be acquired
420  *
421  * \retval              0 for success
422  * \retval              negative error number on failure
423  */
424 int lfsck_ibits_lock(const struct lu_env *env, struct lfsck_instance *lfsck,
425                      struct dt_object *obj, struct lustre_handle *lh,
426                      __u64 bits, enum ldlm_mode mode)
427 {
428         struct ldlm_res_id *resid = &lfsck_env_info(env)->lti_resid;
429
430         LASSERT(!lustre_handle_is_used(lh));
431
432         fid_build_reg_res_name(lfsck_dto2fid(obj), resid);
433         return __lfsck_ibits_lock(env, lfsck, obj, resid, lh, bits, mode);
434 }
435
436 /**
437  * Request the remote LOOKUP lock for the given object.
438  *
439  * If \a pobj is remote, the LOOKUP lock of \a obj is on the MDT where
440  * \a pobj is, acquire LOOKUP lock there.
441  *
442  * \param[in] env       pointer to the thread context
443  * \param[in] lfsck     pointer to the lfsck instance
444  * \param[in] pobj      pointer to parent dt_object
445  * \param[in] obj       pointer to the dt_object to be locked
446  * \param[out] lh       pointer to the lock handle
447  * \param[in] mode      the mode for the ldlm lock to be acquired
448  *
449  * \retval              0 for success
450  * \retval              negative error number on failure
451  */
452 int lfsck_remote_lookup_lock(const struct lu_env *env,
453                              struct lfsck_instance *lfsck,
454                              struct dt_object *pobj, struct dt_object *obj,
455                              struct lustre_handle *lh, enum ldlm_mode mode)
456 {
457         struct ldlm_res_id *resid = &lfsck_env_info(env)->lti_resid;
458
459         LASSERT(!lustre_handle_is_used(lh));
460
461         fid_build_reg_res_name(lfsck_dto2fid(obj), resid);
462         return __lfsck_ibits_lock(env, lfsck, pobj, resid, lh,
463                                   MDS_INODELOCK_LOOKUP, mode);
464 }
465
466 /**
467  * Release the the specified ibits lock.
468  *
469  * If the lock has been acquired before, release it
470  * and cleanup the handle. Otherwise, do nothing.
471  *
472  * \param[in] lh        pointer to the lock handle
473  * \param[in] mode      the mode for the ldlm lock to be released
474  */
475 void lfsck_ibits_unlock(struct lustre_handle *lh, enum ldlm_mode mode)
476 {
477         if (lustre_handle_is_used(lh)) {
478                 ldlm_lock_decref(lh, mode);
479                 memset(lh, 0, sizeof(*lh));
480         }
481 }
482
483 /**
484  * Request compound ibits locks for the given <obj, name> pairs.
485  *
486  * Before the LFSCK modifying on the namespace visible object, it needs to
487  * acquire related ibits ldlm lock. Usually, we can use lfsck_ibits_lock for
488  * the lock purpose. But the simple lfsck_ibits_lock for directory-based
489  * modificationis (such as insert name entry to the directory) may be too
490  * coarse-grained and not efficient.
491  *
492  * The lfsck_lock() will request compound ibits locks on the specified
493  * <obj, name> pairs: the PDO (Parallel Directory Operations) ibits (UPDATE)
494  * lock on the directory object, and the regular ibits lock on the name hash.
495  *
496  * \param[in] env       pointer to the thread context
497  * \param[in] lfsck     pointer to the lfsck instance
498  * \param[in] obj       pointer to the dt_object to be locked
499  * \param[in] name      used for building the PDO lock resource
500  * \param[out] llh      pointer to the lfsck_lock_handle
501  * \param[in] bits      the bits for the ldlm lock to be acquired
502  * \param[in] mode      the mode for the ldlm lock to be acquired
503  *
504  * \retval              0 for success
505  * \retval              negative error number on failure
506  */
507 int lfsck_lock(const struct lu_env *env, struct lfsck_instance *lfsck,
508                struct dt_object *obj, const char *name,
509                struct lfsck_lock_handle *llh, __u64 bits, enum ldlm_mode mode)
510 {
511         struct ldlm_res_id *resid = &lfsck_env_info(env)->lti_resid;
512         int                 rc;
513
514         LASSERT(S_ISDIR(lfsck_object_type(obj)));
515         LASSERT(name != NULL);
516         LASSERT(name[0] != 0);
517         LASSERT(!lustre_handle_is_used(&llh->llh_pdo_lh));
518         LASSERT(!lustre_handle_is_used(&llh->llh_reg_lh));
519
520         switch (mode) {
521         case LCK_EX:
522                 llh->llh_pdo_mode = LCK_EX;
523                 break;
524         case LCK_PW:
525                 llh->llh_pdo_mode = LCK_CW;
526                 break;
527         case LCK_PR:
528                 llh->llh_pdo_mode = LCK_CR;
529                 break;
530         default:
531                 CDEBUG(D_LFSCK, "%s: unexpected PDO lock mode %u on the obj "
532                        DFID"\n", lfsck_lfsck2name(lfsck), mode,
533                        PFID(lfsck_dto2fid(obj)));
534                 LBUG();
535         }
536
537         fid_build_reg_res_name(lfsck_dto2fid(obj), resid);
538         rc = __lfsck_ibits_lock(env, lfsck, obj, resid, &llh->llh_pdo_lh,
539                                 MDS_INODELOCK_UPDATE, llh->llh_pdo_mode);
540         if (rc != 0)
541                 return rc;
542
543         llh->llh_reg_mode = mode;
544         resid->name[LUSTRE_RES_ID_HSH_OFF] = ll_full_name_hash(NULL, name,
545                                                                strlen(name));
546         LASSERT(resid->name[LUSTRE_RES_ID_HSH_OFF] != 0);
547         rc = __lfsck_ibits_lock(env, lfsck, obj, resid, &llh->llh_reg_lh,
548                                 bits, llh->llh_reg_mode);
549         if (rc != 0)
550                 lfsck_ibits_unlock(&llh->llh_pdo_lh, llh->llh_pdo_mode);
551
552         return rc;
553 }
554
555 /**
556  * Release the the compound ibits locks.
557  *
558  * \param[in] llh       pointer to the lfsck_lock_handle to be released
559  */
560 void lfsck_unlock(struct lfsck_lock_handle *llh)
561 {
562         lfsck_ibits_unlock(&llh->llh_reg_lh, llh->llh_reg_mode);
563         lfsck_ibits_unlock(&llh->llh_pdo_lh, llh->llh_pdo_mode);
564 }
565
566 int lfsck_find_mdt_idx_by_fid(const struct lu_env *env,
567                               struct lfsck_instance *lfsck,
568                               const struct lu_fid *fid)
569 {
570         struct seq_server_site  *ss     = lfsck_dev_site(lfsck);
571         struct lu_seq_range     *range  = &lfsck_env_info(env)->lti_range;
572         int                      rc;
573
574         if (unlikely(fid_seq(fid) == FID_SEQ_LOCAL_FILE)) {
575                 /* "ROOT" is always on the MDT0. */
576                 if (lu_fid_eq(fid, &lfsck->li_global_root_fid))
577                         return 0;
578
579                 return lfsck_dev_idx(lfsck);
580         }
581
582         fld_range_set_mdt(range);
583         rc = fld_server_lookup(env, ss->ss_server_fld, fid_seq(fid), range);
584         if (rc == 0)
585                 rc = range->lsr_index;
586
587         return rc;
588 }
589
590 const char dot[] = ".";
591 const char dotdot[] = "..";
592 static const char dotlustre[] = ".lustre";
593 static const char lostfound[] = "lost+found";
594
595 /**
596  * Remove the name entry from the .lustre/lost+found directory.
597  *
598  * No need to care about the object referenced by the name entry,
599  * either the name entry is invalid or redundant, or the referenced
600  * object has been processed or will be handled by others.
601  *
602  * \param[in] env       pointer to the thread context
603  * \param[in] lfsck     pointer to the lfsck instance
604  * \param[in] name      the name for the name entry to be removed
605  *
606  * \retval              0 for success
607  * \retval              negative error number on failure
608  */
609 static int lfsck_lpf_remove_name_entry(const struct lu_env *env,
610                                        struct lfsck_instance *lfsck,
611                                        const char *name)
612 {
613         struct dt_object        *parent = lfsck->li_lpf_root_obj;
614         struct dt_device        *dev    = lfsck_obj2dev(parent);
615         struct thandle          *th;
616         struct lfsck_lock_handle *llh   = &lfsck_env_info(env)->lti_llh;
617         int                      rc;
618         ENTRY;
619
620         rc = lfsck_lock(env, lfsck, parent, name, llh,
621                         MDS_INODELOCK_UPDATE, LCK_PW);
622         if (rc != 0)
623                 RETURN(rc);
624
625         th = lfsck_trans_create(env, dev, lfsck);
626         if (IS_ERR(th))
627                 GOTO(unlock, rc = PTR_ERR(th));
628
629         rc = dt_declare_delete(env, parent, (const struct dt_key *)name, th);
630         if (rc != 0)
631                 GOTO(stop, rc);
632
633         rc = dt_declare_ref_del(env, parent, th);
634         if (rc != 0)
635                 GOTO(stop, rc);
636
637         rc = dt_trans_start_local(env, dev, th);
638         if (rc != 0)
639                 GOTO(stop, rc);
640
641         rc = dt_delete(env, parent, (const struct dt_key *)name, th);
642         if (rc != 0)
643                 GOTO(stop, rc);
644
645         dt_write_lock(env, parent, 0);
646         rc = dt_ref_del(env, parent, th);
647         dt_write_unlock(env, parent);
648
649         GOTO(stop, rc);
650
651 stop:
652         dt_trans_stop(env, dev, th);
653
654 unlock:
655         lfsck_unlock(llh);
656
657         CDEBUG(D_LFSCK, "%s: remove name entry "DFID"/%s: rc = %d\n",
658                lfsck_lfsck2name(lfsck), PFID(lfsck_dto2fid(parent)), name, rc);
659
660         return rc;
661 }
662
663 static int lfsck_create_lpf_local(const struct lu_env *env,
664                                   struct lfsck_instance *lfsck,
665                                   struct dt_object *child,
666                                   struct lu_attr *la,
667                                   struct dt_object_format *dof,
668                                   const char *name)
669 {
670         struct dt_insert_rec    *rec    = &lfsck_env_info(env)->lti_dt_rec;
671         struct dt_object        *parent = lfsck->li_lpf_root_obj;
672         struct dt_device        *dev    = lfsck_obj2dev(child);
673         struct lfsck_bookmark   *bk     = &lfsck->li_bookmark_ram;
674         struct dt_object        *bk_obj = lfsck->li_bookmark_obj;
675         const struct lu_fid     *cfid   = lfsck_dto2fid(child);
676         struct thandle          *th     = NULL;
677         struct linkea_data       ldata  = { NULL };
678         struct lu_buf            linkea_buf;
679         const struct lu_name    *cname;
680         loff_t                   pos    = 0;
681         int                      len    = sizeof(struct lfsck_bookmark);
682         int                      rc;
683         ENTRY;
684
685         cname = lfsck_name_get_const(env, name, strlen(name));
686         rc = linkea_links_new(&ldata, &lfsck_env_info(env)->lti_linkea_buf2,
687                               cname, lfsck_dto2fid(parent));
688         if (rc != 0)
689                 RETURN(rc);
690
691         th = lfsck_trans_create(env, dev, lfsck);
692         if (IS_ERR(th))
693                 RETURN(PTR_ERR(th));
694
695         /* 1a. create child */
696         rc = dt_declare_create(env, child, la, NULL, dof, th);
697         if (rc != 0)
698                 GOTO(stop, rc);
699
700         if (!dt_try_as_dir(env, child))
701                 GOTO(stop, rc = -ENOTDIR);
702
703         /* 2a. increase child nlink */
704         rc = dt_declare_ref_add(env, child, th);
705         if (rc != 0)
706                 GOTO(stop, rc);
707
708         /* 3a. insert dot into child dir */
709         rec->rec_type = S_IFDIR;
710         rec->rec_fid = cfid;
711         rc = dt_declare_insert(env, child, (const struct dt_rec *)rec,
712                                (const struct dt_key *)dot, th);
713         if (rc != 0)
714                 GOTO(stop, rc);
715
716         /* 4a. insert dotdot into child dir */
717         rec->rec_fid = &LU_LPF_FID;
718         rc = dt_declare_insert(env, child, (const struct dt_rec *)rec,
719                                (const struct dt_key *)dotdot, th);
720         if (rc != 0)
721                 GOTO(stop, rc);
722
723         /* 5a. insert linkEA for child */
724         lfsck_buf_init(&linkea_buf, ldata.ld_buf->lb_buf,
725                        ldata.ld_leh->leh_len);
726         rc = dt_declare_xattr_set(env, child, &linkea_buf,
727                                   XATTR_NAME_LINK, 0, th);
728         if (rc != 0)
729                 GOTO(stop, rc);
730
731         /* 6a. insert name into parent dir */
732         rec->rec_type = S_IFDIR;
733         rec->rec_fid = cfid;
734         rc = dt_declare_insert(env, parent, (const struct dt_rec *)rec,
735                                (const struct dt_key *)name, th);
736         if (rc != 0)
737                 GOTO(stop, rc);
738
739         /* 7a. increase parent nlink */
740         rc = dt_declare_ref_add(env, parent, th);
741         if (rc != 0)
742                 GOTO(stop, rc);
743
744         /* 8a. update bookmark */
745         rc = dt_declare_record_write(env, bk_obj,
746                                      lfsck_buf_get(env, bk, len), 0, th);
747         if (rc != 0)
748                 GOTO(stop, rc);
749
750         rc = dt_trans_start_local(env, dev, th);
751         if (rc != 0)
752                 GOTO(stop, rc);
753
754         dt_write_lock(env, child, 0);
755         /* 1b. create child */
756         rc = dt_create(env, child, la, NULL, dof, th);
757         if (rc != 0)
758                 GOTO(unlock, rc);
759
760         /* 2b. increase child nlink */
761         rc = dt_ref_add(env, child, th);
762         if (rc != 0)
763                 GOTO(unlock, rc);
764
765         /* 3b. insert dot into child dir */
766         rec->rec_fid = cfid;
767         rc = dt_insert(env, child, (const struct dt_rec *)rec,
768                        (const struct dt_key *)dot, th);
769         if (rc != 0)
770                 GOTO(unlock, rc);
771
772         /* 4b. insert dotdot into child dir */
773         rec->rec_fid = &LU_LPF_FID;
774         rc = dt_insert(env, child, (const struct dt_rec *)rec,
775                        (const struct dt_key *)dotdot, th);
776         if (rc != 0)
777                 GOTO(unlock, rc);
778
779         /* 5b. insert linkEA for child. */
780         rc = dt_xattr_set(env, child, &linkea_buf,
781                           XATTR_NAME_LINK, 0, th);
782         dt_write_unlock(env, child);
783         if (rc != 0)
784                 GOTO(stop, rc);
785
786         /* 6b. insert name into parent dir */
787         rec->rec_fid = cfid;
788         rc = dt_insert(env, parent, (const struct dt_rec *)rec,
789                        (const struct dt_key *)name, th);
790         if (rc != 0)
791                 GOTO(stop, rc);
792
793         dt_write_lock(env, parent, 0);
794         /* 7b. increase parent nlink */
795         rc = dt_ref_add(env, parent, th);
796         dt_write_unlock(env, parent);
797         if (rc != 0)
798                 GOTO(stop, rc);
799
800         bk->lb_lpf_fid = *cfid;
801         lfsck_bookmark_cpu_to_le(&lfsck->li_bookmark_disk, bk);
802
803         /* 8b. update bookmark */
804         rc = dt_record_write(env, bk_obj,
805                              lfsck_buf_get(env, bk, len), &pos, th);
806
807         GOTO(stop, rc);
808
809 unlock:
810         dt_write_unlock(env, child);
811
812 stop:
813         dt_trans_stop(env, dev, th);
814
815         return rc;
816 }
817
818 static int lfsck_create_lpf_remote(const struct lu_env *env,
819                                    struct lfsck_instance *lfsck,
820                                    struct dt_object *child,
821                                    struct lu_attr *la,
822                                    struct dt_object_format *dof,
823                                    const char *name)
824 {
825         struct dt_insert_rec    *rec    = &lfsck_env_info(env)->lti_dt_rec;
826         struct dt_object        *parent = lfsck->li_lpf_root_obj;
827         struct lfsck_bookmark   *bk     = &lfsck->li_bookmark_ram;
828         struct dt_object        *bk_obj = lfsck->li_bookmark_obj;
829         const struct lu_fid     *cfid   = lfsck_dto2fid(child);
830         struct thandle          *th     = NULL;
831         struct linkea_data       ldata  = { NULL };
832         struct lu_buf            linkea_buf;
833         const struct lu_name    *cname;
834         struct dt_device        *dev;
835         loff_t                   pos    = 0;
836         int                      len    = sizeof(struct lfsck_bookmark);
837         int                      rc;
838         ENTRY;
839
840         cname = lfsck_name_get_const(env, name, strlen(name));
841         rc = linkea_links_new(&ldata, &lfsck_env_info(env)->lti_linkea_buf2,
842                               cname, lfsck_dto2fid(parent));
843         if (rc != 0)
844                 RETURN(rc);
845
846         /* Create .lustre/lost+found/MDTxxxx. */
847
848         /* XXX: Currently, cross-MDT create operation needs to create the child
849          *      object firstly, then insert name into the parent directory. For
850          *      this case, the child object resides on current MDT (local), but
851          *      the parent ".lustre/lost+found" may be on remote MDT. It is not
852          *      easy to contain all the sub-modifications orderly within single
853          *      transaction.
854          *
855          *      To avoid more inconsistency, we split the create operation into
856          *      two transactions:
857          *
858          *      1) create the child and update the lfsck_bookmark::lb_lpf_fid
859          *         locally.
860          *      2) insert the name "MDTXXXX" in the parent ".lustre/lost+found"
861          *         remotely.
862          *
863          *      If 1) done, but 2) failed, then go ahead, the LFSCK will try to
864          *      repair such inconsistency when LFSCK run next time. */
865
866         /* Transaction I: locally */
867
868         dev = lfsck_obj2dev(child);
869         th = lfsck_trans_create(env, dev, lfsck);
870         if (IS_ERR(th))
871                 RETURN(PTR_ERR(th));
872
873         /* 1a. create child */
874         rc = dt_declare_create(env, child, la, NULL, dof, th);
875         if (rc != 0)
876                 GOTO(stop, rc);
877
878         if (!dt_try_as_dir(env, child))
879                 GOTO(stop, rc = -ENOTDIR);
880
881         /* 2a. increase child nlink */
882         rc = dt_declare_ref_add(env, child, th);
883         if (rc != 0)
884                 GOTO(stop, rc);
885
886         /* 3a. insert dot into child dir */
887         rec->rec_type = S_IFDIR;
888         rec->rec_fid = cfid;
889         rc = dt_declare_insert(env, child, (const struct dt_rec *)rec,
890                                (const struct dt_key *)dot, th);
891         if (rc != 0)
892                 GOTO(stop, rc);
893
894         /* 4a. insert dotdot into child dir */
895         rec->rec_fid = &LU_LPF_FID;
896         rc = dt_declare_insert(env, child, (const struct dt_rec *)rec,
897                                (const struct dt_key *)dotdot, th);
898         if (rc != 0)
899                 GOTO(stop, rc);
900
901         /* 5a. insert linkEA for child */
902         lfsck_buf_init(&linkea_buf, ldata.ld_buf->lb_buf,
903                        ldata.ld_leh->leh_len);
904         rc = dt_declare_xattr_set(env, child, &linkea_buf,
905                                   XATTR_NAME_LINK, 0, th);
906         if (rc != 0)
907                 GOTO(stop, rc);
908
909         /* 6a. update bookmark */
910         rc = dt_declare_record_write(env, bk_obj,
911                                      lfsck_buf_get(env, bk, len), 0, th);
912         if (rc != 0)
913                 GOTO(stop, rc);
914
915         rc = dt_trans_start_local(env, dev, th);
916         if (rc != 0)
917                 GOTO(stop, rc);
918
919         dt_write_lock(env, child, 0);
920         /* 1b. create child */
921         rc = dt_create(env, child, la, NULL, dof, th);
922         if (rc != 0)
923                 GOTO(unlock, rc);
924
925         /* 2b. increase child nlink */
926         rc = dt_ref_add(env, child, th);
927         if (rc != 0)
928                 GOTO(unlock, rc);
929
930         /* 3b. insert dot into child dir */
931         rec->rec_type = S_IFDIR;
932         rec->rec_fid = cfid;
933         rc = dt_insert(env, child, (const struct dt_rec *)rec,
934                        (const struct dt_key *)dot, th);
935         if (rc != 0)
936                 GOTO(unlock, rc);
937
938         /* 4b. insert dotdot into child dir */
939         rec->rec_fid = &LU_LPF_FID;
940         rc = dt_insert(env, child, (const struct dt_rec *)rec,
941                        (const struct dt_key *)dotdot, th);
942         if (rc != 0)
943                 GOTO(unlock, rc);
944
945         /* 5b. insert linkEA for child */
946         rc = dt_xattr_set(env, child, &linkea_buf,
947                           XATTR_NAME_LINK, 0, th);
948         if (rc != 0)
949                 GOTO(unlock, rc);
950
951         bk->lb_lpf_fid = *cfid;
952         lfsck_bookmark_cpu_to_le(&lfsck->li_bookmark_disk, bk);
953
954         /* 6b. update bookmark */
955         rc = dt_record_write(env, bk_obj,
956                              lfsck_buf_get(env, bk, len), &pos, th);
957
958         dt_write_unlock(env, child);
959         dt_trans_stop(env, dev, th);
960         if (rc != 0)
961                 RETURN(rc);
962
963         /* Transaction II: remotely */
964
965         dev = lfsck_obj2dev(parent);
966         th = lfsck_trans_create(env, dev, lfsck);
967         if (IS_ERR(th))
968                 RETURN(PTR_ERR(th));
969
970         th->th_sync = 1;
971         /* 5a. insert name into parent dir */
972         rec->rec_fid = cfid;
973         rc = dt_declare_insert(env, parent, (const struct dt_rec *)rec,
974                                (const struct dt_key *)name, th);
975         if (rc != 0)
976                 GOTO(stop, rc);
977
978         /* 6a. increase parent nlink */
979         rc = dt_declare_ref_add(env, parent, th);
980         if (rc != 0)
981                 GOTO(stop, rc);
982
983         rc = dt_trans_start_local(env, dev, th);
984         if (rc != 0)
985                 GOTO(stop, rc);
986
987         /* 5b. insert name into parent dir */
988         rc = dt_insert(env, parent, (const struct dt_rec *)rec,
989                        (const struct dt_key *)name, th);
990         if (rc != 0)
991                 GOTO(stop, rc);
992
993         dt_write_lock(env, parent, 0);
994         /* 6b. increase parent nlink */
995         rc = dt_ref_add(env, parent, th);
996         dt_write_unlock(env, parent);
997
998         GOTO(stop, rc);
999
1000 unlock:
1001         dt_write_unlock(env, child);
1002 stop:
1003         dt_trans_stop(env, dev, th);
1004
1005         if (rc != 0 && dev == lfsck_obj2dev(parent))
1006                 CDEBUG(D_LFSCK, "%s: partially created the object "DFID
1007                        "for orphans, but failed to insert the name %s "
1008                        "to the .lustre/lost+found/. Such inconsistency "
1009                        "will be repaired when LFSCK run next time: rc = %d\n",
1010                        lfsck_lfsck2name(lfsck), PFID(cfid), name, rc);
1011
1012         return rc;
1013 }
1014
1015 /**
1016  * Create the MDTxxxx directory under /ROOT/.lustre/lost+found/
1017  *
1018  * The /ROOT/.lustre/lost+found/MDTxxxx/ directory is used for holding
1019  * orphans and other uncertain inconsistent objects found during the
1020  * LFSCK. Such directory will be created by the LFSCK engine on the
1021  * local MDT before the LFSCK scanning.
1022  *
1023  * \param[in] env       pointer to the thread context
1024  * \param[in] lfsck     pointer to the lfsck instance
1025  *
1026  * \retval              0 for success
1027  * \retval              negative error number on failure
1028  */
1029 static int lfsck_create_lpf(const struct lu_env *env,
1030                             struct lfsck_instance *lfsck)
1031 {
1032         struct lfsck_bookmark    *bk    = &lfsck->li_bookmark_ram;
1033         struct lfsck_thread_info *info  = lfsck_env_info(env);
1034         struct lu_fid            *cfid  = &info->lti_fid2;
1035         struct lu_attr           *la    = &info->lti_la;
1036         struct dt_object_format  *dof   = &info->lti_dof;
1037         struct dt_object         *parent = lfsck->li_lpf_root_obj;
1038         struct dt_object         *child = NULL;
1039         struct lfsck_lock_handle *llh   = &info->lti_llh;
1040         char                      name[8];
1041         int                       node  = lfsck_dev_idx(lfsck);
1042         int                       rc    = 0;
1043         ENTRY;
1044
1045         LASSERT(lfsck->li_master);
1046         LASSERT(parent != NULL);
1047         LASSERT(lfsck->li_lpf_obj == NULL);
1048
1049         snprintf(name, 8, "MDT%04x", node);
1050         rc = lfsck_lock(env, lfsck, parent, name, llh,
1051                         MDS_INODELOCK_UPDATE, LCK_PW);
1052         if (rc != 0)
1053                 RETURN(rc);
1054
1055         if (fid_is_zero(&bk->lb_lpf_fid)) {
1056                 /* There is corner case that: in former LFSCK scanning we have
1057                  * created the .lustre/lost+found/MDTxxxx but failed to update
1058                  * the lfsck_bookmark::lb_lpf_fid successfully. So need lookup
1059                  * it from MDT0 firstly. */
1060                 rc = dt_lookup_dir(env, parent, name, cfid);
1061                 if (rc != 0 && rc != -ENOENT)
1062                         GOTO(unlock, rc);
1063
1064                 if (rc == 0) {
1065                         bk->lb_lpf_fid = *cfid;
1066                         rc = lfsck_bookmark_store(env, lfsck);
1067                 } else {
1068                         rc = lfsck_fid_alloc(env, lfsck, cfid, true);
1069                 }
1070                 if (rc != 0)
1071                         GOTO(unlock, rc);
1072         } else {
1073                 *cfid = bk->lb_lpf_fid;
1074         }
1075
1076         child = lfsck_object_find_bottom_new(env, lfsck, cfid);
1077         if (IS_ERR(child))
1078                 GOTO(unlock, rc = PTR_ERR(child));
1079
1080         if (dt_object_exists(child) != 0) {
1081                 if (unlikely(!dt_try_as_dir(env, child)))
1082                         rc = -ENOTDIR;
1083                 else
1084                         lfsck->li_lpf_obj = child;
1085
1086                 GOTO(unlock, rc);
1087         }
1088
1089         memset(la, 0, sizeof(*la));
1090         la->la_atime = la->la_mtime = la->la_ctime = ktime_get_real_seconds();
1091         la->la_mode = S_IFDIR | S_IRWXU;
1092         la->la_valid = LA_ATIME | LA_MTIME | LA_CTIME | LA_MODE |
1093                        LA_UID | LA_GID | LA_TYPE;
1094         memset(dof, 0, sizeof(*dof));
1095         dof->dof_type = dt_mode_to_dft(S_IFDIR);
1096
1097         if (node == 0)
1098                 rc = lfsck_create_lpf_local(env, lfsck, child, la, dof, name);
1099         else
1100                 rc = lfsck_create_lpf_remote(env, lfsck, child, la, dof, name);
1101         if (rc == 0)
1102                 lfsck->li_lpf_obj = child;
1103
1104         GOTO(unlock, rc);
1105
1106 unlock:
1107         lfsck_unlock(llh);
1108         if (rc != 0 && child != NULL && !IS_ERR(child))
1109                 lfsck_object_put(env, child);
1110
1111         return rc;
1112 }
1113
1114 /**
1115  * Scan .lustre/lost+found for bad name entries and remove them.
1116  *
1117  * The valid name entry should be "MDTxxxx", the "xxxx" is the MDT device
1118  * index in the system. Any other formatted name is invalid and should be
1119  * removed.
1120  *
1121  * \param[in] env       pointer to the thread context
1122  * \param[in] lfsck     pointer to the lfsck instance
1123  *
1124  * \retval              0 for success
1125  * \retval              negative error number on failure
1126  */
1127 static int lfsck_scan_lpf_bad_entries(const struct lu_env *env,
1128                                       struct lfsck_instance *lfsck)
1129 {
1130         struct dt_object        *parent = lfsck->li_lpf_root_obj;
1131         struct lu_dirent        *ent    =
1132                         (struct lu_dirent *)lfsck_env_info(env)->lti_key;
1133         const struct dt_it_ops  *iops   = &parent->do_index_ops->dio_it;
1134         struct dt_it            *it;
1135         int                      rc;
1136         ENTRY;
1137
1138         it = iops->init(env, parent, LUDA_64BITHASH);
1139         if (IS_ERR(it))
1140                 RETURN(PTR_ERR(it));
1141
1142         rc = iops->load(env, it, 0);
1143         if (rc == 0)
1144                 rc = iops->next(env, it);
1145         else if (rc > 0)
1146                 rc = 0;
1147
1148         while (rc == 0) {
1149                 int off = 3;
1150
1151                 rc = iops->rec(env, it, (struct dt_rec *)ent, LUDA_64BITHASH);
1152                 if (rc != 0)
1153                         break;
1154
1155                 ent->lde_namelen = le16_to_cpu(ent->lde_namelen);
1156                 if (name_is_dot_or_dotdot(ent->lde_name, ent->lde_namelen))
1157                         goto next;
1158
1159                 /* name length must be strlen("MDTxxxx") */
1160                 if (ent->lde_namelen != 7)
1161                         goto remove;
1162
1163                 if (memcmp(ent->lde_name, "MDT", off) != 0)
1164                         goto remove;
1165
1166                 while (off < 7 && isxdigit(ent->lde_name[off]))
1167                         off++;
1168
1169                 if (off != 7) {
1170
1171 remove:
1172                         rc = lfsck_lpf_remove_name_entry(env, lfsck,
1173                                                          ent->lde_name);
1174                         if (rc != 0)
1175                                 break;
1176                 }
1177
1178 next:
1179                 rc = iops->next(env, it);
1180         }
1181
1182         iops->put(env, it);
1183         iops->fini(env, it);
1184
1185         RETURN(rc > 0 ? 0 : rc);
1186 }
1187
1188 static int lfsck_update_lpf_entry(const struct lu_env *env,
1189                                   struct lfsck_instance *lfsck,
1190                                   struct dt_object *parent,
1191                                   struct dt_object *child,
1192                                   const char *name,
1193                                   enum lfsck_verify_lpf_types type)
1194 {
1195         int rc;
1196
1197         if (type == LVLT_BY_BOOKMARK) {
1198                 rc = lfsck_update_name_entry(env, lfsck, parent, name,
1199                                              lfsck_dto2fid(child), S_IFDIR);
1200         } else /* if (type == LVLT_BY_NAMEENTRY) */ {
1201                 lfsck->li_bookmark_ram.lb_lpf_fid = *lfsck_dto2fid(child);
1202                 rc = lfsck_bookmark_store(env, lfsck);
1203
1204                 CDEBUG(D_LFSCK, "%s: update LPF fid "DFID
1205                        " in the bookmark file: rc = %d\n",
1206                        lfsck_lfsck2name(lfsck),
1207                        PFID(lfsck_dto2fid(child)), rc);
1208         }
1209
1210         return rc;
1211 }
1212
1213 /**
1214  * Check whether the @child back references the @parent.
1215  *
1216  * Two cases:
1217  * 1) The child's FID is stored in the bookmark file. If the child back
1218  *    references the parent (LU_LPF_FID object) via its ".." entry, then
1219  *    insert the name (MDTxxxx) to the .lustre/lost+found; otherwise, if
1220  *    the child back references another parent2, then:
1221  * 1.1) If the parent2 recognizes the child, then update the bookmark file;
1222  * 1.2) Otherwise, the LFSCK cannot know whether there will be parent3 that
1223  *      references the child. So keep them there. As the LFSCK processing,
1224  *      the parent3 may be found, then when the LFSCK run next time, the
1225  *      inconsistency can be repaired.
1226  *
1227  * 2) The child's FID is stored in the .lustre/lost+found/ sub-directory name
1228  *    entry (MDTxxxx). If the child back references the parent (LU_LPF_FID obj)
1229  *    via its ".." entry, then update the bookmark file, otherwise, if the child
1230  *    back references another parent2, then:
1231  * 2.1) If the parent2 recognizes the child, then remove the sub-directory
1232  *      from .lustre/lost+found/;
1233  * 2.2) Otherwise, if the parent2 does not recognizes the child, trust the
1234  *      sub-directory name entry and update the child;
1235  * 2.3) Otherwise, if we do not know whether the parent2 recognizes the child
1236  *      or not, then keep them there.
1237  *
1238  * \param[in] env       pointer to the thread context
1239  * \param[in] lfsck     pointer to the lfsck instance
1240  * \param[in] child     pointer to the lost+found sub-directory object
1241  * \param[in] name      the name for lost+found sub-directory object
1242  * \param[out] fid      pointer to the buffer to hold the FID of the object
1243  *                      (called it as parent2) that is referenced via the
1244  *                      child's dotdot entry; it also can be the FID that
1245  *                      is referenced by the name entry under the parent2.
1246  * \param[in] type      to indicate where the child's FID is stored in
1247  *
1248  * \retval              positive number for uncertain inconsistency
1249  * \retval              0 for success
1250  * \retval              negative error number on failure
1251  */
1252 static int lfsck_verify_lpf_pairs(const struct lu_env *env,
1253                                   struct lfsck_instance *lfsck,
1254                                   struct dt_object *child, const char *name,
1255                                   struct lu_fid *fid,
1256                                   enum lfsck_verify_lpf_types type)
1257 {
1258         struct dt_object         *parent  = lfsck->li_lpf_root_obj;
1259         struct lfsck_thread_info *info    = lfsck_env_info(env);
1260         char                     *name2   = info->lti_key;
1261         struct lu_fid            *fid2    = &info->lti_fid3;
1262         struct dt_object         *parent2 = NULL;
1263         struct lustre_handle      lh      = { 0 };
1264         int                       rc;
1265         ENTRY;
1266
1267         fid_zero(fid);
1268         rc = dt_lookup_dir(env, child, dotdot, fid);
1269         if (rc != 0)
1270                 GOTO(linkea, rc);
1271
1272         if (!fid_is_sane(fid))
1273                 GOTO(linkea, rc = -EINVAL);
1274
1275         if (lu_fid_eq(fid, &LU_LPF_FID)) {
1276                 const struct lu_name *cname;
1277
1278                 if (lfsck->li_lpf_obj == NULL) {
1279                         lu_object_get(&child->do_lu);
1280                         lfsck->li_lpf_obj = child;
1281                 }
1282
1283                 cname = lfsck_name_get_const(env, name, strlen(name));
1284                 rc = lfsck_verify_linkea(env, lfsck, child, cname, &LU_LPF_FID);
1285                 if (rc == 0)
1286                         rc = lfsck_update_lpf_entry(env, lfsck, parent, child,
1287                                                     name, type);
1288
1289                 GOTO(out_done, rc);
1290         }
1291
1292         parent2 = lfsck_object_find_bottom(env, lfsck, fid);
1293         if (IS_ERR(parent2))
1294                 GOTO(linkea, parent2);
1295
1296         if (!dt_object_exists(parent2)) {
1297                 lfsck_object_put(env, parent2);
1298
1299                 GOTO(linkea, parent2 = ERR_PTR(-ENOENT));
1300         }
1301
1302         if (!dt_try_as_dir(env, parent2)) {
1303                 lfsck_object_put(env, parent2);
1304
1305                 GOTO(linkea, parent2 = ERR_PTR(-ENOTDIR));
1306         }
1307
1308 linkea:
1309         /* To prevent rename/unlink race */
1310         rc = lfsck_ibits_lock(env, lfsck, child, &lh,
1311                               MDS_INODELOCK_UPDATE, LCK_PR);
1312         if (rc != 0)
1313                 GOTO(out_put, rc);
1314
1315         dt_read_lock(env, child, 0);
1316         rc = lfsck_links_get_first(env, child, name2, fid2);
1317         if (rc != 0) {
1318                 dt_read_unlock(env, child);
1319                 lfsck_ibits_unlock(&lh, LCK_PR);
1320
1321                 GOTO(out_put, rc = 1);
1322         }
1323
1324         /* It is almost impossible that the bookmark file (or the name entry)
1325          * and the linkEA hit the same data corruption. Trust the linkEA. */
1326         if (lu_fid_eq(fid2, &LU_LPF_FID) && strcmp(name, name2) == 0) {
1327                 dt_read_unlock(env, child);
1328                 lfsck_ibits_unlock(&lh, LCK_PR);
1329
1330                 *fid = *fid2;
1331                 if (lfsck->li_lpf_obj == NULL) {
1332                         lu_object_get(&child->do_lu);
1333                         lfsck->li_lpf_obj = child;
1334                 }
1335
1336                 /* Update the child's dotdot entry */
1337                 rc = lfsck_update_name_entry(env, lfsck, child, dotdot,
1338                                              &LU_LPF_FID, S_IFDIR);
1339                 if (rc == 0)
1340                         rc = lfsck_update_lpf_entry(env, lfsck, parent, child,
1341                                                     name, type);
1342
1343                 GOTO(out_put, rc);
1344         }
1345
1346         if (parent2 == NULL || IS_ERR(parent2)) {
1347                 dt_read_unlock(env, child);
1348                 lfsck_ibits_unlock(&lh, LCK_PR);
1349
1350                 GOTO(out_done, rc = 1);
1351         }
1352
1353         rc = dt_lookup_dir(env, parent2, name2, fid);
1354         dt_read_unlock(env, child);
1355         lfsck_ibits_unlock(&lh, LCK_PR);
1356         if (rc != 0 && rc != -ENOENT)
1357                 GOTO(out_put, rc);
1358
1359         if (rc == -ENOENT || !lu_fid_eq(fid, lfsck_dto2fid(child))) {
1360                 if (type == LVLT_BY_BOOKMARK)
1361                         GOTO(out_put, rc = 1);
1362
1363                 /* Trust the name entry, update the child's dotdot entry. */
1364                 rc = lfsck_update_name_entry(env, lfsck, child, dotdot,
1365                                              &LU_LPF_FID, S_IFDIR);
1366
1367                 GOTO(out_put, rc);
1368         }
1369
1370         if (type == LVLT_BY_BOOKMARK) {
1371                 /* Invalid FID record in the bookmark file, reset it. */
1372                 fid_zero(&lfsck->li_bookmark_ram.lb_lpf_fid);
1373                 rc = lfsck_bookmark_store(env, lfsck);
1374
1375                 CDEBUG(D_LFSCK, "%s: reset invalid LPF fid "DFID
1376                        " in the bookmark file: rc = %d\n",
1377                        lfsck_lfsck2name(lfsck), PFID(lfsck_dto2fid(child)), rc);
1378         } else /* if (type == LVLT_BY_NAMEENTRY) */ {
1379                 /* The name entry is wrong, remove it. */
1380                 rc = lfsck_lpf_remove_name_entry(env, lfsck, name);
1381         }
1382
1383         GOTO(out_put, rc);
1384
1385 out_put:
1386         if (parent2 != NULL && !IS_ERR(parent2))
1387                 lfsck_object_put(env, parent2);
1388
1389 out_done:
1390         return rc;
1391 }
1392
1393 /**
1394  * Verify the /ROOT/.lustre/lost+found/ directory.
1395  *
1396  * /ROOT/.lustre/lost+found/ is a special directory to hold the objects that
1397  * the LFSCK does not exactly know how to handle, such as orphans. So before
1398  * the LFSCK scanning the system, the consistency of such directory needs to
1399  * be verified firstly to allow the users to use it during the LFSCK.
1400  *
1401  * \param[in] env       pointer to the thread context
1402  * \param[in] lfsck     pointer to the lfsck instance
1403  *
1404  * \retval              positive number for uncertain inconsistency
1405  * \retval              0 for success
1406  * \retval              negative error number on failure
1407  */
1408 int lfsck_verify_lpf(const struct lu_env *env, struct lfsck_instance *lfsck)
1409 {
1410         struct lfsck_thread_info *info   = lfsck_env_info(env);
1411         struct lu_fid            *pfid   = &info->lti_fid;
1412         struct lu_fid            *cfid   = &info->lti_fid2;
1413         struct lfsck_bookmark    *bk     = &lfsck->li_bookmark_ram;
1414         struct dt_object         *parent;
1415         /* child1's FID is in the bookmark file. */
1416         struct dt_object         *child1 = NULL;
1417         /* child2's FID is in the name entry MDTxxxx. */
1418         struct dt_object         *child2 = NULL;
1419         const struct lu_name     *cname;
1420         char                      name[8];
1421         int                       node   = lfsck_dev_idx(lfsck);
1422         int                       rc     = 0;
1423         ENTRY;
1424
1425         LASSERT(lfsck->li_master);
1426
1427         if (lfsck_is_dryrun(lfsck))
1428                 RETURN(0);
1429
1430         if (lfsck->li_lpf_root_obj != NULL)
1431                 RETURN(0);
1432
1433         if (node == 0) {
1434                 parent = lfsck_object_find_by_dev(env, lfsck->li_bottom,
1435                                                   &LU_LPF_FID);
1436         } else {
1437                 struct lfsck_tgt_desc *ltd;
1438
1439                 ltd = lfsck_tgt_get(&lfsck->li_mdt_descs, 0);
1440                 if (unlikely(ltd == NULL))
1441                         RETURN(-ENXIO);
1442
1443                 parent = lfsck_object_find_by_dev(env, ltd->ltd_tgt,
1444                                                   &LU_LPF_FID);
1445                 lfsck_tgt_put(ltd);
1446         }
1447
1448         if (IS_ERR(parent))
1449                 RETURN(PTR_ERR(parent));
1450
1451         LASSERT(dt_object_exists(parent));
1452
1453         if (unlikely(!dt_try_as_dir(env, parent))) {
1454                 lfsck_object_put(env, parent);
1455
1456                 GOTO(put, rc = -ENOTDIR);
1457         }
1458
1459         lfsck->li_lpf_root_obj = parent;
1460         if (node == 0) {
1461                 rc = lfsck_scan_lpf_bad_entries(env, lfsck);
1462                 if (rc != 0)
1463                         CDEBUG(D_LFSCK, "%s: scan .lustre/lost+found/ "
1464                                "for bad sub-directories: rc = %d\n",
1465                                lfsck_lfsck2name(lfsck), rc);
1466         }
1467
1468         /* child2 */
1469         snprintf(name, 8, "MDT%04x", node);
1470         rc = dt_lookup_dir(env, parent, name, cfid);
1471         if (rc == -ENOENT) {
1472                 rc = 0;
1473                 goto find_child1;
1474         }
1475
1476         if (rc != 0)
1477                 GOTO(put, rc);
1478
1479         /* Invalid FID in the name entry, remove the name entry. */
1480         if (!fid_is_norm(cfid)) {
1481                 rc = lfsck_lpf_remove_name_entry(env, lfsck, name);
1482                 if (rc != 0)
1483                         GOTO(put, rc);
1484
1485                 goto find_child1;
1486         }
1487
1488         child2 = lfsck_object_find_bottom(env, lfsck, cfid);
1489         if (IS_ERR(child2))
1490                 GOTO(put, rc = PTR_ERR(child2));
1491
1492         if (unlikely(!dt_object_exists(child2) ||
1493                      dt_object_remote(child2)) ||
1494                      !S_ISDIR(lfsck_object_type(child2))) {
1495                 rc = lfsck_lpf_remove_name_entry(env, lfsck, name);
1496                 if (rc != 0)
1497                         GOTO(put, rc);
1498
1499                 goto find_child1;
1500         }
1501
1502         if (unlikely(!dt_try_as_dir(env, child2)))
1503                 GOTO(put, rc = -ENOTDIR);
1504
1505 find_child1:
1506         if (fid_is_zero(&bk->lb_lpf_fid))
1507                 goto check_child2;
1508
1509         if (likely(lu_fid_eq(cfid, &bk->lb_lpf_fid))) {
1510                 if (lfsck->li_lpf_obj == NULL) {
1511                         lu_object_get(&child2->do_lu);
1512                         lfsck->li_lpf_obj = child2;
1513                 }
1514
1515                 cname = lfsck_name_get_const(env, name, strlen(name));
1516                 rc = lfsck_verify_linkea(env, lfsck, child2, cname,
1517                                          &LU_LPF_FID);
1518
1519                 GOTO(put, rc);
1520         }
1521
1522         if (unlikely(!fid_is_norm(&bk->lb_lpf_fid))) {
1523                 struct lu_fid tfid = bk->lb_lpf_fid;
1524
1525                 /* Invalid FID record in the bookmark file, reset it. */
1526                 fid_zero(&bk->lb_lpf_fid);
1527                 rc = lfsck_bookmark_store(env, lfsck);
1528
1529                 CDEBUG(D_LFSCK, "%s: reset invalid LPF fid "DFID
1530                        " in the bookmark file: rc = %d\n",
1531                        lfsck_lfsck2name(lfsck), PFID(&tfid), rc);
1532
1533                 if (rc != 0)
1534                         GOTO(put, rc);
1535
1536                 goto check_child2;
1537         }
1538
1539         child1 = lfsck_object_find_bottom(env, lfsck, &bk->lb_lpf_fid);
1540         if (IS_ERR(child1)) {
1541                 child1 = NULL;
1542                 goto check_child2;
1543         }
1544
1545         if (unlikely(!dt_object_exists(child1) ||
1546                      dt_object_remote(child1)) ||
1547                      !S_ISDIR(lfsck_object_type(child1))) {
1548                 /* Invalid FID record in the bookmark file, reset it. */
1549                 fid_zero(&bk->lb_lpf_fid);
1550                 rc = lfsck_bookmark_store(env, lfsck);
1551
1552                 CDEBUG(D_LFSCK, "%s: reset invalid LPF fid "DFID
1553                        " in the bookmark file: rc = %d\n",
1554                        lfsck_lfsck2name(lfsck),
1555                        PFID(lfsck_dto2fid(child1)), rc);
1556
1557                 if (rc != 0)
1558                         GOTO(put, rc);
1559
1560                 lfsck_object_put(env, child1);
1561                 child1 = NULL;
1562                 goto check_child2;
1563         }
1564
1565         if (unlikely(!dt_try_as_dir(env, child1))) {
1566                 lfsck_object_put(env, child1);
1567                 child1 = NULL;
1568                 rc = -ENOTDIR;
1569                 goto check_child2;
1570         }
1571
1572         rc = lfsck_verify_lpf_pairs(env, lfsck, child1, name, pfid,
1573                                     LVLT_BY_BOOKMARK);
1574         if (lu_fid_eq(pfid, &LU_LPF_FID))
1575                 GOTO(put, rc);
1576
1577 check_child2:
1578         if (child2 != NULL)
1579                 rc = lfsck_verify_lpf_pairs(env, lfsck, child2, name,
1580                                             pfid, LVLT_BY_NAMEENTRY);
1581
1582         GOTO(put, rc);
1583
1584 put:
1585         if (lfsck->li_lpf_obj != NULL) {
1586                 if (unlikely(!dt_try_as_dir(env, lfsck->li_lpf_obj))) {
1587                         lfsck_object_put(env, lfsck->li_lpf_obj);
1588                         lfsck->li_lpf_obj = NULL;
1589                         rc = -ENOTDIR;
1590                 }
1591         } else if (rc == 0) {
1592                 rc = lfsck_create_lpf(env, lfsck);
1593         }
1594
1595         if (child2 != NULL && !IS_ERR(child2))
1596                 lfsck_object_put(env, child2);
1597         if (child1 != NULL && !IS_ERR(child1))
1598                 lfsck_object_put(env, child1);
1599
1600         return rc;
1601 }
1602
1603 static int lfsck_fid_init(struct lfsck_instance *lfsck)
1604 {
1605         struct lfsck_bookmark   *bk     = &lfsck->li_bookmark_ram;
1606         struct seq_server_site  *ss     = lfsck_dev_site(lfsck);
1607         char                    *prefix;
1608         int                      rc     = 0;
1609         ENTRY;
1610
1611         if (unlikely(ss == NULL))
1612                 RETURN(-ENXIO);
1613
1614         OBD_ALLOC_PTR(lfsck->li_seq);
1615         if (lfsck->li_seq == NULL)
1616                 RETURN(-ENOMEM);
1617
1618         OBD_ALLOC(prefix, MAX_OBD_NAME + 7);
1619         if (prefix == NULL)
1620                 GOTO(out, rc = -ENOMEM);
1621
1622         snprintf(prefix, MAX_OBD_NAME + 7, "lfsck-%s", lfsck_lfsck2name(lfsck));
1623         seq_client_init(lfsck->li_seq, NULL, LUSTRE_SEQ_METADATA, prefix,
1624                              ss->ss_server_seq);
1625         OBD_FREE(prefix, MAX_OBD_NAME + 7);
1626
1627         if (fid_is_sane(&bk->lb_last_fid))
1628                 lfsck->li_seq->lcs_fid = bk->lb_last_fid;
1629
1630         RETURN(0);
1631
1632 out:
1633         OBD_FREE_PTR(lfsck->li_seq);
1634         lfsck->li_seq = NULL;
1635
1636         return rc;
1637 }
1638
1639 static void lfsck_fid_fini(struct lfsck_instance *lfsck)
1640 {
1641         if (lfsck->li_seq != NULL) {
1642                 seq_client_fini(lfsck->li_seq);
1643                 OBD_FREE_PTR(lfsck->li_seq);
1644                 lfsck->li_seq = NULL;
1645         }
1646 }
1647
1648 void lfsck_instance_cleanup(const struct lu_env *env,
1649                             struct lfsck_instance *lfsck)
1650 {
1651         struct ptlrpc_thread    *thread = &lfsck->li_thread;
1652         struct lfsck_component  *com;
1653         struct lfsck_component  *next;
1654         struct lfsck_lmv_unit   *llu;
1655         struct lfsck_lmv_unit   *llu_next;
1656         struct lfsck_lmv        *llmv;
1657         ENTRY;
1658
1659         LASSERT(list_empty(&lfsck->li_link));
1660         LASSERT(thread_is_init(thread) || thread_is_stopped(thread));
1661
1662         if (lfsck->li_obj_oit != NULL) {
1663                 lfsck_object_put(env, lfsck->li_obj_oit);
1664                 lfsck->li_obj_oit = NULL;
1665         }
1666
1667         LASSERT(lfsck->li_obj_dir == NULL);
1668         LASSERT(lfsck->li_lmv == NULL);
1669
1670         list_for_each_entry_safe(llu, llu_next, &lfsck->li_list_lmv, llu_link) {
1671                 llmv = &llu->llu_lmv;
1672
1673                 LASSERTF(atomic_read(&llmv->ll_ref) == 1,
1674                          "still in using: %u\n",
1675                          atomic_read(&llmv->ll_ref));
1676
1677                 lfsck_lmv_put(env, llmv);
1678         }
1679
1680         list_for_each_entry_safe(com, next, &lfsck->li_list_scan, lc_link) {
1681                 lfsck_component_cleanup(env, com);
1682         }
1683
1684         LASSERT(list_empty(&lfsck->li_list_dir));
1685
1686         list_for_each_entry_safe(com, next, &lfsck->li_list_double_scan,
1687                                  lc_link) {
1688                 lfsck_component_cleanup(env, com);
1689         }
1690
1691         list_for_each_entry_safe(com, next, &lfsck->li_list_idle, lc_link) {
1692                 lfsck_component_cleanup(env, com);
1693         }
1694
1695         lfsck_tgt_descs_fini(&lfsck->li_ost_descs);
1696         lfsck_tgt_descs_fini(&lfsck->li_mdt_descs);
1697
1698         if (lfsck->li_lfsck_dir != NULL) {
1699                 lfsck_object_put(env, lfsck->li_lfsck_dir);
1700                 lfsck->li_lfsck_dir = NULL;
1701         }
1702
1703         if (lfsck->li_bookmark_obj != NULL) {
1704                 lfsck_object_put(env, lfsck->li_bookmark_obj);
1705                 lfsck->li_bookmark_obj = NULL;
1706         }
1707
1708         if (lfsck->li_lpf_obj != NULL) {
1709                 lfsck_object_put(env, lfsck->li_lpf_obj);
1710                 lfsck->li_lpf_obj = NULL;
1711         }
1712
1713         if (lfsck->li_lpf_root_obj != NULL) {
1714                 lfsck_object_put(env, lfsck->li_lpf_root_obj);
1715                 lfsck->li_lpf_root_obj = NULL;
1716         }
1717
1718         if (lfsck->li_los != NULL) {
1719                 local_oid_storage_fini(env, lfsck->li_los);
1720                 lfsck->li_los = NULL;
1721         }
1722
1723         lfsck_fid_fini(lfsck);
1724
1725         OBD_FREE_PTR(lfsck);
1726 }
1727
1728 static inline struct lfsck_instance *
1729 __lfsck_instance_find(struct dt_device *key, bool ref, bool unlink)
1730 {
1731         struct lfsck_instance *lfsck;
1732
1733         list_for_each_entry(lfsck, &lfsck_instance_list, li_link) {
1734                 if (lfsck->li_bottom == key) {
1735                         if (ref)
1736                                 lfsck_instance_get(lfsck);
1737                         if (unlink)
1738                                 list_del_init(&lfsck->li_link);
1739
1740                         return lfsck;
1741                 }
1742         }
1743
1744         return NULL;
1745 }
1746
1747 struct lfsck_instance *lfsck_instance_find(struct dt_device *key, bool ref,
1748                                            bool unlink)
1749 {
1750         struct lfsck_instance *lfsck;
1751
1752         spin_lock(&lfsck_instance_lock);
1753         lfsck = __lfsck_instance_find(key, ref, unlink);
1754         spin_unlock(&lfsck_instance_lock);
1755
1756         return lfsck;
1757 }
1758
1759 static inline int lfsck_instance_add(struct lfsck_instance *lfsck)
1760 {
1761         struct lfsck_instance *tmp;
1762
1763         spin_lock(&lfsck_instance_lock);
1764         list_for_each_entry(tmp, &lfsck_instance_list, li_link) {
1765                 if (lfsck->li_bottom == tmp->li_bottom) {
1766                         spin_unlock(&lfsck_instance_lock);
1767                         return -EEXIST;
1768                 }
1769         }
1770
1771         list_add_tail(&lfsck->li_link, &lfsck_instance_list);
1772         spin_unlock(&lfsck_instance_lock);
1773         return 0;
1774 }
1775
1776 void lfsck_bits_dump(struct seq_file *m, int bits, const char *const names[],
1777                      const char *prefix)
1778 {
1779         int flag;
1780         int i;
1781         bool newline = (bits != 0 ? false : true);
1782
1783         seq_printf(m, "%s:%c", prefix, bits != 0 ? ' ' : '\n');
1784
1785         for (i = 0, flag = 1; bits != 0; i++, flag = BIT(i)) {
1786                 if (flag & bits) {
1787                         bits &= ~flag;
1788                         if (names[i] != NULL) {
1789                                 if (bits == 0)
1790                                         newline = true;
1791
1792                                 seq_printf(m, "%s%c", names[i],
1793                                            newline ? '\n' : ',');
1794                         }
1795                 }
1796         }
1797
1798         if (!newline)
1799                 seq_putc(m, '\n');
1800 }
1801
1802 void lfsck_time_dump(struct seq_file *m, time64_t time, const char *name)
1803 {
1804         if (time == 0) {
1805                 seq_printf(m, "%s_time: N/A\n", name);
1806                 seq_printf(m, "time_since_%s: N/A\n", name);
1807         } else {
1808                 seq_printf(m, "%s_time: %lld\n", name, time);
1809                 seq_printf(m, "time_since_%s: %lld seconds\n",
1810                            name, ktime_get_real_seconds() - time);
1811         }
1812 }
1813
1814 void lfsck_pos_dump(struct seq_file *m, struct lfsck_position *pos,
1815                     const char *prefix)
1816 {
1817         if (fid_is_zero(&pos->lp_dir_parent)) {
1818                 if (pos->lp_oit_cookie == 0) {
1819                         seq_printf(m, "%s: N/A, N/A, N/A\n", prefix);
1820                         return;
1821                 }
1822                 seq_printf(m, "%s: %llu, N/A, N/A\n",
1823                            prefix, pos->lp_oit_cookie);
1824         } else {
1825                 seq_printf(m, "%s: %llu, "DFID", %#llx\n",
1826                            prefix, pos->lp_oit_cookie,
1827                            PFID(&pos->lp_dir_parent), pos->lp_dir_cookie);
1828         }
1829 }
1830
1831 void lfsck_pos_fill(const struct lu_env *env, struct lfsck_instance *lfsck,
1832                     struct lfsck_position *pos, bool init)
1833 {
1834         const struct dt_it_ops *iops = &lfsck->li_obj_oit->do_index_ops->dio_it;
1835
1836         if (unlikely(lfsck->li_di_oit == NULL)) {
1837                 memset(pos, 0, sizeof(*pos));
1838                 return;
1839         }
1840
1841         pos->lp_oit_cookie = iops->store(env, lfsck->li_di_oit);
1842         if (!lfsck->li_current_oit_processed && !init)
1843                 pos->lp_oit_cookie--;
1844
1845         if (unlikely(pos->lp_oit_cookie == 0))
1846                 pos->lp_oit_cookie = 1;
1847
1848         spin_lock(&lfsck->li_lock);
1849         if (lfsck->li_di_dir != NULL) {
1850                 struct dt_object *dto = lfsck->li_obj_dir;
1851
1852                 pos->lp_dir_cookie = dto->do_index_ops->dio_it.store(env,
1853                                                         lfsck->li_di_dir);
1854
1855                 if (pos->lp_dir_cookie >= MDS_DIR_END_OFF) {
1856                         fid_zero(&pos->lp_dir_parent);
1857                         pos->lp_dir_cookie = 0;
1858                 } else {
1859                         pos->lp_dir_parent = *lfsck_dto2fid(dto);
1860                 }
1861         } else {
1862                 fid_zero(&pos->lp_dir_parent);
1863                 pos->lp_dir_cookie = 0;
1864         }
1865         spin_unlock(&lfsck->li_lock);
1866 }
1867
1868 bool __lfsck_set_speed(struct lfsck_instance *lfsck, __u32 limit)
1869 {
1870         bool dirty = false;
1871
1872         if (limit != LFSCK_SPEED_NO_LIMIT) {
1873                 if (limit > cfs_time_seconds(1)) {
1874                         lfsck->li_sleep_rate = limit / cfs_time_seconds(1);
1875                         lfsck->li_sleep_jif = 1;
1876                 } else {
1877                         lfsck->li_sleep_rate = 1;
1878                         lfsck->li_sleep_jif = cfs_time_seconds(1) / limit;
1879                 }
1880         } else {
1881                 lfsck->li_sleep_jif = 0;
1882                 lfsck->li_sleep_rate = 0;
1883         }
1884
1885         if (lfsck->li_bookmark_ram.lb_speed_limit != limit) {
1886                 lfsck->li_bookmark_ram.lb_speed_limit = limit;
1887                 dirty = true;
1888         }
1889
1890         return dirty;
1891 }
1892
1893 void lfsck_control_speed(struct lfsck_instance *lfsck)
1894 {
1895         struct ptlrpc_thread *thread = &lfsck->li_thread;
1896
1897         if (lfsck->li_sleep_jif > 0 &&
1898             lfsck->li_new_scanned >= lfsck->li_sleep_rate) {
1899                 wait_event_idle_timeout(thread->t_ctl_waitq,
1900                                         !thread_is_running(thread),
1901                                         lfsck->li_sleep_jif);
1902                 lfsck->li_new_scanned = 0;
1903         }
1904 }
1905
1906 void lfsck_control_speed_by_self(struct lfsck_component *com)
1907 {
1908         struct lfsck_instance   *lfsck  = com->lc_lfsck;
1909         struct ptlrpc_thread    *thread = &lfsck->li_thread;
1910
1911         if (lfsck->li_sleep_jif > 0 &&
1912             com->lc_new_scanned >= lfsck->li_sleep_rate) {
1913                 wait_event_idle_timeout(thread->t_ctl_waitq,
1914                                         !thread_is_running(thread),
1915                                         lfsck->li_sleep_jif);
1916                 com->lc_new_scanned = 0;
1917         }
1918 }
1919
1920 static struct lfsck_thread_args *
1921 lfsck_thread_args_init(struct lfsck_instance *lfsck,
1922                        struct lfsck_component *com,
1923                        struct lfsck_start_param *lsp)
1924 {
1925         struct lfsck_thread_args *lta;
1926         int                       rc;
1927
1928         OBD_ALLOC_PTR(lta);
1929         if (lta == NULL)
1930                 return ERR_PTR(-ENOMEM);
1931
1932         rc = lu_env_init(&lta->lta_env, LCT_MD_THREAD | LCT_DT_THREAD);
1933         if (rc != 0) {
1934                 OBD_FREE_PTR(lta);
1935                 return ERR_PTR(rc);
1936         }
1937
1938         lta->lta_lfsck = lfsck_instance_get(lfsck);
1939         if (com != NULL)
1940                 lta->lta_com = lfsck_component_get(com);
1941
1942         lta->lta_lsp = lsp;
1943
1944         return lta;
1945 }
1946
1947 void lfsck_thread_args_fini(struct lfsck_thread_args *lta)
1948 {
1949         if (lta->lta_com != NULL)
1950                 lfsck_component_put(&lta->lta_env, lta->lta_com);
1951         lfsck_instance_put(&lta->lta_env, lta->lta_lfsck);
1952         lu_env_fini(&lta->lta_env);
1953         OBD_FREE_PTR(lta);
1954 }
1955
1956 struct lfsck_assistant_data *
1957 lfsck_assistant_data_init(const struct lfsck_assistant_operations *lao,
1958                           const char *name)
1959 {
1960         struct lfsck_assistant_data *lad;
1961
1962         OBD_ALLOC_PTR(lad);
1963         if (lad != NULL) {
1964                 lad->lad_bitmap = CFS_ALLOCATE_BITMAP(BITS_PER_LONG);
1965                 if (lad->lad_bitmap == NULL) {
1966                         OBD_FREE_PTR(lad);
1967                         return NULL;
1968                 }
1969
1970                 INIT_LIST_HEAD(&lad->lad_req_list);
1971                 spin_lock_init(&lad->lad_lock);
1972                 INIT_LIST_HEAD(&lad->lad_ost_list);
1973                 INIT_LIST_HEAD(&lad->lad_ost_phase1_list);
1974                 INIT_LIST_HEAD(&lad->lad_ost_phase2_list);
1975                 INIT_LIST_HEAD(&lad->lad_mdt_list);
1976                 INIT_LIST_HEAD(&lad->lad_mdt_phase1_list);
1977                 INIT_LIST_HEAD(&lad->lad_mdt_phase2_list);
1978                 init_waitqueue_head(&lad->lad_thread.t_ctl_waitq);
1979                 lad->lad_ops = lao;
1980                 lad->lad_name = name;
1981         }
1982
1983         return lad;
1984 }
1985
1986 struct lfsck_assistant_object *
1987 lfsck_assistant_object_init(const struct lu_env *env, const struct lu_fid *fid,
1988                             const struct lu_attr *attr, __u64 cookie,
1989                             bool is_dir)
1990 {
1991         struct lfsck_assistant_object   *lso;
1992
1993         OBD_ALLOC_PTR(lso);
1994         if (lso == NULL)
1995                 return ERR_PTR(-ENOMEM);
1996
1997         lso->lso_fid = *fid;
1998         if (attr != NULL)
1999                 lso->lso_attr = *attr;
2000
2001         atomic_set(&lso->lso_ref, 1);
2002         lso->lso_oit_cookie = cookie;
2003         if (is_dir)
2004                 lso->lso_is_dir = 1;
2005
2006         return lso;
2007 }
2008
2009 struct dt_object *
2010 lfsck_assistant_object_load(const struct lu_env *env,
2011                             struct lfsck_instance *lfsck,
2012                             struct lfsck_assistant_object *lso)
2013 {
2014         struct dt_object *obj;
2015
2016         obj = lfsck_object_find_bottom(env, lfsck, &lso->lso_fid);
2017         if (IS_ERR(obj))
2018                 return obj;
2019
2020         if (unlikely(!dt_object_exists(obj) || lfsck_is_dead_obj(obj))) {
2021                 lso->lso_dead = 1;
2022                 lfsck_object_put(env, obj);
2023
2024                 return ERR_PTR(-ENOENT);
2025         }
2026
2027         if (lso->lso_is_dir && unlikely(!dt_try_as_dir(env, obj))) {
2028                 lfsck_object_put(env, obj);
2029
2030                 return ERR_PTR(-ENOTDIR);
2031         }
2032
2033         return obj;
2034 }
2035
2036 /**
2037  * Generic LFSCK asynchronous communication interpretor function.
2038  * The LFSCK RPC reply for both the event notification and status
2039  * querying will be handled here.
2040  *
2041  * \param[in] env       pointer to the thread context
2042  * \param[in] req       pointer to the LFSCK request
2043  * \param[in] args      pointer to the lfsck_async_interpret_args
2044  * \param[in] rc        the result for handling the LFSCK request
2045  *
2046  * \retval              0 for success
2047  * \retval              negative error number on failure
2048  */
2049 int lfsck_async_interpret_common(const struct lu_env *env,
2050                                  struct ptlrpc_request *req,
2051                                  void *args, int rc)
2052 {
2053         struct lfsck_async_interpret_args *laia = args;
2054         struct lfsck_component            *com  = laia->laia_com;
2055         struct lfsck_assistant_data       *lad  = com->lc_data;
2056         struct lfsck_tgt_descs            *ltds = laia->laia_ltds;
2057         struct lfsck_tgt_desc             *ltd  = laia->laia_ltd;
2058         struct lfsck_request              *lr   = laia->laia_lr;
2059
2060         LASSERT(com->lc_lfsck->li_master);
2061
2062         switch (lr->lr_event) {
2063         case LE_START:
2064                 if (unlikely(rc == -EINPROGRESS)) {
2065                         ltd->ltd_retry_start = 1;
2066                         break;
2067                 }
2068
2069                 if (rc != 0) {
2070                         CDEBUG(D_LFSCK, "%s: fail to notify %s %x for %s "
2071                                "start: rc = %d\n",
2072                                lfsck_lfsck2name(com->lc_lfsck),
2073                                (lr->lr_flags & LEF_TO_OST) ? "OST" : "MDT",
2074                                ltd->ltd_index, lad->lad_name, rc);
2075
2076                         if (com->lc_type == LFSCK_TYPE_LAYOUT) {
2077                                 struct lfsck_layout *lo = com->lc_file_ram;
2078
2079                                 if (lr->lr_flags & LEF_TO_OST)
2080                                         lfsck_lad_set_bitmap(env, com,
2081                                                              ltd->ltd_index);
2082                                 else
2083                                         lo->ll_flags |= LF_INCOMPLETE;
2084                         } else {
2085                                 struct lfsck_namespace *ns = com->lc_file_ram;
2086
2087                                 /* If some MDT does not join the namespace
2088                                  * LFSCK, then we cannot know whether there
2089                                  * is some name entry on such MDT that with
2090                                  * the referenced MDT-object on this MDT or
2091                                  * not. So the namespace LFSCK on this MDT
2092                                  * cannot handle orphan MDT-objects properly.
2093                                  * So we mark the LFSCK as LF_INCOMPLETE and
2094                                  * skip orphan MDT-objects handling. */
2095                                 ns->ln_flags |= LF_INCOMPLETE;
2096                         }
2097                         break;
2098                 }
2099
2100                 spin_lock(&ltds->ltd_lock);
2101                 if (ltd->ltd_dead) {
2102                         spin_unlock(&ltds->ltd_lock);
2103                         break;
2104                 }
2105
2106                 if (com->lc_type == LFSCK_TYPE_LAYOUT) {
2107                         struct list_head *list;
2108                         struct list_head *phase_list;
2109
2110                         if (ltd->ltd_layout_done) {
2111                                 spin_unlock(&ltds->ltd_lock);
2112                                 break;
2113                         }
2114
2115                         if (lr->lr_flags & LEF_TO_OST) {
2116                                 list = &lad->lad_ost_list;
2117                                 phase_list = &lad->lad_ost_phase1_list;
2118                         } else {
2119                                 list = &lad->lad_mdt_list;
2120                                 phase_list = &lad->lad_mdt_phase1_list;
2121                         }
2122
2123                         if (list_empty(&ltd->ltd_layout_list))
2124                                 list_add_tail(&ltd->ltd_layout_list, list);
2125                         if (list_empty(&ltd->ltd_layout_phase_list))
2126                                 list_add_tail(&ltd->ltd_layout_phase_list,
2127                                               phase_list);
2128                 } else {
2129                         if (ltd->ltd_namespace_done) {
2130                                 spin_unlock(&ltds->ltd_lock);
2131                                 break;
2132                         }
2133
2134                         if (list_empty(&ltd->ltd_namespace_list))
2135                                 list_add_tail(&ltd->ltd_namespace_list,
2136                                               &lad->lad_mdt_list);
2137                         if (list_empty(&ltd->ltd_namespace_phase_list))
2138                                 list_add_tail(&ltd->ltd_namespace_phase_list,
2139                                               &lad->lad_mdt_phase1_list);
2140                 }
2141                 spin_unlock(&ltds->ltd_lock);
2142                 break;
2143         case LE_STOP:
2144         case LE_PHASE1_DONE:
2145         case LE_PHASE2_DONE:
2146         case LE_PEER_EXIT:
2147                 if (rc != 0 && rc != -EALREADY)
2148                         CDEBUG(D_LFSCK, "%s: fail to notify %s %x for %s: "
2149                               "event = %d, rc = %d\n",
2150                               lfsck_lfsck2name(com->lc_lfsck),
2151                               (lr->lr_flags & LEF_TO_OST) ? "OST" : "MDT",
2152                               ltd->ltd_index, lad->lad_name, lr->lr_event, rc);
2153                 break;
2154         case LE_QUERY: {
2155                 struct lfsck_reply *reply;
2156                 struct list_head *list;
2157                 struct list_head *phase_list;
2158
2159                 if (com->lc_type == LFSCK_TYPE_LAYOUT) {
2160                         list = &ltd->ltd_layout_list;
2161                         phase_list = &ltd->ltd_layout_phase_list;
2162                 } else {
2163                         list = &ltd->ltd_namespace_list;
2164                         phase_list = &ltd->ltd_namespace_phase_list;
2165                 }
2166
2167                 if (rc != 0) {
2168                         if (lr->lr_flags & LEF_QUERY_ALL) {
2169                                 lfsck_reset_ltd_status(ltd, com->lc_type);
2170                                 break;
2171                         }
2172
2173                         spin_lock(&ltds->ltd_lock);
2174                         list_del_init(phase_list);
2175                         list_del_init(list);
2176                         spin_unlock(&ltds->ltd_lock);
2177                         break;
2178                 }
2179
2180                 reply = req_capsule_server_get(&req->rq_pill,
2181                                                &RMF_LFSCK_REPLY);
2182                 if (reply == NULL) {
2183                         rc = -EPROTO;
2184                         CDEBUG(D_LFSCK, "%s: invalid query reply for %s: "
2185                                "rc = %d\n", lfsck_lfsck2name(com->lc_lfsck),
2186                                lad->lad_name, rc);
2187
2188                         if (lr->lr_flags & LEF_QUERY_ALL) {
2189                                 lfsck_reset_ltd_status(ltd, com->lc_type);
2190                                 break;
2191                         }
2192
2193                         spin_lock(&ltds->ltd_lock);
2194                         list_del_init(phase_list);
2195                         list_del_init(list);
2196                         spin_unlock(&ltds->ltd_lock);
2197                         break;
2198                 }
2199
2200                 if (lr->lr_flags & LEF_QUERY_ALL) {
2201                         if (com->lc_type == LFSCK_TYPE_LAYOUT) {
2202                                 ltd->ltd_layout_status = reply->lr_status;
2203                                 ltd->ltd_layout_repaired = reply->lr_repaired;
2204                         } else {
2205                                 ltd->ltd_namespace_status = reply->lr_status;
2206                                 ltd->ltd_namespace_repaired =
2207                                                         reply->lr_repaired;
2208                         }
2209                         break;
2210                 }
2211
2212                 switch (reply->lr_status) {
2213                 case LS_SCANNING_PHASE1:
2214                         break;
2215                 case LS_SCANNING_PHASE2:
2216                         spin_lock(&ltds->ltd_lock);
2217                         list_del_init(phase_list);
2218                         if (ltd->ltd_dead) {
2219                                 spin_unlock(&ltds->ltd_lock);
2220                                 break;
2221                         }
2222
2223                         if (com->lc_type == LFSCK_TYPE_LAYOUT) {
2224                                 if (ltd->ltd_layout_done) {
2225                                         spin_unlock(&ltds->ltd_lock);
2226                                         break;
2227                                 }
2228
2229                                 if (lr->lr_flags & LEF_TO_OST)
2230                                         list_add_tail(phase_list,
2231                                                 &lad->lad_ost_phase2_list);
2232                                 else
2233                                         list_add_tail(phase_list,
2234                                                 &lad->lad_mdt_phase2_list);
2235                         } else {
2236                                 if (ltd->ltd_namespace_done) {
2237                                         spin_unlock(&ltds->ltd_lock);
2238                                         break;
2239                                 }
2240
2241                                 list_add_tail(phase_list,
2242                                               &lad->lad_mdt_phase2_list);
2243                         }
2244                         spin_unlock(&ltds->ltd_lock);
2245                         break;
2246                 default:
2247                         spin_lock(&ltds->ltd_lock);
2248                         list_del_init(phase_list);
2249                         list_del_init(list);
2250                         spin_unlock(&ltds->ltd_lock);
2251                         break;
2252                 }
2253                 break;
2254         }
2255         default:
2256                 CDEBUG(D_LFSCK, "%s: unexpected event: rc = %d\n",
2257                        lfsck_lfsck2name(com->lc_lfsck), lr->lr_event);
2258                 break;
2259         }
2260
2261         if (!laia->laia_shared) {
2262                 lfsck_tgt_put(ltd);
2263                 lfsck_component_put(env, com);
2264         }
2265
2266         return 0;
2267 }
2268
2269 static void lfsck_interpret(const struct lu_env *env,
2270                             struct lfsck_instance *lfsck,
2271                             struct ptlrpc_request *req, void *args, int result)
2272 {
2273         struct lfsck_async_interpret_args *laia = args;
2274         struct lfsck_component            *com;
2275
2276         LASSERT(laia->laia_com == NULL);
2277         LASSERT(laia->laia_shared);
2278
2279         spin_lock(&lfsck->li_lock);
2280         list_for_each_entry(com, &lfsck->li_list_scan, lc_link) {
2281                 laia->laia_com = com;
2282                 lfsck_async_interpret_common(env, req, laia, result);
2283         }
2284
2285         list_for_each_entry(com, &lfsck->li_list_double_scan, lc_link) {
2286                 laia->laia_com = com;
2287                 lfsck_async_interpret_common(env, req, laia, result);
2288         }
2289         spin_unlock(&lfsck->li_lock);
2290 }
2291
2292 static int lfsck_stop_notify(const struct lu_env *env,
2293                              struct lfsck_instance *lfsck,
2294                              struct lfsck_tgt_descs *ltds,
2295                              struct lfsck_tgt_desc *ltd, __u16 type)
2296 {
2297         struct lfsck_component *com;
2298         int                     rc = 0;
2299         ENTRY;
2300
2301         LASSERT(lfsck->li_master);
2302
2303         spin_lock(&lfsck->li_lock);
2304         com = __lfsck_component_find(lfsck, type, &lfsck->li_list_scan);
2305         if (com == NULL)
2306                 com = __lfsck_component_find(lfsck, type,
2307                                              &lfsck->li_list_double_scan);
2308         if (com != NULL)
2309                 lfsck_component_get(com);
2310         spin_unlock(&lfsck->li_lock);
2311
2312         if (com != NULL) {
2313                 struct lfsck_thread_info          *info  = lfsck_env_info(env);
2314                 struct lfsck_async_interpret_args *laia  = &info->lti_laia;
2315                 struct lfsck_request              *lr    = &info->lti_lr;
2316                 struct lfsck_assistant_data       *lad   = com->lc_data;
2317                 struct list_head                  *list;
2318                 struct list_head                  *phase_list;
2319                 struct ptlrpc_request_set         *set;
2320
2321                 set = ptlrpc_prep_set();
2322                 if (set == NULL) {
2323                         lfsck_component_put(env, com);
2324
2325                         RETURN(-ENOMEM);
2326                 }
2327
2328                 if (type == LFSCK_TYPE_LAYOUT) {
2329                         list = &ltd->ltd_layout_list;
2330                         phase_list = &ltd->ltd_layout_phase_list;
2331                 } else {
2332                         list = &ltd->ltd_namespace_list;
2333                         phase_list = &ltd->ltd_namespace_phase_list;
2334                 }
2335
2336                 spin_lock(&ltds->ltd_lock);
2337                 if (list_empty(list)) {
2338                         LASSERT(list_empty(phase_list));
2339                         spin_unlock(&ltds->ltd_lock);
2340                         ptlrpc_set_destroy(set);
2341
2342                         RETURN(0);
2343                 }
2344
2345                 list_del_init(phase_list);
2346                 list_del_init(list);
2347                 spin_unlock(&ltds->ltd_lock);
2348
2349                 memset(lr, 0, sizeof(*lr));
2350                 lr->lr_index = lfsck_dev_idx(lfsck);
2351                 lr->lr_event = LE_PEER_EXIT;
2352                 lr->lr_active = type;
2353                 lr->lr_status = LS_CO_PAUSED;
2354                 if (ltds == &lfsck->li_ost_descs)
2355                         lr->lr_flags = LEF_TO_OST;
2356
2357                 memset(laia, 0, sizeof(*laia));
2358                 laia->laia_com = com;
2359                 laia->laia_ltds = ltds;
2360                 atomic_inc(&ltd->ltd_ref);
2361                 laia->laia_ltd = ltd;
2362                 laia->laia_lr = lr;
2363
2364                 rc = lfsck_async_request(env, ltd->ltd_exp, lr, set,
2365                                          lfsck_async_interpret_common,
2366                                          laia, LFSCK_NOTIFY);
2367                 if (rc != 0) {
2368                         CDEBUG(D_LFSCK, "%s: fail to notify %s %x for "
2369                                "co-stop for %s: rc = %d\n",
2370                                lfsck_lfsck2name(lfsck),
2371                                (lr->lr_flags & LEF_TO_OST) ? "OST" : "MDT",
2372                                ltd->ltd_index, lad->lad_name, rc);
2373                         lfsck_tgt_put(ltd);
2374                 } else {
2375                         rc = ptlrpc_set_wait(env, set);
2376                 }
2377
2378                 ptlrpc_set_destroy(set);
2379                 lfsck_component_put(env, com);
2380         }
2381
2382         RETURN(rc);
2383 }
2384
2385 static int lfsck_async_interpret(const struct lu_env *env,
2386                                  struct ptlrpc_request *req,
2387                                  void *args, int rc)
2388 {
2389         struct lfsck_async_interpret_args *laia = args;
2390         struct lfsck_instance             *lfsck;
2391
2392         lfsck = container_of(laia->laia_ltds, struct lfsck_instance,
2393                              li_mdt_descs);
2394         lfsck_interpret(env, lfsck, req, laia, rc);
2395         lfsck_tgt_put(laia->laia_ltd);
2396         if (rc != 0 && laia->laia_result != -EALREADY)
2397                 laia->laia_result = rc;
2398
2399         return 0;
2400 }
2401
2402 int lfsck_async_request(const struct lu_env *env, struct obd_export *exp,
2403                         struct lfsck_request *lr,
2404                         struct ptlrpc_request_set *set,
2405                         ptlrpc_interpterer_t interpreter,
2406                         void *args, int request)
2407 {
2408         struct lfsck_async_interpret_args *laia;
2409         struct ptlrpc_request             *req;
2410         struct lfsck_request              *tmp;
2411         struct req_format                 *format;
2412         int                                rc;
2413
2414         switch (request) {
2415         case LFSCK_NOTIFY:
2416                 format = &RQF_LFSCK_NOTIFY;
2417                 break;
2418         case LFSCK_QUERY:
2419                 format = &RQF_LFSCK_QUERY;
2420                 break;
2421         default:
2422                 CDEBUG(D_LFSCK, "%s: unknown async request %d: rc = %d\n",
2423                        exp->exp_obd->obd_name, request, -EINVAL);
2424                 return -EINVAL;
2425         }
2426
2427         req = ptlrpc_request_alloc(class_exp2cliimp(exp), format);
2428         if (req == NULL)
2429                 return -ENOMEM;
2430
2431         rc = ptlrpc_request_pack(req, LUSTRE_OBD_VERSION, request);
2432         if (rc != 0) {
2433                 ptlrpc_request_free(req);
2434
2435                 return rc;
2436         }
2437
2438         tmp = req_capsule_client_get(&req->rq_pill, &RMF_LFSCK_REQUEST);
2439         *tmp = *lr;
2440         ptlrpc_request_set_replen(req);
2441
2442         laia = ptlrpc_req_async_args(laia, req);
2443         *laia = *(struct lfsck_async_interpret_args *)args;
2444         if (laia->laia_com != NULL)
2445                 lfsck_component_get(laia->laia_com);
2446         req->rq_interpret_reply = interpreter;
2447         req->rq_allow_intr = 1;
2448         req->rq_no_delay = 1;
2449         ptlrpc_set_add_req(set, req);
2450
2451         return 0;
2452 }
2453
2454 int lfsck_query_all(const struct lu_env *env, struct lfsck_component *com)
2455 {
2456         struct lfsck_thread_info          *info  = lfsck_env_info(env);
2457         struct lfsck_request              *lr    = &info->lti_lr;
2458         struct lfsck_async_interpret_args *laia  = &info->lti_laia;
2459         struct lfsck_instance             *lfsck = com->lc_lfsck;
2460         struct lfsck_tgt_descs            *ltds  = &lfsck->li_mdt_descs;
2461         struct lfsck_tgt_desc             *ltd;
2462         struct ptlrpc_request_set         *set;
2463         int                                idx;
2464         int                                rc;
2465         ENTRY;
2466
2467         memset(lr, 0, sizeof(*lr));
2468         lr->lr_event = LE_QUERY;
2469         lr->lr_active = com->lc_type;
2470         lr->lr_flags = LEF_QUERY_ALL;
2471
2472         memset(laia, 0, sizeof(*laia));
2473         laia->laia_com = com;
2474         laia->laia_lr = lr;
2475
2476         set = ptlrpc_prep_set();
2477         if (set == NULL)
2478                 RETURN(-ENOMEM);
2479
2480 again:
2481         laia->laia_ltds = ltds;
2482         down_read(&ltds->ltd_rw_sem);
2483         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
2484                 ltd = lfsck_tgt_get(ltds, idx);
2485                 LASSERT(ltd != NULL);
2486
2487                 laia->laia_ltd = ltd;
2488                 up_read(&ltds->ltd_rw_sem);
2489                 rc = lfsck_async_request(env, ltd->ltd_exp, lr, set,
2490                                          lfsck_async_interpret_common,
2491                                          laia, LFSCK_QUERY);
2492                 if (rc != 0) {
2493                         struct lfsck_assistant_data *lad = com->lc_data;
2494
2495                         CDEBUG(D_LFSCK, "%s: Fail to query %s %x for stat %s: "
2496                                "rc = %d\n", lfsck_lfsck2name(lfsck),
2497                                (lr->lr_flags & LEF_TO_OST) ? "OST" : "MDT",
2498                                ltd->ltd_index, lad->lad_name, rc);
2499                         lfsck_reset_ltd_status(ltd, com->lc_type);
2500                         lfsck_tgt_put(ltd);
2501                 }
2502                 down_read(&ltds->ltd_rw_sem);
2503         }
2504         up_read(&ltds->ltd_rw_sem);
2505
2506         if (com->lc_type == LFSCK_TYPE_LAYOUT && !(lr->lr_flags & LEF_TO_OST)) {
2507                 ltds = &lfsck->li_ost_descs;
2508                 lr->lr_flags |= LEF_TO_OST;
2509                 goto again;
2510         }
2511
2512         rc = ptlrpc_set_wait(env, set);
2513         ptlrpc_set_destroy(set);
2514
2515         RETURN(rc);
2516 }
2517
2518 int lfsck_start_assistant(const struct lu_env *env, struct lfsck_component *com,
2519                           struct lfsck_start_param *lsp)
2520 {
2521         struct lfsck_instance           *lfsck   = com->lc_lfsck;
2522         struct lfsck_assistant_data     *lad     = com->lc_data;
2523         struct ptlrpc_thread            *mthread = &lfsck->li_thread;
2524         struct ptlrpc_thread            *athread = &lad->lad_thread;
2525         struct lfsck_thread_args        *lta;
2526         struct task_struct              *task;
2527         int                              rc;
2528         ENTRY;
2529
2530         lad->lad_assistant_status = 0;
2531         lad->lad_post_result = 0;
2532         lad->lad_flags = 0;
2533         lad->lad_advance_lock = false;
2534         thread_set_flags(athread, 0);
2535
2536         lta = lfsck_thread_args_init(lfsck, com, lsp);
2537         if (IS_ERR(lta))
2538                 RETURN(PTR_ERR(lta));
2539
2540         task = kthread_run(lfsck_assistant_engine, lta, lad->lad_name);
2541         if (IS_ERR(task)) {
2542                 rc = PTR_ERR(task);
2543                 CERROR("%s: cannot start LFSCK assistant thread for %s: "
2544                        "rc = %d\n", lfsck_lfsck2name(lfsck), lad->lad_name, rc);
2545                 lfsck_thread_args_fini(lta);
2546         } else {
2547                 wait_event_idle(mthread->t_ctl_waitq,
2548                                 thread_is_running(athread) ||
2549                                 thread_is_stopped(athread) ||
2550                                 !thread_is_starting(mthread));
2551                 if (unlikely(!thread_is_starting(mthread)))
2552                         /* stopped by race */
2553                         rc = -ESRCH;
2554                 else if (unlikely(!thread_is_running(athread)))
2555                         rc = lad->lad_assistant_status;
2556                 else
2557                         rc = 0;
2558         }
2559
2560         RETURN(rc);
2561 }
2562
2563 int lfsck_checkpoint_generic(const struct lu_env *env,
2564                              struct lfsck_component *com)
2565 {
2566         struct lfsck_assistant_data     *lad     = com->lc_data;
2567         struct ptlrpc_thread            *mthread = &com->lc_lfsck->li_thread;
2568         struct ptlrpc_thread            *athread = &lad->lad_thread;
2569
2570         wait_event_idle(mthread->t_ctl_waitq,
2571                         list_empty(&lad->lad_req_list) ||
2572                         !thread_is_running(mthread) ||
2573                         thread_is_stopped(athread));
2574
2575         if (!thread_is_running(mthread) || thread_is_stopped(athread))
2576                 return LFSCK_CHECKPOINT_SKIP;
2577
2578         return 0;
2579 }
2580
2581 void lfsck_post_generic(const struct lu_env *env,
2582                         struct lfsck_component *com, int *result)
2583 {
2584         struct lfsck_assistant_data     *lad     = com->lc_data;
2585         struct ptlrpc_thread            *athread = &lad->lad_thread;
2586         struct ptlrpc_thread            *mthread = &com->lc_lfsck->li_thread;
2587
2588         lad->lad_post_result = *result;
2589         if (*result <= 0)
2590                 set_bit(LAD_EXIT, &lad->lad_flags);
2591         set_bit(LAD_TO_POST, &lad->lad_flags);
2592
2593         CDEBUG(D_LFSCK, "%s: waiting for assistant to do %s post, rc = %d\n",
2594                lfsck_lfsck2name(com->lc_lfsck), lad->lad_name, *result);
2595
2596         wake_up(&athread->t_ctl_waitq);
2597         wait_event_idle(mthread->t_ctl_waitq,
2598                         (*result > 0 && list_empty(&lad->lad_req_list)) ||
2599                         thread_is_stopped(athread));
2600
2601         if (lad->lad_assistant_status < 0)
2602                 *result = lad->lad_assistant_status;
2603
2604         CDEBUG(D_LFSCK, "%s: the assistant has done %s post, rc = %d\n",
2605                lfsck_lfsck2name(com->lc_lfsck), lad->lad_name, *result);
2606 }
2607
2608 int lfsck_double_scan_generic(const struct lu_env *env,
2609                               struct lfsck_component *com, int status)
2610 {
2611         struct lfsck_assistant_data     *lad     = com->lc_data;
2612         struct ptlrpc_thread            *mthread = &com->lc_lfsck->li_thread;
2613         struct ptlrpc_thread            *athread = &lad->lad_thread;
2614
2615         if (status != LS_SCANNING_PHASE2)
2616                 set_bit(LAD_EXIT, &lad->lad_flags);
2617         else
2618                 set_bit(LAD_TO_DOUBLE_SCAN, &lad->lad_flags);
2619
2620         CDEBUG(D_LFSCK, "%s: waiting for assistant to do %s double_scan, "
2621                "status %d\n",
2622                lfsck_lfsck2name(com->lc_lfsck), lad->lad_name, status);
2623
2624         wake_up(&athread->t_ctl_waitq);
2625         wait_event_idle(mthread->t_ctl_waitq,
2626                         test_bit(LAD_IN_DOUBLE_SCAN, &lad->lad_flags) ||
2627                         thread_is_stopped(athread));
2628
2629         CDEBUG(D_LFSCK, "%s: the assistant has done %s double_scan, "
2630                "status %d\n", lfsck_lfsck2name(com->lc_lfsck), lad->lad_name,
2631                lad->lad_assistant_status);
2632
2633         if (lad->lad_assistant_status < 0)
2634                 return lad->lad_assistant_status;
2635
2636         return 0;
2637 }
2638
2639 void lfsck_quit_generic(const struct lu_env *env,
2640                         struct lfsck_component *com)
2641 {
2642         struct lfsck_assistant_data     *lad     = com->lc_data;
2643         struct ptlrpc_thread            *mthread = &com->lc_lfsck->li_thread;
2644         struct ptlrpc_thread            *athread = &lad->lad_thread;
2645
2646         set_bit(LAD_EXIT, &lad->lad_flags);
2647         wake_up(&athread->t_ctl_waitq);
2648         wait_event_idle(mthread->t_ctl_waitq,
2649                         thread_is_init(athread) ||
2650                         thread_is_stopped(athread));
2651 }
2652
2653 int lfsck_load_one_trace_file(const struct lu_env *env,
2654                               struct lfsck_component *com,
2655                               struct dt_object *parent,
2656                               struct dt_object **child,
2657                               const struct dt_index_features *ft,
2658                               const char *name, bool reset)
2659 {
2660         struct lfsck_instance *lfsck = com->lc_lfsck;
2661         struct dt_object *obj;
2662         int rc;
2663         ENTRY;
2664
2665         if (*child != NULL) {
2666                 struct dt_it *it;
2667                 const struct dt_it_ops *iops;
2668                 struct lu_fid *fid = &lfsck_env_info(env)->lti_fid3;
2669
2670                 if (!reset)
2671                         RETURN(0);
2672
2673                 obj = *child;
2674                 rc = obj->do_ops->do_index_try(env, obj, ft);
2675                 if (rc)
2676                         /* unlink by force */
2677                         goto unlink;
2678
2679                 iops = &obj->do_index_ops->dio_it;
2680                 it = iops->init(env, obj, 0);
2681                 if (IS_ERR(it))
2682                         /* unlink by force */
2683                         goto unlink;
2684
2685                 fid_zero(fid);
2686                 rc = iops->get(env, it, (const struct dt_key *)fid);
2687                 if (rc >= 0) {
2688                         rc = iops->next(env, it);
2689                         iops->put(env, it);
2690                 }
2691                 iops->fini(env, it);
2692                 if (rc > 0)
2693                         /* "rc > 0" means the index file is empty. */
2694                         RETURN(0);
2695
2696 unlink:
2697                 /* The old index is not empty, remove it firstly. */
2698                 rc = local_object_unlink(env, lfsck->li_bottom, parent, name);
2699                 CDEBUG_LIMIT(rc ? D_ERROR : D_LFSCK,
2700                              "%s: unlink lfsck sub trace file %s: rc = %d\n",
2701                              lfsck_lfsck2name(com->lc_lfsck), name, rc);
2702                 if (rc)
2703                         RETURN(rc);
2704
2705                 if (*child) {
2706                         lfsck_object_put(env, *child);
2707                         *child = NULL;
2708                 }
2709         } else if (reset) {
2710                 goto unlink;
2711         }
2712
2713         obj = local_index_find_or_create(env, lfsck->li_los, parent, name,
2714                                          S_IFREG | S_IRUGO | S_IWUSR, ft);
2715         if (IS_ERR(obj))
2716                 RETURN(PTR_ERR(obj));
2717
2718         rc = obj->do_ops->do_index_try(env, obj, ft);
2719         if (rc) {
2720                 lfsck_object_put(env, obj);
2721                 CDEBUG(D_LFSCK, "%s: LFSCK fail to load "
2722                        "sub trace file %s: rc = %d\n",
2723                        lfsck_lfsck2name(com->lc_lfsck), name, rc);
2724         } else {
2725                 *child = obj;
2726         }
2727
2728         RETURN(rc);
2729 }
2730
2731 int lfsck_load_sub_trace_files(const struct lu_env *env,
2732                                struct lfsck_component *com,
2733                                const struct dt_index_features *ft,
2734                                const char *prefix, bool reset)
2735 {
2736         char *name = lfsck_env_info(env)->lti_key;
2737         struct lfsck_sub_trace_obj *lsto;
2738         int rc;
2739         int i;
2740
2741         for (i = 0, rc = 0, lsto = &com->lc_sub_trace_objs[0];
2742              i < LFSCK_STF_COUNT && rc == 0; i++, lsto++) {
2743                 snprintf(name, NAME_MAX, "%s_%02d", prefix, i);
2744                 rc = lfsck_load_one_trace_file(env, com,
2745                                 com->lc_lfsck->li_lfsck_dir,
2746                                 &lsto->lsto_obj, ft, name, reset);
2747         }
2748
2749         return rc;
2750 }
2751
2752 /* external interfaces */
2753 int lfsck_get_speed(char *buf, struct dt_device *key)
2754 {
2755         struct lu_env           env;
2756         struct lfsck_instance  *lfsck;
2757         int                     rc;
2758         ENTRY;
2759
2760         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD);
2761         if (rc != 0)
2762                 RETURN(rc);
2763
2764         lfsck = lfsck_instance_find(key, true, false);
2765         if (lfsck && buf) {
2766                 rc = sprintf(buf, "%u\n",
2767                              lfsck->li_bookmark_ram.lb_speed_limit);
2768                 lfsck_instance_put(&env, lfsck);
2769         } else {
2770                 rc = -ENXIO;
2771         }
2772
2773         lu_env_fini(&env);
2774
2775         RETURN(rc);
2776 }
2777 EXPORT_SYMBOL(lfsck_get_speed);
2778
2779 int lfsck_set_speed(struct dt_device *key, __u32 val)
2780 {
2781         struct lu_env           env;
2782         struct lfsck_instance  *lfsck;
2783         int                     rc;
2784         ENTRY;
2785
2786         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD);
2787         if (rc != 0)
2788                 RETURN(rc);
2789
2790         lfsck = lfsck_instance_find(key, true, false);
2791         if (likely(lfsck != NULL)) {
2792                 mutex_lock(&lfsck->li_mutex);
2793                 if (__lfsck_set_speed(lfsck, val))
2794                         rc = lfsck_bookmark_store(&env, lfsck);
2795                 mutex_unlock(&lfsck->li_mutex);
2796                 lfsck_instance_put(&env, lfsck);
2797         } else {
2798                 rc = -ENXIO;
2799         }
2800
2801         lu_env_fini(&env);
2802
2803         RETURN(rc);
2804 }
2805 EXPORT_SYMBOL(lfsck_set_speed);
2806
2807 int lfsck_get_windows(char *buf, struct dt_device *key)
2808 {
2809         struct lu_env           env;
2810         struct lfsck_instance  *lfsck;
2811         int                     rc;
2812         ENTRY;
2813
2814         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD);
2815         if (rc != 0)
2816                 RETURN(rc);
2817
2818         lfsck = lfsck_instance_find(key, true, false);
2819         if (likely(lfsck != NULL)) {
2820                 rc = sprintf(buf, "%u\n",
2821                              lfsck->li_bookmark_ram.lb_async_windows);
2822                 lfsck_instance_put(&env, lfsck);
2823         } else {
2824                 rc = -ENXIO;
2825         }
2826
2827         lu_env_fini(&env);
2828
2829         RETURN(rc);
2830 }
2831 EXPORT_SYMBOL(lfsck_get_windows);
2832
2833 int lfsck_set_windows(struct dt_device *key, unsigned int val)
2834 {
2835         struct lu_env           env;
2836         struct lfsck_instance  *lfsck;
2837         int                     rc;
2838         ENTRY;
2839
2840         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD);
2841         if (rc != 0)
2842                 RETURN(rc);
2843
2844         lfsck = lfsck_instance_find(key, true, false);
2845         if (likely(lfsck != NULL)) {
2846                 if (val < 1 || val > LFSCK_ASYNC_WIN_MAX) {
2847                         CWARN("%s: invalid async windows size that may "
2848                               "cause memory issues. The valid range is "
2849                               "[1 - %u].\n",
2850                               lfsck_lfsck2name(lfsck), LFSCK_ASYNC_WIN_MAX);
2851                         rc = -EINVAL;
2852                 } else if (lfsck->li_bookmark_ram.lb_async_windows != val) {
2853                         mutex_lock(&lfsck->li_mutex);
2854                         lfsck->li_bookmark_ram.lb_async_windows = val;
2855                         rc = lfsck_bookmark_store(&env, lfsck);
2856                         mutex_unlock(&lfsck->li_mutex);
2857                 }
2858                 lfsck_instance_put(&env, lfsck);
2859         } else {
2860                 rc = -ENXIO;
2861         }
2862
2863         lu_env_fini(&env);
2864
2865         RETURN(rc);
2866 }
2867 EXPORT_SYMBOL(lfsck_set_windows);
2868
2869 int lfsck_dump(struct seq_file *m, struct dt_device *key, enum lfsck_type type)
2870 {
2871         struct lu_env           env;
2872         struct lfsck_instance  *lfsck;
2873         struct lfsck_component *com;
2874         int                     rc;
2875         ENTRY;
2876
2877         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD);
2878         if (rc != 0)
2879                 RETURN(rc);
2880
2881         lfsck = lfsck_instance_find(key, true, false);
2882         if (likely(lfsck != NULL)) {
2883                 com = lfsck_component_find(lfsck, type);
2884                 if (likely(com != NULL)) {
2885                         com->lc_ops->lfsck_dump(&env, com, m);
2886                         lfsck_component_put(&env, com);
2887                 } else {
2888                         rc = -ENOTSUPP;
2889                 }
2890
2891                 lfsck_instance_put(&env, lfsck);
2892         } else {
2893                 rc = -ENXIO;
2894         }
2895
2896         lu_env_fini(&env);
2897
2898         RETURN(rc);
2899 }
2900 EXPORT_SYMBOL(lfsck_dump);
2901
2902 static int lfsck_stop_all(const struct lu_env *env,
2903                           struct lfsck_instance *lfsck,
2904                           struct lfsck_stop *stop)
2905 {
2906         struct lfsck_thread_info          *info   = lfsck_env_info(env);
2907         struct lfsck_request              *lr     = &info->lti_lr;
2908         struct lfsck_async_interpret_args *laia   = &info->lti_laia;
2909         struct ptlrpc_request_set         *set;
2910         struct lfsck_tgt_descs            *ltds   = &lfsck->li_mdt_descs;
2911         struct lfsck_tgt_desc             *ltd;
2912         struct lfsck_bookmark             *bk     = &lfsck->li_bookmark_ram;
2913         __u32                              idx;
2914         int                                rc     = 0;
2915         int                                rc1    = 0;
2916         ENTRY;
2917
2918         LASSERT(stop->ls_flags & LPF_BROADCAST);
2919
2920         set = ptlrpc_prep_set();
2921         if (unlikely(set == NULL))
2922                 RETURN(-ENOMEM);
2923
2924         memset(lr, 0, sizeof(*lr));
2925         lr->lr_event = LE_STOP;
2926         lr->lr_index = lfsck_dev_idx(lfsck);
2927         lr->lr_status = stop->ls_status;
2928         lr->lr_version = bk->lb_version;
2929         lr->lr_active = LFSCK_TYPES_ALL;
2930         lr->lr_param = stop->ls_flags;
2931
2932         memset(laia, 0, sizeof(*laia));
2933         laia->laia_ltds = ltds;
2934         laia->laia_lr = lr;
2935         laia->laia_shared = 1;
2936
2937         down_read(&ltds->ltd_rw_sem);
2938         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
2939                 ltd = lfsck_tgt_get(ltds, idx);
2940                 LASSERT(ltd != NULL);
2941
2942                 laia->laia_ltd = ltd;
2943                 rc = lfsck_async_request(env, ltd->ltd_exp, lr, set,
2944                                          lfsck_async_interpret, laia,
2945                                          LFSCK_NOTIFY);
2946                 if (rc != 0) {
2947                         lfsck_interpret(env, lfsck, NULL, laia, rc);
2948                         lfsck_tgt_put(ltd);
2949                         CERROR("%s: cannot notify MDT %x for LFSCK stop: "
2950                                "rc = %d\n", lfsck_lfsck2name(lfsck), idx, rc);
2951                         rc1 = rc;
2952                 }
2953         }
2954         up_read(&ltds->ltd_rw_sem);
2955
2956         rc = ptlrpc_set_wait(env, set);
2957         ptlrpc_set_destroy(set);
2958
2959         if (rc == 0)
2960                 rc = laia->laia_result;
2961
2962         if (rc == -EALREADY)
2963                 rc = 0;
2964
2965         if (rc != 0)
2966                 CERROR("%s: fail to stop LFSCK on some MDTs: rc = %d\n",
2967                        lfsck_lfsck2name(lfsck), rc);
2968
2969         RETURN(rc != 0 ? rc : rc1);
2970 }
2971
2972 static int lfsck_start_all(const struct lu_env *env,
2973                            struct lfsck_instance *lfsck,
2974                            struct lfsck_start *start)
2975 {
2976         struct lfsck_thread_info          *info   = lfsck_env_info(env);
2977         struct lfsck_request              *lr     = &info->lti_lr;
2978         struct lfsck_async_interpret_args *laia   = &info->lti_laia;
2979         struct ptlrpc_request_set         *set;
2980         struct lfsck_tgt_descs            *ltds   = &lfsck->li_mdt_descs;
2981         struct lfsck_tgt_desc             *ltd;
2982         struct lfsck_bookmark             *bk     = &lfsck->li_bookmark_ram;
2983         __u32                              idx;
2984         int                                rc     = 0;
2985         bool retry = false;
2986         ENTRY;
2987
2988         LASSERT(start->ls_flags & LPF_BROADCAST);
2989
2990         memset(lr, 0, sizeof(*lr));
2991         lr->lr_event = LE_START;
2992         lr->lr_index = lfsck_dev_idx(lfsck);
2993         lr->lr_speed = bk->lb_speed_limit;
2994         lr->lr_version = bk->lb_version;
2995         lr->lr_active = start->ls_active;
2996         lr->lr_param = start->ls_flags;
2997         lr->lr_async_windows = bk->lb_async_windows;
2998         lr->lr_valid = LSV_SPEED_LIMIT | LSV_ERROR_HANDLE | LSV_DRYRUN |
2999                        LSV_ASYNC_WINDOWS | LSV_CREATE_OSTOBJ |
3000                        LSV_CREATE_MDTOBJ;
3001
3002         memset(laia, 0, sizeof(*laia));
3003         laia->laia_ltds = ltds;
3004         laia->laia_lr = lr;
3005         laia->laia_shared = 1;
3006
3007 again:
3008         set = ptlrpc_prep_set();
3009         if (unlikely(!set))
3010                 RETURN(-ENOMEM);
3011
3012         down_read(&ltds->ltd_rw_sem);
3013         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
3014                 ltd = lfsck_tgt_get(ltds, idx);
3015                 LASSERT(ltd != NULL);
3016
3017                 if (retry && !ltd->ltd_retry_start) {
3018                         lfsck_tgt_put(ltd);
3019                         continue;
3020                 }
3021
3022                 laia->laia_ltd = ltd;
3023                 ltd->ltd_retry_start = 0;
3024                 ltd->ltd_layout_done = 0;
3025                 ltd->ltd_namespace_done = 0;
3026                 ltd->ltd_synced_failures = 0;
3027                 rc = lfsck_async_request(env, ltd->ltd_exp, lr, set,
3028                                          lfsck_async_interpret, laia,
3029                                          LFSCK_NOTIFY);
3030                 if (rc != 0) {
3031                         lfsck_interpret(env, lfsck, NULL, laia, rc);
3032                         lfsck_tgt_put(ltd);
3033                         CERROR("%s: cannot notify MDT %x for LFSCK "
3034                                "start, failout: rc = %d\n",
3035                                lfsck_lfsck2name(lfsck), idx, rc);
3036                         break;
3037                 }
3038         }
3039         up_read(&ltds->ltd_rw_sem);
3040
3041         if (rc != 0) {
3042                 ptlrpc_set_destroy(set);
3043
3044                 RETURN(rc);
3045         }
3046
3047         rc = ptlrpc_set_wait(env, set);
3048         ptlrpc_set_destroy(set);
3049
3050         if (rc == 0)
3051                 rc = laia->laia_result;
3052
3053         if (unlikely(rc == -EINPROGRESS)) {
3054                 retry = true;
3055                 schedule_timeout_interruptible(cfs_time_seconds(1));
3056                 set_current_state(TASK_RUNNING);
3057                 if (!signal_pending(current) &&
3058                     thread_is_running(&lfsck->li_thread))
3059                         goto again;
3060
3061                 rc = -EINTR;
3062         }
3063
3064         if (rc != 0) {
3065                 struct lfsck_stop *stop = &info->lti_stop;
3066
3067                 CERROR("%s: cannot start LFSCK on some MDTs, "
3068                        "stop all: rc = %d\n",
3069                        lfsck_lfsck2name(lfsck), rc);
3070                 if (rc != -EALREADY) {
3071                         stop->ls_status = LS_FAILED;
3072                         stop->ls_flags = LPF_ALL_TGT | LPF_BROADCAST;
3073                         lfsck_stop_all(env, lfsck, stop);
3074                 }
3075         }
3076
3077         RETURN(rc);
3078 }
3079
3080 int lfsck_start(const struct lu_env *env, struct dt_device *key,
3081                 struct lfsck_start_param *lsp)
3082 {
3083         struct lfsck_start              *start  = lsp->lsp_start;
3084         struct lfsck_instance           *lfsck;
3085         struct lfsck_bookmark           *bk;
3086         struct ptlrpc_thread            *thread;
3087         struct lfsck_component          *com;
3088         struct lfsck_thread_args        *lta;
3089         struct task_struct              *task;
3090         struct lfsck_tgt_descs          *ltds;
3091         struct lfsck_tgt_desc           *ltd;
3092         __u32                            idx;
3093         int                              rc     = 0;
3094         __u16                            valid  = 0;
3095         __u16                            flags  = 0;
3096         __u16                            type   = 1;
3097         ENTRY;
3098
3099         if (key->dd_rdonly)
3100                 RETURN(-EROFS);
3101
3102         lfsck = lfsck_instance_find(key, true, false);
3103         if (unlikely(lfsck == NULL))
3104                 RETURN(-ENXIO);
3105
3106         if (unlikely(lfsck->li_stopping))
3107                 GOTO(put, rc = -ENXIO);
3108
3109         /* System is not ready, try again later. */
3110         if (unlikely(lfsck->li_namespace == NULL ||
3111                      lfsck_dev_site(lfsck)->ss_server_fld == NULL))
3112                 GOTO(put, rc = -EINPROGRESS);
3113
3114         /* start == NULL means auto trigger paused LFSCK. */
3115         if (!start) {
3116                 if (list_empty(&lfsck->li_list_scan) ||
3117                     OBD_FAIL_CHECK(OBD_FAIL_LFSCK_NO_AUTO))
3118                         GOTO(put, rc = 0);
3119         } else if (start->ls_flags & LPF_BROADCAST && !lfsck->li_master) {
3120                 CERROR("%s: only allow to specify '-A | -o' via MDS\n",
3121                        lfsck_lfsck2name(lfsck));
3122
3123                 GOTO(put, rc = -EPERM);
3124         }
3125
3126         bk = &lfsck->li_bookmark_ram;
3127         thread = &lfsck->li_thread;
3128         mutex_lock(&lfsck->li_mutex);
3129         spin_lock(&lfsck->li_lock);
3130         if (unlikely(thread_is_stopping(thread))) {
3131                 /* Someone is stopping the LFSCK. */
3132                 spin_unlock(&lfsck->li_lock);
3133                 GOTO(out, rc = -EBUSY);
3134         }
3135
3136         if (!thread_is_init(thread) && !thread_is_stopped(thread)) {
3137                 rc = -EALREADY;
3138                 if (unlikely(start == NULL)) {
3139                         spin_unlock(&lfsck->li_lock);
3140                         GOTO(out, rc);
3141                 }
3142
3143                 while (start->ls_active != 0) {
3144                         if (!(type & start->ls_active)) {
3145                                 type <<= 1;
3146                                 continue;
3147                         }
3148
3149                         com = __lfsck_component_find(lfsck, type,
3150                                                      &lfsck->li_list_scan);
3151                         if (com == NULL)
3152                                 com = __lfsck_component_find(lfsck, type,
3153                                                 &lfsck->li_list_double_scan);
3154                         if (com == NULL) {
3155                                 rc = -EOPNOTSUPP;
3156                                 break;
3157                         }
3158
3159                         if (com->lc_ops->lfsck_join != NULL) {
3160                                 rc = com->lc_ops->lfsck_join( env, com, lsp);
3161                                 if (rc != 0 && rc != -EALREADY)
3162                                         break;
3163                         }
3164                         start->ls_active &= ~type;
3165                         type <<= 1;
3166                 }
3167                 spin_unlock(&lfsck->li_lock);
3168                 GOTO(out, rc);
3169         }
3170         spin_unlock(&lfsck->li_lock);
3171
3172         lfsck->li_status = 0;
3173         lfsck->li_oit_over = 0;
3174         lfsck->li_start_unplug = 0;
3175         lfsck->li_drop_dryrun = 0;
3176         lfsck->li_new_scanned = 0;
3177
3178         /* For auto trigger. */
3179         if (start == NULL)
3180                 goto trigger;
3181
3182         start->ls_version = bk->lb_version;
3183
3184         if (start->ls_active != 0) {
3185                 struct lfsck_component *next;
3186
3187                 if (start->ls_active == LFSCK_TYPES_ALL)
3188                         start->ls_active = LFSCK_TYPES_SUPPORTED;
3189
3190                 if (start->ls_active & ~LFSCK_TYPES_SUPPORTED) {
3191                         start->ls_active &= ~LFSCK_TYPES_SUPPORTED;
3192                         GOTO(out, rc = -ENOTSUPP);
3193                 }
3194
3195                 list_for_each_entry_safe(com, next,
3196                                          &lfsck->li_list_scan, lc_link) {
3197                         if (!(com->lc_type & start->ls_active)) {
3198                                 rc = com->lc_ops->lfsck_post(env, com, 0,
3199                                                              false);
3200                                 if (rc != 0)
3201                                         GOTO(out, rc);
3202                         }
3203                 }
3204
3205                 while (start->ls_active != 0) {
3206                         if (type & start->ls_active) {
3207                                 com = __lfsck_component_find(lfsck, type,
3208                                                         &lfsck->li_list_idle);
3209                                 if (com != NULL)
3210                                         /* The component status will be updated
3211                                          * when its prep() is called later by
3212                                          * the LFSCK main engine. */
3213                                         list_move_tail(&com->lc_link,
3214                                                        &lfsck->li_list_scan);
3215                                 start->ls_active &= ~type;
3216                         }
3217                         type <<= 1;
3218                 }
3219         }
3220
3221         if (list_empty(&lfsck->li_list_scan)) {
3222                 /* The speed limit will be used to control both the LFSCK and
3223                  * low layer scrub (if applied), need to be handled firstly. */
3224                 if (start->ls_valid & LSV_SPEED_LIMIT) {
3225                         if (__lfsck_set_speed(lfsck, start->ls_speed_limit)) {
3226                                 rc = lfsck_bookmark_store(env, lfsck);
3227                                 if (rc != 0)
3228                                         GOTO(out, rc);
3229                         }
3230                 }
3231
3232                 goto trigger;
3233         }
3234
3235         if (start->ls_flags & LPF_RESET)
3236                 flags |= DOIF_RESET;
3237
3238         rc = lfsck_set_param(env, lfsck, start, !!(flags & DOIF_RESET));
3239         if (rc != 0)
3240                 GOTO(out, rc);
3241
3242         list_for_each_entry(com, &lfsck->li_list_scan, lc_link) {
3243                 start->ls_active |= com->lc_type;
3244                 if (flags & DOIF_RESET) {
3245                         rc = com->lc_ops->lfsck_reset(env, com, false);
3246                         if (rc != 0)
3247                                 GOTO(out, rc);
3248                 }
3249         }
3250
3251         ltds = &lfsck->li_mdt_descs;
3252         down_read(&ltds->ltd_rw_sem);
3253         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
3254                 ltd = lfsck_ltd2tgt(ltds, idx);
3255                 LASSERT(ltd != NULL);
3256
3257                 ltd->ltd_layout_done = 0;
3258                 ltd->ltd_namespace_done = 0;
3259                 ltd->ltd_synced_failures = 0;
3260                 lfsck_reset_ltd_status(ltd, LFSCK_TYPE_NAMESPACE);
3261                 lfsck_reset_ltd_status(ltd, LFSCK_TYPE_LAYOUT);
3262                 list_del_init(&ltd->ltd_layout_phase_list);
3263                 list_del_init(&ltd->ltd_layout_list);
3264                 list_del_init(&ltd->ltd_namespace_phase_list);
3265                 list_del_init(&ltd->ltd_namespace_list);
3266         }
3267         up_read(&ltds->ltd_rw_sem);
3268
3269         ltds = &lfsck->li_ost_descs;
3270         down_read(&ltds->ltd_rw_sem);
3271         cfs_foreach_bit(ltds->ltd_tgts_bitmap, idx) {
3272                 ltd = lfsck_ltd2tgt(ltds, idx);
3273                 LASSERT(ltd != NULL);
3274
3275                 ltd->ltd_layout_done = 0;
3276                 ltd->ltd_synced_failures = 0;
3277                 lfsck_reset_ltd_status(ltd, LFSCK_TYPE_LAYOUT);
3278                 list_del_init(&ltd->ltd_layout_phase_list);
3279                 list_del_init(&ltd->ltd_layout_list);
3280         }
3281         up_read(&ltds->ltd_rw_sem);
3282
3283 trigger:
3284         lfsck->li_args_dir = LUDA_64BITHASH | LUDA_VERIFY | LUDA_TYPE;
3285         if (bk->lb_param & LPF_DRYRUN)
3286                 lfsck->li_args_dir |= LUDA_VERIFY_DRYRUN;
3287
3288         if (start != NULL && start->ls_valid & LSV_ERROR_HANDLE) {
3289                 valid |= DOIV_ERROR_HANDLE;
3290                 if (start->ls_flags & LPF_FAILOUT)
3291                         flags |= DOIF_FAILOUT;
3292         }
3293
3294         if (start != NULL && start->ls_valid & LSV_DRYRUN) {
3295                 valid |= DOIV_DRYRUN;
3296                 if (start->ls_flags & LPF_DRYRUN)
3297                         flags |= DOIF_DRYRUN;
3298         }
3299
3300         if (!list_empty(&lfsck->li_list_scan))
3301                 flags |= DOIF_OUTUSED;
3302
3303         lfsck->li_args_oit = (flags << DT_OTABLE_IT_FLAGS_SHIFT) | valid;
3304         lta = lfsck_thread_args_init(lfsck, NULL, lsp);
3305         if (IS_ERR(lta))
3306                 GOTO(out, rc = PTR_ERR(lta));
3307
3308         __lfsck_set_speed(lfsck, bk->lb_speed_limit);
3309         spin_lock(&lfsck->li_lock);
3310         thread_set_flags(thread, SVC_STARTING);
3311         spin_unlock(&lfsck->li_lock);
3312         task = kthread_run(lfsck_master_engine, lta, "lfsck");
3313         if (IS_ERR(task)) {
3314                 rc = PTR_ERR(task);
3315                 CERROR("%s: cannot start LFSCK thread: rc = %d\n",
3316                        lfsck_lfsck2name(lfsck), rc);
3317                 lfsck_thread_args_fini(lta);
3318
3319                 GOTO(out, rc);
3320         }
3321
3322         wait_event_idle(thread->t_ctl_waitq,
3323                         thread_is_running(thread) ||
3324                         thread_is_stopped(thread));
3325         if (start == NULL || !(start->ls_flags & LPF_BROADCAST)) {
3326                 lfsck->li_start_unplug = 1;
3327                 wake_up(&thread->t_ctl_waitq);
3328
3329                 GOTO(out, rc = 0);
3330         }
3331
3332         /* release lfsck::li_mutex to avoid deadlock. */
3333         mutex_unlock(&lfsck->li_mutex);
3334         rc = lfsck_start_all(env, lfsck, start);
3335         if (rc != 0) {
3336                 spin_lock(&lfsck->li_lock);
3337                 if (thread_is_stopped(thread)) {
3338                         spin_unlock(&lfsck->li_lock);
3339                 } else {
3340                         lfsck->li_status = LS_FAILED;
3341                         lfsck->li_flags = 0;
3342                         thread_set_flags(thread, SVC_STOPPING);
3343                         spin_unlock(&lfsck->li_lock);
3344
3345                         lfsck->li_start_unplug = 1;
3346                         wake_up(&thread->t_ctl_waitq);
3347                         wait_event_idle(thread->t_ctl_waitq,
3348                                         thread_is_stopped(thread));
3349                 }
3350         } else {
3351                 lfsck->li_start_unplug = 1;
3352                 wake_up(&thread->t_ctl_waitq);
3353         }
3354
3355         GOTO(put, rc);
3356
3357 out:
3358         mutex_unlock(&lfsck->li_mutex);
3359
3360 put:
3361         lfsck_instance_put(env, lfsck);
3362
3363         return rc < 0 ? rc : 0;
3364 }
3365 EXPORT_SYMBOL(lfsck_start);
3366
3367 int lfsck_stop(const struct lu_env *env, struct dt_device *key,
3368                struct lfsck_stop *stop)
3369 {
3370         struct lfsck_instance   *lfsck;
3371         struct ptlrpc_thread    *thread;
3372         int                      rc     = 0;
3373         int                      rc1    = 0;
3374         ENTRY;
3375
3376         lfsck = lfsck_instance_find(key, true, false);
3377         if (unlikely(lfsck == NULL))
3378                 RETURN(-ENXIO);
3379
3380         thread = &lfsck->li_thread;
3381         if (stop && stop->ls_flags & LPF_BROADCAST && !lfsck->li_master) {
3382                 CERROR("%s: only allow to specify '-A' via MDS\n",
3383                        lfsck_lfsck2name(lfsck));
3384                 GOTO(put, rc = -EPERM);
3385         }
3386
3387         spin_lock(&lfsck->li_lock);
3388         /* The target is umounted */
3389         if (stop && stop->ls_status == LS_PAUSED)
3390                 lfsck->li_stopping = 1;
3391
3392         if (thread_is_init(thread) || thread_is_stopped(thread))
3393                 /* no error if LFSCK stopped already, or not started */
3394                 GOTO(unlock, rc = 0);
3395
3396         if (thread_is_stopping(thread))
3397                 /* Someone is stopping LFSCK. */
3398                 GOTO(unlock, rc = -EINPROGRESS);
3399
3400         if (stop) {
3401                 lfsck->li_status = stop->ls_status;
3402                 lfsck->li_flags = stop->ls_flags;
3403         } else {
3404                 lfsck->li_status = LS_STOPPED;
3405                 lfsck->li_flags = 0;
3406         }
3407
3408         thread_set_flags(thread, SVC_STOPPING);
3409
3410         LASSERT(lfsck->li_task != NULL);
3411         cfs_force_sig(SIGINT, lfsck->li_task);
3412
3413         if (lfsck->li_master) {
3414                 struct lfsck_component *com;
3415                 struct lfsck_assistant_data *lad;
3416
3417                 list_for_each_entry(com, &lfsck->li_list_scan, lc_link) {
3418                         lad = com->lc_data;
3419                         spin_lock(&lad->lad_lock);
3420                         if (lad->lad_task != NULL)
3421                                 cfs_force_sig(SIGINT, lad->lad_task);
3422                         spin_unlock(&lad->lad_lock);
3423                 }
3424
3425                 list_for_each_entry(com, &lfsck->li_list_double_scan, lc_link) {
3426                         lad = com->lc_data;
3427                         spin_lock(&lad->lad_lock);
3428                         if (lad->lad_task != NULL)
3429                                 cfs_force_sig(SIGINT, lad->lad_task);
3430                         spin_unlock(&lad->lad_lock);
3431                 }
3432         }
3433
3434         wake_up(&thread->t_ctl_waitq);
3435         spin_unlock(&lfsck->li_lock);
3436         if (stop && stop->ls_flags & LPF_BROADCAST)
3437                 rc1 = lfsck_stop_all(env, lfsck, stop);
3438
3439         /* It was me set the status as 'stopping' just now, if it is not
3440          * 'stopping' now, then either stopped, or re-started by race. */
3441         wait_event_idle(thread->t_ctl_waitq,
3442                         !thread_is_stopping(thread));
3443
3444         GOTO(put, rc = 0);
3445
3446 unlock:
3447         spin_unlock(&lfsck->li_lock);
3448 put:
3449         lfsck_instance_put(env, lfsck);
3450
3451         return rc != 0 ? rc : rc1;
3452 }
3453 EXPORT_SYMBOL(lfsck_stop);
3454
3455 int lfsck_in_notify_local(const struct lu_env *env, struct dt_device *key,
3456                           struct lfsck_req_local *lrl, struct thandle *th)
3457 {
3458         struct lfsck_instance *lfsck;
3459         struct lfsck_component *com;
3460         int rc = -EOPNOTSUPP;
3461         ENTRY;
3462
3463         lfsck = lfsck_instance_find(key, true, false);
3464         if (unlikely(!lfsck))
3465                 RETURN(-ENXIO);
3466
3467         com = lfsck_component_find(lfsck, lrl->lrl_active);
3468         if (likely(com && com->lc_ops->lfsck_in_notify_local)) {
3469                 rc = com->lc_ops->lfsck_in_notify_local(env, com, lrl, th);
3470                 lfsck_component_put(env, com);
3471         }
3472
3473         lfsck_instance_put(env, lfsck);
3474
3475         RETURN(rc);
3476 }
3477 EXPORT_SYMBOL(lfsck_in_notify_local);
3478
3479 int lfsck_in_notify(const struct lu_env *env, struct dt_device *key,
3480                     struct lfsck_request *lr)
3481 {
3482         int rc = -EOPNOTSUPP;
3483         ENTRY;
3484
3485         switch (lr->lr_event) {
3486         case LE_START: {
3487                 struct lfsck_start       *start = &lfsck_env_info(env)->lti_start;
3488                 struct lfsck_start_param  lsp;
3489
3490                 memset(start, 0, sizeof(*start));
3491                 start->ls_valid = lr->lr_valid;
3492                 start->ls_speed_limit = lr->lr_speed;
3493                 start->ls_version = lr->lr_version;
3494                 start->ls_active = lr->lr_active;
3495                 start->ls_flags = lr->lr_param & ~LPF_BROADCAST;
3496                 start->ls_async_windows = lr->lr_async_windows;
3497
3498                 lsp.lsp_start = start;
3499                 lsp.lsp_index = lr->lr_index;
3500                 lsp.lsp_index_valid = 1;
3501                 rc = lfsck_start(env, key, &lsp);
3502                 break;
3503         }
3504         case LE_STOP: {
3505                 struct lfsck_stop *stop = &lfsck_env_info(env)->lti_stop;
3506
3507                 memset(stop, 0, sizeof(*stop));
3508                 stop->ls_status = lr->lr_status;
3509                 stop->ls_flags = lr->lr_param & ~LPF_BROADCAST;
3510                 rc = lfsck_stop(env, key, stop);
3511                 break;
3512         }
3513         case LE_PHASE1_DONE:
3514         case LE_PHASE2_DONE:
3515         case LE_PEER_EXIT:
3516         case LE_CONDITIONAL_DESTROY:
3517         case LE_SET_LMV_MASTER:
3518         case LE_SET_LMV_SLAVE:
3519         case LE_PAIRS_VERIFY: {
3520                 struct lfsck_instance  *lfsck;
3521                 struct lfsck_component *com;
3522
3523                 lfsck = lfsck_instance_find(key, true, false);
3524                 if (unlikely(lfsck == NULL))
3525                         RETURN(-ENXIO);
3526
3527                 com = lfsck_component_find(lfsck, lr->lr_active);
3528                 if (likely(com)) {
3529                         rc = com->lc_ops->lfsck_in_notify(env, com, lr);
3530                         lfsck_component_put(env, com);
3531                 }
3532
3533                 lfsck_instance_put(env, lfsck);
3534                 break;
3535         }
3536         default:
3537                 break;
3538         }
3539
3540         RETURN(rc);
3541 }
3542 EXPORT_SYMBOL(lfsck_in_notify);
3543
3544 int lfsck_query(const struct lu_env *env, struct dt_device *key,
3545                 struct lfsck_request *req, struct lfsck_reply *rep,
3546                 struct lfsck_query *que)
3547 {
3548         struct lfsck_instance  *lfsck;
3549         struct lfsck_component *com;
3550         int                     i;
3551         int                     rc = 0;
3552         __u16                   type;
3553         ENTRY;
3554
3555         lfsck = lfsck_instance_find(key, true, false);
3556         if (unlikely(lfsck == NULL))
3557                 RETURN(-ENXIO);
3558
3559         if (que != NULL) {
3560                 if (que->lu_types == LFSCK_TYPES_ALL)
3561                         que->lu_types =
3562                                 LFSCK_TYPES_SUPPORTED & ~LFSCK_TYPE_SCRUB;
3563
3564                 if (que->lu_types & ~LFSCK_TYPES_SUPPORTED) {
3565                         que->lu_types &= ~LFSCK_TYPES_SUPPORTED;
3566
3567                         GOTO(out, rc = -ENOTSUPP);
3568                 }
3569
3570                 for (i = 0, type = BIT(i); i < LFSCK_TYPE_BITS;
3571                      i++, type = BIT(i)) {
3572                         if (!(que->lu_types & type))
3573                                 continue;
3574
3575 again:
3576                         com = lfsck_component_find(lfsck, type);
3577                         if (unlikely(com == NULL))
3578                                 GOTO(out, rc = -ENOTSUPP);
3579
3580                         memset(que->lu_mdts_count[i], 0,
3581                                sizeof(__u32) * (LS_MAX + 1));
3582                         memset(que->lu_osts_count[i], 0,
3583                                sizeof(__u32) * (LS_MAX + 1));
3584                         que->lu_repaired[i] = 0;
3585                         rc = com->lc_ops->lfsck_query(env, com, req, rep,
3586                                                       que, i);
3587                         lfsck_component_put(env, com);
3588                         if  (rc < 0)
3589                                 GOTO(out, rc);
3590                 }
3591
3592                 if (!(que->lu_flags & LPF_WAIT))
3593                         GOTO(out, rc);
3594
3595                 for (i = 0, type = BIT(i); i < LFSCK_TYPE_BITS;
3596                      i++, type = BIT(i)) {
3597                         if (!(que->lu_types & type))
3598                                 continue;
3599
3600                         if (que->lu_mdts_count[i][LS_SCANNING_PHASE1] != 0 ||
3601                             que->lu_mdts_count[i][LS_SCANNING_PHASE2] != 0 ||
3602                             que->lu_osts_count[i][LS_SCANNING_PHASE1] != 0 ||
3603                             que->lu_osts_count[i][LS_SCANNING_PHASE2] != 0) {
3604                                 /* If it is required to wait, then sleep
3605                                  * 3 seconds and try to query again.
3606                                  */
3607                                 unsigned long timeout =
3608                                         msecs_to_jiffies(3000) + 1;
3609                                 while (timeout &&
3610                                        !fatal_signal_pending(current))
3611                                         timeout = schedule_timeout_killable(
3612                                                 timeout);
3613                                 if (timeout == 0)
3614                                         goto again;
3615                         }
3616                 }
3617         } else {
3618                 com = lfsck_component_find(lfsck, req->lr_active);
3619                 if (likely(com != NULL)) {
3620                         rc = com->lc_ops->lfsck_query(env, com, req, rep,
3621                                                       que, -1);
3622                         lfsck_component_put(env, com);
3623                 } else {
3624                         rc = -ENOTSUPP;
3625                 }
3626         }
3627
3628         GOTO(out, rc);
3629
3630 out:
3631         lfsck_instance_put(env, lfsck);
3632         return rc;
3633 }
3634 EXPORT_SYMBOL(lfsck_query);
3635
3636 int lfsck_register_namespace(const struct lu_env *env, struct dt_device *key,
3637                              struct ldlm_namespace *ns)
3638 {
3639         struct lfsck_instance  *lfsck;
3640         int                     rc      = -ENXIO;
3641
3642         lfsck = lfsck_instance_find(key, true, false);
3643         if (likely(lfsck != NULL)) {
3644                 lfsck->li_namespace = ns;
3645                 lfsck_instance_put(env, lfsck);
3646                 rc = 0;
3647         }
3648
3649         return rc;
3650 }
3651 EXPORT_SYMBOL(lfsck_register_namespace);
3652
3653 int lfsck_register(const struct lu_env *env, struct dt_device *key,
3654                    struct dt_device *next, struct obd_device *obd,
3655                    lfsck_out_notify notify, void *notify_data, bool master)
3656 {
3657         struct lfsck_instance   *lfsck;
3658         struct dt_object        *root  = NULL;
3659         struct dt_object        *obj   = NULL;
3660         struct lu_fid           *fid   = &lfsck_env_info(env)->lti_fid;
3661         int                      rc;
3662         ENTRY;
3663
3664         lfsck = lfsck_instance_find(key, false, false);
3665         if (unlikely(lfsck != NULL))
3666                 RETURN(-EEXIST);
3667
3668         OBD_ALLOC_PTR(lfsck);
3669         if (lfsck == NULL)
3670                 RETURN(-ENOMEM);
3671
3672         mutex_init(&lfsck->li_mutex);
3673         spin_lock_init(&lfsck->li_lock);
3674         INIT_LIST_HEAD(&lfsck->li_link);
3675         INIT_LIST_HEAD(&lfsck->li_list_scan);
3676         INIT_LIST_HEAD(&lfsck->li_list_dir);
3677         INIT_LIST_HEAD(&lfsck->li_list_double_scan);
3678         INIT_LIST_HEAD(&lfsck->li_list_idle);
3679         INIT_LIST_HEAD(&lfsck->li_list_lmv);
3680         atomic_set(&lfsck->li_ref, 1);
3681         atomic_set(&lfsck->li_double_scan_count, 0);
3682         init_waitqueue_head(&lfsck->li_thread.t_ctl_waitq);
3683         lfsck->li_out_notify = notify;
3684         lfsck->li_out_notify_data = notify_data;
3685         lfsck->li_next = next;
3686         lfsck->li_bottom = key;
3687         lfsck->li_obd = obd;
3688
3689         rc = lfsck_tgt_descs_init(&lfsck->li_ost_descs);
3690         if (rc != 0)
3691                 GOTO(out, rc);
3692
3693         rc = lfsck_tgt_descs_init(&lfsck->li_mdt_descs);
3694         if (rc != 0)
3695                 GOTO(out, rc);
3696
3697         fid->f_seq = FID_SEQ_LOCAL_NAME;
3698         fid->f_oid = 1;
3699         fid->f_ver = 0;
3700         rc = local_oid_storage_init(env, key, fid, &lfsck->li_los);
3701         if (rc != 0)
3702                 GOTO(out, rc);
3703
3704         rc = dt_root_get(env, key, fid);
3705         if (rc != 0)
3706                 GOTO(out, rc);
3707
3708         root = dt_locate(env, key, fid);
3709         if (IS_ERR(root))
3710                 GOTO(out, rc = PTR_ERR(root));
3711
3712         lfsck->li_local_root_fid = *fid;
3713         if (master) {
3714                 lfsck->li_master = 1;
3715                 if (lfsck_dev_idx(lfsck) == 0) {
3716                         struct lu_fid *pfid = &lfsck_env_info(env)->lti_fid2;
3717                         const struct lu_name *cname;
3718
3719                         rc = dt_lookup_dir(env, root, "ROOT",
3720                                            &lfsck->li_global_root_fid);
3721                         if (rc != 0)
3722                                 GOTO(out, rc);
3723
3724                         obj = dt_locate(env, key, &lfsck->li_global_root_fid);
3725                         if (IS_ERR(obj))
3726                                 GOTO(out, rc = PTR_ERR(obj));
3727
3728                         rc = dt_lookup_dir(env, obj, dotlustre, fid);
3729                         if (rc != 0)
3730                                 GOTO(out, rc);
3731
3732                         lfsck_object_put(env, obj);
3733                         obj = dt_locate(env, key, fid);
3734                         if (IS_ERR(obj))
3735                                 GOTO(out, rc = PTR_ERR(obj));
3736
3737                         cname = lfsck_name_get_const(env, dotlustre,
3738                                                      strlen(dotlustre));
3739                         rc = lfsck_verify_linkea(env, lfsck, obj, cname,
3740                                                  &lfsck->li_global_root_fid);
3741                         if (rc != 0)
3742                                 GOTO(out, rc);
3743
3744                         if (unlikely(!dt_try_as_dir(env, obj)))
3745                                 GOTO(out, rc = -ENOTDIR);
3746
3747                         *pfid = *fid;
3748                         rc = dt_lookup_dir(env, obj, lostfound, fid);
3749                         if (rc != 0)
3750                                 GOTO(out, rc);
3751
3752                         lfsck_object_put(env, obj);
3753                         obj = dt_locate(env, key, fid);
3754                         if (IS_ERR(obj))
3755                                 GOTO(out, rc = PTR_ERR(obj));
3756
3757                         cname = lfsck_name_get_const(env, lostfound,
3758                                                      strlen(lostfound));
3759                         rc = lfsck_verify_linkea(env, lfsck, obj, cname, pfid);
3760                         if (rc != 0)
3761                                 GOTO(out, rc);
3762
3763                         lfsck_object_put(env, obj);
3764                         obj = NULL;
3765                 }
3766         }
3767
3768         fid->f_seq = FID_SEQ_LOCAL_FILE;
3769         fid->f_oid = OTABLE_IT_OID;
3770         fid->f_ver = 0;
3771         obj = dt_locate(env, key, fid);
3772         if (IS_ERR(obj))
3773                 GOTO(out, rc = PTR_ERR(obj));
3774
3775         rc = obj->do_ops->do_index_try(env, obj, &dt_otable_features);
3776         if (rc != 0)
3777                 GOTO(out, rc);
3778
3779         lfsck->li_obj_oit = obj;
3780         obj = local_file_find_or_create(env, lfsck->li_los, root, LFSCK_DIR,
3781                                         S_IFDIR | S_IRUGO | S_IWUSR);
3782         if (IS_ERR(obj))
3783                 GOTO(out, rc = PTR_ERR(obj));
3784
3785         lu_object_get(&obj->do_lu);
3786         lfsck->li_lfsck_dir = obj;
3787         rc = lfsck_bookmark_setup(env, lfsck);
3788         if (rc != 0)
3789                 GOTO(out, rc);
3790
3791         if (master) {
3792                 rc = lfsck_fid_init(lfsck);
3793                 if (rc < 0)
3794                         GOTO(out, rc);
3795
3796                 rc = lfsck_namespace_setup(env, lfsck);
3797                 if (rc < 0)
3798                         GOTO(out, rc);
3799         }
3800
3801         rc = lfsck_layout_setup(env, lfsck);
3802         if (rc < 0)
3803                 GOTO(out, rc);
3804
3805         /* XXX: more LFSCK components initialization to be added here. */
3806
3807         rc = lfsck_instance_add(lfsck);
3808         if (rc == 0)
3809                 rc = lfsck_add_target_from_orphan(env, lfsck);
3810 out:
3811         if (obj != NULL && !IS_ERR(obj))
3812                 lfsck_object_put(env, obj);
3813         if (root != NULL && !IS_ERR(root))
3814                 lfsck_object_put(env, root);
3815         if (rc != 0)
3816                 lfsck_instance_cleanup(env, lfsck);
3817         return rc;
3818 }
3819 EXPORT_SYMBOL(lfsck_register);
3820
3821 void lfsck_degister(const struct lu_env *env, struct dt_device *key)
3822 {
3823         struct lfsck_instance *lfsck;
3824
3825         lfsck = lfsck_instance_find(key, false, true);
3826         if (lfsck != NULL)
3827                 lfsck_instance_put(env, lfsck);
3828 }
3829 EXPORT_SYMBOL(lfsck_degister);
3830
3831 int lfsck_add_target(const struct lu_env *env, struct dt_device *key,
3832                      struct dt_device *tgt, struct obd_export *exp,
3833                      __u32 index, bool for_ost)
3834 {
3835         struct lfsck_instance   *lfsck;
3836         struct lfsck_tgt_desc   *ltd;
3837         int                      rc;
3838         ENTRY;
3839
3840         OBD_ALLOC_PTR(ltd);
3841         if (ltd == NULL)
3842                 RETURN(-ENOMEM);
3843
3844         ltd->ltd_tgt = tgt;
3845         ltd->ltd_key = key;
3846         ltd->ltd_exp = exp;
3847         INIT_LIST_HEAD(&ltd->ltd_orphan_list);
3848         INIT_LIST_HEAD(&ltd->ltd_layout_list);
3849         INIT_LIST_HEAD(&ltd->ltd_layout_phase_list);
3850         INIT_LIST_HEAD(&ltd->ltd_namespace_list);
3851         INIT_LIST_HEAD(&ltd->ltd_namespace_phase_list);
3852         atomic_set(&ltd->ltd_ref, 1);
3853         ltd->ltd_index = index;
3854
3855         spin_lock(&lfsck_instance_lock);
3856         lfsck = __lfsck_instance_find(key, true, false);
3857         if (lfsck == NULL) {
3858                 if (for_ost)
3859                         list_add_tail(&ltd->ltd_orphan_list,
3860                                       &lfsck_ost_orphan_list);
3861                 else
3862                         list_add_tail(&ltd->ltd_orphan_list,
3863                                       &lfsck_mdt_orphan_list);
3864                 spin_unlock(&lfsck_instance_lock);
3865
3866                 RETURN(0);
3867         }
3868         spin_unlock(&lfsck_instance_lock);
3869
3870         rc = __lfsck_add_target(env, lfsck, ltd, for_ost, false);
3871         if (rc != 0)
3872                 lfsck_tgt_put(ltd);
3873
3874         lfsck_instance_put(env, lfsck);
3875
3876         RETURN(rc);
3877 }
3878 EXPORT_SYMBOL(lfsck_add_target);
3879
3880 void lfsck_del_target(const struct lu_env *env, struct dt_device *key,
3881                       struct dt_device *tgt, __u32 index, bool for_ost)
3882 {
3883         struct lfsck_instance   *lfsck;
3884         struct lfsck_tgt_descs  *ltds;
3885         struct lfsck_tgt_desc   *ltd;
3886         struct list_head        *head;
3887
3888         if (for_ost)
3889                 head = &lfsck_ost_orphan_list;
3890         else
3891                 head = &lfsck_mdt_orphan_list;
3892
3893         spin_lock(&lfsck_instance_lock);
3894         list_for_each_entry(ltd, head, ltd_orphan_list) {
3895                 if (ltd->ltd_tgt == tgt) {
3896                         list_del_init(&ltd->ltd_orphan_list);
3897                         spin_unlock(&lfsck_instance_lock);
3898                         lfsck_tgt_put(ltd);
3899
3900                         return;
3901                 }
3902         }
3903
3904         ltd = NULL;
3905         lfsck = __lfsck_instance_find(key, true, false);
3906         spin_unlock(&lfsck_instance_lock);
3907         if (unlikely(lfsck == NULL))
3908                 return;
3909
3910         if (for_ost)
3911                 ltds = &lfsck->li_ost_descs;
3912         else
3913                 ltds = &lfsck->li_mdt_descs;
3914
3915         down_write(&ltds->ltd_rw_sem);
3916         LASSERT(ltds->ltd_tgts_bitmap != NULL);
3917
3918         if (unlikely(index >= ltds->ltd_tgts_bitmap->size))
3919                 goto unlock;
3920
3921         ltd = lfsck_ltd2tgt(ltds, index);
3922         if (unlikely(ltd == NULL))
3923                 goto unlock;
3924
3925         LASSERT(ltds->ltd_tgtnr > 0);
3926
3927         ltds->ltd_tgtnr--;
3928         cfs_bitmap_clear(ltds->ltd_tgts_bitmap, index);
3929         lfsck_assign_tgt(ltds, NULL, index);
3930
3931 unlock:
3932         if (ltd == NULL) {
3933                 if (for_ost)
3934                         head = &lfsck->li_ost_descs.ltd_orphan;
3935                 else
3936                         head = &lfsck->li_mdt_descs.ltd_orphan;
3937
3938                 list_for_each_entry(ltd, head, ltd_orphan_list) {
3939                         if (ltd->ltd_tgt == tgt) {
3940                                 list_del_init(&ltd->ltd_orphan_list);
3941                                 break;
3942                         }
3943                 }
3944         }
3945
3946         up_write(&ltds->ltd_rw_sem);
3947         if (ltd != NULL) {
3948                 spin_lock(&ltds->ltd_lock);
3949                 ltd->ltd_dead = 1;
3950                 spin_unlock(&ltds->ltd_lock);
3951                 lfsck_stop_notify(env, lfsck, ltds, ltd, LFSCK_TYPE_NAMESPACE);
3952                 lfsck_stop_notify(env, lfsck, ltds, ltd, LFSCK_TYPE_LAYOUT);
3953                 lfsck_tgt_put(ltd);
3954         }
3955
3956         lfsck_instance_put(env, lfsck);
3957 }
3958 EXPORT_SYMBOL(lfsck_del_target);
3959
3960 static int __init lfsck_init(void)
3961 {
3962         int rc;
3963
3964         lfsck_key_init_generic(&lfsck_thread_key, NULL);
3965         rc = lu_context_key_register(&lfsck_thread_key);
3966         if (!rc) {
3967                 tgt_register_lfsck_in_notify_local(lfsck_in_notify_local);
3968                 tgt_register_lfsck_in_notify(lfsck_in_notify);
3969                 tgt_register_lfsck_query(lfsck_query);
3970         }
3971
3972         return rc;
3973 }
3974
3975 static void __exit lfsck_exit(void)
3976 {
3977         struct lfsck_tgt_desc *ltd;
3978         struct lfsck_tgt_desc *next;
3979
3980         LASSERT(list_empty(&lfsck_instance_list));
3981
3982         list_for_each_entry_safe(ltd, next, &lfsck_ost_orphan_list,
3983                                  ltd_orphan_list) {
3984                 list_del_init(&ltd->ltd_orphan_list);
3985                 lfsck_tgt_put(ltd);
3986         }
3987
3988         list_for_each_entry_safe(ltd, next, &lfsck_mdt_orphan_list,
3989                                  ltd_orphan_list) {
3990                 list_del_init(&ltd->ltd_orphan_list);
3991                 lfsck_tgt_put(ltd);
3992         }
3993
3994         lu_context_key_degister(&lfsck_thread_key);
3995 }
3996
3997 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
3998 MODULE_DESCRIPTION("Lustre File System Checker");
3999 MODULE_VERSION(LUSTRE_VERSION_STRING);
4000 MODULE_LICENSE("GPL");
4001
4002 module_init(lfsck_init);
4003 module_exit(lfsck_exit);