mirrors/lib_um_crypto_rust
1
0
Fork 0
mirror of https://git.um-react.app/um/lib_um_crypto_rust.git synced 2026-03-08 04:29:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: better mp3 detection; drop mp1/mp2 format test

Browse Source
This commit is contained in:
鲁树人
2025-03-31 08:53:24 +09:00
parent f503b5ea4b
commit 8f96da8b20
2 changed files with 22 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
um_audio/src/lib.rs
View File

@@ -103,4 +103,12 @@ mod tests {
let result = detect_audio_type(mp3_data).expect("failed to parse mp3");
assert_eq!(result, AudioType::MP3);
}
#[test]
fn test_mp3_invalid_data() {
let mut mp3_data = [0; 4096];
mp3_data[0..4].copy_from_slice(&[0xff; 4]);
let result = detect_audio_type(&mp3_data).expect("failed to parse mp3");
assert_eq!(result, AudioType::Unknown);
}
}

18
um_audio/src/sync_frame.rs
View File

@@ -1,11 +1,21 @@
pub const SYNC_FRAME_TEST_SIZE: usize = 0xff;
pub fn is_mp3(magic: u32) -> bool {
// Frame sync should have the first 11 bits set to 1.
const MP3_AND_MASK: u32 = 0b1111_1111_1110_0000u32 << 16;
const MP3_EXPECTED: u32 = 0b1111_1111_1110_0000u32 << 16;
// Check for 11-bit sync word, followed by 2 bits of version, and 2 bits of layer.
// MPEG Version: MPEG Version 2 (ISO/IEC 13818-3) or MPEG Version 1 (ISO/IEC 11172-3)
const MP3_AND_MASK: u32 = 0b1111_1111_1111_0110u32 << 16;
const MP3_EXPECTED: u32 = 0b1111_1111_1111_0010u32 << 16;
(magic & MP3_AND_MASK) == MP3_EXPECTED
if (magic & MP3_AND_MASK) != MP3_EXPECTED {
return false;
}
// Check for bitrate index and sampling rate frequency index.
let bitrate = ((magic >> 12) & 0b1111) as u8;
let sampling_rate = ((magic >> 10) & 0b11) as u8;
// They should not be all 1s.
bitrate != 0b1111 && sampling_rate != 0b11
}
pub fn is_aac(magic: u32) -> bool {