gba_save 0.3.0

//! Flash backup memory.
//!
//! The GBA has three different variants of flash backup:
//! - 64KiB
//! - 64KiB Atmel
//! - 128KiB
//!
//! Each of these backup types are interacted with in slightly different ways. Therefore, they are
//! treated separately by this library.
//!
//! To interact with flash backup memory, initialize using [`Flash::new()`]. This will provide the
//! variant of the found flash type for you to interact with.
//!
//! ``` no_run
//! use gba_save::flash::Flash;
//!
//! let flash = unsafe { Flash::new() }.expect("flash not available");
//! match flash {
//!     Flash::Flash64K(flash_64k) => {
//!         // Read, write, etc.
//!     }
//!     Flash::Flash64KAtmel(flash_64k_atmel) => {
//!         // Read, write, etc.
//!     }
//!     Flash::Flash128K(flash_128k) => {
//!         // Read, write, etc.
//!     }
//! }
//! ```
//!
//! If only a subset of flash variants are to be supported, simply handle the unsupported cases as
//! errors. For example, if only 128KiB flash is to be supported, we interact with the flash chip
//! on only that case:
//!
//! ``` no_run
//! use gba_save::flash::Flash;
//!
//! let flash = unsafe { Flash::new() }.expect("flash not available");
//! match flash {
//!     Flash::Flash128K(flash_128k) => {
//!         // Read, write, etc.
//!     }
//!     _ => panic!("unsupported flash type"),
//! }
//! ```
//!
//! [`Flash::new()`]: Flash::new()

mod device;
mod error;
mod reader;
mod writer;

pub use device::UnknownDeviceId;
pub use error::Error;
pub use reader::{Reader64K, Reader128K};
pub use writer::{Writer64K, Writer64KAtmel, Writer128K};

use crate::{
    log,
    mmio::{Cycles, WAITCNT},
    range::translate_range_to_buffer,
};
use core::{
    hint::black_box,
    ops,
    ops::{Bound, RangeBounds},
    time::Duration,
};
use deranged::{RangedU8, RangedUsize};
use device::Device;

const FLASH_MEMORY: *mut u8 = 0x0e00_0000 as *mut u8;
const BANK_SWITCH: *mut Bank = 0x0e00_0000 as *mut Bank;
const COMMAND: *mut Command = 0x0e00_5555 as *mut Command;
const COMMAND_ENABLE: *mut u8 = 0x0e00_2aaa as *mut u8;
const SECTOR_COMMAND: *mut Command = 0x0e00_0000 as *mut Command;
const ENABLE: u8 = 0x55;
const ERASED: u8 = 0xff;
const SIZE_64KB: usize = 0x10000;

#[derive(Debug)]
#[repr(u8)]
enum Command {
    EraseChip = 0x10,
    EraseSector = 0x30,
    Erase = 0x80,
    EnterIDMode = 0x90,
    Write = 0xa0,
    SwitchBank = 0xb0,
    TerminateMode = 0xf0,
    Enable = 0xaa,
}

fn begin_send_command() {
    unsafe {
        COMMAND.write_volatile(Command::Enable);
        COMMAND_ENABLE.write_volatile(ENABLE);
    }
}

fn send_command(command: Command) {
    begin_send_command();
    unsafe {
        COMMAND.write_volatile(command);
    }
}

#[derive(Clone, Copy, Debug)]
enum Bank {
    _0,
    _1,
}

fn switch_bank(bank: Bank) {
    send_command(Command::SwitchBank);
    unsafe {
        BANK_SWITCH.write_volatile(bank);
    }
}

fn wait(amount: Duration) {
    for _ in 0..amount.as_millis() * 1000 {
        black_box(());
    }
}

fn verify_byte(address: *const u8, byte: u8, timeout: Duration) -> Result<(), Error> {
    let mut i = 0;
    loop {
        if unsafe { address.read_volatile() } == byte {
            return Ok(());
        }
        if i >= timeout.as_millis() * 1000 {
            return Err(Error::OperationTimedOut);
        }

        i += 1;
    }
}

