From d4866ab06ab352ebbb928dc69a66766edf1afaa2 Mon Sep 17 00:00:00 2001
From: phyrt9af5 <937290500@qq.com>
Date: Sat, 24 Dec 2022 22:45:30 +0800
Subject: [PATCH] ADD file via upload

---
 client.py | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 client.py

diff --git a/client.py b/client.py
new file mode 100644
index 0000000..871fef3
--- /dev/null
+++ b/client.py
@@ -0,0 +1,67 @@
+import socket
+import threading
+import pickle
+from algorithms import RSAalgorithm
+
+PORT = 4396
+BUFF = 1024
+
+def SendMessage(Sock, test):
+    while True:
+        SendData = input()
+        (encryptdata, PrivateKey) = RSAalgorithm.RsaEncrypt(SendData)
+        print('encrypted data is ' + str(encryptdata))
+        Message = pickle.dumps([encryptdata, PrivateKey])
+        if len(SendData) > 0:
+            Sock.send(Message)
+
+
+def RecvMessage(Sock, test):
+    while True:
+        Message = Sock.recv(BUFF)
+        (recvdata, PrivateKey) = pickle.loads(Message)
+        decryptdata = RSAalgorithm.RsaDecrypt(recvdata, PrivateKey)
+        if len(Message) > 0:
+            print("receive message:" + decryptdata)
+
+#获取本机ip地址
+def get_host_ip():
+    try:
+        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        s.connect(('8.8.8.8', 80))
+        ip = s.getsockname()[0]
+    finally:
+        s.close()
+
+    return ip
+
+
+def main():
+    type = input('您是server还是client？')
+    if type == 'server':
+        IPADD=get_host_ip()
+        print("您的ip地址为：",IPADD)
+        ServerSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        ServerSock.bind((IPADD, PORT))
+        ServerSock.listen(5)
+        print("正在监听中...")
+        while True:
+            ConSock, addr = ServerSock.accept()
+            print('connection succeed' + '\n' + 'you can chat online')
+            thread_1 = threading.Thread(target=SendMessage, args=(ConSock, None))
+            thread_2 = threading.Thread(target=RecvMessage, args=(ConSock, None))
+            thread_1.start()
+            thread_2.start()
+    elif type == 'client':
+        ClientSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        ServerAddr = input("请输入您要通讯的ip地址")
+        ClientSock.connect((ServerAddr, PORT))
+        print('连接成功，可以开始传输消息和文件了')
+        thread_3 = threading.Thread(target=SendMessage, args=(ClientSock, None))
+        thread_4 = threading.Thread(target=RecvMessage, args=(ClientSock, None))
+        thread_3.start()
+        thread_4.start()
+
+
+if __name__ == '__main__':
+    main()