|
|
import socket
|
|
|
import threading
|
|
|
import time
|
|
|
|
|
|
class TcpClient(object):
|
|
|
def __init__(self):
|
|
|
# 配置信息
|
|
|
self.udpPort = 54321
|
|
|
self.broadcastAddr = '<broadcast>'
|
|
|
def discoverServers(self):
|
|
|
print("[*] 正在扫描局域网设备...")
|
|
|
udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
|
udpSock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|
|
udpSock.settimeout(2)
|
|
|
|
|
|
udpSock.sendto(b"DISCOVERY_REQUEST", (self.broadcastAddr, self.udpPort))
|
|
|
|
|
|
servers = []
|
|
|
try:
|
|
|
while True:
|
|
|
data, addr = udpSock.recvfrom(1024)
|
|
|
if data.startswith(b"DISCOVERY_RESPONSE"):
|
|
|
# 修正点：正确解析服务端IP和TCP端口
|
|
|
_, hostname, tcp_port = data.decode('utf-8').split(':')
|
|
|
server_ip = addr # 提取IP字符串
|
|
|
servers.append({
|
|
|
'ip': server_ip, # 存储为字符串
|
|
|
'port': int(tcp_port), # 存储为整数
|
|
|
'hostname': hostname
|
|
|
})
|
|
|
print()
|
|
|
except socket.timeout:
|
|
|
pass
|
|
|
finally:
|
|
|
udpSock.close()
|
|
|
return servers
|
|
|
|
|
|
def connectToServer(self, ip, tcp_port):
|
|
|
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
try:
|
|
|
# 显式转换类型确保安全
|
|
|
client.connect((str(ip), int(tcp_port))) # :ml-citation{ref="2,3" data="citationList"}
|
|
|
print(f"[+] 已连接到 {ip}:{tcp_port}")
|
|
|
client.send(b"Hello from client!")
|
|
|
response = client.recv(1024)
|
|
|
print(f"服务端响应: {response.decode('utf-8')}")
|
|
|
except Exception as e:
|
|
|
print(f"连接失败: {e}")
|
|
|
finally:
|
|
|
client.close()
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
tcpClient = TcpClient()
|
|
|
servers = tcpClient.discoverServers()
|
|
|
if not servers:
|
|
|
print("[-] 未找到可用服务端")
|
|
|
exit()
|
|
|
|
|
|
print("发现以下设备:")
|
|
|
for i, s in enumerate(servers):
|
|
|
print(f"{i+1}. {s['hostname']} ({s['ip']}:{s['port']})") # 显示格式修正
|
|
|
print(servers)
|
|
|
choice = int(input("请输入要连接的设备编号: ")) - 1
|
|
|
selected = servers[choice]
|
|
|
print(selected['ip'][0], selected['port'],88)
|
|
|
tcpClient.connectToServer(selected['ip'][0], selected['port']) # 传递字符串和整数
|