mirror of
				https://github.com/jellyfin/jellyfin.git
				synced 2025-10-31 02:27:18 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			663 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			663 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
| using System;
 | |
| using System.Collections.Generic;
 | |
| using System.Collections.ObjectModel;
 | |
| using System.Linq;
 | |
| using System.Net;
 | |
| using System.Threading;
 | |
| using System.Threading.Tasks;
 | |
| using MediaBrowser.Common.Net;
 | |
| 
 | |
| namespace Rssdp.Infrastructure
 | |
| {
 | |
|     /// <summary>
 | |
|     /// Provides the platform independent logic for publishing SSDP devices (notifications and search responses).
 | |
|     /// </summary>
 | |
|     public class SsdpDevicePublisher : DisposableManagedObjectBase, ISsdpDevicePublisher
 | |
|     {
 | |
|         private readonly INetworkManager _networkManager;
 | |
| 
 | |
|         private ISsdpCommunicationsServer _CommsServer;
 | |
|         private string _OSName;
 | |
|         private string _OSVersion;
 | |
|         private bool _sendOnlyMatchedHost;
 | |
| 
 | |
|         private bool _SupportPnpRootDevice;
 | |
| 
 | |
|         private IList<SsdpRootDevice> _Devices;
 | |
|         private IReadOnlyList<SsdpRootDevice> _ReadOnlyDevices;
 | |
| 
 | |
|         private Timer _RebroadcastAliveNotificationsTimer;
 | |
| 
 | |
|         private IDictionary<string, SearchRequest> _RecentSearchRequests;
 | |
| 
 | |
|         private Random _Random;
 | |
| 
 | |
|         private const string ServerVersion = "1.0";
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Default constructor.
 | |
|         /// </summary>
 | |
|         public SsdpDevicePublisher(ISsdpCommunicationsServer communicationsServer, INetworkManager networkManager,
 | |
|             string osName, string osVersion, bool sendOnlyMatchedHost)
 | |
|         {
 | |
|             if (communicationsServer == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(communicationsServer));
 | |
|             }
 | |
| 
 | |
|             if (networkManager == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(networkManager));
 | |
|             }
 | |
| 
 | |
|             if (osName == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(osName));
 | |
|             }
 | |
| 
 | |
|             if (osName.Length == 0)
 | |
|             {
 | |
|                 throw new ArgumentException("osName cannot be an empty string.", nameof(osName));
 | |
|             }
 | |
| 
 | |
|             if (osVersion == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(osVersion));
 | |
|             }
 | |
| 
 | |
|             if (osVersion.Length == 0)
 | |
|             {
 | |
|                 throw new ArgumentException("osVersion cannot be an empty string.", nameof(osName));
 | |
|             }
 | |
| 
 | |
|             _SupportPnpRootDevice = true;
 | |
|             _Devices = new List<SsdpRootDevice>();
 | |
|             _ReadOnlyDevices = new ReadOnlyCollection<SsdpRootDevice>(_Devices);
 | |
|             _RecentSearchRequests = new Dictionary<string, SearchRequest>(StringComparer.OrdinalIgnoreCase);
 | |
|             _Random = new Random();
 | |
| 
 | |
|             _networkManager = networkManager;
 | |
|             _CommsServer = communicationsServer;
 | |
|             _CommsServer.RequestReceived += CommsServer_RequestReceived;
 | |
|             _OSName = osName;
 | |
|             _OSVersion = osVersion;
 | |
|             _sendOnlyMatchedHost = sendOnlyMatchedHost;
 | |
| 
 | |
|             _CommsServer.BeginListeningForBroadcasts();
 | |
|         }
 | |
| 
 | |
|         public void StartBroadcastingAliveMessages(TimeSpan interval)
 | |
