1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
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);
}
}
}
|
