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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
use alloc::vec::Vec;
use super::{Ec, EcFlash};
pub struct Flasher {
ec: EcFlash,
size: usize,
}
impl Flasher {
pub fn new(mut ec: EcFlash) -> Self {
let size = ec.size();
Self {
ec,
size,
}
}
unsafe fn enter_follow_mode(&mut self) -> Result<(), ()> {
self.ec.cmd(1)
}
unsafe fn spi_cmd(&mut self, cmd: u8) -> Result<(), ()> {
self.ec.cmd(2)?;
self.ec.cmd(cmd)
}
unsafe fn spi_write(&mut self, value: u8) -> Result<(), ()> {
self.ec.cmd(3)?;
self.ec.cmd(value)
}
unsafe fn spi_read(&mut self) -> Result<u8, ()> {
self.ec.cmd(4)?;
self.ec.read()
}
unsafe fn exit_follow_mode(&mut self) -> Result<(), ()> {
self.ec.cmd(5)
}
unsafe fn spi_wait(&mut self) -> Result<(), ()> {
self.enter_follow_mode()?;
self.spi_cmd(5)?;
while self.spi_read()? & 1 > 0 {}
self.exit_follow_mode()
}
unsafe fn spi_write_enable(&mut self) -> Result<(), ()> {
self.spi_wait()?;
self.enter_follow_mode()?;
self.spi_cmd(6)?;
self.enter_follow_mode()?;
self.spi_cmd(5)?;
while self.spi_read()? & 3 != 2 {}
self.exit_follow_mode()
}
unsafe fn spi_write_disable(&mut self) -> Result<(), ()> {
self.spi_wait()?;
self.enter_follow_mode()?;
self.spi_cmd(4)?;
self.enter_follow_mode()?;
self.spi_cmd(5)?;
while self.spi_read()? & 2 > 0 {}
self.exit_follow_mode()
}
pub unsafe fn start(&mut self) -> Result<u8, ()> {
self.ec.cmd(0xDC)?;
self.ec.read()
}
pub unsafe fn read<F: Fn(usize)>(&mut self, callback: F) -> Result<Vec<u8>, ()> {
let mut buf = Vec::with_capacity(self.size);
for sector in 0..self.size/65536 {
self.spi_write_disable()?;
self.spi_wait()?;
self.enter_follow_mode()?;
self.spi_cmd(0x0B)?;
self.spi_write(sector as u8)?;
self.spi_write(0)?;
self.spi_write(0)?;
self.spi_write(0)?;
for _block in 0..64 {
for _ in 0..1024 {
buf.push(self.spi_read()?);
}
callback(buf.len());
}
self.spi_wait()?;
}
Ok(buf)
}
pub unsafe fn erase<F: Fn(usize)>(&mut self, callback: F) -> Result<(), ()> {
for sector in 0..self.size/65536 {
for block in 0..64 {
let index = sector * 65536 + block * 1024;
self.spi_write_enable()?;
self.enter_follow_mode()?;
self.spi_cmd(0xD7)?;
self.spi_write(sector as u8)?;
self.spi_write(block as u8)?;
self.spi_write(0)?;
self.exit_follow_mode()?;
self.spi_wait()?;
callback(index + 1024);
}
self.spi_write_enable()?;
self.enter_follow_mode()?;
self.spi_cmd(0xD8)?;
self.spi_write(sector as u8)?;
self.spi_write(0)?;
self.spi_write(0)?;
self.exit_follow_mode()?;
self.spi_wait()?;
}
Ok(())
}
pub unsafe fn write<F: Fn(usize)>(&mut self, buf: &[u8], callback: F) -> Result<(), ()> {
for sector in 0..self.size/65536 {
self.spi_write_enable()?;
self.enter_follow_mode()?;
self.spi_cmd(0xAD)?;
self.spi_write(sector as u8)?;
self.spi_write((sector >> 8) as u8)?;
self.spi_write((sector >> 16) as u8)?;
for block in 0..64 {
let index = sector * 65536 + block * 1024;
for word in 0..512 {
self.spi_write(buf.get(index + word * 2).map_or(0xFF, |x| *x))?;
self.spi_write(buf.get(index + word * 2 + 1).map_or(0xFF, |x| *x))?;
self.spi_wait()?;
self.enter_follow_mode()?;
self.spi_cmd(0xAD)?;
}
callback(index + 1024);
}
self.spi_write_disable()?;
self.spi_wait()?;
}
Ok(())
}
pub unsafe fn stop(&mut self) -> Result<(), ()> {
self.ec.cmd(0x95)?;
self.ec.cmd(0xFC)
}
}