sangge/mimajingsai_2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sytle: modify style

Browse Source
This commit is contained in:
sangge
2023-11-29 16:22:56 +08:00
parent 32e6af1ae8
commit 217c8c1614
1 changed files with 150 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
数据查询方_new.py
View File

@@ -7,6 +7,7 @@ from io import StringIO
import CoreAlgorithm import CoreAlgorithm
import DataSearch import DataSearch
# 审批回执类消息head：01001001 # 审批回执类消息head：01001001
# sql密文类消息head：01001010 # sql密文类消息head：01001010
# 元数据类消息head：01001100 # 元数据类消息head：01001100
@@ -16,8 +17,9 @@ import tkinter as tk
from tkinter import * from tkinter import *
from tkinter import messagebox, scrolledtext from tkinter import messagebox, scrolledtext
from PIL import ImageTk from PIL import ImageTk
# 美化 # 美化
from ttkbootstrap import Style from ttkbootstrap import Style # pylint: disable=e0401 # type: ignore
from tkinter import ttk from tkinter import ttk
@@ -29,12 +31,13 @@ def generate_key_pair_data():
# 用公钥为本地生成数字证书 # 用公钥为本地生成数字证书
def get_certificate(temp_public_key, cert_name): def get_certificate(temp_public_key, cert_name):
public_key_str = '\n'.join(temp_public_key[i:i + 60] for i in range(0, len(temp_public_key), 60)) public_key_str = "\n".join(
pack_public_key = "-----BEGIN PUBLIC KEY-----\n" + public_key_str + "\n-----END PUBLIC KEY-----" temp_public_key[i : i + 60] for i in range(0, len(temp_public_key), 60)
cert = { )
'public_key': pack_public_key, pack_public_key = (
'name': "<数据查询方>" + cert_name "-----BEGIN PUBLIC KEY-----\n" + public_key_str + "\n-----END PUBLIC KEY-----"
} )
cert = {"public_key": pack_public_key, "name": "<数据查询方>" + cert_name}
return cert return cert
@@ -53,7 +56,7 @@ def str_to_encrypt(message, public_data):
if message.isdigit(): if message.isdigit():
int_message = int(message) int_message = int(message)
else: else:
int_message = int.from_bytes(message.encode(), 'big') int_message = int.from_bytes(message.encode(), "big")
enc_message = public_data.encrypt(int_message) enc_message = public_data.encrypt(int_message)
print("int_message", int_message) print("int_message", int_message)
return enc_message return enc_message
@@ -78,7 +81,10 @@ class MyClient:
self.give_name() self.give_name()
# 生成私钥和公钥信息 # 生成私钥和公钥信息
self.private_key_data, self.public_key_data = generate_key_pair_data() # PaillierPublicKey 类型 (
self.private_key_data,
self.public_key_data,
) = generate_key_pair_data() # PaillierPublicKey 类型
# 生成私钥和公钥字符串 # 生成私钥和公钥字符串
self.private_key, self.public_key = self.generate_key_pair() self.private_key, self.public_key = self.generate_key_pair()
# 获取数字证书 # 获取数字证书
@@ -101,16 +107,18 @@ class MyClient:
# 为本端取名 # 为本端取名
def give_name(self): def give_name(self):
self.window = tk.Toplevel() self.window = tk.Toplevel()
self.window.title('命名窗口') self.window.title("命名窗口")
self.window.geometry('300x320') self.window.geometry("300x320")
# 初始化信息输入框 # 初始化信息输入框
self.name_text = tk.StringVar() self.name_text = tk.StringVar()
tk.Label(self.window, bg='green', text='请为本端取名:').place(x=50, y=100) tk.Label(self.window, bg="green", text="请为本端取名:").place(x=50, y=100)
input_message = tk.Entry(self.window, textvariable=self.name_text) input_message = tk.Entry(self.window, textvariable=self.name_text)
input_message.pack() input_message.pack()
input_message.place(x=80, y=150) input_message.place(x=80, y=150)
# 确认按钮及位置 # 确认按钮及位置
bt_confirm = tk.Button(self.window, bg='green', text='确认', width=10, command=self.get_name) bt_confirm = tk.Button(
self.window, bg="green", text="确认", width=10, command=self.get_name
)
bt_confirm.pack() bt_confirm.pack()
bt_confirm.place(x=125, y=220) bt_confirm.place(x=125, y=220)
@@ -124,15 +132,17 @@ class MyClient:
# 编写sql # 编写sql
def create_sql(self): def create_sql(self):
self.window = tk.Toplevel() # 建立窗口 self.window = tk.Toplevel() # 建立窗口
self.window.title('编写sql语句') # 为窗口命名 self.window.title("编写sql语句") # 为窗口命名
self.window.geometry('580x270') # 设定窗口大小 self.window.geometry("580x270") # 设定窗口大小
# 初始化信息输入框 # 初始化信息输入框
self.sql_text = tk.StringVar() self.sql_text = tk.StringVar()
tk.Label(self.window, bg='green', text='请输入您需要的sql查询:').place(x=70, y=50) tk.Label(self.window, bg="green", text="请输入您需要的sql查询:").place(x=70, y=50)
input_message = tk.Entry(self.window, textvariable=self.sql_text, width=60) input_message = tk.Entry(self.window, textvariable=self.sql_text, width=60)
input_message.place(x=70, y=100) input_message.place(x=70, y=100)
# 初始化确认按钮 # 初始化确认按钮
bt_confirm = tk.Button(self.window, bg='green', text='确认', width=60, command=self.get_sql) bt_confirm = tk.Button(
self.window, bg="green", text="确认", width=60, command=self.get_sql
)
bt_confirm.place(x=70, y=150) bt_confirm.place(x=70, y=150)
# self.sql = input("请编写你的sql:") # self.sql = input("请编写你的sql:")
@@ -149,7 +159,7 @@ class MyClient:
def safe_connect(self): def safe_connect(self):
if self.certificate != {}: if self.certificate != {}:
# 与服务器端地址建立通信, 8080为服务端程序的端口号 # 与服务器端地址建立通信, 8080为服务端程序的端口号
self.client.connect(("localhost",1111)) self.client.connect(("localhost", 1111))
# 为接收消息函数添加线程 # 为接收消息函数添加线程
threading.Thread(target=self.get_message).start() threading.Thread(target=self.get_message).start()
time.sleep(3) time.sleep(3)
@@ -178,7 +188,7 @@ class MyClient:
# 加密封装sql语句 # 加密封装sql语句
def deal_sql(self, context): def deal_sql(self, context):
message = (str(self.client.getsockname()) + "||" + context) message = str(self.client.getsockname()) + "||" + context
global server_public_key_data global server_public_key_data
enc_message = str_to_encrypt(message, server_public_key_data) # 用平台的公钥进行加密 enc_message = str_to_encrypt(message, server_public_key_data) # 用平台的公钥进行加密
return "01001010" + str(enc_message.ciphertext()) # int 型 return "01001010" + str(enc_message.ciphertext()) # int 型
@@ -198,8 +208,14 @@ class MyClient:
print("本地公钥信息为:") print("本地公钥信息为:")
print(self.public_key_data) print(self.public_key_data)
print("打印公钥如下:") print("打印公钥如下:")
public_key_str = '\n'.join(self.public_key[i:i + 60] for i in range(0, len(self.public_key), 60)) public_key_str = "\n".join(
pack_public_key = "\n-----BEGIN PUBLIC KEY-----\n" + public_key_str + "\n-----END PUBLIC KEY-----\n" self.public_key[i : i + 60] for i in range(0, len(self.public_key), 60)
)
pack_public_key = (
"\n-----BEGIN PUBLIC KEY-----\n"
+ public_key_str
+ "\n-----END PUBLIC KEY-----\n"
)
print(pack_public_key) print(pack_public_key)
# show_list() # show_list()
messagebox.showinfo("本地公钥", pack_public_key) # GUI界面显示 messagebox.showinfo("本地公钥", pack_public_key) # GUI界面显示
@@ -208,8 +224,14 @@ class MyClient:
def print_private_key(self): def print_private_key(self):
print("本地私钥信息为:") print("本地私钥信息为:")
print(self.private_key_data) print(self.private_key_data)
private_key_str = '\n'.join(self.private_key[i:i + 60] for i in range(0, len(self.public_key), 60)) private_key_str = "\n".join(
pack_private_key = "-----BEGIN PRIVATE KEY-----\n" + private_key_str + "\n-----END PRIVATE KEY-----\n" self.private_key[i : i + 60] for i in range(0, len(self.public_key), 60)
)
pack_private_key = (
"-----BEGIN PRIVATE KEY-----\n"
+ private_key_str
+ "\n-----END PRIVATE KEY-----\n"
)
print("打印私钥如下:") print("打印私钥如下:")
print(pack_private_key) print(pack_private_key)
messagebox.showinfo("本地私钥", pack_private_key) # GUI界面显示 messagebox.showinfo("本地私钥", pack_private_key) # GUI界面显示
@@ -248,8 +270,11 @@ class MyClient:
print(key, ":\n", value) print(key, ":\n", value)
message += key + ":\n" + value message += key + ":\n" + value
messagebox.showinfo("平台的公钥证书", message) messagebox.showinfo("平台的公钥证书", message)
server_public_key = cert["public_key"].split("-----END PUBLIC KEY-----")[0].split( server_public_key = (
"-----BEGIN PUBLIC KEY-----")[1] cert["public_key"]
.split("-----END PUBLIC KEY-----")[0]
.split("-----BEGIN PUBLIC KEY-----")[1]
)
server_public_key = server_public_key.replace("\n", "") server_public_key = server_public_key.replace("\n", "")
print("提取到的server_public_key:") print("提取到的server_public_key:")
print(server_public_key) print(server_public_key)
@@ -267,14 +292,20 @@ class MyClient:
messagebox.showinfo("平台发来的消息", "与平台的双向认证成功！") messagebox.showinfo("平台发来的消息", "与平台的双向认证成功！")
elif message.startswith("01001010"): elif message.startswith("01001010"):
message = message.split("01001010", 1)[1] # [0]是空字符串，已测试。只切一次是防止密文出现和头部一样的信息被误切。 message = message.split("01001010", 1)[
1
] # [0]是空字符串，已测试。只切一次是防止密文出现和头部一样的信息被误切。
# 使用私钥解密消息,获得sql明文 # 使用私钥解密消息,获得sql明文
print("接收到的sql密文:", message) print("接收到的sql密文:", message)
int_message = int(message) int_message = int(message)
# Ciphertext转EncryptedNumber # Ciphertext转EncryptedNumber
Encrpt_message = CoreAlgorithm.EncryptedNumber(self.public_key_data, int_message, exponent=0) Encrpt_message = CoreAlgorithm.EncryptedNumber(
self.public_key_data, int_message, exponent=0
)
dec_int_message = self.private_key_data.decrypt(Encrpt_message) dec_int_message = self.private_key_data.decrypt(Encrpt_message)
dec_message = dec_int_message.to_bytes((dec_int_message.bit_length() + 7) // 8, 'big').decode() dec_message = dec_int_message.to_bytes(
(dec_int_message.bit_length() + 7) // 8, "big"
).decode()
print("解密后的消息为:", dec_message) print("解密后的消息为:", dec_message)
self.sql = dec_message.split("||")[2] self.sql = dec_message.split("||")[2]
print("收到已授权方的sql:", self.sql) print("收到已授权方的sql:", self.sql)
@@ -285,28 +316,36 @@ class MyClient:
if message != "正在等待接收其他信息...": if message != "正在等待接收其他信息...":
int_message = int(message) int_message = int(message)
# Ciphertext转EncryptedNumber # Ciphertext转EncryptedNumber
Encrpt_message = CoreAlgorithm.EncryptedNumber(self.public_key_data, int_message, exponent=0) Encrpt_message = CoreAlgorithm.EncryptedNumber(
self.public_key_data, int_message, exponent=0
)
dec_int_message = self.private_key_data.decrypt(Encrpt_message) dec_int_message = self.private_key_data.decrypt(Encrpt_message)
print("分析结果:", dec_int_message) print("分析结果:", dec_int_message)
messagebox.showinfo("分析结果", dec_int_message) messagebox.showinfo("分析结果", dec_int_message)
else: else:
messagebox.showinfo("提示", message) messagebox.showinfo("提示", message)
# --------------------------接收sql消息的窗口设计函数------------------------- # --------------------------接收sql消息的窗口设计函数-------------------------
# 接受调用数据的请求 # 接受调用数据的请求
def send_accept_reply(self): def send_accept_reply(self):
self.client.send(("01001001" + str(self.client.getsockname()) + "||安全多方平台同意了你的数据申请。").encode()) self.client.send(
(
"01001001" + str(self.client.getsockname()) + "||安全多方平台同意了你的数据申请。"
).encode()
)
self.window = tk.Toplevel() self.window = tk.Toplevel()
self.window.title('命名窗口') self.window.title("命名窗口")
self.window.geometry('300x320') self.window.geometry("300x320")
# 信息输入框 # 信息输入框
self.data_text = tk.StringVar() self.data_text = tk.StringVar()
tk.Label(self.window, bg='green', text='请输入要调用的信息:').place(x=50, y=100) tk.Label(self.window, bg="green", text="请输入要调用的信息:").place(x=50, y=100)
input_message = tk.Entry(self.window, textvariable=self.data_text) input_message = tk.Entry(self.window, textvariable=self.data_text)
input_message.pack() input_message.pack()
input_message.place(x=80, y=150) input_message.place(x=80, y=150)
# 确认按钮及位置 # 确认按钮及位置
bt_confirm = tk.Button(self.window, bg='green', text='确认', width=10, command=self.send_data) bt_confirm = tk.Button(
self.window, bg="green", text="确认", width=10, command=self.send_data
)
bt_confirm.pack() bt_confirm.pack()
bt_confirm.place(x=125, y=220) bt_confirm.place(x=125, y=220)
@@ -325,7 +364,11 @@ class MyClient:
# 拒绝调用数据的请求 # 拒绝调用数据的请求
def send_decline_reply(self): def send_decline_reply(self):
self.client.send(("01001001" + str(self.client.getsockname()) + "||" + "想得美哦!不给(*^▽^*)").encode()) self.client.send(
(
"01001001" + str(self.client.getsockname()) + "||" + "想得美哦!不给(*^▽^*)"
).encode()
)
messagebox.showinfo("提示", "已拒绝！") messagebox.showinfo("提示", "已拒绝！")
self.window.destroy() self.window.destroy()
@@ -338,24 +381,32 @@ class MyClient:
if self.sql != "": if self.sql != "":
self.window = tk.Toplevel() self.window = tk.Toplevel()
self.window.title("平台发来的消息") self.window.title("平台发来的消息")
self.window.geometry('550x320') # 设定窗口大小 self.window.geometry("550x320") # 设定窗口大小
self.recent_message = "收到已授权方的sql:" + self.sql self.recent_message = "收到已授权方的sql:" + self.sql
tk.Label(self.window, bg='green', text=self.recent_message).place(x=50, y=100) tk.Label(self.window, bg="green", text=self.recent_message).place(
x=50, y=100
)
# 创建接受按钮 # 创建接受按钮
accept_button = tk.Button(self.window, text="接受", width=15, command=self.send_accept_reply) accept_button = tk.Button(
self.window, text="接受", width=15, command=self.send_accept_reply
)
accept_button.place(x=90, y=220) accept_button.place(x=90, y=220)
# 创建拒绝按钮 # 创建拒绝按钮
decline_button = tk.Button(self.window, text="拒绝", width=15, command=self.send_decline_reply) decline_button = tk.Button(
self.window, text="拒绝", width=15, command=self.send_decline_reply
)
decline_button.place(x=225, y=220) decline_button.place(x=225, y=220)
# 创建稍后处理按钮 # 创建稍后处理按钮
decline_button = tk.Button(self.window, text="稍后处理", width=15, command=self.wait_reply) decline_button = tk.Button(
self.window, text="稍后处理", width=15, command=self.wait_reply
)
decline_button.place(x=360, y=220) decline_button.place(x=360, y=220)
else: else:
messagebox.showinfo("提示", "暂无待处理的信息。") messagebox.showinfo("提示", "暂无待处理的信息。")
# -------------------------------------------------------------------------- # --------------------------------------------------------------------------
# 调用本地信息库 # 调用本地信息库
def search_data(self): def search_data(self):
output = StringIO() # 用于保存打印信息 output = StringIO() # 用于保存打印信息
@@ -377,31 +428,71 @@ class MyClient:
# 界面按键 # 界面按键
def create_widgets(self): def create_widgets(self):
# 创建按钮和标签等部件并使用ttk和Style进行美化 # 创建按钮和标签等部件并使用ttk和Style进行美化
style = Style(theme='darkly') # 指定样式主题 style = Style(theme="darkly") # 指定样式主题
self.root = style.master self.root = style.master
button1 = ttk.Button(self.root, text="查看我的公钥", bootstyle="info-outline", button1 = ttk.Button(
command=self.print_public_key, width=30) self.root,
text="查看我的公钥",
bootstyle="info-outline",
command=self.print_public_key,
width=30,
)
button1.place(x=104, y=360) button1.place(x=104, y=360)
button2 = ttk.Button(self.root, text="查看我的私钥", bootstyle="info-outline", button2 = ttk.Button(
command=self.print_private_key, width=30) self.root,
text="查看我的私钥",
bootstyle="info-outline",
command=self.print_private_key,
width=30,
)
button2.place(x=104, y=398) button2.place(x=104, y=398)
button3 = ttk.Button(self.root, text="生成我的证书", bootstyle="info-outline", button3 = ttk.Button(
command=self.print_certificate, width=30) self.root,
text="生成我的证书",
bootstyle="info-outline",
command=self.print_certificate,
width=30,
)
button3.place(x=104, y=433) button3.place(x=104, y=433)
button4 = ttk.Button(self.root, text="获取平台认证", bootstyle="info-outline", button4 = ttk.Button(
command=self.safe_connect, width=30) self.root,
text="获取平台认证",
bootstyle="info-outline",
command=self.safe_connect,
width=30,
)
button4.place(x=104, y=468) button4.place(x=104, y=468)
button5 = ttk.Button(self.root, text="调用SQL查询", bootstyle="info-outline", button5 = ttk.Button(
command=self.create_sql, width=30) self.root,
text="调用SQL查询",
bootstyle="info-outline",
command=self.create_sql,
width=30,
)
button5.place(x=104, y=503) button5.place(x=104, y=503)
button6 = ttk.Button(self.root, text="发送SQL请求", bootstyle="info-outline", button6 = ttk.Button(
command=self.send_sql, width=30) self.root,
text="发送SQL请求",
bootstyle="info-outline",
command=self.send_sql,
width=30,
)
button6.place(x=104, y=538) button6.place(x=104, y=538)
button7 = ttk.Button(self.root, text="处理请求信息", bootstyle="info-outline", button7 = ttk.Button(
command=self.design_window, width=30) self.root,
text="处理请求信息",
bootstyle="info-outline",
command=self.design_window,
width=30,
)
button7.place(x=104, y=573) button7.place(x=104, y=573)
button8 = ttk.Button(self.root, text="查看医疗记录", bootstyle="info-outline", button8 = ttk.Button(
command=self.search_data, width=30) self.root,
text="查看医疗记录",
bootstyle="info-outline",
command=self.search_data,
width=30,
)
button8.place(x=104, y=608) button8.place(x=104, y=608)
# GUI界面 # GUI界面