From c349bc3039027998368f20d2b210230469f13960 Mon Sep 17 00:00:00 2001
From: UniDarkstars <1830018670@qq.com>
Date: Sun, 24 Nov 2024 14:49:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BA=86=E4=B8=80=E4=B8=8B?=
 =?UTF-8?q?=EF=BC=8C=E4=BD=AC=E6=9D=A5=E5=B8=AE=E6=88=91=E8=BF=90=E8=A1=8C?=
 =?UTF-8?q?=E4=B8=80=E4=B8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sender/sender.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/sender/sender.py b/sender/sender.py
index 0535295..2831a8e 100644
--- a/sender/sender.py
+++ b/sender/sender.py
@@ -15,6 +15,7 @@ def sendLetter(letter: Letter, target="192.168.195.162:8426"):
     # 向目标ip和端口发送指定的信件
     pass
 
+
 # 选择对称加密的方法和模式(aes/sm2)
 def selectSymEncryptionChoice():
     encryWay = ""
@@ -47,18 +48,27 @@ def selectSymEncryptionChoice():
 
 
 # 使用对称加密，返回加密后的数据和随机生成的密钥
-def SymEncryption():
-
+def SymEncryption(encryData, key: bytes = None):
     # 获得加密的方法和加密的模式
     way, mode = selectSymEncryptionChoice()
 
     if way == "aes":
-        aesUtils = AESUtils()
-        encryptedData, key = aesUtils.encrypt("test", mode=mode) # 这里data要改为文件内容
+        aesUtils = AESUtils(key)
+        # tureKey 为 真实返回的Key，如果没有设置初始值，那么key是随机生成
+        encryptedData, tureKey = aesUtils.encrypt(encryData, mode=mode)  # 这里encryData要改为文件内容
         return encryptedData
 
     if way == "sm2":
         pass
 
+
 def getKey():
     pass
+
+
+if __name__ == "__main__":
+    # 示例数据和密钥
+    data = "Hello, AES!"
+    key = "1234567890abcdef"
+
+    SymEncryption(data)