Common.TcpLibTcpServerWIOCP

using System;
using System.Text;
using System.Collections;
using System.Collections.Generic;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using System.Runtime.InteropServices;
using System.IO;

namespace Common.TcpLib
{
   
/// <summary>
/// WIOCP Socket处理服务器端
/// </summary>
    public class TcpServerWIOCP : Common.TcpLib.ITcpServer
    {
        private Socket _listener;
        private string _listenerIp = "127.0.0.1";
        private int _port;
        private int _backlog = 100;
        private int _bufferSize = 1024;
        private Thread _thread = null;
        private bool _disposed;
        private bool _debug = false;
        private ManualResetEvent allDone = new ManualResetEvent(false);
        private ManualResetEvent sendDone = new ManualResetEvent(false);
        private object _stopLock = new object();
        private bool _sotpMessage = false;

        private List<Socket> _cs = new List<Socket>();

        #region define delegates
        /// <summary>
        /// 服务器绑定后处理事件 
        /// </summary>
        public _Bgz_OnBindEventDelegate FOnBindEventDelegate;

        /// <summary>
        /// Accept后处理事件
        /// </summary>
        public _Bgz_OnAcceptEventDelegate FOnAcceptEventDelegate;

        /// <summary>
        /// 接收到客户端数据后处理事件
        /// </summary>
        public _Bgz_OnReceiveBeginEventDelegate FOnReceiveBeginEventDelegate;
        public _Bgz_OnReceiveingEventDelegate FOnReceiveingEventDelegate;
        public _Bgz_OnReceiveEndEventDelegate FOnReceiveEndEventDelegate;

        /// <summary>
        /// 报错信息处理
        /// </summary>
        public _Bgz_OnErrorEventDelegate FOnErrorEventDelegate;

        /// <summary>
        /// 断掉与客户机的连接前处理事件 
        /// </summary>
        public _Bgz_OnDropConnectEventDelegate FOnDropConnectEventDelegate;
        #endregion

        #region Event
        private void OnBindEvent(_Bgz_ConnectionState state)
        {
            if (FOnBindEventDelegate != null) FOnBindEventDelegate(state);
        }
        private void OnAcceptEvent(_Bgz_ConnectionState state)
        {
            if (FOnAcceptEventDelegate != null) FOnAcceptEventDelegate(state);
        }
        private void OnReceiveBeginEvent(_Bgz_ConnectionState state)
        {
            if (FOnReceiveBeginEventDelegate != null) FOnReceiveBeginEventDelegate(state);
        }
        private void OnReceiveingEvent(_Bgz_ConnectionState state)
        {
            if (FOnReceiveingEventDelegate != null) FOnReceiveingEventDelegate(state);
        }
        private void OnReceiveEndEvent(_Bgz_ConnectionState state)
        {
            if (FOnReceiveEndEventDelegate != null) FOnReceiveEndEventDelegate(state);
        }
        private void OnErrorEvent(ErrorType errortype, string msg, _Bgz_ConnectionState state)
        {
            if (FOnErrorEventDelegate != null) FOnErrorEventDelegate(errortype, msg, state);
        }
        private void OnDropConnectEvent(_Bgz_ConnectionState state)
        {
            if (FOnDropConnectEventDelegate != null) FOnDropConnectEventDelegate(state);
        }
        #endregion

        #region Constructor and Destructor

        public TcpServerWIOCP(int port)
        {
            _port = port;
        }

        public TcpServerWIOCP(int port, int backlog)
        {
            _port = port;
            _backlog = backlog;
        }

        public TcpServerWIOCP(int port, int backlog, int buffterSize)
        {
            _port = port;
            _backlog = backlog;
            _bufferSize = buffterSize;
        }

        public TcpServerWIOCP(string listenerIp, int port)
        {
            _listenerIp = listenerIp;
            _port = port;
        }

        public TcpServerWIOCP(string listenerIp, int port, int backlog)
        {
            _listenerIp = listenerIp;
            _port = port;
            _backlog = backlog;
        }

