2013-02-23 23:31:51 +01:00
|
|
|
using System;
|
2019-07-07 21:03:26 +02:00
|
|
|
using System.Net;
|
2020-01-12 18:59:10 +01:00
|
|
|
using System.Net.Sockets;
|
2013-02-23 23:31:51 +01:00
|
|
|
using System.Text;
|
2020-01-12 18:59:10 +01:00
|
|
|
using System.Text.Json;
|
2017-02-05 21:44:08 +01:00
|
|
|
using System.Threading;
|
2013-02-23 23:31:51 +01:00
|
|
|
using System.Threading.Tasks;
|
2019-01-13 20:23:38 +01:00
|
|
|
using MediaBrowser.Controller;
|
|
|
|
using MediaBrowser.Model.ApiClient;
|
2020-05-02 18:56:09 +02:00
|
|
|
using Microsoft.Extensions.Configuration;
|
2019-01-13 20:23:38 +01:00
|
|
|
using Microsoft.Extensions.Logging;
|
2013-02-23 23:31:51 +01:00
|
|
|
|
2016-11-04 19:56:47 +01:00
|
|
|
namespace Emby.Server.Implementations.Udp
|
2013-02-23 23:31:51 +01:00
|
|
|
{
|
|
|
|
/// <summary>
|
2020-01-12 18:59:10 +01:00
|
|
|
/// Provides a Udp Server.
|
2013-02-23 23:31:51 +01:00
|
|
|
/// </summary>
|
2020-01-12 18:59:10 +01:00
|
|
|
public sealed class UdpServer : IDisposable
|
2013-02-23 23:31:51 +01:00
|
|
|
{
|
2021-05-28 14:33:54 +02:00
|
|
|
/// <summary>
|
|
|
|
/// Address Override Configuration Key.
|
|
|
|
/// </summary>
|
|
|
|
public const string AddressOverrideConfigKey = "PublishedServerUrl";
|
|
|
|
|
2013-02-23 23:31:51 +01:00
|
|
|
/// <summary>
|
2020-06-16 00:37:52 +02:00
|
|
|
/// The _logger.
|
2013-02-23 23:31:51 +01:00
|
|
|
/// </summary>
|
2013-03-27 23:13:46 +01:00
|
|
|
private readonly ILogger _logger;
|
2014-07-28 00:01:29 +02:00
|
|
|
private readonly IServerApplicationHost _appHost;
|
2020-05-02 18:56:09 +02:00
|
|
|
private readonly IConfiguration _config;
|
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
private Socket _udpSocket;
|
|
|
|
private IPEndPoint _endpoint;
|
|
|
|
private readonly byte[] _receiveBuffer = new byte[8192];
|
2014-07-28 00:01:29 +02:00
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
private bool _disposed = false;
|
2013-02-26 17:10:55 +01:00
|
|
|
|
2013-02-23 23:31:51 +01:00
|
|
|
/// <summary>
|
2020-01-12 18:59:10 +01:00
|
|
|
/// Initializes a new instance of the <see cref="UdpServer" /> class.
|
2013-02-23 23:31:51 +01:00
|
|
|
/// </summary>
|
2021-05-28 14:33:54 +02:00
|
|
|
/// <param name="logger">The logger.</param>
|
|
|
|
/// <param name="appHost">The application host.</param>
|
|
|
|
/// <param name="configuration">The configuration manager.</param>
|
|
|
|
/// <param name="port">The port.</param>
|
|
|
|
public UdpServer(
|
|
|
|
ILogger logger,
|
|
|
|
IServerApplicationHost appHost,
|
|
|
|
IConfiguration configuration,
|
|
|
|
int port)
|
2016-09-03 19:16:36 +02:00
|
|
|
{
|
2020-01-12 18:59:10 +01:00
|
|
|
_logger = logger;
|
|
|
|
_appHost = appHost;
|
2020-05-02 18:56:09 +02:00
|
|
|
_config = configuration;
|
2021-05-28 14:33:54 +02:00
|
|
|
|
|
|
|
_endpoint = new IPEndPoint(IPAddress.Any, port);
|
|
|
|
|
|
|
|
_udpSocket = new Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp);
|
|
|
|
_udpSocket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
|
2016-09-03 19:16:36 +02:00
|
|
|
}
|
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
private async Task RespondToV2Message(string messageText, EndPoint endpoint, CancellationToken cancellationToken)
|
2014-07-28 00:01:29 +02:00
|
|
|
{
|
2021-05-28 14:33:54 +02:00
|
|
|
string? localUrl = _config[AddressOverrideConfigKey];
|
|
|
|
if (string.IsNullOrEmpty(localUrl))
|
|
|
|
{
|
|
|
|
localUrl = _appHost.GetSmartApiUrl(((IPEndPoint)endpoint).Address);
|
|
|
|
}
|
2013-03-27 23:13:46 +01:00
|
|
|
|
2021-05-28 14:33:54 +02:00
|
|
|
if (string.IsNullOrEmpty(localUrl))
|
2014-07-28 00:01:29 +02:00
|
|
|
{
|
2021-05-28 14:33:54 +02:00
|
|
|
_logger.LogWarning("Unable to respond to udp request because the local ip address could not be determined.");
|
|
|
|
return;
|
|
|
|
}
|
2014-07-28 00:01:29 +02:00
|
|
|
|
2021-05-28 14:33:54 +02:00
|
|
|
var response = new ServerDiscoveryInfo(localUrl, _appHost.SystemId, _appHost.FriendlyName);
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
await _udpSocket.SendToAsync(JsonSerializer.SerializeToUtf8Bytes(response), SocketFlags.None, endpoint).ConfigureAwait(false);
|
2014-07-28 00:01:29 +02:00
|
|
|
}
|
2021-05-28 14:33:54 +02:00
|
|
|
catch (SocketException ex)
|
2014-07-28 00:01:29 +02:00
|
|
|
{
|
2021-05-28 14:33:54 +02:00
|
|
|
_logger.LogError(ex, "Error sending response message");
|
2013-03-27 23:13:46 +01:00
|
|
|
}
|
2013-02-23 23:31:51 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
/// Starts the specified port.
|
|
|
|
/// </summary>
|
2021-02-23 16:45:10 +01:00
|
|
|
/// <param name="cancellationToken">The cancellation token to cancel operation.</param>
|
2021-05-28 14:33:54 +02:00
|
|
|
public void Start(CancellationToken cancellationToken)
|
2013-12-07 16:52:38 +01:00
|
|
|
{
|
2020-01-12 18:59:10 +01:00
|
|
|
_udpSocket.Bind(_endpoint);
|
2017-05-24 21:12:55 +02:00
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
_ = Task.Run(async () => await BeginReceiveAsync(cancellationToken).ConfigureAwait(false), cancellationToken).ConfigureAwait(false);
|
2013-12-07 16:52:38 +01:00
|
|
|
}
|
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
private async Task BeginReceiveAsync(CancellationToken cancellationToken)
|
2013-02-23 23:31:51 +01:00
|
|
|
{
|
2021-05-28 14:33:54 +02:00
|
|
|
var infiniteTask = Task.Delay(-1, cancellationToken);
|
2020-01-12 18:59:10 +01:00
|
|
|
while (!cancellationToken.IsCancellationRequested)
|
2013-02-26 17:10:55 +01:00
|
|
|
{
|
2020-01-12 18:59:10 +01:00
|
|
|
try
|
2015-05-17 05:17:23 +02:00
|
|
|
{
|
2020-06-17 19:20:43 +02:00
|
|
|
var task = _udpSocket.ReceiveFromAsync(_receiveBuffer, SocketFlags.None, _endpoint);
|
|
|
|
await Task.WhenAny(task, infiniteTask).ConfigureAwait(false);
|
2013-02-23 23:31:51 +01:00
|
|
|
|
2020-06-17 19:20:43 +02:00
|
|
|
if (!task.IsCompleted)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
var result = task.Result;
|
2017-03-26 21:00:35 +02:00
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
var text = Encoding.UTF8.GetString(_receiveBuffer, 0, result.ReceivedBytes);
|
|
|
|
if (text.Contains("who is JellyfinServer?", StringComparison.OrdinalIgnoreCase))
|
|
|
|
{
|
|
|
|
await RespondToV2Message(text, result.RemoteEndPoint, cancellationToken).ConfigureAwait(false);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (SocketException ex)
|
2017-03-26 21:00:35 +02:00
|
|
|
{
|
2020-05-02 18:56:09 +02:00
|
|
|
_logger.LogError(ex, "Failed to receive data from socket");
|
2020-01-12 18:59:10 +01:00
|
|
|
}
|
|
|
|
catch (OperationCanceledException)
|
|
|
|
{
|
|
|
|
// Don't throw
|
2017-03-26 21:00:35 +02:00
|
|
|
}
|
2013-02-23 23:31:51 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
/// <inheritdoc />
|
|
|
|
public void Dispose()
|
2013-02-23 23:31:51 +01:00
|
|
|
{
|
2020-01-12 18:59:10 +01:00
|
|
|
if (_disposed)
|
2016-12-28 07:08:18 +01:00
|
|
|
{
|
2020-01-12 18:59:10 +01:00
|
|
|
return;
|
2013-02-23 23:31:51 +01:00
|
|
|
}
|
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
_udpSocket?.Dispose();
|
2013-03-12 23:51:10 +01:00
|
|
|
|
2020-01-12 18:59:10 +01:00
|
|
|
GC.SuppressFinalize(this);
|
2013-02-23 23:31:51 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|