From f81d507a618e1594883650dca576d46d79087152 Mon Sep 17 00:00:00 2001
From: px3gkymai <1565756097@qq.com>
Date: Sat, 17 Dec 2022 16:12:08 +0800
Subject: [PATCH] ADD file via upload

---
 server.py | 105 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 server.py

diff --git a/server.py b/server.py
new file mode 100644
index 0000000..2f65425
--- /dev/null
+++ b/server.py
@@ -0,0 +1,105 @@
+import os
+import socket
+from threading import Thread
+from tkinter import LEFT, RIGHT, Button, Label, StringVar, ttk
+from tkinter.tix import Tk  # 升级的组合控件包
+from server_twokey import server_twokey
+from ttkbootstrap import Style
+import tqdm
+
+from server_panduan import panduan
+
+
+def agree():
+    # 设计IP和端口
+    SERVER_HOST = "0.0.0.0"
+    SERVER_PORT = 9799
+
+    # 文件读写缓冲区
+    BUFFER_SIZE = 4096
+    SEPARATOR = "<SEPARATOR>"
+
+    # 创建Server
+    s = socket.socket()
+    s.bind((SERVER_HOST, SERVER_PORT))
+
+    # 设置连接数
+    s.listen(5)
+    print(f"服务器监听{SERVER_HOST}:{SERVER_PORT}")
+
+    while True:
+        # 接受客户端连接
+        client_socket, address = s.accept()
+        zhaiyao = Label(
+            root, text="接收成功！", bg="red", bd=1, font=("Arial", 12), width=10, height=1
+        )
+        zhaiyao.pack()
+        # 输出客户端IP
+        print(f"客户端{address}连接")
+        # 接受客户端信息
+        received = client_socket.recv(BUFFER_SIZE).decode()
+        filename, file_size = received.split(SEPARATOR)
+        # 获取传输文件名和大小
+        filename = os.path.basename(filename)
+        file_size = int(file_size)
+
+        # 文件接受处理
+        progress = tqdm.tqdm(
+            range(file_size),
+            f"接收{filename}",
+            unit="B",
+            unit_divisor=1024,
+            unit_scale=True,
+        )
+        with open(filename, "wb") as f:
+            for _ in progress:
+                # 读取客户端数据
+                bytes_read = client_socket.recv(BUFFER_SIZE)
+                if not bytes_read:
+                    break
+                f.write(bytes_read)
+                progress.update(len(bytes_read))
+        # 关闭
+        client_socket.close()
+        print("客户端关闭")
+    s.close()
+
+
+def socketStart():
+    print("服务器启动")
+    t = Thread(target=agree)
+    t.start()
+
+
+def init_window():
+    global root
+    root = Tk()
+    style = Style(theme='minty')
+    root.title("服务器端")
+    root.geometry("700x500")
+    root.configure(bg='white')
+    root.resizable(width=False, height=False)
+    root.tk.eval("package require Tix")  # 引入升级包，这样才能使用升级的组合控件
+    ttk.Button(root, text="开始接收", command=socketStart,bootstyle="default-outline").place(x=400, y=210, width=200, height=60)
+    model = StringVar()
+    model.set("请选择解密模式")
+    values = ["DES", "AES_ECB", "AES_CBC"]
+    ttk.Combobox(
+        master=root,  # 父容器
+        height=10,  # 高度,下拉显示的条目数量
+        width=20,  # 宽度
+        state="readonly",  # 设置状态 normal(可选可输入)、readonly(只可选)、 disabled
+        cursor="arrow",  # 鼠标移动时样式 arrow, circle, cross, plus...  # 字体
+        textvariable=model,  # 通过StringVar设置可改变的值
+        values=values,
+        bootstyle="info",# 设置下拉框的选项
+    ).place(x=30,y=230,width=180,height=30)
+    ttk.Button(
+        root,text="生成接收端公私钥",command=lambda: server_twokey(),bootstyle="default-outline"
+    ).place(x=400, y=50, width=200, height=60)
+    ttk.Button(root, text="解密", command=lambda: panduan(root=root, op=model.get()),bootstyle="default-outline").place(x=400,y=380,width=200, height=60)
+    root.mainloop()
+
+
+if "__main__" == __name__:
+    init_window()