bootloader: Use distinct USB PIDs depending on location in flash.

[gps-watch.git] / src / bootloader / main.rs

diff --git a/src/bootloader/main.rs b/src/bootloader/main.rs
index 8c6f149e19534d93326a7df4abf290a8a37044ed..ca459fdc2eccf1c406b52b6423e8d40aab307be9 100644 (file)
--- a/src/bootloader/main.rs
+++ b/src/bootloader/main.rs
@@ -49,6 +49,12 @@ const APPLICATION_ADDR: u32 = 0x0;
 #[cfg(bootloader_type = "final")]
 const APPLICATION_ADDR: u32 = 0x8000;
 
+#[cfg(bootloader_type = "intermediate")]
+const USB_PID: u16 = 0x635b;
+
+#[cfg(bootloader_type = "final")]
+const USB_PID: u16 = 0x635c;
+
 unsafe fn application_missing() -> bool {
     let first_app_word = Reg32::new(APPLICATION_ADDR);
 
@@ -71,6 +77,7 @@ fn bootloader_requested() -> bool {
 pub unsafe extern fn main() {
     clock::configure();
     systick::init();
+    port::init();
 
     // Configure lower left push button.
     gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
@@ -78,12 +85,14 @@ pub unsafe extern fn main() {
     port::set_pull(port::PORTE, 24, port::Pull::Up);
 
     if application_missing() || bootloader_requested() {
-        usb_serial::init(0xf055, 0x635c);
+        usb_serial::init(0xf055, USB_PID);
 
         let mut b = bootloader::Bootloader::new();
 
         while b.run() || application_missing() {
         }
+
+        usb_serial::shutdown();
     }
 
     jump_to_application(APPLICATION_ADDR);