fn verify_bytes(address: *const u8, bytes: &[u8], timeout: Duration) -> Result<(), Error> {
    let mut i = 0;
    loop {
        let mut verified = true;
        for (i, &byte) in bytes.iter().enumerate() {
            if unsafe { address.add(i).read_volatile() } != byte {
                verified = false;
                break;
            }
        }
        if verified {
            return Ok(());
        }
        if i >= timeout.as_millis() * 1000 {
            return Err(Error::OperationTimedOut);
        }

        i += 1;
    }
}

fn erase_sector(sector: u8) -> Result<(), Error> {
    // Generic erase command.
    send_command(Command::Erase);

    // Specific erase command for sector.
    begin_send_command();
    let sector_command = unsafe { SECTOR_COMMAND.add(sector as usize * 0x1000) };
    unsafe {
        sector_command.write_volatile(Command::EraseSector);
    }

    verify_byte(
        sector_command as *const u8,
        ERASED,
        Duration::from_millis(20),
    )
}

fn translate_range_to_sectors<const MAX: u8, Range>(range: Range) -> ops::Range<u8>
where
    Range: RangeBounds<RangedU8<0, MAX>>,
{
    #[allow(unused_parens)] // Doesn't compile without the parenthesis.
    (match range.start_bound() {
        Bound::Included(start) => start.get(),
        Bound::Excluded(start) => start.get() + 1,
        Bound::Unbounded => 0,
    }..match range.end_bound() {
        Bound::Included(end) => end.get() + 1,
        Bound::Excluded(end) => end.get(),
        Bound::Unbounded => MAX + 1,
    })
}

/// A flash device with 64KiB of storage.
///
/// This storage type is divided into 16 4KiB sectors. Each sector must be erased before it can be
/// written to. Failing to erase a sector will result in invalid data.
#[derive(Debug)]
pub struct Flash64K;

impl Flash64K {
    /// Returns a reader over the given range.
    pub fn reader<'a, 'b, Range>(&'a mut self, range: Range) -> Reader64K<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 65535>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Reader64K::new_unchecked(address, len) }
    }

    /// Erases the specified sectors.
    ///
    /// This should be called before attempting to write to these sectors. Memory that has already
    /// been written to cannot be written to again without first being erased.
    pub fn erase_sectors<Range>(&mut self, sectors: Range) -> Result<(), Error>
    where
        Range: RangeBounds<RangedU8<0, 15>>,
    {
        for sector in translate_range_to_sectors(sectors) {
            erase_sector(sector)?;
        }
        Ok(())
    }

    /// Returns a writer over the given range.
    pub fn writer<'a, 'b, Range>(&'a mut self, range: Range) -> Writer64K<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 65535>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Writer64K::new_unchecked(address, len) }
    }
}

/// A flash device with 64KiB of storage manufactured by Atmel.
///
/// These devices are handled separately, as they do not require users to manage erasing of
/// sectors. Instead, they can be written to directly, as the sector size is small enough to fit
/// into an internal buffer.
#[derive(Debug)]
pub struct Flash64KAtmel;

impl Flash64KAtmel {
    /// Returns a reader over the given range.
    pub fn reader<'a, 'b, Range>(&'a mut self, range: Range) -> Reader64K<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 65535>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Reader64K::new_unchecked(address, len) }
    }

    /// Returns a writer over the given range.
    pub fn writer<'a, 'b, Range>(&'a mut self, range: Range) -> Writer64KAtmel<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 65535>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Writer64KAtmel::new_unchecked(address, len) }
    }
}

/// A flash device with 128KiB of storage.
///
/// This storage type is divided into 32 4KiB sectors. Each sector must be erased before it can be
/// written to. Failing to erase a sector will result in invalid data.
#[derive(Debug)]
pub struct Flash128K;

impl Flash128K {
    /// Returns a reader over the given range.
    pub fn reader<'a, 'b, Range>(&'a mut self, range: Range) -> Reader128K<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 131071>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Reader128K::new_unchecked(address, len) }
    }

