ble: Wait with setting the product string to after orientation

usb_processing wasn't being setup until AFTER usb was enabled. This was
a problem beacuse the app is so fast that it manages to send the first
requests over ble before usb is setup.

Since "usb_processing" is unrelated to the usb driver, we initialize it
before we initialize the usb driver.

Author: NickeZ

Makefile

@@ -79,6 +79,8 @@ bootloader-btc: | build
 	$(MAKE) -C build bb02-bl-btconly.elf
 bootloader-btc-development: | build
 	$(MAKE) -C build bb02-bl-btconly-development.elf
+bootloader-btc-plus-development: | build
+	$(MAKE) -C build bb02p-bl-btconly-development.elf
 bootloader-btc-production: | build
 	$(MAKE) -C build bb02-bl-btconly-production.elf
 factory-setup: | build
@@ -113,6 +115,8 @@ jlink-flash-bootloader-development: | build
 	JLinkExe -NoGui 1 -if SWD -device ATSAMD51J20 -speed 4000 -autoconnect 1 -CommanderScript ./build/scripts/bb02-bl-multi-development.jlink
 jlink-flash-bootloader-plus-development: | build
 	JLinkExe -NoGui 1 -if SWD -device ATSAMD51J20 -speed 4000 -autoconnect 1 -CommanderScript ./build/scripts/bb02p-bl-multi-development.jlink
+jlink-flash-bootloader-btc-plus-development: | build
+	JLinkExe -NoGui 1 -if SWD -device ATSAMD51J20 -speed 4000 -autoconnect 1 -CommanderScript ./build/scripts/bb02p-bl-btconly-development.jlink
 jlink-flash-bootloader-development-locked: | build
 	JLinkExe -NoGui 1 -if SWD -device ATSAMD51J20 -speed 4000 -autoconnect 1 -CommanderScript ./build/scripts/bb02-bl-multi-development-locked.jlink
 jlink-flash-bootloader: | build

src/bootloader/bootloader.c

@@ -849,14 +849,12 @@ static size_t _api_command(const uint8_t* input, uint8_t* output, const size_t m
 
     case OP_REBOOT: {
 #if PLATFORM_BITBOX02PLUS == 1
-        // da14531_set_product("", 0, &uart_write_queue);
-        da14531_reset(&uart_write_queue);
-        // Send it now, because we are about to reset ourselves
+        da14531_set_product(NULL, 0, &uart_write_queue);
+        //  Send it now, because we are about to reset ourselves
         while (ringbuffer_num(&uart_write_queue)) {
             uart_poll(NULL, 0, NULL, &uart_write_queue);
         }
 #endif
-
         _api_reboot();
     } break;
 
@@ -1046,7 +1044,8 @@ void bootloader_jump(void)
     util_log("Not jumping to firmware");
     _compute_is_app_flash_empty();
     bootloader_render_default_screen();
-    if (usb_start(_api_setup) != ERR_NONE) {
+    _api_setup();
+    if (usb_start() != ERR_NONE) {
         _render_message("Failed to initialize USB", 0);
     }
 }

src/bootloader/startup.c

@@ -23,6 +23,7 @@
 #include <string.h>
 #include <usb/class/hid/hww/hid_hww.h>
 #include <usb/usb_processing.h>
+#include <bootloader/bootloader_version.h>
 
 #if defined(BOOTLOADER_DEVDEVICE) || PLATFORM_BITBOX02PLUS == 1
 #include <qtouch.h>
@@ -34,7 +35,15 @@
 #include <da14531/da14531_protocol.h>
 #include <uart.h>
 #include <utils_ringbuffer.h>
+
+#if PRODUCT_BITBOX_PLUS_MULTI == 1
+#define DEVICE_MODE "{\"p\":\"bb02p-bl-multi\",\"v\":\"" BOOTLOADER_VERSION "\"}"
+#elif PRODUCT_BITBOX_PLUS_BTCONLY == 1
+#define DEVICE_MODE "{\"p\":\"bb02p-bl-btconly\",\"v\":\"" BOOTLOADER_VERSION "\"}"
+#else
+#error "unknown product"
 #endif
+#endif // PLATFORM_BITBOX02PLUS == 1
 
 extern void __attribute__((noreturn)) __stack_chk_fail(void);
 void __attribute__((noreturn)) __stack_chk_fail(void)
@@ -85,10 +94,15 @@ int main(void)
     uint16_t uart_read_buf_len = 0;
 
     ringbuffer_init(&uart_write_queue, &uart_write_buf, UART_OUT_BUF_LEN);
-#define DEVICE_MODE "{\"p\":\"bb02p-bl-multi\",\"v\":\"1.1.0\"}"
-    da14531_set_product(DEVICE_MODE, sizeof(DEVICE_MODE) - 1, &uart_write_queue);
     bool usb_hww_request_seen = false;
 
+    // Set product to bootloader string, this is necessary if we have rebooted from firmware. Must
+    // be done after usb_processing is initalized to avoid getting request from the app to early.
+    da14531_handler_current_product = (const uint8_t*)DEVICE_MODE;
+    da14531_handler_current_product_len = sizeof(DEVICE_MODE) - 1;
+    da14531_set_product(
+        da14531_handler_current_product, da14531_handler_current_product_len, &uart_write_queue);
+
     da14531_protocol_init();
 #endif
     usb_processing_init();

src/da14531/da14531.c

@@ -45,11 +45,16 @@ void da14531_power_down(struct ringbuffer* uart_out)
     }
 }
 
