return 0;
 err:
        set_bit(MD_CLUSTER_HOLDING_MUTEX_FOR_RECVD, &cinfo->state);
-       md_unregister_thread(&cinfo->recovery_thread);
-       md_unregister_thread(&cinfo->recv_thread);
+       md_unregister_thread(mddev, &cinfo->recovery_thread);
+       md_unregister_thread(mddev, &cinfo->recv_thread);
        lockres_free(cinfo->message_lockres);
        lockres_free(cinfo->token_lockres);
        lockres_free(cinfo->ack_lockres);
                resync_bitmap(mddev);
 
        set_bit(MD_CLUSTER_HOLDING_MUTEX_FOR_RECVD, &cinfo->state);
-       md_unregister_thread(&cinfo->recovery_thread);
-       md_unregister_thread(&cinfo->recv_thread);
+       md_unregister_thread(mddev, &cinfo->recovery_thread);
+       md_unregister_thread(mddev, &cinfo->recv_thread);
        lockres_free(cinfo->message_lockres);
        lockres_free(cinfo->token_lockres);
        lockres_free(cinfo->ack_lockres);
 
                mddev->pers->quiesce(mddev, 1);
                mddev->pers->quiesce(mddev, 0);
        }
-       md_unregister_thread(&mddev->thread);
+       md_unregister_thread(mddev, &mddev->thread);
        if (mddev->queue)
                blk_sync_queue(mddev->queue); /* the unplug fn references 'conf'*/
 }
 }
 EXPORT_SYMBOL(md_register_thread);
 
-void md_unregister_thread(struct md_thread __rcu **threadp)
+void md_unregister_thread(struct mddev *mddev, struct md_thread __rcu **threadp)
 {
-       struct md_thread *thread = rcu_dereference_protected(*threadp, true);
+       struct md_thread *thread = rcu_dereference_protected(*threadp,
+                                       lockdep_is_held(&mddev->reconfig_mutex));
 
        if (!thread)
                return;
        bool is_reshaped = false;
 
        /* resync has finished, collect result */
-       md_unregister_thread(&mddev->sync_thread);
+       md_unregister_thread(mddev, &mddev->sync_thread);
        atomic_inc(&mddev->sync_seq);
 
        if (!test_bit(MD_RECOVERY_INTR, &mddev->recovery) &&
 
        void (*run)(struct md_thread *thread),
        struct mddev *mddev,
        const char *name);
-extern void md_unregister_thread(struct md_thread __rcu **threadp);
+extern void md_unregister_thread(struct mddev *mddev, struct md_thread __rcu **threadp);
 extern void md_wakeup_thread(struct md_thread __rcu *thread);
 extern void md_check_recovery(struct mddev *mddev);
 extern void md_reap_sync_thread(struct mddev *mddev);
 
         * RAID1 needs at least one disk in active
         */
        if (conf->raid_disks - mddev->degraded < 1) {
-               md_unregister_thread(&conf->thread);
+               md_unregister_thread(mddev, &conf->thread);
                ret = -EINVAL;
                goto abort;
        }
 
        ret = md_integrity_register(mddev);
        if (ret) {
-               md_unregister_thread(&mddev->thread);
+               md_unregister_thread(mddev, &mddev->thread);
                goto abort;
        }
        return 0;
 
        return 0;
 
 out_free_conf:
-       md_unregister_thread(&mddev->thread);
+       md_unregister_thread(mddev, &mddev->thread);
        raid10_free_conf(conf);
        mddev->private = NULL;
 out:
 
 {
        struct r5l_log *log = conf->log;
 
-       md_unregister_thread(&log->reclaim_thread);
+       md_unregister_thread(conf->mddev, &log->reclaim_thread);
 
        /*
         * 'reconfig_mutex' is held by caller, set 'confg->log' to NULL to
 
 
        return 0;
 abort:
-       md_unregister_thread(&mddev->thread);
+       md_unregister_thread(mddev, &mddev->thread);
        print_raid5_conf(conf);
        free_conf(conf);
        mddev->private = NULL;