private static async Task DoReceiveAsync(Socket udpSocket, ThroughputCounter throughput, CancellationToken cancelToken)
{
// Taking advantage of pre-pinned memory here using the .NET5 POH (pinned object heap).
var buffer = GC.AllocateArray <byte>(PacketSize, true);
var bufferMem = buffer.AsMemory();
while (!cancelToken.IsCancellationRequested)
{
try
{
var result = await udpSocket.ReceiveFromAsync(bufferMem);
// The result tells me where it came from, and gives me the data.
if (result is SocketReceiveFromResult recvResult)
{
throughput.Add(recvResult.ReceivedBytes);
}
else
{
break;
}
}
catch (SocketException)
{
// Socket exception means we are finished.
break;
}
}
}
private static async Task PrintThroughput(ThroughputCounter counter, CancellationToken cancelToken)
{
while (!cancelToken.IsCancellationRequested)
{
await Task.Delay(1000, cancelToken);
var count = counter.SampleAndReset();
var megabytes = count / 1024d / 1024d;
double pps = count / PacketSize;
Console.WriteLine("{0:0.00}MBps ({1:0.00}Mbps) - {2:0.00}pps", megabytes, megabytes * 8, pps);
}
}
static async Task Main(string[] args)
{
using var udpSocket = new Socket(SocketType.Dgram, ProtocolType.Udp);
// Get a cancel source that cancels when the user presses CTRL+C.
var userExitSource = GetUserConsoleCancellationSource();
var cancelToken = userExitSource.Token;
// Discard our socket when the user cancels.
using var cancelReg = cancelToken.Register(() => udpSocket.Dispose());
var throughput = new ThroughputCounter();
// Start a background task to print throughput periodically.
_ = PrintThroughput(throughput, cancelToken);
// Client or server?
if (args.Length > 0 && args[0] == "-c")
{
// Client.
if (args.Length > 1 && IPAddress.TryParse(args[1], out var destination))
{
Console.WriteLine($"Sending to {destination}:9999");
await DoSendAsync(udpSocket, new IPEndPoint(destination, 9999), throughput, cancelToken);
}
else
{
Console.WriteLine("-c argument requires an IP address");
}
}
else
{
// Server.
udpSocket.Bind(new IPEndPoint(IPAddress.Any, 9999));
Console.WriteLine("Listening on 0.0.0.0:9999");
Console.WriteLine("Run with -c <ip address> to be a client.");
await DoReceiveAsync(udpSocket, throughput, cancelToken);
}
}
private static async Task DoReceiveAsync(Socket udpSocket, ThroughputCounter throughput, CancellationToken cancelToken)
{
// Taking advantage of pre-pinned memory here using the .NET5 POH (pinned object heap).
byte[] buffer = GC.AllocateArray <byte>(length: 65527, pinned: true);
Memory <byte> bufferMem = buffer.AsMemory();
while (!cancelToken.IsCancellationRequested)
{
try
{
var result = await udpSocket.ReceiveFromAsync(bufferMem, SocketFlags.None, _blankEndpoint);
throughput.Add(result.ReceivedBytes);
}
catch (SocketException)
{
// Socket exception means we are finished.
break;
}
}
}
private static async Task DoSendAsync(Socket udpSocket, IPEndPoint destination, ThroughputCounter throughput, CancellationToken cancelToken)
{
// Taking advantage of pre-pinned memory here using the .NET 5 POH (pinned object heap).
var buffer = GC.AllocateArray <byte>(PacketSize, true);
var bufferMem = buffer.AsMemory();
// Put something approaching meaningful data in the buffer.
for (var idx = 0; idx < PacketSize; idx++)
{
bufferMem.Span[idx] = (byte)idx;
}
while (!cancelToken.IsCancellationRequested)
{
await udpSocket.SendToAsync(destination, bufferMem);
throughput.Add(bufferMem.Length);
}
}