        public TcpServerWIOCP(string listenerIp, int port, int backlog, int buffterSize)
        {
            _listenerIp = listenerIp;
            _port = port;
            _backlog = backlog;
            _bufferSize = buffterSize;
        }

        ~TcpServerWIOCP()
        {
            if (!_disposed)
            {
                Stop();
            }
        }
        #endregion

        #region Private Methods

        private void Dispose()
        {
            if (!_disposed)
            {
                Stop();
                GC.Collect();
                GC.WaitForPendingFinalizers();
                this.FOnAcceptEventDelegate = null;
                this.FOnBindEventDelegate = null;
                this.FOnDropConnectEventDelegate = null;
                this.FOnErrorEventDelegate = null;
                this.FOnReceiveBeginEventDelegate = null;
                this.FOnReceiveEndEventDelegate = null;
                this.FOnReceiveingEventDelegate = null;
                _disposed = true;
            }
        }

        private void StartListening(object obj)
        {
            try
            {
                bool optionValue = (bool)obj;
                _listener = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);

                #region 处理Socket端口复用
                _listener.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, optionValue);
                #endregion

                _listener.Bind(new IPEndPoint(IPAddress.Parse(_listenerIp), _port));
                _listener.Listen(_backlog);

                _Bgz_ConnectionState stx = new _Bgz_ConnectionState(_listener);
                OnBindEvent(stx);

