1 files changed, 161 insertions, 0 deletions
diff --git a/MediaBrowser.Server.Startup.Common/FFMpeg/FFmpegValidator.cs b/MediaBrowser.Server.Startup.Common/FFMpeg/FFmpegValidator.cs
new file mode 100644
index 000000000..81ba1dc3a
--- /dev/null
+++ b/MediaBrowser.Server.Startup.Common/FFMpeg/FFmpegValidator.cs
@@ -0,0 +1,161 @@
+using MediaBrowser.Common.Configuration;
+using MediaBrowser.Common.Extensions;
+using MediaBrowser.Model.Logging;
+using System;
+using System.Diagnostics;
+using System.Globalization;
+using System.IO;
+using System.Text;
+
+namespace MediaBrowser.Server.Startup.Common.FFMpeg
+{
+    public class FFmpegValidator
+    {
+        private readonly ILogger _logger;
+        private readonly IApplicationPaths _appPaths;
+
+        public FFmpegValidator(ILogger logger, IApplicationPaths appPaths)
+        {
+            _logger = logger;
+            _appPaths = appPaths;
+        }
+
+        public void Validate(FFMpegInfo info)
+        {
+            _logger.Info("FFMpeg: {0}", info.EncoderPath);
+            _logger.Info("FFProbe: {0}", info.ProbePath);
+
+            string cacheKey;
+
+            try
+            {
+                cacheKey = new FileInfo(info.EncoderPath).Length.ToString(CultureInfo.InvariantCulture).GetMD5().ToString("N");
+            }
+            catch (IOException)
+            {
+                // This could happen if ffmpeg is coming from a Path variable and we don't have the full path
+                cacheKey = Guid.NewGuid().ToString("N");
+            }
+            catch
+            {
+                cacheKey = Guid.NewGuid().ToString("N");
+            }
+
+            var cachePath = Path.Combine(_appPaths.CachePath, "1" + cacheKey);
+
+            ValidateCodecs(info.EncoderPath, cachePath);
+        }
+
+        private void ValidateCodecs(string ffmpegPath, string cachePath)
+        {
+            string output = null;
+            try
+            {
+                output = File.ReadAllText(cachePath, Encoding.UTF8);
+            }
+            catch
+            {
+
+            }
+
+            if (string.IsNullOrWhiteSpace(output))
+            {
+                try
+                {
+                    output = GetFFMpegOutput(ffmpegPath, "-encoders");
+                }
+                catch
+                {
+                    return;
+                }
+
+                try
+                {
+                    Directory.CreateDirectory(Path.GetDirectoryName(cachePath));
+                    File.WriteAllText(cachePath, output, Encoding.UTF8);
+                }
+                catch
+                {
+
+                }
+            }
+
+            ValidateCodecsFromOutput(output);
+        }
+
+        private void ValidateCodecsFromOutput(string output)
+        {
+            var required = new[]
+            {
+                "libx264",
+                "mpeg4",
+                "msmpeg4",
+                "libvpx",
+                //"libvpx-vp9",
+                "aac",
+                "ac3",
+                "libmp3lame",
+                "libvorbis",
+                "srt"
+            };
+
+            foreach (var encoder in required)
+            {
+                var srch = " " + encoder + "  ";
+
+                if (output.IndexOf(srch, StringComparison.OrdinalIgnoreCase) == -1)
+                {
+                    _logger.Error("ffmpeg is missing encoder " + encoder);
+                    //throw new ArgumentException("ffmpeg is missing encoder " + encoder);
+                }
+            }
+        }
+
+        private string GetFFMpegOutput(string path, string arguments)
+        {
+            var process = new Process
+            {
+                StartInfo = new ProcessStartInfo
+                {
+                    CreateNoWindow = true,
+                    UseShellExecute = false,
+                    FileName = path,
+                    Arguments = arguments,
+                    WindowStyle = ProcessWindowStyle.Hidden,
+                    ErrorDialog = false,
+                    RedirectStandardOutput = true,
+                    RedirectStandardError = true
+                }
+            };
+
+            using (process)
+            {
+                process.Start();
+
+                try
+                {
+                    process.BeginErrorReadLine();
+
+                    using (var reader = new StreamReader(process.StandardOutput.BaseStream))
+                    {
+                        return reader.ReadToEnd();
+                    }
+                }
+                catch
+                {
+                    // Hate having to do this
+                    try
+                    {
+                        process.Kill();
+                    }
+                    catch (Exception ex1)
+                    {
+                        _logger.ErrorException("Error killing ffmpeg", ex1);
+                    }
+
+                    throw;
+                }
+            }
+        }
+    }
+}