bootloader: Use distinct USB PIDs depending on location in flash.
[gps-watch.git] / src / bootloader / main.rs
1 /*
2  * Copyright (c) 2019 Tilman Sauerbeck (tilman at code-monkey de)
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining
5  * a copy of this software and associated documentation files (the
6  * "Software"), to deal in the Software without restriction, including
7  * without limitation the rights to use, copy, modify, merge, publish,
8  * distribute, sublicense, and/or sell copies of the Software, and to
9  * permit persons to whom the Software is furnished to do so, subject to
10  * the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be
13  * included in all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
19  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
20  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
21  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
22  */
23
24 #![no_std]
25 #![crate_type="staticlib"]
26 #[link(name="libcommon.rlib")]
27
28 extern crate common;
29
30 mod flash;
31 mod bootloader;
32
33 use common::register;
34 use common::clock;
35 use common::systick;
36 use common::port;
37 use common::gpio;
38 use common::usb_serial;
39
40 type Reg32 = register::Register<u32>;
41
42 extern {
43     fn jump_to_application(address: u32);
44 }
45
46 #[cfg(bootloader_type = "intermediate")]
47 const APPLICATION_ADDR: u32 = 0x0;
48
49 #[cfg(bootloader_type = "final")]
50 const APPLICATION_ADDR: u32 = 0x8000;
51
52 #[cfg(bootloader_type = "intermediate")]
53 const USB_PID: u16 = 0x635b;
54
55 #[cfg(bootloader_type = "final")]
56 const USB_PID: u16 = 0x635c;
57
58 unsafe fn application_missing() -> bool {
59     let first_app_word = Reg32::new(APPLICATION_ADDR);
60
61     first_app_word.read() == 0xffffffff
62 }
63
64 fn bootloader_requested() -> bool {
65     let start_ticks = systick::now();
66
67     while !systick::has_timeout_ms(start_ticks, 3000) {
68         if (gpio::get(gpio::GPIOE) & (1 << 24)) == 0 {
69             return true;
70         }
71     }
72
73     false
74 }
75
76 #[no_mangle]
77 pub unsafe extern fn main() {
78     clock::configure();
79     systick::init();
80     port::init();
81
82     // Configure lower left push button.
83     gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
84     port::set_af(port::PORTE, 24, 1);
85     port::set_pull(port::PORTE, 24, port::Pull::Up);
86
87     if application_missing() || bootloader_requested() {
88         usb_serial::init(0xf055, USB_PID);
89
90         let mut b = bootloader::Bootloader::new();
91
92         while b.run() || application_missing() {
93         }
94
95         usb_serial::shutdown();
96     }
97
98     jump_to_application(APPLICATION_ADDR);
99 }