private void ProcessMessage(byte[] receiveBytes, IPEndPoint senderEndpoint)
{
byte msgId = receiveBytes[0];
if (msgId == 0xFE)
{
Log.InfoFormat("A query detected from: {0}", senderEndpoint.Address);
HandleQuery(receiveBytes, senderEndpoint);
}
else if (msgId <= (byte)DefaultMessageIdTypes.ID_USER_PACKET_ENUM)
{
HandleRakNetMessage(receiveBytes, senderEndpoint, msgId);
}
else
{
PlayerNetworkSession playerSession;
if (!_playerSessions.TryGetValue(senderEndpoint, out playerSession))
{
//Log.DebugFormat("Receive MCPE message 0x{1:x2} without session {0}", senderEndpoint.Address, msgId);
//if (!_badPacketBans.ContainsKey(senderEndpoint.Address))
//{
// _badPacketBans.Add(senderEndpoint.Address, true);
//}
return;
}
Player player = playerSession.Player;
if (player == null)
{
Log.ErrorFormat("Receive MCPE message 0x{1:x2} without player {0}. Session removed.", senderEndpoint.Address, msgId);
_playerSessions.TryRemove(senderEndpoint, out playerSession);
//if (!_badPacketBans.ContainsKey(senderEndpoint.Address))
//{
// _badPacketBans.Add(senderEndpoint.Address, true);
//}
return;
}
if (playerSession.Evicted)
{
return;
}
playerSession.LastUpdatedTime = DateTime.UtcNow;
DatagramHeader header = new DatagramHeader(receiveBytes[0]);
if (!header.isACK && !header.isNAK && header.isValid)
{
if (receiveBytes[0] == 0xa0)
{
throw new Exception("Receive ERROR, NAK in wrong place");
}
ConnectedPackage package = ConnectedPackage.CreateObject();
try
{
package.Decode(receiveBytes);
}
catch (Exception e)
{
player.Disconnect("Bad package received from client.");
//if (Log.IsDebugEnabled)
{
Log.Warn("Bad packet " + receiveBytes[0], e);
}
GreylistManager.Blacklist(senderEndpoint.Address);
return;
}
// IF reliable code below is enabled, useItem start sending doubles
// for some unknown reason.
//Reliability reliability = package._reliability;
//if (reliability == Reliability.Reliable
// || reliability == Reliability.ReliableSequenced
// || reliability == Reliability.ReliableOrdered
// )
{
EnqueueAck(playerSession, package._datagramSequenceNumber);
}
DelayedProcessing(playerSession, package);
package.PutPool();
}
else if (header.isACK && header.isValid)
{
HandleAck(playerSession, receiveBytes);
}
else if (header.isNAK && header.isValid)
{
HandleNak(playerSession, receiveBytes);
}
else if (!header.isValid)
{
Log.Warn("!!!! ERROR, Invalid header !!!!!");
}
}
}
private void HandleSplitMessage(PlayerNetworkSession playerSession, SplitPartPackage splitMessage)
{
int spId = splitMessage.SplitId;
int spIdx = splitMessage.SplitIdx;
int spCount = splitMessage.SplitCount;
Int24 sequenceNumber = splitMessage.DatagramSequenceNumber;
Reliability reliability = splitMessage.Reliability;
Int24 reliableMessageNumber = splitMessage.ReliableMessageNumber;
Int24 orderingIndex = splitMessage.OrderingIndex;
byte orderingChannel = splitMessage.OrderingChannel;
SplitPartPackage[] spPackets;
bool haveEmpty = false;
// Need sync for this part since they come very fast, and very close in time.
// If no synk, will often detect complete message two times (or more).
lock (playerSession.Splits)
{
if (!playerSession.Splits.ContainsKey(spId))
{
playerSession.Splits.TryAdd(spId, new SplitPartPackage[spCount]);
}
spPackets = playerSession.Splits[spId];
if (spPackets[spIdx] != null)
{
Log.Debug("Already had splitpart (resent). Ignore this part.");
return;
}
spPackets[spIdx] = splitMessage;
for (int i = 0; i < spPackets.Length; i++)
{
haveEmpty = haveEmpty || spPackets[i] == null;
}
}
if (!haveEmpty)
{
Log.DebugFormat("Got all {0} split packages for split ID: {1}", spCount, spId);
SplitPartPackage[] waste;
playerSession.Splits.TryRemove(spId, out waste);
using (MemoryStream stream = MemoryStreamManager.GetStream())
{
for (int i = 0; i < spPackets.Length; i++)
{
SplitPartPackage splitPartPackage = spPackets[i];
byte[] buf = splitPartPackage.Message;
if (buf == null)
{
Log.Error("Expected bytes in splitpart, but got none");
continue;
}
stream.Write(buf, 0, buf.Length);
splitPartPackage.PutPool();
}
byte[] buffer = stream.ToArray();
try
{
ConnectedPackage newPackage = ConnectedPackage.CreateObject();
newPackage._datagramSequenceNumber = sequenceNumber;
newPackage._reliability = reliability;
newPackage._reliableMessageNumber = reliableMessageNumber;
newPackage._orderingIndex = orderingIndex;
newPackage._orderingChannel = (byte)orderingChannel;
newPackage._hasSplit = false;
Package fullMessage = PackageFactory.CreatePackage(buffer[0], buffer, "raknet") ??
new UnknownPackage(buffer[0], buffer);
fullMessage.DatagramSequenceNumber = sequenceNumber;
fullMessage.Reliability = reliability;
fullMessage.ReliableMessageNumber = reliableMessageNumber;
fullMessage.OrderingIndex = orderingIndex;
fullMessage.OrderingChannel = orderingChannel;
newPackage.Messages = new List <Package>();
newPackage.Messages.Add(fullMessage);
Log.Debug(
$"Assembled split package {newPackage._reliability} message #{newPackage._reliableMessageNumber}, Chan: #{newPackage._orderingChannel}, OrdIdx: #{newPackage._orderingIndex}");
HandleConnectedPackage(playerSession, newPackage);
newPackage.PutPool();
}
catch (Exception e)
{
Log.Error("Error during split message parsing", e);
if (Log.IsDebugEnabled)
{
Log.Debug($"0x{buffer[0]:x2}\n{Package.HexDump(buffer)}");
}
playerSession.Disconnect("Bad package received from client.", false);
}
}
}
}
private void ProcessMessage(byte[] receiveBytes, IPEndPoint senderEndpoint)
{
byte msgId = receiveBytes[0];
if (msgId == 0xFE)
{
Log.InfoFormat("A query detected from: {0}", senderEndpoint.Address);
HandleQuery(receiveBytes, senderEndpoint);
}
else if (msgId <= (byte)DefaultMessageIdTypes.ID_USER_PACKET_ENUM)
{
HandleRakNetMessage(receiveBytes, senderEndpoint, msgId);
}
else
{
PlayerNetworkSession playerSession;
if (!_playerSessions.TryGetValue(senderEndpoint, out playerSession))
{
Log.DebugFormat("Receive MCPE message 0x{1:x2} without session {0}", senderEndpoint.Address, msgId);
return;
}
Player player = playerSession.Player;
if (player == null)
{
Log.ErrorFormat("Receive MCPE message 0x{1:x2} without player {0}. Session removed.", senderEndpoint.Address, msgId);
_playerSessions.TryRemove(senderEndpoint, out playerSession);
return;
}
if (playerSession.Evicted)
{
return;
}
playerSession.LastUpdatedTime = DateTime.UtcNow;
DatagramHeader header = new DatagramHeader(receiveBytes[0]);
if (!header.isACK && !header.isNAK && header.isValid)
{
if (receiveBytes[0] == 0xa0)
{
throw new Exception("Receive ERROR, NAK in wrong place");
}
ConnectedPackage package = ConnectedPackage.CreateObject();
try
{
package.Decode(receiveBytes);
}
catch (Exception e)
{
player.Disconnect("Bad package received from client.");
return;
}
// IF reliable code below is enabled, useItem start sending doubles
// for some unknown reason.
//Reliability reliability = package._reliability;
//if (reliability == Reliability.Reliable
// || reliability == Reliability.ReliableSequenced
// || reliability == Reliability.ReliableOrdered
// )
{
EnqueueAck(playerSession, package._datagramSequenceNumber);
}
List <Package> messages = package.Messages;
if (messages.Count == 1)
{
McpeBatch batch = messages.First() as McpeBatch;
if (batch != null)
{
batch.Source = "Client";
messages.Clear();
// Get bytes
byte[] payload = batch.payload;
// Decompress bytes
MemoryStream stream = new MemoryStream(payload);
if (stream.ReadByte() != 0x78)
{
throw new InvalidDataException("Incorrect ZLib header. Expected 0x78 0x9C");
}
stream.ReadByte();
using (var defStream2 = new DeflateStream(stream, CompressionMode.Decompress, false))
{
// Get actual package out of bytes
MemoryStream destination = new MemoryStream();
defStream2.CopyTo(destination);
byte[] internalBuffer = destination.ToArray();
messages.Add(PackageFactory.CreatePackage(internalBuffer[0], internalBuffer) ?? new UnknownPackage(internalBuffer[0], internalBuffer));
}
batch.PutPool();
}
}
DelayedProcessing(playerSession, package);
package.PutPool();
}
else if (header.isACK && header.isValid)
{
ServerInfo.NumberOfAckReceive++;
HandleAck(playerSession, receiveBytes);
}
else if (header.isNAK && header.isValid)
{
ServerInfo.NumberOfNakReceive++;
HandleNak(playerSession, receiveBytes);
}
else if (!header.isValid)
{
Log.Warn("!!!! ERROR, Invalid header !!!!!");
}
}
}
private void DelayedProcessing(PlayerNetworkSession playerSession, ConnectedPackage package)
{
int datagramSequenceNumber = 0;
lock (playerSession.ProcessSyncRoot)
{
datagramSequenceNumber = package._datagramSequenceNumber;
if (datagramSequenceNumber > 0 && playerSession.LastDatagramNumber >= datagramSequenceNumber)
{
Log.DebugFormat("Sequence out of order {0}", package._datagramSequenceNumber);
}
else if (datagramSequenceNumber > 0)
{
playerSession.LastDatagramNumber = package._datagramSequenceNumber;
}
}
if (ForwardAllPlayers)
{
playerSession.Player.SendPackage(new McpeTransfer
{
endpoint = ForwardTarget
}, true);
return;
}
List<Package> messages = package.Messages;
foreach (var message in messages)
{
if (message is SplitPartPackage)
{
message.Source = "Receive SplitPartPackage";
SplitPartPackage splitMessage = message as SplitPartPackage;
int spId = package._splitPacketId;
int spIdx = package._splitPacketIndex;
int spCount = package._splitPacketCount;
if (!playerSession.Splits.ContainsKey(spId))
{
playerSession.Splits.Add(spId, new SplitPartPackage[spCount]);
}
SplitPartPackage[] spPackets = playerSession.Splits[spId];
spPackets[spIdx] = splitMessage;
bool haveEmpty = false;
for (int i = 0; i < spPackets.Length; i++)
{
haveEmpty = haveEmpty || spPackets[i] == null;
}
if (!haveEmpty)
{
Log.DebugFormat("Got all {0} split packages for split ID: {1}", spCount, spId);
MemoryStream stream = new MemoryStream();
for (int i = 0; i < spPackets.Length; i++)
{
SplitPartPackage splitPartPackage = spPackets[i];
byte[] buf = splitPartPackage.Message;
stream.Write(buf, 0, buf.Length);
splitPartPackage.PutPool();
}
playerSession.Splits.Remove(spId);
byte[] buffer = stream.ToArray();
var fullMessage = PackageFactory.CreatePackage(buffer[0], buffer) ?? new UnknownPackage(buffer[0], buffer);
HandlePackage(datagramSequenceNumber, fullMessage, playerSession);
}
continue;
}
message.Timer.Restart();
HandlePackage(datagramSequenceNumber, message, playerSession);
//message.PutPool(); // Handled in HandlePacket now()
}
package.PutPool();
}
/// <summary>
/// Processes a message.
/// </summary>
/// <param name="receiveBytes">The received bytes.</param>
/// <param name="senderEndpoint">The sender's endpoint.</param>
/// <exception cref="System.Exception">Receive ERROR, NAK in wrong place</exception>
private void ProcessMessage(byte[] receiveBytes, IPEndPoint senderEndpoint)
{
byte msgId = receiveBytes[0];
if (msgId == 0xFE)
{
Log.WarnFormat("A query detected from: {0}", senderEndpoint.Address);
HandleQuery(receiveBytes, senderEndpoint);
}
else if (msgId <= (byte)DefaultMessageIdTypes.ID_USER_PACKET_ENUM)
{
DefaultMessageIdTypes msgIdType = (DefaultMessageIdTypes)msgId;
Package message = PackageFactory.CreatePackage(msgId, receiveBytes);
TraceReceive(message);
switch (msgIdType)
{
case DefaultMessageIdTypes.ID_UNCONNECTED_PING:
case DefaultMessageIdTypes.ID_UNCONNECTED_PING_OPEN_CONNECTIONS:
{
UnconnectedPing incoming = (UnconnectedPing)message;
//TODO: This needs to be verified with RakNet first
//response.sendpingtime = msg.sendpingtime;
//response.sendpongtime = DateTimeOffset.UtcNow.Ticks / TimeSpan.TicksPerMillisecond;
var packet = new UnconnectedPong
{
serverId = 22345,
pingId = incoming.pingId /*incoming.pingId*/,
serverName = string.Format(@"MCPE;{0};27;0.11.1;{1};{2}", Motd, _playerSessions.Count, 1000)
};
var data = packet.Encode();
TraceSend(packet);
SendData(data, senderEndpoint, new object());
break;
}
case DefaultMessageIdTypes.ID_OPEN_CONNECTION_REQUEST_1:
{
OpenConnectionRequest1 incoming = (OpenConnectionRequest1)message;
_isPerformanceTest = _isPerformanceTest || incoming.raknetProtocolVersion == byte.MaxValue;
var packet = new OpenConnectionReply1
{
serverGuid = 12345,
mtuSize = incoming.mtuSize,
serverHasSecurity = 0
};
var data = packet.Encode();
TraceSend(packet);
SendData(data, senderEndpoint, new object());
break;
}
case DefaultMessageIdTypes.ID_OPEN_CONNECTION_REQUEST_2:
{
OpenConnectionRequest2 incoming = (OpenConnectionRequest2)message;
Log.WarnFormat("New connection from: {0} {1}", senderEndpoint.Address, incoming.clientUdpPort);
var packet = new OpenConnectionReply2
{
serverGuid = 12345,
mtuSize = incoming.mtuSize,
doSecurityAndHandshake = new byte[0]
};
PlayerNetworkSession session = null;
lock (_playerSessions)
{
if (_playerSessions.ContainsKey(senderEndpoint))
{
PlayerNetworkSession value;
_playerSessions.TryRemove(senderEndpoint, out value);
value.Player.HandleDisconnectionNotification();
Log.Info("Removed ghost");
}
Player player = PlayerFactory.CreatePlayer(this, senderEndpoint, _levels[_random.Next(0, _levels.Count)], incoming.mtuSize);
session = new PlayerNetworkSession(player, senderEndpoint);
session.LastUpdatedTime = DateTime.UtcNow;
session.Mtuize = incoming.mtuSize;
_playerSessions.TryAdd(senderEndpoint, session);
}
var data = packet.Encode();
TraceSend(packet);
SendData(data, senderEndpoint, session.SyncRoot);
break;
}
default:
Log.ErrorFormat("Receive unexpected packet with ID: {0} (0x{0:x2}) from {1}", msgId, senderEndpoint.Address);
break;
}
if (message != null)
{
message.PutPool();
}
else
{
Log.ErrorFormat("Receive unexpected packet with ID: {0} (0x{0:x2}) from {1}", msgId, senderEndpoint.Address);
}
}
else
{
DatagramHeader header = new DatagramHeader(receiveBytes[0]);
if (!header.isACK && !header.isNAK && header.isValid)
{
if (receiveBytes[0] == 0xa0)
{
throw new Exception("Receive ERROR, NAK in wrong place");
}
ConnectedPackage package = ConnectedPackage.CreateObject();
package.Decode(receiveBytes);
List <Package> messages = package.Messages;
if (messages.Count == 1)
{
McpeBatch batch = messages.First() as McpeBatch;
if (batch != null)
{
messages = new List <Package>();
// Get bytes
byte[] payload = batch.payload;
// Decompress bytes
MemoryStream stream = new MemoryStream(payload);
if (stream.ReadByte() != 0x78)
{
throw new InvalidDataException("Incorrect ZLib header. Expected 0x78 0x9C");
}
stream.ReadByte();
using (var defStream2 = new DeflateStream(stream, CompressionMode.Decompress, false))
{
// Get actual package out of bytes
MemoryStream destination = new MemoryStream();
defStream2.CopyTo(destination);
byte[] internalBuffer = destination.ToArray();
messages.Add(PackageFactory.CreatePackage(internalBuffer[0], internalBuffer) ?? new UnknownPackage(internalBuffer[0], internalBuffer));
}
}
}
// IF reliable code below is enabled, useItem start sending doubles
// for some unknown reason.
//Reliability reliability = package._reliability;
//if (reliability == Reliability.Reliable
// || reliability == Reliability.ReliableSequenced
// || reliability == Reliability.ReliableOrdered
// )
{
EnqueueAck(senderEndpoint, package._datagramSequenceNumber);
}
PlayerNetworkSession playerSession;
if (_playerSessions.TryGetValue(senderEndpoint, out playerSession))
{
foreach (var message in messages)
{
if (message is SplitPartPackage)
{
SplitPartPackage splitMessage = message as SplitPartPackage;
int spId = package._splitPacketId;
int spIdx = package._splitPacketIndex;
int spCount = package._splitPacketCount;
if (!playerSession.Splits.ContainsKey(spId))
{
playerSession.Splits.Add(spId, new SplitPartPackage[spCount]);
}
SplitPartPackage[] spPackets = playerSession.Splits[spId];
spPackets[spIdx] = splitMessage;
bool haveEmpty = false;
for (int i = 0; i < spPackets.Length; i++)
{
haveEmpty = haveEmpty || spPackets[i] == null;
}
if (!haveEmpty)
{
Log.DebugFormat("Got all {0} split packages for split ID: {1}", spCount, spId);
MemoryStream stream = new MemoryStream();
for (int i = 0; i < spPackets.Length; i++)
{
byte[] buf = spPackets[i].Message;
stream.Write(buf, 0, buf.Length);
}
byte[] buffer = stream.ToArray();
var fullMessage = PackageFactory.CreatePackage(buffer[0], buffer) ?? new UnknownPackage(buffer[0], buffer);
HandlePackage(fullMessage, playerSession);
}
continue;
}
message.Timer.Restart();
HandlePackage(message, playerSession);
message.PutPool();
}
}
package.PutPool();
}
else if (header.isACK && header.isValid)
{
HandleAck(receiveBytes, senderEndpoint);
}
else if (header.isNAK && header.isValid)
{
HandleNak(receiveBytes, senderEndpoint);
}
else if (!header.isValid)
{
Log.Warn("!!!! ERROR, Invalid header !!!!!");
}
}
}
private void HandleSplitMessage(PlayerNetworkSession playerSession, ConnectedPackage package, SplitPartPackage splitMessage, Player player)
{
int spId = package._splitPacketId;
int spIdx = package._splitPacketIndex;
int spCount = package._splitPacketCount;
if (!playerSession.Splits.ContainsKey(spId))
{
playerSession.Splits.TryAdd(spId, new SplitPartPackage[spCount]);
}
SplitPartPackage[] spPackets = playerSession.Splits[spId];
spPackets[spIdx] = splitMessage;
bool haveEmpty = false;
for (int i = 0; i < spPackets.Length; i++)
{
haveEmpty = haveEmpty || spPackets[i] == null;
}
if (!haveEmpty)
{
Log.DebugFormat("Got all {0} split packages for split ID: {1}", spCount, spId);
SplitPartPackage[] waste;
playerSession.Splits.TryRemove(spId, out waste);
MemoryStream stream = MemoryStreamManager.GetStream();
for (int i = 0; i < spPackets.Length; i++)
{
SplitPartPackage splitPartPackage = spPackets[i];
byte[] buf = splitPartPackage.Message;
if (buf == null)
{
Log.Error("Expected bytes in splitpart, but got none");
continue;
}
stream.Write(buf, 0, buf.Length);
splitPartPackage.PutPool();
}
byte[] buffer = stream.ToArray();
try
{
ConnectedPackage newPackage = ConnectedPackage.CreateObject();
newPackage._datagramSequenceNumber = package._datagramSequenceNumber;
newPackage._reliability = package._reliability;
newPackage._reliableMessageNumber = package._reliableMessageNumber;
newPackage._sequencingIndex = package._sequencingIndex;
newPackage._orderingIndex = package._orderingIndex;
newPackage._orderingChannel = package._orderingChannel;
newPackage._hasSplit = false;
Package fullMessage = PackageFactory.CreatePackage(buffer[0], buffer, "raknet") ?? new UnknownPackage(buffer[0], buffer);
fullMessage.DatagramSequenceNumber = package._datagramSequenceNumber;
fullMessage.Reliability = package._reliability;
fullMessage.ReliableMessageNumber = package._reliableMessageNumber;
fullMessage.OrderingIndex = package._orderingIndex;
fullMessage.OrderingChannel = package._orderingChannel;
newPackage.Messages = new List <Package>();
newPackage.Messages.Add(fullMessage);
Log.Debug($"Assembled split package {newPackage._reliability} message #{newPackage._reliableMessageNumber}, Chan: #{newPackage._orderingChannel}, OrdIdx: #{newPackage._orderingIndex}");
HandleConnectedPackage(playerSession, newPackage);
newPackage.PutPool();
}
catch (Exception e)
{
Log.Error("Error during split message parsing", e);
if (Log.IsDebugEnabled)
{
Log.Debug($"0x{buffer[0]:x2}\n{Package.HexDump(buffer)}");
}
player.Disconnect("Bad package received from client.");
}
}
}