internal static unsafe Error SendMessage(SafeHandle socket, MessageHeader *messageHeader, SocketFlags flags, long *sent)
{
bool release = false;
try
{
socket.DangerousAddRef(ref release);
return(DangerousSendMessage((int)socket.DangerousGetHandle(), messageHeader, flags, sent));
}
finally
{
if (release)
{
socket.DangerousRelease();
}
}
}
internal static extern unsafe Error SendMessage(int socket, MessageHeader *messageHeader, SocketFlags flags, long *sent);
internal static extern unsafe Error ReceiveMessage(SafeHandle socket, MessageHeader *messageHeader, SocketFlags flags, long *received);
internal static unsafe partial bool TryGetIPPacketInformation(MessageHeader *messageHeader, bool isIPv4, IPPacketInformation *packetInfo);
internal static unsafe partial bool TryGetIPPacketInformation(MessageHeader *messageHeader, [MarshalAs(UnmanagedType.Bool)] bool isIPv4, IPPacketInformation *packetInfo);
private static extern unsafe Error DangerousSendMessage(int socket, MessageHeader *messageHeader, SocketFlags flags, long *sent);
private static unsafe void Receive()
{
// Receive anything sent to us
// Similar setup for sending data
MessageHeader *header = (MessageHeader *)receiveStream.BufferRead->Buffer;
int bytesNeeded = 0;
if (receiveStream.TotalBytes < sizeof(MessageHeader))
{
bytesNeeded = sizeof(MessageHeader) - receiveStream.TotalBytes;
}
else
{
bytesNeeded = sizeof(MessageHeader) + header->bytes - receiveStream.TotalBytes;
}
while (bytesNeeded > 0)
{
MessageStream.MessageStreamBuffer *bufferWrite = receiveStream.BufferWrite;
int bytesAvail = bufferWrite->Capacity - bufferWrite->Size;
#if UNITY_WEBGL
uint actualWritten = WebSocketReceive(bufferWrite->Buffer + bufferWrite->Size,
bytesNeeded <= bytesAvail ? bytesNeeded : bytesAvail);
if (actualWritten == 0xffffffff)
#else
uint actualWritten = Baselib_Socket_TCP_Recv(hSocket, bufferWrite->Buffer + bufferWrite->Size,
(uint)(bytesNeeded <= bytesAvail ? bytesNeeded : bytesAvail), (Baselib_ErrorState *)UnsafeUtility.AddressOf(ref errState));
if (errState.code != Baselib_ErrorCode.Success)
#endif
{
// Something bad happened; lost connection maybe?
// After cleaning up, next time we will try to re-initialize
Disconnect();
initRetryCounter = kInitRetryCounter;
return;
}
if (bytesNeeded > 0 && actualWritten == 0)
{
// Don't do anything with data until we've received everything
return;
}
receiveStream.UpdateSize((int)actualWritten);
bytesNeeded -= (int)actualWritten;
if (bytesNeeded == 0)
{
// Finished receiving header
if (receiveStream.TotalBytes == sizeof(MessageHeader))
{
// De-synced somewhere... reset connection
if (header->magicId != EditorMessageIds.kMagicNumber)
{
Disconnect();
initRetryCounter = kInitRetryCounter;
return;
}
bytesNeeded = header->bytes;
receiveStream.Allocate(bytesNeeded);
}
// Finished receiving message
if (bytesNeeded == 0)
{
// Otherwise bytesNeeded becomes 0 after message is finished, which can be immediately in the
// case of PlayerMessageId.kApplicationQuit (message size 0)
foreach (var e in m_EventMessageList)
{
if (e.messageId == header->messageId)
{
// This could be anything from a 4-byte "bool" to an asset sent from the editor
byte[] messageData = receiveStream.ToByteArray(sizeof(MessageHeader), receiveStream.TotalBytes);
e.Invoke(0, messageData);
}
}
if (header->messageId == EditorMessageIds.kApplicationQuit)
{
UnityEngine.Debug.Log("Unity editor has been closed");
Disconnect();
}
if (header->messageId == EditorMessageIds.kNoFurtherConnections)
{
UnityEngine.Debug.Log("Unity editor can not accept any more connections");
Disconnect();
}
// Poll for next message
receiveStream.RecycleStreamAndFreeExtra();
bytesNeeded = sizeof(MessageHeader);
}
}
}
// This code should not be executable
throw new InvalidOperationException("Internal error receiving network data");
}
private static extern unsafe Error DangerousReceiveMessage(int socket, MessageHeader *messageHeader, SocketFlags flags, long *received);
internal static unsafe partial Error SendMessage(SafeHandle socket, MessageHeader *messageHeader, SocketFlags flags, long *sent);
