private void ProcessMessage(object buffer)
{
try
{
byte[] data = (byte[])buffer;
if (!string.IsNullOrWhiteSpace(this.encryKey))
{
data = AesEncryHelper.AesDecrypt(data, this.encryKey);
}
Message message = EntityBufCore.DeSerialize <Message>(data);
if (message.IsMessage(MessageType.NEGOTIATIONENCRYR))
{
var nmsg = message.GetMessageBody <NegotiationEncryMessage>();
this.encryKey = Encoding.ASCII.GetString(RsaEncryHelper.RsaDecrypt(Convert.FromBase64String(nmsg.EncryKey), this.rsaRrivateKey));
Console.WriteLine("收到加密串:" + encryKey);
_startSign.Set();
}
else
{
OnMessage(message);
}
}
catch (Exception e)
{
OnError(e);
}
}
public static int SendMessage(this Socket s, Message message, string encrykey)
{
try
{
if (s == null)
{
return(0);
}
byte[] data = null;
int bufferindex = -1;
long size = 0;
if (string.IsNullOrWhiteSpace(encrykey))
{
EntityBuf.EntityBufCore.Serialize(message, _sendBufferManger, ref bufferindex, ref size, ref data);
}
else
{
data = EntityBuf.EntityBufCore.Serialize(message);
}
if (bufferindex == -1)
{
if (!string.IsNullOrWhiteSpace(encrykey))
{
using (var ms = new System.IO.MemoryStream())
{
for (int i = 0; i < 8; i++)
{
ms.WriteByte(0);
}
var enbytes = AesEncryHelper.AesEncrypt(data, encrykey);
ms.Write(enbytes, 0, enbytes.Length);
data = ms.ToArray();
}
}
byte[] dataLen = BitConverter.GetBytes(data.Length - 4);
for (int i = 0; i < 4; i++)
{
data[i] = dataLen[i];
}
var crc32 = HashEncrypt.GetCRC32(data, 8);
//LogManager.LogHelper.Instance.Debug("校验值:" + crc32);
var crc32bytes = BitConverter.GetBytes(crc32);
for (int i = 4; i < 8; i++)
{
data[i] = crc32bytes[i - 4];
}
lock (s)
{
var sendcount = s.Send(data, SocketFlags.None);
//if (SocketApplicationEnvironment.TraceSocketDataBag && !string.IsNullOrWhiteSpace(message.MessageHeader.TransactionID))
//{
// LogManager.LogHelper.Instance.Debug(s.Handle + "发送数据:" + message.MessageHeader.TransactionID + "长度:" + data.Length + ", " + Convert.ToBase64String(data));
//}
return(sendcount);
}
}
else
{
try
{
//LogManager.LogHelper.Instance.Error("发送数据bufferindex:" + bufferindex + ",size:" + size);
byte[] dataLen = BitConverter.GetBytes((int)size - 4);
int offset = _sendBufferManger.GetOffset(bufferindex);
for (int i = 0; i < 4; i++)
{
_sendBufferManger.Buffer[i + offset] = dataLen[i];
}
var crc32 = HashEncrypt.GetCRC32(_sendBufferManger.Buffer, offset + 8, (int)size - 8);
//LogManager.LogHelper.Instance.Debug("校验值:" + crc32);
var crc32bytes = BitConverter.GetBytes(crc32);
for (int i = 4; i < 8; i++)
{
_sendBufferManger.Buffer[i + offset] = crc32bytes[i - 4];
}
int sendcount = 0;
lock (s)
{
SocketError senderror = SocketError.Success;
sendcount = s.Send(_sendBufferManger.Buffer, offset, (int)size, SocketFlags.None, out senderror);
//if (SocketApplicationEnvironment.TraceSocketDataBag && !string.IsNullOrWhiteSpace(message.MessageHeader.TransactionID))
//{
// var sendbytes = _sendBufferManger.Buffer.Skip(offset).Take((int)size).ToArray();
// LogManager.LogHelper.Instance.Debug(s.Handle + "发送数据:" + message.MessageHeader.TransactionID + "长度:" + size + ", " + Convert.ToBase64String(sendbytes));
//}
if (senderror != SocketError.Success)
{
throw new Exception(senderror.ToString());
}
}
return(sendcount);
}
finally
{
_sendBufferManger.RealseBuffer(bufferindex);
}
}
}
catch (Exception ex)
{
ex.Data.Add("TransactionID", message.MessageHeader.TransactionID);
throw new SocketApplicationException("SendMessage fail", ex);
}
}
void SocketAsyncEventArgs_Completed(object sender, SocketAsyncEventArgs e)
{
e.Completed -= SocketAsyncEventArgs_Completed;
var args = e as IOCPSocketAsyncEventArgs;
if (args.BytesTransferred == 0 || args.SocketError != SocketError.Success)
{
if (SocketApplication.SocketApplicationEnvironment.TraceSocketDataBag)
{
LogManager.LogHelper.Instance.Debug(e.AcceptSocket.Handle + "异常断开:" + args.SocketError);
}
RemoveSession(args);
return;
}
else
{
bool hasdataerror = false;
try
{
#region 数据逻辑
if (!args.IsReadPackLen)
{
if (args.BytesTransferred != 4)
{
throw new Exception("读取长度失败");
}
var offset = args.BufferIndex == -1 ? 0 : _bufferpoll.GetOffset(args.BufferIndex);
var dataLen = BitConverter.ToInt32(e.Buffer, offset);
if (SocketApplication.SocketApplicationEnvironment.TraceSocketDataBag)
{
LogManager.LogHelper.Instance.Debug(e.AcceptSocket.Handle + "准备接收数据:长度" + dataLen, null);
}
if (dataLen > MaxPackageLength || dataLen <= 0)
{
throw new SocketSessionDataException(string.Format("数据异常，长度:" + dataLen));
}
else
{
args.IsReadPackLen = true;
//byte[] readbuffer = new byte[dataLen];
args.BufferLen = dataLen;
args.BufferRev = 0;
//args.SetBuffer(readbuffer, 0, dataLen);
SetBuffer(args, 0, dataLen);
}
}
else
{
if (SocketApplication.SocketApplicationEnvironment.TraceSocketDataBag)
{
var offset1 = (args.BufferLen == args.Buffer.Length) ? 0 : _bufferpoll.GetOffset(args.BufferIndex);
var bytes = args.Buffer.Skip(offset1 + args.BufferRev).Take(args.BytesTransferred).ToArray();
//if (args.BytesTransferred < args.BufferLen)
{
LogManager.LogHelper.Instance.Debug(string.Format(e.AcceptSocket.Handle + "接收数据{0}/{1}/{2},{3}", args.BufferLen, args.BufferRev, args.BytesTransferred, Convert.ToBase64String(bytes)), null);
}
}
args.BufferRev += args.BytesTransferred;
Exception messageError = null;
if (args.BufferRev == args.BufferLen)
{
byte[] bt = null;
var offset = args.BufferIndex == -1 ? 0 : _bufferpoll.GetOffset(args.BufferIndex);
//校验
var crc32 = BitConverter.ToInt32(args.Buffer, offset);
var calcrc32 = LJC.FrameWork.Comm.HashEncrypt.GetCRC32(args.Buffer, offset + 4, args.BufferLen - 4);
if (calcrc32 == crc32)
{
bt = new byte[args.BufferLen - 4];
for (int i = 4; i < args.BufferLen; i++)
{
bt[i - 4] = args.Buffer[offset + i];
}
ThreadPool.QueueUserWorkItem(new WaitCallback((buf) =>
{
Session connSession = null;
try
{
Message message = null;
if (_connectSocketDic.TryGetValue(args.UserToken.ToString(), out connSession))
{
if (!string.IsNullOrWhiteSpace(connSession.EncryKey))
{
try
{
bt = AesEncryHelper.AesDecrypt(bt, connSession.EncryKey);
}
catch (Exception ex)
{
throw new SocketApplicationException("解密失败", ex);
}
}
try
{
message = EntityBufCore.DeSerialize <Message>(bt);
}
catch (Exception ex)
{
messageError = ex;
}
connSession.LastSessionTime = DateTime.Now;
connSession.BytesRev += bt.Length;
if (messageError == null)
{
//如果是协商加密的
if (message.IsMessage(MessageType.NEGOTIATIONENCRYR))
{
var nmsg = message.GetMessageBody <NegotiationEncryMessage>();
if (string.IsNullOrWhiteSpace(nmsg.PublicKey))
{
throw new SocketApplicationException("公钥错误");
}
var encrykey = connSession.EncryKey;
if (string.IsNullOrWhiteSpace(encrykey))
{
encrykey = Guid.NewGuid().ToString("N");
Console.WriteLine("发送加密串:" + encrykey);
var rep = new Message(MessageType.NEGOTIATIONENCRYR);
rep.SetMessageBody(nmsg);
nmsg.EncryKey = Convert.ToBase64String(RsaEncryHelper.RsaEncrypt(Encoding.ASCII.GetBytes(encrykey), nmsg.PublicKey));;
connSession.SendMessage(rep);
connSession.EncryKey = encrykey;
}
else
{
throw new SocketApplicationException("不允许多次协商密钥");
}
}
else
{
FromApp(message, connSession);
}
}
else
{
OnError(messageError);
}
}
else
{
OnError(new Exception("取会话失败,args.UserToken=" + args.UserToken));
}
}
catch (SocketApplicationException ex)
{
if (connSession != null)
{
connSession.Close();
}
ex.Data.Add("SessionID", connSession.SessionID);
OnError(ex);
}
}), bt);
}
else
{
messageError = new Exception("检查校验码出错");
messageError.Data.Add("crc32", crc32);
messageError.Data.Add("calcrc32", calcrc32);
messageError.Data.Add("data", bt == null ? "" : Convert.ToBase64String(bt));
//LogManager.LogHelper.Instance.Error("接收数据出错", messageError);
Session connSession;
if (_connectSocketDic.TryGetValue(args.UserToken.ToString(), out connSession))
{
connSession.LastSessionTime = DateTime.Now;
connSession.BytesRev += args.BufferRev;
OnError(messageError);
}
else
{
OnError(new Exception("取会话失败,args.UserToken=" + args.UserToken));
}
}
args.IsReadPackLen = false;
//args.SetBuffer(_bufferpoll.Buffer, _bufferpoll.GetOffset(args.BufferIndex), 4);
SetBuffer(args, 0, 4);
}
else
{
SetBuffer(args, args.BufferRev, args.BufferLen - args.BufferRev);
LogManager.LogHelper.Instance.Debug("e.SetBuffer:" + (args.BufferRev) + ",len:" + (args.BufferLen - args.BufferRev), null);
}
}
#endregion
}
catch (SocketSessionDataException ex)
{
RemoveSession(args);
hasdataerror = true;
OnError(ex);
}
catch (Exception ex)
{
OnError(ex);
}
finally
{
if (!hasdataerror)
{
e.Completed += SocketAsyncEventArgs_Completed;
if (!e.AcceptSocket.ReceiveAsync(e))
{
LogManager.LogHelper.Instance.Debug(e.AcceptSocket.Handle + "同步完成，手动处理", null);
SocketAsyncEventArgs_Completed(null, e);
}
}
}
}
}
/// <summary>
/// aes加密
/// </summary>
/// <param name="relatedfilePath"></param>
/// <returns></returns>
public static string DeAes(this string data)
{
return(AesEncryHelper.AESDecrypt(data));
}
