71 lines
2.2 KiB
C#
71 lines
2.2 KiB
C#
//------------------------------------------------------------
|
|
// 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);
|
|
}
|
|
}
|
|
}
|