aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Controller/FFMpeg/FFProbe.cs
blob: f5364821ce8a0d5bbeaae4abbda5607e0046dde4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
using System;
using System.Diagnostics;
using System.IO;
using System.Threading.Tasks;
using MediaBrowser.Common.Logging;
using MediaBrowser.Common.Serialization;
using MediaBrowser.Model.Entities;

namespace MediaBrowser.Controller.FFMpeg
{
    /// <summary>
    /// Runs FFProbe against a media file and returns metadata.
    /// </summary>
    public static class FFProbe
    {
        public async static Task<FFProbeResult> Run(Audio item, string outputCachePath)
        {
            // Use try catch to avoid having to use File.Exists
            try
            {
                using (FileStream stream = File.OpenRead(outputCachePath))
                {
                    return JsonSerializer.DeserializeFromStream<FFProbeResult>(stream);
                }
            }
            catch (FileNotFoundException)
            {
            }

            await Run(item.Path, outputCachePath).ConfigureAwait(false);

            using (FileStream stream = File.OpenRead(outputCachePath))
            {
                return JsonSerializer.DeserializeFromStream<FFProbeResult>(stream);
            }
        }

        public async static Task<FFProbeResult> Run(Video item, string outputCachePath)
        {
            // Use try catch to avoid having to use File.Exists
            try
            {
                using (FileStream stream = File.OpenRead(outputCachePath))
                {
                    return JsonSerializer.DeserializeFromStream<FFProbeResult>(stream);
                }
            }
            catch (FileNotFoundException)
            {
            }

            await Run(item.Path, outputCachePath).ConfigureAwait(false);

            using (FileStream stream = File.OpenRead(outputCachePath))
            {
                return JsonSerializer.DeserializeFromStream<FFProbeResult>(stream);
            }
        }

        private async static Task Run(string input, string output)
        {
            ProcessStartInfo startInfo = new ProcessStartInfo();

            startInfo.CreateNoWindow = true;

            startInfo.UseShellExecute = false;

            // Must consume both or ffmpeg may hang due to deadlocks. See comments below.
            startInfo.RedirectStandardOutput = true;
            startInfo.RedirectStandardError = true;

            startInfo.FileName = Kernel.Instance.ApplicationPaths.FFProbePath;
            startInfo.WorkingDirectory = Kernel.Instance.ApplicationPaths.FFMpegDirectory;
            startInfo.Arguments = string.Format("\"{0}\" -v quiet -print_format json -show_streams -show_format", input);

            //Logger.LogInfo(startInfo.FileName + " " + startInfo.Arguments);

            Process process = new Process();
            process.StartInfo = startInfo;

            FileStream stream = new FileStream(output, FileMode.Create);

            try
            {
                process.Start();

                // MUST read both stdout and stderr asynchronously or a deadlock may occurr
                // If we ever decide to disable the ffmpeg log then you must uncomment the below line.
                process.BeginErrorReadLine();

                await process.StandardOutput.BaseStream.CopyToAsync(stream).ConfigureAwait(false);

                process.WaitForExit();

                stream.Dispose();

                if (process.ExitCode != 0)
                {
                    Logger.LogInfo("FFProbe exited with code {0} for {1}", process.ExitCode, input);
                }
            }
            catch (Exception ex)
            {
                Logger.LogException(ex);

                stream.Dispose();

                // Hate having to do this
                try
                {
                    process.Kill();
                }
                catch
                {
                }
                File.Delete(output);
            }
            finally
            {
                process.Dispose();
            }
        }
    }
}