-void da14531_set_product(const char* product, uint16_t product_len, struct ringbuffer* uart_out)
+void da14531_set_product(
+    volatile const uint8_t* product,
+    volatile uint16_t product_len,
+    struct ringbuffer* uart_out)
 {
     uint8_t payload[64] = {0};
     payload[0] = CTRL_CMD_PRODUCT_STRING;
-    memcpy(&payload[1], product, product_len);
+    for (int i = 0; i < product_len; i++) {
+        payload[1 + i] = product[i];
+    }
     uint8_t tmp[128];
     uint16_t tmp_len = da14531_protocol_format(
         &tmp[0], sizeof(tmp), DA14531_PROTOCOL_PACKET_TYPE_CTRL_DATA, &payload[0], 1 + product_len);

src/da14531/da14531.h

@@ -36,9 +36,12 @@ void da14531_power_down(struct ringbuffer* uart_out);
 
 void da14531_reset(struct ringbuffer* uart_out);
 
-// product is an array of characters to be set as product characteristic
+// product is an array of characters to be set as product characteristic (not null terminated)
 // procuct_len is the number of characters in the product array
 // uart_out is the queue where to put the outgoing serially encoded bytes
-void da14531_set_product(const char* product, uint16_t product_len, struct ringbuffer* uart_out);
+void da14531_set_product(
+    volatile const uint8_t* product,
+    volatile uint16_t product_len,
+    struct ringbuffer* uart_out);
 
 #endif

src/da14531/da14531_handler.c

@@ -25,6 +25,11 @@
 #include <ui/components/ui_images.h>
 #include <ui/fonts/monogram_5X9.h>
 
+// These are set from interrupt context in the orientation workflow :/ therefore they need to be
+// volatile
+volatile const uint8_t* da14531_handler_current_product = NULL;
+volatile uint16_t da14531_handler_current_product_len = 0;
+
 struct da14531_ctrl_frame {
     enum da14531_protocol_packet_type type;
     uint16_t payload_length; // includes length of cmd
@@ -205,12 +210,8 @@ static void _ctrl_handler(struct da14531_ctrl_frame* frame, struct ringbuffer* q
     } break;
     case CTRL_CMD_PRODUCT_STRING: {
         // util_log("da14531: get device mode");
-#if defined(BOOTLOADER)
-#define DEVICE_MODE "{\"p\":\"bb02p-bl-multi\",\"v\":\"1.1.0\"}"
-#else
-#define DEVICE_MODE "{\"p\":\"bb02p-multi\",\"v\":\"9.22.0\"}"
-#endif
-        da14531_set_product(DEVICE_MODE, sizeof(DEVICE_MODE) - 1, queue);
+        da14531_set_product(
+            da14531_handler_current_product, da14531_handler_current_product_len, queue);
     } break;
     case CTRL_CMD_IDENTITY_ADDRESS: {
         // util_log("da14531: get addr");

src/da14531/da14531_handler.h

@@ -18,6 +18,9 @@
 #include "da14531_protocol.h"
 #include <utils_ringbuffer.h>
 
+extern volatile const uint8_t* da14531_handler_current_product;
+extern volatile uint16_t da14531_handler_current_product_len;
+
 void da14531_handler(struct da14531_protocol_frame* frame, struct ringbuffer* queue);
 
 #endif

src/da14531/da14531_protocol.c

@@ -242,7 +242,12 @@ static struct da14531_protocol_frame* _serial_link_in_poll(
     case SERIAL_LINK_STATE_READING: {
         int len = self->buf_in_len;
         for (int i = 0; i < len; i++) {
-            // util_log("i:%d,b:%02x,l:%u", i, self->buf_in[i], self->buf_in_len);
+            // util_log(
+            //     "i:%d,b:%02x,l:%u,h:%s",
+            //     i,
+            //     self->buf_in[i],
+            //     self->buf_in_len,
+            //     util_dbg_hex(&self->frame[0], 10));
             self->buf_in_len--;
             // Reset firmware loader on STX
             if (self->buf_in[i] == STX) {

src/firmware_main_loop.c

@@ -47,21 +47,17 @@
 
 void firmware_main_loop(void)
 {
-    // This starts the async orientation screen workflow, which is processed by the loop below.
-    orientation_screen();
-
-    // TODO: Send out new BLE product string, so app sees that we are booted
-    // Set it to the size of the ringbuffer in the UART driver so we can read out all bytes
+    // Set the size of uart_read_buf to the size of the ringbuffer in the UART driver so we can read
+    // out all bytes
     uint8_t uart_read_buf[USART_0_BUFFER_SIZE] = {0};
     uint16_t uart_read_buf_len = 0;
 
     struct ringbuffer uart_write_queue;
     uint8_t uart_write_buf[UART_OUT_BUF_LEN];
     ringbuffer_init(&uart_write_queue, &uart_write_buf, UART_OUT_BUF_LEN);
 
-// Immediately enqueue the new firmware product string
-#define DEVICE_MODE "{\"p\":\"bb02p-multi\",\"v\":\"9.22.0\"}"
-    da14531_set_product(DEVICE_MODE, sizeof(DEVICE_MODE) - 1, &uart_write_queue);
+    // This starts the async orientation screen workflow, which is processed by the loop below.
+    orientation_screen(&uart_write_queue);
 
     const uint8_t* hww_data = NULL;
     uint8_t hww_frame[USB_REPORT_SIZE] = {0};

src/system.c

@@ -28,7 +28,7 @@ static void _ble_clear_product(void)
     struct ringbuffer uart_queue;
     uint8_t uart_queue_buf[64];
     ringbuffer_init(&uart_queue, &uart_queue_buf[0], sizeof(uart_queue_buf));
-    da14531_set_product("", 0, &uart_queue);
+    da14531_set_product(NULL, 0, &uart_queue);
     while (ringbuffer_num(&uart_queue)) {
 #ifndef TESTING
         uart_poll(NULL, 0, NULL, &uart_queue);