feat: 完善AES加密实现和多个问题的解决方案

- 在common库中添加了完整的AES-128加密解密实现 - 实现了AES-ECB和AES-CBC模式的加密解密函数 - 添加了密钥扩展和所有必要的AES操作函数 - 完成了问题10的AES-CBC模式实现 - 修复了问题11的加密oracle实现，使用正确的ECB检测 - 改进了代码风格，使用现代Rust格式化语法 - 为多个问题添加了common库的依赖 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
2025-07-14 23:49:21 +08:00
parent 66ce722dd2
commit 6f54d41c8e
11 changed files with 653 additions and 14 deletions
--- a/problems/p10/10.txt
+++ b/problems/p10/10.txt
@@ -0,0 +1,64 @@
+CRIwqt4+szDbqkNY+I0qbNXPg1XLaCM5etQ5Bt9DRFV/xIN2k8Go7jtArLIy
+P605b071DL8C+FPYSHOXPkMMMFPAKm+Nsu0nCBMQVt9mlluHbVE/yl6VaBCj
+NuOGvHZ9WYvt51uR/lklZZ0ObqD5UaC1rupZwCEK4pIWf6JQ4pTyPjyiPtKX
+g54FNQvbVIHeotUG2kHEvHGS/w2Tt4E42xEwVfi29J3yp0O/TcL7aoRZIcJj
+MV4qxY/uvZLGsjo1/IyhtQp3vY0nSzJjGgaLYXpvRn8TaAcEtH3cqZenBoox
+BH3MxNjD/TVf3NastEWGnqeGp+0D9bQx/3L0+xTf+k2VjBDrV9HPXNELRgPN
+0MlNo79p2gEwWjfTbx2KbF6htgsbGgCMZ6/iCshy3R8/abxkl8eK/VfCGfA6
+bQQkqs91bgsT0RgxXSWzjjvh4eXTSl8xYoMDCGa2opN/b6Q2MdfvW7rEvp5m
+wJOfQFDtkv4M5cFEO3sjmU9MReRnCpvalG3ark0XC589rm+42jC4/oFWUdwv
+kzGkSeoabAJdEJCifhvtGosYgvQDARUoNTQAO1+CbnwdKnA/WbQ59S9MU61Q
+KcYSuk+jK5nAMDot2dPmvxZIeqbB6ax1IH0cdVx7qB/Z2FlJ/U927xGmC/RU
+FwoXQDRqL05L22wEiF85HKx2XRVB0F7keglwX/kl4gga5rk3YrZ7VbInPpxU
+zgEaE4+BDoEqbv/rYMuaeOuBIkVchmzXwlpPORwbN0/RUL89xwOJKCQQZM8B
+1YsYOqeL3HGxKfpFo7kmArXSRKRHToXuBgDq07KS/jxaS1a1Paz/tvYHjLxw
+Y0Ot3kS+cnBeq/FGSNL/fFV3J2a8eVvydsKat3XZS3WKcNNjY2ZEY1rHgcGL
+5bhVHs67bxb/IGQleyY+EwLuv5eUwS3wljJkGcWeFhlqxNXQ6NDTzRNlBS0W
+4CkNiDBMegCcOlPKC2ZLGw2ejgr2utoNfmRtehr+3LAhLMVjLyPSRQ/zDhHj
+Xu+Kmt4elmTmqLgAUskiOiLYpr0zI7Pb4xsEkcxRFX9rKy5WV7NhJ1lR7BKy
+alO94jWIL4kJmh4GoUEhO+vDCNtW49PEgQkundV8vmzxKarUHZ0xr4feL1ZJ
+THinyUs/KUAJAZSAQ1Zx/S4dNj1HuchZzDDm/nE/Y3DeDhhNUwpggmesLDxF
+tqJJ/BRn8cgwM6/SMFDWUnhkX/t8qJrHphcxBjAmIdIWxDi2d78LA6xhEPUw
+NdPPhUrJcu5hvhDVXcceZLa+rJEmn4aftHm6/Q06WH7dq4RaaJePP6WHvQDp
+zZJOIMSEisApfh3QvHqdbiybZdyErz+yXjPXlKWG90kOz6fx+GbvGcHqibb/
+HUfcDosYA7lY4xY17llY5sibvWM91ohFN5jyDlHtngi7nWQgFcDNfSh77TDT
+zltUp9NnSJSgNOOwoSSNWadm6+AgbXfQNX6oJFaU4LQiAsRNa7vX/9jRfi65
+5uvujM4ob199CZVxEls10UI9pIemAQQ8z/3rgQ3eyL+fViyztUPg/2IvxOHv
+eexE4owH4Fo/bRlhZK0mYIamVxsRADBuBlGqx1b0OuF4AoZZgUM4d8v3iyUu
+feh0QQqOkvJK/svkYHn3mf4JlUb2MTgtRQNYdZKDRgF3Q0IJaZuMyPWFsSNT
+YauWjMVqnj0AEDHh6QUMF8bXLM0jGwANP+r4yPdKJNsoZMpuVoUBJYWnDTV+
+8Ive6ZgBi4EEbPbMLXuqDMpDi4XcLE0UUPJ8VnmO5fAHMQkA64esY2QqldZ+
+5gEhjigueZjEf0917/X53ZYWJIRiICnmYPoM0GSYJRE0k3ycdlzZzljIGk+P
+Q7WgeJhthisEBDbgTuppqKNXLbNZZG/VaTdbpW1ylBv0eqamFOmyrTyh1APS
+Gn37comTI3fmN6/wmVnmV4/FblvVwLuDvGgSCGPOF8i6FVfKvdESs+yr+1AE
+DJXfp6h0eNEUsM3gXaJCknGhnt3awtg1fSUiwpYfDKZxwpPOYUuer8Wi+VCD
+sWsUpkMxhhRqOBKaQaBDQG+kVJu6aPFlnSPQQTi1hxLwi0l0Rr38xkr+lHU7
+ix8LeJVgNsQdtxbovE3i7z3ZcTFY7uJkI9j9E0muDN9x8y/YN25rm6zULYaO
+jUoP/7FQZsSgxPIUvUiXkEq+FU2h0FqAC7H18cr3Za5x5dpw5nwawMArKoqG
+9qlhqc34lXV0ZYwULu58EImFIS8+kITFuu7jOeSXbBgbhx8zGPqavRXeiu0t
+bJd0gWs+YgMLzXtQIbQuVZENMxJSZB4aw5lPA4vr1fFBsiU4unjOEo/XAgwr
+Tc0w0UndJFPvXRr3Ir5rFoIEOdRo+6os5DSlk82SBnUjwbje7BWsxWMkVhYO
+6bOGUm4VxcKWXu2jU66TxQVIHy7WHktMjioVlWJdZC5Hq0g1LHg1nWSmjPY2
+c/odZqN+dBBC51dCt4oi5UKmKtU5gjZsRSTcTlfhGUd6DY4Tp3CZhHjQRH4l
+Zhg0bF/ooPTxIjLKK4r0+yR0lyRjqIYEY27HJMhZDXFDxBQQ1UkUIhAvXacD
+WB2pb3YyeSQjt8j/WSbQY6TzdLq8SreZiuMWcXmQk4EH3xu8bPsHlcvRI+B3
+gxKeLnwrVJqVLkf3m2cSGnWQhSLGbnAtgQPA6z7u3gGbBmRtP0KnAHWSK7q6
+onMoYTH+b5iFjCiVRqzUBVzRRKjAL4rcL2nYeV6Ec3PlnboRzJwZIjD6i7WC
+dcxERr4WVOjOBX4fhhKUiVvlmlcu8CkIiSnZENHZCpI41ypoVqVarHpqh2aP
+/PS624yfxx2N3C2ci7VIuH3DcSYcaTXEKhz/PRLJXkRgVlWxn7QuaJJzDvpB
+oFndoRu1+XCsup/AtkLidsSXMFTo/2Ka739+BgYDuRt1mE9EyuYyCMoxO/27
+sn1QWMMd1jtcv8Ze42MaM4y/PhAMp2RfCoVZALUS2K7XrOLl3s9LDFOdSrfD
+8GeMciBbfLGoXDvv5Oqq0S/OvjdID94UMcadpnSNsist/kcJJV0wtRGfALG2
+UKYzEj/2TOiN75UlRvA5XgwfqajOvmIIXybbdhxpjnSB04X3iY82TNSYTmL
+LAzZlX2vmV9IKRRimZ2SpzNpvLKeB8lDhIyGzGXdiynQjFMNcVjZlmWHsH7e
+ItAKWmCwNkeuAfFwir4TTGrgG1pMje7XA7kMT821cYbLSiPAwtlC0wm77F0T
+a7jdMrLjMO29+1958CEzWPdzdfqKzlfBzsba0+dS6mcW/YTHaB4bDyXechZB
+k/35fUg+4geMj6PBTqLNNWXBX93dFC7fNyda+Lt9cVJnlhIi/61fr0KzxOeX
+NKgePKOC3Rz+fWw7Bm58FlYTgRgN63yFWSKl4sMfzihaQq0R8NMQIOjzuMl3
+Ie5ozSa+y9g4z52RRc69l4n4qzf0aErV/BEe7FrzRyWh4PkDj5wy5ECaRbfO
+7rbs1EHlshFvXfGlLdEfP2kKpT9U32NKZ4h+Gr9ymqZ6isb1KfNov1rw0KSq
+YNP+EyWCyLRJ3EcOYdvVwVb+vIiyzxnRdugB3vNzaNljHG5ypEJQaTLphIQn
+lP02xcBpMNJN69bijVtnASN/TLV5ocYvtnWPTBKu3OyOkcflMaHCEUgHPW0f
+mGfld4i9Tu35zrKvTDzfxkJX7+KJ72d/V+ksNKWvwn/wvMOZsa2EEOfdCidm
+oql027IS5XvSHynQtvFmw0HTk9UXt8HdVNTqcdy/jUFmXpXNP2Wvn8PrU2Dh
+kkIzWhQ5Rxd/vnM2QQr9Cxa2J9GXEV3kGDiZV90+PCDSVGY4VgF8y7GedI1h
--- a/problems/p10/Cargo.toml
+++ b/problems/p10/Cargo.toml
@@ -4,3 +4,7 @@ version = "0.1.0"
 edition = "2024"

 [dependencies]