    /// Erases the specified sectors.
    ///
    /// This should be called before attempting to write to these sectors. Memory that has already
    /// been written to cannot be written to again without first being erased.
    pub fn erase_sectors<Range>(&mut self, sectors: Range) -> Result<(), Error>
    where
        Range: RangeBounds<RangedU8<0, 31>>,
    {
        let sectors_range = translate_range_to_sectors(sectors);
        let mut bank = if sectors_range.start < 16 {
            Bank::_0
        } else {
            Bank::_1
        };
        switch_bank(bank);
        for mut sector in sectors_range {
            if matches!(bank, Bank::_0) {
                if sector >= 16 {
                    bank = Bank::_1;
                    switch_bank(bank);
                }
            }
            if matches!(bank, Bank::_1) {
                sector %= 16;
            }
            erase_sector(sector)?;
        }
        Ok(())
    }

    /// Returns a writer over the given range.
    pub fn writer<'a, 'b, Range>(&'a mut self, range: Range) -> Writer128K<'b>
    where
        'a: 'b,
        Range: RangeBounds<RangedUsize<0, 131071>>,
    {
        let (address, len) = translate_range_to_buffer(range, FLASH_MEMORY);
        unsafe { Writer128K::new_unchecked(address, len) }
    }
}

/// The currently available flash backup device.
///
/// The GBA has three different variants of flash backup:
/// - 64KiB
/// - 64KiB Atmel
/// - 128KiB
///
/// Each of these backup types are interacted with in slightly different ways. Therefore, they are
/// treated separately by this library. This type contains the variant of the currently available
/// flash device. Users should match on the variants of this type and provide specific behavior for
/// each supported variant.
///
/// # Example
/// ``` no_run
/// use gba_save::flash::Flash;
///
/// let flash = unsafe { Flash::new() }.expect("flash not available");
/// match flash {
///     Flash::Flash64K(flash_64k) => {
///         // Read, write, etc.
///     }
///     Flash::Flash64KAtmel(flash_64k_atmel) => {
///         // Read, write, etc.
///     }
///     Flash::Flash128K(flash_128k) => {
///         // Read, write, etc.
///     }
/// }
/// ```
#[derive(Debug)]
pub enum Flash {
    /// 64KiB flash memory.
    Flash64K(Flash64K),
    /// 64KiB flash memory manufactured by Atmel.
    ///
    /// This case is handled separately, as Atmel chips have different sector sizes than other
    /// devices.
    Flash64KAtmel(Flash64KAtmel),
    /// 128KiB flash memory.
    Flash128K(Flash128K),
}

impl Flash {
    /// Returns the variant of the currently available flash device.
    ///
    /// This is the starting point for interacting with the flash backup.
    ///
    /// # Safety
    /// Must have exclusive ownership of both flash RAM memory and WAITCNT's SRAM wait control
    /// setting for the duration of its lifetime.
    pub unsafe fn new() -> Result<Self, UnknownDeviceId> {
        let mut waitstate_control = unsafe { WAITCNT.read_volatile() };
        waitstate_control.set_backup_waitstate(Cycles::_8);
        unsafe { WAITCNT.write_volatile(waitstate_control) };

        send_command(Command::EnterIDMode);
        wait(Duration::from_millis(20));

        // Read u16 from memory.
        let device = u16::from_ne_bytes(unsafe {
            [
                FLASH_MEMORY.read_volatile(),
                FLASH_MEMORY.add(1).read_volatile(),
            ]
        })
        .try_into()?;

        send_command(Command::TerminateMode);
        wait(Duration::from_millis(20));
        // Sanyo 128K device needs to have `TerminateMode` command sent twice.
        if matches!(device, Device::LE26FV10N1TS) {
            send_command(Command::TerminateMode);
            wait(Duration::from_millis(20));
        }

        log::info!("Detected Flash device with ID {device}");

        match device {
            Device::AT29LV512 => Ok(Self::Flash64KAtmel(Flash64KAtmel)),
            Device::MX29L010 | Device::LE26FV10N1TS => Ok(Self::Flash128K(Flash128K)),
            _ => Ok(Self::Flash64K(Flash64K)),
        }
    }

