67 lines
1.5 KiB
Rust
67 lines
1.5 KiB
Rust
use crate::v2_map::QMC2Map;
|
|
use crate::v2_rc4::cipher::QMC2RC4;
|
|
use anyhow::Result;
|
|
use thiserror::Error;
|
|
|
|
pub mod ekey;
|
|
pub mod footer;
|
|
pub mod v1;
|
|
pub mod v2_map;
|
|
pub mod v2_rc4;
|
|
|
|
#[derive(Error, Debug)]
|
|
pub enum QmcCryptoError {
|
|
#[error("QMC V2/Map Cipher: Key is empty")]
|
|
QMCV2MapKeyEmpty,
|
|
}
|
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
|
pub enum QMCv2Cipher {
|
|
MapL(QMC2Map),
|
|
RC4(QMC2RC4),
|
|
}
|
|
|
|
impl QMCv2Cipher {
|
|
pub fn new<T>(key: T) -> Result<Self>
|
|
where
|
|
T: AsRef<[u8]>,
|
|
{
|
|
let key = key.as_ref();
|
|
let cipher = match key.len() {
|
|
0 => Err(QmcCryptoError::QMCV2MapKeyEmpty)?,
|
|
1..=300 => QMCv2Cipher::MapL(QMC2Map::new(key)?),
|
|
_ => QMCv2Cipher::RC4(QMC2RC4::new(key)),
|
|
};
|
|
Ok(cipher)
|
|
}
|
|
|
|
pub fn new_from_ekey<T: AsRef<[u8]>>(ekey_str: T) -> Result<Self> {
|
|
let key = ekey::decrypt(ekey_str)?;
|
|
Self::new(key)
|
|
}
|
|
|
|
pub fn decrypt<T>(&self, data: &mut T, offset: usize)
|
|
where
|
|
T: AsMut<[u8]> + ?Sized,
|
|
{
|
|
match self {
|
|
QMCv2Cipher::MapL(cipher) => cipher.decrypt(data, offset),
|
|
QMCv2Cipher::RC4(cipher) => cipher.decrypt(data, offset),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
pub fn generate_key(len: usize) -> Vec<u8> {
|
|
(1..=len).map(|i| i as u8).collect()
|
|
}
|
|
|
|
#[cfg(test)]
|
|
pub fn generate_key_128() -> [u8; 128] {
|
|
generate_key(128)
|
|
.try_into()
|
|
.expect("failed to make test key")
|
|
}
|
|
}
|