From 42a5916cf742779d016eeefd1c59efbdfab64feb Mon Sep 17 00:00:00 2001 From: metamuffin Date: Mon, 10 Mar 2025 19:53:48 +0100 Subject: successfully read all objects of a file --- src/bin/debug.rs | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++++ src/bin/json.rs | 35 ++++++++++++++++++++++++ src/bin/parse.rs | 81 -------------------------------------------------------- 3 files changed, 112 insertions(+), 81 deletions(-) create mode 100644 src/bin/debug.rs create mode 100644 src/bin/json.rs delete mode 100644 src/bin/parse.rs (limited to 'src/bin') diff --git a/src/bin/debug.rs b/src/bin/debug.rs new file mode 100644 index 0000000..5da7c03 --- /dev/null +++ b/src/bin/debug.rs @@ -0,0 +1,77 @@ +use log::debug; +use std::{ + env::args, + fs::File, + io::{BufReader, Seek, SeekFrom}, +}; +use unity_tools::{object::read_value, serialized_file::read_serialized_file, unityfs::UnityFS}; + +fn main() -> anyhow::Result<()> { + env_logger::init_from_env("LOG"); + let file = BufReader::new(File::open(args().nth(1).unwrap())?); + let mut fs = UnityFS::open(file)?; + + for node in fs.nodes().to_vec() { + if node.name.ends_with(".resource") || node.name.ends_with(".resS") { + continue; + } + let mut cab = fs.read(&node)?; + // let mut writer = File::create(format!("/tmp/{}", node.name))?; + // std::io::copy(&mut cab, &mut writer)?; + // continue; + + let file = read_serialized_file(&mut cab)?; + let e = file.endianness; + + for ob in file.objects { + cab.seek(SeekFrom::Start(ob.data_offset))?; + + // eprintln!("{:#?}", ob); + let typetree = if ob.type_id < 0 { + unimplemented!() + } else { + // file.types + // .iter() + // .find(|t| t.class_id == ob.type_id) + // .expect("unknown type") + &file.types[ob.type_id as usize] + }; + // fn print_types(tt: &TypeTreeNode) { + // println!("{}", tt.type_string); + // for c in &tt.children { + // print_types(&c); + // } + // } + // fn print_crit_types(tt: &TypeTreeNode) { + // let mut crit = tt.byte_size == -1 || tt.children.is_empty(); + // for c in &tt.children { + // print_crit_types(&c); + // crit &= c.byte_size != -1 + // } + // if crit { + // println!("{}", tt.type_string); + // } + // } + // if let Some(tree) = &typetree.type_tree { + // println!("{}", tree.type_string); + // print_crit_types(tree); + // print_types(tree); + // } + // eprintln!("{typetree:#?}"); + + let value = read_value(typetree.type_tree.as_ref().unwrap(), e, &mut cab)?; + + // if let Value::Object { class, .. } = &value { + // println!("{class}") + // } + + debug!( + "{}", + serde_json::to_string_pretty(&value.to_json()).unwrap() + ) + } + // eprintln!("{:#?}", file.types); + } + + Ok(()) +} diff --git a/src/bin/json.rs b/src/bin/json.rs new file mode 100644 index 0000000..a9685ed --- /dev/null +++ b/src/bin/json.rs @@ -0,0 +1,35 @@ +use std::{ + env::args, + fs::File, + io::{BufReader, Seek, SeekFrom, stdout}, +}; +use unity_tools::{object::read_value, serialized_file::read_serialized_file, unityfs::UnityFS}; + +fn main() -> anyhow::Result<()> { + env_logger::init_from_env("LOG"); + let file = BufReader::new(File::open(args().nth(1).unwrap())?); + let mut fs = UnityFS::open(file)?; + + for node in fs.nodes().to_vec() { + if node.name.ends_with(".resource") || node.name.ends_with(".resS") { + continue; + } + let mut cab = fs.read(&node)?; + let file = read_serialized_file(&mut cab)?; + let e = file.endianness; + + for ob in file.objects { + cab.seek(SeekFrom::Start(ob.data_offset))?; + let typetree = if ob.type_id < 0 { + unimplemented!() + } else { + &file.types[ob.type_id as usize] + }; + let value = read_value(typetree.type_tree.as_ref().unwrap(), e, &mut cab)?; + serde_json::to_writer(stdout(), &value.to_json()).unwrap(); + println!() + } + } + + Ok(()) +} diff --git a/src/bin/parse.rs b/src/bin/parse.rs deleted file mode 100644 index 273cffd..0000000 --- a/src/bin/parse.rs +++ /dev/null @@ -1,81 +0,0 @@ -use log::debug; -use std::{ - env::args, - fs::File, - io::{BufReader, Read, Seek, SeekFrom}, -}; -use unity_tools::{ - object::{Value, read_value}, - serialized_file::{TypeTreeNode, read_serialized_file}, - unityfs::UnityFS, -}; - -fn main() -> anyhow::Result<()> { - env_logger::init_from_env("LOG"); - let file = BufReader::new(File::open(args().nth(1).unwrap())?); - let mut fs = UnityFS::open(file)?; - - for node in fs.nodes().to_vec() { - if node.name.ends_with(".resource") || node.name.ends_with(".resS") { - continue; - } - let mut cab = fs.read(&node)?; - // let mut writer = File::create(format!("/tmp/{}", node.name))?; - // std::io::copy(&mut cab, &mut writer)?; - // continue; - - let file = read_serialized_file(&mut cab)?; - let e = file.endianness; - - for ob in file.objects { - cab.seek(SeekFrom::Start(ob.data_offset))?; - - // eprintln!("{:#?}", ob); - let typetree = if ob.type_id < 0 { - unimplemented!() - } else { - // file.types - // .iter() - // .find(|t| t.class_id == ob.type_id) - // .expect("unknown type") - &file.types[ob.type_id as usize] - }; - // fn print_types(tt: &TypeTreeNode) { - // println!("{}", tt.type_string); - // for c in &tt.children { - // print_types(&c); - // } - // } - // fn print_crit_types(tt: &TypeTreeNode) { - // let mut crit = tt.byte_size == -1; - // for c in &tt.children { - // print_crit_types(&c); - // crit &= c.byte_size != -1 - // } - // if crit { - // println!("{}", tt.type_string); - // } - // } - if let Some(tree) = &typetree.type_tree { - println!("{}", tree.type_string); - // print_crit_types(tree); - // print_types(tree); - } - // eprintln!("{typetree:#?}"); - - let value = read_value(typetree.type_tree.as_ref().unwrap(), e, &mut cab)?; - - // if let Value::Object { class, .. } = &value { - // println!("{class}") - // } - - debug!( - "{}", - serde_json::to_string_pretty(&value.to_json()).unwrap() - ) - } - // eprintln!("{:#?}", file.types); - } - - Ok(()) -} -- cgit v1.2.3-70-g09d2