public async Task ReadWriteMessageWithInvalidMagicCookieAsync()
{
byte[] invalidCookie = { 0x01, 0x02 };
Assert.False(TcpTransport.MagicCookie.SequenceEqual(invalidCookie));
var listener = new TcpListener(IPAddress.Any, 0);
listener.Start();
var client = new TcpClient();
await client.ConnectAsync("127.0.0.1", ((IPEndPoint)listener.LocalEndpoint).Port);
TcpClient listenerSocket = await listener.AcceptTcpClientAsync();
AppProtocolVersion version = AppProtocolVersion.Sign(new PrivateKey(), 1);
TcpTransport transport = CreateTcpTransport(appProtocolVersion: version);
try
{
var message = new Ping
{
Identity = Guid.NewGuid().ToByteArray(),
};
var codec = new TcpMessageCodec();
byte[] serialized = codec.Encode(
message,
new PrivateKey(),
transport.AsPeer,
DateTimeOffset.UtcNow,
version);
int length = serialized.Length;
var buffer = new byte[invalidCookie.Length + sizeof(int) + length];
invalidCookie.CopyTo(buffer, 0);
BitConverter.GetBytes(length).CopyTo(buffer, invalidCookie.Length);
serialized.CopyTo(buffer, invalidCookie.Length + sizeof(int));
NetworkStream stream = client.GetStream();
await stream.WriteAsync(buffer, 0, buffer.Length, default);
await Assert.ThrowsAsync <InvalidMagicCookieException>(
async() => await transport.ReadMessageAsync(listenerSocket, default));
byte[] shortBuffer = { 0x00 };
await stream.WriteAsync(shortBuffer, 0, shortBuffer.Length, default);
await Assert.ThrowsAsync <InvalidMagicCookieException>(
async() => await transport.ReadMessageAsync(listenerSocket, default));
}
finally
{
listenerSocket.Dispose();
client.Dispose();
}
}
public async Task ReadMessageCancelAsync()
{
var cts = new CancellationTokenSource();
cts.CancelAfter(TimeSpan.FromSeconds(3));
var listener = new TcpListener(IPAddress.Any, 0);
listener.Start();
var client = new TcpClient();
await client.ConnectAsync("127.0.0.1", ((IPEndPoint)listener.LocalEndpoint).Port);
TcpClient listenerSocket = await listener.AcceptTcpClientAsync();
TcpTransport transport = CreateTcpTransport(
appProtocolVersion: AppProtocolVersion.Sign(new PrivateKey(), 1));
try
{
await Assert.ThrowsAsync <TaskCanceledException>(async() =>
await transport.ReadMessageAsync(listenerSocket, cts.Token));
}
finally
{
listenerSocket.Dispose();
client.Dispose();
}
}
public async Task ReadWriteMessageAsync()
{
var listener = new TcpListener(IPAddress.Any, 0);
listener.Start();
var client = new TcpClient();
await client.ConnectAsync("127.0.0.1", ((IPEndPoint)listener.LocalEndpoint).Port);
TcpClient listenerSocket = await listener.AcceptTcpClientAsync();
TcpTransport transport = CreateTcpTransport(
appProtocolVersion: AppProtocolVersion.Sign(new PrivateKey(), 1));
try
{
var message1 = new Ping
{
Identity = Guid.NewGuid().ToByteArray(),
};
var message2 = new Pong
{
Identity = Guid.NewGuid().ToByteArray(),
};
await transport.WriteMessageAsync(message1, client, default);
await transport.WriteMessageAsync(message2, client, default);
Message[] messages = new Message[2];
messages[0] = await transport.ReadMessageAsync(listenerSocket, default);
messages[1] = await transport.ReadMessageAsync(listenerSocket, default);
Assert.Equal(2, messages.Length);
Assert.Contains(messages, message => message is Ping);
Assert.Contains(messages, message => message is Pong);
}
finally
{
listenerSocket.Dispose();
client.Dispose();
}
}