|         {
 | |
|             _RebroadcastAliveNotificationsTimer = new Timer(SendAllAliveNotifications, null, TimeSpan.FromSeconds(5), interval);
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Adds a device (and it's children) to the list of devices being published by this server, making them discoverable to SSDP clients.
 | |
|         /// </summary>
 | |
|         /// <remarks>
 | |
|         /// <para>Adding a device causes "alive" notification messages to be sent immediately, or very soon after. Ensure your device/description service is running before adding the device object here.</para>
 | |
|         /// <para>Devices added here with a non-zero cache life time will also have notifications broadcast periodically.</para>
 | |
|         /// <para>This method ignores duplicate device adds (if the same device instance is added multiple times, the second and subsequent add calls do nothing).</para>
 | |
|         /// </remarks>
 | |
|         /// <param name="device">The <see cref="SsdpDevice"/> instance to add.</param>
 | |
|         /// <exception cref="ArgumentNullException">Thrown if the <paramref name="device"/> argument is null.</exception>
 | |
|         /// <exception cref="InvalidOperationException">Thrown if the <paramref name="device"/> contains property values that are not acceptable to the UPnP 1.0 specification.</exception>
 | |
|         [System.Diagnostics.CodeAnalysis.SuppressMessage("Microsoft.Performance", "CA1804:RemoveUnusedLocals", MessageId = "t", Justification = "Capture task to local variable supresses compiler warning, but task is not really needed.")]
 | |
|         public void AddDevice(SsdpRootDevice device)
 | |
|         {
 | |
|             if (device == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(device));
 | |
|             }
 | |
| 
 | |
|             ThrowIfDisposed();
 | |
| 
 | |
|             bool wasAdded = false;
 | |
|             lock (_Devices)
 | |
|             {
 | |
|                 if (!_Devices.Contains(device))
 | |
|                 {
 | |
|                     _Devices.Add(device);
 | |
|                     wasAdded = true;
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             if (wasAdded)
 | |
|             {
 | |
|                 WriteTrace("Device Added", device);
 | |
| 
 | |
|                 SendAliveNotifications(device, true, CancellationToken.None);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Removes a device (and it's children) from the list of devices being published by this server, making them undiscoverable.
 | |
|         /// </summary>
 | |
|         /// <remarks>
 | |
|         /// <para>Removing a device causes "byebye" notification messages to be sent immediately, advising clients of the device/service becoming unavailable. We recommend removing the device from the published list before shutting down the actual device/service, if possible.</para>
 | |
|         /// <para>This method does nothing if the device was not found in the collection.</para>
 | |
|         /// </remarks>
 | |
|         /// <param name="device">The <see cref="SsdpDevice"/> instance to add.</param>
 | |
|         /// <exception cref="ArgumentNullException">Thrown if the <paramref name="device"/> argument is null.</exception>
 | |
|         public async Task RemoveDevice(SsdpRootDevice device)
 | |
|         {
 | |
|             if (device == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException(nameof(device));
 | |
|             }
 | |
| 
 | |
|             bool wasRemoved = false;
 | |
|             lock (_Devices)
 | |
|             {
 | |
|                 if (_Devices.Contains(device))
 | |
|                 {
 | |
|                     _Devices.Remove(device);
 | |
|                     wasRemoved = true;
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             if (wasRemoved)
 | |
|             {
 | |
|                 WriteTrace("Device Removed", device);
 | |
| 
 | |
|                 await SendByeByeNotifications(device, true, CancellationToken.None).ConfigureAwait(false);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Returns a read only list of devices being published by this instance.
 | |
|         /// </summary>
 | |
|         public IEnumerable<SsdpRootDevice> Devices
 | |
|         {
 | |
|             get
 | |
|             {
 | |
|                 return _ReadOnlyDevices;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// If true (default) treats root devices as both upnp:rootdevice and pnp:rootdevice types.
 | |
|         /// </summary>
 | |
|         /// <remarks>
 | |
|         /// <para>Enabling this option will cause devices to show up in Microsoft Windows Explorer's network screens (if discovery is enabled etc.). Windows Explorer appears to search only for pnp:rootdeivce and not upnp:rootdevice.</para>
 | |
|         /// <para>If false, the system will only use upnp:rootdevice for notifiation broadcasts and and search responses, which is correct according to the UPnP/SSDP spec.</para>
 | |
|         /// </remarks>
 | |
|         public bool SupportPnpRootDevice
 | |
|         {
 | |
|             get { return _SupportPnpRootDevice; }
 | |
| 
 | |
|             set
 | |
|             {
 | |
|                 _SupportPnpRootDevice = value;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Stops listening for requests, stops sending periodic broadcasts, disposes all internal resources.
 | |
|         /// </summary>
 | |
|         /// <param name="disposing"></param>
 | |
|         protected override void Dispose(bool disposing)
 | |
|         {
 | |
|             if (disposing)
 | |
|             {
 | |
|                 DisposeRebroadcastTimer();
 | |
| 
 | |
|                 var commsServer = _CommsServer;
 | |
|                 if (commsServer != null)
 | |
|                 {
 | |
|                     commsServer.RequestReceived -= this.CommsServer_RequestReceived;
 | |
|                 }
 | |
| 
 | |
|                 var tasks = Devices.ToList().Select(RemoveDevice).ToArray();
 | |
|                 Task.WaitAll(tasks);
 | |
| 
 | |
|                 _CommsServer = null;
 | |
|                 if (commsServer != null)
 | |
|                 {
 | |
|                     if (!commsServer.IsShared)
 | |
|                     {
 | |
|                         commsServer.Dispose();
 | |
|                     }
 | |
|                 }
 | |
| 
 | |
|                 _RecentSearchRequests = null;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private void ProcessSearchRequest(
 | |
|             string mx,
 | |
|             string searchTarget,
 | |
|             IPEndPoint remoteEndPoint,
 | |
|             IPAddress receivedOnlocalIpAddress,
 | |
|             CancellationToken cancellationToken)
 | |
|         {
 | |
|             if (String.IsNullOrEmpty(searchTarget))
 | |
|             {
 | |
|                 WriteTrace(String.Format("Invalid search request received From {0}, Target is null/empty.", remoteEndPoint.ToString()));
 | |
|                 return;
 | |
|             }
 | |
| 
 | |
|             // WriteTrace(String.Format("Search Request Received From {0}, Target = {1}", remoteEndPoint.ToString(), searchTarget));
 | |
| 
 | |
|             if (IsDuplicateSearchRequest(searchTarget, remoteEndPoint))
 | |
|             {
 | |
|                 // WriteTrace("Search Request is Duplicate, ignoring.");
 | |
|                 return;
 | |
|             }
 | |
| 
 | |
|             // Wait on random interval up to MX, as per SSDP spec.
 | |
|             // Also, as per UPnP 1.1/SSDP spec ignore missing/bank MX header. If over 120, assume random value between 0 and 120.
 | |
|             // Using 16 as minimum as that's often the minimum system clock frequency anyway.
 | |
|             int maxWaitInterval = 0;
 | |
|             if (String.IsNullOrEmpty(mx))
 | |
|             {
 | |
|                 // Windows Explorer is poorly behaved and doesn't supply an MX header value.
 | |
|                 // if (this.SupportPnpRootDevice)
 | |
|                 mx = "1";
 | |
|                 // else
 | |
|                 // return;
 | |
|             }
 | |
| 
 | |
|             if (!Int32.TryParse(mx, out maxWaitInterval) || maxWaitInterval <= 0)
 | |
|             {
 | |
|                 return;
 | |
|             }
 | |
| 
 | |
|             if (maxWaitInterval > 120)
 | |
|             {
 | |
|                 maxWaitInterval = _Random.Next(0, 120);
 | |
|             }
 | |
| 
 | |
|             // Do not block synchronously as that may tie up a threadpool thread for several seconds.
 | |
|             Task.Delay(_Random.Next(16, (maxWaitInterval * 1000))).ContinueWith((parentTask) =>
 | |
|             {
 | |
|                 // Copying devices to local array here to avoid threading issues/enumerator exceptions.
 | |
|                 IEnumerable<SsdpDevice> devices = null;
 | |
|                 lock (_Devices)
 | |
|                 {
 | |
|                     if (String.Compare(SsdpConstants.SsdpDiscoverAllSTHeader, searchTarget, StringComparison.OrdinalIgnoreCase) == 0)
 | |
|                     {
 | |
|                         devices = GetAllDevicesAsFlatEnumerable().ToArray();
 | |
|                     }
 | |
|                     else if (String.Compare(SsdpConstants.UpnpDeviceTypeRootDevice, searchTarget, StringComparison.OrdinalIgnoreCase) == 0 || (this.SupportPnpRootDevice && String.Compare(SsdpConstants.PnpDeviceTypeRootDevice, searchTarget, StringComparison.OrdinalIgnoreCase) == 0))
 | |
|                     {
 | |
|                         devices = _Devices.ToArray();
 | |
|                     }
 | |
|                     else if (searchTarget.Trim().StartsWith("uuid:", StringComparison.OrdinalIgnoreCase))
 | |
|                     {
 | |
|                         devices = (from device in GetAllDevicesAsFlatEnumerable() where String.Compare(device.Uuid, searchTarget.Substring(5), StringComparison.OrdinalIgnoreCase) == 0 select device).ToArray();
 | |
|                     }
 | |
|                     else if (searchTarget.StartsWith("urn:", StringComparison.OrdinalIgnoreCase))
 | |
|                     {
 | |
|                         devices = (from device in GetAllDevicesAsFlatEnumerable() where String.Compare(device.FullDeviceType, searchTarget, StringComparison.OrdinalIgnoreCase) == 0 select device).ToArray();
 | |
|                     }
 | |
|                 }
 | |
| 
 | |
|                 if (devices != null)
 | |
|                 {
 | |
|                     var deviceList = devices.ToList();
 | |
|                     // WriteTrace(String.Format("Sending {0} search responses", deviceList.Count));
 | |
| 
 | |
|                     foreach (var device in deviceList)
 | |
|                     {
 | |
|                         if (!_sendOnlyMatchedHost ||
 | |
|                             _networkManager.IsInSameSubnet(device.ToRootDevice().Address, remoteEndPoint.Address, device.ToRootDevice().SubnetMask))
 | |
|                         {
 | |
|                             SendDeviceSearchResponses(device, remoteEndPoint, receivedOnlocalIpAddress, cancellationToken);
 | |
|                         }
 | |
|                     }
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     // WriteTrace(String.Format("Sending 0 search responses."));
 | |
|                 }
 | |
|             });
 | |
|         }
 | |
| 
 | |
|         private IEnumerable<SsdpDevice> GetAllDevicesAsFlatEnumerable()
 | |
|         {
 | |
|             return _Devices.Union(_Devices.SelectManyRecursive<SsdpDevice>((d) => d.Devices));
 | |
|         }
 | |
| 
 | |
|         private void SendDeviceSearchResponses(
 | |
|             SsdpDevice device,
 | |
|             IPEndPoint endPoint,
 | |
|             IPAddress receivedOnlocalIpAddress,
 | |
|             CancellationToken cancellationToken)
 | |
|         {
 | |
|             bool isRootDevice = (device as SsdpRootDevice) != null;
 | |
|             if (isRootDevice)
 | |
|             {
 | |
|                 SendSearchResponse(SsdpConstants.UpnpDeviceTypeRootDevice, device, GetUsn(device.Udn, SsdpConstants.UpnpDeviceTypeRootDevice), endPoint, receivedOnlocalIpAddress, cancellationToken);
 | |
|                 if (this.SupportPnpRootDevice)
 | |
|                 {
 | |
|                     SendSearchResponse(SsdpConstants.PnpDeviceTypeRootDevice, device, GetUsn(device.Udn, SsdpConstants.PnpDeviceTypeRootDevice), endPoint, receivedOnlocalIpAddress, cancellationToken);
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             SendSearchResponse(device.Udn, device, device.Udn, endPoint, receivedOnlocalIpAddress, cancellationToken);
 | |
| 
 | |
|             SendSearchResponse(device.FullDeviceType, device, GetUsn(device.Udn, device.FullDeviceType), endPoint, receivedOnlocalIpAddress, cancellationToken);
 | |
|         }
 | |
| 
 | |
|         private string GetUsn(string udn, string fullDeviceType)
 | |
|         {
 | |
|             return String.Format("{0}::{1}", udn, fullDeviceType);
 | |
|         }
 | |
| 
 | |
|         private async void SendSearchResponse(
 | |
|             string searchTarget,
 | |
|             SsdpDevice device,
 | |
|             string uniqueServiceName,
 | |
|             IPEndPoint endPoint,
 | |
|             IPAddress receivedOnlocalIpAddress,
 | |
|             CancellationToken cancellationToken)
 | |
|         {
 | |
|             var rootDevice = device.ToRootDevice();
 | |
| 
 | |
|             // var additionalheaders = FormatCustomHeadersForResponse(device);
 | |
| 
 | |
|             const string header = "HTTP/1.1 200 OK";
 | |
| 
 | |
|             var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
 | |
| 
 | |
|             values["EXT"] = "";
 | |
|             values["DATE"] = DateTime.UtcNow.ToString("r");
 | |
|             values["CACHE-CONTROL"] = "max-age = " + rootDevice.CacheLifetime.TotalSeconds;
 | |
|             values["ST"] = searchTarget;
 | |
|             values["SERVER"] = string.Format("{0}/{1} UPnP/1.0 RSSDP/{2}", _OSName, _OSVersion, ServerVersion);
 | |
|             values["USN"] = uniqueServiceName;
 | |
|             values["LOCATION"] = rootDevice.Location.ToString();
 | |
| 
 | |
|             var message = BuildMessage(header, values);
 | |
| 
 | |
|             try
 | |
|             {
 | |
|                 await _CommsServer.SendMessage(
 | |
|                         System.Text.Encoding.UTF8.GetBytes(message),
 | |
|                         endPoint,
 | |
|                         receivedOnlocalIpAddress,
 | |
|                         cancellationToken)
 | |
|                     .ConfigureAwait(false);
 | |
|             }
 | |
|             catch (Exception)
 | |
|             {
 | |
|             }
 | |
| 
 | |
|             // WriteTrace(String.Format("Sent search response to " + endPoint.ToString()), device);
 | |
|         }
 | |
| 
 | |
|         private bool IsDuplicateSearchRequest(string searchTarget, IPEndPoint endPoint)
 | |
|         {
 | |
|             var isDuplicateRequest = false;
 | |
| 
 | |
|             var newRequest = new SearchRequest() { EndPoint = endPoint, SearchTarget = searchTarget, Received = DateTime.UtcNow };
 | |
|             lock (_RecentSearchRequests)
 | |
|             {
 | |
|                 if (_RecentSearchRequests.ContainsKey(newRequest.Key))
 | |
|                 {
 | |
|                     var lastRequest = _RecentSearchRequests[newRequest.Key];
 | |
|                     if (lastRequest.IsOld())
 | |
|                     {
 | |
|                         _RecentSearchRequests[newRequest.Key] = newRequest;
 | |
|                     }
 | |
|                     else
 | |
|                     {
 | |
|                         isDuplicateRequest = true;
 | |
|                     }
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     _RecentSearchRequests.Add(newRequest.Key, newRequest);
 | |
|                     if (_RecentSearchRequests.Count > 10)
 | |
|                     {
 | |
|                         CleanUpRecentSearchRequestsAsync();
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             return isDuplicateRequest;
 | |
|         }
 | |
| 
 | |
|         private void CleanUpRecentSearchRequestsAsync()
 | |
|         {
 | |
|             lock (_RecentSearchRequests)
 | |
|             {
 | |
|                 foreach (var requestKey in (from r in _RecentSearchRequests where r.Value.IsOld() select r.Key).ToArray())
 | |
|                 {
 | |
|                     _RecentSearchRequests.Remove(requestKey);
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private void SendAllAliveNotifications(object state)
 | |
|         {
 | |
|             try
 | |
|             {
 | |
|                 if (IsDisposed)
 | |
|                 {
 | |
|                     return;
 | |
|                 }
 | |
| 
 | |
|                 // WriteTrace("Begin Sending Alive Notifications For All Devices");
 | |
| 
 | |
|                 SsdpRootDevice[] devices;
 | |
|                 lock (_Devices)
 | |
|                 {
 | |
|                     devices = _Devices.ToArray();
 | |
|                 }
 | |
| 
 | |
|                 foreach (var device in devices)
 | |
|                 {
 | |
|                     if (IsDisposed)
 | |
|                     {
 | |
|                         return;
 | |
|                     }
 | |
| 
 | |
|                     SendAliveNotifications(device, true, CancellationToken.None);
 | |
|                 }
 | |
| 
 | |
|                 // WriteTrace("Completed Sending Alive Notifications For All Devices");
 | |
|             }
 | |
|             catch (ObjectDisposedException ex)
 | |
|             {
 | |
|                 WriteTrace("Publisher stopped, exception " + ex.Message);
 | |
|                 Dispose();
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private void SendAliveNotifications(SsdpDevice device, bool isRoot, CancellationToken cancellationToken)
 | |
|         {
 | |
|             if (isRoot)
 | |
|             {
 | |
|                 SendAliveNotification(device, SsdpConstants.UpnpDeviceTypeRootDevice, GetUsn(device.Udn, SsdpConstants.UpnpDeviceTypeRootDevice), cancellationToken);
 | |
|                 if (this.SupportPnpRootDevice)
 | |
|                 {
 | |
|                     SendAliveNotification(device, SsdpConstants.PnpDeviceTypeRootDevice, GetUsn(device.Udn, SsdpConstants.PnpDeviceTypeRootDevice), cancellationToken);
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             SendAliveNotification(device, device.Udn, device.Udn, cancellationToken);
 | |
|             SendAliveNotification(device, device.FullDeviceType, GetUsn(device.Udn, device.FullDeviceType), cancellationToken);
 | |
| 
 | |
|             foreach (var childDevice in device.Devices)
 | |
|             {
 | |
|                 SendAliveNotifications(childDevice, false, cancellationToken);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private void SendAliveNotification(SsdpDevice device, string notificationType, string uniqueServiceName, CancellationToken cancellationToken)
 | |
|         {
 | |
|             var rootDevice = device.ToRootDevice();
 | |
| 
 | |
|             const string header = "NOTIFY * HTTP/1.1";
 | |
| 
 | |
|             var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
 | |
| 
 | |
|             // If needed later for non-server devices, these headers will need to be dynamic
 | |
|             values["HOST"] = "239.255.255.250:1900";
 | |
|             values["DATE"] = DateTime.UtcNow.ToString("r");
 | |
|             values["CACHE-CONTROL"] = "max-age = " + rootDevice.CacheLifetime.TotalSeconds;
 | |
|             values["LOCATION"] = rootDevice.Location.ToString();
 | |
|             values["SERVER"] = string.Format("{0}/{1} UPnP/1.0 RSSDP/{2}", _OSName, _OSVersion, ServerVersion);
 | |
|             values["NTS"] = "ssdp:alive";
 | |
|             values["NT"] = notificationType;
 | |
|             values["USN"] = uniqueServiceName;
 | |
| 
 | |
|             var message = BuildMessage(header, values);
 | |
| 
 | |
|             _CommsServer.SendMulticastMessage(message, _sendOnlyMatchedHost ? rootDevice.Address : null, cancellationToken);
 | |
| 
 | |
|             // WriteTrace(String.Format("Sent alive notification"), device);
 | |
|         }
 | |
| 
 | |
|         private Task SendByeByeNotifications(SsdpDevice device, bool isRoot, CancellationToken cancellationToken)
 | |
|         {
 | |
|             var tasks = new List<Task>();
 | |
|             if (isRoot)
 | |
|             {
 | |
|                 tasks.Add(SendByeByeNotification(device, SsdpConstants.UpnpDeviceTypeRootDevice, GetUsn(device.Udn, SsdpConstants.UpnpDeviceTypeRootDevice), cancellationToken));
 | |
|                 if (this.SupportPnpRootDevice)
 | |
|                 {
 | |
|                     tasks.Add(SendByeByeNotification(device, "pnp:rootdevice", GetUsn(device.Udn, "pnp:rootdevice"), cancellationToken));
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             tasks.Add(SendByeByeNotification(device, device.Udn, device.Udn, cancellationToken));
 | |
|             tasks.Add(SendByeByeNotification(device, String.Format("urn:{0}", device.FullDeviceType), GetUsn(device.Udn, device.FullDeviceType), cancellationToken));
 | |
| 
 | |
|             foreach (var childDevice in device.Devices)
 | |
|             {
 | |
|                 tasks.Add(SendByeByeNotifications(childDevice, false, cancellationToken));
 | |
|             }
 | |
| 
 | |
|             return Task.WhenAll(tasks);
 | |
|         }
 | |
| 
 | |
|         [System.Diagnostics.CodeAnalysis.SuppressMessage("Microsoft.Naming", "CA2204:Literals should be spelled correctly", MessageId = "byebye", Justification = "Correct value for this type of notification in SSDP.")]
 | |
|         private Task SendByeByeNotification(SsdpDevice device, string notificationType, string uniqueServiceName, CancellationToken cancellationToken)
 | |
|         {
 | |
|             const string header = "NOTIFY * HTTP/1.1";
 | |
| 
 | |
|             var values = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
 | |
| 
 | |
|             // If needed later for non-server devices, these headers will need to be dynamic
 | |
|             values["HOST"] = "239.255.255.250:1900";
 | |
|             values["DATE"] = DateTime.UtcNow.ToString("r");
 | |
|             values["SERVER"] = string.Format("{0}/{1} UPnP/1.0 RSSDP/{2}", _OSName, _OSVersion, ServerVersion);
 | |
|             values["NTS"] = "ssdp:byebye";
 | |
|             values["NT"] = notificationType;
 | |
|             values["USN"] = uniqueServiceName;
 | |
| 
 | |
|             var message = BuildMessage(header, values);
 | |
| 
 | |
|             var sendCount = IsDisposed ? 1 : 3;
 | |
|             WriteTrace(String.Format("Sent byebye notification"), device);
 | |
|             return _CommsServer.SendMulticastMessage(message, sendCount, _sendOnlyMatchedHost ? device.ToRootDevice().Address : null, cancellationToken);
 | |
|         }
 | |
| 
 | |
|         private void DisposeRebroadcastTimer()
 | |
|         {
 | |
|             var timer = _RebroadcastAliveNotificationsTimer;
 | |
|             _RebroadcastAliveNotificationsTimer = null;
 | |
|             if (timer != null)
 | |
|             {
 | |
|                 timer.Dispose();
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private TimeSpan GetMinimumNonZeroCacheLifetime()
 | |
|         {
 | |
|             var nonzeroCacheLifetimesQuery = (
 | |
|                 from device
 | |
|                 in _Devices
 | |
|                 where device.CacheLifetime != TimeSpan.Zero
 | |
|                 select device.CacheLifetime).ToList();
 | |
| 
 | |
|             if (nonzeroCacheLifetimesQuery.Any())
 | |
|             {
 | |
|                 return nonzeroCacheLifetimesQuery.Min();
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 return TimeSpan.Zero;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private string GetFirstHeaderValue(System.Net.Http.Headers.HttpRequestHeaders httpRequestHeaders, string headerName)
 | |
|         {
 | |
|             string retVal = null;
 | |
|             IEnumerable<String> values = null;
 | |
|             if (httpRequestHeaders.TryGetValues(headerName, out values) && values != null)
 | |
|             {
 | |
|                 retVal = values.FirstOrDefault();
 | |
|             }
 | |
| 
 | |
|             return retVal;
 | |
|         }
 | |
| 
 | |
|         public Action<string> LogFunction { get; set; }
 | |
| 
 | |
|         private void WriteTrace(string text)
 | |
|         {
 | |
|             if (LogFunction != null)
 | |
|             {
 | |
|                 LogFunction(text);
 | |
|             }
 | |
|             // System.Diagnostics.Debug.WriteLine(text, "SSDP Publisher");
 | |
|         }
 | |
| 
 | |
|         private void WriteTrace(string text, SsdpDevice device)
 | |
|         {
 | |
|             var rootDevice = device as SsdpRootDevice;
 | |
|             if (rootDevice != null)
 | |
|             {
 | |
|                 WriteTrace(text + " " + device.DeviceType + " - " + device.Uuid + " - " + rootDevice.Location);
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 WriteTrace(text + " " + device.DeviceType + " - " + device.Uuid);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private void CommsServer_RequestReceived(object sender, RequestReceivedEventArgs e)
 | |
|         {
 | |
|             if (this.IsDisposed)
 | |
|             {
 | |
|                 return;
 | |
|             }
 | |
| 
 | |
|             if (string.Equals(e.Message.Method.Method, SsdpConstants.MSearchMethod, StringComparison.OrdinalIgnoreCase))
 | |
|             {
 | |
|                 // According to SSDP/UPnP spec, ignore message if missing these headers.
 | |
|                 // Edit: But some devices do it anyway
 | |
|                 // if (!e.Message.Headers.Contains("MX"))
 | |
|                 //    WriteTrace("Ignoring search request - missing MX header.");
 | |
|                 // else if (!e.Message.Headers.Contains("MAN"))
 | |
|                 //    WriteTrace("Ignoring search request - missing MAN header.");
 | |
|                 // else
 | |
|                 ProcessSearchRequest(GetFirstHeaderValue(e.Message.Headers, "MX"), GetFirstHeaderValue(e.Message.Headers, "ST"), e.ReceivedFrom, e.LocalIpAddress, CancellationToken.None);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private class SearchRequest
 | |
|         {
 | |
|             public IPEndPoint EndPoint { get; set; }
 | |
| 
 | |
|             public DateTime Received { get; set; }
 | |
| 
 | |
|             public string SearchTarget { get; set; }
 | |
| 
 | |
|             public string Key
 | |
|             {
 | |
|                 get { return this.SearchTarget + ":" + this.EndPoint.ToString(); }
 | |
|             }
 | |
| 
 | |
|             public bool IsOld()
 | |
|             {
 | |
|                 return DateTime.UtcNow.Subtract(this.Received).TotalMilliseconds > 500;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| }
 |