2021-09-11 12:39:52 +02:00
|
|
|
|
// The MIT License (MIT)
|
|
|
|
|
//
|
|
|
|
|
// Copyright (c) .NET Foundation and Contributors
|
|
|
|
|
//
|
|
|
|
|
// All rights reserved.
|
|
|
|
|
//
|
|
|
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
|
|
|
// in the Software without restriction, including without limitation the rights
|
|
|
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
|
|
|
// furnished to do so, subject to the following conditions:
|
|
|
|
|
//
|
|
|
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
|
|
|
// copies or substantial portions of the Software.
|
|
|
|
|
//
|
|
|
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
|
// SOFTWARE.
|
|
|
|
|
|
|
|
|
|
using System;
|
2021-09-11 12:31:29 +02:00
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
using MediaBrowser.Model.IO;
|
|
|
|
|
using Microsoft.AspNetCore.Http;
|
|
|
|
|
using Microsoft.AspNetCore.Http.Extensions;
|
|
|
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
|
using Microsoft.AspNetCore.Mvc.Infrastructure;
|
|
|
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
|
using Microsoft.Net.Http.Headers;
|
|
|
|
|
|
|
|
|
|
namespace Jellyfin.Server.Infrastructure
|
|
|
|
|
{
|
|
|
|
|
/// <inheritdoc />
|
|
|
|
|
public class SymlinkFollowingPhysicalFileResultExecutor : PhysicalFileResultExecutor
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Initializes a new instance of the <see cref="SymlinkFollowingPhysicalFileResultExecutor"/> class.
|
|
|
|
|
/// </summary>
|
2021-09-11 12:47:01 +02:00
|
|
|
|
/// <param name="loggerFactory">An instance of the <see cref="ILoggerFactory"/> interface.</param>
|
2021-09-11 12:31:29 +02:00
|
|
|
|
public SymlinkFollowingPhysicalFileResultExecutor(ILoggerFactory loggerFactory) : base(loggerFactory)
|
|
|
|
|
{
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
|
|
|
|
protected override FileMetadata GetFileInfo(string path)
|
|
|
|
|
{
|
|
|
|
|
var fileInfo = new FileInfo(path);
|
|
|
|
|
var length = fileInfo.Length;
|
|
|
|
|
// This may or may not be fixed in .NET 6, but looks like it will not https://github.com/dotnet/aspnetcore/issues/34371
|
|
|
|
|
if ((fileInfo.Attributes & FileAttributes.ReparsePoint) == FileAttributes.ReparsePoint)
|
|
|
|
|
{
|
2021-09-11 14:09:22 +02:00
|
|
|
|
using Stream thisFileStream = File.OpenRead(path);
|
2021-09-11 12:31:29 +02:00
|
|
|
|
length = thisFileStream.Length;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new FileMetadata
|
|
|
|
|
{
|
|
|
|
|
Exists = fileInfo.Exists,
|
|
|
|
|
Length = length,
|
|
|
|
|
LastModified = fileInfo.LastWriteTimeUtc
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
2021-02-14 15:11:46 +01:00
|
|
|
|
protected override Task WriteFileAsync(ActionContext context, PhysicalFileResult result, RangeItemHeaderValue? range, long rangeLength)
|
2021-09-11 12:31:29 +02:00
|
|
|
|
{
|
|
|
|
|
if (context == null)
|
|
|
|
|
{
|
|
|
|
|
throw new ArgumentNullException(nameof(context));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (result == null)
|
|
|
|
|
{
|
|
|
|
|
throw new ArgumentNullException(nameof(result));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (range != null && rangeLength == 0)
|
|
|
|
|
{
|
|
|
|
|
return Task.CompletedTask;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// It's a bit of wasted IO to perform this check again, but non-symlinks shouldn't use this code
|
|
|
|
|
if (!IsSymLink(result.FileName))
|
|
|
|
|
{
|
|
|
|
|
return base.WriteFileAsync(context, result, range, rangeLength);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var response = context.HttpContext.Response;
|
|
|
|
|
|
|
|
|
|
if (range != null)
|
|
|
|
|
{
|
2021-09-11 12:47:01 +02:00
|
|
|
|
return SendFileAsync(
|
|
|
|
|
result.FileName,
|
2021-09-11 12:31:29 +02:00
|
|
|
|
response,
|
|
|
|
|
offset: range.From ?? 0L,
|
|
|
|
|
count: rangeLength);
|
|
|
|
|
}
|
|
|
|
|
|
2021-09-11 12:47:01 +02:00
|
|
|
|
return SendFileAsync(
|
|
|
|
|
result.FileName,
|
2021-09-11 12:31:29 +02:00
|
|
|
|
response,
|
|
|
|
|
offset: 0,
|
|
|
|
|
count: null);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task SendFileAsync(string filePath, HttpResponse response, long offset, long? count)
|
|
|
|
|
{
|
|
|
|
|
var fileInfo = GetFileInfo(filePath);
|
|
|
|
|
if (offset < 0 || offset > fileInfo.Length)
|
|
|
|
|
{
|
|
|
|
|
throw new ArgumentOutOfRangeException(nameof(offset), offset, string.Empty);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (count.HasValue
|
|
|
|
|
&& (count.Value < 0 || count.Value > fileInfo.Length - offset))
|
|
|
|
|
{
|
|
|
|
|
throw new ArgumentOutOfRangeException(nameof(count), count, string.Empty);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Copied from SendFileFallback.SendFileAsync
|
2021-09-11 14:19:50 +02:00
|
|
|
|
const int BufferSize = 1024 * 16;
|
2021-09-11 12:31:29 +02:00
|
|
|
|
|
|
|
|
|
await using var fileStream = new FileStream(
|
|
|
|
|
filePath,
|
|
|
|
|
FileMode.Open,
|
|
|
|
|
FileAccess.Read,
|
|
|
|
|
FileShare.ReadWrite,
|
2021-09-11 15:06:56 +02:00
|
|
|
|
bufferSize: BufferSize,
|
2021-09-25 19:44:40 +02:00
|
|
|
|
options: FileOptions.Asynchronous | FileOptions.SequentialScan);
|
2021-09-11 12:31:29 +02:00
|
|
|
|
|
|
|
|
|
fileStream.Seek(offset, SeekOrigin.Begin);
|
|
|
|
|
await StreamCopyOperation
|
2021-09-11 15:07:09 +02:00
|
|
|
|
.CopyToAsync(fileStream, response.Body, count, BufferSize, CancellationToken.None)
|
2021-09-11 12:31:29 +02:00
|
|
|
|
.ConfigureAwait(true);
|
|
|
|
|
}
|
|
|
|
|
|
2021-09-11 13:32:59 +02:00
|
|
|
|
private static bool IsSymLink(string path) => (File.GetAttributes(path) & FileAttributes.ReparsePoint) == FileAttributes.ReparsePoint;
|
2021-09-11 12:31:29 +02:00
|
|
|
|
}
|
|
|
|
|
}
|