136 lines
3.1 KiB
C#
136 lines
3.1 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2011 The OpenRA Developers (see AUTHORS)
|
|
* This file is part of OpenRA, which is free software. It is made
|
|
* available to you under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation. For more information,
|
|
* see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net.Sockets;
|
|
using System.Net.NetworkInformation;
|
|
using System.Threading;
|
|
|
|
namespace OpenRA.Server
|
|
{
|
|
public class Connection
|
|
{
|
|
public Socket socket;
|
|
public List<byte> data = new List<byte>();
|
|
public ReceiveState State = ReceiveState.Header;
|
|
public int ExpectLength = 8;
|
|
public int Frame = 0;
|
|
public int MostRecentFrame = 0;
|
|
public string RemoteAddress;
|
|
public int Latency = -1;
|
|
|
|
/* client data */
|
|
public int PlayerIndex;
|
|
|
|
public byte[] PopBytes(int n)
|
|
{
|
|
var result = data.GetRange(0, n);
|
|
data.RemoveRange(0, n);
|
|
return result.ToArray();
|
|
}
|
|
|
|
bool ReadDataInner( Server server )
|
|
{
|
|
var rx = new byte[1024];
|
|
var len = 0;
|
|
|
|
for (; ; )
|
|
{
|
|
try
|
|
{
|
|
// NOTE(jsd): Poll the socket first to see if there's anything there.
|
|
// This avoids the exception with SocketErrorCode == `SocketError.WouldBlock` thrown
|
|
// from `socket.Receive(rx)`.
|
|
if (!socket.Poll(0, SelectMode.SelectRead)) break;
|
|
|
|
if (0 < (len = socket.Receive(rx)))
|
|
data.AddRange(rx.Take(len));
|
|
else
|
|
{
|
|
if (len == 0)
|
|
server.DropClient(this);
|
|
break;
|
|
}
|
|
}
|
|
catch (SocketException e)
|
|
{
|
|
// NOTE(jsd): This should no longer be needed with the socket.Poll call above.
|
|
if (e.SocketErrorCode == SocketError.WouldBlock) break;
|
|
|
|
server.DropClient(this);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public void ReadData( Server server )
|
|
{
|
|
if (ReadDataInner(server))
|
|
while (data.Count >= ExpectLength)
|
|
{
|
|
var bytes = PopBytes(ExpectLength);
|
|
switch (State)
|
|
{
|
|
case ReceiveState.Header:
|
|
{
|
|
ExpectLength = BitConverter.ToInt32(bytes, 0) - 4;
|
|
Frame = BitConverter.ToInt32(bytes, 4);
|
|
State = ReceiveState.Data;
|
|
} break;
|
|
|
|
case ReceiveState.Data:
|
|
{
|
|
server.DispatchOrders(this, Frame, bytes);
|
|
MostRecentFrame = Frame;
|
|
ExpectLength = 8;
|
|
State = ReceiveState.Header;
|
|
|
|
server.UpdateInFlightFrames(this);
|
|
} break;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool hasBeenPinged;
|
|
public void Ping()
|
|
{
|
|
if (!hasBeenPinged)
|
|
{
|
|
hasBeenPinged = true;
|
|
var pingSender = new Ping();
|
|
pingSender.PingCompleted += new PingCompletedEventHandler(pongRecieved);
|
|
AutoResetEvent waiter = new AutoResetEvent(false);
|
|
pingSender.SendAsync(RemoteAddress, waiter);
|
|
}
|
|
}
|
|
|
|
void pongRecieved(object sender, PingCompletedEventArgs e)
|
|
{
|
|
if (e.Cancelled || e.Error != null)
|
|
Latency = -1;
|
|
else
|
|
{
|
|
PingReply pong = e.Reply;
|
|
if (pong != null && pong.Status == IPStatus.Success)
|
|
Latency = (int)pong.RoundtripTime;
|
|
else
|
|
Latency = -1;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
public enum ReceiveState { Header, Data };
|
|
}
|