mirror of
https://github.com/lifegpc/msg-tool.git
synced 2026-06-06 12:58:45 +08:00
Add AkabeiCrypt (tested game: https://vndb.org/v23242 )
This commit is contained in:
@@ -171,6 +171,10 @@ enum CryptType {
|
||||
TokidokiCrypt,
|
||||
SourireCrypt,
|
||||
HibikiCrypt,
|
||||
#[serde(rename_all = "PascalCase")]
|
||||
AkabeiCrypt {
|
||||
seed: u32,
|
||||
},
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Deserialize)]
|
||||
@@ -269,6 +273,7 @@ impl Schema {
|
||||
CryptType::TokidokiCrypt => Box::new(TokidokiCrypt::new(self.base.clone())),
|
||||
CryptType::SourireCrypt => Box::new(SourireCrypt::new(self.base.clone())),
|
||||
CryptType::HibikiCrypt => Box::new(HibikiCrypt::new(self.base.clone())),
|
||||
CryptType::AkabeiCrypt { seed } => Box::new(AkabeiCrypt::new(self.base.clone(), *seed)),
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -1053,7 +1058,7 @@ impl<R: Read> Read for SourireCryptReader<R> {
|
||||
|
||||
seek_crypt_filehash_key_impl!(HibikiCrypt, HibikiCryptReader<T>);
|
||||
|
||||
impl <R: Read> Read for HibikiCryptReader<R> {
|
||||
impl<R: Read> Read for HibikiCryptReader<R> {
|
||||
fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
|
||||
let readed = self.inner.read(buf)?;
|
||||
let key1 = (self.key >> 5) as u8;
|
||||
@@ -1072,6 +1077,77 @@ impl <R: Read> Read for HibikiCryptReader<R> {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct AkabeiCrypt {
|
||||
base: BaseSchema,
|
||||
seed: u32,
|
||||
}
|
||||
|
||||
impl AkabeiCrypt {
|
||||
pub fn new(base: BaseSchema, seed: u32) -> Self {
|
||||
Self { base, seed }
|
||||
}
|
||||
|
||||
fn get_key(&self, mut hash: u32) -> [u8; 0x20] {
|
||||
let mut state = [0; 0x20];
|
||||
hash = (hash ^ self.seed) & 0x7FFFFFFF;
|
||||
hash = hash << 31 | hash;
|
||||
for i in 0..0x20 {
|
||||
state[i] = (hash & 0xFF) as u8;
|
||||
hash = (hash & 0xFFFFFFFE) << 23 | hash >> 8;
|
||||
}
|
||||
state
|
||||
}
|
||||
}
|
||||
|
||||
impl Crypt for AkabeiCrypt {
|
||||
base_schema_impl!();
|
||||
fn decrypt_supported(&self) -> bool {
|
||||
true
|
||||
}
|
||||
fn decrypt_seek_supported(&self) -> bool {
|
||||
true
|
||||
}
|
||||
fn decrypt<'a>(
|
||||
&self,
|
||||
entry: &Xp3Entry,
|
||||
cur_seg: &Segment,
|
||||
stream: Box<dyn Read + 'a>,
|
||||
) -> Result<Box<dyn ReadDebug + 'a>> {
|
||||
Ok(Box::new(AkabeiCryptReader::new(
|
||||
stream,
|
||||
cur_seg,
|
||||
self.get_key(entry.file_hash),
|
||||
)))
|
||||
}
|
||||
fn decrypt_with_seek<'a>(
|
||||
&self,
|
||||
entry: &Xp3Entry,
|
||||
cur_seg: &Segment,
|
||||
stream: Box<dyn ReadSeek + 'a>,
|
||||
) -> Result<Box<dyn ReadSeek + 'a>> {
|
||||
Ok(Box::new(AkabeiCryptReader::new(
|
||||
stream,
|
||||
cur_seg,
|
||||
self.get_key(entry.file_hash),
|
||||
)))
|
||||
}
|
||||
}
|
||||
|
||||
seek_reader_key_impl!(AkabeiCryptReader<T>, [u8; 0x20]);
|
||||
|
||||
impl<R: Read> Read for AkabeiCryptReader<R> {
|
||||
fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
|
||||
let readed = self.inner.read(buf)?;
|
||||
for (i, t) in (&mut buf[..readed]).iter_mut().enumerate() {
|
||||
let offset = self.seg_start + self.pos + i as u64;
|
||||
*t ^= self.key[(offset & 0x1F) as usize];
|
||||
}
|
||||
self.pos += readed as u64;
|
||||
Ok(readed)
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_deserialize_crypt() {
|
||||
for (key, schema) in CRYPT_SCHEMA.iter() {
|
||||
|
||||
@@ -134,7 +134,7 @@ impl ScriptBuilder for Xp3ArchiveBuilder {
|
||||
}
|
||||
|
||||
fn extensions(&self) -> &'static [&'static str] {
|
||||
&["xp3"]
|
||||
&["xp3", "bin"]
|
||||
}
|
||||
|
||||
fn script_type(&self) -> &'static ScriptType {
|
||||
@@ -154,6 +154,13 @@ impl ScriptBuilder for Xp3ArchiveBuilder {
|
||||
) -> Result<Box<dyn Archive>> {
|
||||
Ok(Box::new(Xp3ArchiveWriter::new(filename, files, config)?))
|
||||
}
|
||||
|
||||
fn is_this_format(&self, _filename: &str, buf: &[u8], buf_len: usize) -> Option<u8> {
|
||||
if buf_len >= 11 && buf.starts_with(consts::XP3_MAGIC) {
|
||||
return Some(100);
|
||||
}
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
|
||||
Reference in New Issue
Block a user