Implement decompression of .rpgsave files

This commit is contained in:
Rekai Musuka 2020-06-17 18:27:32 -05:00
parent 17672ce114
commit 30d4a861f2
3 changed files with 172 additions and 93 deletions

6
Cargo.lock generated
View File

@ -62,12 +62,18 @@ version = "0.2.71"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9457b06509d27052635f90d6466700c65095fdf75409b3fbdd903e988b886f49"
[[package]]
name = "lz-string"
version = "0.0.1"
source = "git+https://github.com/adumbidiot/lz-string-rs#f968dd54b0b7d0a560235f50f204557d84d1f0c0"
[[package]]
name = "rpgmv_decryptor"
version = "0.1.0"
dependencies = [
"anyhow",
"clap",
"lz-string",
]
[[package]]

View File

@ -9,3 +9,4 @@ edition = "2018"
[dependencies]
anyhow = "1.0.31"
clap = "2.33.1"
lz-string = { git = "https://github.com/adumbidiot/lz-string-rs" }

View File

@ -1,11 +1,10 @@
use anyhow::{anyhow, Context, Result};
use clap::{App, Arg};
use std::fs::File;
use std::io::{Read, Write};
use std::io::Write;
use std::path::Path;
/// # RPGMV Decryptor
///
fn main() -> Result<()> {
let matches = App::new("RPGMV Decoder")
.version("0.1.0")
@ -25,15 +24,34 @@ fn main() -> Result<()> {
.long("key")
.value_name("KEY")
.help("The Encryption Key (look in www/data/System.json or www/js/rpg_core.js")
.takes_value(true)
.required(true),
.takes_value(true),
)
.get_matches();
let path_string = matches.value_of("path").context("No file path provided")?;
let key_string = matches.value_of("key").context("No key provided")?;
let path = Path::new(path_string);
let filename = path
.file_stem()
.context("Unable to determine file stem")?
.to_str()
.context("file stem is not valid UTF-8")?;
let path_str = path.to_string_lossy();
match path.extension() {
Some(ext) => match ext.to_str() {
Some("rpgsave") => {
use rpgmv_save::*;
let decompressed = decompress_save(path)?;
write_decompressed_save(format!("{}.json", filename), &decompressed)?;
Ok(())
}
Some("rpgmvp") | Some("rpgmvm") | Some("rpgmvo") => {
use rpgmv_media::*;
let key_string = matches.value_of("key").context("No key provided")?;
let key = get_key_bytes(key_string)?;
// let default_rpgmv_header: [u8; 16] = [
@ -46,42 +64,76 @@ fn main() -> Result<()> {
let mut file;
let filename = path
.file_stem()
.context("Unable to determine file stem")?
.to_str()
.context("Unable to convert file stem to UTF-8 string")?;
match detect_original_extension(path)? {
RPGFile::Photo => file = File::create(format!("{}.png", filename))?, // .rpgmvps are .pngs
RPGFile::Video => file = File::create(format!("{}.m4a", filename))?, // .rpgmvms are .m4as
RPGFile::Audio => file = File::create(format!("{}.ogg", filename))?, // .rpgmvos are .oggs
}
file.write_all(&decrypted)?;
Ok(file.write_all(&decrypted)?)
}
Some(ext_str) => Err(anyhow!("{} is not a supported .ext", ext_str)),
None => Err(anyhow!("{}'s .ext is not valid UTF-8", path_str)),
},
None => Err(anyhow!("Unable to determine the extension of {}", path_str)),
}
}
mod rpgmv_save {
use anyhow::{anyhow, Result};
use std::char::decode_utf16;
use std::fs::File;
use std::io::{Read, Write};
use std::path::Path;
pub fn decompress_save<P: AsRef<Path>>(path: P) -> Result<String> {
let mut save_file = File::open(path.as_ref())?;
let mut base64_string = String::new();
save_file.read_to_string(&mut base64_string)?;
let maybe_compressed = convert_base64_string(&base64_string);
match maybe_compressed {
Some(compressed) => Ok(lz_string::decompress(&compressed, 32)
.ok_or_else(|| anyhow!("Failed to decompress .rpgsave"))?),
None => Err(anyhow!("Failed to convert the .rpgsave from base64")),
}
}
pub fn write_decompressed_save<P: AsRef<Path>>(path: P, decompressed: &str) -> Result<()> {
let mut out = File::create(path.as_ref())?;
out.write_all(decompressed.as_bytes())?;
Ok(())
}
/// Determines the original file type for the encrypted .rpgmv* file
fn detect_original_extension<P: AsRef<Path>>(path: P) -> Result<RPGFile> {
match path.as_ref().extension() {
Some(ext) => {
let ext = ext
.to_str()
.context("File extension was not UTF-8 Compatible")?;
pub fn convert_base64_string(base64: &str) -> Option<String> {
let codes = convert_to_utf16_char_codes(base64)?;
match ext {
"rpgmvp" => Ok(RPGFile::Photo),
"rpgmvm" => Ok(RPGFile::Video),
"rpgmvo" => Ok(RPGFile::Audio),
_ => Err(anyhow!("File was lacking a supported file extension")),
}
}
None => Err(anyhow!("File is lacking a file extension")),
let res = decode_utf16(codes.iter().cloned())
.map(|res| res.expect("UTF-16 Decoding failed"))
.collect();
Some(res)
}
pub fn convert_to_utf16_char_codes(base64: &str) -> Option<Vec<u16>> {
let alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
let mut codes: Vec<u16> = Vec::with_capacity(base64.chars().count());
for c in base64.chars() {
codes.push(alphabet.find(c)? as u16);
}
Some(codes)
}
}
mod rpgmv_media {
use anyhow::{anyhow, Context, Result};
use std::fs::File;
use std::io::Read;
use std::path::Path;
/// get_key_bytes takes a 32-character hexadecimal string and interperets
/// every two characters as an unsined 8-bit integer.
///
@ -89,7 +141,7 @@ fn detect_original_extension<P: AsRef<Path>>(path: P) -> Result<RPGFile> {
/// `71e7bdd9b28010980c2a4e8a1386e100` will be split up as:
///
/// `0x71`, `0xe7`, `0xbd`, `0xd9`, `0xb2`, `0x80`, `0x10`, `0x98`, `0x0c`, `0x2a`, `0x4e`, `0x8a`, `0x13`, `0x86`, `0xe1`, and `0x00`
fn get_key_bytes(key: &str) -> Result<Vec<u8>> {
pub fn get_key_bytes(key: &str) -> Result<Vec<u8>> {
let sub_len = 2;
let mut key_buf = Vec::with_capacity(key.len() / 2);
@ -108,7 +160,7 @@ fn get_key_bytes(key: &str) -> Result<Vec<u8>> {
}
/// get_file_data returns a buffer containing a data of a file which is on disk.
fn get_file_data<P: AsRef<Path>>(path: P) -> Result<Vec<u8>> {
pub fn get_file_data<P: AsRef<Path>>(path: P) -> Result<Vec<u8>> {
let mut buf = Vec::new();
let mut file = File::open(path.as_ref())?;
@ -128,7 +180,7 @@ fn _encrypt(_bytes: Vec<u8>, _key: Vec<u8>) -> Vec<u8> {
/// 16 values which we can use to perform a bitwise XOR. The bitwise XOR will undo the cipher, returning the header to
/// it's original position. At this point, byte #17 -> end of file will now be recognized as their respected unencrypted
// file formats.
fn decrypt(bytes: Vec<u8>, key: Vec<u8>) -> Vec<u8> {
pub fn decrypt(bytes: Vec<u8>, key: Vec<u8>) -> Vec<u8> {
let mut decrypted = bytes[16..].to_owned(); // Throw out the first 16 bytes (the RPGMV Header)
for i in 0..16 as usize {
@ -138,9 +190,29 @@ fn decrypt(bytes: Vec<u8>, key: Vec<u8>) -> Vec<u8> {
decrypted
}
/// Determines the original file type for the encrypted .rpgmv* file
pub fn detect_original_extension<P: AsRef<Path>>(path: P) -> Result<RPGFile> {
match path.as_ref().extension() {
Some(ext) => {
let ext = ext
.to_str()
.context("File extension was not UTF-8 Compatible")?;
match ext {
"rpgmvp" => Ok(RPGFile::Photo),
"rpgmvm" => Ok(RPGFile::Video),
"rpgmvo" => Ok(RPGFile::Audio),
_ => Err(anyhow!("File was lacking a supported file extension")),
}
}
None => Err(anyhow!("File is lacking a file extension")),
}
}
/// An enum that represents the types of supported RPGMV media files.
enum RPGFile {
pub enum RPGFile {
Photo,
Video,
Audio,
}
}