Added a few debug tools for stability testing
This commit is contained in:
RogueException
31
src/Discord.Net.DebugTools/Discord.Net.DebugTools.csproj
Normal file
31
src/Discord.Net.DebugTools/Discord.Net.DebugTools.csproj
Normal file
@@ -0,0 +1,31 @@
|
||||
<Project ToolsVersion="15.0" Sdk="Microsoft.NET.Sdk" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<PropertyGroup>
|
||||
<VersionPrefix>1.0.0</VersionPrefix>
|
||||
<VersionSuffix Condition="'$(BuildNumber)' == ''">rc-dev</VersionSuffix>
|
||||
<VersionSuffix Condition="'$(BuildNumber)' != ''">rc-$(BuildNumber)</VersionSuffix>
|
||||
<TargetFrameworks>netstandard1.6</TargetFrameworks>
|
||||
<AssemblyName>Discord.Net.DebugTools</AssemblyName>
|
||||
<Authors>RogueException</Authors>
|
||||
<Description>A Discord.Net extension adding random helper classes for diagnosing issues.</Description>
|
||||
<PackageTags>discord;discordapp</PackageTags>
|
||||
<PackageProjectUrl>https://github.com/RogueException/Discord.Net</PackageProjectUrl>
|
||||
<PackageLicenseUrl>http://opensource.org/licenses/MIT</PackageLicenseUrl>
|
||||
<RepositoryType>git</RepositoryType>
|
||||
<RepositoryUrl>git://github.com/RogueException/Discord.Net</RepositoryUrl>
|
||||
<RootNamespace>Discord</RootNamespace>
|
||||
<DisableImplicitFrameworkReferences>true</DisableImplicitFrameworkReferences>
|
||||
</PropertyGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\Discord.Net.Core\Discord.Net.Core.csproj" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<PackageReference Include="System.Net.NameResolution" Version="4.3.0" />
|
||||
<PackageReference Include="System.Net.Sockets" Version="4.3.0" />
|
||||
<PackageReference Include="System.Net.WebSockets.Client" Version="4.3.0" />
|
||||
</ItemGroup>
|
||||
<PropertyGroup Condition=" '$(Configuration)' == 'Release' ">
|
||||
<NoWarn>$(NoWarn);CS1573;CS1591</NoWarn>
|
||||
<WarningsAsErrors>true</WarningsAsErrors>
|
||||
<GenerateDocumentationFile>true</GenerateDocumentationFile>
|
||||
</PropertyGroup>
|
||||
</Project>
|
||||
142
src/Discord.Net.DebugTools/UnstableUdpClient.cs
Normal file
142
src/Discord.Net.DebugTools/UnstableUdpClient.cs
Normal file
@@ -0,0 +1,142 @@
|
||||
using Discord.Net.Udp;
|
||||
using System;
|
||||
using System.Net;
|
||||
using System.Net.Sockets;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Discord.Net.Providers.UnstableUdpSocket
|
||||
{
|
||||
internal class UnstableUdpSocket : IUdpSocket, IDisposable
|
||||
{
|
||||
private const double FailureRate = 0.10; //10%
|
||||
|
||||
public event Func<byte[], int, int, Task> ReceivedDatagram;
|
||||
|
||||
private readonly SemaphoreSlim _lock;
|
||||
private readonly Random _rand;
|
||||
private UdpClient _udp;
|
||||
private IPEndPoint _destination;
|
||||
private CancellationTokenSource _cancelTokenSource;
|
||||
private CancellationToken _cancelToken, _parentToken;
|
||||
private Task _task;
|
||||
private bool _isDisposed;
|
||||
|
||||
public UnstableUdpSocket()
|
||||
{
|
||||
_lock = new SemaphoreSlim(1, 1);
|
||||
_rand = new Random();
|
||||
_cancelTokenSource = new CancellationTokenSource();
|
||||
}
|
||||
private void Dispose(bool disposing)
|
||||
{
|
||||
if (!_isDisposed)
|
||||
{
|
||||
if (disposing)
|
||||
StopInternalAsync(true).GetAwaiter().GetResult();
|
||||
_isDisposed = true;
|
||||
}
|
||||
}
|
||||
public void Dispose()
|
||||
{
|
||||
Dispose(true);
|
||||
}
|
||||
|
||||
|
||||
public async Task StartAsync()
|
||||
{
|
||||
await _lock.WaitAsync().ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await StartInternalAsync(_cancelToken).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
public async Task StartInternalAsync(CancellationToken cancelToken)
|
||||
{
|
||||
await StopInternalAsync().ConfigureAwait(false);
|
||||
|
||||
_cancelTokenSource = new CancellationTokenSource();
|
||||
_cancelToken = CancellationTokenSource.CreateLinkedTokenSource(_parentToken, _cancelTokenSource.Token).Token;
|
||||
|
||||
_udp = new UdpClient(0);
|
||||
|
||||
_task = RunAsync(_cancelToken);
|
||||
}
|
||||
public async Task StopAsync()
|
||||
{
|
||||
await _lock.WaitAsync().ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await StopInternalAsync().ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
public async Task StopInternalAsync(bool isDisposing = false)
|
||||
{
|
||||
try { _cancelTokenSource.Cancel(false); } catch { }
|
||||
|
||||
if (!isDisposing)
|
||||
await (_task ?? Task.Delay(0)).ConfigureAwait(false);
|
||||
|
||||
if (_udp != null)
|
||||
{
|
||||
try { _udp.Dispose(); }
|
||||
catch { }
|
||||
_udp = null;
|
||||
}
|
||||
}
|
||||
|
||||
public void SetDestination(string host, int port)
|
||||
{
|
||||
var entry = Dns.GetHostEntryAsync(host).GetAwaiter().GetResult();
|
||||
_destination = new IPEndPoint(entry.AddressList[0], port);
|
||||
}
|
||||
public void SetCancelToken(CancellationToken cancelToken)
|
||||
{
|
||||
_parentToken = cancelToken;
|
||||
_cancelToken = CancellationTokenSource.CreateLinkedTokenSource(_parentToken, _cancelTokenSource.Token).Token;
|
||||
}
|
||||
|
||||
public async Task SendAsync(byte[] data, int index, int count)
|
||||
{
|
||||
if (!UnstableCheck())
|
||||
return;
|
||||
|
||||
if (index != 0) //Should never happen?
|
||||
{
|
||||
var newData = new byte[count];
|
||||
Buffer.BlockCopy(data, index, newData, 0, count);
|
||||
data = newData;
|
||||
}
|
||||
|
||||
await _udp.SendAsync(data, count, _destination).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
private async Task RunAsync(CancellationToken cancelToken)
|
||||
{
|
||||
var closeTask = Task.Delay(-1, cancelToken);
|
||||
while (!cancelToken.IsCancellationRequested)
|
||||
{
|
||||
var receiveTask = _udp.ReceiveAsync();
|
||||
var task = await Task.WhenAny(closeTask, receiveTask).ConfigureAwait(false);
|
||||
if (task == closeTask)
|
||||
break;
|
||||
|
||||
var result = receiveTask.Result;
|
||||
await ReceivedDatagram(result.Buffer, 0, result.Buffer.Length).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
|
||||
private bool UnstableCheck()
|
||||
{
|
||||
return _rand.NextDouble() > FailureRate;
|
||||
}
|
||||
}
|
||||
}
|
||||
9
src/Discord.Net.DebugTools/UnstableUdpClientProvider.cs
Normal file
9
src/Discord.Net.DebugTools/UnstableUdpClientProvider.cs
Normal file
@@ -0,0 +1,9 @@
|
||||
using Discord.Net.Udp;
|
||||
|
||||
namespace Discord.Net.Providers.UnstableUdpSocket
|
||||
{
|
||||
public static class UnstableUdpSocketProvider
|
||||
{
|
||||
public static readonly UdpSocketProvider Instance = () => new UnstableUdpSocket();
|
||||
}
|
||||
}
|
||||
237
src/Discord.Net.DebugTools/UnstableWebSocketClient.cs
Normal file
237
src/Discord.Net.DebugTools/UnstableWebSocketClient.cs
Normal file
@@ -0,0 +1,237 @@
|
||||
using Discord.Net.WebSockets;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel;
|
||||
using System.IO;
|
||||
using System.Net.WebSockets;
|
||||
using System.Text;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Discord.Net.Providers.UnstableWebSocket
|
||||
{
|
||||
internal class UnstableWebSocketClient : IWebSocketClient, IDisposable
|
||||
{
|
||||
public const int ReceiveChunkSize = 16 * 1024; //16KB
|
||||
public const int SendChunkSize = 4 * 1024; //4KB
|
||||
private const int HR_TIMEOUT = -2147012894;
|
||||
private const double FailureRate = 0.10; //10%
|
||||
|
||||
public event Func<byte[], int, int, Task> BinaryMessage;
|
||||
public event Func<string, Task> TextMessage;
|
||||
public event Func<Exception, Task> Closed;
|
||||
|
||||
private readonly SemaphoreSlim _lock;
|
||||
private readonly Dictionary<string, string> _headers;
|
||||
private readonly Random _rand;
|
||||
private ClientWebSocket _client;
|
||||
private Task _task;
|
||||
private CancellationTokenSource _cancelTokenSource;
|
||||
private CancellationToken _cancelToken, _parentToken;
|
||||
private bool _isDisposed;
|
||||
|
||||
public UnstableWebSocketClient()
|
||||
{
|
||||
_lock = new SemaphoreSlim(1, 1);
|
||||
_rand = new Random();
|
||||
_cancelTokenSource = new CancellationTokenSource();
|
||||
_cancelToken = CancellationToken.None;
|
||||
_parentToken = CancellationToken.None;
|
||||
_headers = new Dictionary<string, string>();
|
||||
}
|
||||
private void Dispose(bool disposing)
|
||||
{
|
||||
if (!_isDisposed)
|
||||
{
|
||||
if (disposing)
|
||||
DisconnectInternalAsync(true).GetAwaiter().GetResult();
|
||||
_isDisposed = true;
|
||||
}
|
||||
}
|
||||
public void Dispose()
|
||||
{
|
||||
Dispose(true);
|
||||
}
|
||||
|
||||
public async Task ConnectAsync(string host)
|
||||
{
|
||||
await _lock.WaitAsync().ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await ConnectInternalAsync(host).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
private async Task ConnectInternalAsync(string host)
|
||||
{
|
||||
await DisconnectInternalAsync().ConfigureAwait(false);
|
||||
|
||||
_cancelTokenSource = new CancellationTokenSource();
|
||||
_cancelToken = CancellationTokenSource.CreateLinkedTokenSource(_parentToken, _cancelTokenSource.Token).Token;
|
||||
|
||||
_client = new ClientWebSocket();
|
||||
_client.Options.Proxy = null;
|
||||
_client.Options.KeepAliveInterval = TimeSpan.Zero;
|
||||
foreach (var header in _headers)
|
||||
{
|
||||
if (header.Value != null)
|
||||
_client.Options.SetRequestHeader(header.Key, header.Value);
|
||||
}
|
||||
|
||||
await _client.ConnectAsync(new Uri(host), _cancelToken).ConfigureAwait(false);
|
||||
_task = RunAsync(_cancelToken);
|
||||
}
|
||||
|
||||
public async Task DisconnectAsync()
|
||||
{
|
||||
await _lock.WaitAsync().ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await DisconnectInternalAsync().ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
private async Task DisconnectInternalAsync(bool isDisposing = false)
|
||||
{
|
||||
try { _cancelTokenSource.Cancel(false); } catch { }
|
||||
|
||||
if (!isDisposing)
|
||||
await (_task ?? Task.Delay(0)).ConfigureAwait(false);
|
||||
|
||||
if (_client != null && _client.State == WebSocketState.Open)
|
||||
{
|
||||
var token = new CancellationToken();
|
||||
if (!isDisposing)
|
||||
{
|
||||
try { await _client.CloseOutputAsync(WebSocketCloseStatus.NormalClosure, "", token); }
|
||||
catch { }
|
||||
}
|
||||
try { _client.Dispose(); }
|
||||
catch { }
|
||||
_client = null;
|
||||
}
|
||||
}
|
||||
|
||||
public void SetHeader(string key, string value)
|
||||
{
|
||||
_headers[key] = value;
|
||||
}
|
||||
public void SetCancelToken(CancellationToken cancelToken)
|
||||
{
|
||||
_parentToken = cancelToken;
|
||||
_cancelToken = CancellationTokenSource.CreateLinkedTokenSource(_parentToken, _cancelTokenSource.Token).Token;
|
||||
}
|
||||
|
||||
public async Task SendAsync(byte[] data, int index, int count, bool isText)
|
||||
{
|
||||
await _lock.WaitAsync().ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
if (!UnstableCheck())
|
||||
return;
|
||||
|
||||
if (_client == null) return;
|
||||
|
||||
int frameCount = (int)Math.Ceiling((double)count / SendChunkSize);
|
||||
|
||||
for (int i = 0; i < frameCount; i++, index += SendChunkSize)
|
||||
{
|
||||
bool isLast = i == (frameCount - 1);
|
||||
|
||||
int frameSize;
|
||||
if (isLast)
|
||||
frameSize = count - (i * SendChunkSize);
|
||||
else
|
||||
frameSize = SendChunkSize;
|
||||
|
||||
var type = isText ? WebSocketMessageType.Text : WebSocketMessageType.Binary;
|
||||
await _client.SendAsync(new ArraySegment<byte>(data, index, count), type, isLast, _cancelToken).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
|
||||
private async Task RunAsync(CancellationToken cancelToken)
|
||||
{
|
||||
var buffer = new ArraySegment<byte>(new byte[ReceiveChunkSize]);
|
||||
|
||||
try
|
||||
{
|
||||
while (!cancelToken.IsCancellationRequested)
|
||||
{
|
||||
WebSocketReceiveResult socketResult = await _client.ReceiveAsync(buffer, cancelToken).ConfigureAwait(false);
|
||||
byte[] result;
|
||||
int resultCount;
|
||||
|
||||
if (socketResult.MessageType == WebSocketMessageType.Close)
|
||||
{
|
||||
var _ = Closed(new WebSocketClosedException((int)socketResult.CloseStatus, socketResult.CloseStatusDescription));
|
||||
return;
|
||||
}
|
||||
|
||||
if (!socketResult.EndOfMessage)
|
||||
{
|
||||
//This is a large message (likely just READY), lets create a temporary expandable stream
|
||||
using (var stream = new MemoryStream())
|
||||
{
|
||||
stream.Write(buffer.Array, 0, socketResult.Count);
|
||||
do
|
||||
{
|
||||
if (cancelToken.IsCancellationRequested) return;
|
||||
socketResult = await _client.ReceiveAsync(buffer, cancelToken).ConfigureAwait(false);
|
||||
stream.Write(buffer.Array, 0, socketResult.Count);
|
||||
}
|
||||
while (socketResult == null || !socketResult.EndOfMessage);
|
||||
|
||||
//Use the internal buffer if we can get it
|
||||
resultCount = (int)stream.Length;
|
||||
ArraySegment<byte> streamBuffer;
|
||||
if (stream.TryGetBuffer(out streamBuffer))
|
||||
result = streamBuffer.Array;
|
||||
else
|
||||
result = stream.ToArray();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
//Small message
|
||||
resultCount = socketResult.Count;
|
||||
result = buffer.Array;
|
||||
}
|
||||
|
||||
if (socketResult.MessageType == WebSocketMessageType.Text)
|
||||
{
|
||||
string text = Encoding.UTF8.GetString(result, 0, resultCount);
|
||||
await TextMessage(text).ConfigureAwait(false);
|
||||
}
|
||||
else
|
||||
await BinaryMessage(result, 0, resultCount).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
catch (Win32Exception ex) when (ex.HResult == HR_TIMEOUT)
|
||||
{
|
||||
var _ = Closed(new Exception("Connection timed out.", ex));
|
||||
}
|
||||
catch (OperationCanceledException) { }
|
||||
catch (Exception ex)
|
||||
{
|
||||
//This cannot be awaited otherwise we'll deadlock when DiscordApiClient waits for this task to complete.
|
||||
var _ = Closed(ex);
|
||||
}
|
||||
}
|
||||
|
||||
private bool UnstableCheck()
|
||||
{
|
||||
return _rand.NextDouble() > FailureRate;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,9 @@
|
||||
using Discord.Net.WebSockets;
|
||||
|
||||
namespace Discord.Net.Providers.UnstableWebSocket
|
||||
{
|
||||
public static class UnstableWebSocketProvider
|
||||
{
|
||||
public static readonly WebSocketProvider Instance = () => new UnstableWebSocketClient();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user