github/Discord.Net
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added UDPClient Provider

Browse Source
This commit is contained in:
RogueException
2016-12-23 15:07:59 -04:00
parent 357f42ffdb
commit ca6eb6aff4
4 changed files with 201 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/Discord.Net.Providers.UDPClient/Discord.Net.Providers.UdpClient.csproj Normal file
View File

@@ -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
View 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
View 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
View 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": {}
}
}