Added UDPClient Provider
This commit is contained in:
@@ -0,0 +1,27 @@
|
|||||||
|
<Project ToolsVersion="15.0" Sdk="Microsoft.NET.Sdk" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||||
|
<PropertyGroup>
|
||||||
|
<Description>An optional UDP client provider for Discord.Net using System.Net.UdpClient</Description>
|
||||||
|
<VersionPrefix>1.0.0-beta2</VersionPrefix>
|
||||||
|
<TargetFrameworks>net45</TargetFrameworks>
|
||||||
|
<AllowUnsafeBlocks>true</AllowUnsafeBlocks>
|
||||||
|
<AssemblyName>Discord.Net.Providers.UDPClient</AssemblyName>
|
||||||
|
<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.Providers.UDPClient</RootNamespace>
|
||||||
|
</PropertyGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<Compile Include="**\*.cs" />
|
||||||
|
<EmbeddedResource Include="**\*.resx" />
|
||||||
|
</ItemGroup>
|
||||||
|
<ItemGroup>
|
||||||
|
<ProjectReference Include="..\Discord.Net.Core\Discord.Net.Core.csproj" />
|
||||||
|
</ItemGroup>
|
||||||
|
<PropertyGroup Condition=" '$(Configuration)' == 'Release' ">
|
||||||
|
<NoWarn>$(NoWarn);CS1573;CS1591</NoWarn>
|
||||||
|
<WarningsAsErrors>true</WarningsAsErrors>
|
||||||
|
<GenerateDocumentationFile>true</GenerateDocumentationFile>
|
||||||
|
</PropertyGroup>
|
||||||
|
</Project>
|
||||||
128
src/Discord.Net.Providers.UDPClient/UDPClient.cs
Normal file
128
src/Discord.Net.Providers.UDPClient/UDPClient.cs
Normal file
@@ -0,0 +1,128 @@
|
|||||||
|
using Discord.Net.Udp;
|
||||||
|
using System;
|
||||||
|
using System.Net;
|
||||||
|
using System.Threading;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using UdpSocket = System.Net.Sockets.UdpClient;
|
||||||
|
|
||||||
|
namespace Discord.Net.Providers.UDPClient
|
||||||
|
{
|
||||||
|
internal class UDPClient : IUdpSocket, IDisposable
|
||||||
|
{
|
||||||
|
public event Func<byte[], int, int, Task> ReceivedDatagram;
|
||||||
|
|
||||||
|
private readonly SemaphoreSlim _lock;
|
||||||
|
private UdpSocket _udp;
|
||||||
|
private IPEndPoint _destination;
|
||||||
|
private CancellationTokenSource _cancelTokenSource;
|
||||||
|
private CancellationToken _cancelToken, _parentToken;
|
||||||
|
private Task _task;
|
||||||
|
private bool _isDisposed;
|
||||||
|
|
||||||
|
public UDPClient()
|
||||||
|
{
|
||||||
|
_lock = new SemaphoreSlim(1, 1);
|
||||||
|
}
|
||||||
|
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 UdpSocket();
|
||||||
|
|
||||||
|
_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.Close(); }
|
||||||
|
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 (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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
9
src/Discord.Net.Providers.UDPClient/UDPClientProvider.cs
Normal file
9
src/Discord.Net.Providers.UDPClient/UDPClientProvider.cs
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
using Discord.Net.Udp;
|
||||||
|
|
||||||
|
namespace Discord.Net.Providers.UDPClient
|
||||||
|
{
|
||||||
|
public static class UDPClientProvider
|
||||||
|
{
|
||||||
|
public static readonly UdpSocketProvider Instance = () => new UDPClient();
|
||||||
|
}
|
||||||
|
}
|
||||||
37
src/Discord.Net.Providers.UDPClient/project.json
Normal file
37
src/Discord.Net.Providers.UDPClient/project.json
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
{
|
||||||
|
"version": "1.0.0-*",
|
||||||
|
"description": "An optional UDP client provider for Discord.Net using System.Net.UdpClient.",
|
||||||
|
"authors": [ "RogueException" ],
|
||||||
|
|
||||||
|
"packOptions": {
|
||||||
|
"tags": [ "discord", "discordapp" ],
|
||||||
|
"licenseUrl": "http://opensource.org/licenses/MIT",
|
||||||
|
"projectUrl": "https://github.com/RogueException/Discord.Net",
|
||||||
|
"repository": {
|
||||||
|
"type": "git",
|
||||||
|
"url": "git://github.com/RogueException/Discord.Net"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
"configurations": {
|
||||||
|
"Release": {
|
||||||
|
"buildOptions": {
|
||||||
|
"define": [ "RELEASE" ],
|
||||||
|
"nowarn": [ "CS1573", "CS1591" ],
|
||||||
|
"optimize": true,
|
||||||
|
"warningsAsErrors": true,
|
||||||
|
"xmlDoc": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
"dependencies": {
|
||||||
|
"Discord.Net.Core": {
|
||||||
|
"target": "project"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
"frameworks": {
|
||||||
|
"net45": {}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user