sftool_lib/common/
erase_flash.rs

1use crate::common::ram_command::{Command, RamCommand};
2use crate::utils::Utils;
3use crate::{Error, Result, SifliToolTrait};
4
5/// 通用的Flash擦除操作实现
6pub struct EraseOps;
7
8impl EraseOps {
9    /// 擦除整个Flash的通用实现
10    pub fn erase_all<T>(tool: &mut T, address: u32) -> Result<()>
11    where
12        T: SifliToolTrait + RamCommand,
13    {
14        let progress = tool.progress();
15        let progress_bar =
16            progress.create_spinner(format!("Erasing entire flash at 0x{:08X}...", address));
17
18        // 发送擦除所有命令
19        let _ = tool.command(Command::EraseAll { address });
20
21        let mut buffer = Vec::new();
22        let now = std::time::SystemTime::now();
23
24        // 等待擦除完成
25        loop {
26            let elapsed = now.elapsed().unwrap().as_millis();
27            if elapsed > 30000 {
28                return Err(Error::timeout(format!(
29                    "erasing flash at 0x{:08X}",
30                    address
31                )));
32            }
33
34            let mut byte = [0];
35            let ret = tool.port().read_exact(&mut byte);
36            if ret.is_err() {
37                continue;
38            }
39            buffer.push(byte[0]);
40
41            if buffer.windows(2).any(|window| window == b"OK") {
42                break;
43            }
44        }
45
46        progress_bar.finish_with_message("Erase complete");
47
48        Ok(())
49    }
50
51    /// 擦除指定区域的通用实现
52    pub fn erase_region<T>(tool: &mut T, address: u32, len: u32) -> Result<()>
53    where
54        T: SifliToolTrait + RamCommand,
55    {
56        let progress = tool.progress();
57        let progress_bar = progress.create_spinner(format!(
58            "Erasing region at 0x{:08X} (size: 0x{:08X})...",
59            address, len
60        ));
61
62        // 发送擦除区域命令
63        let _ = tool.command(Command::Erase { address, len });
64
65        let mut buffer = Vec::new();
66        let now = std::time::SystemTime::now();
67
68        // 等待擦除完成
69        loop {
70            let elapsed = now.elapsed().unwrap().as_millis();
71            if elapsed > 30000 {
72                return Err(Error::timeout(format!(
73                    "erasing region 0x{:08X}..0x{:08X}",
74                    address,
75                    address + len.saturating_sub(1)
76                )));
77            }
78
79            let mut byte = [0];
80            let ret = tool.port().read_exact(&mut byte);
81            if ret.is_err() {
82                continue;
83            }
84            buffer.push(byte[0]);
85
86            if buffer.windows(2).any(|window| window == b"OK") {
87                break;
88            }
89        }
90
91        let end_address = address + len - 1;
92        progress_bar.finish_with_message(format!(
93            "Region erased successfully for 0x{:08X}..0x{:08X}",
94            address, end_address
95        ));
96
97        Ok(())
98    }
99
100    /// 解析擦除地址参数
101    pub fn parse_address(address_str: &str) -> Result<u32> {
102        Utils::str_to_u32(address_str)
103            .map_err(|e| Error::invalid_input(format!("Invalid address '{}': {}", address_str, e)))
104    }
105
106    /// 解析区域参数 (address:size格式)
107    pub fn parse_region(region_spec: &str) -> Result<(u32, u32)> {
108        let Some((addr_str, size_str)) = region_spec.split_once(':') else {
109            return Err(Error::invalid_input(format!(
110                "Invalid region format: {}. Expected: address:size",
111                region_spec
112            )));
113        };
114
115        let address = Utils::str_to_u32(addr_str)
116            .map_err(|e| Error::invalid_input(format!("Invalid address '{}': {}", addr_str, e)))?;
117        let len = Utils::str_to_u32(size_str)
118            .map_err(|e| Error::invalid_input(format!("Invalid size '{}': {}", size_str, e)))?;
119
120        Ok((address, len))
121    }
122}
sftool_lib/common/erase_flash.rs

sftool_lib/common/
erase_flash.rs
