rework filestream

This commit is contained in:
Luke Pulverenti 2017-03-13 00:08:23 -04:00
parent 5e82194749
commit b38b7a7062
20 changed files with 108 additions and 34 deletions

View File

@ -100,7 +100,7 @@ namespace Emby.Common.Implementations.Net
#if NET46 #if NET46
public Task SendFile(string path, byte[] preBuffer, byte[] postBuffer, CancellationToken cancellationToken) public Task SendFile(string path, byte[] preBuffer, byte[] postBuffer, CancellationToken cancellationToken)
{ {
var options = TransmitFileOptions.Disconnect | TransmitFileOptions.ReuseSocket | TransmitFileOptions.UseKernelApc; var options = TransmitFileOptions.UseKernelApc;
var completionSource = new TaskCompletionSource<bool>(); var completionSource = new TaskCompletionSource<bool>();

View File

@ -97,6 +97,27 @@ namespace Emby.Common.Implementations.Net
} }
} }
public ISocket CreateUdpBroadcastSocket(int localPort)
{
if (localPort < 0) throw new ArgumentException("localPort cannot be less than zero.", "localPort");
var retVal = new Socket(AddressFamily.InterNetwork, System.Net.Sockets.SocketType.Dgram, System.Net.Sockets.ProtocolType.Udp);
try
{
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Broadcast, 1);
return new UdpSocket(retVal, localPort, IPAddress.Any);
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
/// <summary> /// <summary>
/// Creates a new UDP acceptSocket that is a member of the SSDP multicast local admin group and binds it to the specified local port. /// Creates a new UDP acceptSocket that is a member of the SSDP multicast local admin group and binds it to the specified local port.
/// </summary> /// </summary>

View File

@ -27,6 +27,8 @@ namespace Emby.Server.Implementations.HttpServer
private static readonly CultureInfo UsCulture = new CultureInfo("en-US"); private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
public List<Cookie> Cookies { get; private set; } public List<Cookie> Cookies { get; private set; }
public FileShareMode FileShare { get; set; }
/// <summary> /// <summary>
/// The _options /// The _options
/// </summary> /// </summary>
@ -69,6 +71,7 @@ namespace Emby.Server.Implementations.HttpServer
SetRangeValues(); SetRangeValues();
} }
FileShare = FileShareMode.Read;
Cookies = new List<Cookie>(); Cookies = new List<Cookie>();
} }
@ -153,11 +156,11 @@ namespace Emby.Server.Implementations.HttpServer
if (string.IsNullOrWhiteSpace(RangeHeader) || (RangeStart <= 0 && RangeEnd >= TotalContentLength - 1)) if (string.IsNullOrWhiteSpace(RangeHeader) || (RangeStart <= 0 && RangeEnd >= TotalContentLength - 1))
{ {
Logger.Info("Transmit file {0}", Path); Logger.Info("Transmit file {0}", Path);
await response.TransmitFile(Path, 0, 0, cancellationToken).ConfigureAwait(false); await response.TransmitFile(Path, 0, 0, FileShare, cancellationToken).ConfigureAwait(false);
return; return;
} }
await response.TransmitFile(Path, RangeStart, RangeEnd, cancellationToken).ConfigureAwait(false); await response.TransmitFile(Path, RangeStart, RangeEnd, FileShare, cancellationToken).ConfigureAwait(false);
} }
finally finally
{ {

View File

@ -556,12 +556,13 @@ namespace Emby.Server.Implementations.HttpServer
{ {
var rangeHeader = requestContext.Headers.Get("Range"); var rangeHeader = requestContext.Headers.Get("Range");
if (!isHeadRequest && !string.IsNullOrWhiteSpace(options.Path) && options.FileShare == FileShareMode.Read) if (!isHeadRequest && !string.IsNullOrWhiteSpace(options.Path))
{ {
return new FileWriter(options.Path, contentType, rangeHeader, _logger, _fileSystem) return new FileWriter(options.Path, contentType, rangeHeader, _logger, _fileSystem)
{ {
OnComplete = options.OnComplete, OnComplete = options.OnComplete,
OnError = options.OnError OnError = options.OnError,
FileShare = options.FileShare
}; };
} }

View File

@ -193,9 +193,9 @@ namespace Emby.Server.Implementations.HttpServer.SocketSharp
{ {
} }
public Task TransmitFile(string path, long offset, long count, CancellationToken cancellationToken) public Task TransmitFile(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken)
{ {
return _response.TransmitFile(path, offset, count, cancellationToken); return _response.TransmitFile(path, offset, count, fileShareMode, cancellationToken);
} }
} }
} }

