Compare commits
No commits in common. 'ebddfeff74711c41e8eb92c441ed8e627db13799' and '2bbb0962a150faf16febfad1e14395e694ea1a77' have entirely different histories.
ebddfeff74
...
2bbb0962a1
@ -1,141 +0,0 @@
|
|||||||
import hashlib
|
|
||||||
import json
|
|
||||||
import selectors
|
|
||||||
import socket
|
|
||||||
import threading
|
|
||||||
from time import sleep
|
|
||||||
|
|
||||||
import select
|
|
||||||
|
|
||||||
from entity.Letter import Letter, json_to_obj
|
|
||||||
import recv
|
|
||||||
from sender import sender
|
|
||||||
|
|
||||||
|
|
||||||
# from entity.Letter import Letter
|
|
||||||
|
|
||||||
|
|
||||||
# def main():
|
|
||||||
# # 用户输入各种数据填充letter字段
|
|
||||||
# # 获取用户的公私钥对进行签名
|
|
||||||
# # 使用对方的公钥进行加密
|
|
||||||
# # 发送信件
|
|
||||||
# pass
|
|
||||||
#
|
|
||||||
#
|
|
||||||
# def sendLetter(letter: Letter, target="192.168.195.162:8426"):
|
|
||||||
# # 向目标ip和端口发送指定的信件
|
|
||||||
# pass
|
|
||||||
|
|
||||||
class recver_net():
|
|
||||||
def __init__(self, port, client_host, client_port):
|
|
||||||
self.client_host = client_host
|
|
||||||
self.client_port = client_port
|
|
||||||
self.letter = Letter()
|
|
||||||
self.port = port
|
|
||||||
self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
||||||
self.server.bind(('0.0.0.0', self.port))
|
|
||||||
self.server.listen(10)
|
|
||||||
|
|
||||||
def recver_thread(self):
|
|
||||||
while True:
|
|
||||||
conn, addr = self.server.accept()
|
|
||||||
data = conn.recv(10240)
|
|
||||||
msg = json.loads(data.decode())
|
|
||||||
if int(msg['flag']) == 0:
|
|
||||||
print("[+] haved received data:" + msg["data"])
|
|
||||||
elif int(msg['flag']) == 1:
|
|
||||||
self.recv_file(msg['data'])
|
|
||||||
else:
|
|
||||||
print("[x] Error")
|
|
||||||
|
|
||||||
def send_thread(self, conn):
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
flag: int = int(input())
|
|
||||||
if flag == 0:
|
|
||||||
data = input("[-] Enter data to send: ")
|
|
||||||
msg = {"flag": 0, "data": data}
|
|
||||||
msg = json.dumps(msg).encode("utf-8")
|
|
||||||
self.send_data(conn, msg)
|
|
||||||
elif flag == 1:
|
|
||||||
print("[+] Calling file transfer module...")
|
|
||||||
# file_name = input("Enter file name to send: ")
|
|
||||||
self.send_file(conn)
|
|
||||||
else:
|
|
||||||
print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
|
|
||||||
except ValueError:
|
|
||||||
print("[x] Error, please enter a valid number.")
|
|
||||||
|
|
||||||
def send_data(self, conn, data):
|
|
||||||
conn.sendall(data)
|
|
||||||
|
|
||||||
def send_file(self, conn):
|
|
||||||
self.letter: Letter = sender.main()
|
|
||||||
letter = self.letter.to_dict()
|
|
||||||
msg = {"flag": 1, "data": letter}
|
|
||||||
msg = json.dumps(msg).encode("utf-8")
|
|
||||||
conn.sendall(msg)
|
|
||||||
|
|
||||||
def recv_data(self, conn):
|
|
||||||
con, addr = conn.accept()
|
|
||||||
data = con.recv(1024)
|
|
||||||
if not data:
|
|
||||||
return None
|
|
||||||
# print(data.decode())
|
|
||||||
return data.decode()
|
|
||||||
|
|
||||||
def recv_file(self, data):
|
|
||||||
letter: Letter = json_to_obj(data)
|
|
||||||
recv.handleLetter(letter)
|
|
||||||
|
|
||||||
def main(self):
|
|
||||||
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
client = socket.socket() # 定义协议类型,相当于生命socket类型,同时生成socket连接对象
|
|
||||||
client.connect((self.client_host, self.client_port))
|
|
||||||
print("[*] Connected...")
|
|
||||||
print("[*] if you want to send data, enter 0, if you want to send file, enter 1.")
|
|
||||||
sleep(1)
|
|
||||||
break
|
|
||||||
except socket.error:
|
|
||||||
print("[*] Waiting for ...")
|
|
||||||
# 加入线程
|
|
||||||
threading.Thread(target=self.recver_thread, ).start()
|
|
||||||
threading.Thread(target=self.send_thread, args=(client,)).start()
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
threading.Thread(target=self.main).start()
|
|
||||||
|
|
||||||
|
|
||||||
def input_verify():
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
port = int(input("[*] Enter to the listen port: "))
|
|
||||||
break
|
|
||||||
except ValueError:
|
|
||||||
print("[x] Error, please enter a valid port number.")
|
|
||||||
|
|
||||||
while True:
|
|
||||||
addr = input("[*] Enter the address to connect to(127.0.0.1:8424): ")
|
|
||||||
if ':' in addr:
|
|
||||||
client_host, client_port = addr.split(":")
|
|
||||||
if client_host and client_port.isdigit():
|
|
||||||
client_port = int(client_port)
|
|
||||||
if 0 <= client_port <= 65535: # 检查端口范围
|
|
||||||
break # 输入有效，跳出循环
|
|
||||||
else:
|
|
||||||
print("Port must be between 0 and 65535.")
|
|
||||||
else:
|
|
||||||
print("Invalid address format. Please enter in the format 'host:port'.")
|
|
||||||
else:
|
|
||||||
print("Invalid address format. Please enter in the format 'host:port'.")
|
|
||||||
return port, client_host, client_port
|
|
||||||
|
|
||||||
|
|
||||||
# test
|
|
||||||
if __name__ == '__main__':
|
|
||||||
port, client_host, client_port = input_verify()
|
|
||||||
a = recver_net(port, client_host, client_port)
|
|
||||||
a.run()
|
|
||||||
@ -1,143 +0,0 @@
|
|||||||
import hashlib
|
|
||||||
import json
|
|
||||||
import selectors
|
|
||||||
import socket
|
|
||||||
import threading
|
|
||||||
from time import sleep
|
|
||||||
from entity.Letter import Letter,json_to_obj
|
|
||||||
import select
|
|
||||||
import sender
|
|
||||||
from recv import recv
|
|
||||||
|
|
||||||
|
|
||||||
# from entity.Letter import Letter
|
|
||||||
|
|
||||||
|
|
||||||
# def main():
|
|
||||||
# # 用户输入各种数据填充letter字段
|
|
||||||
# # 获取用户的公私钥对进行签名
|
|
||||||
# # 使用对方的公钥进行加密
|
|
||||||
# # 发送信件
|
|
||||||
# pass
|
|
||||||
#
|
|
||||||
#
|
|
||||||
# def sendLetter(letter: Letter, target="192.168.195.162:8426"):
|
|
||||||
# # 向目标ip和端口发送指定的信件
|
|
||||||
# pass
|
|
||||||
|
|
||||||
class sender_net():
|
|
||||||
def __init__(self,port=8424,client_host='127.0.0.1',client_port=8424):
|
|
||||||
self.client_host = client_host
|
|
||||||
self.client_port = client_port
|
|
||||||
self.letter = Letter()
|
|
||||||
self.port=port
|
|
||||||
self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
||||||
self.server.bind(('0.0.0.0', self.port))
|
|
||||||
self.server.listen(10)
|
|
||||||
|
|
||||||
|
|
||||||
def recver_thread(self):
|
|
||||||
while True:
|
|
||||||
conn, addr = self.server.accept()
|
|
||||||
data = conn.recv(10240)
|
|
||||||
msg = json.loads(data.decode())
|
|
||||||
if int(msg['flag']) == 0:
|
|
||||||
print("[+] haved received data:" + msg["data"])
|
|
||||||
elif int(msg['flag']) == 1:
|
|
||||||
self.recv_file(msg['data'])
|
|
||||||
else:
|
|
||||||
print("[x] Error")
|
|
||||||
|
|
||||||
def send_thread(self, conn):
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
flag: int = int(input())
|
|
||||||
if flag == 0:
|
|
||||||
data = input("[-] Enter data to send: ")
|
|
||||||
msg = {"flag": 0, "data": data}
|
|
||||||
msg = json.dumps(msg).encode("utf-8")
|
|
||||||
self.send_data(conn, msg)
|
|
||||||
elif flag == 1:
|
|
||||||
print("[+] Calling file transfer module...")
|
|
||||||
# file_name = input("Enter file name to send: ")
|
|
||||||
self.send_file(conn)
|
|
||||||
else:
|
|
||||||
print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
|
|
||||||
except ValueError:
|
|
||||||
print("[x] Error, please enter a valid number.")
|
|
||||||
|
|
||||||
|
|
||||||
def send_data(self, conn, msg):
|
|
||||||
conn.sendall(msg)
|
|
||||||
|
|
||||||
def send_file(self, conn):
|
|
||||||
self.letter: Letter = sender.main()
|
|
||||||
letter = self.letter.to_dict()
|
|
||||||
msg = {"flag": 1, "data": letter}
|
|
||||||
print(letter)
|
|
||||||
msg = json.dumps(msg).encode("utf-8")
|
|
||||||
conn.sendall(msg)
|
|
||||||
|
|
||||||
def recv_data(self, conn):
|
|
||||||
con, addr = conn.accept()
|
|
||||||
data = con.recv(1024)
|
|
||||||
if not data:
|
|
||||||
return None
|
|
||||||
# print(data.decode())
|
|
||||||
return data.decode()
|
|
||||||
|
|
||||||
def recv_file(self, data):
|
|
||||||
letter: Letter = json_to_obj(data)
|
|
||||||
recv.handleLetter(letter)
|
|
||||||
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
threading.Thread(target=self.main).start()
|
|
||||||
|
|
||||||
def main(self):
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
client = socket.socket() # 定义协议类型,相当于生命socket类型,同时生成socket连接对象
|
|
||||||
client.connect((self.client_host, self.client_port))
|
|
||||||
print(" [*] Connected...")
|
|
||||||
print("[*] if you want to send data, enter 0, if you want to send file, enter 1.")
|
|
||||||
sleep(1)
|
|
||||||
break
|
|
||||||
except socket.error:
|
|
||||||
print("[*] Waiting for ...")
|
|
||||||
#加入线程
|
|
||||||
threading.Thread(target=self.recver_thread, ).start()
|
|
||||||
threading.Thread(target=self.send_thread, args=(client,)).start()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def input_verify():
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
port = int(input("[*] Enter to the listen port: "))
|
|
||||||
break
|
|
||||||
except ValueError:
|
|
||||||
print("[x] Error, please enter a valid port number.")
|
|
||||||
|
|
||||||
while True:
|
|
||||||
addr = input("[*] Enter the address to connect to(127.0.0.1:8424): ")
|
|
||||||
if ':' in addr:
|
|
||||||
client_host, client_port = addr.split(":")
|
|
||||||
if client_host and client_port.isdigit():
|
|
||||||
client_port = int(client_port)
|
|
||||||
if 0 <= client_port <= 65535: # 检查端口范围
|
|
||||||
break # 输入有效，跳出循环
|
|
||||||
else:
|
|
||||||
print("Port must be between 0 and 65535.")
|
|
||||||
else:
|
|
||||||
print("Invalid address format. Please enter in the format 'host:port'.")
|
|
||||||
else:
|
|
||||||
print("Invalid address format. Please enter in the format 'host:port'.")
|
|
||||||
return port, client_host, client_port
|
|
||||||
|
|
||||||
|
|
||||||
#test
|
|
||||||
if __name__ == '__main__':
|
|
||||||
port, client_host, client_port = input_verify()
|
|
||||||
a = sender_net(port, client_host, client_port)
|
|
||||||
a.run()
|
|
||||||
Loading…
Reference in new issue