private void ProcessSend(SocketAsyncEventArgs ioEventArgs)
{
DataToken dataToken = (DataToken)ioEventArgs.UserToken;
if (ioEventArgs.SocketError == SocketError.Success)
{
dataToken.messageBytesDone += ioEventArgs.BytesTransferred;
if (dataToken.messageBytesDone != dataToken.messageLength)
{
PostSend(ioEventArgs);
}
else
{
dataToken.Reset(true);
try
{
TryDequeueAndPostSend(dataToken.Socket, ioEventArgs);
}
catch
{
dataToken.Socket.ResetSendFlag();
throw;
}
}
}
else
{
dataToken.Socket.ResetSendFlag();
HandleCloseSocket(ioEventArgs);
}
}
public override bool TryReadMeaage(DataToken dataToken, byte[] buffer, out List<DataMeaage> messageList)
{
messageList = new List<DataMeaage>();
int remainingBytesToProcess = buffer.Length;
bool needPostAnother = true;
do
{
if (dataToken.prefixBytesDone < 4)
{
remainingBytesToProcess = HandlePrefix(buffer, dataToken.bufferSkip, dataToken, remainingBytesToProcess);
if (dataToken.prefixBytesDone == 4 && (dataToken.messageLength > 10 * 1024 * 1024 || dataToken.messageLength <= 0))
{
byte[] bufferBytes = BufferUtils.MergeBytes(dataToken.byteArrayForPrefix, dataToken.byteArrayForMessage);
string data = Encoding.UTF8.GetString(bufferBytes);
//消息头已接收完毕，并且接收到的消息长度大于10M，socket传输的数据已紊乱，关闭掉
TraceLog.ReleaseWriteDebug("Receive Ip {1} message length error:{0}\r\nData:{2}", dataToken.messageLength, (dataToken != null ? dataToken.Socket.RemoteEndPoint.ToNotNullString() : ""), data);
needPostAnother = false;
break;
}
//if (logger.IsDebugEnabled) logger.Debug("处理消息头，消息长度[{0}]，剩余字节[{1}]", dataToken.messageLength, remainingBytesToProcess);
if (remainingBytesToProcess == 0) break;
}
remainingBytesToProcess = HandleMessage(buffer, dataToken.bufferSkip, dataToken, remainingBytesToProcess);
if (dataToken.IsMessageReady)
{
//if (logger.IsDebugEnabled) logger.Debug("完整封包 长度[{0}],总传输[{1}],剩余[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, remainingBytesToProcess);
messageList.Add(new DataMeaage() { Data = dataToken.byteArrayForMessage, OpCode = OpCode.Binary });
if (remainingBytesToProcess != 0)
{
//if (logger.IsDebugEnabled) logger.Debug("重置缓冲区,buffskip指针[{0}]。", dataToken.bufferSkip);
dataToken.Reset(false);
}
}
else
{
//if (logger.IsDebugEnabled) logger.Debug("不完整封包 长度[{0}],总传输[{1}],已接收[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, dataToken.messageBytesDone);
}
} while (remainingBytesToProcess != 0);
if (needPostAnother)
{
//继续处理下个请求包
if (dataToken.prefixBytesDone == 4 && dataToken.IsMessageReady)
{
dataToken.Reset(true);
}
dataToken.bufferSkip = 0;
}
return needPostAnother;
}
/// <summary>
///
/// </summary>
/// <param name="dataToken"></param>
/// <param name="buffer"></param>
/// <param name="messageList"></param>
/// <returns></returns>
public override bool TryReadMeaage(DataToken dataToken, byte[] buffer, out List <DataMeaage> messageList)
{
messageList = new List <DataMeaage>();
int remainingBytesToProcess = buffer.Length;
bool needPostAnother = true;
do
{
if (dataToken.prefixBytesDone < 4)
{
remainingBytesToProcess = HandlePrefix(buffer, dataToken.bufferSkip, dataToken, remainingBytesToProcess);
if (dataToken.prefixBytesDone == 4 && (dataToken.messageLength > 10 * 1024 * 1024 || dataToken.messageLength <= 0))
{
byte[] bufferBytes = dataToken.byteArrayForMessage == null?
BufferUtils.MergeBytes(dataToken.byteArrayForPrefix, BufferUtils.GetBytes(buffer, dataToken.bufferSkip, buffer.Length - dataToken.bufferSkip)) :
BufferUtils.MergeBytes(dataToken.byteArrayForPrefix, dataToken.byteArrayForMessage, buffer);
string data = Encoding.UTF8.GetString(bufferBytes);
//消息头已接收完毕，并且接收到的消息长度大于10M，socket传输的数据已紊乱，关闭掉
TraceLog.Write("The host[{0}] request parser head byte error, byte len={1}\r\ndata:{2}",
dataToken.Socket.RemoteEndPoint.ToNotNullString(),
dataToken.messageLength,
data);
needPostAnother = false;
break;
}
//if (logger.IsDebugEnabled) logger.Debug("处理消息头，消息长度[{0}]，剩余字节[{1}]", dataToken.messageLength, remainingBytesToProcess);
if (remainingBytesToProcess == 0)
{
break;
}
}
remainingBytesToProcess = HandleMessage(buffer, dataToken.bufferSkip, dataToken, remainingBytesToProcess);
if (dataToken.IsMessageReady)
{
//if (logger.IsDebugEnabled) logger.Debug("完整封包 长度[{0}],总传输[{1}],剩余[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, remainingBytesToProcess);
messageList.Add(new DataMeaage()
{
Data = dataToken.byteArrayForMessage, OpCode = OpCode.Binary
});
if (remainingBytesToProcess != 0)
{
//if (logger.IsDebugEnabled) logger.Debug("重置缓冲区,buffskip指针[{0}]。", dataToken.bufferSkip);
dataToken.Reset(false);
}
}
else
{
//if (logger.IsDebugEnabled) logger.Debug("不完整封包 长度[{0}],总传输[{1}],已接收[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, dataToken.messageBytesDone);
}
} while (remainingBytesToProcess != 0);
if (needPostAnother)
{
//继续处理下个请求包
if (dataToken.prefixBytesDone == 4 && dataToken.IsMessageReady)
{
dataToken.Reset(true);
}
dataToken.bufferSkip = 0;
}
return(needPostAnother);
}
/// <summary>
/// 处理数据接收回调
/// </summary>
/// <param name="ioEventArgs"></param>
private void ProcessReceive(SocketAsyncEventArgs ioEventArgs)
{
DataToken dataToken = (DataToken)ioEventArgs.UserToken;
if (ioEventArgs.SocketError != SocketError.Success)
{
//Socket错误
//if (logger.IsDebugEnabled) logger.Debug("ProcessReceive:{0}", ioEventArgs.SocketError);
HandleCloseSocket(ioEventArgs);
return;
}
if (ioEventArgs.BytesTransferred == 0)
{
//对方主动关闭socket
//if (logger.IsDebugEnabled) logger.Debug("对方关闭Socket");
HandleCloseSocket(ioEventArgs);
return;
}
var exSocket = dataToken.Socket;
#region 数据解析
List <byte[]> msgs = new List <byte[]>();
int remainingBytesToProcess = ioEventArgs.BytesTransferred;
bool needPostAnother = true;
do
{
if (dataToken.prefixBytesDone < 4)
{
remainingBytesToProcess = prefixHandler.HandlePrefix(ioEventArgs, dataToken, remainingBytesToProcess);
if (dataToken.prefixBytesDone == 4 && (dataToken.messageLength > 10 * 1024 * 1024 || dataToken.messageLength <= 0))
{
//消息头已接收完毕，并且接收到的消息长度大于10M，socket传输的数据已紊乱，关闭掉
logger.Warn("接收到的消息长度错误:{0}", dataToken.messageLength);
needPostAnother = false;
HandleCloseSocket(ioEventArgs);
break;
}
//if (logger.IsDebugEnabled) logger.Debug("处理消息头，消息长度[{0}]，剩余字节[{1}]", dataToken.messageLength, remainingBytesToProcess);
if (remainingBytesToProcess == 0)
{
break;
}
}
remainingBytesToProcess = messageHandler.HandleMessage(ioEventArgs, dataToken, remainingBytesToProcess);
if (dataToken.IsMessageReady)
{
//if (logger.IsDebugEnabled) logger.Debug("完整封包 长度[{0}],总传输[{1}],剩余[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, remainingBytesToProcess);
msgs.Add(dataToken.byteArrayForMessage);
if (remainingBytesToProcess != 0)
{
//if (logger.IsDebugEnabled) logger.Debug("重置缓冲区,buffskip指针[{0}]。", dataToken.bufferSkip);
dataToken.Reset(false);
}
}
else
{
//if (logger.IsDebugEnabled) logger.Debug("不完整封包 长度[{0}],总传输[{1}],已接收[{2}]", dataToken.messageLength, ioEventArgs.BytesTransferred, dataToken.messageBytesDone);
}
} while (remainingBytesToProcess != 0);
#endregion
if (needPostAnother)
{
if (dataToken.prefixBytesDone == 4 && dataToken.IsMessageReady)
{
dataToken.Reset(true);
}
dataToken.bufferSkip = 0;
PostReceive(ioEventArgs);
}
foreach (var m in msgs)
{
try
{
OnDataReceived(new ConnectionEventArgs {
Socket = exSocket, Data = m
});
}
catch (Exception ex)
{
TraceLog.WriteError("OnDataReceived error:{0}", ex);
}
}
}
/// <summary>
/// 接收处理
/// </summary>
/// <param name="e"></param>
private void ProcessReceive(SocketAsyncEventArgs e)
{
DataToken session = e.UserToken as DataToken;
if (e.BytesTransferred > 0)
{
if (e.SocketError == SocketError.Success)
{
//是否有已接收但未处理的数据
int remainingBytesToProcess = e.BytesTransferred;
do
{
if (session.prefixBytesDone < 4)
{
remainingBytesToProcess = _prefixHandler.HandlePrefix(e, session, remainingBytesToProcess);
if (remainingBytesToProcess == 0)
{
session.bufferSkip = 0;
StartReceive(e);
return;
}
}
remainingBytesToProcess = _messageHandler.HandleMessage(e, session, remainingBytesToProcess);
if (session.IsMessageReady)
{
// 触发收到消息事件
if (ReceiveCompleted != null)
{
byte[] data = session.byteArrayForMessage;
ReceiveCompleted.BeginInvoke(new SocketProcessEventArgs()
{
Socket = session.Socket, Data = data
}, null, null);
}
if (remainingBytesToProcess != 0)
{
session.Reset(false);
}
}
}while (remainingBytesToProcess != 0);
if (session.prefixBytesDone == 4 && session.IsMessageReady)
{
session.Reset(true);
}
session.bufferSkip = 0;
StartReceive(e);
}
else
{
session.Reset(true);
ProcessSocketError(e);
}
}
else
{
session.Reset(true);
CloseClientSocket(e);
}
}