View File

@ -513,6 +513,11 @@ namespace Emby.Server.Implementations.Library
} }
public Guid GetNewItemId(string key, Type type) public Guid GetNewItemId(string key, Type type)
{
return GetNewItemIdInternal(key, type, false);
}
private Guid GetNewItemIdInternal(string key, Type type, bool forceCaseInsensitive)
{ {
if (string.IsNullOrWhiteSpace(key)) if (string.IsNullOrWhiteSpace(key))
{ {
@ -531,7 +536,7 @@ namespace Emby.Server.Implementations.Library
.Replace("/", "\\"); .Replace("/", "\\");
} }
if (!ConfigurationManager.Configuration.EnableCaseSensitiveItemIds) if (forceCaseInsensitive || !ConfigurationManager.Configuration.EnableCaseSensitiveItemIds)
{ {
key = key.ToLower(); key = key.ToLower();
} }
@ -865,7 +870,7 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{Person}.</returns> /// <returns>Task{Person}.</returns>
public Person GetPerson(string name) public Person GetPerson(string name)
{ {
return CreateItemByName<Person>(Person.GetPath(name), name); return CreateItemByName<Person>(Person.GetPath, name);
} }
/// <summary> /// <summary>
@ -875,7 +880,7 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{Studio}.</returns> /// <returns>Task{Studio}.</returns>
public Studio GetStudio(string name) public Studio GetStudio(string name)
{ {
return CreateItemByName<Studio>(Studio.GetPath(name), name); return CreateItemByName<Studio>(Studio.GetPath, name);
} }
/// <summary> /// <summary>
@ -885,7 +890,7 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{Genre}.</returns> /// <returns>Task{Genre}.</returns>
public Genre GetGenre(string name) public Genre GetGenre(string name)
{ {
return CreateItemByName<Genre>(Genre.GetPath(name), name); return CreateItemByName<Genre>(Genre.GetPath, name);
} }
/// <summary> /// <summary>
@ -895,7 +900,7 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{MusicGenre}.</returns> /// <returns>Task{MusicGenre}.</returns>
public MusicGenre GetMusicGenre(string name) public MusicGenre GetMusicGenre(string name)
{ {
return CreateItemByName<MusicGenre>(MusicGenre.GetPath(name), name); return CreateItemByName<MusicGenre>(MusicGenre.GetPath, name);
} }
/// <summary> /// <summary>
@ -905,7 +910,7 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{GameGenre}.</returns> /// <returns>Task{GameGenre}.</returns>
public GameGenre GetGameGenre(string name) public GameGenre GetGameGenre(string name)
{ {
return CreateItemByName<GameGenre>(GameGenre.GetPath(name), name); return CreateItemByName<GameGenre>(GameGenre.GetPath, name);
} }
/// <summary> /// <summary>
@ -923,7 +928,7 @@ namespace Emby.Server.Implementations.Library
var name = value.ToString(CultureInfo.InvariantCulture); var name = value.ToString(CultureInfo.InvariantCulture);
return CreateItemByName<Year>(Year.GetPath(name), name); return CreateItemByName<Year>(Year.GetPath, name);
} }
/// <summary> /// <summary>
@ -933,10 +938,10 @@ namespace Emby.Server.Implementations.Library
/// <returns>Task{Genre}.</returns> /// <returns>Task{Genre}.</returns>
public MusicArtist GetArtist(string name) public MusicArtist GetArtist(string name)
{ {
return CreateItemByName<MusicArtist>(MusicArtist.GetPath(name), name); return CreateItemByName<MusicArtist>(MusicArtist.GetPath, name);
} }
private T CreateItemByName<T>(string path, string name) private T CreateItemByName<T>(Func<string,string> getPathFn, string name)
where T : BaseItem, new() where T : BaseItem, new()
{ {
if (typeof(T) == typeof(MusicArtist)) if (typeof(T) == typeof(MusicArtist))
@ -957,7 +962,9 @@ namespace Emby.Server.Implementations.Library
} }
} }
var id = GetNewItemId(path, typeof(T)); var path = getPathFn(name);
var forceCaseInsensitiveId = ConfigurationManager.Configuration.EnableNormalizedItemByNameIds;
var id = GetNewItemIdInternal(path, typeof(T), forceCaseInsensitiveId);
var item = GetItemById(id) as T; var item = GetItemById(id) as T;

