Imported Upstream version 4.6.0.125

Former-commit-id: a2155e9bd80020e49e72e86c44da02a8ac0e57a4

mcs/class/referencesource/System.ServiceModel/System/ServiceModel/Channels/TimeoutStream.cs

@@ -0,0 +1,70 @@
+//------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+//------------------------------------------------------------
+namespace System.ServiceModel.Channels
+{
+    using System.IO;
+    using System.Runtime;
+
+    // Enforces an overall timeout based on the TimeoutHelper passed in
+    class TimeoutStream : DelegatingStream
+    {
+        TimeoutHelper timeoutHelper;
+        public TimeoutStream(Stream stream, ref TimeoutHelper timeoutHelper)
+            : base(stream)
+        {
+            if (!stream.CanTimeout)
+            {
+                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgument("stream", SR.GetString(SR.StreamDoesNotSupportTimeout));
+            }
+
+            this.timeoutHelper = timeoutHelper;
+        }
+
+        void UpdateReadTimeout()
+        {
+            this.ReadTimeout = TimeoutHelper.ToMilliseconds(this.timeoutHelper.RemainingTime());
+        }
+
+        void UpdateWriteTimeout()
+        {
+            this.WriteTimeout = TimeoutHelper.ToMilliseconds(this.timeoutHelper.RemainingTime());
+        }
+
+        public override IAsyncResult BeginRead(byte[] buffer, int offset, int count, AsyncCallback callback, object state)
+        {
+            UpdateReadTimeout();
+            return base.BeginRead(buffer, offset, count, callback, state);
+        }
+
+        public override IAsyncResult BeginWrite(byte[] buffer, int offset, int count, AsyncCallback callback, object state)
+        {
+            UpdateWriteTimeout();
+            return base.BeginWrite(buffer, offset, count, callback, state);
+        }
+
+        public override int Read(byte[] buffer, int offset, int count)
+        {
+            UpdateReadTimeout();
+            return base.Read(buffer, offset, count);
+        }
+
+        public override int ReadByte()
+        {
+            UpdateReadTimeout();
+            return base.ReadByte();
+        }
+
+        public override void Write(byte[] buffer, int offset, int count)
+        {
+            UpdateWriteTimeout();
+            base.Write(buffer, offset, count);
+        }
+
+        public override void WriteByte(byte value)
+        {
+            UpdateWriteTimeout();
+            base.WriteByte(value);
+        }
+    }
+}