lgladsden/jellyfin-server
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Limit size for playbacktest

Browse Source
This commit is contained in:
Bond-009 2019-12-19 17:46:17 +01:00
parent a5cd11735c
commit bb62dd14c2
1 changed files with 27 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
MediaBrowser.Api/Playback/MediaInfoService.cs
View File

@ -1,4 +1,5 @@
using System;
using System.Buffers;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
@ -54,7 +55,7 @@ namespace MediaBrowser.Api.Playback
public class GetBitrateTestBytes
{
[ApiMember(Name = "Size", Description = "Size", IsRequired = true, DataType = "int", ParameterType = "query", Verb = "GET")]
public long Size { get; set; }
public int Size { get; set; }
public GetBitrateTestBytes()
{
@ -101,8 +102,31 @@ namespace MediaBrowser.Api.Playback
public object Get(GetBitrateTestBytes request)
{
var bytes = new byte[request.Size];
return ResultFactory.GetResult(null, bytes, "application/octet-stream");
const int MaxSize = 10_000_000;
var size = request.Size;
if (size <= 0)
{
throw new ArgumentException($"The requested size can't be equal or smaller than 0.", nameof(request));
}
if (size > MaxSize)
{
throw new ArgumentException($"The requested size can't be larger than the max allowed value ({MaxSize}).", nameof(request));
}
byte[] buffer = ArrayPool<byte>.Shared.Rent(size);
try
{
// ArrayPool<byte>.Shared.Rent doesn't guarantee that the returned buffer is zeroed
Array.Fill<byte>(buffer, 0);
return ResultFactory.GetResult(null, buffer, "application/octet-stream");
}
finally
{
ArrayPool<byte>.Shared.Return(buffer);
}
}
public async Task<object> Get(GetPlaybackInfo request)