2017-03-12 20:27:26 +01:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Globalization;
|
|
|
|
|
using System.Net;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
using MediaBrowser.Model.IO;
|
|
|
|
|
using MediaBrowser.Model.Logging;
|
|
|
|
|
using MediaBrowser.Model.Services;
|
2018-09-12 19:26:21 +02:00
|
|
|
|
using System.Linq;
|
2017-03-12 20:27:26 +01:00
|
|
|
|
|
|
|
|
|
namespace Emby.Server.Implementations.HttpServer
|
|
|
|
|
{
|
|
|
|
|
public class FileWriter : IHttpResult
|
|
|
|
|
{
|
|
|
|
|
private ILogger Logger { 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; }
|
|
|
|
|
public Action OnError { get; set; }
|
|
|
|
|
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
|
|
|
|
|
public List<Cookie> Cookies { get; private set; }
|
|
|
|
|
|
2017-03-13 05:08:23 +01:00
|
|
|
|
public FileShareMode FileShare { get; set; }
|
|
|
|
|
|
2017-03-12 20:27:26 +01:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _options
|
|
|
|
|
/// </summary>
|
|
|
|
|
private readonly IDictionary<string, string> _options = new Dictionary<string, string>();
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets the options.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <value>The options.</value>
|
|
|
|
|
public IDictionary<string, string> Headers
|
|
|
|
|
{
|
|
|
|
|
get { return _options; }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string Path { get; set; }
|
|
|
|
|
|
|
|
|
|
public FileWriter(string path, string contentType, string rangeHeader, ILogger logger, IFileSystem fileSystem)
|
|
|
|
|
{
|
|
|
|
|
if (string.IsNullOrEmpty(contentType))
|
|
|
|
|
{
|
|
|
|
|
throw new ArgumentNullException("contentType");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Path = path;
|
|
|
|
|
Logger = logger;
|
|
|
|
|
RangeHeader = rangeHeader;
|
|
|
|
|
|
|
|
|
|
Headers["Content-Type"] = contentType;
|
|
|
|
|
|
|
|
|
|
TotalContentLength = fileSystem.GetFileInfo(path).Length;
|
2017-05-05 19:55:38 +02:00
|
|
|
|
Headers["Accept-Ranges"] = "bytes";
|
2017-03-12 20:27:26 +01:00
|
|
|
|
|
|
|
|
|
if (string.IsNullOrWhiteSpace(rangeHeader))
|
|
|
|
|
{
|
|
|
|
|
Headers["Content-Length"] = TotalContentLength.ToString(UsCulture);
|
|
|
|
|
StatusCode = HttpStatusCode.OK;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
StatusCode = HttpStatusCode.PartialContent;
|
|
|
|
|
SetRangeValues();
|
|
|
|
|
}
|
|
|
|
|
|
2017-03-13 05:08:23 +01:00
|
|
|
|
FileShare = FileShareMode.Read;
|
2017-03-12 20:27:26 +01:00
|
|
|
|
Cookies = new List<Cookie>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Sets the range values.
|
|
|
|
|
/// </summary>
|
|
|
|
|
private void SetRangeValues()
|
|
|
|
|
{
|
|
|
|
|
var requestedRange = RequestedRanges[0];
|
|
|
|
|
|
|
|
|
|
// 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
|
2017-05-05 19:55:38 +02:00
|
|
|
|
var lengthString = RangeLength.ToString(UsCulture);
|
|
|
|
|
Headers["Content-Length"] = lengthString;
|
|
|
|
|
var rangeString = string.Format("bytes {0}-{1}/{2}", RangeStart, RangeEnd, TotalContentLength);
|
|
|
|
|
Headers["Content-Range"] = rangeString;
|
|
|
|
|
|
|
|
|
|
Logger.Info("Setting range response values for {0}. RangeRequest: {1} Content-Length: {2}, Content-Range: {3}", Path, RangeHeader, lengthString, rangeString);
|
2017-03-12 20:27:26 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <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;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
|
private string[] SkipLogExtensions = new string[]
|
|
|
|
|
{
|
|
|
|
|
".js",
|
|
|
|
|
".html",
|
|
|
|
|
".css"
|
|
|
|
|
};
|
|
|
|
|
|
2017-03-12 20:27:26 +01:00
|
|
|
|
public async Task WriteToAsync(IResponse response, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
// Headers only
|
|
|
|
|
if (IsHeadRequest)
|
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
|
var path = Path;
|
|
|
|
|
|
2017-03-12 20:27:26 +01:00
|
|
|
|
if (string.IsNullOrWhiteSpace(RangeHeader) || (RangeStart <= 0 && RangeEnd >= TotalContentLength - 1))
|
|
|
|
|
{
|
2018-09-12 19:26:21 +02:00
|
|
|
|
var extension = System.IO.Path.GetExtension(path);
|
|
|
|
|
|
|
|
|
|
if (extension == null || !SkipLogExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase))
|
|
|
|
|
{
|
|
|
|
|
Logger.Debug("Transmit file {0}", path);
|
|
|
|
|
}
|
2017-09-04 21:28:22 +02:00
|
|
|
|
|
|
|
|
|
//var count = FileShare == FileShareMode.ReadWrite ? TotalContentLength : 0;
|
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
|
await response.TransmitFile(path, 0, 0, FileShare, cancellationToken).ConfigureAwait(false);
|
2017-03-12 20:27:26 +01:00
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
|
await response.TransmitFile(path, RangeStart, RangeLength, FileShare, cancellationToken).ConfigureAwait(false);
|
2017-03-12 20:27:26 +01:00
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
if (OnComplete != null)
|
|
|
|
|
{
|
|
|
|
|
OnComplete();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string ContentType { get; set; }
|
|
|
|
|
|
|
|
|
|
public IRequest RequestContext { get; set; }
|
|
|
|
|
|
|
|
|
|
public object Response { get; set; }
|
|
|
|
|
|
|
|
|
|
public int Status { get; set; }
|
|
|
|
|
|
|
|
|
|
public HttpStatusCode StatusCode
|
|
|
|
|
{
|
|
|
|
|
get { return (HttpStatusCode)Status; }
|
|
|
|
|
set { Status = (int)value; }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string StatusDescription { get; set; }
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|