dcx: better API

src/bin/ironring.rs

@@ -56,6 +56,11 @@ fn main() {
             .arg(Arg::with_name("output")
                 .takes_value(true)
                 .required(false)))
+        .subcommand(SubCommand::with_name("bnd")
+            .about("?TODO?")
+            .arg(Arg::with_name("file")
+                .takes_value(true)
+                .required(true)))
         .get_matches();
 
     process::exit(match matches.subcommand() {

src/unpackers/dcx.rs

@@ -7,7 +7,16 @@ use nom::Err::{Error as NomError, Failure as NomFailure};
 use crate::parsers::dcx;
 use crate::unpackers::errors::{self as unpackers_errors, UnpackError};
 
+/// Extract DCX file content.
 pub fn extract_dcx(dcx_path: &str, output_path: &str) -> Result<(), UnpackError> {
+    let (_dcx, decomp_data) = load_dcx(dcx_path)?;
+    let mut output_file = fs::File::create(output_path)?;
+    output_file.write_all(&decomp_data)?;
+    Ok(())
+}
+
+/// Load a DCX file in memory along with its decompressed content.
+pub fn load_dcx(dcx_path: &str) -> Result<(dcx::Dcx, Vec<u8>), UnpackError> {
     let mut dcx_file = fs::File::open(dcx_path)?;
     let file_len = dcx_file.metadata()?.len() as usize;
     let mut dcx_data = vec![0u8; file_len];
@@ -24,16 +33,13 @@ pub fn extract_dcx(dcx_path: &str, output_path: &str) -> Result<(), UnpackError>
     };
 
     let decomp_data = decompress_dcx(&dcx, data)?;
-
+    Ok((dcx, decomp_data))
-    let mut output_file = fs::File::create(output_path)?;
-    output_file.write_all(&decomp_data)?;
-    Ok(())
 }
 
 fn decompress_dcx(dcx: &dcx::Dcx, comp_data: &[u8]) -> Result<Vec<u8>, UnpackError> {
     let method: &[u8] = dcx.params.method.as_slice();
     if method == b"DFLT" {
-        decompress_dcx_deflate(dcx, comp_data)
+        decompress_dcx_dflt(dcx, comp_data)
     } else {
         let method_string = match std::str::from_utf8(method) {
             Ok(s) => { String::from(s) }
@@ -43,7 +49,7 @@ fn decompress_dcx(dcx: &dcx::Dcx, comp_data: &[u8]) -> Result<Vec<u8>, UnpackErr
     }
 }
 
-fn decompress_dcx_deflate(dcx: &dcx::Dcx, comp_data: &[u8]) -> Result<Vec<u8>, UnpackError> {
+fn decompress_dcx_dflt(dcx: &dcx::Dcx, comp_data: &[u8]) -> Result<Vec<u8>, UnpackError> {
     let mut data = vec![0u8; dcx.sizes.uncompressed_size as usize];
     let mut deflater = ZlibDecoder::new(comp_data);
     deflater.read_exact(&mut data)?;