新的生成文件

rsa_key_generate.py

@@ -1,24 +1,24 @@
-from OpenSSL import crypto
+from OpenSSL import crypto
-import tkinter as tk
+import tkinter as tk
-import tkinter.filedialog
+import tkinter.filedialog
-# 生成 RSA 密钥对
+# 生成 RSA 密钥对
-key = crypto.PKey()
+key = crypto.PKey()
-key.generate_key(crypto.TYPE_RSA, 2048)  # 2048 是密钥位大小，也可以是 1024, 4096 等
+key.generate_key(crypto.TYPE_RSA, 2048)  # 2048 是密钥位大小，也可以是 1024, 4096 等
-# 提取私钥并将其转换为 PEM 格式
+# 提取私钥并将其转换为 PEM 格式
-private_key_pem = crypto.dump_privatekey(crypto.FILETYPE_PEM, key).decode('utf-8')
+private_key_pem = crypto.dump_privatekey(crypto.FILETYPE_ASN1, key)
-# 提取公钥并将其转换为 PEM 格式
+# 提取公钥并将其转换为 PEM 格式
-public_key_pem = crypto.dump_publickey(crypto.FILETYPE_PEM, key).decode('utf-8')
+public_key_pem = crypto.dump_publickey(crypto.FILETYPE_PEM, key)
-# 将私钥保存到文件
+# 将私钥保存到文件
-root = tk.Tk()
+root = tk.Tk()
-root.withdraw()
+root.withdraw()
-try:
+try:
-    file_path = tk.filedialog.asksaveasfilename(title="私钥保存",defaultextension=".pem", filetypes=[("PEM files", "*.pem")])
+    file_path = tk.filedialog.asksaveasfilename(title="私钥保存",defaultextension=".pem", filetypes=[("PEM files", "*.pem")])
-    with open(file_path, 'w') as f:
+    with open(file_path, 'wb') as f:
-        f.write(private_key_pem)
+        f.write(private_key_pem)
-except:
+except:
-    print("私钥保存失败")
+    print("私钥保存失败")
-# 将公钥保存到文件
+# 将公钥保存到文件
-file_path = tk.filedialog.asksaveasfilename(title="公钥保存",defaultextension=".pem", filetypes=[("PEM files", "*.pem")])
+file_path = tk.filedialog.asksaveasfilename(title="公钥保存",defaultextension=".pem", filetypes=[("PEM files", "*.pem")])
-with open(file_path, 'w') as f:
+with open(file_path, 'wb') as f:
-    f.write(public_key_pem)
+    f.write(public_key_pem)
 print("RSA 密钥对已成功生成并保存为 PEM 文件。")