much more robust server

This commit is contained in:
Chris Forbes
2009-11-18 21:57:07 +13:00
parent d6d1846ee3
commit e427e6b16e

View File

@@ -4,45 +4,72 @@ using System.Linq;
using System.Text; using System.Text;
using System.Net.Sockets; using System.Net.Sockets;
using System.Threading; using System.Threading;
using System.IO;
using System.Net;
namespace OpenRA.Server namespace OpenRA.Server
{ {
class Program static class Program
{ {
static void Main(string[] args) static void Main(string[] args)
{ {
var left = new TcpListener(1234); var listener = new TcpListener(IPAddress.Any, 1234);
left.Start(); var clients = new List<TcpClient>();
var right = new TcpListener(1235);
right.Start();
var lc = left.AcceptTcpClient(); listener.Start();
lc.NoDelay = true;
var l = lc.GetStream();
var rc = right.AcceptTcpClient(); for (; ; )
rc.NoDelay = true; {
var r = rc.GetStream(); try
{
var conn = listener.AcceptTcpClient();
conn.NoDelay = true;
Console.WriteLine("Accepted connection from {0}",
conn.Client.RemoteEndPoint.ToString());
var ll = new Thread(RW(l, r)); new Thread(() =>
var rr = new Thread(RW(r, l)); {
ll.Start(); lock (clients)
rr.Start(); clients.Add(conn);
ll.Join(); var ns = conn.GetStream();
try
{
for (; ; )
{
var frame = BitConverter.ToInt32(ns.Read(4), 0);
var length = BitConverter.ToInt32(ns.Read(4), 0);
var data = ns.Read(length);
lock (clients)
foreach (var c in clients)
if (c != conn)
{
var otherStream = c.GetStream();
otherStream.Write(BitConverter.GetBytes(frame));
otherStream.Write(BitConverter.GetBytes(length));
otherStream.Write(data);
}
}
}
catch (Exception e)
{
Console.WriteLine("Client dropped: {0}", conn.Client.RemoteEndPoint.ToString());
lock (clients)
clients.Remove(conn);
}
}) { IsBackground = true }.Start();
}
catch (Exception e)
{
Console.WriteLine(e.ToString());
}
}
} }
static ThreadStart RW(NetworkStream a, NetworkStream b) public static void Write(this Stream s, byte[] data) { s.Write(data, 0, data.Length); }
{ public static byte[] Read(this Stream s, int len) { var data = new byte[len]; s.Read(data, 0, len); return data; }
return () =>
{
var buf = new byte[4096];
while (true)
{
var len = a.Read(buf, 0, 4096);
b.Write(buf, 0, len);
}
};
}
} }
} }