View File

@ -120,6 +120,7 @@ namespace MediaBrowser.Api
config.EnableSeriesPresentationUniqueKey = true; config.EnableSeriesPresentationUniqueKey = true;
config.EnableLocalizedGuids = true; config.EnableLocalizedGuids = true;
config.EnableSimpleArtistDetection = true; config.EnableSimpleArtistDetection = true;
config.EnableNormalizedItemByNameIds = true;
} }
public void Post(UpdateStartupConfiguration request) public void Post(UpdateStartupConfiguration request)

View File

@ -289,7 +289,12 @@ namespace MediaBrowser.Controller.Entities.Audio
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -118,7 +118,12 @@ namespace MediaBrowser.Controller.Entities.Audio
return LibraryManager.GetItemList(query); return LibraryManager.GetItemList(query);
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -96,7 +96,12 @@ namespace MediaBrowser.Controller.Entities
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -108,7 +108,12 @@ namespace MediaBrowser.Controller.Entities
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -133,7 +133,12 @@ namespace MediaBrowser.Controller.Entities
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validFilename = normalizeName ? var validFilename = normalizeName ?

View File

@ -114,7 +114,12 @@ namespace MediaBrowser.Controller.Entities
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -122,7 +122,12 @@ namespace MediaBrowser.Controller.Entities
} }
} }
public static string GetPath(string name, bool normalizeName = true) public static string GetPath(string name)
{
return GetPath(name, true);
}
public static string GetPath(string name, bool normalizeName)
{ {
// Trim the period at the end because windows will have a hard time with that // Trim the period at the end because windows will have a hard time with that
var validName = normalizeName ? var validName = normalizeName ?

View File

@ -44,6 +44,8 @@ namespace MediaBrowser.Controller.LiveTv
/// <param name="cancellationToken">The cancellation token.</param> /// <param name="cancellationToken">The cancellation token.</param>
/// <returns>Task&lt;List&lt;MediaSourceInfo&gt;&gt;.</returns> /// <returns>Task&lt;List&lt;MediaSourceInfo&gt;&gt;.</returns>
Task<List<MediaSourceInfo>> GetChannelStreamMediaSources(string channelId, CancellationToken cancellationToken); Task<List<MediaSourceInfo>> GetChannelStreamMediaSources(string channelId, CancellationToken cancellationToken);
Task<List<TunerHostInfo>> DiscoverDevices(int discoveryDurationMs);
} }
public interface IConfigurableTunerHost public interface IConfigurableTunerHost
{ {

View File

@ -48,6 +48,7 @@ namespace MediaBrowser.Model.Configuration
public bool EnableHttps { get; set; } public bool EnableHttps { get; set; }
public bool EnableSeriesPresentationUniqueKey { get; set; } public bool EnableSeriesPresentationUniqueKey { get; set; }
public bool EnableLocalizedGuids { get; set; } public bool EnableLocalizedGuids { get; set; }
public bool EnableNormalizedItemByNameIds { get; set; }
/// <summary> /// <summary>
/// Gets or sets the value pointing to the file system where the ssl certiifcate is located.. /// Gets or sets the value pointing to the file system where the ssl certiifcate is located..

View File

@ -14,6 +14,8 @@ namespace MediaBrowser.Model.Net
/// <returns>A <see cref="ISocket"/> implementation.</returns> /// <returns>A <see cref="ISocket"/> implementation.</returns>
ISocket CreateUdpSocket(int localPort); ISocket CreateUdpSocket(int localPort);
ISocket CreateUdpBroadcastSocket(int localPort);
ISocket CreateTcpSocket(IpAddressInfo remoteAddress, int remotePort); ISocket CreateTcpSocket(IpAddressInfo remoteAddress, int remotePort);
/// <summary> /// <summary>

View File

@ -4,6 +4,7 @@ using System.IO;
using System.Net; using System.Net;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
using MediaBrowser.Model.IO;
namespace MediaBrowser.Model.Services namespace MediaBrowser.Model.Services
{ {
@ -154,6 +155,6 @@ namespace MediaBrowser.Model.Services
//Add Metadata to Response //Add Metadata to Response
Dictionary<string, object> Items { get; } Dictionary<string, object> Items { get; }
Task TransmitFile(string path, long offset, long count, CancellationToken cancellationToken); Task TransmitFile(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken);
} }
} }

