jellyfin/Emby.Server.Implementations/SocketSharp/WebSocketSharpRequest.cs

247 lines
8.0 KiB
C#
Raw Normal View History

2019-02-25 22:34:32 +00:00
using System;
using System.Collections.Generic;
using System.IO;
using System.Net;
2019-02-24 02:16:19 +00:00
using System.Linq;
2019-04-12 13:25:18 +00:00
using MediaBrowser.Common.Net;
2019-02-25 22:34:32 +00:00
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.Extensions;
2019-02-25 22:34:32 +00:00
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Primitives;
using Microsoft.Net.Http.Headers;
using IHttpRequest = MediaBrowser.Model.Services.IHttpRequest;
namespace Emby.Server.Implementations.SocketSharp
{
public partial class WebSocketSharpRequest : IHttpRequest
{
public const string FormUrlEncoded = "application/x-www-form-urlencoded";
public const string MultiPartFormData = "multipart/form-data";
public const string Soap11 = "text/xml; charset=utf-8";
private string _remoteIp;
private Dictionary<string, object> _items;
private string _responseContentType;
2019-02-25 22:34:32 +00:00
public WebSocketSharpRequest(HttpRequest httpRequest, HttpResponse httpResponse, string operationName, ILogger logger)
2019-02-25 22:34:32 +00:00
{
this.OperationName = operationName;
this.Request = httpRequest;
this.Response = httpResponse;
2019-02-25 22:34:32 +00:00
}
public string Accept => StringValues.IsNullOrEmpty(Request.Headers[HeaderNames.Accept]) ? null : Request.Headers[HeaderNames.Accept].ToString();
2019-02-25 22:34:32 +00:00
public string Authorization => StringValues.IsNullOrEmpty(Request.Headers[HeaderNames.Authorization]) ? null : Request.Headers[HeaderNames.Authorization].ToString();
2019-02-25 22:34:32 +00:00
public HttpRequest Request { get; }
2019-02-25 22:34:32 +00:00
public HttpResponse Response { get; }
2019-02-25 22:34:32 +00:00
public string OperationName { get; set; }
2019-02-25 22:34:32 +00:00
public string RawUrl => Request.GetEncodedPathAndQuery();
2019-02-25 22:34:32 +00:00
public string AbsoluteUri => Request.GetDisplayUrl().TrimEnd('/');
2019-02-25 22:34:32 +00:00
2019-02-16 15:54:18 +00:00
public string RemoteIp
{
get
{
if (_remoteIp != null)
2019-02-16 15:54:18 +00:00
{
return _remoteIp;
2019-02-16 15:54:18 +00:00
}
2019-02-25 22:34:32 +00:00
IPAddress ip;
2019-02-16 15:54:18 +00:00
// "Real" remote ip might be in X-Forwarded-For of X-Real-Ip
// (if the server is behind a reverse proxy for example)
2019-04-12 13:25:18 +00:00
if (!IPAddress.TryParse(GetHeader(CustomHeaderNames.XForwardedFor), out ip))
2019-02-16 15:54:18 +00:00
{
2019-04-12 13:25:18 +00:00
if (!IPAddress.TryParse(GetHeader(CustomHeaderNames.XRealIP), out ip))
2019-02-25 22:34:32 +00:00
{
ip = Request.HttpContext.Connection.RemoteIpAddress;
2019-02-25 22:34:32 +00:00
}
}
return _remoteIp = NormalizeIp(ip).ToString();
2019-02-25 22:34:32 +00:00
}
}
public string[] AcceptTypes => Request.Headers.GetCommaSeparatedValues(HeaderNames.Accept);
public Dictionary<string, object> Items => _items ?? (_items = new Dictionary<string, object>());
public string ResponseContentType
{
get =>
_responseContentType
?? (_responseContentType = GetResponseContentType(Request));
set => this._responseContentType = value;
}
public string PathInfo => Request.Path.Value;
public string UserAgent => Request.Headers[HeaderNames.UserAgent];
public IHeaderDictionary Headers => Request.Headers;
public IQueryCollection QueryString => Request.Query;
public bool IsLocal => Request.HttpContext.Connection.LocalIpAddress.Equals(Request.HttpContext.Connection.RemoteIpAddress);
public string HttpMethod => Request.Method;
public string Verb => HttpMethod;
public string ContentType => Request.ContentType;
public Uri UrlReferrer => Request.GetTypedHeaders().Referer;
public Stream InputStream => Request.Body;
public long ContentLength => Request.ContentLength ?? 0;
private string GetHeader(string name) => Request.Headers[name].ToString();
private static IPAddress NormalizeIp(IPAddress ip)
2019-02-25 22:34:32 +00:00
{
if (ip.IsIPv4MappedToIPv6)
2019-02-25 22:34:32 +00:00
{
return ip.MapToIPv4();
2019-02-25 22:34:32 +00:00
}
return ip;
}
public static string GetResponseContentType(HttpRequest httpReq)
{
var specifiedContentType = GetQueryStringContentType(httpReq);
if (!string.IsNullOrEmpty(specifiedContentType))
{
return specifiedContentType;
}
const string serverDefaultContentType = "application/json";
var acceptContentTypes = httpReq.Headers.GetCommaSeparatedValues(HeaderNames.Accept);
2019-02-25 22:34:32 +00:00
string defaultContentType = null;
if (HasAnyOfContentTypes(httpReq, FormUrlEncoded, MultiPartFormData))
{
defaultContentType = serverDefaultContentType;
}
var acceptsAnything = false;
var hasDefaultContentType = defaultContentType != null;
if (acceptContentTypes != null)
{
foreach (var acceptsType in acceptContentTypes)
{
// TODO: @bond move to Span when Span.Split lands
// https://github.com/dotnet/corefx/issues/26528
var contentType = acceptsType?.Split(';')[0].Trim();
acceptsAnything = contentType.Equals("*/*", StringComparison.OrdinalIgnoreCase);
if (acceptsAnything)
{
break;
}
}
if (acceptsAnything)
{
if (hasDefaultContentType)
{
return defaultContentType;
}
else
{
return serverDefaultContentType;
}
}
}
if (acceptContentTypes == null && httpReq.ContentType == Soap11)
{
return Soap11;
}
// We could also send a '406 Not Acceptable', but this is allowed also
return serverDefaultContentType;
}
public static bool HasAnyOfContentTypes(HttpRequest request, params string[] contentTypes)
{
if (contentTypes == null || request.ContentType == null)
{
return false;
}
foreach (var contentType in contentTypes)
{
if (IsContentType(request, contentType))
{
return true;
}
}
return false;
}
public static bool IsContentType(HttpRequest request, string contentType)
{
return request.ContentType.StartsWith(contentType, StringComparison.OrdinalIgnoreCase);
}
private static string GetQueryStringContentType(HttpRequest httpReq)
{
2019-02-26 21:40:25 +00:00
ReadOnlySpan<char> format = httpReq.Query["format"].ToString().AsSpan();
2019-02-25 22:34:32 +00:00
if (format == null)
{
const int formatMaxLength = 4;
2019-02-26 21:40:25 +00:00
ReadOnlySpan<char> pi = httpReq.Path.ToString().AsSpan();
2019-02-25 22:34:32 +00:00
if (pi == null || pi.Length <= formatMaxLength)
{
return null;
}
if (pi[0] == '/')
{
2019-02-26 21:40:25 +00:00
pi = pi.Slice(1);
2019-02-25 22:34:32 +00:00
}
format = LeftPart(pi, '/');
if (format.Length > formatMaxLength)
{
return null;
}
}
format = LeftPart(format, '.');
2019-02-26 21:40:25 +00:00
if (format.Contains("json".AsSpan(), StringComparison.OrdinalIgnoreCase))
2019-02-25 22:34:32 +00:00
{
return "application/json";
}
2019-02-26 21:40:25 +00:00
else if (format.Contains("xml".AsSpan(), StringComparison.OrdinalIgnoreCase))
2019-02-25 22:34:32 +00:00
{
return "application/xml";
}
return null;
}
2019-02-26 21:40:25 +00:00
public static ReadOnlySpan<char> LeftPart(ReadOnlySpan<char> strVal, char needle)
2019-02-25 22:34:32 +00:00
{
if (strVal == null)
{
return null;
}
2019-02-26 21:40:25 +00:00
var pos = strVal.IndexOf(needle);
2019-02-25 22:34:32 +00:00
return pos == -1 ? strVal : strVal.Slice(0, pos);
}
}
}