ServerTCP.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace FLY.Modbus.WithThread
{
    public class ServerTCP : IModbusServer,INotifyPropertyChanged
    {
        /// <summary>
        /// 通信超时, client 长时间连接上，但不通讯，会被断开
        /// </summary>
        public TimeSpan Timeout { get; set; } = TimeSpan.FromMinutes(1);

        /// <summary>
        /// 本地地址
        /// </summary>
        public IPEndPoint LocalEP { get; set; }

        /// <summary>
        /// 当前连接数
        /// </summary>
        public int ConnectedCnt { get; private set; }

        /// <summary>
        /// 工作中
        /// </summary>
        public bool IsRunning { get; set; }

        /// <summary>
        /// socket 错误信息
        /// </summary>
        public string ErrMsg { get; set; }

        /// <summary>
        /// 
        /// </summary>
        public GetValueHander GetValue { get; private set; }

        /// <summary>
        /// 
        /// </summary>
        public SetValueHander SetValue { get; private set; }

        Socket listen_socket;
        List<Client> clients = new List<Client>();

        CancellationTokenSource cts;

        public ServerTCP(IPEndPoint ep, GetValueHander getValue, SetValueHander setValue)
        {
            LocalEP = ep;
            GetValue = getValue;
            SetValue = setValue;
        }

        public void Start()
        {
            lock (this)
            {
                if (IsRunning)
                    return;
                IsRunning = true;
                cts = new CancellationTokenSource();
                Task.Factory.StartNew(OnPoll_listen, cts.Token);
            }
        }
        public void Stop()
        {
            lock (this)
            {
                if (!IsRunning)
                    return;
                cts.Cancel();
                listen_socket.Close();

                while (true)
                {
                    Client client;
                    lock (clients)
                    {
                        if (clients.Count() == 0)
                            break;
                        client = clients.First();
                        clients.Remove(client);
                    }
                    client.Close();
                }
            }
        }
        void OnPoll_listen()
        {
            listen_socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
            listen_socket.Bind(LocalEP);
            listen_socket.Listen(5);

            while (true)
            {
                if (cts.IsCancellationRequested)
                    goto _end;
                try
                {
                    Socket socket = listen_socket.Accept();
                    Client client = new Client(socket, this);
                    client.Closed += Client_Closed;
                    client.Start();
                    lock (client)
                    {
                        clients.Add(client);
                        ConnectedCnt = clients.Count();
                    }
                }
                catch (Exception e)
                {
                    goto _end;
                }
            }

        _end:
            IsRunning = false;
        }

        private void Client_Closed(object obj)
        {
            lock (clients)
            {
                clients.Remove(obj as Client);
            }
        }

        public delegate void ClosedHandler(object obj);
        class Client
        {


            public Socket socket;
            public CancellationTokenSource cts;
            ServerTCP server;
            public Client(Socket socket, ServerTCP server)
            {
                this.socket = socket;
                this.server = server;
            }
            public void Start()
            {
                cts = new CancellationTokenSource();
                Task.Factory.StartNew(OnPoll_client, cts.Token);
            }

            bool PullRequest(out Pack_Proto request)
            {
                //格式——————————————                
                //MBAP
                //域                         长度          描述
                //事务标识符         2byte           MODBUS请求/响应的识别码    
                //协议标识符         2byte           0=MODBUS协议
                //——长度—         2byte           以下字节的数量
                //单元标识符         1byte           串行链路或其它总线过来的识别码


                //PDU
                //功能码——         1byte
                //数据                    nbyte


                byte[] recvBytes = new byte[socket.ReceiveBufferSize];
                socket.ReceiveTimeout = (int)server.Timeout.TotalMilliseconds;
                request = new Pack_Proto();
                int offset = 0;
                //读取数据
                while (true)
                {
                    int size = recvBytes.Length - offset;

                    if (size <= 0)
                    {
                        //异常
                        server.ErrMsg = "client error: rec to many!!!";
                        return false;
                    }

                    int bytes = 0;
                    try
                    {
                        bytes = socket.Receive(recvBytes, offset, size, SocketFlags.None);//从客户端接受信息,可能会分了多次接收
                    }
                    catch (SocketException e)
                    {
                        //异常
                        server.ErrMsg = $"client error: {e.Message}";
                        return false;
                    }
                    if (bytes == 0)
                    {
                        //超时，或已经断开连接
                        server.ErrMsg = "client error: TimeOut";
                        return false;
                    }
                    //肯定只回复一条，只需要翻译一条
                    if (!request.TryParse(recvBytes, 0, bytes, out int rlen))
                    {
                        //处理失败,数据不够，继续接收
                        offset = bytes;
                    }
                    else
                    {
                        //完成！！！！
                        return true;
                    }
                }
            }

            bool PushReponse(Pack_Proto reponse)
            {
                try
                {
                    socket.Send(reponse.ToBytes());
                }
                catch (Exception e)
                {
                    server.ErrMsg = e.Message;
                    return false;
                }
                return true;
            }

            void OnPoll_client()
            {
                while (true)
                {
                    if(cts.IsCancellationRequested)
                        goto _error;

                    bool ret = PullRequest(out Pack_Proto request);
                    if(!ret)
                        goto _error;

                    ParsePacket(request, out Pack_Proto reponse);

                    
                    ret = PushReponse(reponse);
                    if (!ret)
                        goto _error;
                    
                }
            _error:
                Close();
            }

            public void Close()
            {
                cts.Cancel();
                socket.Close();

                Closed?.Invoke(this);
            }
            public event ClosedHandler Closed;


            void ParsePacket(Pack_Proto request, out Pack_Proto reponse)
            {
                switch (request.func)
                {
                    case 0x03:      // read holding registers
                        Do_03(request, out reponse);
                        break;
                    case 0x01:      // read coils
                        Do_01(request, out reponse);
                        break;
                    case 0x10:      // write multiple registers
                        Do_10(request, out reponse);
                        break;
                    case 0x06:      // Write single register
                        Do_06(request, out reponse);
                        break;
                    case 0x05:      // Write single coil
                        Do_05(request, out reponse);
                        break;
                    case 0x0f:      // Write mutiple coils
                        Do_0f(request, out reponse);
                        break;
                    default:
                        Do_Exception(request, 0x01, out reponse);
                        break;
                }
            }

            #region 处理
            /// <summary>
            /// 异常处理
            /// </summary>
            void Do_Exception(Pack_Proto request, byte e_code, out Pack_Proto reponse)
            {
                byte func = request.func;
                Misc.MyBase.SIGNBIT(ref func, 7);
                reponse = new Pack_Proto()
                {
                    tranid = request.tranid,
                    unitid = request.unitid,
                    func = func,
                    buf = new byte[1] { e_code }
                };
            }
            #region 功能
            /// <summary>
            /// 读多个 COIL
            /// </summary>
            void Do_01(Pack_Proto request, out Pack_Proto reponse)
            {
                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 num = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;

                if ((num < 0x01) || (num > 0x07D0))
                {
                    Do_Exception(request, 0x03, out reponse);
                    return;
                }
                bool[] value = new bool[num];
                server.GetValue(start, value);
                
                List<byte> data = new List<byte>();
                data.Add((byte)(Math.Ceiling(num / 8.0)));//返回的字节数

                for (int i = 0, j = 0; i < num; i++)
                {
                    if (j == 0)
                        data.Add(0);

                    if (value[i])
                    {
                        data[data.Count - 1] = (byte)(data[data.Count - 1] | Misc.MyBase.BIT(i % 8));
                    }
                    j++;
                    if (j == 8)
                    {
                        j = 0;
                    }
                }

                reponse = new Pack_Proto()
                {
                    tranid = request.tranid,
                    unitid = request.unitid,
                    func = request.func,
                    buf = data.ToArray()
                };
            }


            /// <summary>
            /// 读多个REG
            /// </summary>
            /// <param name="request"></param>
            /// <param name="reponse"></param>
            void Do_03(Pack_Proto request, out Pack_Proto reponse)
            {

                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 num = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;





                if ((num < 0x01) || (num > 0x007D))
                {
                    Do_Exception(request, 0x03, out reponse);
                    return;
                }


                UInt16[] value = new UInt16[num];
                server.GetValue(start, value);
                

                List<byte> data = new List<byte>();
                data.Add((byte)(num * 2));
                for (int i = 0; i < num; i++)
                {
                    data.AddRange(value[i].GetBytes_Big_endian());
                }

                reponse = new Pack_Proto()
                {
                    tranid = request.tranid,
                    unitid = request.unitid,
                    func = request.func,
                    buf = data.ToArray()
                };
            }

            /// <summary>
            /// 写单个coil
            /// </summary>
            void Do_05(Pack_Proto request, out Pack_Proto reponse)
            {
                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 value = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;


                if ((value != 0xff00) && (value != 0x0000))
                {
                    Do_Exception(request, 0x03, out reponse);
                    return;
                }



                bool[] v = new bool[1];

                if (value == 0xff00)
                    v[0] = true;
                else
                    v[0] = false;
                
                server.SetValue(start, v);

                reponse = request;
            }
            /// <summary>
            /// 写单个REG
            /// </summary>
            void Do_06(Pack_Proto request, out Pack_Proto reponse)
            {

                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 value = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;

                UInt16[] v = new UInt16[1];
                v[0] = value;
                server.SetValue(start, v);

                reponse = request;
            }

            /// <summary>
            /// 写多个coil
            /// </summary>
            /// <param name="request"></param>
            /// <param name="reponse"></param>
            void Do_0f(Pack_Proto request, out Pack_Proto reponse)
            {
                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 num = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;

                if ((num < 0x01) || (num > 0x007D))
                {
                    Do_Exception(request, 0x03, out reponse);
                    return;
                }


                byte byte_cnt = request.buf[index];
                index++;

                bool[] v = new bool[num];
                for (int i = 0, j = 0; i < num; i++)
                {
                    if (Misc.MyBase.CHECKBIT(request.buf[index], j))
                        v[i] = true;
                    else
                        v[i] = false;
                    j++;
                    if (j == 8)
                    {
                        j = 0;
                        index++;
                    }
                }

                server.SetValue(start, v);

                List<byte> data = new List<byte>();
                data.AddRange(start.GetBytes_Big_endian());
                data.AddRange(num.GetBytes_Big_endian());

                reponse = new Pack_Proto()
                {
                    tranid = request.tranid,
                    unitid = request.unitid,
                    func = request.func,
                    buf = data.ToArray()
                };
            }


            /// <summary>
            /// 写多个REG
            /// </summary>
            /// <param name="request"></param>
            /// <param name="reponse"></param>
            void Do_10(Pack_Proto request, out Pack_Proto reponse)
            {
                int index = 0;
                //开始地址
                UInt16 start = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                //数量
                UInt16 num = COMMON.ToUInt16_Big_endian(request.buf, index);
                index += 2;
                byte byte_cnt = request.buf[index];
                index++;

                if ((num < 0x01) || (num > 0x007D))
                {
                    Do_Exception(request, 0x03, out reponse);
                    return;
                }



                UInt16[] v = new UInt16[num];
                for (int i = 0; i < num; i++)
                {
                    v[i] = COMMON.ToUInt16_Big_endian(request.buf, index);
                    index += 2;
                }

                server.SetValue(start, v);

                List<byte> data = new List<byte>();
                data.AddRange(start.GetBytes_Big_endian());
                data.AddRange(num.GetBytes_Big_endian());

                reponse = new Pack_Proto()
                {
                    tranid = request.tranid,
                    unitid = request.unitid,
                    func = request.func,
                    buf = data.ToArray()
                };
            }
            #endregion
            #endregion
        }





        public event PropertyChangedEventHandler PropertyChanged;
    }
}