jellyfin/Emby.Server.Implementations/HttpServer/ResponseFilter.cs

104 lines
3.7 KiB
C#
Raw Normal View History

using System;
2013-12-07 15:52:38 +00:00
using System.Globalization;
using System.Text;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.Services;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;
using Microsoft.Net.Http.Headers;
2013-12-07 15:52:38 +00:00
2016-11-08 18:44:23 +00:00
namespace Emby.Server.Implementations.HttpServer
2013-12-07 15:52:38 +00:00
{
2019-11-01 17:38:54 +00:00
/// <summary>
/// Class ResponseFilter.
/// </summary>
2013-12-07 15:52:38 +00:00
public class ResponseFilter
{
private readonly ILogger _logger;
2019-11-01 17:38:54 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="ResponseFilter"/> class.
/// </summary>
/// <param name="logger">The logger.</param>
2016-05-29 21:04:49 +00:00
public ResponseFilter(ILogger logger)
2013-12-07 15:52:38 +00:00
{
_logger = logger;
}
/// <summary>
/// Filters the response.
/// </summary>
/// <param name="req">The req.</param>
/// <param name="res">The res.</param>
/// <param name="dto">The dto.</param>
public void FilterResponse(IRequest req, HttpResponse res, object dto)
2013-12-07 15:52:38 +00:00
{
// Try to prevent compatibility view
res.Headers.Add("Access-Control-Allow-Headers", "Accept, Accept-Language, Authorization, Cache-Control, Content-Disposition, Content-Encoding, Content-Language, Content-Length, Content-MD5, Content-Range, Content-Type, Date, Host, If-Match, If-Modified-Since, If-None-Match, If-Unmodified-Since, Origin, OriginToken, Pragma, Range, Slug, Transfer-Encoding, Want-Digest, X-MediaBrowser-Token, X-Emby-Authorization");
res.Headers.Add("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, PATCH, OPTIONS");
res.Headers.Add("Access-Control-Allow-Origin", "*");
2015-06-13 04:14:48 +00:00
2019-01-06 19:35:36 +00:00
if (dto is Exception exception)
2013-12-07 15:52:38 +00:00
{
2018-12-20 12:11:26 +00:00
_logger.LogError(exception, "Error processing request for {RawUrl}", req.RawUrl);
2013-12-07 15:52:38 +00:00
if (!string.IsNullOrEmpty(exception.Message))
{
2019-11-01 17:38:54 +00:00
var error = exception.Message.Replace(Environment.NewLine, " ", StringComparison.Ordinal);
2013-12-07 15:52:38 +00:00
error = RemoveControlCharacters(error);
res.Headers.Add("X-Application-Error-Code", error);
2013-12-07 15:52:38 +00:00
}
}
2019-01-06 19:35:36 +00:00
if (dto is IHasHeaders hasHeaders)
2013-12-07 15:52:38 +00:00
{
if (!hasHeaders.Headers.ContainsKey(HeaderNames.Server))
2016-01-14 21:41:23 +00:00
{
hasHeaders.Headers[HeaderNames.Server] = "Microsoft-NetCore/2.0, UPnP/1.0 DLNADOC/1.50";
2016-01-14 21:41:23 +00:00
}
2015-05-22 19:16:14 +00:00
2013-12-07 15:52:38 +00:00
// Content length has to be explicitly set on on HttpListenerResponse or it won't be happy
if (hasHeaders.Headers.TryGetValue(HeaderNames.ContentLength, out string contentLength)
2019-01-06 19:35:36 +00:00
&& !string.IsNullOrEmpty(contentLength))
2013-12-07 15:52:38 +00:00
{
2019-11-01 17:38:54 +00:00
var length = long.Parse(contentLength, CultureInfo.InvariantCulture);
2013-12-07 15:52:38 +00:00
if (length > 0)
{
res.ContentLength = length;
2013-12-07 15:52:38 +00:00
}
}
}
}
/// <summary>
/// Removes the control characters.
/// </summary>
/// <param name="inString">The in string.</param>
/// <returns>System.String.</returns>
2014-10-28 23:17:55 +00:00
public static string RemoveControlCharacters(string inString)
2013-12-07 15:52:38 +00:00
{
if (inString == null)
{
return null;
}
2020-04-19 13:18:28 +00:00
else if (inString.Length == 0)
{
return inString;
}
2013-12-07 15:52:38 +00:00
var newString = new StringBuilder(inString.Length);
2013-12-07 15:52:38 +00:00
foreach (var ch in inString)
{
if (!char.IsControl(ch))
{
newString.Append(ch);
}
}
2013-12-07 15:52:38 +00:00
return newString.ToString();
}
}
}