public static FramingConnectionDuplexSession CreateSession(FramingDuplexSessionChannel channel,
StreamUpgradeAcceptor upgradeAcceptor)
{
if (!(upgradeAcceptor is StreamSecurityUpgradeAcceptor security))
{
return(new FramingConnectionDuplexSession(channel));
}
public ServerFramingDuplexSessionChannel(FramingConnection connection, ITransportFactorySettings settings,
bool exposeConnectionProperty, IServiceProvider serviceProvider)
: base(connection, settings, exposeConnectionProperty)
{
Connection = connection;
upgradeAcceptor = connection.StreamUpgradeAcceptor;
_serviceProvider = serviceProvider;
SetMessageSource(new ServerSessionConnectionMessageSource(connection));
}
public ServerFramingDuplexSessionChannel(FramingConnection connection, ITransportFactorySettings settings,
bool exposeConnectionProperty, IServiceProvider serviceProvider)
: base(connection, settings, exposeConnectionProperty)
{
Connection = connection;
upgradeAcceptor = connection.StreamUpgradeAcceptor;
_serviceProvider = serviceProvider;
//if (upgradeAcceptor != null)
//{
// this.channelBindingProvider = upgrade.GetProperty<IStreamUpgradeChannelBindingProvider>();
// this.upgradeAcceptor = upgrade.CreateUpgradeAcceptor();
//}
}
public static FramingConnectionDuplexSession CreateSession(FramingDuplexSessionChannel channel,
StreamUpgradeAcceptor upgradeAcceptor)
{
StreamSecurityUpgradeAcceptor security = upgradeAcceptor as StreamSecurityUpgradeAcceptor;
if (security == null)
{
return(new FramingConnectionDuplexSession(channel));
}
else
{
return(new SecureConnectionDuplexSession(channel));
}
}
void SetupSecurityIfNecessary(StreamUpgradeAcceptor upgradeAcceptor)
{
StreamSecurityUpgradeAcceptor securityUpgradeAcceptor = upgradeAcceptor as StreamSecurityUpgradeAcceptor;
if (securityUpgradeAcceptor != null)
{
security = securityUpgradeAcceptor.GetRemoteSecurity();
if (security == null)
{
Exception securityFailedException = new ProtocolException(
SR.Format(SR.RemoteSecurityNotNegotiatedOnStreamUpgrade, Via));
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(securityFailedException);
}
}
}
ChannelBinding IStreamUpgradeChannelBindingProvider.GetChannelBinding(StreamUpgradeAcceptor upgradeAcceptor, ChannelBindingKind kind)
{
if (upgradeAcceptor == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull(nameof(upgradeAcceptor));
}
if (!(upgradeAcceptor is SslStreamSecurityUpgradeAcceptor sslupgradeAcceptor))
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgument(nameof(upgradeAcceptor), SR.Format(SR.UnsupportedUpgradeAcceptor, upgradeAcceptor.GetType()));
}
if (kind != ChannelBindingKind.Endpoint)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgument(nameof(kind), SR.Format(SR.StreamUpgradeUnsupportedChannelBindingKind, GetType(), kind));
}
return(sslupgradeAcceptor.ChannelBinding);
}
public ServerFramingDuplexSessionChannel(ConnectionOrientedTransportChannelListener channelListener, ServerSessionPreambleConnectionReader preambleReader,
EndpointAddress localAddress, bool exposeConnectionProperty, ConnectionDemuxer connectionDemuxer)
: base(channelListener, localAddress, preambleReader.Via, exposeConnectionProperty)
{
this.channelListener = channelListener;
this.connectionDemuxer = connectionDemuxer;
Connection = preambleReader.Connection;
decoder = preambleReader.Decoder;
connectionBuffer = preambleReader.connectionBuffer;
offset = preambleReader.BufferOffset;
size = preambleReader.BufferSize;
rawConnection = preambleReader.RawConnection;
StreamUpgradeProvider upgrade = channelListener.Upgrade;
if (upgrade != null)
{
channelBindingProvider = upgrade.GetProperty <IStreamUpgradeChannelBindingProvider>();
upgradeAcceptor = upgrade.CreateUpgradeAcceptor();
}
}
public static async Task <IConnection> UpgradeConnectionAsync(IConnection connection, StreamUpgradeAcceptor upgradeAcceptor, IDefaultCommunicationTimeouts defaultTimeouts)
{
ConnectionStream connectionStream = new ConnectionStream(connection, defaultTimeouts);
Stream stream = await upgradeAcceptor.AcceptUpgradeAsync(connectionStream);
return(new StreamConnection(stream, connectionStream));
}
public async Task <IConnection> CompletePreambleAsync(TimeSpan timeout)
{
var timeoutHelper = new TimeoutHelper(timeout);
var parent = this;
if (!transportSettings.MessageEncoderFactory.Encoder.IsContentTypeSupported(Decoder.ContentType))
{
SendFault(FramingEncodingString.ContentTypeInvalidFault, ref timeoutHelper);
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(new ProtocolException(SR.Format(
SR.ContentTypeMismatch, Decoder.ContentType, parent.transportSettings.MessageEncoderFactory.Encoder.ContentType)));
}
IStreamUpgradeChannelBindingProvider channelBindingProvider = null;
StreamUpgradeAcceptor upgradeAcceptor = null;
if (transportSettings.Upgrade != null)
{
channelBindingProvider = transportSettings.Upgrade.GetProperty <IStreamUpgradeChannelBindingProvider>();
upgradeAcceptor = transportSettings.Upgrade.CreateUpgradeAcceptor();
}
var currentConnection = Connection;
UpgradeState upgradeState = UpgradeState.None;
while (true)
{
if (size == 0 && CanReadAndDecode(upgradeState))
{
size = await currentConnection.ReadAsync(0, connectionBuffer.Length, timeoutHelper.RemainingTime());
if (size == 0)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(Decoder.CreatePrematureEOFException());
}
}
while (true)
{
if (CanReadAndDecode(upgradeState))
{
int bytesRead = Decoder.Decode(connectionBuffer, offset, size);
if (bytesRead > 0)
{
offset += bytesRead;
size -= bytesRead;
}
}
switch (Decoder.CurrentState)
{
case ServerSingletonDecoder.State.UpgradeRequest:
switch (upgradeState)
{
case UpgradeState.None:
//change the state so that we don't read/decode until it is safe
ChangeUpgradeState(ref upgradeState, UpgradeState.VerifyingUpgradeRequest);
break;
case UpgradeState.VerifyingUpgradeRequest:
if (upgradeAcceptor == null)
{
SendFault(FramingEncodingString.UpgradeInvalidFault, ref timeoutHelper);
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
new ProtocolException(SR.Format(SR.UpgradeRequestToNonupgradableService, Decoder.Upgrade)));
}
if (!upgradeAcceptor.CanUpgrade(Decoder.Upgrade))
{
SendFault(FramingEncodingString.UpgradeInvalidFault, ref timeoutHelper);
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(new ProtocolException(SR.Format(SR.UpgradeProtocolNotSupported, Decoder.Upgrade)));
}
ChangeUpgradeState(ref upgradeState, UpgradeState.WritingUpgradeAck);
// accept upgrade
await currentConnection.WriteAsync(ServerSingletonEncoder.UpgradeResponseBytes, 0, ServerSingletonEncoder.UpgradeResponseBytes.Length,
true, timeoutHelper.RemainingTime());
ChangeUpgradeState(ref upgradeState, UpgradeState.UpgradeAckSent);
break;
case UpgradeState.UpgradeAckSent:
IConnection connectionToUpgrade = currentConnection;
if (size > 0)
{
connectionToUpgrade = new PreReadConnection(connectionToUpgrade, connectionBuffer, offset, size);
}
ChangeUpgradeState(ref upgradeState, UpgradeState.BeginUpgrade);
break;
case UpgradeState.BeginUpgrade:
try
{
currentConnection = await InitialServerConnectionReader.UpgradeConnectionAsync(currentConnection, upgradeAcceptor, transportSettings);
connectionBuffer = currentConnection.AsyncReadBuffer;
if (channelBindingProvider != null &&
channelBindingProvider.IsChannelBindingSupportEnabled &&
channelBindingToken == null) //first one wins in the case of multiple upgrades.
{
channelBindingToken = channelBindingProvider.GetChannelBinding(upgradeAcceptor, ChannelBindingKind.Endpoint);
}
ChangeUpgradeState(ref upgradeState, UpgradeState.EndUpgrade);
ChangeUpgradeState(ref upgradeState, UpgradeState.UpgradeComplete);
}
catch (Exception exception)
{
if (Fx.IsFatal(exception))
{
throw;
}
WriteAuditFailure(upgradeAcceptor as StreamSecurityUpgradeAcceptor, exception);
throw;
}
break;
case UpgradeState.UpgradeComplete:
//Client is doing more than one upgrade, reset the state
ChangeUpgradeState(ref upgradeState, UpgradeState.VerifyingUpgradeRequest);
break;
}
break;
case ServerSingletonDecoder.State.Start:
SetupSecurityIfNecessary(upgradeAcceptor);
if (upgradeState == UpgradeState.UpgradeComplete || //We have done at least one upgrade, but we are now done.
upgradeState == UpgradeState.None) //no upgrade, just send the preample end bytes
{
ChangeUpgradeState(ref upgradeState, UpgradeState.WritingPreambleEnd);
// we've finished the preamble. Ack and return.
await currentConnection.WriteAsync(ServerSessionEncoder.AckResponseBytes, 0, ServerSessionEncoder.AckResponseBytes.Length,
true, timeoutHelper.RemainingTime());
//terminal state
ChangeUpgradeState(ref upgradeState, UpgradeState.PreambleEndSent);
}
//we are done, this.currentConnection is the upgraded connection
return(currentConnection);
}
if (size == 0)
{
break;
}
}
}
}