main.py

@@ -1,6 +1,6 @@
 import pyfiglet # 程序greet
 
-from senders import sender_net
+from sender import sender_net
 
 # 程序greet
 print(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")

recv/recv_net.py

@@ -8,8 +8,8 @@ from time import sleep
 import select
 
 from entity.Letter import Letter, json_to_obj
-import recvs.recv
+import recv
-from senders import sender
+from sender import sender
 
 
 # from entity.Letter import Letter
@@ -87,7 +87,7 @@ class recver_net():
 
     def recv_file(self, data):
         letter: Letter = json_to_obj(data)
-        recvs.recv.handleLetter(letter)
+        recv.handleLetter(letter)
 
     def main(self):
 

recv/test_recv.py

@@ -0,0 +1,32 @@
+from unittest.mock import patch
+
+import recv
+from sender import sender
+
+
+@patch('builtins.input', side_effect=['../README.md', 'aes', 'ecb'])
+def test_handleLetter_aes_ecb(mock_input):
+    letter = sender.main()
+    recv.handleLetter(letter)
+    pass
+
+
+@patch('builtins.input', side_effect=['../README.md', 'aes', 'cbc'])
+def test_handleLetter_aes_cbc(mock_input):
+    letter = sender.main()
+    recv.handleLetter(letter)
+    pass
+
+
+@patch('builtins.input', side_effect=['../README.md', 'sm4', 'ecb'])
+def test_handleLetter_sm4_ecb(mock_input):
+    letter = sender.main()
+    recv.handleLetter(letter)
+    pass
+
+
+@patch('builtins.input', side_effect=['../README.md', 'sm4', 'cbc'])
+def test_handleLetter_sm4_cbc(mock_input):
+    letter = sender.main()
+    recv.handleLetter(letter)
+    pass

recvs/test_recv.py

@@ -1,32 +0,0 @@
-from unittest.mock import patch
-
-import recv
-from senders import sender
-
-
-@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem','aes', 'ecb'])
-def test_handleLetter_aes_ecb(mock_input):
-    letter = sender.main()
-    recv.handleLetter(letter)
-    pass
-
-
-@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'aes', 'cbc'])
-def test_handleLetter_aes_cbc(mock_input):
-    letter = sender.main()
-    recv.handleLetter(letter)
-    pass
-
-
-@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'sm4', 'ecb'])
-def test_handleLetter_sm4_ecb(mock_input):
-    letter = sender.main()
-    recv.handleLetter(letter)
-    pass
-
-
-@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'sm4', 'cbc'])
-def test_handleLetter_sm4_cbc(mock_input):
-    letter = sender.main()
-    recv.handleLetter(letter)
-    pass

sender/sender.py

@@ -1,25 +1,26 @@
 # 模式，文件，自己的公钥从哪里来，别人的公钥从哪里来
+import pyfiglet
+
 import base64
 import os
 
-import pyfiglet
-from Crypto.Random import get_random_bytes
 
 from entity.Letter import Letter
 from tool import PriKeyHelper
 from tool.PriKeyHelper import getUserKey
 from tool.asymmetric import RSA
-from tool.hash import Segwit
 from tool.symmetric.AES import AESUtils
 from tool.symmetric.SM4 import encrypt_ecb, encrypt_cbc_with_iv
+from Crypto.Random import get_random_bytes
+
 
 # 三个全局变量，记录信封
 letterWay = ""
 letterMode = ""
 letterSymKey = b""
 
-
 def main():
