jellyfin-server/MediaBrowser.MediaEncoding/Encoder/EncoderValidator.cs

198 lines
5.7 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Globalization;
2019-01-02 00:23:49 +00:00
using System.Linq;
using System.Text.RegularExpressions;
using MediaBrowser.Model.Diagnostics;
2018-12-14 23:48:06 +00:00
using Microsoft.Extensions.Logging;
namespace MediaBrowser.MediaEncoding.Encoder
{
public class EncoderValidator
{
private readonly ILogger _logger;
private readonly IProcessFactory _processFactory;
public EncoderValidator(ILogger logger, IProcessFactory processFactory)
{
_logger = logger;
_processFactory = processFactory;
}
2019-01-02 00:23:49 +00:00
public (IEnumerable<string> decoders, IEnumerable<string> encoders) Validate(string encoderPath)
{
2019-01-02 00:23:49 +00:00
_logger.LogInformation("Validating media encoder at {EncoderPath}", encoderPath);
var decoders = GetCodecs(encoderPath, Codec.Decoder);
var encoders = GetCodecs(encoderPath, Codec.Encoder);
2018-12-14 23:48:06 +00:00
_logger.LogInformation("Encoder validation complete");
2019-01-02 00:23:49 +00:00
return (decoders, encoders);
}
public bool ValidateVersion(string encoderAppPath, bool logOutput)
{
2019-01-02 00:23:49 +00:00
string output = null;
try
{
output = GetProcessOutput(encoderAppPath, "-version");
}
catch (Exception ex)
{
if (logOutput)
{
2018-12-20 12:11:26 +00:00
_logger.LogError(ex, "Error validating encoder");
}
}
if (string.IsNullOrWhiteSpace(output))
{
return false;
}
2019-01-05 21:40:33 +00:00
_logger.LogDebug("ffmpeg output: {Output}", output);
if (output.IndexOf("Libav developers", StringComparison.OrdinalIgnoreCase) != -1)
{
return false;
}
output = " " + output + " ";
for (var i = 2013; i <= 2015; i++)
{
var yearString = i.ToString(CultureInfo.InvariantCulture);
if (output.IndexOf(" " + yearString + " ", StringComparison.OrdinalIgnoreCase) != -1)
{
return false;
}
}
return true;
}
private static readonly string[] requiredDecoders = new[]
{
"mpeg2video",
"h264_qsv",
"hevc_qsv",
"mpeg2_qsv",
"vc1_qsv",
"h264_cuvid",
"hevc_cuvid",
"dts",
"ac3",
"aac",
"mp3",
"h264",
"hevc"
};
private static readonly string[] requiredEncoders = new[]
{
"libx264",
"libx265",
"mpeg4",
"msmpeg4",
"libvpx",
"libvpx-vp9",
"aac",
"libmp3lame",
"libopus",
"libvorbis",
"srt",
"h264_nvenc",
"hevc_nvenc",
"h264_qsv",
"hevc_qsv",
"h264_omx",
"hevc_omx",
"h264_vaapi",
"hevc_vaapi",
"ac3"
};
private enum Codec
{
Encoder,
Decoder
}
private IEnumerable<string> GetCodecs(string encoderAppPath, Codec codec)
{
string codecstr = codec == Codec.Encoder ? "encoders" : "decoders";
string output = null;
try
{
output = GetProcessOutput(encoderAppPath, "-" + codecstr);
}
catch (Exception ex)
{
_logger.LogError(ex, "Error detecting available {Codec}", codecstr);
}
if (string.IsNullOrWhiteSpace(output))
{
return Enumerable.Empty<string>();
}
var required = codec == Codec.Encoder ? requiredEncoders : requiredDecoders;
2019-01-02 10:32:35 +00:00
var found = Regex
.Matches(output, @"^\s\S{6}\s(?<codec>[\w|-]+)\s+.+$", RegexOptions.Multiline)
.Cast<Match>()
.Select(x => x.Groups["codec"].Value)
.Where(x => required.Contains(x));
_logger.LogInformation("Available {Codec}: {Codecs}", codecstr, found);
return found;
}
private string GetProcessOutput(string path, string arguments)
{
IProcess process = _processFactory.Create(new ProcessOptions
{
CreateNoWindow = true,
UseShellExecute = false,
FileName = path,
Arguments = arguments,
IsHidden = true,
ErrorDialog = false,
2019-01-05 21:40:33 +00:00
RedirectStandardOutput = true,
// ffmpeg uses stderr to log info, don't show this
RedirectStandardError = true
});
2019-01-05 21:40:33 +00:00
_logger.LogDebug("Running {Path} {Arguments}", path, arguments);
using (process)
{
process.Start();
try
{
return process.StandardOutput.ReadToEnd();
}
catch
{
2019-01-05 21:40:33 +00:00
_logger.LogWarning("Killing process {Path} {Arguments}", path, arguments);
// Hate having to do this
try
{
process.Kill();
}
2018-12-20 12:11:26 +00:00
catch (Exception ex)
{
2018-12-20 12:11:26 +00:00
_logger.LogError(ex, "Error killing process");
}
throw;
}
}
}
}
}