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
pub use self::error::HeaderError;
mod error;
#[derive(Clone)]
pub struct Header {
raw_blocksize: u8,
max_blocksize: u32,
}
impl Header {
pub fn parse(buf: [u8; 4]) -> Result<Self, HeaderError> {
let signature = &buf[..2];
if signature != b"BZ" {
return Err(HeaderError::InvalidSignature);
}
let version = buf[2];
if version != b'h' {
return Err(HeaderError::UnsupportedVersion);
}
let hundred_k_blocksize = buf[3];
match hundred_k_blocksize {
b'1'..=b'9' => {
let raw_blocksize = hundred_k_blocksize - b'0';
Self::from_raw_blocksize(raw_blocksize)
}
_ => Err(HeaderError::InvalidBlockSize),
}
}
pub fn from_raw_blocksize(raw_blocksize: u8) -> Result<Self, HeaderError> {
if raw_blocksize < 1 || raw_blocksize > 9 {
return Err(HeaderError::InvalidBlockSize);
}
let max_blocksize = 100 * 1000 * u32::from(raw_blocksize);
Ok(Self {
raw_blocksize,
max_blocksize,
})
}
pub fn raw_blocksize(&self) -> u8 {
self.raw_blocksize
}
pub fn max_blocksize(&self) -> u32 {
self.max_blocksize
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn valid_9k() {
let header = Header::parse(*b"BZh9").unwrap();
assert_eq!(header.raw_blocksize(), 9);
assert_eq!(header.max_blocksize(), 900000);
}
}