|
|
|
|
|
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()
|
