From 5694b97dda2457f96c413cb735832bff42dbb3e5 Mon Sep 17 00:00:00 2001
From: jflyper <shinoda@jaist.ac.jp>
Date: Mon, 29 May 2017 09:28:18 +0900
Subject: [PATCH] Prevent CMS from freezing after flash erase

---
 src/main/cms/cms.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/cms/cms.c b/src/main/cms/cms.c
index 1cb93d925..5e97c5eeb 100644
--- a/src/main/cms/cms.c
+++ b/src/main/cms/cms.c
@@ -990,7 +990,11 @@ void cmsUpdate(uint32_t currentTimeUs)
             lastCmsHeartBeatMs = currentTimeMs;
         }
     }
-    lastCalledMs = currentTimeMs;
+
+    // Some key (command), notably flash erase, takes too long to use the
+    // currentTimeMs to be used as lastCalledMs (freezes CMS for a minute or so
+    // if used).
+    lastCalledMs = millis();
 }
 
 void cmsHandler(timeUs_t currentTimeUs)