14 Commits
22 changed files with 1275 additions and 175 deletions
Generated
+59
View File
@@ -53,6 +53,12 @@ version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c583acf993cf4245c4acb0a2cc2ab1f9cc097de73411bb6d3647ff6af2b1013d" checksum = "c583acf993cf4245c4acb0a2cc2ab1f9cc097de73411bb6d3647ff6af2b1013d"
[[package]]
name = "array-init"
version = "2.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3d62b7694a562cdf5a74227903507c56ab2cc8bdd1f781ed5cb4cf9c9f810bfc"
[[package]] [[package]]
name = "autocfg" name = "autocfg"
version = "1.5.1" version = "1.5.1"
@@ -96,6 +102,29 @@ dependencies = [
"vsimd", "vsimd",
] ]
[[package]]
name = "binary_serde"
version = "1.0.25"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c4b2fe974c561ed04fb89673dd221bed36fc4794b71ca6ae96ffe19949045686"
dependencies = [
"array-init",
"binary_serde_macros",
"recursive_array",
"thiserror-no-std",
]
[[package]]
name = "binary_serde_macros"
version = "1.0.25"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4790b8dbcfa9d3adf2020ebf5fd36c6bc89a77d434131615ea349bc9620e8831"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.119",
]
[[package]] [[package]]
name = "bitfield" name = "bitfield"
version = "0.17.0" version = "0.17.0"
@@ -319,6 +348,7 @@ name = "creaturedex"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"base64 0.23.1", "base64 0.23.1",
"binary_serde",
"ch1115", "ch1115",
"critical-section", "critical-section",
"display-interface-spi", "display-interface-spi",
@@ -332,17 +362,20 @@ dependencies = [
"embedded-hal-async", "embedded-hal-async",
"embedded-hal-bus", "embedded-hal-bus",
"embedded-io 0.7.1", "embedded-io 0.7.1",
"embedded-storage",
"esp-alloc", "esp-alloc",
"esp-bootloader-esp-idf", "esp-bootloader-esp-idf",
"esp-hal", "esp-hal",
"esp-hal-wifimanager", "esp-hal-wifimanager",
"esp-println", "esp-println",
"esp-rtos", "esp-rtos",
"esp-storage",
"i2c-character-display", "i2c-character-display",
"ili9341", "ili9341",
"log", "log",
"num_enum", "num_enum",
"pn532", "pn532",
"static_assertions",
] ]
[[package]] [[package]]
@@ -2796,6 +2829,12 @@ dependencies = [
"crossbeam-utils", "crossbeam-utils",
] ]
[[package]]
name = "recursive_array"
version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "69bad83913c3b39011ad9d43b7ac0cae139cec5d7e7288fbea5bb4b4e3cc78f6"
[[package]] [[package]]
name = "redox_syscall" name = "redox_syscall"
version = "0.5.18" version = "0.5.18"
@@ -3273,6 +3312,26 @@ dependencies = [
"syn 3.0.3", "syn 3.0.3",
] ]
[[package]]
name = "thiserror-impl-no-std"
version = "2.0.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "58e6318948b519ba6dc2b442a6d0b904ebfb8d411a3ad3e07843615a72249758"
dependencies = [
"proc-macro2",
"quote",
"syn 1.0.109",
]
[[package]]
name = "thiserror-no-std"
version = "2.0.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a3ad459d94dd517257cc96add8a43190ee620011bb6e6cdc82dafd97dfafafea"
dependencies = [
"thiserror-impl-no-std",
]
[[package]] [[package]]
name = "thread_local" name = "thread_local"
version = "1.1.10" version = "1.1.10"
+13 -3
View File
@@ -9,7 +9,11 @@ name = "creaturedex"
path = "./src/bin/main.rs" path = "./src/bin/main.rs"
[dependencies] [dependencies]
esp-hal = { version = "~1.1.0", features = ["esp32s3", "log-04", "unstable"] } #, "embassy", "embassy-time-timg0", "embassy-executor-thread"] } esp-hal = { version = "~1.1.0", features = [
"esp32s3",
"log-04",
"unstable",
] } #, "embassy", "embassy-time-timg0", "embassy-executor-thread"] }
esp-rtos = { version = "0.3.0", features = [ esp-rtos = { version = "0.3.0", features = [
"embassy", "embassy",
@@ -24,8 +28,7 @@ log = "0.4.27"
critical-section = "1.2.0" critical-section = "1.2.0"
embedded-io = "0.7.1" embedded-io = "0.7.1"
embassy-executor = { version = "0.10.0", features = [ embassy-executor = { version = "0.10.0", features = [] }
] }
embassy-time = { version = "0.5.1", default-features = false } embassy-time = { version = "0.5.1", default-features = false }
embassy-futures = "0.1.2" embassy-futures = "0.1.2"
esp-alloc = "0.10.0" esp-alloc = "0.10.0"
@@ -47,6 +50,13 @@ embassy-sync = "0.8.0"
embedded-hal-async = "1.0.0" embedded-hal-async = "1.0.0"
i2c-character-display = "0.5.1" i2c-character-display = "0.5.1"
ch1115 = { version = "0.1.2", features = ["graphics"] } ch1115 = { version = "0.1.2", features = ["graphics"] }
esp-storage = "0.9.0"
embedded-storage = "0.3.1"
binary_serde = "1.0.25"
static_assertions = { version = "1.1.0", default-features = false }
[build-dependencies]
log = "0.4.27"
[patch.crates-io] [patch.crates-io]
+163
View File
@@ -1,4 +1,5 @@
fn main() { fn main() {
generate_filter_snippet();
linker_be_nice(); linker_be_nice();
// make sure linkall.x is the last linker script (otherwise might cause problems with flip-link) // make sure linkall.x is the last linker script (otherwise might cause problems with flip-link)
println!("cargo:rustc-link-arg=-Tlinkall.x"); println!("cargo:rustc-link-arg=-Tlinkall.x");
@@ -68,3 +69,165 @@ fn linker_be_nice() {
std::env::current_exe().unwrap().display() std::env::current_exe().unwrap().display()
); );
} }
// Taken over from esp_println because impossible to just extend their existing env logger.
// https://github.com/esp-rs/esp-hal/blob/8fa6d7e985e724d6ad998e34307429b0920da370/esp-println/build.rs
fn generate_filter_snippet() {
use log::LevelFilter;
use std::{env, path::Path};
let out_dir = env::var_os("OUT_DIR").unwrap();
let dest_path = Path::new(&out_dir).join("log_filter.rs");
let filter = env::var("ESP_LOG");
let snippet = if let Ok(filter) = filter {
let res = parse_spec(&filter);
if !res.errors.is_empty() {
panic!("Error parsing `ESP_LOG`: {:?}", res.errors);
} else {
let max = res
.directives
.iter()
.map(|v| v.level)
.max()
.unwrap_or(LevelFilter::Off);
let max = match max {
LevelFilter::Off => "Off",
LevelFilter::Error => "Error",
LevelFilter::Warn => "Warn",
LevelFilter::Info => "Info",
LevelFilter::Debug => "Debug",
LevelFilter::Trace => "Trace",
};
let mut snippet = String::new();
snippet.push_str(&format!(
"pub(crate) const FILTER_MAX: log::LevelFilter = log::LevelFilter::{max};"
));
snippet
.push_str("pub(crate) fn is_enabled(level: log::Level, _target: &str) -> bool {");
let mut global_level = None;
for directive in res.directives {
let level = match directive.level {
LevelFilter::Off => "Off",
LevelFilter::Error => "Error",
LevelFilter::Warn => "Warn",
LevelFilter::Info => "Info",
LevelFilter::Debug => "Debug",
LevelFilter::Trace => "Trace",
};
if let Some(name) = directive.name {
// If a prefix matches, don't continue to the next directive
snippet.push_str(&format!(
"if _target.starts_with(\"{name}\") {{ return level <= log::LevelFilter::{level}; }}"
));
} else {
if global_level.is_some() {
panic!("Multiple global log levels specified in `ESP_LOG`");
}
global_level = Some(level);
}
}
// Place the fallback rule at the end
if let Some(level) = global_level {
snippet.push_str(&format!("level <= log::LevelFilter::{level}"));
} else {
snippet.push_str(" false");
}
snippet.push('}');
snippet
}
} else {
"pub(crate) const FILTER_MAX: log::LevelFilter = log::LevelFilter::Off; pub(crate) fn is_enabled(_level: log::Level, _target: &str) -> bool { true }".to_string()
};
std::fs::write(&dest_path, &snippet).unwrap();
}
#[derive(Default, Debug)]
struct ParseResult {
pub(crate) directives: Vec<Directive>,
pub(crate) errors: Vec<String>,
}
impl ParseResult {
fn add_directive(&mut self, directive: Directive) {
self.directives.push(directive);
}
fn add_error(&mut self, message: String) {
self.errors.push(message);
}
}
#[derive(Debug)]
struct Directive {
pub(crate) name: Option<String>,
pub(crate) level: log::LevelFilter,
}
/// Parse a logging specification string (e.g:
/// `crate1,crate2::mod3,crate3::x=error/foo`) and return a vector with log
/// directives.
fn parse_spec(spec: &str) -> ParseResult {
use log::LevelFilter;
let mut result = ParseResult::default();
let mut parts = spec.split('/');
let mods = parts.next();
if let Some(m) = mods {
for s in m.split(',').map(|ss| ss.trim()) {
if s.is_empty() {
continue;
}
let mut parts = s.split('=');
let (log_level, name) =
match (parts.next(), parts.next().map(|s| s.trim()), parts.next()) {
(Some(part0), None, None) => {
// if the single argument is a log-level string or number,
// treat that as a global fallback
match part0.parse() {
Ok(num) => (num, None),
Err(_) => (LevelFilter::max(), Some(part0)),
}
}
(Some(part0), Some(""), None) => (LevelFilter::max(), Some(part0)),
(Some(part0), Some(part1), None) => {
if let Ok(num) = part1.parse() {
(num, Some(part0))
} else {
result.add_error(format!("invalid logging spec '{part1}'"));
continue;
}
}
_ => {
result.add_error(format!("invalid logging spec '{s}'"));
continue;
}
};
result.add_directive(Directive {
name: name.map(|s| s.to_owned()),
level: log_level,
});
}
}
// Sort by length so that the most specific prefixes come first
result
.directives
.sort_by(|a, b| match (a.name.as_ref(), b.name.as_ref()) {
(Some(a), Some(b)) => b.len().cmp(&a.len()),
_ => std::cmp::Ordering::Equal,
});
result
}
+22 -6
View File
@@ -1,4 +1,6 @@
use crate::storage::cardstore::CardStore;
use embedded_hal_bus::i2c::AtomicDevice; use embedded_hal_bus::i2c::AtomicDevice;
use esp_hal::rtc_cntl::Rtc;
use esp_hal::{Blocking, i2c::master::I2c, time::Instant}; use esp_hal::{Blocking, i2c::master::I2c, time::Instant};
use crate::{ use crate::{
@@ -14,6 +16,8 @@ use crate::{
)] )]
pub async fn nfc_scanner( pub async fn nfc_scanner(
mut nfc_driver: NfcPn532Driver<AtomicDevice<'static, I2c<'static, Blocking>>>, mut nfc_driver: NfcPn532Driver<AtomicDevice<'static, I2c<'static, Blocking>>>,
mut card_store: CardStore,
rtc: Rtc<'static>,
) { ) {
loop { loop {
while CARD_DATA.lock().await.is_none() { while CARD_DATA.lock().await.is_none() {
@@ -24,10 +28,22 @@ pub async fn nfc_scanner(
log::error!("Failed to split NFC card data"); log::error!("Failed to split NFC card data");
continue; continue;
}; };
CARD_DATA let card = match Card::try_from(split_data) {
.lock() Ok(card) => card,
.await Err(err) => {
.replace(Card::try_from(split_data).unwrap()); log::error!("Invalid card data: {err}");
continue;
}
};
CARD_DATA.lock().await.replace(card);
match card_store.save_raw_card(split_data, rtc.current_time_us()) {
Ok(()) => {}
Err(error) => {
log::error!("Error storing card in flash: {:?}", error)
}
}
let elapsed_ms = start.elapsed().as_millis(); let elapsed_ms = start.elapsed().as_millis();
log::info!( log::info!(
"NFC read duration: {} ms, card data length: {}", "NFC read duration: {} ms, card data length: {}",
@@ -35,8 +51,8 @@ pub async fn nfc_scanner(
card_data.len() card_data.len()
); );
} }
Err(e) => { Err(_) => {
log::error!("NFC read failed: {e}"); log::info!("No NFC card found");
} }
} }
} }
+53 -13
View File
@@ -12,11 +12,14 @@ use creaturedex::drivers::i2c_bus::I2cBusPinConfiguration;
use creaturedex::drivers::nfc_pn532::NfcPn532Driver; use creaturedex::drivers::nfc_pn532::NfcPn532Driver;
use creaturedex::drivers::secondary_oled::SecondaryDisplay; use creaturedex::drivers::secondary_oled::SecondaryDisplay;
use creaturedex::drivers::spi_bus; use creaturedex::drivers::spi_bus;
use creaturedex::drivers::tertiary_lcd::{TertiaryDisplay, init_tertiary_lcd, write_wrapped}; use creaturedex::drivers::tertiary_lcd::{
BOX, EMPTY, EXAMPLE, HEART, TertiaryDisplay, TertiaryDisplayPinConfiguration,
};
use creaturedex::navigation::inputs::Inputs; use creaturedex::navigation::inputs::Inputs;
use creaturedex::navigation::navigation::{self}; use creaturedex::navigation::navigation::{self};
use creaturedex::navigation::outputs::Outputs; use creaturedex::navigation::outputs::Outputs;
use creaturedex::network::wifi::setup_wifi; use creaturedex::storage::cardstore::CardStore;
use creaturedex::storage::store::Store;
use embassy_executor::Spawner; use embassy_executor::Spawner;
use embedded_graphics::pixelcolor::BinaryColor; use embedded_graphics::pixelcolor::BinaryColor;
use embedded_graphics::{ use embedded_graphics::{
@@ -30,6 +33,8 @@ use esp_hal::gpio::{Input, InputConfig, Pull};
use esp_hal::timer::timg::TimerGroup; use esp_hal::timer::timg::TimerGroup;
use log::error; use log::error;
use esp_hal::rtc_cntl::Rtc;
#[panic_handler] #[panic_handler]
fn panic(panic_info: &core::panic::PanicInfo) -> ! { fn panic(panic_info: &core::panic::PanicInfo) -> ! {
error!("{}", panic_info); error!("{}", panic_info);
@@ -50,7 +55,7 @@ esp_bootloader_esp_idf::esp_app_desc!();
)] )]
#[esp_rtos::main] #[esp_rtos::main]
async fn main(spawner: Spawner) { async fn main(spawner: Spawner) {
esp_println::logger::init_logger_from_env(); creaturedex::logging::init();
let config = esp_hal::Config::default().with_cpu_clock(CpuClock::max()); let config = esp_hal::Config::default().with_cpu_clock(CpuClock::max());
let peripherals = esp_hal::init(config); let peripherals = esp_hal::init(config);
@@ -62,7 +67,9 @@ async fn main(spawner: Spawner) {
let timg0 = TimerGroup::new(peripherals.TIMG0); let timg0 = TimerGroup::new(peripherals.TIMG0);
esp_rtos::start(timg0.timer0, sw_interrupt.software_interrupt0); esp_rtos::start(timg0.timer0, sw_interrupt.software_interrupt0);
let (primary, mut secondaries) = spi_bus::DisplayPinConfiguration { let (primary, mut secondaries) = retry(
5,
spi_bus::DisplayPinConfiguration {
spi_peripheral: peripherals.SPI2, spi_peripheral: peripherals.SPI2,
sck: peripherals.GPIO36, sck: peripherals.GPIO36,
mosi: peripherals.GPIO35, mosi: peripherals.GPIO35,
@@ -74,27 +81,43 @@ async fn main(spawner: Spawner) {
reset_primary: peripherals.GPIO10, reset_primary: peripherals.GPIO10,
reset_secondary: peripherals.GPIO47, reset_secondary: peripherals.GPIO47,
dc_pin: peripherals.GPIO12, dc_pin: peripherals.GPIO12,
} },
.build(); spi_bus::DisplayPinConfiguration::build,
);
// Initialize the shared I2C bus using the refactored i2c_bus module. // Initialize the shared I2C bus using the refactored i2c_bus module.
let shared_i2c = { let shared_i2c = {
let config = I2cBusPinConfiguration { use alloc::boxed::Box;
let bus = retry(5, I2cBusPinConfiguration {
i2c_peripheral: peripherals.I2C0, i2c_peripheral: peripherals.I2C0,
scl: peripherals.GPIO18, scl: peripherals.GPIO18,
sda: peripherals.GPIO17, sda: peripherals.GPIO17,
}; }, I2cBusPinConfiguration::build);
config.build() Box::leak(Box::new(bus))
}; };
let mut lcd: TertiaryDisplay = init_tertiary_lcd(AtomicDevice::new(shared_i2c)); let mut lcd: TertiaryDisplay = retry(
if let Err(e) = write_wrapped(&mut lcd, "Testing more than 16 chars what happens now?") { 5,
TertiaryDisplayPinConfiguration { i2c: shared_i2c },
TertiaryDisplayPinConfiguration::build,
);
lcd.load_charset(EXAMPLE).unwrap();
let h = EXAMPLE[HEART];
let e = EXAMPLE[EMPTY];
let b = EXAMPLE[BOX];
if let Err(e) = lcd.write(&alloc::format!(
"Custom symbols: {h}{e}{b} And more stuff to wrap and cut off please."
)) {
log::error!( log::error!(
"Tertiary LCD startup text write failed over shared I2C bus; check the shared bus, wiring, and device responses: {e}" "Tertiary LCD startup text write failed over shared I2C bus; check the shared bus, wiring, and device responses: {e}"
); );
} }
setup_wifi(peripherals.WIFI, peripherals.FLASH, &spawner).await; let store = Store::new(peripherals.FLASH).expect("Could not initialize Flash Store");
let cardstore = CardStore::new(store).expect("Could not initialize Card Store");
let rtc = Rtc::new(peripherals.LPWR);
//setup_wifi(peripherals.WIFI, peripherals.FLASH, &spawner).await;
let mut nfc_driver = NfcPn532Driver::new(AtomicDevice::new(shared_i2c)); let mut nfc_driver = NfcPn532Driver::new(AtomicDevice::new(shared_i2c));
if nfc_driver.configure_sam().await.is_ok() { if nfc_driver.configure_sam().await.is_ok() {
@@ -131,7 +154,10 @@ async fn main(spawner: Spawner) {
log::info!("Setup complete, entering main loop"); log::info!("Setup complete, entering main loop");
spawner.spawn(navigation::run(inputs, outputs).expect("run task failed")); spawner.spawn(navigation::run(inputs, outputs).expect("run task failed"));
spawner.spawn(background_tasks::nfc_scanner(nfc_driver).expect("nfc scanner task failed")); spawner.spawn(
background_tasks::nfc_scanner(nfc_driver, cardstore, rtc).expect("nfc scanner task failed"),
);
core::future::pending::<()>().await
} }
fn display_shit(oled: &mut SecondaryDisplay<'static>, text: &str) { fn display_shit(oled: &mut SecondaryDisplay<'static>, text: &str) {
@@ -143,3 +169,17 @@ fn display_shit(oled: &mut SecondaryDisplay<'static>, text: &str) {
.unwrap(); .unwrap();
oled.flush().unwrap(); oled.flush().unwrap();
} }
fn retry<T, U, F: FnMut(T) -> Result<U, T>>(repetitions: u32, input: T, mut action: F) -> U {
let mut initial_state = input;
for iteration in 0..repetitions {
log::debug!("Trying to initialize. Retries: {iteration}");
match action(initial_state) {
Ok(success) => return success,
Err(recovered_input) => initial_state = recovered_input,
}
}
panic!("Init failed after {repetitions} retries.")
}
+11 -10
View File
@@ -1,22 +1,23 @@
use core::fmt::Display; use core::fmt::Display;
use crate::card::model::NfcPayload; use crate::card::model::RawCard;
use alloc::string::{String, ToString}; use alloc::string::{String, ToString};
pub fn split_nfc_hex(payload: &[u8]) -> Option<NfcPayload<'_>> { pub fn split_nfc_hex(payload: &[u8]) -> Option<RawCard> {
if payload.len() < 0x35A { if payload.len() < 0x35A {
return None; return None;
} }
Some(NfcPayload { Some(RawCard {
event_encoding: &payload[0..2], event_encoding: payload[0..2].try_into().ok()?,
card_type: payload[2], card_type: payload[2],
card_uuid: &payload[3..5], card_uuid: payload[3..5].try_into().ok()?,
_reserved: &payload[5..9], _reserved: payload[5..9].try_into().ok()?,
sprite: &payload[9..0x2DB], sprite: payload[9..0x2DB].try_into().ok()?,
packed_card_text: &payload[0x2DB..0x311], packed_card_text: payload[0x2DB..0x311].try_into().ok()?,
secret: &payload[0x311..0x329], secret: payload[0x311..0x329].try_into().ok()?,
_opaque_trailer: &payload[0x329..0x35A], _opaque_trailer: payload[0x329..0x35A].try_into().ok()?,
_padding: Default::default(),
}) })
} }
+26 -18
View File
@@ -6,18 +6,24 @@ use num_enum::FromPrimitive;
use crate::card::decoder::{Events, decode_known_event, decode_packed_card_text, decode_secret}; use crate::card::decoder::{Events, decode_known_event, decode_packed_card_text, decode_secret};
use embedded_graphics::pixelcolor::{Rgb565, Rgb888}; use embedded_graphics::pixelcolor::{Rgb565, Rgb888};
#[derive(Debug, Clone, Copy)] use binary_serde::BinarySerde;
pub struct NfcPayload<'a> {
pub event_encoding: &'a [u8], #[derive(Debug, BinarySerde, PartialEq, Eq, Clone, Copy)]
#[repr(C)]
pub struct RawCard {
pub event_encoding: [u8; 2],
pub card_type: u8, pub card_type: u8,
pub card_uuid: &'a [u8], pub card_uuid: [u8; 2],
pub _reserved: &'a [u8], pub _reserved: [u8; 4],
pub sprite: &'a [u8], pub sprite: [u8; 722],
pub packed_card_text: &'a [u8], pub packed_card_text: [u8; 54],
pub secret: &'a [u8], pub secret: [u8; 24],
pub _opaque_trailer: &'a [u8], pub _opaque_trailer: [u8; 49],
pub _padding: [u8; 2],
} }
static_assertions::const_assert!(RawCard::SERIALIZED_SIZE.is_multiple_of(4));
#[derive(Debug, Clone, Eq, PartialEq, FromPrimitive)] #[derive(Debug, Clone, Eq, PartialEq, FromPrimitive)]
#[repr(u8)] #[repr(u8)]
pub enum CardType { pub enum CardType {
@@ -222,7 +228,7 @@ pub struct Sprite {
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct Card { pub struct Card {
pub uuid: String, pub uuid: u32,
pub event: Events, pub event: Events,
pub cardtype: CardType, pub cardtype: CardType,
pub name: String, pub name: String,
@@ -235,17 +241,19 @@ pub struct Card {
pub sprite: Sprite, pub sprite: Sprite,
} }
impl<'a> TryFrom<NfcPayload<'a>> for Card { impl TryFrom<RawCard> for Card {
type Error = &'static str; type Error = &'static str;
fn try_from(payload: NfcPayload<'a>) -> Result<Self, Self::Error> { fn try_from(raw_card: RawCard) -> Result<Self, Self::Error> {
let uuid: String = "empty for now".into(); //str::from_utf8(payload.card_uuid).unwrap().into(); let uuid: u32 = (u16::from_ne_bytes(raw_card.event_encoding) as u32) << 16
let event = decode_known_event(payload.event_encoding); | (u16::from_ne_bytes(raw_card.card_uuid) as u32);
let cardtype = CardType::from(payload.card_type);
let (name, trait1, trait2, trait3) = decode_packed_card_text(payload.packed_card_text); let event = decode_known_event(&raw_card.event_encoding);
let secret = decode_secret(payload.secret); let cardtype = CardType::from(raw_card.card_type);
let (name, trait1, trait2, trait3) = decode_packed_card_text(&raw_card.packed_card_text);
let secret = decode_secret(&raw_card.secret);
let sprite = Sprite { let sprite = Sprite {
data: payload.sprite.to_vec(), data: raw_card.sprite.to_vec(),
}; };
Ok(Self { Ok(Self {
uuid, uuid,
+19 -17
View File
@@ -1,11 +1,12 @@
use alloc::boxed::Box; use embedded_hal_bus::i2c::AtomicDevice;
use embedded_hal_bus::util::AtomicCell; use embedded_hal_bus::util::AtomicCell;
use esp_hal::gpio::interconnect::PeripheralInput; use esp_hal::gpio::interconnect::PeripheralInput;
use esp_hal::i2c::master::Config; use esp_hal::i2c::master::Config;
use esp_hal::{gpio::interconnect::PeripheralOutput, i2c::master::I2c}; use esp_hal::{gpio::interconnect::PeripheralOutput, i2c::master::I2c};
type InnerI2cBus = esp_hal::i2c::master::I2c<'static, esp_hal::Blocking>; type InnerI2cBus<'a> = esp_hal::i2c::master::I2c<'a, esp_hal::Blocking>;
pub type I2cBus = &'static AtomicCell<InnerI2cBus>; pub type I2cBus<'a> = &'a AtomicCell<InnerI2cBus<'a>>;
pub type I2cDevice<'a> = AtomicDevice<'a, esp_hal::i2c::master::I2c<'a, esp_hal::Blocking>>;
pub struct I2cBusPinConfiguration<I2C, SCL, SDA> { pub struct I2cBusPinConfiguration<I2C, SCL, SDA> {
pub i2c_peripheral: I2C, pub i2c_peripheral: I2C,
@@ -13,24 +14,25 @@ pub struct I2cBusPinConfiguration<I2C, SCL, SDA> {
pub sda: SDA, pub sda: SDA,
} }
impl<I2C, SCL, SDA> I2cBusPinConfiguration<I2C, SCL, SDA> impl<'a, I2C, SCL, SDA> I2cBusPinConfiguration<I2C, SCL, SDA>
where where
I2C: esp_hal::i2c::master::Instance + 'static, I2C: esp_hal::i2c::master::Instance + 'a,
SCL: PeripheralOutput<'static> + PeripheralInput<'static>, SCL: PeripheralOutput<'a> + PeripheralInput<'a>,
SDA: PeripheralOutput<'static> + PeripheralInput<'static>, SDA: PeripheralOutput<'a> + PeripheralInput<'a>,
{ {
pub fn build(self) -> I2cBus { pub fn build(self) -> Result<AtomicCell<InnerI2cBus<'a>>, Self> {
let config = Config::default().with_frequency(esp_hal::time::Rate::from_khz(400)); let config = Config::default().with_frequency(esp_hal::time::Rate::from_khz(400));
let i2c = match I2c::new(self.i2c_peripheral, config) { match I2c::new(self.i2c_peripheral, config) {
Ok(bus) => bus.with_sda(self.sda).with_scl(self.scl), Ok(bus) => {
let i2c = bus.with_sda(self.sda).with_scl(self.scl);
log::info!("Shared I2C bus ready on GPIO17 (SDA) and GPIO18 (SCL)");
Ok(AtomicCell::new(i2c))
}
Err(e) => { Err(e) => {
log::error!("Shared I2C bus initialization failed for GPIO17/GPIO18: {e:?}"); log::error!("Shared I2C bus initialization failed for GPIO17/GPIO18: {e:?}");
panic!("Shared I2C bus initialization failed"); unimplemented!("Partial move makes it impossible to retry.");
} // Err(self)
}; }
let shared_i2c = Box::leak(Box::new(AtomicCell::new(i2c))); }
log::info!("Shared I2C bus ready on GPIO17 (SDA) and GPIO18 (SCL)");
shared_i2c
} }
} }
+2 -2
View File
@@ -135,7 +135,7 @@ where
RES2: OutputPin + 'static, RES2: OutputPin + 'static,
DC: OutputPin + 'static, DC: OutputPin + 'static,
{ {
pub fn build(self) -> (PrimaryDisplay<'static>, [SecondaryDisplay<'static>; 3]) { pub fn build(self) -> Result<(PrimaryDisplay<'static>, [SecondaryDisplay<'static>; 3]), Self> {
let spi = Spi::new( let spi = Spi::new(
self.spi_peripheral, self.spi_peripheral,
SpiConfig::default() SpiConfig::default()
@@ -178,7 +178,7 @@ where
} }
.build(); .build();
(primary, secondaries) Ok((primary, secondaries))
} }
} }
+379 -72
View File
@@ -1,103 +1,410 @@
use embedded_hal::i2c::I2c;
use embedded_hal_bus::i2c::AtomicDevice; use embedded_hal_bus::i2c::AtomicDevice;
use esp_hal::delay::Delay; use esp_hal::delay::Delay;
use i2c_character_display::{CharacterDisplayPCF8574T, LcdDisplayType}; use i2c_character_display::{CharacterDisplayPCF8574T, LcdDisplayType};
use log::error; use log::error;
type GenericTertiaryDisplay<I2C> = CharacterDisplayPCF8574T<I2C, Delay>; use crate::create_custom_char_set;
pub type TertiaryDisplay<'a> = use crate::drivers::i2c_bus::{I2cBus, I2cDevice};
GenericTertiaryDisplay<AtomicDevice<'a, esp_hal::i2c::master::I2c<'a, esp_hal::Blocking>>>;
pub fn init_tertiary_lcd<I2C>(i2c: I2C) -> GenericTertiaryDisplay<I2C> type GenericTertiaryDisplay<I2C> = CharacterDisplayPCF8574T<I2C, Delay>;
where type TertiaryDisplayRaw<'a> = GenericTertiaryDisplay<I2cDevice<'a>>;
I2C: I2c, pub type TertiaryDisplayError<'a> = i2c_character_display::CharacterDisplayError<I2cDevice<'a>>;
{
let mut lcd = CharacterDisplayPCF8574T::new(i2c, LcdDisplayType::Lcd16x2, Delay::new()); pub struct TertiaryDisplayPinConfiguration<'a> {
pub i2c: I2cBus<'a>,
}
impl<'a> TertiaryDisplayPinConfiguration<'a> {
pub fn build(self) -> Result<TertiaryDisplay<'a>, Self> {
let device = AtomicDevice::new(self.i2c);
let mut lcd = CharacterDisplayPCF8574T::new(device, LcdDisplayType::Lcd16x2, Delay::new());
match lcd.init() { match lcd.init() {
Ok(()) => { Ok(()) => {
log::info!("I2C tertiary LCD initialized on PCF8574T at address 0x27"); log::info!("I2C tertiary LCD initialized on PCF8574T at address 0x27");
lcd Ok(TertiaryDisplay {
display: lcd,
charset: create_custom_char_set!(),
})
}
Err(e) => {
log::error!("I2C tertiary LCD init failed on PCF8574T at address 0x27: {e}");
Err(self)
} }
Err(_) => {
error!(
"I2C tertiary LCD init failed on PCF8574T at address 0x27; check bus wiring, power, and device address"
);
panic!("tertiary LCD initialization failed");
} }
} }
} }
pub fn write_line<I2C>( #[derive(Debug, Clone)]
lcd: &mut GenericTertiaryDisplay<I2C>, pub enum TertiaryDisplayWriteError<'a> {
line: u8, Display(TertiaryDisplayError<'a>),
text: &str, Format(LinesError),
) -> Result<(), i2c_character_display::CharacterDisplayError<I2C>>
where
I2C: I2c,
{
if let Err(e) = lcd.set_cursor(0, line) {
error!("I2C LCD cursor move failed on line {line}; bus or device may be unavailable");
return Err(e);
} }
if let Err(e) = lcd.print(text) {
error!( impl<'a> From<TertiaryDisplayError<'a>> for TertiaryDisplayWriteError<'a> {
"I2C LCD print failed on line {line}; payload length={} and device may be busy or disconnected", fn from(value: TertiaryDisplayError<'a>) -> Self {
text.len() Self::Display(value)
); }
}
impl From<LinesError> for TertiaryDisplayWriteError<'_> {
fn from(value: LinesError) -> Self {
Self::Format(value)
}
}
impl<'a> core::fmt::Display for TertiaryDisplayWriteError<'a> {
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
match self {
Self::Display(e) => write!(f, "display error: {e}"),
Self::Format(e) => write!(f, "format error: {e}"),
}
}
}
pub struct TertiaryDisplay<'a> {
display: TertiaryDisplayRaw<'a>,
charset: CustomCharset,
}
impl<'a> TertiaryDisplay<'a> {
pub const WIDTH: usize = 16;
pub const ROWS: usize = 2;
pub fn inner_mut(&mut self) -> &mut TertiaryDisplayRaw<'a> {
&mut self.display
}
pub fn load_charset(&mut self, charset: CustomCharset) -> Result<(), TertiaryDisplayError<'a>> {
self.charset = charset;
for (index, charmap) in self
.charset
.set
.into_iter()
.enumerate()
.take(self.charset.len)
{
self.display
.create_char(index as u8, charmap)
.inspect_err(|e| error!("I2C Custom character {index} init failed: {e}"))?;
}
Ok(())
}
pub fn clear(&mut self) -> Result<(), TertiaryDisplayError<'a>> {
if let Err(e) = self.display.clear() {
error!("I2C LCD clear failed; bus may be busy or device unresponsive");
return Err(e); return Err(e);
} }
Ok(()) Ok(())
} }
pub fn write_wrapped<I2C>( /// Writes text to the LCD. Normal strings wrap by default but this can be controlled
lcd: &mut GenericTertiaryDisplay<I2C>, /// via explicit [`Lines::checked`].
text: &str, pub fn write<'b, L: Into<Lines<'b>>>(
) -> Result<(), i2c_character_display::CharacterDisplayError<I2C>> &mut self,
where text: L,
I2C: I2c, ) -> Result<(), TertiaryDisplayWriteError<'a>> {
{ let lines = text.into();
const DISPLAY_WIDTH: usize = 16;
const DISPLAY_ROWS: usize = 2;
if let Err(e) = lcd.clear() { // Use original to get correct position in original string
error!( lines
"I2C LCD clear failed before wrapped write; check the shared bus and display response" .original
); .iter()
return Err(e); .try_for_each(|text| Lines::invalid_chars(text, Some(&self.charset)))?;
let lines = lines.into_result()?;
self.display
.clear()
.inspect_err(|e| error!("I2C LCD clear failed before wrapped write: {e}"))?;
for (row, line) in lines.into_iter().enumerate() {
self.display.set_cursor(0, row as u8).inspect_err(|e| {
error!("I2C LCD cursor move failed while writing wrapped row {row}: {e}")
})?;
self.display
.print(line)
.inspect_err(|e| error!("I2C LCD wrapped write failed on row {row}: {e}"))?;
} }
let mut offset = 0; Ok(())
for row in 0..DISPLAY_ROWS { }
let remaining = &text[offset..];
if remaining.is_empty() {
return Ok(());
} }
let chunk = if remaining.len() > DISPLAY_WIDTH { #[derive(Debug, PartialEq, Eq, Clone)]
&remaining[..DISPLAY_WIDTH] pub enum LinesError {
} else { ColumnOverflow { line: u8, columns: usize },
remaining LineOverflow,
InvalidChar { position: usize, symbol: char },
}
impl core::fmt::Display for LinesError {
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
match self {
Self::ColumnOverflow { line, columns } => {
write!(
f,
"text too long for line {line}: {columns} > {}",
TertiaryDisplay::WIDTH
)
}
Self::LineOverflow => write!(f, "text has too many lines"),
Self::InvalidChar { position, symbol } => {
write!(f, "Invalid character {symbol} at offset {position}")
}
}
}
}
#[derive(Debug, PartialEq, Eq, Clone)]
pub struct Lines<'a> {
pub lines: Result<[&'a str; TertiaryDisplay::ROWS], LinesError>,
/// Unmodified original string. Only first entry is populated unless constructed with [`by_line`]
pub original: [&'a str; TertiaryDisplay::ROWS],
}
impl<'a> Lines<'a> {
pub fn into_result(self) -> Result<[&'a str; TertiaryDisplay::ROWS], LinesError> {
self.lines
}
/// Validates that the text contains only non-control ASCII characters,
/// newlines, and contiguous custom characters up to `charset.len`.
/// If `charset` is `None`, allow any custom char up to and including maximum.
pub fn invalid_chars(text: &str, charset: Option<&CustomCharset>) -> Result<(), LinesError> {
for (pos, c) in text.char_indices() {
let is_standard_valid = c == '\n' || (' '..='~').contains(&c);
if is_standard_valid {
continue;
}
let first_invalid_custom_char =
charset.map(|c| c.len).unwrap_or(CUSTOM_CHAR_COUNT.into());
if (c as usize) >= first_invalid_custom_char {
return Err(LinesError::InvalidChar {
position: pos,
symbol: c,
});
}
}
Ok(())
}
/// Checks pre-split lines strictly for length and valid characters.
pub fn by_line(lines: [&'a str; TertiaryDisplay::ROWS]) -> Self {
let this = |res| Self {
lines: res,
original: lines,
}; };
if let Err(e) = lcd.set_cursor(0, row as u8) { for line in lines.iter() {
error!( if let Err(e) = Self::invalid_chars(line, None) {
"I2C LCD cursor move failed while writing wrapped row {row}; bus may be busy or device unresponsive" return this(Err(e));
);
return Err(e);
}
if let Err(e) = lcd.print(chunk) {
error!(
"I2C LCD wrapped write failed on row {row}; chunk length={} and text may exceed display width",
chunk.len()
);
return Err(e);
}
offset += chunk.len();
if offset >= text.len() {
return Ok(());
} }
} }
Ok(()) for (i, line) in lines.iter().enumerate() {
if line.len() > TertiaryDisplay::WIDTH {
return this(Err(LinesError::ColumnOverflow {
line: i as u8,
columns: line.len(),
}));
} }
}
this(Ok(lines))
}
/// Strictly checks an input string.
/// Fails if there are invalid chars, too many newlines, or if any segment exceeds WIDTH.
pub fn checked(input: &'a str) -> Self {
let this = |res| Self {
lines: res,
original: [input, ""],
};
if let Err(e) = Self::invalid_chars(input, None) {
return this(Err(e));
}
let mut out = [""; TertiaryDisplay::ROWS];
let mut parts = input.split('\n');
for (i, out) in out.iter_mut().enumerate() {
if let Some(part) = parts.next() {
if part.len() > TertiaryDisplay::WIDTH {
return this(Err(LinesError::ColumnOverflow {
line: i as u8,
columns: part.len(),
}));
}
*out = part;
}
}
if parts.next().is_some() {
return this(Err(LinesError::LineOverflow));
}
this(Ok(out))
}
/// Wraps text automatically at `WIDTH` or at newlines.
/// Discards (clamps) any leftover text that exceeds `ROWS`.
/// Fails immediately if invalid characters are present.
pub fn clamped(input: &'a str) -> Self {
let this = |res| Self {
lines: res,
original: [input, ""],
};
if let Err(e) = Self::invalid_chars(input, None) {
return this(Err(e));
}
let mut out = [""; TertiaryDisplay::ROWS];
let mut remaining = input;
for out in out.iter_mut() {
if remaining.is_empty() {
break;
}
let (segment, rest) = Self::next_segment(remaining, TertiaryDisplay::WIDTH);
*out = segment;
remaining = rest;
}
this(Ok(out))
}
/// Helper function to slice an ASCII string up to `max_width`
/// or until the next newline, whichever comes first.
fn next_segment(s: &str, max_width: usize) -> (&str, &str) {
let nl_pos = s.find('\n');
let split_pos = match nl_pos {
Some(pos) if pos <= max_width => pos,
_ => core::cmp::min(s.len(), max_width),
};
let segment = &s[..split_pos];
let mut rest = &s[split_pos..];
if let Some(stripped) = rest.strip_prefix('\n') {
rest = stripped;
}
(segment, rest)
}
}
impl<'a> From<&'a str> for Lines<'a> {
fn from(value: &'a str) -> Self {
Self::clamped(value)
}
}
impl<'a> From<&'a alloc::string::String> for Lines<'a> {
fn from(value: &'a alloc::string::String) -> Self {
Self::clamped(value)
}
}
/// 5x8 pixel charmap. Each set bit is a black pixel.
pub type CharMap = [u8; 8];
pub struct CustomCharset {
set: [CharMap; CUSTOM_CHAR_COUNT as usize],
len: usize,
}
impl core::ops::Index<CharMap> for CustomCharset {
type Output = &'static str;
fn index(&self, map: CharMap) -> &Self::Output {
let encoded = self.set[..self.len]
.iter()
.position(|&m| m == map)
.expect("Unknown char map in this set.");
match encoded {
0 => &"\x00",
1 => &"\x01",
2 => &"\x02",
3 => &"\x03",
4 => &"\x04",
5 => &"\x05",
6 => &"\x06",
7 => &"\x07",
8 => &"\x08",
9 => &"\x09",
10 => &"\x0A",
11 => &"\x0B",
12 => &"\x0C",
13 => &"\x0D",
14 => &"\x0E",
15 => &"\x0F",
pos => unreachable!(
"More than {CUSTOM_CHAR_COUNT} (=max) elements in array? Found element at {pos}."
),
}
}
}
#[macro_export]
macro_rules! create_custom_char_set {
() => {{
CustomCharset { set: [EMPTY; CUSTOM_CHAR_COUNT as usize], len: 0 }
}};
($($char:expr),+ $(,)?) => {{
// Count how many characters were passed using macro repetition length
const LEN: usize = [$(stringify!($char)),+].len();
// Ensure we don't exceed the defined maximum capacity
const _: () = assert!(
LEN <= CUSTOM_CHAR_COUNT as usize,
"Too many characters provided for CUSTOM_CHAR_COUNT"
);
// Build a fixed-size array populated with the provided characters,
// and pad the rest with empty CharMaps (zeros) to fit CUSTOM_CHAR_COUNT.
const SET: [CharMap; CUSTOM_CHAR_COUNT as usize] = {
let mut buffer = [[0u8; 8]; CUSTOM_CHAR_COUNT as usize];
let input = [$($char),*];
let mut i = 0;
while i < input.len() {
buffer[i] = input[i];
i += 1;
}
buffer
};
CustomCharset {
set: SET,
len: LEN,
}
}};
}
pub const CUSTOM_CHAR_COUNT: u8 = 16;
pub const HEART: CharMap = [
0b00000, // Top row
0b01010, // * *
0b11111, // *****
0b11111, // *****
0b01110, // ***
0b00100, // *
0b00000, // Bottom row
0b00000, // Cursor line (usually left blank)
];
pub const EMPTY: CharMap = [0; 8];
pub const BOX: CharMap = [
0b11111, 0b11111, 0b11111, 0b11111, 0b11111, 0b11111, 0b11111,
0b00000, // Cursor line (usually left blank)
];
pub const EXAMPLE: CustomCharset = create_custom_char_set!(HEART, BOX, EMPTY);
+2
View File
@@ -1,4 +1,5 @@
#![no_std] #![no_std]
#![feature(vec_into_chunks)]
extern crate alloc; extern crate alloc;
@@ -6,6 +7,7 @@ pub mod background_tasks;
pub mod card; pub mod card;
pub mod display; pub mod display;
pub mod drivers; pub mod drivers;
pub mod logging;
pub mod navigation; pub mod navigation;
pub mod network; pub mod network;
pub mod storage; pub mod storage;
+83
View File
@@ -0,0 +1,83 @@
mod logging_config {
include!(concat!(env!("OUT_DIR"), "/log_filter.rs"));
}
struct EnvLogger;
impl log::Log for EnvLogger {
fn enabled(&self, metadata: &log::Metadata) -> bool {
let level = metadata.level();
let target = metadata.target();
logging_config::is_enabled(level, target)
}
fn log(&self, record: &log::Record) {
if !self.enabled(record.metadata()) {
return;
}
// print with color? default to true if env var unset or set but empty.
let with_color = 1
== option_env!("ESP_LOG_COLOR")
.unwrap_or("1")
.parse()
.unwrap_or(1);
let (color, reset) = if with_color {
const RESET: &str = "\u{001B}[0m";
const RED: &str = "\u{001B}[31m";
const GREEN: &str = "\u{001B}[32m";
const YELLOW: &str = "\u{001B}[33m";
const BLUE: &str = "\u{001B}[34m";
const CYAN: &str = "\u{001B}[35m";
let color = match record.level() {
log::Level::Error => RED,
log::Level::Warn => YELLOW,
log::Level::Info => GREEN,
log::Level::Debug => BLUE,
log::Level::Trace => CYAN,
};
let reset = RESET;
(color, reset)
} else {
("", "")
};
let [now_s, now_ms] = {
let now = esp_hal::time::Instant::now()
.duration_since_epoch()
.as_millis();
[now / 1000, now % 1000]
};
let level = match record.level() {
log::Level::Error => "E",
log::Level::Warn => "W",
log::Level::Info => "I",
log::Level::Debug => "D",
log::Level::Trace => "T",
};
let args = record.args();
let line = record.line().unwrap_or(0);
let target = record.target();
let module = record.module_path().unwrap_or("???");
let (module, divider) = if module == target {
("", "")
} else {
(module, " ")
};
esp_println::println!(
"{color}{now_s:>4}.{now_ms:03}s {level} {target}{divider}{module}:{line} - {args}{reset}"
);
}
fn flush(&self) {}
}
pub fn init() {
log::set_logger(&EnvLogger).expect("Failed to init logger.");
log::set_max_level(logging_config::FILTER_MAX);
}
+31 -2
View File
@@ -1,2 +1,31 @@
pub mod dex_db; pub struct MemoryRegion {
pub mod nvs; offset: u32,
size: usize,
}
impl MemoryRegion {
pub const fn new(offset: u32, size: usize) -> Self {
assert!(offset.is_multiple_of(4), "Flash requires 4 bytes alignment");
assert!(size.is_multiple_of(4), "Flash requires 4 bytes alignment");
Self { offset, size }
}
pub const fn end(&self) -> usize {
self.offset as usize + self.size
}
}
const MAX_FLASH: usize = 16 * 1024 * 1024;
const MAGIC_REGION: MemoryRegion = MemoryRegion::new(0, 4);
const SETTINGS_REGION: MemoryRegion = MemoryRegion::new(MAGIC_REGION.end() as u32, 1024);
const CARDSTORE_REGION: MemoryRegion = MemoryRegion::new(
SETTINGS_REGION.end() as u32,
MAX_FLASH - SETTINGS_REGION.end(),
);
pub mod cardstore;
pub mod settings;
pub mod store;
+214
View File
@@ -0,0 +1,214 @@
use alloc::collections::BTreeMap;
use alloc::vec::Vec;
use binary_serde::{BinarySerde, DeserializeError, Endianness};
//use crate::card::model::Nfcraw_card;
use crate::card::model::RawCard;
use crate::storage::store::{Store, StoreError};
use crate::storage::{CARDSTORE_REGION, MemoryRegion};
const MAX_CARDS: usize = 20000;
pub const COUNT_REGION: MemoryRegion = MemoryRegion::new(CARDSTORE_REGION.offset, size_of::<u32>());
const ALLOCATION_TABLE_REGION: MemoryRegion = MemoryRegion::new(
COUNT_REGION.end() as u32,
MAX_CARDS * AllocationTableEntry::SERIALIZED_SIZE,
);
const CARDS_REGION: MemoryRegion = MemoryRegion::new(
ALLOCATION_TABLE_REGION.end() as u32,
CARDSTORE_REGION.size - ALLOCATION_TABLE_REGION.end(),
);
#[derive(Debug, BinarySerde, PartialEq, Eq, Clone)]
#[repr(C)]
pub struct AllocationTableEntry {
uuid: u32,
// this is the numeric offset of the AllocationTableEntry
// as well as the card entry in flash memory
// size_of::<AllocationTableEntry>() * offset
// size_of::<RawCard>() * offset
offset: u32,
creation_date: u64,
last_read_date: u64,
// how often a card has been read by nfc
read_count: u32,
deleted: bool,
reserved: [u64; 4],
_padding: [u8; 3],
}
static_assertions::const_assert!(AllocationTableEntry::SERIALIZED_SIZE.is_multiple_of(4));
impl AllocationTableEntry {
fn new(uuid: u32, offset: u32, timestamp: u64) -> Self {
Self {
uuid,
offset,
creation_date: timestamp,
last_read_date: timestamp,
read_count: 0,
deleted: true,
reserved: Default::default(),
_padding: Default::default(),
}
}
}
#[derive(Debug)]
pub enum CardStoreError {
Store(StoreError),
CorruptedEntry(DeserializeError),
NoEntry,
}
impl From<StoreError> for CardStoreError {
fn from(error: StoreError) -> CardStoreError {
CardStoreError::Store(error)
}
}
impl From<DeserializeError> for CardStoreError {
fn from(error: DeserializeError) -> CardStoreError {
CardStoreError::CorruptedEntry(error)
}
}
pub struct CardStore {
store: Store,
count: u32,
pub allocation_table: BTreeMap<u32, AllocationTableEntry>,
}
impl CardStore {
pub fn new(mut store: Store) -> Result<Self, CardStoreError> {
let mut count_bytes: [u8; COUNT_REGION.size] = [0; COUNT_REGION.size];
store.read(COUNT_REGION.offset, &mut count_bytes)?;
let count: u32 = u32::from_ne_bytes(count_bytes);
log::info!("Found {count} cards in flash.");
let mut allocation_table_entries: Vec<u8> =
Vec::with_capacity(AllocationTableEntry::SERIALIZED_SIZE * count as usize);
store.read(
ALLOCATION_TABLE_REGION.offset,
&mut allocation_table_entries,
)?;
let chunks =
allocation_table_entries.into_chunks::<{ AllocationTableEntry::SERIALIZED_SIZE }>();
let mut allocation_table = BTreeMap::<u32, AllocationTableEntry>::new();
for chunk in chunks {
let allocation_table_entry =
AllocationTableEntry::binary_deserialize(chunk.as_ref(), Endianness::Little)?;
allocation_table.insert(allocation_table_entry.uuid, allocation_table_entry);
}
Ok(Self {
store,
count,
allocation_table,
})
}
pub fn get_card_by_uuid(&mut self, uuid: u32) -> Result<RawCard, CardStoreError> {
let entry = self
.allocation_table
.get(&uuid)
.ok_or(CardStoreError::NoEntry)?;
let mut raw_card: [u8; size_of::<RawCard>()] = [0; size_of::<RawCard>()];
self.store.read(entry.offset, &mut raw_card)?;
Ok(RawCard::binary_deserialize(
raw_card.as_ref(),
Endianness::Little,
)?)
}
pub fn save_raw_card(
&mut self,
raw_card: RawCard,
timestamp: u64,
) -> Result<(), CardStoreError> {
let uuid: u32 = (u16::from_ne_bytes(raw_card.event_encoding) as u32) << 16
| (u16::from_ne_bytes(raw_card.card_uuid) as u32);
log::info!("Saving card with uuid {:x}", uuid);
// check if we already got the card
// if not find the next deleted entry,
// if none create a new one at the end
let (entry, is_new) = match self.allocation_table.get_mut(&uuid) {
Some(entry) => {
log::debug!("old entry");
(entry, false)
}
None => match self
.allocation_table
.values_mut()
.find(|entry| entry.deleted)
{
Some(entry) => {
log::debug!("deleted entry");
(entry, true)
}
None => {
log::debug!("new entry");
let entry = AllocationTableEntry::new(uuid, self.count + 1, timestamp);
self.allocation_table.insert(uuid, entry);
(
self.allocation_table
.get_mut(&uuid)
.expect("entry was just inserted"),
true,
)
}
},
};
// only write card data if it is new. else it is already written in flash
if is_new {
log::debug!("writing RawCard to flash");
let mut serialized_card: [u8; RawCard::SERIALIZED_SIZE] = [0; RawCard::SERIALIZED_SIZE];
raw_card.binary_serialize(&mut serialized_card, Endianness::Little);
self.store.write_erase(
CARDS_REGION.offset + entry.offset * RawCard::SERIALIZED_SIZE as u32,
&mut serialized_card,
)?;
} else {
entry.last_read_date = timestamp;
entry.read_count += 1;
}
entry.deleted = false;
log::debug!("writing AllocationTableEntry to flash");
let mut serialized_entry: [u8; AllocationTableEntry::SERIALIZED_SIZE] =
[0; AllocationTableEntry::SERIALIZED_SIZE];
entry.binary_serialize(&mut serialized_entry, Endianness::Little);
self.store.write_erase(
ALLOCATION_TABLE_REGION.offset
+ entry.offset * AllocationTableEntry::SERIALIZED_SIZE as u32,
&mut serialized_entry,
)?;
self.count += 1;
log::debug!("writing card count to flash");
let mut serialized_count: [u8; 4] = self.count.to_le_bytes();
self.store
.write(COUNT_REGION.offset, serialized_count.as_mut_slice())?;
Ok(())
}
}
-1
View File
@@ -1 +0,0 @@
// Storage dex_db module
-1
View File
@@ -1 +0,0 @@
// Storage nvs module
View File
+78
View File
@@ -0,0 +1,78 @@
use embedded_storage::nor_flash::NorFlash;
use embedded_storage::nor_flash::ReadNorFlash;
use esp_storage::{FlashStorage, FlashStorageError};
use crate::storage::MAGIC_REGION;
pub type StoreError = FlashStorageError;
const FLASH_ADDR: u32 = 0x9000;
const INITIALIZATION_SIZE: usize = 0x1000;
const FLASH_INITIALIZE_MAGIC: u32 = 0xde6de6de;
pub struct Store {
pub storage: FlashStorage<'static>,
}
impl Store {
pub fn new(flash: esp_hal::peripherals::FLASH<'static>) -> Result<Self, StoreError> {
let mut storage = FlashStorage::new(flash);
let mut magic: [u8; 4] = [0, 0, 0, 0];
storage.read(FLASH_ADDR + MAGIC_REGION.offset, &mut magic)?;
log::debug!("Magic bytes are {magic:?}");
if Self::has_magic_bytes(&mut storage)? {
log::info!("Skipped flash initialization, magic bytes already present.");
} else {
log::info!("Initializing flash because no magic bytes found.");
Self::initialize_flash(&mut storage)?;
}
Ok(Self { storage })
}
pub fn read(&mut self, offset: u32, bytes: &mut [u8]) -> Result<(), StoreError> {
self.storage.read(FLASH_ADDR + offset, bytes)
}
pub fn write(&mut self, offset: u32, bytes: &mut [u8]) -> Result<(), StoreError> {
log::debug!("Writing {} bytes at offset 0x{offset:08x}", bytes.len());
self.storage.write(FLASH_ADDR + offset, bytes)
}
pub fn write_erase(&mut self, offset: u32, bytes: &mut [u8]) -> Result<(), StoreError> {
log::debug!(
"Erasing and Writing {} bytes at offset 0x{offset:08x}",
bytes.len()
);
self.storage
.erase(FLASH_ADDR + offset, FLASH_ADDR + bytes.len() as u32)?;
self.storage.write(FLASH_ADDR + offset, bytes)
}
pub fn has_magic_bytes(storage: &mut FlashStorage) -> Result<bool, StoreError> {
let mut magic: [u8; 4] = [0, 0, 0, 0];
storage.read(FLASH_ADDR + MAGIC_REGION.offset, &mut magic)?;
log::debug!("Magic bytes are {magic:?}");
Ok(u32::from_ne_bytes(magic) == FLASH_INITIALIZE_MAGIC)
}
pub fn initialize_flash(storage: &mut FlashStorage) -> Result<(), StoreError> {
storage.erase(FLASH_ADDR, FLASH_ADDR + INITIALIZATION_SIZE as u32)?;
let zeros: [u8; INITIALIZATION_SIZE] = [0; INITIALIZATION_SIZE];
storage.write(FLASH_ADDR + MAGIC_REGION.end() as u32, &zeros)?;
storage.write(
FLASH_ADDR + MAGIC_REGION.offset,
&FLASH_INITIALIZE_MAGIC.to_ne_bytes(),
)?;
if Self::has_magic_bytes(storage)? {
Ok(())
} else {
Err(StoreError::Other(FLASH_INITIALIZE_MAGIC as i32))
}
}
}
+1
View File
@@ -1,4 +1,5 @@
pub mod card_view; pub mod card_view;
pub mod flash_info_view;
pub mod journal_view; pub mod journal_view;
pub mod main_menu; pub mod main_menu;
pub mod menu_item; pub mod menu_item;
+49
View File
@@ -0,0 +1,49 @@
use crate::navigation::inputs::ButtonAction;
use crate::navigation::navigation::Action;
use crate::navigation::navigation::Navigable;
use crate::navigation::navigation::NewState;
use crate::navigation::outputs::Outputs;
use crate::views::view::View;
use embedded_graphics::mono_font::MonoTextStyle;
use embedded_graphics::mono_font::ascii::FONT_10X20;
use embedded_graphics::pixelcolor::Rgb565;
use embedded_graphics::prelude::DrawTarget;
use embedded_graphics::prelude::RgbColor;
use embedded_graphics::text::Text;
use embedded_graphics::prelude::Point;
#[derive(Debug, Clone)]
pub struct FlashInfoView {}
impl Navigable for FlashInfoView {
fn display(&self, outputs: &mut Outputs) -> impl core::future::Future<Output = ()> + Send {
let style = MonoTextStyle::new(&FONT_10X20, Rgb565::WHITE);
Text::new("Flash Initialized: {:?}", Point::new(40, 40), style);
core::future::ready(())
}
fn handle_input(&self, input: Action) -> impl core::future::Future<Output = NewState> + Send {
async move {
let Action::Button(input) = input else {
// Skip timer inputs
return NewState {
view: View::FlashInfo(self.clone()),
replace_view: true,
redraw: false,
};
};
let new_menu = match input {
_ => View::FlashInfo(self.clone()),
};
NewState {
replace_view: matches!(new_menu, View::FlashInfo(_)),
view: new_menu,
redraw: true,
}
}
}
}
+42 -5
View File
@@ -1,11 +1,14 @@
use crate::navigation::navigation::{Action, Navigable, NewState}; use crate::navigation::navigation::{Action, Navigable, NewState};
use crate::navigation::outputs::Outputs; use crate::navigation::outputs::Outputs;
use crate::views::{menu_item, view::View}; use crate::views::{flash_info_view::FlashInfoView, menu_item, view::View};
use crate::navigation::inputs::ButtonAction;
use embedded_graphics::pixelcolor::Rgb565; use embedded_graphics::pixelcolor::Rgb565;
use embedded_graphics::prelude::RgbColor; use embedded_graphics::prelude::RgbColor;
use embedded_graphics_core::draw_target::DrawTarget; use embedded_graphics_core::draw_target::DrawTarget;
pub const MAX_SELECTED: i8 = 2;
pub const MAX_SELECTED: i32 = 2;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct SettingsMenu { pub struct SettingsMenu {
@@ -18,16 +21,50 @@ impl Navigable for SettingsMenu {
display.clear(Rgb565::BLACK).unwrap(); display.clear(Rgb565::BLACK).unwrap();
menu_item::show(display, "System Inforation", 0, self.selected); menu_item::show(display, "System Inforation", 0, self.selected);
menu_item::show(display, "Wifi Information", 1, self.selected); menu_item::show(display, "Wifi Information", 1, self.selected);
menu_item::show(display, "Flash Information", 2, self.selected);
core::future::ready(()) core::future::ready(())
} }
fn handle_input(&self, input: Action) -> impl core::future::Future<Output = NewState> + Send { fn handle_input(&self, input: Action) -> impl core::future::Future<Output = NewState> + Send {
async move { async move {
NewState { let Action::Button(input) = input else {
view: View::Settings(SettingsMenu { selected: 0 }), // Skip timer inputs
return NewState {
view: View::Settings(self.clone()),
replace_view: true, replace_view: true,
redraw: !matches!(input, Action::Timer), redraw: false,
};
};
let new_menu = match input {
ButtonAction::Up => {
let new_selected = if (self.selected - 1) < 0 {
MAX_SELECTED - 1
} else {
self.selected - 1
};
View::Settings(SettingsMenu {
selected: new_selected,
})
}
ButtonAction::Down => {
let new_selected = self.selected.wrapping_add(1) % MAX_SELECTED;
View::Settings(SettingsMenu {
selected: new_selected,
})
}
ButtonAction::Ok => match self.selected {
2 => View::FlashInfo(FlashInfoView {}),
_ => View::Settings(self.clone()),
},
_ => View::Settings(self.clone()),
};
NewState {
replace_view: matches!(new_menu, View::Settings(_)),
view: new_menu,
redraw: true,
} }
} }
} }
+5 -2
View File
@@ -1,8 +1,8 @@
use crate::navigation::outputs::Outputs; use crate::navigation::outputs::Outputs;
use crate::views::card_view::CardView; use crate::views::card_view::CardView;
use crate::views::{ use crate::views::{
journal_view::JournalView, main_menu::MainMenu, scan_menu::ScanMenu, flash_info_view::FlashInfoView, journal_view::JournalView, main_menu::MainMenu,
settings_menu::SettingsMenu, scan_menu::ScanMenu, settings_menu::SettingsMenu,
}; };
use crate::navigation::navigation::{Navigable, NewState}; use crate::navigation::navigation::{Navigable, NewState};
@@ -14,6 +14,7 @@ pub enum View {
Settings(SettingsMenu), Settings(SettingsMenu),
Journal(JournalView), Journal(JournalView),
Card(CardView), Card(CardView),
FlashInfo(FlashInfoView),
} }
impl Navigable for View { impl Navigable for View {
@@ -25,6 +26,7 @@ impl Navigable for View {
View::Settings(settings_menu) => settings_menu.display(outputs).await, View::Settings(settings_menu) => settings_menu.display(outputs).await,
View::Journal(journal_view) => journal_view.display(outputs).await, View::Journal(journal_view) => journal_view.display(outputs).await,
View::Card(card_view) => card_view.display(outputs).await, View::Card(card_view) => card_view.display(outputs).await,
View::FlashInfo(flash_info_view) => flash_info_view.display(outputs).await,
} }
} }
} }
@@ -40,6 +42,7 @@ impl Navigable for View {
View::Settings(settings_menu) => settings_menu.handle_input(input).await, View::Settings(settings_menu) => settings_menu.handle_input(input).await,
View::Journal(journal_view) => journal_view.handle_input(input).await, View::Journal(journal_view) => journal_view.handle_input(input).await,
View::Card(card_view) => card_view.handle_input(input).await, View::Card(card_view) => card_view.handle_input(input).await,
View::FlashInfo(flash_info_view) => flash_info_view.handle_input(input).await,
} }
} }
} }