public void IndicateANonWebSocketConnection()
{
var extension = new Mock <IWebSocketMessageExtension>();
extension.Setup(x => x.Name).Returns("test-extension");
var ext = new WebSocketExtension("test-extension", new List <WebSocketExtensionOption>(new[]
{
new WebSocketExtensionOption("optionA")
}));
IWebSocketMessageExtensionContext ctx;
extension.Setup(x => x.TryNegotiate(It.IsAny <WebSocketHttpRequest>(), out ext, out ctx))
.Returns(true);
var factory = new WebSocketFactoryRfc6455();
factory.MessageExtensions.Add(extension.Object);
var factories = new WebSocketFactoryCollection();
factories.Add(factory);
var handshaker = new WebSocketHandshaker(factories, new WebSocketListenerOptions
{
Logger = this.logger,
SubProtocols = new[]
{
"superchat"
}
});
using (var connectionInput = new MemoryStream())
using (var connectionOutput = new MemoryStream())
using (var connection = new DummyNetworkConnection(connectionInput, connectionOutput))
{
using (var sw = new StreamWriter(connectionInput, Encoding.ASCII, 1024, true))
{
sw.WriteLine(@"GET /chat HTTP/1.1");
sw.WriteLine(@"Host: server.example.com");
}
connectionInput.Seek(0, SeekOrigin.Begin);
var result = handshaker.HandshakeAsync(connection).Result;
Assert.NotNull(result);
Assert.False((bool)result.IsWebSocketRequest);
Assert.False((bool)result.IsVersionSupported);
connectionOutput.Seek(0, SeekOrigin.Begin);
var sb = new StringBuilder();
sb.AppendLine(@"HTTP/1.1 400 Bad Request");
sb.AppendLine(@"Connection: close");
sb.AppendLine();
using (var sr = new StreamReader(connectionOutput))
{
var s = sr.ReadToEnd();
Assert.Equal(sb.ToString(), s);
}
}
}
public void WebSocketHandshaker_FailsWhenBadExtensionRequest()
{
var extension = new Mock <IWebSocketMessageExtension>();
extension.Setup(x => x.Name).Returns("test-extension");
WebSocketExtension ext = new WebSocketExtension("test-extension", new List <WebSocketExtensionOption>(new[] { new WebSocketExtensionOption()
{
ClientAvailableOption = false, Name = "optionA"
} }));
IWebSocketMessageExtensionContext ctx;
extension.Setup(x => x.TryNegotiate(It.IsAny <WebSocketHttpRequest>(), out ext, out ctx))
.Returns(true);
var config = new WebSocketListenerConfig(new WebSocketListenerOptions()
{
SubProtocols = new[] { "superchat" }
});
config.Standards.RegisterStandard(new WebSocketFactoryRfc6455());
config.MessageExtensions.RegisterExtension(extension.Object);
var handshaker = new WebSocketHandshaker(config);
using (var ms = new MemoryStream())
{
using (var sw = new StreamWriter(ms, Encoding.ASCII, 1024, true))
{
sw.WriteLine(@"GET /chat HTTP/1.1");
sw.WriteLine(@"Host: server.example.com");
sw.WriteLine(@"Upgrade: websocket");
sw.WriteLine(@"Connection: Upgrade");
sw.WriteLine(@"Cookie: key=W9g/8FLW8RAFqSCWBvB9Ag==#5962c0ace89f4f780aa2a53febf2aae5;");
sw.WriteLine(@"Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==");
sw.WriteLine(@"Sec-WebSocket-Protocol: chat, superchat");
sw.WriteLine(@"Sec-WebSocket-Extensions: test-extension;;Dsf,,optionA");
sw.WriteLine(@"Sec-WebSocket-Version: 13");
sw.WriteLine(@"Origin: http://example.com");
}
var position = ms.Position;
ms.Seek(0, SeekOrigin.Begin);
var result = handshaker.HandshakeAsync(ms).Result;
Assert.IsTrue(result.IsWebSocketRequest);
Assert.IsTrue(result.IsVersionSupported);
Assert.IsTrue(result.NegotiatedMessageExtensions.Count() == 0);
ms.Seek(position, SeekOrigin.Begin);
StringBuilder sb = new StringBuilder();
sb.AppendLine(@"HTTP/1.1 400 Bad Request");
sb.AppendLine();
using (var sr = new StreamReader(ms))
{
var s = sr.ReadToEnd();
Assert.AreEqual(sb.ToString(), s);
}
}
}
public bool TryNegotiate(WebSocketHttpRequest request, out WebSocketExtension extensionResponse, out IWebSocketMessageExtensionContext context)
{
if (request == null)
{
throw new ArgumentNullException(nameof(request));
}
extensionResponse = DefaultResponse;
context = new WebSocketDeflateContext();
return(true);
}
public void WebSocketHandshaker_CanIndicateANonWebSocketConnection()
{
var extension = new Mock <IWebSocketMessageExtension>();
extension.Setup(x => x.Name).Returns("test-extension");
WebSocketExtension ext = new WebSocketExtension("test-extension", new List <WebSocketExtensionOption>(new[] { new WebSocketExtensionOption()
{
ClientAvailableOption = false, Name = "optionA"
} }));
IWebSocketMessageExtensionContext ctx;
extension.Setup(x => x.TryNegotiate(It.IsAny <WebSocketHttpRequest>(), out ext, out ctx))
.Returns(true);
var factory = new WebSocketFactoryRfc6455();
factory.MessageExtensions.RegisterExtension(extension.Object);
var factories = new WebSocketFactoryCollection();
factories.RegisterStandard(factory);
WebSocketHandshaker handshaker = new WebSocketHandshaker(factories, new WebSocketListenerOptions()
{
SubProtocols = new[] { "superchat" }
});
using (var ms = new MemoryStream())
{
using (var sw = new StreamWriter(ms, Encoding.ASCII, 1024, true))
{
sw.WriteLine(@"GET /chat HTTP/1.1");
sw.WriteLine(@"Host: server.example.com");
}
var position = ms.Position;
ms.Seek(0, SeekOrigin.Begin);
var result = handshaker.HandshakeAsync(ms).Result;
Assert.IsNotNull(result);
Assert.IsFalse(result.IsWebSocketRequest);
Assert.IsFalse(result.IsVersionSupported);
ms.Seek(position, SeekOrigin.Begin);
StringBuilder sb = new StringBuilder();
sb.AppendLine(@"HTTP/1.1 400 Bad Request");
sb.AppendLine();
using (var sr = new StreamReader(ms))
{
var s = sr.ReadToEnd();
Assert.AreEqual(sb.ToString(), s);
}
}
}
public void WebSocketHandshaker_CanNegotiateAnExtensionWithParameters()
{
var extension = new Mock <IWebSocketMessageExtension>();
extension.Setup(x => x.Name).Returns("test-extension");
WebSocketExtension ext = new WebSocketExtension("test-extension", new List <WebSocketExtensionOption>(new [] { new WebSocketExtensionOption()
{
ClientAvailableOption = false, Name = "optionA"
} }));
IWebSocketMessageExtensionContext ctx;
extension.Setup(x => x.TryNegotiate(It.IsAny <WebSocketHttpRequest>(), out ext, out ctx))
.Returns(true);
var factory = new WebSocketFactoryRfc6455();
factory.MessageExtensions.RegisterExtension(extension.Object);
var factories = new WebSocketFactoryCollection();
factories.RegisterStandard(factory);
WebSocketHandshaker handshaker = new WebSocketHandshaker(factories, new WebSocketListenerOptions()
{
SubProtocols = new[] { "superchat" }
});
using (var ms = new MemoryStream())
{
using (var sw = new StreamWriter(ms, Encoding.ASCII, 1024, true))
{
sw.WriteLine(@"GET /chat HTTP/1.1");
sw.WriteLine(@"Host: server.example.com");
sw.WriteLine(@"Upgrade: websocket");
sw.WriteLine(@"Connection: Upgrade");
sw.WriteLine(@"Cookie: key=W9g/8FLW8RAFqSCWBvB9Ag==#5962c0ace89f4f780aa2a53febf2aae5;");
sw.WriteLine(@"Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==");
sw.WriteLine(@"Sec-WebSocket-Protocol: chat, superchat");
sw.WriteLine(@"Sec-WebSocket-Extensions: test-extension;optionA");
sw.WriteLine(@"Sec-WebSocket-Version: 13");
sw.WriteLine(@"Origin: http://example.com");
}
var position = ms.Position;
ms.Seek(0, SeekOrigin.Begin);
var result = handshaker.HandshakeAsync(ms).Result;
Assert.IsNotNull(result);
Assert.IsTrue(result.IsWebSocketRequest);
Assert.AreEqual(new Uri("http://example.com"), result.Request.Headers.Origin);
Assert.AreEqual("superchat", result.Response.WebSocketProtocol);
ms.Seek(position, SeekOrigin.Begin);
StringBuilder sb = new StringBuilder();
sb.AppendLine(@"HTTP/1.1 101 Switching Protocols");
sb.AppendLine(@"Upgrade: websocket");
sb.AppendLine(@"Connection: Upgrade");
sb.AppendLine(@"Sec-WebSocket-Accept: HSmrc0sMlYUkAGmm5OPpG2HaGWk=");
sb.AppendLine(@"Sec-WebSocket-Protocol: superchat");
sb.AppendLine(@"Sec-WebSocket-Extensions: test-extension;optionA");
sb.AppendLine();
using (var sr = new StreamReader(ms))
{
var s = sr.ReadToEnd();
Assert.AreEqual(sb.ToString(), s);
}
}
}
public bool TryNegotiate(WebSocketHttpRequest request, out WebSocketExtension extensionResponse, out IWebSocketMessageExtensionContext context)
{
extensionResponse = _response;
context = new WebSocketDeflateContext();
return true;
}
public bool TryNegotiate(WebSocketHttpRequest request, out WebSocketExtension extensionResponse, out IWebSocketMessageExtensionContext context)
{
extensionResponse = _response;
context = new WebSocketDeflateContext();
return(true);
}
public void NegotiateAnExtensionWithParameters()
{
var extension = new Mock <IWebSocketMessageExtension>();
extension.Setup(x => x.Name).Returns("test-extension");
var ext = new WebSocketExtension("test-extension", new List <WebSocketExtensionOption>(new[]
{
new WebSocketExtensionOption("optionA")
}));
IWebSocketMessageExtensionContext ctx;
extension.Setup(x => x.TryNegotiate(It.IsAny <WebSocketHttpRequest>(), out ext, out ctx))
.Returns(true);
var factory = new WebSocketFactoryRfc6455();
factory.MessageExtensions.Add(extension.Object);
var factories = new WebSocketFactoryCollection();
factories.Add(factory);
var handshaker = new WebSocketHandshaker(factories, new WebSocketListenerOptions
{
Logger = this.logger,
SubProtocols = new[]
{
"superchat"
}
});
using (var connectionInput = new MemoryStream())
using (var connectionOutput = new MemoryStream())
using (var connection = new DummyNetworkConnection(connectionInput, connectionOutput))
{
using (var sw = new StreamWriter(connectionInput, Encoding.ASCII, 1024, true))
{
sw.WriteLine(@"GET /chat HTTP/1.1");
sw.WriteLine(@"Host: server.example.com");
sw.WriteLine(@"Upgrade: websocket");
sw.WriteLine(@"Connection: Upgrade");
sw.WriteLine(@"Cookie: key=W9g/8FLW8RAFqSCWBvB9Ag==#5962c0ace89f4f780aa2a53febf2aae5;");
sw.WriteLine(@"Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==");
sw.WriteLine(@"Sec-WebSocket-Protocol: chat, superchat");
sw.WriteLine(@"Sec-WebSocket-Extensions: test-extension;optionA");
sw.WriteLine(@"Sec-WebSocket-Version: 13");
sw.WriteLine(@"Origin: http://example.com");
}
connectionInput.Seek(0, SeekOrigin.Begin);
var result = handshaker.HandshakeAsync(connection).Result;
Assert.NotNull(result);
Assert.True((bool)result.IsWebSocketRequest);
Assert.Equal(new Uri("http://example.com"), new Uri(result.Request.Headers[RequestHeader.Origin]));
Assert.Equal((string)"superchat", (string)result.Response.Headers[ResponseHeader.WebSocketProtocol]);
connectionOutput.Seek(0, SeekOrigin.Begin);
var sb = new StringBuilder();
sb.AppendLine(@"HTTP/1.1 101 Switching Protocols");
sb.AppendLine(@"Upgrade: websocket");
sb.AppendLine(@"Connection: Upgrade");
sb.AppendLine(@"Sec-WebSocket-Accept: HSmrc0sMlYUkAGmm5OPpG2HaGWk=");
sb.AppendLine(@"Sec-WebSocket-Protocol: superchat");
sb.AppendLine(@"Sec-WebSocket-Extensions: test-extension;optionA");
sb.AppendLine();
using (var sr = new StreamReader(connectionOutput))
{
var s = sr.ReadToEnd();
Assert.Equal(sb.ToString(), s);
}
}
}
