xtex/huextract
1
0
Fork 0
mirror of https://github.com/echo-devim/huextract.git synced 2024-10-18 01:24:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Optionally disable checksum verification in Input

Browse source
This commit is contained in:
Romain Garbage 2022-09-08 01:09:24 +02:00
parent 58c1e18eb8
commit 4ed3331eac
No known key found for this signature in database
GPG key ID: 9DFBAECAE17DF2EE
1 changed files with 18 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/input.rs
View file

@ -118,30 +118,32 @@ impl Input {
}
/// Extract the content of the img files to disk
pub fn extract_img(&mut self) -> Result<(), Error> {
pub fn extract_img(&mut self, no_checksum_verification: bool) -> Result<(), Error> {
for part in self.img_parts.clone() {
let filename = format!("{}.img", part.header.filename()?);
let offset = part.offset + part.header.headersize();
let size = part.header.filesize() as usize;
print!("Extracting {filename}... ");
self.write_to_disk(&filename, offset, size)?;
print!("Done");
// Verify file checksum
let mut checksum = Vec::new();
self.write_to(
&mut checksum,
part.offset + img_header::FILE_CHECKSUM_OFFSET,
part.header.filechecksumsize(),
)?;
let mut img = BufReader::new(File::open(&filename)?);
print!("... Verifying checksum... ");
if checksum
== Crc::new(part.header.blocksize() as usize).compute_file_checksum(&mut img)?
{
println!("OK");
if no_checksum_verification {
println!("Done");
} else {
println!("Error");
// Verify file checksum
let mut checksum = Vec::new();
self.write_to(
&mut checksum,
part.offset + img_header::FILE_CHECKSUM_OFFSET,
part.header.filechecksumsize(),
)?;
let mut img = BufReader::new(File::open(&filename)?);
if checksum
== Crc::new(part.header.blocksize() as usize).compute_file_checksum(&mut img)?
{
println!("Checksum OK");
} else {
println!("Checksum error");
}
}
}
Ok(())