jellyfin-server/MediaBrowser.Server.Implementations/HttpServer/Security/AuthorizationContext.cs

197 lines
6.2 KiB
C#
Raw Normal View History

2015-04-04 00:41:16 +00:00
using MediaBrowser.Controller.Connect;
using MediaBrowser.Controller.Net;
2014-11-15 02:31:03 +00:00
using MediaBrowser.Controller.Security;
using ServiceStack.Web;
2014-11-15 02:31:03 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
namespace MediaBrowser.Server.Implementations.HttpServer.Security
{
public class AuthorizationContext : IAuthorizationContext
{
2014-11-15 02:31:03 +00:00
private readonly IAuthenticationRepository _authRepo;
2015-04-04 00:41:16 +00:00
private readonly IConnectManager _connectManager;
2014-11-15 02:31:03 +00:00
2015-04-04 00:41:16 +00:00
public AuthorizationContext(IAuthenticationRepository authRepo, IConnectManager connectManager)
2014-11-15 02:31:03 +00:00
{
_authRepo = authRepo;
2015-04-04 00:41:16 +00:00
_connectManager = connectManager;
2014-11-15 02:31:03 +00:00
}
public AuthorizationInfo GetAuthorizationInfo(object requestContext)
{
2014-11-15 02:31:03 +00:00
var req = new ServiceStackServiceRequest((IRequest)requestContext);
return GetAuthorizationInfo(req);
}
public AuthorizationInfo GetAuthorizationInfo(IServiceRequest requestContext)
{
object cached;
if (requestContext.Items.TryGetValue("AuthorizationInfo", out cached))
{
return (AuthorizationInfo)cached;
}
return GetAuthorization(requestContext);
}
/// <summary>
/// Gets the authorization.
/// </summary>
/// <param name="httpReq">The HTTP req.</param>
/// <returns>Dictionary{System.StringSystem.String}.</returns>
2014-11-15 02:31:03 +00:00
private AuthorizationInfo GetAuthorization(IServiceRequest httpReq)
{
var auth = GetAuthorizationDictionary(httpReq);
string userId = null;
string deviceId = null;
string device = null;
string client = null;
string version = null;
if (auth != null)
{
2014-11-15 02:31:03 +00:00
// TODO: Remove this
auth.TryGetValue("UserId", out userId);
2014-11-15 02:31:03 +00:00
auth.TryGetValue("DeviceId", out deviceId);
auth.TryGetValue("Device", out device);
auth.TryGetValue("Client", out client);
auth.TryGetValue("Version", out version);
}
2015-10-04 22:04:56 +00:00
var token = httpReq.Headers["X-Emby-Token"];
2015-10-04 22:04:56 +00:00
if (string.IsNullOrWhiteSpace(token))
{
token = httpReq.Headers["X-MediaBrowser-Token"];
}
if (string.IsNullOrWhiteSpace(token))
{
token = httpReq.QueryString["api_key"];
}
2014-11-15 02:31:03 +00:00
var info = new AuthorizationInfo
{
Client = client,
Device = device,
DeviceId = deviceId,
UserId = userId,
2014-07-02 18:34:08 +00:00
Version = version,
Token = token
};
2014-11-15 02:31:03 +00:00
if (!string.IsNullOrWhiteSpace(token))
{
var result = _authRepo.Get(new AuthenticationInfoQuery
{
AccessToken = token
});
var tokenInfo = result.Items.FirstOrDefault();
if (tokenInfo != null)
{
info.UserId = tokenInfo.UserId;
2015-02-12 03:54:31 +00:00
// TODO: Remove these checks for IsNullOrWhiteSpace
if (string.IsNullOrWhiteSpace(info.Client))
{
info.Client = tokenInfo.AppName;
}
if (string.IsNullOrWhiteSpace(info.Device))
{
info.Device = tokenInfo.DeviceName;
}
if (string.IsNullOrWhiteSpace(info.DeviceId))
{
info.DeviceId = tokenInfo.DeviceId;
}
2014-11-15 02:31:03 +00:00
}
2015-04-04 00:41:16 +00:00
else
{
var user = _connectManager.GetUserFromExchangeToken(token);
if (user != null)
{
info.UserId = user.Id.ToString("N");
}
}
2014-11-15 02:31:03 +00:00
httpReq.Items["OriginalAuthenticationInfo"] = tokenInfo;
}
httpReq.Items["AuthorizationInfo"] = info;
return info;
}
/// <summary>
/// Gets the auth.
/// </summary>
/// <param name="httpReq">The HTTP req.</param>
/// <returns>Dictionary{System.StringSystem.String}.</returns>
2014-11-15 02:31:03 +00:00
private Dictionary<string, string> GetAuthorizationDictionary(IServiceRequest httpReq)
{
2015-07-03 11:51:45 +00:00
var auth = httpReq.Headers["X-Emby-Authorization"];
if (string.IsNullOrWhiteSpace(auth))
{
auth = httpReq.Headers["Authorization"];
}
return GetAuthorization(auth);
}
/// <summary>
/// Gets the authorization.
/// </summary>
/// <param name="authorizationHeader">The authorization header.</param>
/// <returns>Dictionary{System.StringSystem.String}.</returns>
2014-09-05 03:48:53 +00:00
private Dictionary<string, string> GetAuthorization(string authorizationHeader)
{
if (authorizationHeader == null) return null;
2014-09-05 03:48:53 +00:00
var parts = authorizationHeader.Split(new[] { ' ' }, 2);
// There should be at least to parts
2014-09-05 03:48:53 +00:00
if (parts.Length != 2) return null;
// It has to be a digest request
if (!string.Equals(parts[0], "MediaBrowser", StringComparison.OrdinalIgnoreCase))
{
return null;
}
// Remove uptil the first space
2014-09-05 03:48:53 +00:00
authorizationHeader = parts[1];
parts = authorizationHeader.Split(',');
var result = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
foreach (var item in parts)
{
var param = item.Trim().Split(new[] { '=' }, 2);
2014-09-05 03:48:53 +00:00
if (param.Length == 2)
{
var value = NormalizeValue (param[1].Trim(new[] { '"' }));
result.Add(param[0], value);
2014-09-05 03:48:53 +00:00
}
}
return result;
}
private string NormalizeValue(string value)
{
if (string.IsNullOrWhiteSpace (value))
{
return value;
}
return System.Net.WebUtility.HtmlEncode(value);
}
}
}