                _sotpMessage = false;
                _stopLock = new object();
                while (true)
                {
                    if (_listener == null)
                    {
                        lock (_stopLock)
                        {
                            if (!_sotpMessage)
                            {
                                OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                                _sotpMessage = true;
                            }
                        }
                        return;
                    }
                    allDone.Reset();
                    _listener.BeginAccept(new AsyncCallback(AcceptCallback), _listener);
                    allDone.WaitOne();
                }

            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "Start Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        private void AcceptCallback(IAsyncResult ar)
        {
            Socket handler = null;
            _Bgz_ConnectionState stx = new _Bgz_ConnectionState();
            try
            {
                Socket listener = ar.AsyncState as Socket;
                if (_listener == null)
                {
                    lock (_stopLock)
                    {
                        if (!_sotpMessage)
                        {
                            OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                            _sotpMessage = true;
                        }
                    }
                    return;
                }
                handler = listener.EndAccept(ar);
                _cs.Add(handler);
                stx._conn = handler;
                OnAcceptEvent(stx);
               
                #region 处理Socket网络异常断开
                uint dummy = 0;
                byte[] inOptionValues = new byte[Marshal.SizeOf(dummy) * 3];
                BitConverter.GetBytes((uint)1).CopyTo(inOptionValues, 0);
                BitConverter.GetBytes((uint)50000).CopyTo(inOptionValues, Marshal.SizeOf(dummy));
                BitConverter.GetBytes((uint)50000).CopyTo(inOptionValues, Marshal.SizeOf(dummy) * 2);
                _listener.IOControl(IOControlCode.KeepAliveValues, inOptionValues, null);
                #endregion

                handler.BeginReceive(stx._buffer, 0, 0, 0, new AsyncCallback(ReadCallBack), handler);
            }
            catch (Exception ex)
            {
                DropConnection(stx);
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "AcceptCallback Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
            finally
            {
                allDone.Set();
            }
        }

        private void DropConnection(_Bgz_ConnectionState stx)
        {
            try
            {
                lock (this)
                {
                    if (_listener == null)
                    {
                        lock (_stopLock)
                        {
                            if (!_sotpMessage)
                            {
                                OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                                _sotpMessage = true;
                            }
                        }
                        return;
                    }
                    OnDropConnectEvent(stx);
                    if (_cs.Contains(stx._conn))
                        _cs.Remove(stx._conn);

                    if (stx._conn != null)
                    {
                        IPEndPoint info = (IPEndPoint)stx._conn.RemoteEndPoint;
                        string str = info.Address.ToString() + ":" + info.Port.ToString();
                        OnErrorEvent(ErrorType.DisAccept, "关闭了远程连接[" + str + "]！", null);
                        if (stx._conn.Connected)
                            stx._conn.Close();
                    }
                }
            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "DropConnection Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        private void ReadCallBack(IAsyncResult ar)
        {
            Socket handler = null;
            try
            {

                handler = (Socket)ar.AsyncState;
                _Bgz_ConnectionState stx = new _Bgz_ConnectionState(handler);

                 if (this._listener == null)
                {
                    lock (_stopLock)
                    {
                        if (!_sotpMessage)
                        {
                            OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                            _sotpMessage = true;
                        }
                    }
                    return;
                }

                if (handler == null) return;

                handler.EndReceive(ar);

                if (handler.Available == 0)
                {
                    DropConnection(stx);
                    return;
                }

                try
                {
                    lock (this)
                    {
                        OnReceiveBeginEvent(stx);

                        #region receive

                        stx._buffer = new byte[0];
                        stx._count = 0;
                        stx._dataStream.SetLength(0);
                        stx._dataStream.Position = 0;
                        if (stx._getonceall)
                        {
                            stx._buffer = new byte[handler.Available];
                            int ret = handler.Receive(stx._buffer, 0, stx._buffer.Length, SocketFlags.None);
                            if (ret > 0)
                            {
                                stx._dataStream.Write(stx._buffer, 0, stx._buffer.Length);
                                stx._count++;
                                OnReceiveingEvent(stx);
                            }
                        }
                        else
                        {
                            while (handler.Available > 0)
                            {
                                if (handler.Available > _bufferSize)
                                    stx._buffer = new byte[_bufferSize];
                                else
                                    stx._buffer = new byte[handler.Available];
                                int ret = handler.Receive(stx._buffer, 0, stx._buffer.Length, SocketFlags.None);
                                if (ret > 0)
                                {
                                    stx._dataStream.Write(stx._buffer, 0, stx._buffer.Length);
                                    stx._count++;
                                    OnReceiveingEvent(stx);
                                }
                            }
                        }
                        #endregion

                        OnReceiveEndEvent(stx);
                    }
                }
                catch (Exception ex)
                {
                    DropConnection(stx);
                    if (Debug)
                    {
                        OnErrorEvent(ErrorType.Catch, "ReadCallBack.1 Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", stx);
                    }
                }

                if (handler == null) return;
                if (handler.Connected)
                    handler.BeginReceive(stx._buffer, 0, 0, 0, new AsyncCallback(ReadCallBack), handler);
            }
            catch (Exception ex)
            {
                if (this._listener == null)
                {
                    lock (_stopLock)
                    {
                        if (!_sotpMessage)
                        {
                            OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                            _sotpMessage = true;
                        }
                    }
                    return;
                }

                if (handler != null)
                {
                    _Bgz_ConnectionState stx = new _Bgz_ConnectionState(handler);
                    DropConnection(stx);
                }
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "ReadCallBack.2 Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        private void SendCallback(IAsyncResult ar)
        {
            try
            {
                Socket client = (Socket)ar.AsyncState;
                int bytesSent = client.EndSend(ar);
            }
            catch (Exception ex)
            {
                OnErrorEvent(ErrorType.Catch, "出错信息提示:\r\n" + ex.Message, null);
            }
            finally
            {
                sendDone.Set();
            }
        }

        #endregion

        #region Public Methods
        public void Start()
        {
            Start(false);
        }

        public void Start(bool optionValue)
        {
            _thread = new Thread(new ParameterizedThreadStart(StartListening));
            _thread.Name = "TcpServerIOCP.Listening";
            _thread.Start(optionValue);
        }

        public void Send(byte[] msg)
        {
            try
            {
                foreach (Socket s in ClientSockets)
                {
                    Send(s, msg);
                }
            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "Send Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        public void Send(Socket ClientSocket, byte[] msg)
        {
            try
            {
                if (ClientSocket != null)
                {
                    if (ClientSocket.Connected)
                        ClientSocket.Send(msg);
                    else
                    {
                        IPEndPoint info = (IPEndPoint)ClientSocket.RemoteEndPoint;
                        string str = info.Address.ToString() + ":" + info.Port.ToString();
                        OnErrorEvent(ErrorType.DisConnect, "客户端[" + str + "]的连接未建立，不能发送信息！", null);
                        ClientSocket.Close();
                    }
                }
                else
                {
                    OnErrorEvent(ErrorType.DisAccept, "客户端的连接未建立，不能发送信息！", null);
                }
            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "Send Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        public void SendAsync(Socket ClientSocket , byte[] msg)
        {
            try
            {
                if (ClientSocket.Connected == false)
                {
                    throw (new Exception("没有连接客户端不可以发送信息!"));
                }

                byte[] buffer;
                int count = msg.Length;
                int cur = 0;
                while (count > 0)
                {
                    if (count > this._bufferSize)
                        buffer = new byte[this._bufferSize];
                    else
                        buffer = new byte[count];

                    for (int i = 0; i < buffer.Length; i++)
                    {
                        buffer[i] = msg[cur + i];
                    }
                    count -= buffer.Length;

                    cur += buffer.Length;

                    ClientSocket.BeginSend(buffer, 0, buffer.Length, 0, new AsyncCallback(SendCallback), ClientSocket);
                    sendDone.WaitOne();
                }
            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "Send Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        public void SendAsync(Socket ClientSocket, Stream Astream)
        {
            try
            {
                if (ClientSocket.Connected == false)
                {
                    throw (new Exception("没有连接客户端不可以发送信息!"));
                }
               
                byte[] buffer ;
                long count = Astream.Length;
                while (count > 0)
                {
                    if (count > this._bufferSize)
                        buffer = new byte[this._bufferSize];
                    else
                        buffer = new byte[count];

                    Astream.Read(buffer, 0, buffer.Length);
                    count -= buffer.Length;
                    ClientSocket.BeginSend(buffer, 0, buffer.Length, 0, new AsyncCallback(SendCallback), ClientSocket);
                    sendDone.WaitOne();
                }
            }
            catch (Exception ex)
            {
                if (Debug)
                {
                    OnErrorEvent(ErrorType.Catch, "Send Error！[Message]:\r\n" + ex.Message + "[StackTrace]:\r\n" + ex.StackTrace + "\r\n", null);
                }
            }
        }

        public void Stop()
        {
            try
            {
                if (_listener == null)
                {
                    lock (_stopLock)
                    {
                        if (!_sotpMessage)
                        {
                            OnErrorEvent(ErrorType.DisAccept, "服务器关闭了监听！", null);
                            _sotpMessage = true;
                        }
                    }
                    return;
                }
                if (_thread != null)
                {
                    foreach (Socket s in _cs)
                    {
                        if (s != null)
                            if (s.Connected)
                            {

                                IPEndPoint info = (IPEndPoint)s.RemoteEndPoint;
                                string str = info.Address.ToString() + ":" + info.Port.ToString();
                                OnErrorEvent(ErrorType.DisAccept, "关闭了远程连接[" + str + "]！", null);
                                s.Close();
                            }
                    }

                    _cs.Clear();
                    _listener.Close();
                    _listener = null;
                    _thread.Join();
                    _thread = null;
                }
            }
            catch (Exception ex)
            {
                OnErrorEvent(ErrorType.Catch, "出错信息提示:\r\n" + ex.Message, null);
            }
        }

        #endregion

        #region Property
        /// <summary>
        /// 获取挂起连接队列的最大长度
        /// </summary>
        public int BackLog
        {
            get
            {
                return _backlog;
            }
        }

        public int BufferSize
        {
            get
            {
                return _bufferSize;
            }
        }

        public Socket[] ClientSockets
        {
            get
            {
                Socket[] sockets = new Socket[this._cs.Count];
                _cs.CopyTo(sockets);
                return sockets;
            }
        }

        public bool Debug
        {
            get
            {
                return _debug;
            }
            set
            {
                _debug = value;
            }
        }
        #endregion
    }
}