mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
Add Argument*Exceptions now use proper nameof operators. Added exception messages to quite a few Argument*Exceptions. Fixed rethorwing to be proper syntax. Added a ton of null checkes. (This is only a start, there are about 500 places that need proper null handling) Added some TODOs to log certain exceptions. Fix sln again. Fixed all AssemblyInfo's and added proper copyright (where I could find them) We live in *current year*. Fixed the use of braces. Fixed a ton of properties, and made a fair amount of functions static that should be and can be static. Made more Methods that should be static static. You can now use static to find bad functions! Removed unused variable. And added one more proper XML comment.
226 lines
7.2 KiB
C#
226 lines
7.2 KiB
C#
using Microsoft.Extensions.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Model.Services;
|
|
|
|
namespace Emby.Server.Implementations.HttpServer
|
|
{
|
|
public class RangeRequestWriter : IAsyncStreamWriter, IHttpResult
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets the source stream.
|
|
/// </summary>
|
|
/// <value>The source stream.</value>
|
|
private Stream SourceStream { get; set; }
|
|
private string RangeHeader { get; set; }
|
|
private bool IsHeadRequest { get; set; }
|
|
|
|
private long RangeStart { get; set; }
|
|
private long RangeEnd { get; set; }
|
|
private long RangeLength { get; set; }
|
|
private long TotalContentLength { get; set; }
|
|
|
|
public Action OnComplete { get; set; }
|
|
private readonly ILogger _logger;
|
|
|
|
private const int BufferSize = 81920;
|
|
|
|
/// <summary>
|
|
/// The _options
|
|
/// </summary>
|
|
private readonly Dictionary<string, string> _options = new Dictionary<string, string>();
|
|
|
|
/// <summary>
|
|
/// The us culture
|
|
/// </summary>
|
|
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
|
|
|
|
public List<Cookie> Cookies { get; private set; }
|
|
|
|
/// <summary>
|
|
/// Additional HTTP Headers
|
|
/// </summary>
|
|
/// <value>The headers.</value>
|
|
public IDictionary<string, string> Headers => _options;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="StreamWriter" /> class.
|
|
/// </summary>
|
|
/// <param name="rangeHeader">The range header.</param>
|
|
/// <param name="source">The source.</param>
|
|
/// <param name="contentType">Type of the content.</param>
|
|
/// <param name="isHeadRequest">if set to <c>true</c> [is head request].</param>
|
|
public RangeRequestWriter(string rangeHeader, long contentLength, Stream source, string contentType, bool isHeadRequest, ILogger logger)
|
|
{
|
|
if (string.IsNullOrEmpty(contentType))
|
|
{
|
|
throw new ArgumentNullException(nameof(contentType));
|
|
}
|
|
|
|
RangeHeader = rangeHeader;
|
|
SourceStream = source;
|
|
IsHeadRequest = isHeadRequest;
|
|
this._logger = logger;
|
|
|
|
ContentType = contentType;
|
|
Headers["Content-Type"] = contentType;
|
|
Headers["Accept-Ranges"] = "bytes";
|
|
StatusCode = HttpStatusCode.PartialContent;
|
|
|
|
Cookies = new List<Cookie>();
|
|
SetRangeValues(contentLength);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets the range values.
|
|
/// </summary>
|
|
private void SetRangeValues(long contentLength)
|
|
{
|
|
var requestedRange = RequestedRanges[0];
|
|
|
|
TotalContentLength = contentLength;
|
|
|
|
// If the requested range is "0-", we can optimize by just doing a stream copy
|
|
if (!requestedRange.Value.HasValue)
|
|
{
|
|
RangeEnd = TotalContentLength - 1;
|
|
}
|
|
else
|
|
{
|
|
RangeEnd = requestedRange.Value.Value;
|
|
}
|
|
|
|
RangeStart = requestedRange.Key;
|
|
RangeLength = 1 + RangeEnd - RangeStart;
|
|
|
|
// Content-Length is the length of what we're serving, not the original content
|
|
Headers["Content-Length"] = RangeLength.ToString(UsCulture);
|
|
Headers["Content-Range"] = string.Format("bytes {0}-{1}/{2}", RangeStart, RangeEnd, TotalContentLength);
|
|
|
|
if (RangeStart > 0 && SourceStream.CanSeek)
|
|
{
|
|
SourceStream.Position = RangeStart;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// The _requested ranges
|
|
/// </summary>
|
|
private List<KeyValuePair<long, long?>> _requestedRanges;
|
|
/// <summary>
|
|
/// Gets the requested ranges.
|
|
/// </summary>
|
|
/// <value>The requested ranges.</value>
|
|
protected List<KeyValuePair<long, long?>> RequestedRanges
|
|
{
|
|
get
|
|
{
|
|
if (_requestedRanges == null)
|
|
{
|
|
_requestedRanges = new List<KeyValuePair<long, long?>>();
|
|
|
|
// Example: bytes=0-,32-63
|
|
var ranges = RangeHeader.Split('=')[1].Split(',');
|
|
|
|
foreach (var range in ranges)
|
|
{
|
|
var vals = range.Split('-');
|
|
|
|
long start = 0;
|
|
long? end = null;
|
|
|
|
if (!string.IsNullOrEmpty(vals[0]))
|
|
{
|
|
start = long.Parse(vals[0], UsCulture);
|
|
}
|
|
if (!string.IsNullOrEmpty(vals[1]))
|
|
{
|
|
end = long.Parse(vals[1], UsCulture);
|
|
}
|
|
|
|
_requestedRanges.Add(new KeyValuePair<long, long?>(start, end));
|
|
}
|
|
}
|
|
|
|
return _requestedRanges;
|
|
}
|
|
}
|
|
|
|
public async Task WriteToAsync(Stream responseStream, CancellationToken cancellationToken)
|
|
{
|
|
try
|
|
{
|
|
// Headers only
|
|
if (IsHeadRequest)
|
|
{
|
|
return;
|
|
}
|
|
|
|
using (var source = SourceStream)
|
|
{
|
|
// If the requested range is "0-", we can optimize by just doing a stream copy
|
|
if (RangeEnd >= TotalContentLength - 1)
|
|
{
|
|
await source.CopyToAsync(responseStream, BufferSize).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await CopyToInternalAsync(source, responseStream, RangeLength).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
if (OnComplete != null)
|
|
{
|
|
OnComplete();
|
|
}
|
|
}
|
|
}
|
|
|
|
private static async Task CopyToInternalAsync(Stream source, Stream destination, long copyLength)
|
|
{
|
|
var array = new byte[BufferSize];
|
|
int bytesRead;
|
|
while ((bytesRead = await source.ReadAsync(array, 0, array.Length).ConfigureAwait(false)) != 0)
|
|
{
|
|
if (bytesRead == 0)
|
|
{
|
|
break;
|
|
}
|
|
|
|
var bytesToCopy = Math.Min(bytesRead, copyLength);
|
|
|
|
await destination.WriteAsync(array, 0, Convert.ToInt32(bytesToCopy)).ConfigureAwait(false);
|
|
|
|
copyLength -= bytesToCopy;
|
|
|
|
if (copyLength <= 0)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
public string ContentType { get; set; }
|
|
|
|
public IRequest RequestContext { get; set; }
|
|
|
|
public object Response { get; set; }
|
|
|
|
public int Status { get; set; }
|
|
|
|
public HttpStatusCode StatusCode
|
|
{
|
|
get => (HttpStatusCode)Status;
|
|
set => Status = (int)value;
|
|
}
|
|
|
|
public string StatusDescription { get; set; }
|
|
}
|
|
}
|