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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
//-----------------------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//-----------------------------------------------------------------------------
namespace System.ServiceModel.Channels
{
using System.IO;
using System.Runtime.Diagnostics;
using System.ServiceModel.Security;
abstract class StreamSecurityUpgradeAcceptorBase : StreamSecurityUpgradeAcceptor
{
SecurityMessageProperty remoteSecurity;
bool securityUpgraded;
string upgradeString;
EventTraceActivity eventTraceActivity;
protected StreamSecurityUpgradeAcceptorBase(string upgradeString)
{
this.upgradeString = upgradeString;
}
internal EventTraceActivity EventTraceActivity
{
get
{
if (this.eventTraceActivity == null)
{
this.eventTraceActivity = EventTraceActivity.GetFromThreadOrCreate();
}
return this.eventTraceActivity;
}
}
public override Stream AcceptUpgrade(Stream stream)
{
if (stream == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("stream");
}
Stream result = this.OnAcceptUpgrade(stream, out this.remoteSecurity);
this.securityUpgraded = true;
return result;
}
public override IAsyncResult BeginAcceptUpgrade(Stream stream, AsyncCallback callback, object state)
{
if (stream == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("stream");
}
return this.OnBeginAcceptUpgrade(stream, callback, state);
}
public override bool CanUpgrade(string contentType)
{
if (this.securityUpgraded)
{
return false;
}
return (contentType == this.upgradeString);
}
public override Stream EndAcceptUpgrade(IAsyncResult result)
{
if (result == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("result");
}
Stream retValue = this.OnEndAcceptUpgrade(result, out this.remoteSecurity);
this.securityUpgraded = true;
return retValue;
}
public override SecurityMessageProperty GetRemoteSecurity()
{
// this could be null if upgrade not completed.
return this.remoteSecurity;
}
protected abstract Stream OnAcceptUpgrade(Stream stream, out SecurityMessageProperty remoteSecurity);
protected abstract IAsyncResult OnBeginAcceptUpgrade(Stream stream, AsyncCallback callback, object state);
protected abstract Stream OnEndAcceptUpgrade(IAsyncResult result,
out SecurityMessageProperty remoteSecurity);
}
}
|
