private async ValueTask <IConnection?> InternalConnectAsync(ICap cap, string serviceType, CancellationToken cancellationToken)
{
using var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
linkedTokenSource.CancelAfter(TimeSpan.FromSeconds(20));
var baseConnectionOptions = new BaseConnectionOptions()
{
MaxSendByteCount = 4 * 1024 * 1024,
MaxReceiveByteCount = 4 * 1024 * 1024,
BytesPool = _bytesPool,
};
var baseConnection = new BaseConnection(cap, _baseConnectionDispatcher, baseConnectionOptions);
var omniSecureConnectionOptions = new OmniSecureConnectionOptions()
{
Type = OmniSecureConnectionType.Connected,
BufferPool = _bytesPool,
};
var omniSecureConnection = new OmniSecureConnection(baseConnection, omniSecureConnectionOptions);
await omniSecureConnection.HandshakeAsync(linkedTokenSource.Token);
var helloMessage = new ConnectionHelloMessage(serviceType);
await omniSecureConnection.EnqueueAsync((bufferWriter) => helloMessage.Export(bufferWriter, _bytesPool), linkedTokenSource.Token);
return(omniSecureConnection);
}
public void RandomSendAndReceiveTest()
{
var random = new Random();
var caseList = new List <int>();
caseList.AddRange(Enumerable.Range(1, 32));
caseList.AddRange(new int[] { 100, 1000, 10000, 1024 * 1024, 1024 * 1024 * 32 });
var(socket1, socket2) = SocketHelpers.GetSockets();
var options = new OmniNonblockingConnectionOptions()
{
MaxReceiveByteCount = 1024 * 1024 * 256,
MaxSendByteCount = 1024 * 1024 * 256,
BufferPool = BufferPool.Shared,
};
using (var baseConnection1 = new OmniNonblockingConnection(new SocketCap(socket1, false), options))
using (var baseConnection2 = new OmniNonblockingConnection(new SocketCap(socket2, false), options))
using (var connection1 = new OmniSecureConnection(baseConnection1, new OmniSecureConnectionOptions()
{
Type = OmniSecureConnectionType.Connect
}))
using (var connection2 = new OmniSecureConnection(baseConnection2, new OmniSecureConnectionOptions()
{
Type = OmniSecureConnectionType.Accept
}))
{
// ハンドシェイクを行う
{
var valueTask1 = connection1.Handshake();
var valueTask2 = connection2.Handshake();
var stopwatch = Stopwatch.StartNew();
while (!valueTask1.IsCompleted || !valueTask2.IsCompleted)
{
if (stopwatch.Elapsed.TotalSeconds > 60)
{
throw new TimeoutException("Handshake");
}
Thread.Sleep(100);
connection1.DoEvents();
connection2.DoEvents();
}
}
foreach (var bufferSize in caseList)
{
var buffer1 = new byte[bufferSize];
var buffer2 = new byte[bufferSize];
random.NextBytes(buffer1);
var valueTask1 = connection1.EnqueueAsync((bufferWriter) =>
{
bufferWriter.Write(buffer1);
});
var valueTask2 = connection2.DequeueAsync((sequence) =>
{
sequence.CopyTo(buffer2);
});
var stopwatch = Stopwatch.StartNew();
while (!valueTask1.IsCompleted || !valueTask2.IsCompleted)
{
if (stopwatch.Elapsed.TotalSeconds > 60)
{
throw new TimeoutException("SendAndReceive");
}
Thread.Sleep(100);
connection1.DoEvents();
connection2.DoEvents();
}
Assert.True(BytesOperations.SequenceEqual(buffer1, buffer2));
}
}
}