From 26f9b5ba8106f504fcc77aa80c6152b245e5e958 Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <pavol@rusnak.io>
Date: Sun, 10 Feb 2019 13:21:31 +0100
Subject: [PATCH] flash: unify lock/unlock sequences

---
 firmware/bl_check.c | 3 +++
 supervise.c         | 1 +
 2 files changed, 4 insertions(+)

diff --git a/firmware/bl_check.c b/firmware/bl_check.c
index 92c0a76e1..5ca0a41c3 100644
--- a/firmware/bl_check.c
+++ b/firmware/bl_check.c
@@ -78,6 +78,8 @@ void check_bootloader(void)
 
 	for (int tries = 0; tries < 10; tries++) {
 		// replace bootloader
+		flash_wait_for_last_operation();
+		flash_clear_status_flags();
 		flash_unlock();
 		for (int i = FLASH_BOOT_SECTOR_FIRST; i <= FLASH_BOOT_SECTOR_LAST; i++) {
 			flash_erase_sector(i, FLASH_CR_PROGRAM_X32);
@@ -86,6 +88,7 @@ void check_bootloader(void)
 			const uint32_t *w = (const uint32_t *)(bl_data + i * 4);
 			flash_program_word(FLASH_BOOT_START + i * 4, *w);
 		}
+		flash_wait_for_last_operation();
 		flash_lock();
 		// check whether the write was OK
 		r = memory_bootloader_hash(hash);
diff --git a/supervise.c b/supervise.c
index 3c21ea0e4..7669cc5ed 100644
--- a/supervise.c
+++ b/supervise.c
@@ -25,6 +25,7 @@
 #if !EMULATOR
 
 static void svhandler_flash_unlock(void) {
+	flash_wait_for_last_operation();
 	flash_clear_status_flags();
 	flash_unlock();
 }