From fe0f4a51dfa04b14f69d162ffc7c7e7d8057f68a Mon Sep 17 00:00:00 2001 From: sheep <1976965681@qq.com> Date: Wed, 25 Dec 2024 21:37:56 +0800 Subject: [PATCH 1/3] 111 --- Initialization.py | 54 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100644 Initialization.py diff --git a/Initialization.py b/Initialization.py new file mode 100644 index 0000000..ba4ef81 --- /dev/null +++ b/Initialization.py @@ -0,0 +1,54 @@ +## Python file to initialize the overall progress of the file transmit +## Finished May 31 2:20 p.m. + +import os +import shutil + + +def does_not_exist(directory_path): + if os.path.exists(directory_path): + return False + return True + + +def not_empty_dir(directory_path): + if len(os.listdir(directory_path)) == 0: + return False + return True + + +if does_not_exist('inbox'): + os.mkdir('inbox') + +if does_not_exist('outbox'): + os.mkdir('outbox') + +if does_not_exist('sent'): + os.mkdir('sent') + +if does_not_exist('sandbox_receiver'): + os.mkdir('sandbox_receiver') + +if not_empty_dir('inbox'): + shutil.rmtree(r'inbox') + os.mkdir(r'inbox') + +if not_empty_dir('sent'): + shutil.rmtree(r'sent') + os.mkdir(r'sent') + +if not_empty_dir('outbox'): + shutil.rmtree(r'outbox') + os.mkdir(r'outbox') + +try: + os.unlink('public_key_sender.txt') + os.unlink('public_key_sender.pem') + os.unlink('public_key_receiver.txt') + os.unlink('sandbox_sender/private_key_sender.txt') + os.unlink('sandbox_sender/private_key_sender.pem') + os.unlink('sandbox_receiver/private_key_receiver.txt') +except Exception as ex: + print("文件已被初始化") + +print('Initialization Complete') \ No newline at end of file -- 2.34.1 From ed67d86351354fcf6d7c81959f523827629d0370 Mon Sep 17 00:00:00 2001 From: sheep <1976965681@qq.com> Date: Wed, 25 Dec 2024 21:52:02 +0800 Subject: [PATCH 2/3] 111 --- signature_1.py | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 signature_1.py diff --git a/signature_1.py b/signature_1.py new file mode 100644 index 0000000..818714a --- /dev/null +++ b/signature_1.py @@ -0,0 +1,50 @@ + +import hashlib +from Crypto.PublicKey import RSA +from Crypto.Cipher import DES +from cryptography.hazmat.primitives import hashes +from cryptography.hazmat.primitives.asymmetric import padding +from cryptography.hazmat.primitives import serialization +from cryptography.hazmat.backends import default_backend +from Crypto.Cipher import PKCS1_OAEP +def sign(): + # 读取文件中的消息内容 + with open('sandbox_sender/sample.txt', 'rb') as original_file: + message = original_file.read() # 以二进制方式读取消息 + + # 读取私钥文件并加载私钥 + with open('sandbox_sender/private_key_sender.pem', 'rb') as private_key_file: + private_key = serialization.load_pem_private_key( + private_key_file.read(), + password=None, + backend=default_backend() + ) + + # 创建消息的哈希 + message_hash = hashes.Hash(hashes.SHA256(), backend=default_backend()) + message_hash.update(message) # 使用文件内容更新哈希 + digest = message_hash.finalize() + + # 使用 PSS 填充对消息进行签名 + signature = private_key.sign( + digest, + padding.PSS( + mgf=padding.MGF1(hashes.SHA256()), + salt_length=padding.PSS.MAX_LENGTH + ), + hashes.SHA256() + ) + + # 将签名转换为十六进制字符串 + signature_hex = signature.hex() + + # 打印签名 + #print(f"Signature (Hex): {signature_hex}") + + # 保存签名为十六进制字符串到文件 + with open('outbox/signature.txt', 'w') as f: + f.write(signature_hex) + + print("Signature saved as hex in 'outbox/signature.txt'.") + + return signature \ No newline at end of file -- 2.34.1 From 928d48f28073a2a975efe28c929981dfed95e0e0 Mon Sep 17 00:00:00 2001 From: sheep <1976965681@qq.com> Date: Wed, 25 Dec 2024 21:52:17 +0800 Subject: [PATCH 3/3] 111 --- signature_2.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 signature_2.py diff --git a/signature_2.py b/signature_2.py new file mode 100644 index 0000000..d7bf8d9 --- /dev/null +++ b/signature_2.py @@ -0,0 +1,45 @@ +from Crypto.Cipher import DES +import binascii +from Crypto.Cipher import PKCS1_OAEP +from Crypto.PublicKey import RSA +from cryptography.hazmat.primitives import hashes +from cryptography.hazmat.primitives.asymmetric import padding +from cryptography.hazmat.primitives import serialization +from cryptography.hazmat.backends import default_backend +def verify(): + # 读取解密后的消息内容 + with open('inbox/decrypted_msg.txt', 'rb') as decrypted_file: + message = decrypted_file.read() # 以二进制方式读取消息 + + # 从 PEM 文件中加载公钥 + with open('public_key_sender.pem', 'rb') as f: # 以二进制模式打开 + public_key = serialization.load_pem_public_key(f.read(), backend=default_backend()) + + # 读取十六进制格式的签名 + with open('sent/signature.txt', 'r') as signature_file: + hex_signature = signature_file.read().strip() # 读取并去掉两端的空白字符 + + # 将十六进制签名转换为字节数据 + signature = bytes.fromhex(hex_signature) + + # 计算消息的哈希（应与签名时一致） + message_hash = hashes.Hash(hashes.SHA256(), backend=default_backend()) + message_hash.update(message) + digest = message_hash.finalize() + + try: + # 使用公钥对签名进行验证 + public_key.verify( + signature, + digest, # 使用哈希值进行验证 + padding.PSS( + mgf=padding.MGF1(hashes.SHA256()), + salt_length=padding.PSS.MAX_LENGTH + ), + hashes.SHA256() + ) + print("Signature verified successfully.") + return True # 验证成功 + except Exception as e: + print(f"Verification failed: {e}") + return False # 验证失败 \ No newline at end of file -- 2.34.1