+anyhow = "1.0"
+hex = "0.4"
+base64 = "0.22"
+common = { path = "../../common" }
--- a/problems/p10/src/main.rs
+++ b/problems/p10/src/main.rs
@@ -1,3 +1,141 @@
-fn main() {
-    println!("Hello, world!");
+#![allow(dead_code)]
+use anyhow::{Result, anyhow};
+use base64::{Engine, engine::general_purpose::STANDARD};
+use common::{
+    add_round_key, expand_key, inv_mix_columns, inv_shift_rows, inv_sub_bytes, mix_columns,
+    shift_rows, sub_bytes,
+};
+use std::fs::read_to_string;
+
+fn aes_cbc_enc(input: &[u8], key: &[u8; 16], iv: &[u8; 16]) -> Result<Vec<u8>> {
+    if input.len() % 16 != 0 {
+        return Err(anyhow!("Invalid input length"));
+    }
+    let mut cipher: Vec<u8> = Vec::new();
+    let round_keys = expand_key(key);
+    let mut prev_block = *iv;
+    for i in 0..(input.len() / 16) {
+        let mut block: [u8; 16] = input[(i * 16)..(i * 16 + 16)].try_into()?;
+        block = block
+            .iter()
+            .zip(prev_block.iter())
+            .map(|(b, iv)| b ^ iv)
+            .collect::<Vec<u8>>()
+            .try_into()
+            .unwrap();
+        block = add_round_key(&block, &round_keys[0]);
+        for round_key in round_keys.iter().take(10).skip(1) {
+            block = sub_bytes(&block);
+            block = shift_rows(&block);
+            block = mix_columns(&block);
+            block = add_round_key(&block, round_key);
+        }
+        block = sub_bytes(&block);
+        block = shift_rows(&block);
+        block = add_round_key(&block, &round_keys[10]);
+        cipher.extend(block);
+        prev_block = block;
+    }
+
+    Ok(cipher)
+}
+
+fn aes_cbc_dec(input: &[u8], key: &[u8; 16], iv: &[u8; 16]) -> Result<Vec<u8>> {
+    if input.len() % 16 != 0 {
+        return Err(anyhow!("Invalid input length"));
+    }
+
+    let mut plaintext: Vec<u8> = Vec::new();
+    let round_keys = expand_key(key);
+    let mut prev_block = *iv;
+    for i in 0..(input.len() / 16) {
+        let mut block: [u8; 16] = input[(i * 16)..(i * 16 + 16)].try_into()?;
+        block = add_round_key(&block, &round_keys[10]);
+        block = inv_shift_rows(&block);
+        block = inv_sub_bytes(&block);
+        for j in 0..9 {
+            block = add_round_key(&block, &round_keys[9 - j]);
+            block = inv_mix_columns(&block);
+            block = inv_shift_rows(&block);
+            block = inv_sub_bytes(&block);
+        }
+        block = add_round_key(&block, &round_keys[0]);
+        block = block
+            .iter()
+            .zip(prev_block.iter())
+            .map(|(b, iv)| b ^ iv)
+            .collect::<Vec<u8>>()
+            .try_into()
+            .unwrap();
+        plaintext.extend(block);
+        prev_block = input[(i * 16)..(i * 16 + 16)].try_into()?;
+    }
+
+    Ok(plaintext)
+}
+
+fn main() {
+    let key = "YELLOW SUBMARINE".as_bytes();
+    let iv = [0u8; 16];
+    let file_path = "./problems/p10/10.txt";
+    let b64_cipher = read_to_string(file_path).expect("Failed to read file");
+    let cipher = STANDARD
+        .decode(b64_cipher.trim().replace('\n', ""))
+        .expect("Failed to decode base64");
+    let key_array: [u8; 16] = key.try_into().expect("Key must be 16 bytes long");
+    let decrypted = aes_cbc_dec(&cipher, &key_array, &iv).expect("Decryption failed");
+    let decrypted_str = String::from_utf8(decrypted).expect("Failed to convert to string");
+    println!("Decrypted text: {decrypted_str}");
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_aes_cbc_enc_dec() {
+        let key: [u8; 16] = [0x00; 16];
+        let iv: [u8; 16] = [0x00; 16];
+        let plaintext: Vec<u8> = vec![0x01; 32]; // 2 blocks of 16 bytes
+
+        let ciphertext = aes_cbc_enc(&plaintext, &key, &iv).unwrap();
+        let decrypted = aes_cbc_dec(&ciphertext, &key, &iv).unwrap();
+        assert_eq!(decrypted, plaintext);
+    }
+
+    #[test]
+    fn test_aes_cbc_enc_with_openssl_cmd() {
+        use std::io::Write;
+        use std::process::{Command, Stdio};
+
+        let key: [u8; 16] = [0x00; 16];
+        let iv: [u8; 16] = [0x00; 16];
+        let plaintext: Vec<u8> = vec![0x01; 32]; // 2 blocks of 16 bytes
+
+        // Encrypt using our implementation
+        let ciphertext = aes_cbc_enc(&plaintext, &key, &iv).unwrap();
+
+        // Call openssl with stdin
+        let mut child = Command::new("openssl")
+            .args([
+                "enc",
+                "-aes-128-cbc",
+                "-K",
+                &hex::encode(key),
+                "-iv",
+                &hex::encode(iv),
+                "-nopad",
+            ])
+            .stdin(Stdio::piped())
+            .stdout(Stdio::piped())
+            .spawn()
+            .expect("failed to execute openssl");
+
+        let mut stdin = child.stdin.take().expect("failed to get stdin");
+        stdin.write_all(&plaintext).unwrap();
+        drop(stdin);
+
+        let output = child.wait_with_output().unwrap();
+        assert_eq!(ciphertext, output.stdout);
+    }
 }