    /// Erase the entirety of the flash backup memory.
    pub fn reset(&mut self) -> Result<(), Error> {
        send_command(Command::Erase);
        send_command(Command::EraseChip);

        // Verify.
        verify_byte(FLASH_MEMORY, ERASED, Duration::from_millis(20))
    }
}

#[cfg(test)]
mod tests {
    use super::{Error, Flash, UnknownDeviceId, wait};
    use claims::{assert_err_eq, assert_ok, assert_ok_eq};
    use core::time::Duration;
    use deranged::{RangedU8, RangedUsize};
    use embedded_io::{Read, Write};
    use gba_test::test;

    macro_rules! assert_flash_64k {
        ($expr:expr) => {
            match $expr {
                Flash::Flash64K(flash_64k) => flash_64k,
                flash => panic!(
                    "assertion failed, expected Flash::Flash64K(..), got {:?}",
                    flash
                ),
            }
        };
    }

    macro_rules! assert_flash_64k_atmel {
        ($expr:expr) => {
            match $expr {
                Flash::Flash64KAtmel(flash_64k_atmel) => flash_64k_atmel,
                flash => panic!(
                    "assertion failed, expected Flash::Flash64KAtmel(..), got {:?}",
                    flash
                ),
            }
        };
    }

    macro_rules! assert_flash_128k {
        ($expr:expr) => {
            match $expr {
                Flash::Flash128K(flash_128k) => flash_128k,
                flash => panic!(
                    "assertion failed, expected Flash::Flash129K(..), got {:?}",
                    flash
                ),
            }
        };
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn new_64k() {
        assert_flash_64k!(assert_ok!(unsafe { Flash::new() }));
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn empty_range_read_64k() {
        let mut flash = assert_flash_64k!(assert_ok!(unsafe { Flash::new() }));
        let mut buffer = [1, 2, 3, 4];

        assert_ok_eq!(
            flash
                .reader(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .read(&mut buffer),
            0
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn empty_range_write_64k() {
        let mut flash = assert_flash_64k!(assert_ok!(unsafe { Flash::new() }));

        assert_err_eq!(
            flash
                .writer(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .write(&[1, 2, 3, 4]),
            Error::EndOfWriter
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn full_range_64k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k = assert_flash_64k!(flash);
        let mut writer = flash_64k.writer(..);

        for i in 0..16384 {
            assert_ok_eq!(
                writer.write(&[
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ]),
                4
            );
        }

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader = flash_64k.reader(..);
        let mut buf = [0, 0, 0, 0];

        for i in 0..16384 {
            assert_ok_eq!(reader.read(&mut buf), 4);
            assert_eq!(
                buf,
                [
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ],
            );
        }
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn partial_range_64k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k = assert_flash_64k!(flash);
        let mut writer =
            flash_64k.writer(RangedUsize::new_static::<42>()..RangedUsize::new_static::<100>());

        assert_ok_eq!(writer.write(&[b'a'; 100]), 58);

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader =
            flash_64k.reader(RangedUsize::new_static::<51>()..RangedUsize::new_static::<60>());
        let mut buf = [0; 20];

        assert_ok_eq!(reader.read(&mut buf), 9);
        assert_eq!(
            buf,
            [
                b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
                0
            ]
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn erase_one_sector_64k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k = assert_flash_64k!(flash);
        // Write some data to it.
        let mut writer = flash_64k.writer(..RangedUsize::new_static::<13>());
        assert_ok_eq!(writer.write(b"hello, world!"), 13);

        assert_ok!(
            flash_64k.erase_sectors(RangedU8::new_static::<0>()..RangedU8::new_static::<1>())
        );

        let mut reader =
            flash_64k.reader(RangedUsize::new_static::<0>()..RangedUsize::new_static::<13>());
        let mut buf = [0; 13];

        assert_ok_eq!(reader.read(&mut buf), 13);
        assert_eq!(buf, [0xff; 13],);
    }

    #[test]
    #[cfg_attr(
        not(flash_64k),
        ignore = "This test requires a Flash 64KiB chip. Ensure Flash 64KiB is configured and pass `--cfg flash_64k` to enable."
    )]
    fn erase_all_sectors_64k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k = assert_flash_64k!(flash);
        // Write some data to it.
        let mut writer = flash_64k.writer(..);
        for i in 0..16384 {
            assert_ok_eq!(
                writer.write(&[
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ]),
                4
            );
        }

        assert_ok!(flash_64k.erase_sectors(..));

        let mut reader = flash_64k.reader(..);
        let mut buf = [0; 4];

        for _ in 0..16384 {
            assert_ok_eq!(reader.read(&mut buf), 4);
            assert_eq!(buf, [0xff; 4],);
        }
    }

    #[test]
    #[cfg_attr(
        not(flash_64k_atmel),
        ignore = "This test requires a Flash 64KiB Atmel chip. Ensure Flash 64KiB Atmel is configured and pass `--cfg flash_64k_atmel` to enable."
    )]
    fn new_64k_atmel() {
        assert_flash_64k_atmel!(assert_ok!(unsafe { Flash::new() }));
    }

    #[test]
    #[cfg_attr(
        not(flash_64k_atmel),
        ignore = "This test requires a Flash 64KiB Atmel chip. Ensure Flash 64KiB Atmel is configured and pass `--cfg flash_64k_atmel` to enable."
    )]
    fn empty_range_read_64k_atmel() {
        let mut flash = assert_flash_64k_atmel!(assert_ok!(unsafe { Flash::new() }));
        let mut buffer = [1, 2, 3, 4];

        assert_ok_eq!(
            flash
                .reader(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .read(&mut buffer),
            0
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_64k_atmel),
        ignore = "This test requires a Flash 64KiB Atmel chip. Ensure Flash 64KiB Atmel is configured and pass `--cfg flash_64k_atmel` to enable."
    )]
    fn empty_range_write_64k_atmel() {
        let mut flash = assert_flash_64k_atmel!(assert_ok!(unsafe { Flash::new() }));

        assert_err_eq!(
            flash
                .writer(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .write(&[1, 2, 3, 4]),
            Error::EndOfWriter
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_64k_atmel),
        ignore = "This test requires a Flash 64KiB Atmel chip. Ensure Flash 64KiB Atmel is configured and pass `--cfg flash_64k_atmel` to enable."
    )]
    fn full_range_64k_atmel() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k_atmel = assert_flash_64k_atmel!(flash);
        let mut writer = flash_64k_atmel.writer(..);

        for i in 0..16384 {
            assert_ok_eq!(
                writer.write(&[
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ]),
                4
            );
        }
        drop(writer);

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader = flash_64k_atmel.reader(..);
        let mut buf = [0, 0, 0, 0];

        for i in 0..16384 {
            assert_ok_eq!(reader.read(&mut buf), 4);
            assert_eq!(
                buf,
                [
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ],
                "i: {}",
                i,
            );
        }
    }

