unixtract/src/formats/slp.rs

use std::path::{Path};
use std::fs::{self, File, OpenOptions};
use std::io::{Write, Seek, SeekFrom};
use binrw::{BinRead, BinReaderExt};

use crate::utils::common;

#[derive(BinRead)]
struct Header {
    #[br(count = 4)] _magic_bytes: Vec<u8>,
    #[br(count = 8)] version_bytes: Vec<u8>,
    #[br(count = 16)] model_bytes: Vec<u8>,
    #[br(count = 16)] firmware_bytes: Vec<u8>,
    _unk: u32,
    #[br(count = 8)] check: Vec<u8>,
    #[br(count = 8)] _unk2: Vec<u8>,
}
impl Header {
    fn version(&self) -> String {
        common::string_from_bytes(&self.version_bytes)
    }
    fn model(&self) -> String {
        common::string_from_bytes(&self.model_bytes)
    }
    fn firmware(&self) -> String {
        common::string_from_bytes(&self.firmware_bytes)
    }
    fn is_new_type(&self) -> bool {
        self.check == b"\x01VER_PR1"
    }
}

#[derive(BinRead)]
struct EntryOld {
    size: u32,
    _unk: u32,
    offset: u32,
    _unk2: u32,
}

#[derive(BinRead)]
struct EntryNew {
    size: u32,
    _unk: u32,
    offset: u32,
    #[br(count = 12)] _unk2: Vec<u8>,
}

pub fn is_slp_file(file: &File) -> bool {
    let header = common::read_file(&file, 0, 4).expect("Failed to read from file.");
    if header == b"SLP\x00" {
        true
    } else {
        false
    }
}

pub fn extract_slp(mut file: &File, output_folder: &str) -> Result<(), Box<dyn std::error::Error>> {
    let header: Header = file.read_le()?;

    println!("\nFile info:\nModel: {}\nVersion: {}\nFirmware: {}\nNew type: {}\n",
            header.model(), header.version(), header.firmware(), header.is_new_type());

    let mut first_entry_offset = 0;
    let mut entries: Vec<EntryOld> = Vec::new();

    for i in 0..100 {
        if (i != 0) && (file.stream_position()? >= first_entry_offset) {
            break
        }
        let offset;
        let size;
        if header.is_new_type() {
            let entry: EntryNew = file.read_le()?;
            offset = entry.offset;
            size = entry.size;
        } else {
            let entry: EntryOld = file.read_le()?;
            offset = entry.offset;
            size = entry.size;
        }
        if i == 0 {
            first_entry_offset = offset as u64;
        }
        println!("{}. Offset: {}, Size: {}", i + 1, offset, size);
        entries.push(EntryOld {size: size, _unk: 0, offset: offset, _unk2: 0});
    }

    let mut i = 1;
    for entry in &entries {
        println!("\nExtracting {}/{}", i, &entries.len());
        file.seek(SeekFrom::Start(entry.offset.into()))?;
        let data = common::read_exact(&mut file, entry.size as usize)?;

        let output_path = Path::new(&output_folder).join(format!("{}.bin", i));

        fs::create_dir_all(&output_folder)?;
        let mut out_file = OpenOptions::new()
            .write(true)
            .create(true)
            .open(output_path)?;
            
        out_file.write_all(&data)?;

        println!("- Saved file!");

        i += 1;
    }

    println!("\nExtraction finished!");

    Ok(())
}