internal static bool HandleMessage(SocketAsyncEventArgs arg, DataUserToken userToken, int remainingBytesToProcess)
{
bool incomingTcpMessageIsReady = false;
if (userToken.receivedMessageBytesDoneCount == 0)
{
userToken.dataMessageReceived = new Byte[userToken.lengthOfCurrentIncomingMessage];
}
if (remainingBytesToProcess + userToken.receivedMessageBytesDoneCount == userToken.lengthOfCurrentIncomingMessage)
{
// Write/append the bytes received to the byte array in the
// DataHolder object that we are using to store our data.
Buffer.BlockCopy(arg.Buffer, userToken.receiveMessageOffset, userToken.dataMessageReceived, userToken.receivedMessageBytesDoneCount, remainingBytesToProcess);
incomingTcpMessageIsReady = true;
}
else
{
Buffer.BlockCopy(arg.Buffer, userToken.receiveMessageOffset, userToken.dataMessageReceived, userToken.receivedMessageBytesDoneCount, remainingBytesToProcess);
userToken.receiveMessageOffset = userToken.receiveMessageOffset - userToken.recPrefixBytesDoneThisOp;
userToken.receivedMessageBytesDoneCount += remainingBytesToProcess;
}
return(incomingTcpMessageIsReady);
}
internal static int HandlePrefix(SocketAsyncEventArgs arg, DataUserToken userToken, int remainingBytesToProcess, ref bool getLengthInfoSuccessfully)
{
if (userToken.receivedPrefixBytesDoneCount.Equals(0))
{
userToken.byteArrayForPrefix = new byte[userToken.receivePrefixLength];
}
if (remainingBytesToProcess >= (userToken.receivePrefixLength - userToken.receivedPrefixBytesDoneCount))
{
//接收数据足够多，甚至已经包含了具体的内容数据
Buffer.BlockCopy(arg.Buffer, userToken.receiveMessageOffset - userToken.receivePrefixLength + userToken.receivedPrefixBytesDoneCount, userToken.byteArrayForPrefix, userToken.receivedPrefixBytesDoneCount, userToken.receivePrefixLength - userToken.receivedPrefixBytesDoneCount);
remainingBytesToProcess = remainingBytesToProcess - userToken.receivePrefixLength + userToken.receivedPrefixBytesDoneCount;
userToken.recPrefixBytesDoneThisOp = userToken.receivePrefixLength - userToken.receivedPrefixBytesDoneCount;
userToken.receivedPrefixBytesDoneCount = userToken.receivePrefixLength;
userToken.lengthOfCurrentIncomingMessage = BitConverter.ToInt32(userToken.byteArrayForPrefix, 0);
if (userToken.lengthOfCurrentIncomingMessage.Equals(0))
{
remainingBytesToProcess = 0;
}
getLengthInfoSuccessfully = true;
}
else
{
//接收数据太小，不足以得到数据长度头信息，需要再次接收补全
Buffer.BlockCopy(arg.Buffer, userToken.receiveMessageOffset - userToken.receivePrefixLength + userToken.receivedPrefixBytesDoneCount, userToken.byteArrayForPrefix, userToken.receivedPrefixBytesDoneCount, remainingBytesToProcess);
userToken.recPrefixBytesDoneThisOp = remainingBytesToProcess;
userToken.receivedPrefixBytesDoneCount += remainingBytesToProcess;
remainingBytesToProcess = 0;
}
if (remainingBytesToProcess == 0)
{
//必须的处理步骤！确保后续数据的正确处理
userToken.receiveMessageOffset = userToken.receiveMessageOffset - userToken.recPrefixBytesDoneThisOp;
userToken.recPrefixBytesDoneThisOp = 0;
}
return(remainingBytesToProcess);
}
