1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#![no_std]
#![no_main]
#![macro_use]
#[cfg(feature = "defmt")]
use defmt_rtt as _;
use embassy_boot::State;
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());
#[cfg(not(feature = "nrf54"))]
let mut button = Input::new(p.P0_11, Pull::Up);
#[cfg(not(feature = "nrf54"))]
let mut led = Output::new(p.P0_13, Level::Low, OutputDrive::Standard);
#[cfg(not(feature = "nrf54"))]
let mut led_reverted = Output::new(p.P0_14, Level::High, OutputDrive::Standard);
// nRF54 DK
#[cfg(feature = "nrf54")]
let mut button = Input::new(p.P1_13, Pull::Up);
#[cfg(feature = "nrf54")]
let mut led = Output::new(p.P1_14, Level::Low, OutputDrive::Standard);
#[cfg(feature = "nrf54")]
let mut led_reverted = Output::new(p.P2_09, Level::High, 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.
#[cfg(feature = "nrf54")]
let wdt = p.WDT0;
#[cfg(not(feature = "nrf54"))]
let wdt = p.WDT;
let wdt_config = wdt::Config::try_new(&wdt).unwrap();
let (_wdt, [_wdt_handle]) = match Watchdog::try_new(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();
}
}
};
// RRAMC for nRF54
#[cfg(feature = "nrf54")]
let nvmc = Nvmc::new(p.RRAMC);
#[cfg(not(feature = "nrf54"))]
let nvmc = Nvmc::new(p.NVMC);
let nvmc = Mutex::new(BlockingAsync::new(nvmc));
let config = FirmwareUpdaterConfig::from_linkerfile(&nvmc, &nvmc);
let mut magic = [0; 16];
let mut updater = FirmwareUpdater::new(config, &mut magic);
let state = updater.get_state().await.unwrap();
if state == State::Revert {
led_reverted.set_low();
} else {
led_reverted.set_high();
}
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();
}
}
}
|
