commit 51c9d707fbab00e2ba419894eb790385fa322fb0
author Rajavardhan Gundi <rajavardhan.gundi@intel.com> Wed Feb 20 14:08:52 2019 +0530
committer Andrzej Puzdrowski <andrzej.puzdrowski@nordicsemi.no> Fri Mar 08 16:31:48 2019 +0100
tree 9142b20d55642dc66569745336cdc59b808e8a2c
parent 048168aa377e40f5e103a48d634a1c7b7035bedd

dfu: Introduce CONFIG_BOOT_WAIT_FOR_USB_DFU

This Kconfig parameter makes MCUboot wait for a prescribed duration
of time to allow for USB DFU to be invoked.

Signed-off-by: Rajavardhan Gundi <rajavardhan.gundi@intel.com>

boot/zephyr/main.c

diff --git a/boot/zephyr/main.c b/boot/zephyr/main.c
index 236f80c..e276454 100644
--- a/boot/zephyr/main.c
+++ b/boot/zephyr/main.c
@@ -40,6 +40,10 @@
 };
 #endif
 
+#ifdef CONFIG_BOOT_WAIT_FOR_USB_DFU
+#include <usb/class/usb_dfu.h>
+#endif
+
 MCUBOOT_LOG_MODULE_REGISTER(mcuboot);
 
 void os_heap_init(void);
@@ -197,6 +201,12 @@
     }
 #endif
 
+#ifdef CONFIG_BOOT_WAIT_FOR_USB_DFU
+    BOOT_LOG_INF("Waiting for USB DFU");
+    wait_for_usb_dfu();
+    BOOT_LOG_INF("USB DFU wait time elapsed");
+#endif
+
     rc = boot_go(&rsp);
     if (rc != 0) {
         BOOT_LOG_ERR("Unable to find bootable image");