93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
import socket
|
|||
|
|
import threading
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
#SERVER_A = ("127.0.0.1", 10) # 终端IP地址
|
|||
|
|
#LISTEN_B = ("127.0.0.1", 8080) # 本地监听端口，模拟服务端B，使用组网虚拟IP：172.16.1.59:8080
|
|||
|
|
|
|||
|
|
# 从config.txt读取配置
|
|||
|
|
with open('./config.txt', 'r') as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
# 第一行为SERVER_A的IP和端口，格式为0.0.0.0:10
|
|||
|
|
server_a_parts = lines[0].strip().split(':')
|
|||
|
|
SERVER_A = (server_a_parts[0], int(server_a_parts[1]))
|
|||
|
|
|
|||
|
|
# 第二行为LISTEN_B的IP和端口，格式为0.0.0.0:10
|
|||
|
|
listen_b_parts = lines[1].strip().split(':')
|
|||
|
|
LISTEN_B = (listen_b_parts[0], int(listen_b_parts[1]))
|
|||
|
|
clients = [] # 保存所有连接到B的客户端
|
|||
|
|
|
|||
|
|
def handle_client(conn, addr):
|
|||
|
|
"""服务端B处理客户端连接"""
|
|||
|
|
print(f"[B] 客户端已连接: {addr}")
|
|||
|
|
clients.append(conn)
|
|||
|
|
try:
|
|||
|
|
while True:
|
|||
|
|
data = conn.recv(1024)
|
|||
|
|
if not data:
|
|||
|
|
break
|
|||
|
|
print(f"[B] 收到客户端消息（一般不用处理）: {data!r}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[B] 客户端异常: {e}")
|
|||
|
|
finally:
|
|||
|
|
clients.remove(conn)
|
|||
|
|
conn.close()
|
|||
|
|
print(f"[B] 客户端断开: {addr}")
|
|||
|
|
|
|||
|
|
def start_server_b():
|
|||
|
|
"""启动服务端B"""
|
|||
|
|
global server_b_started
|
|||
|
|
try:
|
|||
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|||
|
|
s.bind(LISTEN_B)
|
|||
|
|
s.listen(5)
|
|||
|
|
server_b_started = True
|
|||
|
|
print(f"[B] 模拟服务端B已启动，监听 {LISTEN_B}")
|
|||
|
|
while True:
|
|||
|
|
conn, addr = s.accept()
|
|||
|
|
threading.Thread(target=handle_client, args=(conn, addr), daemon=True).start()
|
|||
|
|
except Exception as e:
|
|||
|
|
server_b_started = False
|
|||
|
|
print(f"[B] 服务端B异常: {e}")
|
|||
|
|
|
|||
|
|
def connect_server_a():
|
|||
|
|
"""连接服务端A，并转发数据到B的所有客户端"""
|
|||
|
|
global server_b_started
|
|||
|
|
while True: # 添加外层循环以实现重连
|
|||
|
|
try:
|
|||
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||
|
|
s.connect(SERVER_A)
|
|||
|
|
print(f"[A] 已连接到服务端A: {SERVER_A}")
|
|||
|
|
while True:
|
|||
|
|
# 检查服务端B是否正常启动
|
|||
|
|
if not server_b_started:
|
|||
|
|
print("[B] 警告: 服务端B没有正常启动，可能是使用了本机不存在的IP地址")
|
|||
|
|
data = s.recv(4096)
|
|||
|
|
if not data:
|
|||
|
|
print("[A] 连接关闭")
|
|||
|
|
break
|
|||
|
|
print(f"[A] 收到数据 {len(data)} 字节，转发给B的客户端")
|
|||
|
|
# 转发给所有已连接B的客户端
|
|||
|
|
for c in clients[:]:
|
|||
|
|
try:
|
|||
|
|
c.sendall(data)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[B] 转发失败: {e}")
|
|||
|
|
clients.remove(c)
|
|||
|
|
c.close()
|
|||
|
|
s.close() # 正常关闭连接
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[A] 连接失败: {e}")
|
|||
|
|
print("[A] 10秒后尝试重新连接...")
|
|||
|
|
time.sleep(10) # 等待10秒后重新连接
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 服务端B启动状态标志
|
|||
|
|
server_b_started = False
|
|||
|
|
# 启动服务端B
|
|||
|
|
threading.Thread(target=start_server_b, daemon=True).start()
|
|||
|
|
# 启动客户端去连接服务端A
|
|||
|
|
connect_server_a()
|