public void InitializeMessage()
{
var transactionId = StunTransactionId.Generate();
var message = new StunBuilderStep().SetTransactionId(transactionId)
.SetType(StunMessageType.BindingRequest)
.Build();
var result = StunMessage.TryParse(message, out var parsedMessage);
Assert.True(result);
Assert.Equal(StunMessageType.BindingRequest, parsedMessage.Type);
Assert.Equal(transactionId, parsedMessage.TransactionId);
}
public async Task SendingAndReceiving()
{
var transationId = StunTransactionId.Generate();
var mappedAddresses = new List <IPEndPoint>();
var datagramBlock = new DatagramBlock(new DatagramOptions
{
ListeningPoint = new IPEndPoint(IPAddress.Any, 50000)
});
var stunMessage = new StunBuilderStep().SetType(StunMessageType.BindingRequest)
.SetTransactionId(transationId)
.Build();
var datagram = new Datagram(stunMessage, new byte[] { 64, 233, 161, 127 }, 19302); //stun1.l.google.com:19302
var stunMessageHandler = new ActionBlock <StunMessage>(response =>
{
if (response.Type != StunMessageType.BindingResponse || response.TransactionId != transationId)
{
return;
}
foreach (var attribute in response.Attributes)
{
if (StunMappedAddressAttribute.TryParse(attribute, out var mappedAddressAttribute))
{
mappedAddresses.Add(new IPEndPoint(new IPAddress(mappedAddressAttribute.EndPoint.Address.ToArray()), mappedAddressAttribute.EndPoint.Port));
}
}
});
var stunMessagePipe = new StunPipeBlock();
datagramBlock.LinkTo(stunMessagePipe, new DataflowLinkOptions {
PropagateCompletion = true
});
stunMessagePipe.LinkTo(stunMessageHandler, new DataflowLinkOptions {
PropagateCompletion = true
});
stunMessagePipe.LinkTo(DataflowBlock.NullTarget <Datagram>());
using (new Timer(context => datagramBlock.Post(datagram), null, TimeSpan.FromSeconds(1), TimeSpan.FromSeconds(1)))
{
datagramBlock.Start();
await datagramBlock.Initialization;
await Task.Delay(4000);
datagramBlock.Complete();
await Task.WhenAll(datagramBlock.Completion,
stunMessagePipe.Completion,
stunMessageHandler.Completion);
}
mappedAddresses.Should().HaveCountGreaterOrEqualTo(3);
}
public void InitializeAttributeList()
{
var firstAttribute = new StunAttribute(StunAttributeType.ChangedAddress, new byte[] { 1, 2, 3 });
var secondAttribute = new StunAttribute(StunAttributeType.MappedAddress, new byte[] { 4, 5, 6 });
var message = new StunBuilderStep().AddAttributes(new StunAttribute[] { firstAttribute, secondAttribute })
.Build();
var result = StunMessage.TryParse(message, out var parsedMessage);
var parsedAttributes = parsedMessage.Attributes.GetEnumerator();
Assert.True(result);
Assert.True(parsedAttributes.MoveNext());
Assert.Equal(firstAttribute.Type, parsedAttributes.Current.Type);
Assert.True(firstAttribute.Content.Span.SequenceEqual(parsedAttributes.Current.Content.Span));
Assert.True(parsedAttributes.MoveNext());
Assert.Equal(secondAttribute.Type, parsedAttributes.Current.Type);
Assert.True(secondAttribute.Content.Span.SequenceEqual(parsedAttributes.Current.Content.Span));
}
