parent
ce85e8e898
commit
353d5e6651
@ -0,0 +1,75 @@
|
||||
use std::collections::HashMap;
|
||||
use std::fs;
|
||||
use std::io::{self, Read, Seek, Write};
|
||||
use std::path;
|
||||
|
||||
use nom::Err::{Error as NomError, Failure as NomFailure};
|
||||
|
||||
use crate::name_hashes;
|
||||
use crate::parsers::bhd;
|
||||
use crate::utils::fs as fs_utils;
|
||||
|
||||
/// Parse a BHD file and extract its content.
|
||||
pub fn extract_bhd(
|
||||
bhd_path: &str,
|
||||
names: &HashMap<String, String>,
|
||||
output_path: &str
|
||||
) -> Result<(), io::Error> {
|
||||
let mut bhd_file = fs::File::open(bhd_path)?;
|
||||
let file_len = bhd_file.metadata()?.len() as usize;
|
||||
let mut bhd_data = vec![0u8; file_len];
|
||||
bhd_file.read_exact(&mut bhd_data)?;
|
||||
let bhd = match bhd::parse(&bhd_data) {
|
||||
Ok((_, bhd)) => { bhd }
|
||||
Err(NomError(e)) | Err(NomFailure(e)) => {
|
||||
let (_, kind) = e;
|
||||
let reason = format!("{:?} {:?}", kind, kind.description());
|
||||
eprintln!("BHD parsing failed: {}", reason); return Ok(())
|
||||
}
|
||||
e => {
|
||||
eprintln!("Unknown error: {:?}", e); return Ok(())
|
||||
}
|
||||
};
|
||||
|
||||
let bdt_path = path::PathBuf::from(bhd_path).with_extension("bdt");
|
||||
let mut bdt_file = fs::File::open(bdt_path.to_str().unwrap())?;
|
||||
|
||||
extract_files(&bhd, &mut bdt_file, &names, &output_path)?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Extract files from a BHD/BDT pair.
|
||||
pub fn extract_files(
|
||||
bhd: &bhd::Bhd,
|
||||
bdt_file: &mut fs::File,
|
||||
names: &HashMap<String, String>,
|
||||
output_path: &str,
|
||||
) -> Result<(), io::Error> {
|
||||
let output_path = path::Path::new(output_path);
|
||||
fs_utils::ensure_dir_exists(output_path)?;
|
||||
|
||||
for bucket in &bhd.buckets {
|
||||
for entry in bucket {
|
||||
bdt_file.seek(io::SeekFrom::Start(entry.offset))?;
|
||||
let mut data = vec![0; entry.size as usize];
|
||||
bdt_file.read_exact(&mut data)?;
|
||||
|
||||
let hash_str = name_hashes::hash_as_string(entry.hash);
|
||||
let rel_path: &str = match names.get(&hash_str) {
|
||||
Some(path) => {
|
||||
path.trim_start_matches("/")
|
||||
}
|
||||
_ => {
|
||||
eprintln!("No name for {}, using hash as name.", hash_str);
|
||||
&hash_str
|
||||
}
|
||||
};
|
||||
let file_path = output_path.join(rel_path);
|
||||
fs_utils::ensure_dir_exists(file_path.parent().unwrap())?;
|
||||
let mut output_file = fs::File::create(file_path)?;
|
||||
output_file.write_all(&data)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
@ -0,0 +1,14 @@
|
||||
use std::fs;
|
||||
use std::io;
|
||||
use std::path::Path;
|
||||
|
||||
/// Ensure a directory exists, creating it with parents if necessary.
|
||||
pub fn ensure_dir_exists(path: &Path) -> Result<(), io::Error> {
|
||||
if !path.is_dir() {
|
||||
if path.exists() {
|
||||
return Err(io::Error::new(io::ErrorKind::AlreadyExists, "Not a directory."));
|
||||
}
|
||||
fs::create_dir_all(&path)?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
Loading…
Reference in new issue