mirror of
https://github.com/xavo95/repak.git
synced 2025-01-18 19:04:07 +00:00
Simplify matrix test
This commit is contained in:
Truman Kilen
parent
8564cf6268
commit
f94e6c460d
1 changed files with 130 additions and 122 deletions
|
|
@ -82,79 +82,144 @@ mod test {
|
||||||
|
|
||||||
static AES_KEY: &str = "lNJbw660IOC+kU7cnVQ1oeqrXyhk4J6UAZrCBbcnp94=";
|
static AES_KEY: &str = "lNJbw660IOC+kU7cnVQ1oeqrXyhk4J6UAZrCBbcnp94=";
|
||||||
|
|
||||||
macro_rules! matrix_test_read {
|
fn test_read(version: repak::Version, bytes: &[u8]) {
|
||||||
( ($($version:literal $exp_version:expr),* $(,)?), $compress:tt, $encrypt:tt, $encryptindex:tt ) => {
|
use aes::cipher::KeyInit;
|
||||||
$( mt_compress_read!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
use base64::{engine::general_purpose, Engine as _};
|
||||||
|
let key = general_purpose::STANDARD
|
||||||
|
.decode(AES_KEY)
|
||||||
|
.as_ref()
|
||||||
|
.map_err(|_| repak::Error::Base64)
|
||||||
|
.and_then(|bytes| aes::Aes256::new_from_slice(bytes).map_err(|_| repak::Error::Aes))
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let mut inner_reader = std::io::Cursor::new(bytes);
|
||||||
|
let len = inner_reader.seek(SeekFrom::End(0)).unwrap();
|
||||||
|
let mut reader = ReadCounter::new_size(inner_reader, len as usize);
|
||||||
|
|
||||||
|
let pak = repak::PakReader::new_any(&mut reader, Some(key)).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(pak.mount_point(), "../mount/point/root/");
|
||||||
|
assert_eq!(pak.version(), version);
|
||||||
|
use std::collections::HashSet;
|
||||||
|
let files: HashSet<String> = HashSet::from_iter(pak.files());
|
||||||
|
assert_eq!(
|
||||||
|
files,
|
||||||
|
HashSet::from_iter(
|
||||||
|
vec!["test.txt", "test.png", "zeros.bin", "directory/nested.txt"]
|
||||||
|
.into_iter()
|
||||||
|
.map(String::from)
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
||||||
|
for file in files {
|
||||||
|
let mut buf = vec![];
|
||||||
|
let mut writer = std::io::Cursor::new(&mut buf);
|
||||||
|
pak.read_file(&file, &mut reader, &mut writer).unwrap();
|
||||||
|
match file.as_str() {
|
||||||
|
"test.txt" => assert_eq!(
|
||||||
|
buf,
|
||||||
|
include_bytes!("pack/root/test.txt"),
|
||||||
|
"test.txt incorrect contents"
|
||||||
|
),
|
||||||
|
"test.png" => assert_eq!(
|
||||||
|
buf,
|
||||||
|
include_bytes!("pack/root/test.png"),
|
||||||
|
"test.png incorrect contents"
|
||||||
|
),
|
||||||
|
"zeros.bin" => assert_eq!(
|
||||||
|
buf,
|
||||||
|
include_bytes!("pack/root/zeros.bin"),
|
||||||
|
"zeros.bin incorrect contents"
|
||||||
|
),
|
||||||
|
"directory/nested.txt" => assert_eq!(
|
||||||
|
buf,
|
||||||
|
include_bytes!("pack/root/directory/nested.txt"),
|
||||||
|
"nested.txt incorrect contents"
|
||||||
|
),
|
||||||
|
name => panic!("unrecognized file {}", name),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for r in reader.into_reads() {
|
||||||
|
// sanity check. a pak file can be constructed with a lot of dead space
|
||||||
|
// which wouldn't have to be read, but so far all bytes in paks generated
|
||||||
|
// by UnrealPak are meaningful
|
||||||
|
assert!(r > 0, "every byte has been read at least once");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_write(_version: repak::Version, bytes: &[u8]) {
|
||||||
|
use aes::cipher::KeyInit;
|
||||||
|
use base64::{engine::general_purpose, Engine as _};
|
||||||
|
let key = general_purpose::STANDARD
|
||||||
|
.decode(AES_KEY)
|
||||||
|
.as_ref()
|
||||||
|
.map_err(|_| repak::Error::Base64)
|
||||||
|
.and_then(|bytes| aes::Aes256::new_from_slice(bytes).map_err(|_| repak::Error::Aes))
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let mut reader = std::io::Cursor::new(bytes);
|
||||||
|
let pak_reader = repak::PakReader::new_any(&mut reader, Some(key)).unwrap();
|
||||||
|
|
||||||
|
let writer = Cursor::new(vec![]);
|
||||||
|
let mut pak_writer = repak::PakWriter::new(
|
||||||
|
writer,
|
||||||
|
None,
|
||||||
|
pak_reader.version(),
|
||||||
|
pak_reader.mount_point().to_owned(),
|
||||||
|
Some(0x205C5A7D),
|
||||||
|
);
|
||||||
|
|
||||||
|
for path in pak_reader.files() {
|
||||||
|
let data = pak_reader.get(&path, &mut reader).unwrap();
|
||||||
|
pak_writer
|
||||||
|
.write_file(&path, &mut std::io::Cursor::new(data))
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
pak_writer.write_index().unwrap().into_inner(),
|
||||||
|
reader.into_inner()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! matrix_test {
|
||||||
|
( $name:literal, ($($version:literal $exp_version:expr),* $(,)?), $compress:tt, $encrypt:tt, $encryptindex:tt, $body:tt ) => {
|
||||||
|
$( matrix_test_compress!($name, $version, $exp_version, $compress, $encrypt, $encryptindex, $body); )*
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
macro_rules! mt_compress_read {
|
macro_rules! matrix_test_compress {
|
||||||
( $version:literal, $exp_version:expr, ($($compress:literal),* $(,)?), $encrypt:tt, $encryptindex:tt ) => {
|
( $name:literal, $version:literal, $exp_version:expr, ($($compress:literal),* $(,)?), $encrypt:tt, $encryptindex:tt, $body:tt ) => {
|
||||||
$( mt_encrypt_read!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
$( matrix_test_encrypt!($name, $version, $exp_version, $compress, $encrypt, $encryptindex, $body); )*
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
macro_rules! mt_encrypt_read {
|
macro_rules! matrix_test_encrypt {
|
||||||
( $version:literal, $exp_version:expr, $compress:literal, ($($encrypt:literal),* $(,)?), $encryptindex:tt ) => {
|
( $name:literal, $version:literal, $exp_version:expr, $compress:literal, ($($encrypt:literal),* $(,)?), $encryptindex:tt, $body:tt ) => {
|
||||||
$( mt_encryptindex_read!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
$( matrix_test_encryptindex!($name, $version, $exp_version, $compress, $encrypt, $encryptindex, $body); )*
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
macro_rules! mt_encryptindex_read {
|
macro_rules! matrix_test_encryptindex {
|
||||||
( $version:literal, $exp_version:expr, $compress:literal, $encrypt:literal, ($($encryptindex:literal),* $(,)?) ) => {
|
( $name:literal, $version:literal, $exp_version:expr, $compress:literal, $encrypt:literal, ($($encryptindex:literal),* $(,)?), $body:tt ) => {
|
||||||
$(
|
$( matrix_test_body!($name, $version, $exp_version, $compress, $encrypt, $encryptindex, $body); )*
|
||||||
paste! {
|
};
|
||||||
#[test]
|
}
|
||||||
fn [< test_read_version_ $version $compress $encrypt $encryptindex >]() {
|
|
||||||
use aes::cipher::KeyInit;
|
|
||||||
use base64::{engine::general_purpose, Engine as _};
|
|
||||||
let key = general_purpose::STANDARD
|
|
||||||
.decode(AES_KEY)
|
|
||||||
.as_ref()
|
|
||||||
.map_err(|_| repak::Error::Base64)
|
|
||||||
.and_then(|bytes| {
|
|
||||||
aes::Aes256::new_from_slice(bytes).map_err(|_| repak::Error::Aes)
|
|
||||||
}).unwrap();
|
|
||||||
|
|
||||||
|
macro_rules! matrix_test_body {
|
||||||
let mut inner_reader = std::io::Cursor::new(include_bytes!(concat!("packs/pack_", $version, $compress, $encrypt, $encryptindex, ".pak")));
|
( $name:literal, $version:literal, $exp_version:expr, $compress:literal, $encrypt:literal, $encryptindex:literal, $body:expr ) => {
|
||||||
let len = inner_reader.seek(SeekFrom::End(0)).unwrap();
|
paste! {
|
||||||
let mut reader = ReadCounter::new_size(inner_reader, len as usize);
|
#[test]
|
||||||
|
fn [< test_ $name _version_ $version $compress $encrypt $encryptindex >]() {
|
||||||
let pak = repak::PakReader::new_any(&mut reader, Some(key)).unwrap();
|
$body($exp_version, include_bytes!(concat!("packs/pack_", $version, $compress, $encrypt, $encryptindex, ".pak")));
|
||||||
|
|
||||||
assert_eq!(pak.mount_point(), "../mount/point/root/");
|
|
||||||
assert_eq!(pak.version(), $exp_version);
|
|
||||||
use std::collections::HashSet;
|
|
||||||
let files: HashSet<String> = HashSet::from_iter(pak.files());
|
|
||||||
assert_eq!(files, HashSet::from_iter(vec!["test.txt", "test.png", "zeros.bin", "directory/nested.txt"].into_iter().map(String::from)));
|
|
||||||
|
|
||||||
for file in files {
|
|
||||||
let mut buf = vec![];
|
|
||||||
let mut writer = std::io::Cursor::new(&mut buf);
|
|
||||||
pak.read_file(&file, &mut reader, &mut writer).unwrap();
|
|
||||||
match file.as_str() {
|
|
||||||
"test.txt" => assert_eq!(buf, include_bytes!("pack/root/test.txt"), "test.txt incorrect contents"),
|
|
||||||
"test.png" => assert_eq!(buf, include_bytes!("pack/root/test.png"), "test.png incorrect contents"),
|
|
||||||
"zeros.bin" => assert_eq!(buf, include_bytes!("pack/root/zeros.bin"), "zeros.bin incorrect contents"),
|
|
||||||
"directory/nested.txt" => assert_eq!(buf, include_bytes!("pack/root/directory/nested.txt"), "nested.txt incorrect contents"),
|
|
||||||
name => panic!("unrecognized file {}", name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for r in reader.into_reads() {
|
|
||||||
// sanity check. a pak file can be constructed with a lot of dead space
|
|
||||||
// which wouldn't have to be read, but so far all bytes in paks generated
|
|
||||||
// by UnrealPak are meaningful
|
|
||||||
assert!(r > 0, "every byte has been read at least once");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
)*
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
matrix_test_read!(
|
matrix_test!(
|
||||||
|
"read",
|
||||||
(
|
(
|
||||||
"v5" repak::Version::V5,
|
"v5" repak::Version::V5,
|
||||||
"v7" repak::Version::V7,
|
"v7" repak::Version::V7,
|
||||||
|
|
@ -165,70 +230,12 @@ matrix_test_read!(
|
||||||
),
|
),
|
||||||
("", "_compress"),
|
("", "_compress"),
|
||||||
("", "_encrypt"),
|
("", "_encrypt"),
|
||||||
("", "_encryptindex")
|
("", "_encryptindex"),
|
||||||
|
test_read
|
||||||
);
|
);
|
||||||
|
|
||||||
macro_rules! matrix_test_write {
|
matrix_test!(
|
||||||
( ($($version:literal $exp_version:expr),* $(,)?), $compress:tt, $encrypt:tt, $encryptindex:tt ) => {
|
"write",
|
||||||
$( mt_compress_write!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! mt_compress_write {
|
|
||||||
( $version:literal, $exp_version:expr, ($($compress:literal),* $(,)?), $encrypt:tt, $encryptindex:tt ) => {
|
|
||||||
$( mt_encrypt_write!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! mt_encrypt_write {
|
|
||||||
( $version:literal, $exp_version:expr, $compress:literal, ($($encrypt:literal),* $(,)?), $encryptindex:tt ) => {
|
|
||||||
$( mt_encryptindex_write!($version, $exp_version, $compress, $encrypt, $encryptindex); )*
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! mt_encryptindex_write {
|
|
||||||
( $version:literal, $exp_version:expr, $compress:literal, $encrypt:literal, ($($encryptindex:literal),* $(,)?) ) => {
|
|
||||||
$(
|
|
||||||
paste! {
|
|
||||||
#[test]
|
|
||||||
fn [< test_write_version_ $version $compress $encrypt $encryptindex >]() {
|
|
||||||
use aes::cipher::KeyInit;
|
|
||||||
use base64::{engine::general_purpose, Engine as _};
|
|
||||||
let key = general_purpose::STANDARD
|
|
||||||
.decode(AES_KEY)
|
|
||||||
.as_ref()
|
|
||||||
.map_err(|_| repak::Error::Base64)
|
|
||||||
.and_then(|bytes| {
|
|
||||||
aes::Aes256::new_from_slice(bytes).map_err(|_| repak::Error::Aes)
|
|
||||||
}).unwrap();
|
|
||||||
|
|
||||||
let mut reader = std::io::Cursor::new(include_bytes!(concat!("packs/pack_", $version, $compress, $encrypt, $encryptindex, ".pak")));
|
|
||||||
let pak_reader = repak::PakReader::new_any(&mut reader, Some(key)).unwrap();
|
|
||||||
|
|
||||||
let writer = Cursor::new(vec![]);
|
|
||||||
let mut pak_writer = repak::PakWriter::new(
|
|
||||||
writer,
|
|
||||||
None,
|
|
||||||
pak_reader.version(),
|
|
||||||
pak_reader.mount_point().to_owned(),
|
|
||||||
Some(0x205C5A7D),
|
|
||||||
);
|
|
||||||
|
|
||||||
for path in pak_reader.files() {
|
|
||||||
let data = pak_reader.get(&path, &mut reader).unwrap();
|
|
||||||
pak_writer
|
|
||||||
.write_file(&path, &mut std::io::Cursor::new(data))
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
assert_eq!(pak_writer.write_index().unwrap().into_inner(), reader.into_inner());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)*
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
matrix_test_write!(
|
|
||||||
(
|
(
|
||||||
"v5" repak::Version::V5,
|
"v5" repak::Version::V5,
|
||||||
"v7" repak::Version::V7,
|
"v7" repak::Version::V7,
|
||||||
|
|
@ -239,5 +246,6 @@ matrix_test_write!(
|
||||||
),
|
),
|
||||||
("", /*"_compress"*/),
|
("", /*"_compress"*/),
|
||||||
("", /*"_encrypt"*/),
|
("", /*"_encrypt"*/),
|
||||||
("", /*"_encryptindex"*/)
|
("", /*"_encryptindex"*/),
|
||||||
|
test_write
|
||||||
);
|
);
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue