feat: add read_element function for raw bloc or field parsing
This commit is contained in:
parent
1b2303a47c
commit
2ba548aa28
@ -54,6 +54,20 @@ pub fn read_element_size(bytes: &[u8]) -> Result<ElementSize, MemoryParsingError
|
|||||||
Ok(element_size)
|
Ok(element_size)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct Element {
|
||||||
|
pub pad: usize,
|
||||||
|
pub contenu: Vec<u8>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn read_element(bytes: &[u8]) -> Element {
|
||||||
|
let ElementSize { size, pad } = read_element_size(bytes).unwrap();
|
||||||
|
let contenu = bytes[pad..pad+size].to_vec();
|
||||||
|
Element {
|
||||||
|
pad: pad+size,
|
||||||
|
contenu,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test_read_element_size {
|
mod test_read_element_size {
|
||||||
use super::*;
|
use super::*;
|
||||||
@ -117,3 +131,35 @@ mod test_read_element_size {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test_read_element {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn short_size() {
|
||||||
|
let bytes = [51,
|
||||||
|
1, 48, 1, 56, 11, 57, 57, 55, 48, 48,
|
||||||
|
53, 50, 52, 49, 57, 52, 1, 52, 2, 50,
|
||||||
|
50, 17, 80, 72, 65, 82, 77, 65, 67, 73,
|
||||||
|
69, 78, 48, 48, 53, 50, 52, 49, 57, 9,
|
||||||
|
70, 82, 65, 78, 67, 79, 73, 83, 69, 1,
|
||||||
|
84,
|
||||||
|
];
|
||||||
|
let element = read_element(&bytes);
|
||||||
|
assert_eq!(element.pad, 52);
|
||||||
|
assert_eq!(element.contenu[..5], [1, 48, 1, 56, 11]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn long_size() {
|
||||||
|
let mut bytes = vec![0b_1000_0010_u8,
|
||||||
|
0b_0000_0001_u8, 0b_0000_0000_u8, // size = 256
|
||||||
|
];
|
||||||
|
// Add 256 bytes to the content
|
||||||
|
bytes.append(&mut vec![1; 256]);
|
||||||
|
let element = read_element(&bytes);
|
||||||
|
assert_eq!(element.pad, 259);
|
||||||
|
assert_eq!(element.contenu.len(), 256);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user