View File

@ -515,9 +515,9 @@ namespace SocketHttpListener.Net
cookies.Add(cookie); cookies.Add(cookie);
} }
public Task TransmitFile(string path, long offset, long count, CancellationToken cancellationToken) public Task TransmitFile(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken)
{ {
return ((ResponseStream)OutputStream).TransmitFile(path, offset, count, cancellationToken); return ((ResponseStream)OutputStream).TransmitFile(path, offset, count, fileShareMode, cancellationToken);
} }
} }
} }

View File

@ -307,13 +307,13 @@ namespace SocketHttpListener.Net
throw new NotSupportedException(); throw new NotSupportedException();
} }
public Task TransmitFile(string path, long offset, long count, CancellationToken cancellationToken) public Task TransmitFile(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken)
{ {
//if (_supportsDirectSocketAccess && offset == 0 && count == 0 && !response.SendChunked) //if (_supportsDirectSocketAccess && offset == 0 && count == 0 && !response.SendChunked)
//{ //{
// return TransmitFileOverSocket(path, offset, count, cancellationToken); // return TransmitFileOverSocket(path, offset, count, cancellationToken);
//} //}
return TransmitFileManaged(path, offset, count, cancellationToken); return TransmitFileManaged(path, offset, count, fileShareMode, cancellationToken);
} }
private readonly byte[] _emptyBuffer = new byte[] { }; private readonly byte[] _emptyBuffer = new byte[] { };
@ -334,7 +334,7 @@ namespace SocketHttpListener.Net
await _socket.SendFile(path, buffer, _emptyBuffer, cancellationToken).ConfigureAwait(false); await _socket.SendFile(path, buffer, _emptyBuffer, cancellationToken).ConfigureAwait(false);
} }
private async Task TransmitFileManaged(string path, long offset, long count, CancellationToken cancellationToken) private async Task TransmitFileManaged(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken)
{ {
var chunked = response.SendChunked; var chunked = response.SendChunked;
@ -343,7 +343,7 @@ namespace SocketHttpListener.Net
await WriteAsync(_emptyBuffer, 0, 0, cancellationToken).ConfigureAwait(false); await WriteAsync(_emptyBuffer, 0, 0, cancellationToken).ConfigureAwait(false);
} }
using (var fs = _fileSystem.GetFileStream(path, FileOpenMode.Open, FileAccessMode.Read, FileShareMode.Read, true)) using (var fs = _fileSystem.GetFileStream(path, FileOpenMode.Open, FileAccessMode.Read, fileShareMode, true))
{ {
if (offset > 0) if (offset > 0)
{ {