- Initial commit

Runtime/Implementations/Utils/LockedBufferSegmenStream.cs

@@ -0,0 +1,56 @@
+#if (!UNITY_WEBGL || UNITY_EDITOR) && !BESTHTTP_DISABLE_ALTERNATE_SSL
+using Best.HTTP.Shared.PlatformSupport.Memory;
+using Best.HTTP.Shared.Streams;
+
+namespace Best.WebSockets.Implementations.Utils
+{
+    public sealed class LockedBufferSegmenStream : BufferSegmentStream
+    {
+        public bool IsClosed { get; private set; }
+
+        public override int Read(byte[] buffer, int offset, int count)
+        {
+            lock (base.bufferList)
+            {
+                if (this.IsClosed && base.bufferList.Count == 0)
+                    return 0;
+
+                int sumReadCount = base.Read(buffer, offset, count);
+
+                return sumReadCount == 0 ? -1 : sumReadCount;
+            }
+        }
+
+        public override void Write(BufferSegment bufferSegment)
+        {
+            lock (base.bufferList)
+            {
+                if (this.IsClosed)
+                    return;
+
+                base.Write(bufferSegment);
+            }
+        }
+
+        public override void Reset()
+        {
+            lock (base.bufferList)
+            {
+                base.Reset();
+            }
+        }
+
+        protected override void Dispose(bool disposing)
+        {
+            base.Dispose(disposing);
+
+            Reset();
+        }
+
+        public override void Close()
+        {
+            this.IsClosed = true;
+        }
+    }
+}
+#endif