huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add timeout to Negotiate (#736)

This commit is contained in:
BrennanConroy 2017-08-18 15:51:01 -07:00 committed by GitHub
parent a4053acd06
commit 345190e6a9
2 changed files with 44 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/Microsoft.AspNetCore.SignalR/HubEndPoint.cs
View File

@ -27,6 +27,7 @@ namespace Microsoft.AspNetCore.SignalR
{
private static readonly Base64Encoder Base64Encoder = new Base64Encoder();
private static readonly PassThroughEncoder PassThroughEncoder = new PassThroughEncoder();
private static readonly TimeSpan NegotiateTimeout = TimeSpan.FromSeconds(5);
private readonly Dictionary<string, HubMethodDescriptor> _methods = new Dictionary<string, HubMethodDescriptor>(StringComparer.OrdinalIgnoreCase);
@ -108,35 +109,46 @@ namespace Microsoft.AspNetCore.SignalR
private async Task<bool> ProcessNegotiate(HubConnectionContext connection)
{
while (await connection.Input.WaitToReadAsync())
try
{
while (connection.Input.TryRead(out var buffer))
using (var cts = new CancellationTokenSource())
{
if (NegotiationProtocol.TryParseMessage(buffer, out var negotiationMessage))
cts.CancelAfter(NegotiateTimeout);
while (await connection.Input.WaitToReadAsync(cts.Token))
{
var protocol = _protocolResolver.GetProtocol(negotiationMessage.Protocol, connection);
while (connection.Input.TryRead(out var buffer))
{
if (NegotiationProtocol.TryParseMessage(buffer, out var negotiationMessage))
{
var protocol = _protocolResolver.GetProtocol(negotiationMessage.Protocol, connection);
var transportCapabilities = connection.Features.Get<IConnectionTransportFeature>()?.TransportCapabilities
?? throw new InvalidOperationException("Unable to read transport capabilities.");
var transportCapabilities = connection.Features.Get<IConnectionTransportFeature>()?.TransportCapabilities
?? throw new InvalidOperationException("Unable to read transport capabilities.");
var dataEncoder = (protocol.Type == ProtocolType.Binary && (transportCapabilities & TransferMode.Binary) == 0)
? (IDataEncoder)Base64Encoder
: PassThroughEncoder;
var dataEncoder = (protocol.Type == ProtocolType.Binary && (transportCapabilities & TransferMode.Binary) == 0)
? (IDataEncoder)Base64Encoder
: PassThroughEncoder;
var transferModeFeature = connection.Features.Get<ITransferModeFeature>() ??
throw new InvalidOperationException("Unable to read transfer mode.");
var transferModeFeature = connection.Features.Get<ITransferModeFeature>() ??
throw new InvalidOperationException("Unable to read transfer mode.");
transferModeFeature.TransferMode =
(protocol.Type == ProtocolType.Binary && (transportCapabilities & TransferMode.Binary) != 0)
? TransferMode.Binary
: TransferMode.Text;
transferModeFeature.TransferMode =
(protocol.Type == ProtocolType.Binary && (transportCapabilities & TransferMode.Binary) != 0)
? TransferMode.Binary
: TransferMode.Text;
connection.ProtocolReaderWriter = new HubProtocolReaderWriter(protocol, dataEncoder);
connection.ProtocolReaderWriter = new HubProtocolReaderWriter(protocol, dataEncoder);
return true;
return true;
}
}
}
}
}
catch (OperationCanceledException)
{
_logger.LogDebug("Negotiate was canceled.");
}
return false;
}

15
test/Microsoft.AspNetCore.SignalR.Tests/HubEndpointTests.cs
View File

@ -59,6 +59,21 @@ namespace Microsoft.AspNetCore.SignalR.Tests
}
}
[Fact]
public async Task NegotiateTimesOut()
{
var serviceProvider = CreateServiceProvider();
var endPoint = serviceProvider.GetService<HubEndPoint<SimpleHub>>();
using (var client = new TestClient())
{
// TestClient automatically writes negotiate, for this test we want to assume negotiate never gets sent
client.Connection.Transport.In.TryRead(out var item);
await endPoint.OnConnectedAsync(client.Connection).OrTimeout(TimeSpan.FromSeconds(10));
}
}
[Fact]
public async Task LifetimeManagerOnDisconnectedAsyncCalledIfLifetimeManagerOnConnectedAsyncThrows()
{