jellyfin/MediaBrowser.Dlna/Ssdp/SsdpHandler.cs

479 lines
16 KiB
C#
Raw Normal View History

2014-11-07 00:16:16 +00:00
using MediaBrowser.Common.Configuration;
2014-06-22 05:52:31 +00:00
using MediaBrowser.Common.Events;
2014-04-27 03:42:05 +00:00
using MediaBrowser.Controller.Configuration;
2014-04-25 17:30:41 +00:00
using MediaBrowser.Dlna.Server;
2014-03-24 12:47:39 +00:00
using MediaBrowser.Model.Logging;
using System;
2014-04-10 15:06:54 +00:00
using System.Collections.Concurrent;
2014-03-24 12:47:39 +00:00
using System.Collections.Generic;
2014-11-07 00:16:16 +00:00
using System.Globalization;
2014-03-24 12:47:39 +00:00
using System.Linq;
using System.Net;
using System.Net.Sockets;
2014-06-29 17:35:05 +00:00
using System.Text;
2014-04-10 15:06:54 +00:00
using System.Threading;
2014-11-07 00:16:16 +00:00
using System.Threading.Tasks;
2014-03-24 12:47:39 +00:00
2014-04-25 17:30:41 +00:00
namespace MediaBrowser.Dlna.Ssdp
2014-03-24 12:47:39 +00:00
{
public class SsdpHandler : IDisposable
{
2014-04-25 17:30:41 +00:00
private Socket _socket;
2014-04-10 15:06:54 +00:00
2014-03-24 12:47:39 +00:00
private readonly ILogger _logger;
private readonly IServerConfigurationManager _config;
const string SSDPAddr = "239.255.255.250";
const int SSDPPort = 1900;
2014-04-25 17:30:41 +00:00
private readonly string _serverSignature;
2014-03-24 12:47:39 +00:00
private readonly IPAddress _ssdpIp = IPAddress.Parse(SSDPAddr);
2014-04-25 17:30:41 +00:00
private readonly IPEndPoint _ssdpEndp = new IPEndPoint(IPAddress.Parse(SSDPAddr), SSDPPort);
2014-03-24 12:47:39 +00:00
2014-04-10 15:06:54 +00:00
private Timer _queueTimer;
private Timer _notificationTimer;
2014-04-27 03:42:05 +00:00
2014-04-25 17:30:41 +00:00
private readonly AutoResetEvent _datagramPosted = new AutoResetEvent(false);
private readonly ConcurrentQueue<Datagram> _messageQueue = new ConcurrentQueue<Datagram>();
2014-04-10 15:06:54 +00:00
2014-04-25 17:30:41 +00:00
private bool _isDisposed;
private readonly ConcurrentDictionary<Guid, List<UpnpDevice>> _devices = new ConcurrentDictionary<Guid, List<UpnpDevice>>();
2014-04-27 03:42:05 +00:00
2014-03-24 12:47:39 +00:00
public SsdpHandler(ILogger logger, IServerConfigurationManager config, string serverSignature)
{
_logger = logger;
_config = config;
_serverSignature = serverSignature;
2014-04-28 15:05:28 +00:00
2014-06-29 17:35:05 +00:00
_config.NamedConfigurationUpdated += _config_ConfigurationUpdated;
2014-04-28 15:05:28 +00:00
}
2014-06-29 17:35:05 +00:00
void _config_ConfigurationUpdated(object sender, ConfigurationUpdateEventArgs e)
2014-04-28 15:05:28 +00:00
{
2014-06-29 17:35:05 +00:00
if (string.Equals(e.Key, "dlna", StringComparison.OrdinalIgnoreCase))
{
ReloadAliveNotifier();
}
2014-04-25 17:30:41 +00:00
}
public event EventHandler<SsdpMessageEventArgs> MessageReceived;
2014-03-24 12:47:39 +00:00
2014-08-12 02:59:51 +00:00
private async void OnMessageReceived(SsdpMessageEventArgs args)
2014-04-25 17:30:41 +00:00
{
if (string.Equals(args.Method, "M-SEARCH", StringComparison.OrdinalIgnoreCase))
{
TimeSpan delay = GetSearchDelay(args.Headers);
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-08-12 02:59:51 +00:00
{
_logger.Debug("Delaying search response by {0} seconds", delay.TotalSeconds);
2014-08-12 02:59:51 +00:00
}
await Task.Delay(delay).ConfigureAwait(false);
2014-08-12 02:59:51 +00:00
2014-04-25 17:30:41 +00:00
RespondToSearch(args.EndPoint, args.Headers["st"]);
}
2014-04-27 03:42:05 +00:00
EventHelper.FireEventIfNotNull(MessageReceived, this, args, _logger);
2014-03-24 12:47:39 +00:00
}
2014-04-25 17:30:41 +00:00
public IEnumerable<UpnpDevice> RegisteredDevices
2014-03-24 12:47:39 +00:00
{
get
{
2014-04-25 17:30:41 +00:00
return _devices.Values.SelectMany(i => i).ToList();
2014-03-24 12:47:39 +00:00
}
}
2014-04-27 03:42:05 +00:00
2014-04-25 17:30:41 +00:00
public void Start()
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
_socket = CreateMulticastSocket();
2014-03-24 12:47:39 +00:00
_logger.Info("SSDP service started");
Receive();
2014-04-10 15:06:54 +00:00
2014-04-28 15:05:28 +00:00
ReloadAliveNotifier();
2014-03-24 12:47:39 +00:00
}
public void SendSearchMessage(EndPoint localIp)
2014-06-22 05:52:31 +00:00
{
var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
values["HOST"] = "239.255.255.250:1900";
values["USER-AGENT"] = "UPnP/1.0 DLNADOC/1.50 Platinum/1.0.4.2";
2014-07-30 03:31:35 +00:00
2014-06-22 05:52:31 +00:00
values["MAN"] = "\"ssdp:discover\"";
2014-07-30 03:31:35 +00:00
// Search target
values["ST"] = "ssdp:all";
// Seconds to delay response
values["MX"] = "3";
2014-06-22 05:52:31 +00:00
SendDatagram("M-SEARCH * HTTP/1.1", values, localIp);
}
2014-04-25 17:30:41 +00:00
public void SendDatagram(string header,
Dictionary<string, string> values,
EndPoint localAddress,
2014-04-25 17:30:41 +00:00
int sendCount = 1)
{
2014-10-08 02:25:24 +00:00
SendDatagram(header, values, _ssdpEndp, localAddress, false, sendCount);
2014-04-25 17:30:41 +00:00
}
2014-04-27 03:42:05 +00:00
public void SendDatagram(string header,
Dictionary<string, string> values,
EndPoint endpoint,
EndPoint localAddress,
2014-10-08 02:25:24 +00:00
bool ignoreBindFailure,
2014-04-25 17:30:41 +00:00
int sendCount = 1)
{
var msg = new SsdpMessageBuilder().BuildMessage(header, values);
2015-01-31 23:12:22 +00:00
var queued = false;
2014-04-25 17:30:41 +00:00
2015-01-31 23:12:22 +00:00
for (var i = 0; i < sendCount; i++)
2014-04-25 17:30:41 +00:00
{
2015-01-31 23:12:22 +00:00
var dgram = new Datagram(endpoint, localAddress, _logger, msg, ignoreBindFailure);
if (_messageQueue.Count == 0)
{
dgram.Send();
}
else
{
_messageQueue.Enqueue(dgram);
queued = true;
}
2014-04-25 17:30:41 +00:00
}
2015-01-31 23:12:22 +00:00
if (queued)
{
StartQueueTimer();
}
2014-04-25 17:30:41 +00:00
}
/// <summary>
/// According to the spec: http://www.upnp.org/specs/arch/UPnP-arch-DeviceArchitecture-v1.0-20080424.pdf
/// Device responses should be delayed a random duration between 0 and this many seconds to balance
/// load for the control point when it processes responses. In my testing kodi times out after mx
/// so we will generate from mx - 1
/// </summary>
/// <param name="headers">The mx headers</param>
/// <returns>A timepsan for the amount to delay before returning search result.</returns>
private TimeSpan GetSearchDelay(Dictionary<string, string> headers)
{
string mx;
headers.TryGetValue("mx", out mx);
int delaySeconds = 0;
if (!string.IsNullOrWhiteSpace(mx)
&& int.TryParse(mx, NumberStyles.Any, CultureInfo.InvariantCulture, out delaySeconds)
&& delaySeconds > 1)
{
delaySeconds = new Random().Next(delaySeconds - 1);
}
return TimeSpan.FromSeconds(delaySeconds);
}
private void RespondToSearch(EndPoint endpoint, string deviceType)
2014-04-25 17:30:41 +00:00
{
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-04-25 17:30:41 +00:00
{
_logger.Debug("RespondToSearch");
}
const string header = "HTTP/1.1 200 OK";
foreach (var d in RegisteredDevices)
{
if (string.Equals(deviceType, "ssdp:all", StringComparison.OrdinalIgnoreCase) ||
string.Equals(deviceType, d.Type, StringComparison.OrdinalIgnoreCase))
{
var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
values["CACHE-CONTROL"] = "max-age = 600";
values["DATE"] = DateTime.Now.ToString("R");
values["EXT"] = "";
values["LOCATION"] = d.Descriptor.ToString();
values["SERVER"] = _serverSignature;
values["ST"] = d.Type;
values["USN"] = d.USN;
2015-01-31 23:12:22 +00:00
SendDatagram(header, values, endpoint, null, true, 1);
SendDatagram(header, values, endpoint, new IPEndPoint(d.Address, 0), true, 1);
2014-10-08 02:25:24 +00:00
//SendDatagram(header, values, endpoint, null, true);
2014-04-25 17:30:41 +00:00
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-06-04 03:34:36 +00:00
{
_logger.Debug("{1} - Responded to a {0} request to {2}", d.Type, endpoint, d.Address.ToString());
}
2014-04-25 17:30:41 +00:00
}
2014-04-27 03:42:05 +00:00
}
2014-04-25 17:30:41 +00:00
}
private readonly object _queueTimerSyncLock = new object();
private void StartQueueTimer()
{
lock (_queueTimerSyncLock)
{
if (_queueTimer == null)
{
2015-01-31 23:12:22 +00:00
_queueTimer = new Timer(QueueTimerCallback, null, 500, Timeout.Infinite);
2014-04-25 17:30:41 +00:00
}
else
{
2015-01-31 23:12:22 +00:00
_queueTimer.Change(500, Timeout.Infinite);
2014-04-25 17:30:41 +00:00
}
}
}
private void QueueTimerCallback(object state)
{
2015-01-31 23:12:22 +00:00
Datagram msg;
while (_messageQueue.TryDequeue(out msg))
2014-04-25 17:30:41 +00:00
{
2015-01-31 23:12:22 +00:00
msg.Send();
2014-04-25 17:30:41 +00:00
}
_datagramPosted.Set();
if (_messageQueue.Count > 0)
{
StartQueueTimer();
}
else
{
DisposeQueueTimer();
}
}
2014-03-24 12:47:39 +00:00
private void Receive()
{
try
{
2014-04-25 17:30:41 +00:00
var buffer = new byte[1024];
EndPoint endpoint = new IPEndPoint(IPAddress.Any, SSDPPort);
_socket.BeginReceiveFrom(buffer, 0, buffer.Length, SocketFlags.None, ref endpoint, ReceiveCallback, buffer);
2014-03-24 12:47:39 +00:00
}
catch (ObjectDisposedException)
{
}
2014-11-07 00:16:16 +00:00
catch (Exception ex)
{
_logger.Debug("Error in BeginReceiveFrom", ex);
}
2014-03-24 12:47:39 +00:00
}
private void ReceiveCallback(IAsyncResult result)
{
2014-04-25 17:30:41 +00:00
if (_isDisposed)
{
return;
}
2014-03-24 12:47:39 +00:00
try
{
2014-04-25 17:30:41 +00:00
EndPoint endpoint = new IPEndPoint(IPAddress.Any, SSDPPort);
2014-06-22 05:52:31 +00:00
var length = _socket.EndReceiveFrom(result, ref endpoint);
2014-04-25 17:30:41 +00:00
var received = (byte[])result.AsyncState;
2014-03-24 12:47:39 +00:00
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-06-22 05:52:31 +00:00
{
_logger.Debug(Encoding.ASCII.GetString(received));
}
var args = SsdpHelper.ParseSsdpResponse(received);
args.EndPoint = endpoint;
2014-03-24 12:47:39 +00:00
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-03-24 12:47:39 +00:00
{
2014-04-27 03:42:05 +00:00
var headerTexts = args.Headers.Select(i => string.Format("{0}={1}", i.Key, i.Value));
var headerText = string.Join(",", headerTexts.ToArray());
2014-03-24 12:47:39 +00:00
_logger.Debug("{0} message received from {1} on {3}. Headers: {2}", args.Method, args.EndPoint, headerText, _socket.LocalEndPoint);
2014-03-24 12:47:39 +00:00
}
2014-04-27 03:42:05 +00:00
OnMessageReceived(args);
2014-03-24 12:47:39 +00:00
}
catch (Exception ex)
{
_logger.ErrorException("Failed to read SSDP message", ex);
}
2014-04-25 17:30:41 +00:00
if (_socket != null)
2014-03-24 12:47:39 +00:00
{
Receive();
}
}
2014-04-25 17:30:41 +00:00
public void Dispose()
2014-03-24 12:47:39 +00:00
{
2014-06-29 17:35:05 +00:00
_config.NamedConfigurationUpdated -= _config_ConfigurationUpdated;
2014-04-28 15:05:28 +00:00
2014-04-25 17:30:41 +00:00
_isDisposed = true;
while (_messageQueue.Count != 0)
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
_datagramPosted.WaitOne();
2014-03-24 12:47:39 +00:00
}
2014-04-10 15:06:54 +00:00
2014-04-25 17:30:41 +00:00
DisposeSocket();
DisposeQueueTimer();
DisposeNotificationTimer();
2014-04-10 15:06:54 +00:00
2014-04-25 17:30:41 +00:00
_datagramPosted.Dispose();
2014-03-24 12:47:39 +00:00
}
2014-04-25 17:30:41 +00:00
private void DisposeSocket()
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
if (_socket != null)
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
_socket.Close();
_socket.Dispose();
_socket = null;
2014-03-24 12:47:39 +00:00
}
2014-04-10 15:06:54 +00:00
}
2014-04-25 17:30:41 +00:00
private void DisposeQueueTimer()
2014-04-10 15:06:54 +00:00
{
2014-04-25 17:30:41 +00:00
lock (_queueTimerSyncLock)
2014-04-10 15:06:54 +00:00
{
2014-04-25 17:30:41 +00:00
if (_queueTimer != null)
2014-04-10 15:06:54 +00:00
{
2014-04-25 17:30:41 +00:00
_queueTimer.Dispose();
_queueTimer = null;
2014-04-10 15:06:54 +00:00
}
}
2014-04-25 17:30:41 +00:00
}
2014-04-10 15:06:54 +00:00
2014-04-25 17:30:41 +00:00
private Socket CreateMulticastSocket()
{
var socket = new Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp);
socket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Broadcast, true);
socket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
socket.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.MulticastTimeToLive, 4);
socket.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.AddMembership, new MulticastOption(_ssdpIp, 0));
2014-04-10 15:06:54 +00:00
2014-04-25 17:30:41 +00:00
socket.Bind(new IPEndPoint(IPAddress.Any, SSDPPort));
return socket;
2014-03-24 12:47:39 +00:00
}
private void NotifyAll()
{
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-04-21 16:02:30 +00:00
{
_logger.Debug("Sending alive notifications");
}
2014-04-25 17:30:41 +00:00
foreach (var d in RegisteredDevices)
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
NotifyDevice(d, "alive");
2014-03-24 12:47:39 +00:00
}
}
2014-04-25 17:30:41 +00:00
private void NotifyDevice(UpnpDevice dev, string type, int sendCount = 1)
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
const string header = "NOTIFY * HTTP/1.1";
2014-04-20 05:21:08 +00:00
2014-04-25 17:30:41 +00:00
var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
2014-03-24 12:47:39 +00:00
2014-04-25 17:30:41 +00:00
// If needed later for non-server devices, these headers will need to be dynamic
values["HOST"] = "239.255.255.250:1900";
values["CACHE-CONTROL"] = "max-age = 600";
values["LOCATION"] = dev.Descriptor.ToString();
values["SERVER"] = _serverSignature;
values["NTS"] = "ssdp:" + type;
values["NT"] = dev.Type;
values["USN"] = dev.USN;
2014-04-10 15:06:54 +00:00
2014-06-29 17:35:05 +00:00
if (_config.GetDlnaConfiguration().EnableDebugLogging)
2014-04-21 16:02:30 +00:00
{
_logger.Debug("{0} said {1}", dev.USN, type);
}
2014-06-22 05:52:31 +00:00
SendDatagram(header, values, new IPEndPoint(dev.Address, 0), sendCount);
2014-03-24 12:47:39 +00:00
}
2014-04-25 17:30:41 +00:00
public void RegisterNotification(Guid uuid, Uri descriptionUri, IPAddress address, IEnumerable<string> services)
2014-03-24 12:47:39 +00:00
{
List<UpnpDevice> list;
lock (_devices)
{
2014-04-10 15:06:54 +00:00
if (!_devices.TryGetValue(uuid, out list))
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
_devices.TryAdd(uuid, list = new List<UpnpDevice>());
2014-03-24 12:47:39 +00:00
}
}
2014-04-25 17:30:41 +00:00
list.AddRange(services.Select(i => new UpnpDevice(uuid, i, descriptionUri, address)));
2014-03-24 12:47:39 +00:00
NotifyAll();
2014-04-25 17:30:41 +00:00
_logger.Debug("Registered mount {0} at {1}", uuid, descriptionUri);
2014-03-24 12:47:39 +00:00
}
2014-04-25 17:30:41 +00:00
public void UnregisterNotification(Guid uuid)
2014-03-24 12:47:39 +00:00
{
List<UpnpDevice> dl;
2014-04-25 17:30:41 +00:00
if (_devices.TryRemove(uuid, out dl))
2014-03-24 12:47:39 +00:00
{
2014-04-25 17:30:41 +00:00
foreach (var d in dl.ToList())
2014-04-10 15:06:54 +00:00
{
2014-04-25 17:30:41 +00:00
NotifyDevice(d, "byebye", 2);
2014-04-10 15:06:54 +00:00
}
2014-04-25 17:30:41 +00:00
_logger.Debug("Unregistered mount {0}", uuid);
2014-04-10 15:06:54 +00:00
}
}
private readonly object _notificationTimerSyncLock = new object();
2014-04-28 15:05:28 +00:00
private int _aliveNotifierIntervalMs;
private void ReloadAliveNotifier()
2014-04-10 15:06:54 +00:00
{
2014-06-29 17:35:05 +00:00
if (!_config.GetDlnaConfiguration().BlastAliveMessages)
2014-04-20 05:21:08 +00:00
{
2014-04-28 15:05:28 +00:00
DisposeNotificationTimer();
2014-04-20 05:21:08 +00:00
return;
}
2014-06-29 17:35:05 +00:00
var intervalMs = _config.GetDlnaConfiguration().BlastAliveMessageIntervalSeconds * 1000;
2014-04-10 15:06:54 +00:00
2014-04-28 15:05:28 +00:00
if (_notificationTimer == null || _aliveNotifierIntervalMs != intervalMs)
2014-04-10 15:06:54 +00:00
{
2014-04-28 15:05:28 +00:00
lock (_notificationTimerSyncLock)
2014-04-10 15:06:54 +00:00
{
2014-04-28 15:05:28 +00:00
if (_notificationTimer == null)
{
_logger.Debug("Starting alive notifier");
const int initialDelayMs = 3000;
_notificationTimer = new Timer(state => NotifyAll(), null, initialDelayMs, intervalMs);
}
else
{
_logger.Debug("Updating alive notifier");
_notificationTimer.Change(intervalMs, intervalMs);
}
_aliveNotifierIntervalMs = intervalMs;
2014-04-10 15:06:54 +00:00
}
}
}
private void DisposeNotificationTimer()
{
lock (_notificationTimerSyncLock)
{
if (_notificationTimer != null)
{
2014-04-28 15:05:28 +00:00
_logger.Debug("Stopping alive notifier");
2014-04-10 15:06:54 +00:00
_notificationTimer.Dispose();
_notificationTimer = null;
}
}
2014-03-24 12:47:39 +00:00
}
}
}