public void NoHelloMessage()
{
var responseStream = new MemoryStream ();
var stream = new TestStream (new MemoryStream (), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock<IServer<byte,byte>> ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock<IClient<byte,byte>> ();
mockByteClient.Setup (x => x.Stream).Returns (stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer (byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow ();
server.Start ();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionArgs<byte,byte> (byteClient);
mockByteServer.Raise (m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse (eventArgs.Request.ShouldAllow);
Assert.IsTrue (eventArgs.Request.ShouldDeny);
Assert.AreEqual (0, responseStream.Length);
}
public void NoConnectionMessage()
{
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock <IClient <byte, byte> > ();
mockByteClient.Setup(x => x.Stream).Returns(stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsFalse(eventArgs.Request.ShouldDeny);
Assert.AreEqual(responseStream.ToArray().Length, 0);
}
public void WrongConnectionType()
{
var responseStream = new MemoryStream();
var connectionMessage = TestingTools.CreateConnectionRequest(Type.Rpc, clientId.ToByteArray());
var stream = new TestStream(new MemoryStream(connectionMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock <IClient <byte, byte> > ();
mockByteClient.Setup(x => x.Stream).Returns(stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsTrue(eventArgs.Request.ShouldDeny);
TestingTools.CheckConnectionResponse(responseStream.ToArray(), 120, Status.WrongType,
"Connection request was for the rpc server, but this is the stream server. " +
"Did you connect to the wrong port number?", 0);
}
public void InvalidConnectionMessageIdentifier()
{
var connectionMessage = TestingTools.CreateConnectionRequest(Type.Stream, "123456".ToBytes());
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(connectionMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock <IClient <byte, byte> > ();
mockByteClient.Setup(x => x.Stream).Returns(stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsTrue(eventArgs.Request.ShouldDeny);
TestingTools.CheckConnectionResponse(responseStream.ToArray(), 40, Status.MalformedMessage, "Client identifier must be 16 bytes.", 0);
}
public void ValidConnectionMessage()
{
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(TestingTools.CreateConnectionRequest(Type.Stream, clientId.ToByteArray())), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var byteClient = new TestClient(stream);
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsTrue(eventArgs.Request.ShouldAllow);
Assert.IsFalse(eventArgs.Request.ShouldDeny);
server.Update();
Assert.AreEqual(1, server.Clients.Count());
Assert.AreEqual(clientId, server.Clients.First().Guid);
TestingTools.CheckConnectionResponse(responseStream.ToArray(), 19, Status.Ok, string.Empty, 16);
}
public void ShortHelloMessageHeader()
{
var shortHelloMessage = new byte[5];
Array.Copy(helloMessage, shortHelloMessage, shortHelloMessage.Length);
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(shortHelloMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock <IClient <byte, byte> > ();
mockByteClient.Setup(x => x.Stream).Returns(stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsTrue(eventArgs.Request.ShouldDeny);
Assert.AreEqual(0, responseStream.Length);
}
public void ValidHelloMessage()
{
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(helloMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var byteClient = new TestClient(stream);
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsTrue(eventArgs.Request.ShouldAllow);
Assert.IsFalse(eventArgs.Request.ShouldDeny);
server.Update();
Assert.AreEqual(1, server.Clients.Count());
Assert.AreEqual(clientGuid, server.Clients.First().Guid);
byte[] bytes = responseStream.ToArray();
byte[] expectedBytes = { 0x4F, 0x4B };
Assert.IsTrue(expectedBytes.SequenceEqual(bytes));
}
public void ValidConnectionRequest()
{
var clientGuid = new Guid("1234567890abcdef1234567890abcdef".ToBytes());
var clientGuidBase64 = Convert.ToBase64String(clientGuid.ToByteArray());
var request = Encoding.ASCII.GetBytes(
"GET /?id=" + clientGuidBase64 + " HTTP/1.1\r\n" +
"Host: localhost\r\n" +
"Upgrade: websocket\r\n" +
"Connection: Upgrade\r\n" +
"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" +
"Sec-WebSocket-Version: 13\r\n\r\n"
);
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(request), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var byteClient = new TestClient(stream);
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsTrue(eventArgs.Request.ShouldAllow);
Assert.IsFalse(eventArgs.Request.ShouldDeny);
server.Update();
Assert.AreEqual(1, server.Clients.Count());
Assert.AreEqual(clientGuid, server.Clients.First().Guid);
var response = Encoding.ASCII.GetString(responseStream.ToArray());
Assert.AreEqual(
"HTTP/1.1 101 Switching Protocols\r\n" +
"Upgrade: websocket\r\n" +
"Connection: Upgrade\r\n" +
"Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n\r\n",
response
);
}
public void InvalidConnectionRequestNoGuid()
{
var ascii = Encoding.ASCII;
var request = ascii.GetBytes(
"GET / HTTP/1.1\r\n" +
"Host: localhost\r\n" +
"Upgrade: websocket\r\n" +
"Connection: Upgrade\r\n" +
"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" +
"Sec-WebSocket-Version: 13\r\n\r\n"
);
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(request), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var byteClient = new TestClient(stream);
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsTrue(eventArgs.Request.ShouldDeny);
server.Update();
Assert.AreEqual(0, server.Clients.Count());
var response = ascii.GetString(responseStream.ToArray());
Assert.AreEqual(
"HTTP/1.1 400 Bad Request\r\n\r\n",
response
);
}
public void InvalidConnectionMessageHeader()
{
var connectionMessage = TestingTools.CreateConnectionRequest(Type.Stream, clientId.ToByteArray());
connectionMessage [2] ^= 0x42;
connectionMessage [3] ^= 0x42;
connectionMessage [4] ^= 0x42;
connectionMessage [5] ^= 0x42;
var responseStream = new MemoryStream();
var stream = new TestStream(new MemoryStream(connectionMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock <IServer <byte, byte> > ();
var byteServer = mockByteServer.Object;
var mockByteClient = new Mock <IClient <byte, byte> > ();
mockByteClient.Setup(x => x.Stream).Returns(stream);
var byteClient = mockByteClient.Object;
var server = new StreamServer(byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow();
server.Start();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionEventArgs <byte, byte> (byteClient);
mockByteServer.Raise(m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsFalse(eventArgs.Request.ShouldAllow);
Assert.IsTrue(eventArgs.Request.ShouldDeny);
TestingTools.CheckConnectionResponse(responseStream.ToArray(), 209, Status.MalformedMessage,
"While parsing a protocol message, the input ended unexpectedly in the middle of a field. " +
"This could mean either that the input has been truncated or that an embedded message misreported its own length.", 0);
}
public void ValidHelloMessage()
{
var responseStream = new MemoryStream ();
var stream = new TestStream (new MemoryStream (helloMessage), responseStream);
// Create mock byte server and client
var mockByteServer = new Mock<IServer<byte,byte>> ();
var byteServer = mockByteServer.Object;
var byteClient = new TestClient (stream);
var server = new StreamServer (byteServer);
server.OnClientRequestingConnection += (sender, e) => e.Request.Allow ();
server.Start ();
// Fire a client connection event
var eventArgs = new ClientRequestingConnectionArgs<byte,byte> (byteClient);
mockByteServer.Raise (m => m.OnClientRequestingConnection += null, eventArgs);
Assert.IsTrue (eventArgs.Request.ShouldAllow);
Assert.IsFalse (eventArgs.Request.ShouldDeny);
server.Update ();
Assert.AreEqual (1, server.Clients.Count ());
Assert.AreEqual (clientGuid, server.Clients.First ().Guid);
byte[] bytes = responseStream.ToArray ();
byte[] expectedBytes = { 0x4F, 0x4B };
Assert.IsTrue (expectedBytes.SequenceEqual (bytes));
}
/// <summary>
/// Start the server
/// </summary>
public void Start()
{
rpcServer.Start();
streamServer.Start();
ClearStats();
}