mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 11:57:55 +02:00
331 lines
11 KiB
C#
331 lines
11 KiB
C#
using MediaBrowser.Common.Net;
|
|
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;
|
|
|
|
namespace MediaBrowser.Server.Implementations.Udp
|
|
{
|
|
/// <summary>
|
|
/// Provides a Udp Server
|
|
/// </summary>
|
|
public class UdpServer : IDisposable
|
|
{
|
|
/// <summary>
|
|
/// The _logger
|
|
/// </summary>
|
|
private readonly ILogger _logger;
|
|
|
|
/// <summary>
|
|
/// The _network manager
|
|
/// </summary>
|
|
private readonly INetworkManager _networkManager;
|
|
|
|
private bool _isDisposed;
|
|
|
|
private readonly List<Tuple<string, byte[], Action<string, Encoding>>> _responders = new List<Tuple<string, byte[], Action<string, Encoding>>>();
|
|
|
|
private readonly IServerApplicationHost _appHost;
|
|
private readonly IJsonSerializer _json;
|
|
|
|
/// <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)
|
|
{
|
|
_logger = logger;
|
|
_networkManager = networkManager;
|
|
_appHost = appHost;
|
|
_json = json;
|
|
|
|
AddMessageResponder("who is EmbyServer?", RespondToV2Message);
|
|
AddMessageResponder("who is MediaBrowserServer_v2?", RespondToV2Message);
|
|
AddMessageResponder("who is MediaBrowserServer?", RespondToV1Message);
|
|
}
|
|
|
|
private void AddMessageResponder(string message, Action<string, Encoding> responder)
|
|
{
|
|
var expectedMessageBytes = Encoding.UTF8.GetBytes(message);
|
|
|
|
_responders.Add(new Tuple<string, byte[], Action<string, Encoding>>(message, expectedMessageBytes, responder));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Raises the <see cref="E:MessageReceived" /> event.
|
|
/// </summary>
|
|
/// <param name="e">The <see cref="UdpMessageReceivedEventArgs"/> instance containing the event data.</param>
|
|
private void OnMessageReceived(UdpMessageReceivedEventArgs e)
|
|
{
|
|
var responder = _responders.FirstOrDefault(i => i.Item2.SequenceEqual(e.Bytes));
|
|
var encoding = Encoding.UTF8;
|
|
|
|
if (responder == null)
|
|
{
|
|
var text = Encoding.Unicode.GetString(e.Bytes);
|
|
responder = _responders.FirstOrDefault(i => string.Equals(i.Item1, text, StringComparison.OrdinalIgnoreCase));
|
|
|
|
if (responder != null)
|
|
{
|
|
encoding = Encoding.Unicode;
|
|
}
|
|
}
|
|
|
|
if (responder != null)
|
|
{
|
|
try
|
|
{
|
|
responder.Item3(e.RemoteEndPoint, encoding);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error in OnMessageReceived", ex);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async void RespondToV1Message(string endpoint, Encoding encoding)
|
|
{
|
|
var localUrl = await _appHost.GetLocalApiUrl().ConfigureAwait(false);
|
|
|
|
if (!string.IsNullOrEmpty(localUrl))
|
|
{
|
|
// This is how we did the old v1 search, so need to strip off the protocol
|
|
var index = localUrl.IndexOf("://", StringComparison.OrdinalIgnoreCase);
|
|
|
|
if (index != -1)
|
|
{
|
|
localUrl = localUrl.Substring(index + 3);
|
|
}
|
|
|
|
// Send a response back with our ip address and port
|
|
var response = String.Format("MediaBrowserServer|{0}", localUrl);
|
|
|
|
await SendAsync(Encoding.UTF8.GetBytes(response), endpoint);
|
|
}
|
|
else
|
|
{
|
|
_logger.Warn("Unable to respond to udp request because the local ip address could not be determined.");
|
|
}
|
|
}
|
|
|
|
private async void RespondToV2Message(string endpoint, Encoding encoding)
|
|
{
|
|
var localUrl = await _appHost.GetLocalApiUrl().ConfigureAwait(false);
|
|
|
|
if (!string.IsNullOrEmpty(localUrl))
|
|
{
|
|
var response = new ServerDiscoveryInfo
|
|
{
|
|
Address = localUrl,
|
|
Id = _appHost.SystemId,
|
|
Name = _appHost.FriendlyName
|
|
};
|
|
|
|
await SendAsync(encoding.GetBytes(_json.SerializeToString(response)), endpoint);
|
|
}
|
|
else
|
|
{
|
|
_logger.Warn("Unable to respond to udp request because the local ip address could not be determined.");
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// The _udp client
|
|
/// </summary>
|
|
private UdpClient _udpClient;
|
|
|
|
/// <summary>
|
|
/// Starts the specified port.
|
|
/// </summary>
|
|
/// <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);
|
|
|
|
Task.Run(() => StartListening());
|
|
}
|
|
|
|
private async void StartListening()
|
|
{
|
|
while (!_isDisposed)
|
|
{
|
|
try
|
|
{
|
|
var result = await GetResult().ConfigureAwait(false);
|
|
|
|
OnMessageReceived(result);
|
|
}
|
|
catch (ObjectDisposedException)
|
|
{
|
|
break;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error in StartListening", 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(UdpReceiveResult message)
|
|
{
|
|
if (message.RemoteEndPoint.Port == 0)
|
|
{
|
|
return;
|
|
}
|
|
var bytes = message.Buffer;
|
|
|
|
try
|
|
{
|
|
OnMessageReceived(new UdpMessageReceivedEventArgs
|
|
{
|
|
Bytes = bytes,
|
|
RemoteEndPoint = message.RemoteEndPoint.ToString()
|
|
});
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error handling UDP message", ex);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
|
|
/// </summary>
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Stops this instance.
|
|
/// </summary>
|
|
public void Stop()
|
|
{
|
|
_isDisposed = true;
|
|
|
|
if (_udpClient != null)
|
|
{
|
|
_udpClient.Close();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Releases unmanaged and - optionally - managed resources.
|
|
/// </summary>
|
|
/// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
|
|
protected virtual void Dispose(bool dispose)
|
|
{
|
|
if (dispose)
|
|
{
|
|
Stop();
|
|
}
|
|
}
|
|
|
|
/// <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)
|
|
{
|
|
if (bytes == null)
|
|
{
|
|
throw new ArgumentNullException("bytes");
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(remoteEndPoint))
|
|
{
|
|
throw new ArgumentNullException("remoteEndPoint");
|
|
}
|
|
|
|
try
|
|
{
|
|
await _udpClient.SendAsync(bytes, bytes.Length, _networkManager.Parse(remoteEndPoint)).ConfigureAwait(false);
|
|
|
|
_logger.Info("Udp message sent to {0}", remoteEndPoint);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error sending message to {0}", ex, remoteEndPoint);
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|