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 #![no_main]
  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 #[cfg(bootloader_type = "intermediate")]
  69 fn bootloader_requested() -> bool {
  70     true
  71 }
  72
  73 #[cfg(bootloader_type = "final")]
  74 fn bootloader_requested() -> bool {
  75     let start_ticks = systick::now();
  76
  77     while !systick::has_timeout_ms(start_ticks, 3000) {
  78         if (gpio::get(gpio::GPIOE) & (1 << 24)) == 0 {
  79             return true;
  80         }
  81     }
  82
  83     false
  84 }
  85
  86 #[inline(never)]
  87 #[panic_handler]
  88 fn panic(_info: &core::panic::PanicInfo) -> ! {
  89     loop {
  90     }
  91 }
  92
  93 #[no_mangle]
  94 pub unsafe extern "C" fn _start() -> ! {
  95     watchdog::disable();
  96     clock::configure();
  97     systick::init();
  98     port::init();
  99
 100     // Configure lower left push button.
 101     gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
 102     port::set_af(port::PORTE, 24, 1);
 103     port::set_pull(port::PORTE, 24, port::Pull::Up);
 104
 105     enable_interrupts();
 106
 107     if application_missing() || bootloader_requested() {
 108         usb_serial::init(0xf055, USB_PID);
 109
 110         let mut b = bootloader::Bootloader::new();
 111
 112         while b.run() || application_missing() {
 113         }
 114
 115         usb_serial::shutdown();
 116     }
 117
 118     disable_interrupts();
 119
 120     clock::reset();
 121
 122     jump_to_application(APPLICATION_ADDR);
 123 }