@ -1,18 +1,16 @@
using MediaBrowser.Common.Net ;
using MediaBrowser.Controller ;
using MediaBrowser.Controller ;
using MediaBrowser.Model.ApiClient ;
using MediaBrowser.Model.Logging ;
using MediaBrowser.Model.Serialization ;
using System ;
using System.Collections.Generic ;
using System.Linq ;
using System.Net ;
using System.Net.Sockets ;
using System.Text ;
using System.Threading.Tasks ;
using Emby.Common.Implementations.Networking ;
using MediaBrowser.Model.Events ;
using MediaBrowser.Model.Net ;
namespace MediaBrowser .Server.Implementations.Udp
namespace Emby .Server.Implementations.Udp
{
/// <summary>
/// Provides a Udp Server
@ -24,14 +22,9 @@ namespace MediaBrowser.Server.Implementations.Udp
/// </summary>
private readonly ILogger _logger ;
/// <summary>
/// The _network manager
/// </summary>
private readonly INetworkManager _networkManager ;
private bool _isDisposed ;
private readonly List < Tuple < string , bool , Func < string , string , Encoding , Task > > > _responders = new List < Tuple < string , bool , Func < string , string , Encoding , Task > > > ( ) ;
private readonly List < Tuple < string , bool , Func < string , IpEndPointInfo , Encoding , Task > > > _responders = new List < Tuple < string , bool , Func < string , IpEndPointInfo , Encoding , Task > > > ( ) ;
private readonly IServerApplicationHost _appHost ;
private readonly IJsonSerializer _json ;
@ -39,46 +32,43 @@ namespace MediaBrowser.Server.Implementations.Udp
/// <summary>
/// Initializes a new instance of the <see cref="UdpServer" /> class.
/// </summary>
/// <param name="logger">The logger.</param>
/// <param name="networkManager">The network manager.</param>
/// <param name="appHost">The application host.</param>
/// <param name="json">The json.</param>
public UdpServer ( ILogger logger , INetworkManager networkManager , IServerApplicationHost appHost , IJsonSerializer json )
public UdpServer ( ILogger logger , IServerApplicationHost appHost , IJsonSerializer json , ISocketFactory socketFactory )
{
_logger = logger ;
_networkManager = networkManager ;
_appHost = appHost ;
_json = json ;
_socketFactory = socketFactory ;
AddMessageResponder ( "who is EmbyServer?" , true , RespondToV2Message ) ;
AddMessageResponder ( "who is MediaBrowserServer_v2?" , false , RespondToV2Message ) ;
}
private void AddMessageResponder ( string message , bool isSubstring , Func < string , string , Encoding , Task > responder )
private void AddMessageResponder ( string message , bool isSubstring , Func < string , IpEndPointInfo , Encoding , Task > responder )
{
_responders . Add ( new Tuple < string , bool , Func < string , string , Encoding , Task > > ( message , isSubstring , responder ) ) ;
_responders . Add ( new Tuple < string , bool , Func < string , IpEndPointInfo , Encoding , Task > > ( message , isSubstring , responder ) ) ;
}
/// <summary>
/// Raises the <see cref="E:MessageReceived" /> event.
/// </summary>
/// <param name="e">The <see cref="UdpMessageReceivedEventArgs"/> instance containing the event data.</param>
private async void OnMessageReceived ( UdpMessageReceivedEventArgs e )
private async void OnMessageReceived ( GenericEventArgs < SocketReceiveResult > e )
{
var message = e . Argument ;
var encoding = Encoding . UTF8 ;
var responder = GetResponder ( e . Bytes , encoding ) ;
var responder = GetResponder ( messag e. Buffer, message . Received Bytes, encoding ) ;
if ( responder = = null )
{
encoding = Encoding . Unicode ;
responder = GetResponder ( . , encoding ) ;
responder = GetResponder ( messag e. Buffer, message . Received Bytes, encoding ) ;
}
if ( responder ! = null )
{
try
{
await responder . Item2 . Item3 ( responder . Item1 , . RemoteEndPoint , encoding ) . ConfigureAwait ( false ) ;
await responder . Item2 . Item3 ( responder . Item1 , messag e. RemoteEndPoint , encoding ) . ConfigureAwait ( false ) ;
}
catch ( Exception ex )
{
@ -87,9 +77,9 @@ namespace MediaBrowser.Server.Implementations.Udp
}
}
private Tuple < string , Tuple < string , bool , Func < string , string , Encoding , Task > > > GetResponder ( byte [ ] b ytes, Encoding encoding )
private Tuple < string , Tuple < string , bool , Func < string , IpEndPointInfo , Encoding , Task > > > GetResponder ( byte [ ] b uffer, int b ytesReceived , Encoding encoding )
{
var text = encoding . GetString ( b ytes) ;
var text = encoding . GetString ( b uffer, 0 , b ytesReceived ) ;
var responder = _responders . FirstOrDefault ( i = >
{
if ( i . Item2 )
@ -103,10 +93,10 @@ namespace MediaBrowser.Server.Implementations.Udp
{
return null ;
}
return new Tuple < string , Tuple < string , bool , Func < string , string , Encoding , Task > > > ( text , responder ) ;
return new Tuple < string , Tuple < string , bool , Func < string , IpEndPointInfo , Encoding , Task > > > ( text , responder ) ;
}
private async Task RespondToV2Message ( string messageText , string endpoint , Encoding encoding )
private async Task RespondToV2Message ( string messageText , IpEndPointInfo endpoint , Encoding encoding )
{
var parts = messageText . Split ( '|' ) ;
@ -122,7 +112,7 @@ namespace MediaBrowser.Server.Implementations.Udp
} ;
await SendAsync ( encoding . GetBytes ( _json . SerializeToString ( response ) ) , endpoint ) . ConfigureAwait ( false ) ;
if ( parts . Length > 1 )
{
_appHost . EnableLoopback ( parts [ 1 ] ) ;
@ -137,7 +127,8 @@ namespace MediaBrowser.Server.Implementations.Udp
/// <summary>
/// The _udp client
/// </summary>
private UdpClient _udpClient ;
private IUdpSocket _udpClient ;
private readonly ISocketFactory _socketFactory ;
/// <summary>
/// Starts the specified port.
@ -145,9 +136,7 @@ namespace MediaBrowser.Server.Implementations.Udp
/// <param name="port">The port.</param>
public void Start ( int port )
{
_udpClient = new UdpClient ( new IPEndPoint ( IPAddress . Any , port ) ) ;
_udpClient . Client . SetSocketOption ( SocketOptionLevel . Socket , SocketOptionName . ReuseAddress , true ) ;
_udpClient = _socketFactory . CreateUdpSocket ( port ) ;
Task . Run ( ( ) = > StartListening ( ) ) ;
}
@ -158,56 +147,36 @@ namespace MediaBrowser.Server.Implementations.Udp
{
try
{
var result = await GetResult ( ) . ConfigureAwait ( false ) ;
var result = await _udpClient. ReceiveAsync ( ) . ConfigureAwait ( false ) ;
OnMessageReceived ( result ) ;
}
catch ( ObjectDisposedException )
{
break ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error in StartListening ", ex ) ;
_logger . ErrorException ( "Error receiving udp message ", ex ) ;
}
}
}
private Task < UdpReceiveResult > GetResult ( )
{
try
{
return _udpClient . ReceiveAsync ( ) ;
}
catch ( ObjectDisposedException )
{
return Task . FromResult ( new UdpReceiveResult ( new byte [ ] { } , new IPEndPoint ( IPAddress . Any , 0 ) ) ) ;
}
catch ( Exception ex )
{
_logger . ErrorException ( "Error receiving udp message" , ex ) ;
return Task . FromResult ( new UdpReceiveResult ( new byte [ ] { } , new IPEndPoint ( IPAddress . Any , 0 ) ) ) ;
}
}
/// <summary>
/// Called when [message received].
/// </summary>
/// <param name="message">The message.</param>
private void OnMessageReceived ( Udp ReceiveResult message )
private void OnMessageReceived ( SocketReceiveResult message )
{
if ( message . RemoteEndPoint . Port = = 0 )
{
return ;
}
var bytes = message . Buffer ;
try
{
OnMessageReceived ( new UdpMessageReceivedEventArgs
OnMessageReceived ( new GenericEventArgs < SocketReceiveResult >
{
Bytes = bytes ,
RemoteEndPoint = message . RemoteEndPoint . ToString ( )
Argument = message
} ) ;
}
catch ( Exception ex )
@ -234,7 +203,7 @@ namespace MediaBrowser.Server.Implementations.Udp
if ( _udpClient ! = null )
{
_udpClient . Cl ose( ) ;
_udpClient . Disp ose( ) ;
}
}
@ -250,71 +219,21 @@ namespace MediaBrowser.Server.Implementations.Udp
}
}
/// <summary>
/// Sends the async.
/// </summary>
/// <param name="data">The data.</param>
/// <param name="ipAddress">The ip address.</param>
/// <param name="port">The port.</param>
/// <returns>Task{System.Int32}.</returns>
/// <exception cref="System.ArgumentNullException">data</exception>
public Task SendAsync ( string data , string ipAddress , int port )
{
return SendAsync ( Encoding . UTF8 . GetBytes ( data ) , ipAddress , port ) ;
}
/// <summary>
/// Sends the async.
/// </summary>
/// <param name="bytes">The bytes.</param>
/// <param name="ipAddress">The ip address.</param>
/// <param name="port">The port.</param>
/// <returns>Task{System.Int32}.</returns>
/// <exception cref="System.ArgumentNullException">bytes</exception>
public Task SendAsync ( byte [ ] bytes , string ipAddress , int port )
{
if ( bytes = = null )
{
throw new ArgumentNullException ( "bytes" ) ;
}
if ( string . IsNullOrEmpty ( ipAddress ) )
{
throw new ArgumentNullException ( "ipAddress" ) ;
}
return _udpClient . SendAsync ( bytes , bytes . Length , ipAddress , port ) ;
}
/// <summary>
/// Sends the async.
/// </summary>
/// <param name="bytes">The bytes.</param>
/// <param name="remoteEndPoint">The remote end point.</param>
/// <returns>Task.</returns>
/// <exception cref="System.ArgumentNullException">
/// bytes
/// or
/// remoteEndPoint
/// </exception>
public async Task SendAsync ( byte [ ] bytes , string remoteEndPoint )
public async Task SendAsync ( byte [ ] bytes , IpEndPointInfo remoteEndPoint )
{
if ( bytes = = null )
{
throw new ArgumentNullException ( "bytes" ) ;
}
if ( string . IsNullOrEmpty ( remoteEndPoint ) )
if ( remoteEndPoint = = null )
{
throw new ArgumentNullException ( "remoteEndPoint" ) ;
}
try
{
// Need to do this until Common will compile with this method
var nativeNetworkManager = ( BaseNetworkManager ) _networkManager ;
await _udpClient . SendAsync ( bytes , bytes . Length , nativeNetworkManager . Parse ( remoteEndPoint ) ) . ConfigureAwait ( false ) ;
await _udpClient . SendAsync ( bytes , bytes . Length , remoteEndPoint ) . ConfigureAwait ( false ) ;
_logger . Info ( "Udp message sent to {0}" , remoteEndPoint ) ;
}
Luke Pulverenti
8 years ago