    #[test]
    #[cfg_attr(
        not(flash_64k_atmel),
        ignore = "This test requires a Flash 64KiB Atmel chip. Ensure Flash 64KiB Atmel is configured and pass `--cfg flash_64k_atmel` to enable."
    )]
    fn partial_range_64k_atmel() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_64k_atmel = assert_flash_64k_atmel!(flash);
        let mut writer = flash_64k_atmel
            .writer(RangedUsize::new_static::<42>()..RangedUsize::new_static::<130>());

        assert_ok_eq!(writer.write(&[b'a'; 100]), 88);
        drop(writer);

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader = flash_64k_atmel
            .reader(RangedUsize::new_static::<121>()..RangedUsize::new_static::<130>());
        let mut buf = [0; 20];

        assert_ok_eq!(reader.read(&mut buf), 9);
        assert_eq!(
            buf,
            [
                b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
                0
            ]
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn new_128k() {
        assert_flash_128k!(assert_ok!(unsafe { Flash::new() }));
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn empty_range_read_128k() {
        let mut flash = assert_flash_128k!(assert_ok!(unsafe { Flash::new() }));
        let mut buffer = [1, 2, 3, 4];

        assert_ok_eq!(
            flash
                .reader(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .read(&mut buffer),
            0
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn empty_range_write_128k() {
        let mut flash = assert_flash_128k!(assert_ok!(unsafe { Flash::new() }));

        assert_err_eq!(
            flash
                .writer(RangedUsize::new_static::<0>()..RangedUsize::new_static::<0>())
                .write(&[1, 2, 3, 4]),
            Error::EndOfWriter
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn full_range_128k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_128k = assert_flash_128k!(flash);
        let mut writer = flash_128k.writer(..);

        for i in 0..32768 {
            assert_ok_eq!(
                writer.write(&[
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ]),
                4
            );
        }

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader = flash_128k.reader(..);
        let mut buf = [0, 0, 0, 0];

        for i in 0..32768 {
            assert_ok_eq!(reader.read(&mut buf), 4);
            assert_eq!(
                buf,
                [
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ],
            );
        }
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn partial_range_128k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_128k = assert_flash_128k!(flash);
        let mut writer =
            flash_128k.writer(RangedUsize::new_static::<42>()..RangedUsize::new_static::<100>());

        assert_ok_eq!(writer.write(&[b'a'; 100]), 58);

        // Wait for the data to be available.
        wait(Duration::from_millis(1));

        let mut reader =
            flash_128k.reader(RangedUsize::new_static::<51>()..RangedUsize::new_static::<60>());
        let mut buf = [0; 20];

        assert_ok_eq!(reader.read(&mut buf), 9);
        assert_eq!(
            buf,
            [
                b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', b'a', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
                0
            ]
        );
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn erase_one_sector_128k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_128k = assert_flash_128k!(flash);
        // Write some data to it.
        let mut writer = flash_128k.writer(..RangedUsize::new_static::<13>());
        assert_ok_eq!(writer.write(b"hello, world!"), 13);

        assert_ok!(
            flash_128k.erase_sectors(RangedU8::new_static::<0>()..RangedU8::new_static::<1>())
        );

        let mut reader =
            flash_128k.reader(RangedUsize::new_static::<0>()..RangedUsize::new_static::<13>());
        let mut buf = [0; 13];

        assert_ok_eq!(reader.read(&mut buf), 13);
        assert_eq!(buf, [0xff; 13],);
    }

    #[test]
    #[cfg_attr(
        not(flash_128k),
        ignore = "This test requires a Flash 128KiB chip. Ensure Flash 128KiB is configured and pass `--cfg flash_128k` to enable."
    )]
    fn erase_all_sectors_128k() {
        let mut flash = assert_ok!(unsafe { Flash::new() });
        assert_ok!(flash.reset());
        let mut flash_128k = assert_flash_128k!(flash);
        // Write some data to it.
        let mut writer = flash_128k.writer(..);
        for i in 0..32768 {
            assert_ok_eq!(
                writer.write(&[
                    0u8.wrapping_add(i as u8),
                    1u8.wrapping_add(i as u8),
                    2u8.wrapping_add(i as u8),
                    3u8.wrapping_add(i as u8)
                ]),
                4
            );
        }

        assert_ok!(flash_128k.erase_sectors(..));

        let mut reader = flash_128k.reader(..);
        let mut buf = [0; 4];

        for _ in 0..32768 {
            assert_ok_eq!(reader.read(&mut buf), 4);
            assert_eq!(buf, [0xff; 4],);
        }
    }

    #[test]
    #[cfg_attr(
        any(flash_64k, flash_64k_atmel, flash_128k),
        ignore = "This test cannot be run with a Flash chip. Ensure Flash is not configured and do not pass `--cfg flash_64k`, `--cfg flash_64k_atmel`, or `--cfg flash_128k` to enable."
    )]
    fn new_unknown() {
        assert_err_eq!(unsafe { Flash::new() }, UnknownDeviceId(0xffff));
    }

    #[test]
    #[cfg_attr(
        all(not(flash_64k), not(flash_64k_atmel), not(flash_128k)),
        ignore = "This test requires a Flash chip. Ensure Flash is configured and pass `--cfg flash_64k`, `--cfg flash_64k_atmel`, or `--cfg flash_128k` to enable."
    )]
    fn reset() {
        let mut flash = assert_ok!(unsafe { Flash::new() });

        assert_ok!(flash.reset());
    }
}