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