diff options
| -rw-r--r-- | drivers/md/dm-integrity.c | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/drivers/md/dm-integrity.c b/drivers/md/dm-integrity.c index 54b3fe1403a8..42be03bbfafa 100644 --- a/drivers/md/dm-integrity.c +++ b/drivers/md/dm-integrity.c @@ -15,6 +15,7 @@ #include <linux/rbtree.h> #include <linux/delay.h> #include <linux/random.h> +#include <linux/reboot.h> #include <crypto/hash.h> #include <crypto/skcipher.h> #include <linux/async_tx.h> @@ -257,6 +258,8 @@ struct dm_integrity_c { struct alg_spec journal_mac_alg; atomic64_t number_of_mismatches; + + struct notifier_block reboot_notifier; }; struct dm_integrity_range { @@ -2717,11 +2720,27 @@ clear_journal: init_journal_node(&ic->journal_tree[i]); } +static int dm_integrity_reboot(struct notifier_block *n, unsigned long code, void *x) +{ + struct dm_integrity_c *ic = container_of(n, struct dm_integrity_c, reboot_notifier); + + if (ic->mode == 'B') { + DEBUG_print("dm_integrity_reboot\n"); + cancel_delayed_work_sync(&ic->bitmap_flush_work); + queue_delayed_work(ic->commit_wq, &ic->bitmap_flush_work, 0); + flush_workqueue(ic->commit_wq); + } + + return NOTIFY_DONE; +} + static void dm_integrity_postsuspend(struct dm_target *ti) { struct dm_integrity_c *ic = (struct dm_integrity_c *)ti->private; int r; + WARN_ON(unregister_reboot_notifier(&ic->reboot_notifier)); + del_timer_sync(&ic->autocommit_timer); WRITE_ONCE(ic->suspending, 1); @@ -2829,6 +2848,11 @@ static void dm_integrity_resume(struct dm_target *ti) recalc_write_super(ic); } } + + ic->reboot_notifier.notifier_call = dm_integrity_reboot; + ic->reboot_notifier.next = NULL; + ic->reboot_notifier.priority = INT_MAX - 1; /* be notified after md and before hardware drivers */ + WARN_ON(register_reboot_notifier(&ic->reboot_notifier)); } static void dm_integrity_status(struct dm_target *ti, status_type_t type, |