jellyfin/MediaBrowser.MediaEncoding/Encoder/EncoderValidator.cs

215 lines
5.8 KiB
C#
Raw Normal View History

2016-06-30 04:23:52 +00:00
using System;
using System.Collections.Generic;
2014-11-15 02:31:03 +00:00
using System.Diagnostics;
2016-11-06 17:30:44 +00:00
using System.Globalization;
2016-11-01 04:07:12 +00:00
using MediaBrowser.Model.Diagnostics;
2016-06-30 04:23:52 +00:00
using MediaBrowser.Model.Logging;
2014-11-15 02:31:03 +00:00
2016-06-30 04:23:52 +00:00
namespace MediaBrowser.MediaEncoding.Encoder
2014-11-15 02:31:03 +00:00
{
2016-06-30 04:23:52 +00:00
public class EncoderValidator
2014-11-15 02:31:03 +00:00
{
private readonly ILogger _logger;
2016-11-01 04:07:12 +00:00
private readonly IProcessFactory _processFactory;
2014-11-15 02:31:03 +00:00
2016-11-01 04:07:12 +00:00
public EncoderValidator(ILogger logger, IProcessFactory processFactory)
2014-11-15 02:31:03 +00:00
{
_logger = logger;
2016-11-01 04:07:12 +00:00
_processFactory = processFactory;
2014-11-15 02:31:03 +00:00
}
2016-06-30 04:23:52 +00:00
public Tuple<List<string>, List<string>> Validate(string encoderPath)
2014-11-15 02:31:03 +00:00
{
2016-06-30 04:23:52 +00:00
_logger.Info("Validating media encoder at {0}", encoderPath);
2016-06-20 06:19:28 +00:00
var decoders = GetDecoders(encoderPath);
var encoders = GetEncoders(encoderPath);
2014-11-29 02:40:46 +00:00
2016-06-30 04:23:52 +00:00
_logger.Info("Encoder validation complete");
return new Tuple<List<string>, List<string>>(decoders, encoders);
2014-11-18 02:48:22 +00:00
}
2016-10-28 18:35:17 +00:00
public bool ValidateVersion(string encoderAppPath, bool logOutput)
2016-09-22 06:57:31 +00:00
{
string output = string.Empty;
try
{
output = GetProcessOutput(encoderAppPath, "-version");
}
2016-11-13 21:04:21 +00:00
catch (Exception ex)
2016-09-22 06:57:31 +00:00
{
2016-11-13 21:04:21 +00:00
if (logOutput)
{
_logger.ErrorException("Error validating encoder", ex);
}
2016-09-22 06:57:31 +00:00
}
2016-11-13 21:04:21 +00:00
if (string.IsNullOrWhiteSpace(output))
2016-10-28 18:35:17 +00:00
{
2016-11-13 21:04:21 +00:00
return false;
2016-10-28 18:35:17 +00:00
}
2016-11-13 21:04:21 +00:00
if (logOutput)
2016-09-22 06:57:31 +00:00
{
2016-11-13 21:04:21 +00:00
_logger.Info("ffmpeg info: {0}", output);
2016-09-22 06:57:31 +00:00
}
if (output.IndexOf("Libav developers", StringComparison.OrdinalIgnoreCase) != -1)
{
return false;
}
2016-11-06 17:30:44 +00:00
output = " " + output + " ";
for (var i = 2013; i <= 2015; i++)
{
var yearString = i.ToString(CultureInfo.InvariantCulture);
if (output.IndexOf(" " + yearString + " ", StringComparison.OrdinalIgnoreCase) != -1)
{
return false;
}
}
2016-09-22 06:57:31 +00:00
return true;
}
2016-06-30 18:59:18 +00:00
private List<string> GetDecoders(string encoderAppPath)
2014-11-18 02:48:22 +00:00
{
string output = string.Empty;
2014-11-18 02:48:22 +00:00
try
{
2016-06-30 18:59:18 +00:00
output = GetProcessOutput(encoderAppPath, "-decoders");
2014-11-18 02:48:22 +00:00
}
2016-11-19 07:51:07 +00:00
catch (Exception )
2014-11-15 02:31:03 +00:00
{
2016-11-19 07:51:07 +00:00
//_logger.ErrorException("Error detecting available decoders", ex);
2014-11-15 02:31:03 +00:00
}
2014-11-18 02:48:22 +00:00
var found = new List<string>();
var required = new[]
2014-11-18 02:48:22 +00:00
{
2017-03-16 17:21:24 +00:00
"mpeg2video",
"h264_qsv",
"hevc_qsv",
"mpeg2_qsv",
"vc1_qsv"
};
foreach (var codec in required)
{
var srch = " " + codec + " ";
2014-11-18 02:48:22 +00:00
2016-06-30 18:59:18 +00:00
if (output.IndexOf(srch, StringComparison.OrdinalIgnoreCase) != -1)
2014-11-18 02:48:22 +00:00
{
2016-06-30 18:59:18 +00:00
_logger.Info("Decoder available: " + codec);
found.Add(codec);
2014-11-18 02:48:22 +00:00
}
}
return found;
2014-11-15 02:31:03 +00:00
}
2016-06-30 18:59:18 +00:00
private List<string> GetEncoders(string encoderAppPath)
2014-11-15 02:31:03 +00:00
{
string output = null;
try
{
2016-06-30 18:59:18 +00:00
output = GetProcessOutput(encoderAppPath, "-encoders");
}
catch
{
}
var found = new List<string>();
2014-11-15 02:31:03 +00:00
var required = new[]
{
"libx264",
2015-01-11 03:06:16 +00:00
"libx265",
2014-11-15 02:31:03 +00:00
"mpeg4",
"msmpeg4",
2016-07-01 01:06:18 +00:00
"libvpx",
"libvpx-vp9",
2014-11-15 02:31:03 +00:00
"aac",
"libmp3lame",
"libopus",
2016-07-01 01:06:18 +00:00
"libvorbis",
2016-06-30 18:59:18 +00:00
"srt",
"h264_nvenc",
"hevc_nvenc",
2016-07-08 03:21:06 +00:00
"h264_qsv",
"hevc_qsv",
2016-08-05 05:12:25 +00:00
"h264_omx",
"hevc_omx",
2016-08-07 20:13:30 +00:00
"h264_vaapi",
"hevc_vaapi",
2016-07-08 03:21:06 +00:00
"ac3"
2014-11-15 02:31:03 +00:00
};
2016-07-01 01:06:18 +00:00
output = output ?? string.Empty;
2016-07-08 03:21:06 +00:00
var index = 0;
foreach (var codec in required)
2014-11-15 02:31:03 +00:00
{
var srch = " " + codec + " ";
2014-11-15 02:31:03 +00:00
2016-06-30 18:59:18 +00:00
if (output.IndexOf(srch, StringComparison.OrdinalIgnoreCase) != -1)
{
2016-07-08 03:21:06 +00:00
if (index < required.Length - 1)
{
_logger.Info("Encoder available: " + codec);
}
found.Add(codec);
2014-11-15 02:31:03 +00:00
}
2016-07-08 03:21:06 +00:00
index++;
2014-11-15 02:31:03 +00:00
}
return found;
2014-11-15 02:31:03 +00:00
}
2016-06-30 18:59:18 +00:00
private string GetProcessOutput(string path, string arguments)
2014-11-15 02:31:03 +00:00
{
2016-11-01 04:07:12 +00:00
var process = _processFactory.Create(new ProcessOptions
2014-11-15 02:31:03 +00:00
{
2016-11-01 04:07:12 +00:00
CreateNoWindow = true,
UseShellExecute = false,
FileName = path,
Arguments = arguments,
IsHidden = true,
ErrorDialog = false,
RedirectStandardOutput = true
});
2014-11-15 02:31:03 +00:00
2016-10-16 17:11:32 +00:00
_logger.Info("Running {0} {1}", path, arguments);
2014-11-15 02:31:03 +00:00
using (process)
{
process.Start();
try
{
2016-06-30 04:41:15 +00:00
return process.StandardOutput.ReadToEnd();
2014-11-15 02:31:03 +00:00
}
catch
{
2016-06-30 04:41:15 +00:00
_logger.Info("Killing process {0} {1}", path, arguments);
2014-11-15 02:31:03 +00:00
// Hate having to do this
try
{
process.Kill();
}
catch (Exception ex1)
{
2016-06-30 18:59:18 +00:00
_logger.ErrorException("Error killing process", ex1);
2014-11-15 02:31:03 +00:00
}
throw;
}
}
}
}
}