2021-12-10 23:38:40 +08:00
|
|
|
|
using HunterProtobufCore;
|
|
|
|
|
using ProtoBuf;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Net;
|
|
|
|
|
using System.Net.Sockets;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace HaoYueNet.ServerNetwork
|
|
|
|
|
{
|
|
|
|
|
public class SocketManager
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 心跳包数据
|
|
|
|
|
/// </summary>
|
|
|
|
|
private byte[] HeartbeatData = new byte[5] { 0x05, 0x00, 0x00, 0x00, 0x00 };
|
|
|
|
|
//响应倒计时计数最大值
|
|
|
|
|
//public int MaxRevIndexNum { get; set; } = 5;
|
|
|
|
|
////发送倒计时计数最大值
|
|
|
|
|
//public int MaxSendIndexNum { get; set; } = 3;
|
|
|
|
|
|
|
|
|
|
//响应倒计时计数最大值
|
|
|
|
|
public int MaxRevIndexNum { get; set; } = 50;
|
|
|
|
|
//发送倒计时计数最大值
|
|
|
|
|
public int MaxSendIndexNum { get; set; } = 3;
|
|
|
|
|
|
|
|
|
|
//计时器间隔
|
|
|
|
|
private static int TimerInterval = 3000;
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 心跳包计数器
|
|
|
|
|
/// </summary>
|
|
|
|
|
private System.Timers.Timer _heartTimer;
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
public int m_maxConnectNum; //最大连接数
|
|
|
|
|
public int m_revBufferSize; //最大接收字节数
|
|
|
|
|
public BufferManager m_bufferManager;
|
|
|
|
|
public const int opsToAlloc = 2;
|
|
|
|
|
Socket listenSocket; //监听Socket
|
|
|
|
|
public SocketEventPool m_pool;
|
|
|
|
|
|
|
|
|
|
public SocketEventPool m_Sendpool;
|
|
|
|
|
public TokenMsgPool msg_pool;
|
|
|
|
|
//public Dictionary<AsyncUserToken, SocketAsyncEventArgs> _DictAsyncUserTokenSendSAEA = new Dictionary<AsyncUserToken, SocketAsyncEventArgs>();
|
|
|
|
|
public int m_clientCount; //连接的客户端数量
|
|
|
|
|
public Semaphore m_maxNumberAcceptedClients;
|
|
|
|
|
|
|
|
|
|
List<AsyncUserToken> m_clients; //客户端列表
|
|
|
|
|
public Dictionary<Socket, AsyncUserToken> _DictSocketAsyncUserToken = new Dictionary<Socket, AsyncUserToken>();
|
|
|
|
|
|
|
|
|
|
#region 定义委托
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 客户端连接数量变化时触发
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="num">当前增加客户的个数(用户退出时为负数,增加时为正数,一般为1)</param>
|
|
|
|
|
/// <param name="token">增加用户的信息</param>
|
|
|
|
|
public delegate void OnClientNumberChange(int num, AsyncUserToken token);
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 接收到客户端的数据
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="token">客户端</param>
|
|
|
|
|
/// <param name="buff">客户端数据</param>
|
|
|
|
|
public delegate void OnReceiveData(AsyncUserToken token, byte[] buff);
|
|
|
|
|
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
#region 定义事件
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 客户端连接数量变化事件
|
|
|
|
|
/// </summary>
|
|
|
|
|
public event OnClientNumberChange ClientNumberChange;
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 接收到客户端的数据事件
|
|
|
|
|
/// </summary>
|
|
|
|
|
public event OnReceiveData ReceiveClientData;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
#region 定义属性
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 获取客户端列表
|
|
|
|
|
/// </summary>
|
|
|
|
|
public List<AsyncUserToken> ClientList { get { return m_clients; } }
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 构造函数
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="numConnections">最大连接数</param>
|
|
|
|
|
/// <param name="receiveBufferSize">缓存区大小</param>
|
|
|
|
|
public SocketManager(int numConnections, int receiveBufferSize)
|
|
|
|
|
{
|
|
|
|
|
m_clientCount = 0;
|
|
|
|
|
m_maxConnectNum = numConnections;
|
|
|
|
|
m_revBufferSize = receiveBufferSize;
|
|
|
|
|
// allocate buffers such that the maximum number of sockets can have one outstanding read and
|
|
|
|
|
//write posted to the socket simultaneously
|
|
|
|
|
m_bufferManager = new BufferManager(receiveBufferSize * numConnections * opsToAlloc, receiveBufferSize);
|
|
|
|
|
|
|
|
|
|
m_pool = new SocketEventPool(numConnections);
|
|
|
|
|
m_Sendpool = new SocketEventPool(numConnections);
|
|
|
|
|
|
|
|
|
|
msg_pool = new TokenMsgPool(numConnections);
|
|
|
|
|
|
|
|
|
|
m_maxNumberAcceptedClients = new Semaphore(numConnections, numConnections);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 初始化
|
|
|
|
|
/// </summary>
|
|
|
|
|
public void Init()
|
|
|
|
|
{
|
|
|
|
|
// Allocates one large byte buffer which all I/O operations use a piece of. This gaurds
|
|
|
|
|
// against memory fragmentation
|
|
|
|
|
m_bufferManager.InitBuffer();
|
|
|
|
|
m_clients = new List<AsyncUserToken>();
|
|
|
|
|
// preallocate pool of SocketAsyncEventArgs objects
|
|
|
|
|
SocketAsyncEventArgs readWriteEventArg;
|
|
|
|
|
|
|
|
|
|
for (int i = 0; i < m_maxConnectNum; i++)
|
|
|
|
|
{
|
|
|
|
|
readWriteEventArg = new SocketAsyncEventArgs();
|
|
|
|
|
readWriteEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(IO_Completed);
|
|
|
|
|
readWriteEventArg.UserToken = new AsyncUserToken();
|
|
|
|
|
|
|
|
|
|
// assign a byte buffer from the buffer pool to the SocketAsyncEventArg object
|
|
|
|
|
m_bufferManager.SetBuffer(readWriteEventArg);
|
|
|
|
|
// add SocketAsyncEventArg to the pool
|
|
|
|
|
m_pool.Push(readWriteEventArg);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//尝试
|
|
|
|
|
for (int i = 0; i < m_maxConnectNum; i++)
|
|
|
|
|
{
|
|
|
|
|
readWriteEventArg = new SocketAsyncEventArgs();
|
|
|
|
|
readWriteEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(IO_Completed2);
|
|
|
|
|
readWriteEventArg.UserToken = new AsyncUserToken();
|
|
|
|
|
|
|
|
|
|
// assign a byte buffer from the buffer pool to the SocketAsyncEventArg object
|
|
|
|
|
m_bufferManager.SetBuffer(readWriteEventArg);
|
|
|
|
|
// add SocketAsyncEventArg to the pool
|
|
|
|
|
m_Sendpool.Push(readWriteEventArg);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 启动服务
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="localEndPoint"></param>
|
|
|
|
|
public bool Start(IPEndPoint localEndPoint)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
m_clients.Clear();
|
|
|
|
|
listenSocket = new Socket(localEndPoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
|
|
|
|
|
listenSocket.Bind(localEndPoint);
|
|
|
|
|
// start the server with a listen backlog of 100 connections
|
|
|
|
|
listenSocket.Listen(m_maxConnectNum);
|
|
|
|
|
// post accepts on the listening socket
|
|
|
|
|
StartAccept(null);
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
_heartTimer = new System.Timers.Timer();
|
|
|
|
|
_heartTimer.Interval = TimerInterval;
|
|
|
|
|
_heartTimer.Elapsed += CheckUpdatetimer_Elapsed;
|
|
|
|
|
_heartTimer.AutoReset = true;
|
|
|
|
|
_heartTimer.Enabled = true;
|
|
|
|
|
//Console.WriteLine("开启心跳包定时器");
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
catch (Exception)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 停止服务
|
|
|
|
|
/// </summary>
|
|
|
|
|
public void Stop()
|
|
|
|
|
{
|
|
|
|
|
foreach (AsyncUserToken token in m_clients)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
token.Socket.Shutdown(SocketShutdown.Both);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
}
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
listenSocket.Shutdown(SocketShutdown.Both);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
|
|
|
|
|
listenSocket.Close();
|
|
|
|
|
int c_count = m_clients.Count;
|
|
|
|
|
lock (m_clients) { m_clients.Clear(); }
|
|
|
|
|
//补充处理
|
|
|
|
|
lock (_DictSocketAsyncUserToken) { _DictSocketAsyncUserToken.Clear(); }
|
|
|
|
|
|
|
|
|
|
if (ClientNumberChange != null)
|
|
|
|
|
ClientNumberChange(-c_count, null);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public void CloseClient(AsyncUserToken token)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
token.Socket.Shutdown(SocketShutdown.Both);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Begins an operation to accept a connection request from the client
|
|
|
|
|
//
|
|
|
|
|
// <param name="acceptEventArg">The context object to use when issuing
|
|
|
|
|
// the accept operation on the server's listening socket</param>
|
|
|
|
|
public void StartAccept(SocketAsyncEventArgs acceptEventArg)
|
|
|
|
|
{
|
|
|
|
|
if (acceptEventArg == null)
|
|
|
|
|
{
|
|
|
|
|
acceptEventArg = new SocketAsyncEventArgs();
|
|
|
|
|
acceptEventArg.Completed += new EventHandler<SocketAsyncEventArgs>(AcceptEventArg_Completed);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// socket must be cleared since the context object is being reused
|
|
|
|
|
acceptEventArg.AcceptSocket = null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
m_maxNumberAcceptedClients.WaitOne();
|
|
|
|
|
if (!listenSocket.AcceptAsync(acceptEventArg))
|
|
|
|
|
{
|
|
|
|
|
ProcessAccept(acceptEventArg);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// This method is the callback method associated with Socket.AcceptAsync
|
|
|
|
|
// operations and is invoked when an accept operation is complete
|
|
|
|
|
//
|
|
|
|
|
void AcceptEventArg_Completed(object sender, SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
ProcessAccept(e);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void ProcessAccept(SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Interlocked.Increment(ref m_clientCount);
|
|
|
|
|
// Get the socket for the accepted client connection and put it into the
|
|
|
|
|
//ReadEventArg object user token
|
|
|
|
|
SocketAsyncEventArgs readEventArgs = m_pool.Pop();
|
|
|
|
|
AsyncUserToken userToken = (AsyncUserToken)readEventArgs.UserToken;
|
|
|
|
|
userToken.Socket = e.AcceptSocket;
|
|
|
|
|
userToken.ConnectTime = DateTime.Now;
|
|
|
|
|
userToken.Remote = e.AcceptSocket.RemoteEndPoint;
|
|
|
|
|
userToken.IPAddress = ((IPEndPoint)(e.AcceptSocket.RemoteEndPoint)).Address;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
userToken.RevIndex = MaxRevIndexNum;
|
|
|
|
|
userToken.SendIndex = MaxSendIndexNum;
|
|
|
|
|
|
|
|
|
|
lock (m_clients) { m_clients.Add(userToken); }
|
|
|
|
|
|
|
|
|
|
//补充处理
|
|
|
|
|
lock (_DictSocketAsyncUserToken) { _DictSocketAsyncUserToken.Add(userToken.Socket, userToken); }
|
|
|
|
|
|
|
|
|
|
if (ClientNumberChange != null)
|
|
|
|
|
ClientNumberChange(1, userToken);
|
|
|
|
|
if (!e.AcceptSocket.ReceiveAsync(readEventArgs))
|
|
|
|
|
{
|
|
|
|
|
ProcessReceive(readEventArgs);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (Exception me)
|
|
|
|
|
{
|
|
|
|
|
//RuncomLib.Log.LogUtils.Info(me.Message + "\r\n" + me.StackTrace);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Accept the next connection request
|
|
|
|
|
if (e.SocketError == SocketError.OperationAborted) return;
|
|
|
|
|
StartAccept(e);
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
void IO_Completed(object sender, SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
// determine which type of operation just completed and call the associated handler
|
|
|
|
|
|
|
|
|
|
switch (e.LastOperation)
|
|
|
|
|
{
|
|
|
|
|
case SocketAsyncOperation.Receive:
|
|
|
|
|
ProcessReceive(e);
|
|
|
|
|
break;
|
|
|
|
|
case SocketAsyncOperation.Send:
|
|
|
|
|
ProcessSend(e);
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
throw new ArgumentException("The last operation completed on the socket was not a receive or send");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
void IO_Completed2(object sender, SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
// determine which type of operation just completed and call the associated handler
|
2021-12-27 23:52:27 +08:00
|
|
|
|
Console.WriteLine("就他妈从来没进过");
|
|
|
|
|
//switch (e.LastOperation)
|
|
|
|
|
//{
|
|
|
|
|
// case SocketAsyncOperation.Receive:
|
|
|
|
|
// ProcessReceive(e);
|
|
|
|
|
// break;
|
|
|
|
|
// case SocketAsyncOperation.Send:
|
|
|
|
|
// ProcessSend2(e);
|
|
|
|
|
// break;
|
|
|
|
|
// default:
|
|
|
|
|
// throw new ArgumentException("The last operation completed on the socket was not a receive or send");
|
|
|
|
|
//}
|
2021-12-10 23:38:40 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// This method is invoked when an asynchronous receive operation completes.
|
|
|
|
|
// If the remote host closed the connection, then the socket is closed.
|
|
|
|
|
// If data was received then the data is echoed back to the client.
|
|
|
|
|
//
|
|
|
|
|
private void ProcessReceive(SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
// check if the remote host closed the connection
|
|
|
|
|
AsyncUserToken token = (AsyncUserToken)e.UserToken;
|
|
|
|
|
if (e.BytesTransferred > 0 && e.SocketError == SocketError.Success)
|
|
|
|
|
{
|
|
|
|
|
//读取数据
|
|
|
|
|
byte[] data = new byte[e.BytesTransferred];
|
|
|
|
|
Array.Copy(e.Buffer, e.Offset, data, 0, e.BytesTransferred);
|
|
|
|
|
lock (token.Buffer)
|
|
|
|
|
{
|
|
|
|
|
token.Buffer.AddRange(data);
|
|
|
|
|
}
|
|
|
|
|
//注意:你一定会问,这里为什么要用do-while循环?
|
|
|
|
|
//如果当客户发送大数据流的时候,e.BytesTransferred的大小就会比客户端发送过来的要小,
|
|
|
|
|
//需要分多次接收.所以收到包的时候,先判断包头的大小.够一个完整的包再处理.
|
|
|
|
|
//如果客户短时间内发送多个小数据包时, 服务器可能会一次性把他们全收了.
|
|
|
|
|
//这样如果没有一个循环来控制,那么只会处理第一个包,
|
|
|
|
|
//剩下的包全部留在token.Buffer中了,只有等下一个数据包过来后,才会放出一个来.
|
|
|
|
|
do
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
//如果包头不完整
|
|
|
|
|
if (token.Buffer.Count < 4)
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
//判断包的长度
|
|
|
|
|
byte[] lenBytes = token.Buffer.GetRange(0, 4).ToArray();
|
|
|
|
|
int packageLen = BitConverter.ToInt32(lenBytes, 0) - 4;
|
|
|
|
|
if (packageLen > token.Buffer.Count - 4)
|
|
|
|
|
{ //长度不够时,退出循环,让程序继续接收
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//包够长时,则提取出来,交给后面的程序去处理
|
|
|
|
|
byte[] rev = token.Buffer.GetRange(4, packageLen).ToArray();
|
|
|
|
|
//从数据池中移除这组数据
|
|
|
|
|
lock (token.Buffer)
|
|
|
|
|
{
|
|
|
|
|
token.Buffer.RemoveRange(0, packageLen + 4);
|
|
|
|
|
}
|
|
|
|
|
//将数据包交给后台处理,这里你也可以新开个线程来处理.加快速度.
|
|
|
|
|
if (ReceiveClientData != null)
|
|
|
|
|
ReceiveClientData(token, rev);
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
DataCallBackReady(token, rev);
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
//这里API处理完后,并没有返回结果,当然结果是要返回的,却不是在这里, 这里的代码只管接收.
|
|
|
|
|
//若要返回结果,可在API处理中调用此类对象的SendMessage方法,统一打包发送.不要被微软的示例给迷惑了.
|
|
|
|
|
} while (token.Buffer.Count > 4);
|
|
|
|
|
|
|
|
|
|
//继续接收. 为什么要这么写,请看Socket.ReceiveAsync方法的说明
|
|
|
|
|
if (!token.Socket.ReceiveAsync(e))
|
|
|
|
|
this.ProcessReceive(e);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
CloseClientSocket(e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (Exception xe)
|
|
|
|
|
{
|
|
|
|
|
//RuncomLib.Log.LogUtils.Info(xe.Message + "\r\n" + xe.StackTrace);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// This method is invoked when an asynchronous send operation completes.
|
|
|
|
|
// The method issues another receive on the socket to read any additional
|
|
|
|
|
// data sent from the client
|
|
|
|
|
//
|
|
|
|
|
// <param name="e"></param>
|
|
|
|
|
private void ProcessSend(SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
if (e.SocketError == SocketError.Success)
|
|
|
|
|
{
|
|
|
|
|
// done echoing data back to the client
|
|
|
|
|
AsyncUserToken token = (AsyncUserToken)e.UserToken;
|
|
|
|
|
// read the next block of data send from the client
|
|
|
|
|
bool willRaiseEvent = token.Socket.ReceiveAsync(e);
|
|
|
|
|
if (!willRaiseEvent)
|
|
|
|
|
{
|
|
|
|
|
ProcessReceive(e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
CloseClientSocket(e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void ProcessSend2(SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
if (e.SocketError == SocketError.Success)
|
|
|
|
|
{
|
|
|
|
|
// done echoing data back to the client
|
|
|
|
|
//AsyncUserToken token = (AsyncUserToken)e.UserToken;
|
|
|
|
|
// read the next block of data send from the client
|
|
|
|
|
//bool willRaiseEvent = token.Socket.ReceiveAsync(e);
|
|
|
|
|
//if (!willRaiseEvent)
|
|
|
|
|
//{
|
|
|
|
|
// ProcessReceive(e);
|
|
|
|
|
//}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
CloseClientSocket(e);
|
|
|
|
|
}
|
|
|
|
|
e.SetBuffer(null, 0, 0);
|
|
|
|
|
m_Sendpool.Push(e);
|
|
|
|
|
//Console.WriteLine("发送完毕压进回对象池");
|
|
|
|
|
|
|
|
|
|
SendForMsgPool();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//关闭客户端
|
|
|
|
|
private void CloseClientSocket(SocketAsyncEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
AsyncUserToken token = e.UserToken as AsyncUserToken;
|
|
|
|
|
|
|
|
|
|
//调用关闭连接
|
|
|
|
|
OnClose(token);
|
|
|
|
|
|
|
|
|
|
lock (m_clients) { m_clients.Remove(token); }
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
//补充处理
|
|
|
|
|
lock (_DictSocketAsyncUserToken) { _DictSocketAsyncUserToken.Remove(token.Socket); }
|
|
|
|
|
|
|
|
|
|
////尝试1
|
|
|
|
|
//m_Sendpool.Push(_DictAsyncUserTokenSendSAEA[token]);
|
|
|
|
|
//lock (_DictAsyncUserTokenSendSAEA) { _DictAsyncUserTokenSendSAEA.Remove(token); }
|
|
|
|
|
////尝试1结束
|
|
|
|
|
|
|
|
|
|
//如果有事件,则调用事件,发送客户端数量变化通知
|
|
|
|
|
if (ClientNumberChange != null)
|
|
|
|
|
ClientNumberChange(-1, token);
|
|
|
|
|
// close the socket associated with the client
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
token.Socket.Shutdown(SocketShutdown.Send);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
token.Socket.Close();
|
|
|
|
|
// decrement the counter keeping track of the total number of clients connected to the server
|
|
|
|
|
Interlocked.Decrement(ref m_clientCount);
|
|
|
|
|
m_maxNumberAcceptedClients.Release();
|
|
|
|
|
// Free the SocketAsyncEventArg so they can be reused by another client
|
|
|
|
|
e.UserToken = new AsyncUserToken();
|
|
|
|
|
m_pool.Push(e);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 对数据进行打包,然后再发送
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="token"></param>
|
|
|
|
|
/// <param name="message"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
//public void SendMessage(AsyncUserToken token, byte[] message)
|
|
|
|
|
//{
|
|
|
|
|
// if (token == null || token.Socket == null || !token.Socket.Connected)
|
|
|
|
|
// return;
|
|
|
|
|
// try
|
|
|
|
|
// {
|
|
|
|
|
// //对要发送的消息,制定简单协议,头4字节指定包的大小,方便客户端接收(协议可以自己定)
|
|
|
|
|
// byte[] buff = new byte[message.Length + 4];
|
|
|
|
|
// byte[] len = BitConverter.GetBytes(message.Length);
|
|
|
|
|
// Array.Copy(len, buff, 4);
|
|
|
|
|
// Array.Copy(message, 0, buff, 4, message.Length);
|
|
|
|
|
// //token.Socket.Send(buff); //这句也可以发送, 可根据自己的需要来选择
|
|
|
|
|
// //新建异步发送对象, 发送消息
|
|
|
|
|
// SocketAsyncEventArgs sendArg = new SocketAsyncEventArgs();
|
|
|
|
|
// sendArg.UserToken = token;
|
|
|
|
|
// sendArg.SetBuffer(buff, 0, buff.Length); //将数据放置进去.
|
|
|
|
|
// token.Socket.SendAsync(sendArg);
|
|
|
|
|
// }
|
|
|
|
|
// catch (Exception e)
|
|
|
|
|
// {
|
|
|
|
|
// //RuncomLib.Log.LogUtils.Info("SendMessage - Error:" + e.Message);
|
|
|
|
|
// }
|
|
|
|
|
//}
|
|
|
|
|
//bool flag_SendForMsgPool = false;
|
|
|
|
|
int sendrun = 0;
|
|
|
|
|
public void SendForMsgPool()
|
|
|
|
|
{
|
|
|
|
|
//if (flag_SendForMsgPool) return;
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
if (sendrun < msg_pool.Count || msg_pool.Count < 1)
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
sendrun++;
|
|
|
|
|
while (msg_pool.Count > 0)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
TokenWithMsg msg = msg_pool.Dequeue();
|
|
|
|
|
//Console.WriteLine("从信息池取出发送");
|
|
|
|
|
SendMessage(msg.token, msg.message);
|
|
|
|
|
msg = null;
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine("==============================================>");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
sendrun--;
|
|
|
|
|
Console.WriteLine("!!!!!!!!!!!!!!!!!!!!!!!!!!");
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
catch (Exception ex)
|
2021-12-10 23:38:40 +08:00
|
|
|
|
{
|
|
|
|
|
Console.WriteLine(ex.ToString());
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void SendMessage(AsyncUserToken token, byte[] message)
|
|
|
|
|
{
|
|
|
|
|
if (token == null || token.Socket == null || !token.Socket.Connected)
|
|
|
|
|
return;
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
message = SendDataWithHead(message);
|
|
|
|
|
|
|
|
|
|
//尝试2 (发送的时候从队列取,动态绑定
|
|
|
|
|
//Console.WriteLine("队列取出 并 发送!!!!");
|
|
|
|
|
if (m_Sendpool.Count > 0)
|
|
|
|
|
{
|
|
|
|
|
SocketAsyncEventArgs myreadEventArgs = m_Sendpool.Pop();
|
|
|
|
|
myreadEventArgs.UserToken = token;
|
|
|
|
|
myreadEventArgs.AcceptSocket = token.Socket;
|
|
|
|
|
myreadEventArgs.SetBuffer(message, 0, message.Length); //将数据放置进去.
|
|
|
|
|
token.Socket.SendAsync(myreadEventArgs);
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
|
|
|
|
//得了,先回去吧
|
|
|
|
|
m_Sendpool.Push(myreadEventArgs);
|
2021-12-10 23:38:40 +08:00
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
msg_pool.Enqueue(new TokenWithMsg() { token = token, message = message });
|
|
|
|
|
//Console.WriteLine("!!!!压入消息发送队列MSG_Pool");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//尝试结束
|
|
|
|
|
|
|
|
|
|
////尝试1
|
|
|
|
|
//Console.WriteLine("发送!!!!");
|
|
|
|
|
//SocketAsyncEventArgs myreadEventArgs;
|
|
|
|
|
//if (!_DictAsyncUserTokenSendSAEA.ContainsKey(token))
|
|
|
|
|
//{
|
|
|
|
|
// myreadEventArgs = m_Sendpool.Pop();
|
|
|
|
|
// myreadEventArgs.UserToken = token;
|
|
|
|
|
// _DictAsyncUserTokenSendSAEA.Add(token, myreadEventArgs);
|
|
|
|
|
// myreadEventArgs.AcceptSocket = token.Socket;
|
|
|
|
|
// //myreadEventArgs.AcceptSocket.RemoteEndPoint = token.Remote;
|
|
|
|
|
// token.IPAddress = ((IPEndPoint)(myreadEventArgs.AcceptSocket.RemoteEndPoint)).Address;
|
|
|
|
|
//}
|
|
|
|
|
//else
|
|
|
|
|
//{
|
|
|
|
|
// myreadEventArgs = _DictAsyncUserTokenSendSAEA[token];
|
|
|
|
|
//}
|
|
|
|
|
|
|
|
|
|
//myreadEventArgs.SetBuffer(message, 0, message.Length); //将数据放置进去.
|
|
|
|
|
//token.Socket.SendAsync(myreadEventArgs);
|
|
|
|
|
|
|
|
|
|
//return;
|
|
|
|
|
////尝试1结束
|
|
|
|
|
|
|
|
|
|
//新建异步发送对象, 发送消息
|
|
|
|
|
SocketAsyncEventArgs sendArg = new SocketAsyncEventArgs();
|
|
|
|
|
sendArg.UserToken = token;
|
|
|
|
|
sendArg.SetBuffer(message, 0, message.Length); //将数据放置进去.
|
|
|
|
|
token.Socket.SendAsync(sendArg);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine(e.ToString());
|
|
|
|
|
//RuncomLib.Log.LogUtils.Info("SendMessage - Error:" + e.Message);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//拼接长度
|
|
|
|
|
private static byte[] SendDataWithHead(byte[] message)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
MemoryStream memoryStream = new MemoryStream();//创建一个内存流
|
|
|
|
|
|
|
|
|
|
byte[] BagHead = BitConverter.GetBytes(message.Length + 4);//往字节数组中写入包头(包头自身的长度和消息体的长度)的长度
|
|
|
|
|
|
|
|
|
|
memoryStream.Write(BagHead, 0, BagHead.Length);//将包头写入内存流
|
|
|
|
|
|
|
|
|
|
memoryStream.Write(message, 0, message.Length);//将消息体写入内存流
|
|
|
|
|
|
|
|
|
|
byte[] HeadAndBody = memoryStream.ToArray();//将内存流中的数据写入字节数组
|
|
|
|
|
|
|
|
|
|
memoryStream.Close();//关闭内存
|
|
|
|
|
memoryStream.Dispose();//释放资源
|
|
|
|
|
|
|
|
|
|
return HeadAndBody;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#region
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// 用于调用者回调的虚函数
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="data"></param>
|
|
|
|
|
public virtual void DataCallBack(AsyncUserToken sk, int CMDID, byte[] data)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 断开连接
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="sk"></param>
|
|
|
|
|
public virtual void OnClose(AsyncUserToken sk)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public virtual void OnCloseReady(AsyncUserToken token)
|
|
|
|
|
{
|
|
|
|
|
OnClose(token);
|
|
|
|
|
|
|
|
|
|
lock (m_clients) { m_clients.Remove(token); }
|
|
|
|
|
|
|
|
|
|
//补充处理
|
|
|
|
|
lock (_DictSocketAsyncUserToken) { _DictSocketAsyncUserToken.Remove(token.Socket); }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//如果有事件,则调用事件,发送客户端数量变化通知
|
|
|
|
|
if (ClientNumberChange != null)
|
|
|
|
|
ClientNumberChange(-1, token);
|
|
|
|
|
// close the socket associated with the client
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
token.Socket.Shutdown(SocketShutdown.Send);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
token.Socket.Close();
|
|
|
|
|
// decrement the counter keeping track of the total number of clients connected to the server
|
|
|
|
|
Interlocked.Decrement(ref m_clientCount);
|
|
|
|
|
m_maxNumberAcceptedClients.Release();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//试着加入一个释放
|
|
|
|
|
//token.Socket.Dispose();
|
|
|
|
|
|
|
|
|
|
// Free the SocketAsyncEventArg so they can be reused by another client
|
|
|
|
|
//e.UserToken = new AsyncUserToken();
|
|
|
|
|
//m_pool.Push(e);
|
|
|
|
|
//这里直接注释了进程池,需要验证是否会出问题
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 发送数据并计数
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="data"></param>
|
|
|
|
|
private void SendWithIndex(AsyncUserToken token, byte[] data)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
//发送数据
|
|
|
|
|
SendMessage(token, data);
|
|
|
|
|
|
|
|
|
|
token.SendIndex = MaxSendIndexNum;
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
OnCloseReady(token);
|
|
|
|
|
}
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
public AsyncUserToken GetAsyncUserTokenForSocket(Socket sk)
|
|
|
|
|
{
|
|
|
|
|
return _DictSocketAsyncUserToken.ContainsKey(sk) ? _DictSocketAsyncUserToken[sk] : null;
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// 对外暴露的发送消息
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="CMDID"></param>
|
|
|
|
|
/// <param name="data">序列化之后的数据</param>
|
|
|
|
|
public void SendToSocket(Socket sk, int CMDID, int ERRCODE, byte[] data)
|
|
|
|
|
{
|
|
|
|
|
AsyncUserToken token = GetAsyncUserTokenForSocket(sk);
|
|
|
|
|
HunterNet_S2C _s2cdata = new HunterNet_S2C();
|
|
|
|
|
_s2cdata.HunterNetCore_CmdID = CMDID;
|
|
|
|
|
_s2cdata.HunterNetCore_Data = data;
|
|
|
|
|
_s2cdata.HunterNetCore_ERRORCode = ERRCODE;
|
|
|
|
|
byte[] _finaldata = Serizlize<HunterNet_S2C>(_s2cdata);
|
|
|
|
|
SendWithIndex(token, _finaldata);
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// 发送心跳包
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="sk"></param>
|
|
|
|
|
///
|
|
|
|
|
private void SendHeartbeat(AsyncUserToken token)
|
|
|
|
|
{
|
|
|
|
|
if (token == null || token.Socket == null || !token.Socket.Connected)
|
|
|
|
|
return;
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine(DateTime.Now.ToString() + "发送心跳包");
|
|
|
|
|
token.SendIndex = MaxSendIndexNum;
|
|
|
|
|
|
|
|
|
|
//新建异步发送对象, 发送消息
|
|
|
|
|
SocketAsyncEventArgs sendArg = new SocketAsyncEventArgs();
|
|
|
|
|
sendArg.UserToken = token;
|
|
|
|
|
sendArg.SetBuffer(HeartbeatData, 0, HeartbeatData.Length); //将数据放置进去.
|
|
|
|
|
token.Socket.SendAsync(sendArg);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
|
|
|
|
OnCloseReady(token);
|
|
|
|
|
}
|
|
|
|
|
}
|
2021-12-27 23:52:27 +08:00
|
|
|
|
|
2021-12-10 23:38:40 +08:00
|
|
|
|
private void DataCallBackReady(AsyncUserToken sk, byte[] data)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
//增加接收计数
|
|
|
|
|
sk.RevIndex = MaxRevIndexNum;
|
|
|
|
|
|
|
|
|
|
if (data.Length == 1 && data[0] == 0x00)//心跳包
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine("收到心跳包");
|
|
|
|
|
//无处理
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
HunterNet_C2S _s2c = DeSerizlize<HunterNet_C2S>(data);
|
|
|
|
|
DataCallBack(sk, (int)_s2c.HunterNetCore_CmdID, _s2c.HunterNetCore_Data);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine("数据解析错误");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 心跳包时钟事件
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="sender"></param>
|
|
|
|
|
/// <param name="e"></param>
|
|
|
|
|
private void CheckUpdatetimer_Elapsed(object sender, System.Timers.ElapsedEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
for (int i = 0; i < m_clients.Count(); i++)
|
|
|
|
|
{
|
|
|
|
|
//Console.WriteLine("RevIndex->{0} SendIndex->{1}", m_clients[i].RevIndex, m_clients[i].SendIndex);
|
|
|
|
|
//接收服务器数据计数
|
|
|
|
|
m_clients[i].RevIndex--;
|
|
|
|
|
if (m_clients[i].RevIndex <= 0)
|
|
|
|
|
{
|
|
|
|
|
//判定掉线
|
|
|
|
|
OnCloseReady(m_clients[i]);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//发送计数
|
|
|
|
|
m_clients[i].SendIndex--;
|
|
|
|
|
if (m_clients[i].SendIndex <= 0)//需要发送心跳包了
|
|
|
|
|
{
|
|
|
|
|
//重置倒计时计数
|
|
|
|
|
m_clients[i].SendIndex = MaxSendIndexNum;
|
|
|
|
|
SendHeartbeat(m_clients[i]);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
public static byte[] Serizlize<T>(T MsgObj)
|
|
|
|
|
{
|
|
|
|
|
using (MemoryStream ms = new MemoryStream())
|
|
|
|
|
{
|
|
|
|
|
Serializer.Serialize<T>(ms, MsgObj);
|
|
|
|
|
byte[] data1 = ms.ToArray();
|
|
|
|
|
return data1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static T DeSerizlize<T>(byte[] MsgObj)
|
|
|
|
|
{
|
|
|
|
|
using (MemoryStream ms = new MemoryStream(MsgObj))
|
|
|
|
|
{
|
|
|
|
|
var ds_obj = Serializer.Deserialize<T>(ms);
|
|
|
|
|
//ds_obj = MySet(ds_obj);
|
|
|
|
|
return ds_obj;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|