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::watchdog;
  39 use common::usb_serial;
  40
  41 type Reg32 = register::Register<u32>;
  42
  43 extern {
  44     fn enable_interrupts();
  45     fn disable_interrupts();
  46
  47     fn jump_to_application(address: u32);
  48 }
  49
  50 #[cfg(bootloader_type = "intermediate")]
  51 const APPLICATION_ADDR: u32 = 0x0;
  52
  53 #[cfg(bootloader_type = "final")]
  54 const APPLICATION_ADDR: u32 = 0x8000;
  55
  56 #[cfg(bootloader_type = "intermediate")]
  57 const USB_PID: u16 = 0x635b;
  58
  59 #[cfg(bootloader_type = "final")]
  60 const USB_PID: u16 = 0x635c;
  61
  62 unsafe fn application_missing() -> bool {
  63     let first_app_word = Reg32::new(APPLICATION_ADDR);
  64
  65     first_app_word.read() == 0xffffffff
  66 }
  67
  68 fn bootloader_requested() -> bool {
  69     let start_ticks = systick::now();
  70
  71     while !systick::has_timeout_ms(start_ticks, 3000) {
  72         if (gpio::get(gpio::GPIOE) & (1 << 24)) == 0 {
  73             return true;
  74         }
  75     }
  76
  77     false
  78 }
  79
  80 #[no_mangle]
  81 pub unsafe extern fn main() {
  82     watchdog::disable();
  83     clock::configure();
  84     systick::init();
  85     port::init();
  86
  87     // Configure lower left push button.
  88     gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
  89     port::set_af(port::PORTE, 24, 1);
  90     port::set_pull(port::PORTE, 24, port::Pull::Up);
  91
  92     enable_interrupts();
  93
  94     if application_missing() || bootloader_requested() {
  95         usb_serial::init(0xf055, USB_PID);
  96
  97         let mut b = bootloader::Bootloader::new();
  98
  99         while b.run() || application_missing() {
 100         }
 101
 102         usb_serial::shutdown();
 103     }
 104
 105     disable_interrupts();
 106
 107     clock::reset();
 108
 109     jump_to_application(APPLICATION_ADDR);
 110 }