sangge
e400b87e9f
- 重写p1实现纯Rust的hex到base64转换 - 完成p13 ECB剪切粘贴攻击和破解脚本 - 实现p33 Diffie-Hellman密钥交换算法 - 修复p9的PKCS#7测试用例 - 在common库添加gen_random_key和pkcs7_unpadding函数 - 更新workspace依赖管理
45 lines
1.4 KiB
Rust
45 lines
1.4 KiB
Rust
use common::{aes_cbc_enc, aes_ecb_enc, gen_random_key, is_ecb, pkcs7_padding};
|
|
use rand::prelude::*;
|
|
|
|
fn encryption_oracle(plaintext: &[u8]) -> Vec<u8> {
|
|
let key = gen_random_key();
|
|
let mut rng = rand::rng();
|
|
let enc_mode = rng.random_range(0..2); // 0 for ECB, 1 for CBC
|
|
let plaintext = plaintext.to_vec();
|
|
let random_prefix_len = rng.random_range(5..11);
|
|
let random_suffix_len = rng.random_range(5..11);
|
|
let mut random_padding = [0u8; 20];
|
|
rng.fill(&mut random_padding);
|
|
let random_prefix = random_padding[..random_prefix_len].to_vec();
|
|
let random_suffix = random_padding[random_padding.len() - random_suffix_len..].to_vec();
|
|
|
|
let mut data = Vec::new();
|
|
data.extend(random_prefix);
|
|
data.extend(plaintext);
|
|
data.extend(random_suffix);
|
|
|
|
pkcs7_padding(&mut data, 16);
|
|
|
|
if enc_mode == 0 {
|
|
// ECB mode in p7
|
|
println!("Using ECB mode.");
|
|
aes_ecb_enc(&data, &key).unwrap()
|
|
} else {
|
|
// CBC mode in p10
|
|
let iv = [0u8; 16];
|
|
println!("Using CBC mode.");
|
|
aes_cbc_enc(&data, &key, &iv).unwrap()
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let plaintext = b"YELLOW SUBMARINE".repeat(32);
|
|
let encrypted_data = encryption_oracle(&plaintext);
|
|
// in p8
|
|
if is_ecb(&encrypted_data) {
|
|
println!("ECB mode detected.");
|
|
} else {
|
|
println!("CBC mode detected.");
|
|
}
|
|
}
|