[joox] feat #1: add joox decipher implementation

This commit is contained in:
鲁树人
2024-09-17 21:45:51 +01:00
parent b3fc9f8318
commit 12199616c2
13 changed files with 293 additions and 1 deletions

49
um_cli/src/cmd/joox.rs Normal file
View File

@@ -0,0 +1,49 @@
use crate::Cli;
use anyhow::Result;
use clap::Args;
use std::fs::File;
use std::io::{BufReader, BufWriter, Read, Seek, SeekFrom, Write};
use std::path::PathBuf;
use umc_joox::decrypt::JooxDecipher;
use umc_joox::header::Header;
/// Decrypt a ofl_en file (Joox Music, Encryption V4)
#[derive(Args)]
pub struct ArgsJoox {
/// Path to output file, e.g. /export/Music/song.flac
#[clap(short, long)]
output: PathBuf,
/// Path to input file, e.g. /export/Music/song.ofl_en
#[arg(name = "input")]
input: PathBuf,
/// Device GUID
#[clap(short = 'u', long)]
device_guid: String,
}
impl ArgsJoox {
pub fn run(&self, _cli: &Cli) -> Result<i32> {
let mut file_input = File::open(&self.input)?;
let mut header_buffer = [0u8; 0x0c];
file_input.read_exact(&mut header_buffer)?;
let header = Header::from_buffer(&header_buffer, &self.device_guid)?;
file_input.seek(SeekFrom::Start(header.audio_start_offset as u64))?;
let reader = BufReader::new(file_input);
let mut input_stream = reader.take(header.original_file_len);
let file_output = File::create(&self.output)?;
let mut writer = BufWriter::new(file_output);
let mut buffer = vec![0u8; header.get_audio_block_size()].into_boxed_slice();
while let Ok(()) = input_stream.read_exact(&mut buffer) {
let result = header.decrypt_audio_block(&mut buffer)?;
writer.write_all(result)?;
}
Ok(0)
}
}

View File

@@ -1,5 +1,6 @@
use clap::Subcommand;
pub mod joox;
pub mod kgm;
pub mod ncm;
pub mod qmc1;
@@ -15,4 +16,6 @@ pub enum Commands {
NCM(ncm::ArgsNCM),
#[command(name = "kgm")]
KGM(kgm::ArgsKGM),
#[command(name = "joox")]
JOOX(joox::ArgsJoox),
}

View File

@@ -31,6 +31,7 @@ fn run_command(cli: &Cli) -> Result<i32> {
Some(Commands::QMCv2(cmd)) => cmd.run(&cli),
Some(Commands::NCM(cmd)) => cmd.run(&cli),
Some(Commands::KGM(cmd)) => cmd.run(&cli),
Some(Commands::JOOX(cmd)) => cmd.run(&cli),
None => {
// https://github.com/clap-rs/clap/issues/3857#issuecomment-1161796261
todo!("implement a sensible default command, similar to um/cli");