2019-01-13 20:22:24 +01:00
|
|
|
using System;
|
|
|
|
using System.Threading.Tasks;
|
2019-01-06 21:50:43 +01:00
|
|
|
using MediaBrowser.Controller.Net;
|
2013-05-09 19:38:02 +02:00
|
|
|
using MediaBrowser.Controller.Session;
|
2015-03-08 20:48:30 +01:00
|
|
|
using MediaBrowser.Model.Events;
|
2019-02-27 14:23:39 +01:00
|
|
|
using Microsoft.AspNetCore.Http;
|
2019-01-13 20:22:24 +01:00
|
|
|
using Microsoft.Extensions.Logging;
|
2013-05-09 19:38:02 +02:00
|
|
|
|
2016-11-04 00:35:19 +01:00
|
|
|
namespace Emby.Server.Implementations.Session
|
2013-05-09 19:38:02 +02:00
|
|
|
{
|
|
|
|
/// <summary>
|
|
|
|
/// Class SessionWebSocketListener
|
|
|
|
/// </summary>
|
2019-12-17 23:15:02 +01:00
|
|
|
public sealed class SessionWebSocketListener : IWebSocketListener, IDisposable
|
2013-05-09 19:38:02 +02:00
|
|
|
{
|
|
|
|
/// <summary>
|
|
|
|
/// The _session manager
|
|
|
|
/// </summary>
|
|
|
|
private readonly ISessionManager _sessionManager;
|
|
|
|
|
|
|
|
/// <summary>
|
2013-05-10 14:18:07 +02:00
|
|
|
/// The _logger
|
|
|
|
/// </summary>
|
|
|
|
private readonly ILogger _logger;
|
2019-12-17 23:15:02 +01:00
|
|
|
private readonly ILoggerFactory _loggerFactory;
|
2013-05-10 14:18:07 +02:00
|
|
|
|
2015-03-08 20:48:30 +01:00
|
|
|
private readonly IHttpServer _httpServer;
|
|
|
|
|
2013-05-10 14:18:07 +02:00
|
|
|
/// <summary>
|
|
|
|
/// Initializes a new instance of the <see cref="SessionWebSocketListener" /> class.
|
2013-05-09 19:38:02 +02:00
|
|
|
/// </summary>
|
2019-12-17 23:15:02 +01:00
|
|
|
/// <param name="logger">The logger.</param>
|
2013-05-09 19:38:02 +02:00
|
|
|
/// <param name="sessionManager">The session manager.</param>
|
2018-12-13 14:18:25 +01:00
|
|
|
/// <param name="loggerFactory">The logger factory.</param>
|
2015-03-08 20:48:30 +01:00
|
|
|
/// <param name="httpServer">The HTTP server.</param>
|
2019-12-17 23:15:02 +01:00
|
|
|
public SessionWebSocketListener(
|
|
|
|
ILogger<SessionWebSocketListener> logger,
|
|
|
|
ISessionManager sessionManager,
|
|
|
|
ILoggerFactory loggerFactory,
|
|
|
|
IHttpServer httpServer)
|
2013-05-09 19:38:02 +02:00
|
|
|
{
|
2019-12-17 23:15:02 +01:00
|
|
|
_logger = logger;
|
2013-05-09 19:38:02 +02:00
|
|
|
_sessionManager = sessionManager;
|
2019-12-17 23:15:02 +01:00
|
|
|
_loggerFactory = loggerFactory;
|
2015-03-08 20:48:30 +01:00
|
|
|
_httpServer = httpServer;
|
2019-12-17 23:15:02 +01:00
|
|
|
|
|
|
|
httpServer.WebSocketConnected += OnServerManagerWebSocketConnected;
|
2015-03-08 20:48:30 +01:00
|
|
|
}
|
|
|
|
|
2019-12-17 23:15:02 +01:00
|
|
|
private void OnServerManagerWebSocketConnected(object sender, GenericEventArgs<IWebSocketConnection> e)
|
2015-03-08 20:48:30 +01:00
|
|
|
{
|
2019-12-17 23:15:02 +01:00
|
|
|
var session = GetSession(e.Argument.QueryString, e.Argument.RemoteEndPoint.ToString());
|
2015-03-08 20:48:30 +01:00
|
|
|
if (session != null)
|
|
|
|
{
|
2018-09-12 19:26:21 +02:00
|
|
|
EnsureController(session, e.Argument);
|
2015-03-08 20:48:30 +01:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-12-26 20:57:46 +01:00
|
|
|
_logger.LogWarning("Unable to determine session based on query string: {0}", e.Argument.QueryString);
|
2015-03-08 20:48:30 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-27 14:23:39 +01:00
|
|
|
private SessionInfo GetSession(IQueryCollection queryString, string remoteEndpoint)
|
2015-03-08 20:48:30 +01:00
|
|
|
{
|
2015-03-16 17:47:14 +01:00
|
|
|
if (queryString == null)
|
|
|
|
{
|
2019-11-24 15:27:58 +01:00
|
|
|
return null;
|
2015-03-16 17:47:14 +01:00
|
|
|
}
|
|
|
|
|
2015-03-08 20:48:30 +01:00
|
|
|
var token = queryString["api_key"];
|
2015-03-10 02:47:52 +01:00
|
|
|
if (string.IsNullOrWhiteSpace(token))
|
|
|
|
{
|
2018-09-12 19:26:21 +02:00
|
|
|
return null;
|
2015-03-10 02:47:52 +01:00
|
|
|
}
|
2019-11-24 15:27:58 +01:00
|
|
|
|
2015-03-13 02:55:22 +01:00
|
|
|
var deviceId = queryString["deviceId"];
|
|
|
|
return _sessionManager.GetSessionByAuthenticationToken(token, deviceId, remoteEndpoint);
|
2015-03-08 20:48:30 +01:00
|
|
|
}
|
|
|
|
|
2019-12-17 23:15:02 +01:00
|
|
|
/// <inheritdoc />
|
2015-03-08 20:48:30 +01:00
|
|
|
public void Dispose()
|
|
|
|
{
|
2019-12-17 23:15:02 +01:00
|
|
|
_httpServer.WebSocketConnected -= OnServerManagerWebSocketConnected;
|
2013-05-09 19:38:02 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
/// Processes the message.
|
|
|
|
/// </summary>
|
|
|
|
/// <param name="message">The message.</param>
|
|
|
|
/// <returns>Task.</returns>
|
2019-02-24 03:16:19 +01:00
|
|
|
public Task ProcessMessageAsync(WebSocketMessageInfo message)
|
|
|
|
=> Task.CompletedTask;
|
2013-10-03 03:22:50 +02:00
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
private void EnsureController(SessionInfo session, IWebSocketConnection connection)
|
2014-04-16 04:17:48 +02:00
|
|
|
{
|
2019-12-17 23:15:02 +01:00
|
|
|
var controllerInfo = session.EnsureController<WebSocketController>(
|
|
|
|
s => new WebSocketController(_loggerFactory.CreateLogger<WebSocketController>(), s, _sessionManager));
|
2014-04-16 04:17:48 +02:00
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
var controller = (WebSocketController)controllerInfo.Item1;
|
|
|
|
controller.AddWebSocket(connection);
|
2013-10-03 03:22:50 +02:00
|
|
|
}
|
2013-05-09 19:38:02 +02:00
|
|
|
}
|
|
|
|
}
|