+
     # greet
     print("")
     print(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
@@ -34,13 +35,14 @@ def main():
     # 用户输入各种数据填充letter字段
     path = selectFile()
 
-    with open(path, "rb") as f:
+    with open(path,"rb") as f:
         data = f.read()
 
+
     letter.fileName = getFileName(path)
     letter.recvPubKey = getRecvPubKey()
     letter.senderPubKey = getSenderPubKey()
-    letter.fileBase64, akey = SymEncryption(base64.b64encode(data).decode("utf-8"), letterSymKey)
+    letter.fileBase64, akey = SymEncryption(base64.b64encode(data).decode("utf-8"),letterSymKey)
     letter.encryptKey = getEncryptKey(letter.recvPubKey)
     letter.encryptType = getEncryptType()
 
@@ -54,19 +56,12 @@ def main():
     print(letter.recvPubKey)
     print(letter.senderPubKey)
 
-    # by Timmoc
-    print(
-        f"已以 {Segwit.encodeSegwit(letter.senderPubKey.encode('utf-8'))} 的身份向"
-        f" {Segwit.encodeSegwit(letter.recvPubKey.encode('utf-8'))} 生成加密信件")
-    # by Timmoc
 
     # 获取用户的公私钥对进行签名
     # 使用对方的公钥进行加密
     # 发送信件
-    return letter  # 方便recv测试，以后可以删除。
+    return letter # 方便recv测试，以后可以删除。
     pass
-
-
 def selectFile() -> str:
     # s = "public.pem"
     while True:
@@ -74,7 +69,7 @@ def selectFile() -> str:
             s = input("输入文件路径:")
             with open(s, "rb") as _:
                 pass
-            return s
+            return  s
         except FileNotFoundError:
             print("错误: 文件未找到，请输入正确的文件路径。")
         except PermissionError:
@@ -82,13 +77,11 @@ def selectFile() -> str:
         except Exception as e:
             print(f"发生未知错误: {e}")
 
-
 # 获得文件名
-def getFileName(fName: str) -> str:
+def getFileName(fName:str) -> str:
     filePath = os.path.split(fName)
     return filePath[-1]
 
-
 def sendLetter(letter: Letter, target="192.168.195.162:8426"):
     # 向目标ip和端口发送指定的信件
     pass
@@ -136,6 +129,7 @@ def selectSymEncryptionChoice():
 
 # 使用对称加密，返回加密后的数据和随机生成的密钥
 def SymEncryption(encryData, key: bytes = None):
+
     global letterSymKey
     # 获得加密的方法和加密的模式
     way, mode = selectSymEncryptionChoice()
@@ -145,7 +139,7 @@ def SymEncryption(encryData, key: bytes = None):
         encryptedData, trueKey = aesUtils.encrypt(encryData, mode=mode)  # 这里encryData要改为文件内容
         letterSymKey = trueKey
 
-        return encryptedData, trueKey
+        return encryptedData,trueKey
 
     if way == "sm4":
         key = get_random_bytes(16)
@@ -157,30 +151,27 @@ def SymEncryption(encryData, key: bytes = None):
         letterSymKey = key
         return encrypted_data, key
 
-
 # 获得签名
 def getSign(document_bytes):
-    # 计算文件bytes
+
+    #计算文件bytes
     priKey, pubKey = PriKeyHelper.getUserKey()
     signDocuHash = RSA.sign_message(document_bytes, priKey)
 
     return signDocuHash
 
-
 # 获得加密的方法和模式，封装信封
 def getEncryptType():
     encryType = f"{letterWay}_{letterMode}".upper()
 
     return encryType
 
-
 # 对称密钥，返回的是使用接收方公钥加密后的对称密钥
 def getEncryptKey(getRecvPubKey):
     rsaEncrySymKey = RSA.encrypt_message(letterSymKey, getRecvPubKey)
 
     return base64.b64encode(rsaEncrySymKey).decode("utf-8")
 
-
 # 获得接收方的公钥
 def getRecvPubKey():
     # recPubKey = input(" plz input Receiver's Public Key: ")
@@ -191,7 +182,6 @@ def getRecvPubKey():
         recPubKey = base64.b64encode(data).decode('utf-8')
     return recPubKey
 
-
 # 获得发送方的公钥
 def getSenderPubKey():
     privateKey, publicKey = getUserKey()
@@ -208,4 +198,4 @@ if __name__ == "__main__":
     #
     # encryptType = f"{letterWay}_{letterMode}".upper()
     # print(encryptType)
-    main()
+    main()

sender/sender_net.py

@@ -4,9 +4,9 @@ import socket
 import threading
 from time import sleep
 
-import senders.sender
+import sender
 from entity.Letter import Letter, json_to_obj
-from recvs import recv
+from recv import recv
 
 
 # from entity.Letter import Letter
@@ -70,7 +70,7 @@ class sender_net():
         conn.sendall(msg)
 
     def send_file(self, conn):
-        self.letter: Letter = senders.sender.main()
+        self.letter: Letter = sender.main()
         letter = self.letter.to_dict()
         msg = {"flag": 1, "data": letter}
         print(letter)