internal static void ReplyPong(NetMessage pingMessage, NetConnection connection)
{
byte nr = pingMessage.ReadByte(6);
NetMessage pong = new NetMessage(NetMessageType.PingPong, 3);
pong.Write(true); // means pong
pong.Write(false); // means NOT optimize info
pong.Write(nr, 6);
pong.WriteSendStamp();
connection.Parent.SendSingleMessageAtOnce(pong, connection, connection.RemoteEndpoint);
}
internal static int SendConnect(NetClient client, IPEndPoint remoteEndpoint, byte[] customData)
{
if (client.Configuration.UsesEncryption)
{
client.Log.Debug("Sending Connect containing key: " + Convert.ToBase64String(client.ServerConnection.m_encryption.SymmetricEncryptionKeyBytes));
}
else
{
client.Log.Debug("Sending Connect - Unencrypted connection!");
}
NetMessage msg = new NetMessage(NetMessageType.Handshake, 3);
msg.Write((byte)NetHandshakeType.Connect);
msg.WriteSendStamp();
// encrypt symmetric key using server public key
if (client.Configuration.UsesEncryption)
{
byte[] encryptedKey = client.ServerConnection.m_encryption.EncryptRSA(client.ServerConnection.m_encryption.SymmetricEncryptionKeyBytes);
msg.Write((ushort)encryptedKey.Length);
msg.Write(encryptedKey);
}
else
{
// request no encryption
msg.Write((ushort)0);
}
if (customData == null || customData.Length < 1)
{
msg.Write7BitEncodedUInt(0);
}
else
{
msg.Write7BitEncodedUInt((uint)customData.Length);
msg.Write(customData);
}
return(client.SendSingleMessageAtOnce(msg, null, remoteEndpoint));
}
internal static void ReplyPong(NetMessage pingMessage, NetConnection connection)
{
byte nr = pingMessage.ReadByte(6);
NetMessage pong = new NetMessage(NetMessageType.PingPong, 3);
pong.Write(true); // means pong
pong.Write(false); // means NOT optimize info
pong.Write(nr, 6);
pong.WriteSendStamp();
connection.Parent.SendSingleMessageAtOnce(pong, connection, connection.RemoteEndpoint);
}
internal override void HandlePacket(NetBuffer buffer, int bytesReceived, IPEndPoint sender)
{
double now = NetTime.Now;
try
{
NetMessage msg;
if (m_serverConnection == null || m_serverConnection.Status == NetConnectionStatus.Disconnected)
{
//
// unconnected packet
//
msg = NetMessage.Decode(buffer);
if (msg == null)
{
Log.Warning("Malformed NetMessage?");
return; // malformed?
}
Log.Verbose("Received unconnected message: " + msg);
// discovery response?
if (msg.m_type == NetMessageType.Discovery)
{
NetServerInfo info = NetDiscovery.DecodeResponse(msg, sender);
if (ServerDiscovered != null)
{
ServerDiscovered(this, new NetServerDiscoveredEventArgs(info));
}
return;
}
if (m_serverConnection != null && sender.Equals(m_serverConnection.RemoteEndpoint))
{
// we have m_serverConnection, but status is disconnected
Log.Warning("Received " + buffer.LengthBytes + " from server; but we're disconnected!");
return;
}
Log.Warning("Received " + buffer.LengthBytes + " bytes from non-server source: " + sender + "(server is " + m_serverConnection.RemoteEndpoint + ")");
return;
}
// decrypt in place
if (m_serverConnection.m_encryption.SymmetricEncryptionKeyBytes != null)
{
byte[] savedBuffer = null;
if (m_serverConnection.Status == NetConnectionStatus.Connecting)
{
// special case; save buffer in case we get unencrypted response
savedBuffer = new byte[buffer.LengthBytes];
Array.Copy(buffer.Data, 0, savedBuffer, 0, buffer.LengthBytes);
}
bool ok = m_serverConnection.m_encryption.DecryptSymmetric(buffer);
if (!ok)
{
// failed to decrypt; drop this packet UNLESS we're in a connecting state,
// in which case the server may want to tell us something
if (m_serverConnection.Status == NetConnectionStatus.Connecting)
{
// ok let this one thru unencrypted
Array.Copy(savedBuffer, buffer.Data, savedBuffer.Length);
}
else
{
Log.Warning("Failed to decrypt packet from server!");
return;
}
}
}
m_serverConnection.m_lastHeardFromRemote = now;
int messagesReceived = 0;
int usrMessagesReceived = 0;
int ackMessagesReceived = 0;
while (buffer.ReadBitsLeft > 7)
{
msg = NetMessage.Decode(buffer);
if (msg == null)
{
break; // done
}
messagesReceived++;
msg.Sender = m_serverConnection;
switch (msg.m_type)
{
case NetMessageType.Handshake:
NetHandshakeType tp = (NetHandshakeType)msg.ReadByte();
if (tp == NetHandshakeType.Connect || tp == NetHandshakeType.ConnectionEstablished)
{
Log.Warning("Client received " + tp + "?!");
}
else if (tp == NetHandshakeType.ConnectResponse)
{
if (m_serverConnection.Status != NetConnectionStatus.Connecting)
{
Log.Verbose("Received redundant ConnectResponse!");
break;
}
// Log.Debug("ConnectResponse received");
m_serverConnection.SetStatus(NetConnectionStatus.Connected, "Connected");
Log.Info("Connected to " + m_serverConnection.RemoteEndpoint);
// initialize ping to now - m_firstSentConnect
float initRoundtrip = (float)(now - m_serverConnection.m_firstSentHandshake);
m_serverConnection.m_ping.Initialize(initRoundtrip);
ushort remoteValue = msg.ReadUInt16();
m_serverConnection.RemoteClockOffset = NetTime.CalculateOffset(now, remoteValue, initRoundtrip);
Log.Verbose("Initializing remote clock offset to " + m_serverConnection.RemoteClockOffset + " ms (roundtrip " + NetUtil.SecToMil(initRoundtrip) + " ms)");
NetMessage established = new NetMessage(NetMessageType.Handshake, 3);
established.Write((byte)NetHandshakeType.ConnectionEstablished);
established.WriteSendStamp();
SendSingleMessageAtOnce(established, m_serverConnection, m_serverConnection.RemoteEndpoint);
}
else
{ // Disconnected
string reason = msg.ReadString();
m_serverConnection.SetStatus(NetConnectionStatus.Disconnected, reason);
}
break;
case NetMessageType.Acknowledge:
//Log.Debug("Received ack " + msg.SequenceChannel + "|" + msg.SequenceNumber);
m_serverConnection.ReceiveAcknowledge(msg);
ackMessagesReceived++;
break;
case NetMessageType.PingPong:
bool isPong = msg.ReadBoolean();
bool isOptimizeInfo = msg.ReadBoolean();
if (isOptimizeInfo)
{
m_serverConnection.m_ping.HandleOptimizeInfo(now, msg);
}
else if (isPong)
{
if (m_serverConnection.Status == NetConnectionStatus.Connected)
{
m_serverConnection.m_ping.HandlePong(now, msg);
}
}
else
{
NetPing.ReplyPong(msg, m_serverConnection);
}
break;
case NetMessageType.User:
case NetMessageType.UserFragmented:
//Log.Debug("User message received; " + msg.m_buffer.LengthBytes + " bytes");
m_serverConnection.ReceiveMessage(now, msg);
usrMessagesReceived++;
break;
case NetMessageType.Discovery:
NetServerInfo info = NetDiscovery.DecodeResponse(msg, sender);
if (ServerDiscovered != null)
{
ServerDiscovered(this, new NetServerDiscoveredEventArgs(info));
}
break;
default:
Log.Warning("Client received " + msg.m_type + "?!");
break;
}
}
// add statistics
NetStatistics stats = m_serverConnection.Statistics;
stats.PacketsReceived++;
stats.MessagesReceived += messagesReceived;
stats.UserMessagesReceived += usrMessagesReceived;
stats.AckMessagesReceived += ackMessagesReceived;
stats.BytesReceived += bytesReceived;
}
catch (Exception ex)
{
Log.Error("Failed to parse packet correctly; read/write mismatch? " + ex);
}
}
internal static int SendConnect(NetClient client, IPEndPoint remoteEndpoint, byte[] customData)
{
if (client.Configuration.UsesEncryption)
client.Log.Debug("Sending Connect containing key: " + Convert.ToBase64String(client.ServerConnection.m_encryption.SymmetricEncryptionKeyBytes));
else
client.Log.Debug("Sending Connect - Unencrypted connection!");
NetMessage msg = new NetMessage(NetMessageType.Handshake, 3);
msg.Write((byte)NetHandshakeType.Connect);
msg.WriteSendStamp();
// encrypt symmetric key using server public key
if (client.Configuration.UsesEncryption)
{
byte[] encryptedKey = client.ServerConnection.m_encryption.EncryptRSA(client.ServerConnection.m_encryption.SymmetricEncryptionKeyBytes);
msg.Write((ushort)encryptedKey.Length);
msg.Write(encryptedKey);
}
else
{
// request no encryption
msg.Write((ushort)0);
}
if (customData == null || customData.Length < 1)
{
msg.Write7BitEncodedUInt(0);
}
else
{
msg.Write7BitEncodedUInt((uint)customData.Length);
msg.Write(customData);
}
return client.SendSingleMessageAtOnce(msg, null, remoteEndpoint);
}
internal override void HandlePacket(NetBuffer buffer, int bytesReceived, IPEndPoint sender)
{
double now = NetTime.Now;
try
{
NetMessage msg;
if (m_serverConnection == null || m_serverConnection.Status == NetConnectionStatus.Disconnected)
{
//
// unconnected packet
//
msg = NetMessage.Decode(buffer);
if (msg == null)
{
Log.Warning("Malformed NetMessage?");
return; // malformed?
}
Log.Verbose("Received unconnected message: " + msg);
// discovery response?
if (msg.m_type == NetMessageType.Discovery)
{
NetServerInfo info = NetDiscovery.DecodeResponse(msg, sender);
if (ServerDiscovered != null)
ServerDiscovered(this, new NetServerDiscoveredEventArgs(info));
return;
}
if (m_serverConnection != null && sender.Equals(m_serverConnection.RemoteEndpoint))
{
// we have m_serverConnection, but status is disconnected
Log.Warning("Received " + buffer.LengthBytes + " from server; but we're disconnected!");
return;
}
Log.Warning("Received " + buffer.LengthBytes + " bytes from non-server source: " + sender + "(server is " + m_serverConnection.RemoteEndpoint + ")");
return;
}
// decrypt in place
if (m_serverConnection.m_encryption.SymmetricEncryptionKeyBytes != null)
{
byte[] savedBuffer = null;
if (m_serverConnection.Status == NetConnectionStatus.Connecting)
{
// special case; save buffer in case we get unencrypted response
savedBuffer = new byte[buffer.LengthBytes];
Array.Copy(buffer.Data, 0, savedBuffer, 0, buffer.LengthBytes);
}
bool ok = m_serverConnection.m_encryption.DecryptSymmetric(buffer);
if (!ok)
{
// failed to decrypt; drop this packet UNLESS we're in a connecting state,
// in which case the server may want to tell us something
if (m_serverConnection.Status == NetConnectionStatus.Connecting)
{
// ok let this one thru unencrypted
Array.Copy(savedBuffer, buffer.Data, savedBuffer.Length);
}
else
{
Log.Warning("Failed to decrypt packet from server!");
return;
}
}
}
m_serverConnection.m_lastHeardFromRemote = now;
int messagesReceived = 0;
int usrMessagesReceived = 0;
int ackMessagesReceived = 0;
while (buffer.ReadBitsLeft > 7)
{
msg = NetMessage.Decode(buffer);
if (msg == null)
break; // done
messagesReceived++;
msg.Sender = m_serverConnection;
switch (msg.m_type)
{
case NetMessageType.Handshake:
NetHandshakeType tp = (NetHandshakeType)msg.ReadByte();
if (tp == NetHandshakeType.Connect || tp == NetHandshakeType.ConnectionEstablished)
{
Log.Warning("Client received " + tp + "?!");
}
else if (tp == NetHandshakeType.ConnectResponse)
{
if (m_serverConnection.Status != NetConnectionStatus.Connecting)
{
Log.Verbose("Received redundant ConnectResponse!");
break;
}
// Log.Debug("ConnectResponse received");
m_serverConnection.SetStatus(NetConnectionStatus.Connected, "Connected");
Log.Info("Connected to " + m_serverConnection.RemoteEndpoint);
// initialize ping to now - m_firstSentConnect
float initRoundtrip = (float)(now - m_serverConnection.m_firstSentHandshake);
m_serverConnection.m_ping.Initialize(initRoundtrip);
ushort remoteValue = msg.ReadUInt16();
m_serverConnection.RemoteClockOffset = NetTime.CalculateOffset(now, remoteValue, initRoundtrip);
Log.Verbose("Initializing remote clock offset to " + m_serverConnection.RemoteClockOffset + " ms (roundtrip " + NetUtil.SecToMil(initRoundtrip) + " ms)");
NetMessage established = new NetMessage(NetMessageType.Handshake, 3);
established.Write((byte)NetHandshakeType.ConnectionEstablished);
established.WriteSendStamp();
SendSingleMessageAtOnce(established, m_serverConnection, m_serverConnection.RemoteEndpoint);
}
else
{ // Disconnected
string reason = msg.ReadString();
m_serverConnection.SetStatus(NetConnectionStatus.Disconnected, reason);
}
break;
case NetMessageType.Acknowledge:
//Log.Debug("Received ack " + msg.SequenceChannel + "|" + msg.SequenceNumber);
m_serverConnection.ReceiveAcknowledge(msg);
ackMessagesReceived++;
break;
case NetMessageType.PingPong:
bool isPong = msg.ReadBoolean();
bool isOptimizeInfo = msg.ReadBoolean();
if (isOptimizeInfo)
{
m_serverConnection.m_ping.HandleOptimizeInfo(now, msg);
} else if (isPong)
{
if (m_serverConnection.Status == NetConnectionStatus.Connected)
m_serverConnection.m_ping.HandlePong(now, msg);
}
else
{
NetPing.ReplyPong(msg, m_serverConnection);
}
break;
case NetMessageType.User:
case NetMessageType.UserFragmented:
//Log.Debug("User message received; " + msg.m_buffer.LengthBytes + " bytes");
m_serverConnection.ReceiveMessage(now, msg);
usrMessagesReceived++;
break;
case NetMessageType.Discovery:
NetServerInfo info = NetDiscovery.DecodeResponse(msg, sender);
if (ServerDiscovered != null)
ServerDiscovered(this, new NetServerDiscoveredEventArgs(info));
break;
default:
Log.Warning("Client received " + msg.m_type + "?!");
break;
}
}
// add statistics
NetStatistics stats = m_serverConnection.Statistics;
stats.PacketsReceived++;
stats.MessagesReceived += messagesReceived;
stats.UserMessagesReceived += usrMessagesReceived;
stats.AckMessagesReceived += ackMessagesReceived;
stats.BytesReceived += bytesReceived;
}
catch (Exception ex)
{
Log.Error("Failed to parse packet correctly; read/write mismatch? " + ex);
}
}