using System;
using System.Collections.Generic;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
class P2PCoordinator
{
private const int ServerPort = 11000;
private TcpListener _tcpListener;
private readonly Dictionary<string, IPEndPoint> _tcpEndpoints = new();
private readonly Dictionary<string, TcpClient> _tcpConnections = new();
private readonly object _lock = new object();
private bool _isRunning = true;
public P2PCoordinator()
{
try
{
_tcpListener = new TcpListener(IPAddress.Any, ServerPort);
Console.WriteLine($"P2P協(xié)調(diào)服務器啟動，監(jiān)聽端口: {ServerPort}");
}
catch (Exception ex)
{
Console.WriteLine($"服務器初始化失敗: {ex.Message}");
Environment.Exit(1);
}
}
public void Start()
{
new Thread(ListenTcp).Start();
Console.WriteLine("服務器運行中. 按任意鍵停止...");
Console.ReadKey();
Stop();
}
private void ListenTcp()
{
try
{
_tcpListener.Start();
Console.WriteLine("TCP監(jiān)聽已啟動");
while (_isRunning)
{
TcpClient client = _tcpListener.AcceptTcpClient();
new Thread(() => HandleTcpClient(client)).Start();
}
}
catch (Exception ex)
{
if (_isRunning) Console.WriteLine($"TCP監(jiān)聽錯誤: {ex.Message}");
}
}
private void HandleTcpClient(TcpClient client)
{
string clientId = null;
try
{
NetworkStream stream = client.GetStream();
byte[] buffer = new byte[1024];
int bytesRead = stream.Read(buffer, 0, buffer.Length);
if (bytesRead == 0) return;
string message = Encoding.ASCII.GetString(buffer, 0, bytesRead);
string[] parts = message.Split(':');
if (parts.Length < 2) return;
string command = parts[0];
clientId = parts[1];
var clientEP = (IPEndPoint)client.Client.RemoteEndPoint;
Console.WriteLine($"[TCP] {clientId} 連接: {clientEP}");
lock (_lock)
{
if (command == "REGISTER")
{
_tcpEndpoints[clientId] = clientEP;
_tcpConnections[clientId] = client;
Console.WriteLine($"[注冊] {clientId} -> {clientEP}");
SendTcpResponse(client, "REGISTERED");
}
else if (command == "CONNECT" && parts.Length > 2)
{
string targetId = parts[2];
Console.WriteLine($"[連接請求] {clientId} -> {targetId}");
HandleConnectionRequest(clientId, targetId);
}
}
// 處理后續(xù)消息
while (_isRunning)
{
bytesRead = stream.Read(buffer, 0, buffer.Length);
if (bytesRead == 0) break;
message = Encoding.ASCII.GetString(buffer, 0, bytesRead);
parts = message.Split(':');
command = parts[0];
if (command == "HEARTBEAT")
{
// 心跳檢測
SendTcpResponse(client, "HEARTBEAT_ACK");
}
}
}
catch (Exception ex)
{
Console.WriteLine($"處理TCP客戶端錯誤: {ex.Message}");
}
finally
{
if (clientId != null)
{
lock (_lock)
{
_tcpConnections.Remove(clientId);
_tcpEndpoints.Remove(clientId);
}
}
client.Close();
}
}
private void HandleConnectionRequest(string fromClient, string toClient)
{
lock (_lock)
{
if (!_tcpConnections.ContainsKey(fromClient))
{
Console.WriteLine($"[錯誤] 源客戶端未連接: {fromClient}");
return;
}
if (!_tcpConnections.ContainsKey(toClient))
{
SendTcpResponse(_tcpConnections[fromClient], "ERROR:目標客戶端未連接");
Console.WriteLine($"[錯誤] 目標客戶端未連接: {toClient}");
return;
}
if (!_tcpEndpoints.TryGetValue(fromClient, out IPEndPoint fromEp) ||
!_tcpEndpoints.TryGetValue(toClient, out IPEndPoint toEp))
{
SendTcpResponse(_tcpConnections[fromClient], "ERROR:端點信息不完整");
Console.WriteLine($"[錯誤] 端點信息不完整: {fromClient} -> {toClient}");
return;
}
// 交換端點信息
string fromMessage = $"PEER_INFO:{toClient}:{toEp.Address}:{toEp.Port}";
string toMessage = $"PEER_INFO:{fromClient}:{fromEp.Address}:{fromEp.Port}";
SendTcpResponse(_tcpConnections[fromClient], fromMessage);
SendTcpResponse(_tcpConnections[toClient], toMessage);
Console.WriteLine($"[端點交換] {fromClient} <-> {toClient}");
Console.WriteLine($" {fromEp} <-> {toEp}");
}
}
private void SendTcpResponse(TcpClient client, string message)
{
try
{
if (client.Connected)
{
NetworkStream stream = client.GetStream();
byte[] data = Encoding.ASCII.GetBytes(message);
stream.Write(data, 0, data.Length);
}
}
catch (Exception ex)
{
Console.WriteLine($"發(fā)送TCP響應錯誤: {ex.Message}");
}
}
private void Stop()
{
Console.WriteLine("停止服務器...");
_isRunning = false;
try
{
lock (_lock)
{
foreach (var client in _tcpConnections.Values)
{
try { client.Close(); } catch { }
}
_tcpConnections.Clear();
_tcpEndpoints.Clear();
}
_tcpListener.Stop();
Console.WriteLine("服務器已停止");
}
catch (Exception ex)
{
Console.WriteLine($"停止服務器錯誤: {ex.Message}");
}
}
static void Main(string[] args)
{
Console.Title = "P2P協(xié)調(diào)服務器";
Console.WriteLine("=== P2P協(xié)調(diào)服務器 ===");
Console.WriteLine("用于遠程桌面直連");
Console.WriteLine("端口: " + ServerPort);
Console.WriteLine(new string('=', 50));
var server = new P2PCoordinator();
server.Start();
}
}
400 186 1886
