From 5d5c7ee6404c70e13a6afbd703c893a3498807b4 Mon Sep 17 00:00:00 2001
From: Bertrand Songis <bsongis@gmail.com>
Date: Thu, 25 Apr 2019 17:18:23 +0200
Subject: [PATCH] Bluetooth flash firmware continued

---
 radio/src/bluetooth.cpp | 37 +++++++++++++++++++++++++++++++------
 radio/src/bluetooth.h   |  6 ++++--
 2 files changed, 35 insertions(+), 8 deletions(-)

diff --git a/radio/src/bluetooth.cpp b/radio/src/bluetooth.cpp
index bac0503d9..ef6ed1cbf 100644
--- a/radio/src/bluetooth.cpp
+++ b/radio/src/bluetooth.cpp
@@ -507,7 +507,7 @@ uint8_t Bluetooth::read(uint8_t * data, uint8_t size, uint32_t timeout)
   return len;
 }
 
-const char * Bluetooth::waitBootloaderResponse()
+const char * Bluetooth::waitBootloaderCommandResponse()
 {
   uint8_t response[2];
   if (read(response, 2) != 2) {
@@ -525,18 +525,43 @@ const char * Bluetooth::waitBootloaderResponse()
   return "Bluetooth error";
 }
 
+enum
+{
+  CMD_GET_CHIP_ID = 0x28,
+};
+
+const char * Bluetooth::doFlashFirmware(const char * filename)
+{
+  const char * result;
+
+  // Dummy command
+  sendBootloaderCommand(0);
+  result = waitBootloaderCommandResponse();
+  if (result)
+    return result;
+
+  // Get chip ID
+  sendBootloaderCommand(CMD_GET_CHIP_ID);
+  result = waitBootloaderCommandResponse();
+  if (result)
+    return result;
+  uint8_t id[4];
+
+
+  return result;
+}
+
 void Bluetooth::flashFirmware(const char * filename)
 {
+  drawProgressScreen(getBasename(filename), "Bluetooth reset...", 0, 0);
+
   state = BLUETOOTH_STATE_FLASH_FIRMWARE;
 
   pausePulses();
 
-  bluetoothInit(BLUETOOTH_BOOTLOADER_BAUDRATE, false); /* go to bootloader mode */
+  bluetoothInit(BLUETOOTH_BOOTLOADER_BAUDRATE, false); // bootloader mode
 
-  drawProgressScreen(getBasename(filename), "Bluetooth reset...", 0, 0);
-
-  sendBootloaderCommand(0);
-  const char * result = waitBootloaderResponse();
+  const char * result = doFlashFirmware(filename);
 
   AUDIO_PLAY(AU_SPECIAL_SOUND_BEEP1 );
   BACKLIGHT_ENABLE();
diff --git a/radio/src/bluetooth.h b/radio/src/bluetooth.h
index 75c935c90..8d4ae23e6 100644
--- a/radio/src/bluetooth.h
+++ b/radio/src/bluetooth.h
@@ -64,7 +64,7 @@ class Bluetooth
 
   protected:
     void pushByte(uint8_t byte);
-    uint8_t read(uint8_t * data, uint8_t size, uint32_t timeout=1000/*ms*/);
+    uint8_t read(uint8_t * data, uint8_t size, uint32_t timeout=100/*ms*/);
     void write(const uint8_t * data, uint8_t length);
     void appendTrainerByte(uint8_t data);
     void processTrainerFrame(const uint8_t * buffer);
@@ -74,7 +74,9 @@ class Bluetooth
 
     uint8_t bootloaderChecksum(uint8_t command, const uint8_t * data, uint8_t size);
     void sendBootloaderCommand(uint8_t command, const uint8_t * data = nullptr, uint8_t size = 0);
-    const char * waitBootloaderResponse();
+    const char * waitBootloaderCommandResponse();
+
+    const char * doFlashFirmware(const char * filename);
 
     uint8_t buffer[BLUETOOTH_LINE_LENGTH+1];
     uint8_t bufferIndex = 0;