void OnCommandReceived(UdpBitStream buffer)
{
if (ParseHeader(ref buffer))
{
stats.CommandsReceived += 1;
buffer.Ptr = UdpHeader.GetSize(socket);
UdpCommandType cmd = (UdpCommandType)buffer.ReadByte(8);
switch (cmd)
{
case UdpCommandType.Connect: OnCommandConnect(buffer); break;
case UdpCommandType.Accepted: OnCommandAccepted(buffer); break;
case UdpCommandType.Refused: OnCommandRefused(buffer); break;
case UdpCommandType.Disconnected: OnCommandDisconnected(buffer); break;
case UdpCommandType.Ping: OnCommandPing(buffer); break;
default: ConnectionError(UdpConnectionError.IncorrectCommand); break;
}
}
}
void RecvUnconnectedPacket(UdpBitStream buff, UdpEndPoint ep)
{
buff.Ptr = UdpHeader.GetSize(this);
if (buff.ReadByte(8) == (byte)UdpCommandType.Connect)
{
if (Config.AllowIncommingConnections && ((connLookup.Count + pendingConnections.Count) < Config.ConnectionLimit || Config.ConnectionLimit == -1))
{
if (Config.AutoAcceptIncommingConnections)
{
AcceptConnection(ep);
}
else
{
if (pendingConnections.Add(ep))
{
Raise(UdpEvent.PUBLIC_CONNECT_REQUEST, ep);
}
}
}
else
{
SendRefusedCommand(ep);
}
}
else
{
UdpLog.Debug("received invalid header byte in unconnected packet from {0}", ep.ToString());
}
}
internal void SendObject(object o)
{
serializer.SendNext(o);
while (serializer.HasQueuedObjects)
{
UdpSendFailReason reason = CheckCanSend(false);
if (reason != UdpSendFailReason.None)
{
while (serializer.HasQueuedObjects)
{
socket.Raise(UdpEvent.PUBLIC_OBJECT_SEND_FAILED, this, serializer.NextObject(), reason);
}
break;
}
UdpBitStream stream = new UdpBitStream(socket.GetWriteBuffer(), mtu, UdpHeader.GetSize(socket));
object obj = serializer.NextObject();
if (serializer.Pack(ref stream, ref obj))
{
if (stream.Overflowing && (socket.Config.AllowPacketOverflow == false))
{
UdpLog.Error("stream to {0} is overflowing, not sending", endpoint.ToString());
socket.Raise(UdpEvent.PUBLIC_OBJECT_SEND_FAILED, this, obj, UdpSendFailReason.StreamOverflow);
return;
}
UdpHeader header = MakeHeader(true);
header.Pack(new UdpBitStream(stream.Data, mtu, 0), socket);
UdpHandle handle = MakeHandle(ref header);
handle.Object = obj;
if (SendStream(stream, handle, alwaysSendMtu) == false)
{
socket.Raise(UdpEvent.PUBLIC_OBJECT_SEND_FAILED, this, obj, UdpSendFailReason.SocketError);
}
else
{
stats.PacketSent();
}
}
else
{
socket.Raise(UdpEvent.PUBLIC_OBJECT_SEND_FAILED, this, obj, UdpSendFailReason.SerializerReturnedFalse);
}
}
}
internal void SendCommand(UdpCommandType cmd)
{
if (CheckCanSend(true) == UdpSendFailReason.None)
{
UdpBitStream stream = new UdpBitStream(socket.GetWriteBuffer(), mtu, UdpHeader.GetSize(socket));
stream.WriteByte((byte)cmd, 8);
UdpHeader header = MakeHeader(false);
header.Pack(new UdpBitStream(stream.Data, mtu, 0), socket);
UdpHandle handle = MakeHandle(ref header);
handle.Object = null;
if (SendStream(stream, handle, false) == false)
{
// do something here?
}
stats.CommandSent += 1;
}
}
void OnObjectReceived(UdpBitStream buffer)
{
EnsureClientIsConnected();
if (CheckState(UdpConnectionState.Connected) == false)
{
return;
}
if (ParseHeader(ref buffer))
{
object obj = null;
buffer.Ptr = UdpHeader.GetSize(socket);
if (serializer.Unpack(ref buffer, ref obj))
{
socket.Raise(UdpEvent.PUBLIC_OBJECT_RECEIVED, this, obj);
}
stats.PacketReceived();
}
}
void SendRefusedCommand(UdpEndPoint endpoint)
{
UdpBitStream stream = new UdpBitStream(GetWriteBuffer(), Config.DefaultMtu, UdpHeader.GetSize(this));
stream.WriteByte((byte)UdpCommandType.Refused, 8);
UdpHeader header = new UdpHeader();
header.IsObject = false;
header.AckHistory = 0;
header.AckSequence = 1;
header.ObjSequence = 1;
header.Now = 0;
header.Pack(new UdpBitStream(stream.Data, Config.DefaultMtu, 0), this);
if (Send(endpoint, stream.Data, UdpMath.BytesRequired(stream.Ptr)) == false)
{
// do something here?
}
}
