From ca27e6d24f3d77fadbf061df59154359fc34d8b7 Mon Sep 17 00:00:00 2001
From: UniDarkstars <1830018670@qq.com>
Date: Tue, 26 Nov 2024 12:40:04 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=8F=E6=94=B9=EF=BC=8C=E6=97=A0=E4=BC=A4?=
 =?UTF-8?q?=E5=A4=A7=E9=9B=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 recv/recv.py          | 2 +-
 recv/test_recv.py     | 2 +-
 tool/symmetric/AES.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/recv/recv.py b/recv/recv.py
index 98679c3..ac9d829 100644
--- a/recv/recv.py
+++ b/recv/recv.py
@@ -72,6 +72,6 @@ def handleLetter(letter: Letter):
             break
     with open(path, "wb") as f:
         f.write(data)
-        print(f"确认收到来自 {Segwit.encodeSegwit(letter.senderPubKey.encode("utf-8"))} 的文件")
+        print(f"确认收到来自 {Segwit.encodeSegwit(letter.senderPubKey.encode('utf-8'))} 的文件")
         print(f"签名验证有效，已将文件 {newName} 保存至 {download_dir} 下")
     return
diff --git a/recv/test_recv.py b/recv/test_recv.py
index 1106e10..8a80588 100644
--- a/recv/test_recv.py
+++ b/recv/test_recv.py
@@ -25,7 +25,7 @@ def test_handleLetter_sm4_ecb(mock_input):
     pass
 
 
-@patch('builtins.input', side_effect=['./public.pem', 'sm4', 'cbc'])
+@patch('builtins.input', side_effect=['./public.pem', 'aes', 'cbc'])
 def test_handleLetter_sm4_cbc(mock_input):
     letter = sender.main()
     recv.handleLetter(letter)
diff --git a/tool/symmetric/AES.py b/tool/symmetric/AES.py
index 8c9a439..87e9589 100644
--- a/tool/symmetric/AES.py
+++ b/tool/symmetric/AES.py
@@ -10,7 +10,7 @@ from fontTools.misc.eexec import encrypt
 class AESUtils:
     def __init__(self, key: bytes = None):
         """生成一个随机密钥"""
-        self.key = get_random_bytes(32)
+        self.key = get_random_bytes(16)
 
     def encrypt(self, data: str, mode: str = 'ECB') -> (str,str):
         """加密数据"""