NinjaCheetah/rustii
1
0
Fork 0
mirror of https://github.com/NinjaCheetah/rustii.git synced 2026-03-04 11:55:27 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added WAD parsing, allows for WAD packing/unpacking

Browse Source
This commit is contained in:
Campbell
2025-03-18 18:01:17 -04:00
parent 93f2103763
commit 83dc83d2d6
8 changed files with 243 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/bin/rustii/main.rs
View File

@@ -1,28 +1,37 @@
// Sample file for testing rustii library stuff.
use std::fs;
use rustii::title::{tmd, ticket, content};
use rustii::title::{tmd, ticket, content, crypto, wad};
fn main() {
let data = fs::read("title.tmd").unwrap();
let tmd = tmd::TMD::from_bytes(&data).unwrap();
let data = fs::read("sm.wad").unwrap();
let wad = wad::WAD::from_bytes(&data).unwrap();
println!("size of tmd: {:?}", wad.tmd().len());
let tmd = tmd::TMD::from_bytes(&wad.tmd()).unwrap();
println!("num content records: {:?}", tmd.content_records.len());
println!("first record data: {:?}", tmd.content_records.first().unwrap());
assert_eq!(data, tmd.to_vec().unwrap());
assert_eq!(wad.tmd(), tmd.to_bytes().unwrap());
let data = fs::read("tik").unwrap();
let tik = ticket::Ticket::from_bytes(&data).unwrap();
let tik = ticket::Ticket::from_bytes(&wad.ticket()).unwrap();
println!("title version from ticket is: {:?}", tik.title_version);
println!("title key (enc): {:?}", tik.title_key);
println!("title key (dec): {:?}", tik.dec_title_key());
assert_eq!(data, tik.to_vec().unwrap());
assert_eq!(wad.ticket(), tik.to_bytes().unwrap());
let data = fs::read("content-blob").unwrap();
let content_region = content::ContentRegion::from_bytes(&data, tmd.content_records).unwrap();
assert_eq!(data, content_region.to_bytes().unwrap());
let content_region = content::ContentRegion::from_bytes(&wad.content(), tmd.content_records).unwrap();
assert_eq!(wad.content(), content_region.to_bytes().unwrap());
println!("content OK");
let content_dec = content_region.get_content_by_index(0, tik.dec_title_key()).unwrap();
println!("content dec from index: {:?}", content_dec);
let content = content_region.get_content_by_cid(150, tik.dec_title_key()).unwrap();
println!("content dec from cid: {:?}", content);
let content = content_region.get_enc_content_by_index(0).unwrap();
assert_eq!(content, crypto::encrypt_content(&content_dec, tik.dec_title_key(), 0, content_region.content_records[0].content_size));
println!("content re-encrypted OK");
println!("wad header: {:?}", wad.header);
let repacked = wad.to_bytes().unwrap();
assert_eq!(repacked, data);
println!("wad packed OK");
}