import socket
import threading
import time

class TcpServer(object):
    def __init__(self):
# 配置信息
        self.tcpPort = 12345         # 服务端TCP监听端口
        self.udpPort = 54321  # 设备发现UDP端口
        self.broadcastAddr = '<broadcast>'  # 广播地址

# TCP服务端主逻辑
    def tcpServer(self):
        server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        server.bind(('0.0.0.0', self.tcpPort))
        server.listen(5)
        print(f"[*] TCP服务端已启动，监听端口 {self.tcpPort}")

        while True:
            clientSock, addr = server.accept()
            print(f"[+] 接收到来自 {addr} 的连接")
            # 创建线程处理客户端请求
            client_thread = threading.Thread(target=self.handleClient, args=(clientSock,))
            client_thread.start()

    # 处理客户端请求
    def handleClient(self, clientSock):
        try:
            while True:
                data = clientSock.recv(1024)
                if not data:
                    break
                print(f"收到消息: {data.decode('utf-8')}")
                clientSock.send(b"ACK")  # 返回确认
        except Exception as e:
            print(f"客户端异常断开: {e}")
        finally:
            clientSock.close()

    # UDP设备发现服务
    def udpDiscoveryServer(self):
        udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        udpSock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        udpSock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
        udpSock.bind(('', self.udpPort))
        print(f"[*] UDP发现服务已启动，监听端口 {self.udpPort}")

        while True:
            data, addr = udpSock.recvfrom(1024)
            if data.decode('utf-8') == "DISCOVERY_REQUEST":
                print(f"[+] 收到来自 {addr} 的发现请求")
                response = f"DISCOVERY_RESPONSE:{socket.gethostname()}:{self.tcpPort}"
                udpSock.sendto(response.encode('utf-8'), addr)

if __name__ == '__main__':
    server = TcpServer()
    # 启动TCP服务端和UDP发现服务
    threading.Thread(target=server.tcpServer, daemon=True).start()
    threading.Thread(target=server.udpDiscoveryServer, daemon=True).start()
    # 防止主线程退出
    while True:
        time.sleep(1)