初始化上传

常用工具集/Utility/Network/Modbus/SharpModbus/ModbusSocketStream.cs

@@ -0,0 +1,56 @@
+using System;
+using System.Threading;
+using System.Net.Sockets;
+
+namespace SharpModbus
+{
+    public class ModbusSocketStream : IModbusStream
+    {
+        private readonly TcpClient socket;
+        private readonly Action<char, byte[], int> monitor;
+
+        public ModbusSocketStream(TcpClient socket, int timeout, Action<char, byte[], int> monitor = null)
+        {
+            socket.ReceiveTimeout = timeout;
+            socket.SendTimeout = timeout;
+            this.monitor = monitor;
+            this.socket = socket;
+        }
+
+        public void Dispose()
+        {
+            Tools.Dispose(socket);
+        }
+
+        public void Write(byte[] data)
+        {
+            //implicit discard to allow retry after timeout as per #5
+            while (socket.Available > 0) socket.GetStream().ReadByte();
+            if (monitor != null) monitor('>', data, data.Length);
+            socket.GetStream().Write(data, 0, data.Length);
+        }
+
+        public int Read(byte[] data)
+        {
+            var count = 0;
+            var dl = DateTime.Now.AddMilliseconds(socket.ReceiveTimeout);
+            while (count < data.Length)
+            {
+                var available = socket.Available;
+                if (available == 0)
+                {
+                    if (DateTime.Now > dl) break;
+                    Thread.Sleep(1);
+                }
+                else
+                {
+                    var size = (int)Math.Min(available, data.Length - count);
+                    count += socket.GetStream().Read(data, count, size);
+                    dl = DateTime.Now; //should come in single packet
+                }
+            }
+            if (monitor != null) monitor('<', data, count);
+            return count;
+        }
+    }
+}