a34331ae5f
* Allow manipulating state without accessing DFU partition. * Provide aligned buffer when creating updater to reduce potential wrong parameters passed.
74 lines
2.8 KiB
Rust
74 lines
2.8 KiB
Rust
#![no_std]
|
|
#![no_main]
|
|
#![macro_use]
|
|
#![feature(type_alias_impl_trait)]
|
|
|
|
use embassy_boot_nrf::{FirmwareUpdater, FirmwareUpdaterConfig};
|
|
use embassy_embedded_hal::adapter::BlockingAsync;
|
|
use embassy_executor::Spawner;
|
|
use embassy_nrf::gpio::{Input, Level, Output, OutputDrive, Pull};
|
|
use embassy_nrf::nvmc::Nvmc;
|
|
use embassy_nrf::wdt::{self, Watchdog};
|
|
use embassy_sync::mutex::Mutex;
|
|
use panic_reset as _;
|
|
|
|
#[cfg(feature = "skip-include")]
|
|
static APP_B: &[u8] = &[0, 1, 2, 3];
|
|
#[cfg(not(feature = "skip-include"))]
|
|
static APP_B: &[u8] = include_bytes!("../../b.bin");
|
|
|
|
#[embassy_executor::main]
|
|
async fn main(_spawner: Spawner) {
|
|
let p = embassy_nrf::init(Default::default());
|
|
|
|
let mut button = Input::new(p.P0_11, Pull::Up);
|
|
let mut led = Output::new(p.P0_13, Level::Low, OutputDrive::Standard);
|
|
|
|
//let mut led = Output::new(p.P1_10, Level::Low, OutputDrive::Standard);
|
|
//let mut button = Input::new(p.P1_02, Pull::Up);
|
|
|
|
// nRF91 DK
|
|
// let mut led = Output::new(p.P0_02, Level::Low, OutputDrive::Standard);
|
|
// let mut button = Input::new(p.P0_06, Pull::Up);
|
|
|
|
// The following code block illustrates how to obtain a watchdog that is configured
|
|
// as per the existing watchdog. Ordinarily, we'd use the handle returned to "pet" the
|
|
// watchdog periodically. If we don't, and we're not going to for this example, then
|
|
// the watchdog will cause the device to reset as per its configured timeout in the bootloader.
|
|
// This helps is avoid a situation where new firmware might be bad and block our executor.
|
|
// If firmware is bad in this way then the bootloader will revert to any previous version.
|
|
let wdt_config = wdt::Config::try_new(&p.WDT).unwrap();
|
|
let (_wdt, [_wdt_handle]) = match Watchdog::try_new(p.WDT, wdt_config) {
|
|
Ok(x) => x,
|
|
Err(_) => {
|
|
// Watchdog already active with the wrong number of handles, waiting for it to timeout...
|
|
loop {
|
|
cortex_m::asm::wfe();
|
|
}
|
|
}
|
|
};
|
|
|
|
let nvmc = Nvmc::new(p.NVMC);
|
|
let nvmc = Mutex::new(BlockingAsync::new(nvmc));
|
|
|
|
let config = FirmwareUpdaterConfig::from_linkerfile(&nvmc);
|
|
let mut magic = [0; 4];
|
|
let mut updater = FirmwareUpdater::new(config, &mut magic);
|
|
loop {
|
|
led.set_low();
|
|
button.wait_for_any_edge().await;
|
|
if button.is_low() {
|
|
let mut offset = 0;
|
|
for chunk in APP_B.chunks(4096) {
|
|
let mut buf: [u8; 4096] = [0; 4096];
|
|
buf[..chunk.len()].copy_from_slice(chunk);
|
|
updater.write_firmware(offset, &buf).await.unwrap();
|
|
offset += chunk.len();
|
|
}
|
|
updater.mark_updated().await.unwrap();
|
|
led.set_high();
|
|
cortex_m::peripheral::SCB::sys_reset();
|
|
}
|
|
}
|
|
}
|