mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +02:00
592 lines
20 KiB
C#
592 lines
20 KiB
C#
#pragma warning disable CS1591
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Globalization;
|
|
using System.Linq;
|
|
using System.Text.RegularExpressions;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace MediaBrowser.MediaEncoding.Encoder
|
|
{
|
|
public partial class EncoderValidator
|
|
{
|
|
private static readonly string[] _requiredDecoders = new[]
|
|
{
|
|
"h264",
|
|
"hevc",
|
|
"vp8",
|
|
"libvpx",
|
|
"vp9",
|
|
"libvpx-vp9",
|
|
"av1",
|
|
"libdav1d",
|
|
"mpeg2video",
|
|
"mpeg4",
|
|
"msmpeg4",
|
|
"dca",
|
|
"ac3",
|
|
"aac",
|
|
"mp3",
|
|
"flac",
|
|
"truehd",
|
|
"h264_qsv",
|
|
"hevc_qsv",
|
|
"mpeg2_qsv",
|
|
"vc1_qsv",
|
|
"vp8_qsv",
|
|
"vp9_qsv",
|
|
"av1_qsv",
|
|
"h264_cuvid",
|
|
"hevc_cuvid",
|
|
"mpeg2_cuvid",
|
|
"vc1_cuvid",
|
|
"mpeg4_cuvid",
|
|
"vp8_cuvid",
|
|
"vp9_cuvid",
|
|
"av1_cuvid"
|
|
};
|
|
|
|
private static readonly string[] _requiredEncoders = new[]
|
|
{
|
|
"libx264",
|
|
"libx265",
|
|
"libsvtav1",
|
|
"mpeg4",
|
|
"msmpeg4",
|
|
"libvpx",
|
|
"libvpx-vp9",
|
|
"aac",
|
|
"aac_at",
|
|
"libfdk_aac",
|
|
"ac3",
|
|
"dca",
|
|
"libmp3lame",
|
|
"libopus",
|
|
"libvorbis",
|
|
"flac",
|
|
"truehd",
|
|
"srt",
|
|
"h264_amf",
|
|
"hevc_amf",
|
|
"av1_amf",
|
|
"h264_qsv",
|
|
"hevc_qsv",
|
|
"av1_qsv",
|
|
"h264_nvenc",
|
|
"hevc_nvenc",
|
|
"av1_nvenc",
|
|
"h264_vaapi",
|
|
"hevc_vaapi",
|
|
"av1_vaapi",
|
|
"h264_v4l2m2m",
|
|
"h264_videotoolbox",
|
|
"hevc_videotoolbox"
|
|
};
|
|
|
|
private static readonly string[] _requiredFilters = new[]
|
|
{
|
|
// sw
|
|
"alphasrc",
|
|
"zscale",
|
|
// qsv
|
|
"scale_qsv",
|
|
"vpp_qsv",
|
|
"deinterlace_qsv",
|
|
"overlay_qsv",
|
|
// cuda
|
|
"scale_cuda",
|
|
"yadif_cuda",
|
|
"tonemap_cuda",
|
|
"overlay_cuda",
|
|
"hwupload_cuda",
|
|
// opencl
|
|
"scale_opencl",
|
|
"tonemap_opencl",
|
|
"overlay_opencl",
|
|
// vaapi
|
|
"scale_vaapi",
|
|
"deinterlace_vaapi",
|
|
"tonemap_vaapi",
|
|
"procamp_vaapi",
|
|
"overlay_vaapi",
|
|
"hwupload_vaapi",
|
|
// vulkan
|
|
"libplacebo",
|
|
"scale_vulkan",
|
|
"overlay_vulkan",
|
|
"hwupload_vaapi",
|
|
// videotoolbox
|
|
"yadif_videotoolbox"
|
|
};
|
|
|
|
private static readonly Dictionary<int, string[]> _filterOptionsDict = new Dictionary<int, string[]>
|
|
{
|
|
{ 0, new string[] { "scale_cuda", "Output format (default \"same\")" } },
|
|
{ 1, new string[] { "tonemap_cuda", "GPU accelerated HDR to SDR tonemapping" } },
|
|
{ 2, new string[] { "tonemap_opencl", "bt2390" } },
|
|
{ 3, new string[] { "overlay_opencl", "Action to take when encountering EOF from secondary input" } },
|
|
{ 4, new string[] { "overlay_vaapi", "Action to take when encountering EOF from secondary input" } },
|
|
{ 5, new string[] { "overlay_vulkan", "Action to take when encountering EOF from secondary input" } }
|
|
};
|
|
|
|
// These are the library versions that corresponds to our minimum ffmpeg version 4.x according to the version table below
|
|
private static readonly Dictionary<string, Version> _ffmpegMinimumLibraryVersions = new Dictionary<string, Version>
|
|
{
|
|
{ "libavutil", new Version(56, 14) },
|
|
{ "libavcodec", new Version(58, 18) },
|
|
{ "libavformat", new Version(58, 12) },
|
|
{ "libavdevice", new Version(58, 3) },
|
|
{ "libavfilter", new Version(7, 16) },
|
|
{ "libswscale", new Version(5, 1) },
|
|
{ "libswresample", new Version(3, 1) },
|
|
{ "libpostproc", new Version(55, 1) }
|
|
};
|
|
|
|
private readonly ILogger _logger;
|
|
|
|
private readonly string _encoderPath;
|
|
|
|
public EncoderValidator(ILogger logger, string encoderPath)
|
|
{
|
|
_logger = logger;
|
|
_encoderPath = encoderPath;
|
|
}
|
|
|
|
private enum Codec
|
|
{
|
|
Encoder,
|
|
Decoder
|
|
}
|
|
|
|
// When changing this, also change the minimum library versions in _ffmpegMinimumLibraryVersions
|
|
public static Version MinVersion { get; } = new Version(4, 0);
|
|
|
|
public static Version? MaxVersion { get; } = null;
|
|
|
|
[GeneratedRegex(@"^ffmpeg version n?((?:[0-9]+\.?)+)")]
|
|
private static partial Regex FfmpegVersionRegex();
|
|
|
|
[GeneratedRegex(@"((?<name>lib\w+)\s+(?<major>[0-9]+)\.\s*(?<minor>[0-9]+))", RegexOptions.Multiline)]
|
|
private static partial Regex LibraryRegex();
|
|
|
|
public bool ValidateVersion()
|
|
{
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-version", false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error validating encoder");
|
|
return false;
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(output))
|
|
{
|
|
_logger.LogError("FFmpeg validation: The process returned no result");
|
|
return false;
|
|
}
|
|
|
|
_logger.LogDebug("ffmpeg output: {Output}", output);
|
|
|
|
return ValidateVersionInternal(output);
|
|
}
|
|
|
|
internal bool ValidateVersionInternal(string versionOutput)
|
|
{
|
|
if (versionOutput.Contains("Libav developers", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
_logger.LogError("FFmpeg validation: avconv instead of ffmpeg is not supported");
|
|
return false;
|
|
}
|
|
|
|
// Work out what the version under test is
|
|
var version = GetFFmpegVersionInternal(versionOutput);
|
|
|
|
_logger.LogInformation("Found ffmpeg version {Version}", version is not null ? version.ToString() : "unknown");
|
|
|
|
if (version is null)
|
|
{
|
|
if (MaxVersion is not null) // Version is unknown
|
|
{
|
|
if (MinVersion == MaxVersion)
|
|
{
|
|
_logger.LogWarning("FFmpeg validation: We recommend version {MinVersion}", MinVersion);
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning("FFmpeg validation: We recommend a minimum of {MinVersion} and maximum of {MaxVersion}", MinVersion, MaxVersion);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning("FFmpeg validation: We recommend minimum version {MinVersion}", MinVersion);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
if (version < MinVersion) // Version is below what we recommend
|
|
{
|
|
_logger.LogWarning("FFmpeg validation: The minimum recommended version is {MinVersion}", MinVersion);
|
|
return false;
|
|
}
|
|
|
|
if (MaxVersion is not null && version > MaxVersion) // Version is above what we recommend
|
|
{
|
|
_logger.LogWarning("FFmpeg validation: The maximum recommended version is {MaxVersion}", MaxVersion);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public IEnumerable<string> GetDecoders() => GetCodecs(Codec.Decoder);
|
|
|
|
public IEnumerable<string> GetEncoders() => GetCodecs(Codec.Encoder);
|
|
|
|
public IEnumerable<string> GetHwaccels() => GetHwaccelTypes();
|
|
|
|
public IEnumerable<string> GetFilters() => GetFFmpegFilters();
|
|
|
|
public IDictionary<int, bool> GetFiltersWithOption() => GetFFmpegFiltersWithOption();
|
|
|
|
public Version? GetFFmpegVersion()
|
|
{
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-version", false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error validating encoder");
|
|
return null;
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(output))
|
|
{
|
|
_logger.LogError("FFmpeg validation: The process returned no result");
|
|
return null;
|
|
}
|
|
|
|
_logger.LogDebug("ffmpeg output: {Output}", output);
|
|
|
|
return GetFFmpegVersionInternal(output);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Using the output from "ffmpeg -version" work out the FFmpeg version.
|
|
/// For pre-built binaries the first line should contain a string like "ffmpeg version x.y", which is easy
|
|
/// to parse. If this is not available, then we try to match known library versions to FFmpeg versions.
|
|
/// If that fails then we test the libraries to determine if they're newer than our minimum versions.
|
|
/// </summary>
|
|
/// <param name="output">The output from "ffmpeg -version".</param>
|
|
/// <returns>The FFmpeg version.</returns>
|
|
internal Version? GetFFmpegVersionInternal(string output)
|
|
{
|
|
// For pre-built binaries the FFmpeg version should be mentioned at the very start of the output
|
|
var match = FfmpegVersionRegex().Match(output);
|
|
|
|
if (match.Success)
|
|
{
|
|
if (Version.TryParse(match.Groups[1].ValueSpan, out var result))
|
|
{
|
|
return result;
|
|
}
|
|
}
|
|
|
|
var versionMap = GetFFmpegLibraryVersions(output);
|
|
|
|
var allVersionsValidated = true;
|
|
|
|
foreach (var minimumVersion in _ffmpegMinimumLibraryVersions)
|
|
{
|
|
if (versionMap.TryGetValue(minimumVersion.Key, out var foundVersion))
|
|
{
|
|
if (foundVersion >= minimumVersion.Value)
|
|
{
|
|
_logger.LogInformation("Found {Library} version {FoundVersion} ({MinimumVersion})", minimumVersion.Key, foundVersion, minimumVersion.Value);
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning("Found {Library} version {FoundVersion} lower than recommended version {MinimumVersion}", minimumVersion.Key, foundVersion, minimumVersion.Value);
|
|
allVersionsValidated = false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_logger.LogError("{Library} version not found", minimumVersion.Key);
|
|
allVersionsValidated = false;
|
|
}
|
|
}
|
|
|
|
return allVersionsValidated ? MinVersion : null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Grabs the library names and major.minor version numbers from the 'ffmpeg -version' output
|
|
/// and condenses them on to one line. Output format is "name1=major.minor,name2=major.minor,etc.".
|
|
/// </summary>
|
|
/// <param name="output">The 'ffmpeg -version' output.</param>
|
|
/// <returns>The library names and major.minor version numbers.</returns>
|
|
private static Dictionary<string, Version> GetFFmpegLibraryVersions(string output)
|
|
{
|
|
var map = new Dictionary<string, Version>();
|
|
|
|
foreach (Match match in LibraryRegex().Matches(output))
|
|
{
|
|
var version = new Version(
|
|
int.Parse(match.Groups["major"].ValueSpan, CultureInfo.InvariantCulture),
|
|
int.Parse(match.Groups["minor"].ValueSpan, CultureInfo.InvariantCulture));
|
|
|
|
map.Add(match.Groups["name"].Value, version);
|
|
}
|
|
|
|
return map;
|
|
}
|
|
|
|
public bool CheckVaapiDeviceByDriverName(string driverName, string renderNodePath)
|
|
{
|
|
if (!OperatingSystem.IsLinux())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(driverName) || string.IsNullOrEmpty(renderNodePath))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
try
|
|
{
|
|
var output = GetProcessOutput(_encoderPath, "-v verbose -hide_banner -init_hw_device vaapi=va:" + renderNodePath, true, null);
|
|
return output.Contains(driverName, StringComparison.Ordinal);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting the given vaapi render node path");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public bool CheckVulkanDrmDeviceByExtensionName(string renderNodePath, string[] vulkanExtensions)
|
|
{
|
|
if (!OperatingSystem.IsLinux())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(renderNodePath))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
try
|
|
{
|
|
var command = "-v verbose -hide_banner -init_hw_device drm=dr:" + renderNodePath + " -init_hw_device vulkan=vk@dr";
|
|
var output = GetProcessOutput(_encoderPath, command, true, null);
|
|
foreach (string ext in vulkanExtensions)
|
|
{
|
|
if (!output.Contains(ext, StringComparison.Ordinal))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting the given drm render node path");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private IEnumerable<string> GetHwaccelTypes()
|
|
{
|
|
string? output = null;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-hwaccels", false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting available hwaccel types");
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(output))
|
|
{
|
|
return Enumerable.Empty<string>();
|
|
}
|
|
|
|
var found = output.Split(new char[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries).Skip(1).Distinct().ToList();
|
|
_logger.LogInformation("Available hwaccel types: {Types}", found);
|
|
|
|
return found;
|
|
}
|
|
|
|
public bool CheckFilterWithOption(string filter, string option)
|
|
{
|
|
if (string.IsNullOrEmpty(filter) || string.IsNullOrEmpty(option))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-h filter=" + filter, false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting the given filter");
|
|
return false;
|
|
}
|
|
|
|
if (output.Contains("Filter " + filter, StringComparison.Ordinal))
|
|
{
|
|
return output.Contains(option, StringComparison.Ordinal);
|
|
}
|
|
|
|
_logger.LogWarning("Filter: {Name} with option {Option} is not available", filter, option);
|
|
|
|
return false;
|
|
}
|
|
|
|
public bool CheckSupportedRuntimeKey(string keyDesc)
|
|
{
|
|
if (string.IsNullOrEmpty(keyDesc))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-hide_banner -f lavfi -i nullsrc=s=1x1:d=500 -f null -", true, "?");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error checking supported runtime key");
|
|
return false;
|
|
}
|
|
|
|
return output.Contains(keyDesc, StringComparison.Ordinal);
|
|
}
|
|
|
|
private IEnumerable<string> GetCodecs(Codec codec)
|
|
{
|
|
string codecstr = codec == Codec.Encoder ? "encoders" : "decoders";
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-" + codecstr, false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting available {Codec}", codecstr);
|
|
return Enumerable.Empty<string>();
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(output))
|
|
{
|
|
return Enumerable.Empty<string>();
|
|
}
|
|
|
|
var required = codec == Codec.Encoder ? _requiredEncoders : _requiredDecoders;
|
|
|
|
var found = CodecRegex()
|
|
.Matches(output)
|
|
.Select(x => x.Groups["codec"].Value)
|
|
.Where(x => required.Contains(x));
|
|
|
|
_logger.LogInformation("Available {Codec}: {Codecs}", codecstr, found);
|
|
|
|
return found;
|
|
}
|
|
|
|
private IEnumerable<string> GetFFmpegFilters()
|
|
{
|
|
string output;
|
|
try
|
|
{
|
|
output = GetProcessOutput(_encoderPath, "-filters", false, null);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error detecting available filters");
|
|
return Enumerable.Empty<string>();
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(output))
|
|
{
|
|
return Enumerable.Empty<string>();
|
|
}
|
|
|
|
var found = FilterRegex()
|
|
.Matches(output)
|
|
.Select(x => x.Groups["filter"].Value)
|
|
.Where(x => _requiredFilters.Contains(x));
|
|
|
|
_logger.LogInformation("Available filters: {Filters}", found);
|
|
|
|
return found;
|
|
}
|
|
|
|
private Dictionary<int, bool> GetFFmpegFiltersWithOption()
|
|
{
|
|
Dictionary<int, bool> dict = new Dictionary<int, bool>();
|
|
for (int i = 0; i < _filterOptionsDict.Count; i++)
|
|
{
|
|
if (_filterOptionsDict.TryGetValue(i, out var val) && val.Length == 2)
|
|
{
|
|
dict.Add(i, CheckFilterWithOption(val[0], val[1]));
|
|
}
|
|
}
|
|
|
|
return dict;
|
|
}
|
|
|
|
private string GetProcessOutput(string path, string arguments, bool readStdErr, string? testKey)
|
|
{
|
|
var redirectStandardIn = !string.IsNullOrEmpty(testKey);
|
|
using (var process = new Process
|
|
{
|
|
StartInfo = new ProcessStartInfo(path, arguments)
|
|
{
|
|
CreateNoWindow = true,
|
|
UseShellExecute = false,
|
|
WindowStyle = ProcessWindowStyle.Hidden,
|
|
ErrorDialog = false,
|
|
RedirectStandardInput = redirectStandardIn,
|
|
RedirectStandardOutput = true,
|
|
RedirectStandardError = true
|
|
}
|
|
})
|
|
{
|
|
_logger.LogDebug("Running {Path} {Arguments}", path, arguments);
|
|
|
|
process.Start();
|
|
|
|
if (redirectStandardIn)
|
|
{
|
|
using var writer = process.StandardInput;
|
|
writer.Write(testKey);
|
|
}
|
|
|
|
using var reader = readStdErr ? process.StandardError : process.StandardOutput;
|
|
return reader.ReadToEnd();
|
|
}
|
|
}
|
|
|
|
[GeneratedRegex("^\\s\\S{6}\\s(?<codec>[\\w|-]+)\\s+.+$", RegexOptions.Multiline)]
|
|
private static partial Regex CodecRegex();
|
|
|
|
[GeneratedRegex("^\\s\\S{3}\\s(?<filter>[\\w|-]+)\\s+.+$", RegexOptions.Multiline)]
|
|
private static partial Regex FilterRegex();
|
|
}
|
|
}
|