.NET进程 - 重定向stdin和stdout而不会导致死锁

时间:2017-07-21 01:29:08

标签: c# .net ffmpeg stdout deadlock

我尝试使用FFmpeg从我的程序生成的帧中编码视频文件,然后将FFmpeg的输出重定向回我的程序,以避免出现中间视频文件。

但是,在文档here的备注中提到的 System.Diagnostic.Process 中重定向输出时,我遇到了一个相当常见的问题,如果同步运行会导致死锁。

在我的头发上撕了一天,并尝试在网上找到几个建议的解决方案后,我仍然找不到让它工作的方法。我得到了一些数据,但是这个过程总是会在它结束之前冻结。

以下是产生上述问题的代码段:

static void Main(string[] args)
    {

        Process proc = new Process();

        proc.StartInfo.FileName = @"ffmpeg.exe";
        proc.StartInfo.Arguments = String.Format("-f rawvideo -vcodec rawvideo -s {0}x{1} -pix_fmt rgb24 -r {2} -i - -an -codec:v libx264 -preset veryfast -f mp4 -movflags frag_keyframe+empty_moov -",
            16, 9, 30);
        proc.StartInfo.UseShellExecute = false;
        proc.StartInfo.RedirectStandardInput = true;
        proc.StartInfo.RedirectStandardOutput = true;

        FileStream fs = new FileStream(@"out.mp4", FileMode.Create, FileAccess.Write);

        //read output asynchronously
        using (AutoResetEvent outputWaitHandle = new AutoResetEvent(false))
        {
            proc.OutputDataReceived += (sender, e) =>
            {
                if (e.Data == null)
                {
                    outputWaitHandle.Set();
                }
                else
                {
                    string str = e.Data;
                    byte[] bytes = new byte[str.Length * sizeof(char)];
                    System.Buffer.BlockCopy(str.ToCharArray(), 0, bytes, 0, bytes.Length);
                    fs.Write(bytes, 0, bytes.Length);
                }
            };
        }


        proc.Start();
        proc.BeginOutputReadLine();

        //Generate frames and write to stdin
        for (int i = 0; i < 30*60*60; i++)
        {
            byte[] myArray = Enumerable.Repeat((byte)Math.Min(i,255), 9*16*3).ToArray();
            proc.StandardInput.BaseStream.Write(myArray, 0, myArray.Length);
        }

        proc.WaitForExit();
        fs.Close();
        Console.WriteLine("Done!");
        Console.ReadKey();

    }

目前我还试图将输出写入文件以进行调试,但这并不是最终如何使用数据。

如果有人知道解决方案,我们将非常感激。

1 个答案:

答案 0 :(得分:2)

您的进程挂起,因为您从未告诉ffmpeg您已写完StandardInput流。所以它仍然坐在那里等着你发送更多的数据。

就我个人而言,我认为使用常规文件作为ffmpeg的输入和输出选项(即命令中的 infile outfile )将更可靠,更容易编码行参数)。然后您不需要重定向任何流。

但是,如果你确实想要或需要重定向自己 - 例如,你实际上是在独立于某个文件生成输入数据,并且你以某种方式使用输出数据而不是文件 - 你可以得到它通过正确使用Process类及其属性来工作。

特别是,这意味着一些事情:

  1. 当程序为流生成原始字节数据时,不能将输出数据视为字符。
  2. 完成后, 必须关闭输入流,以便程序知道已到达输入流的末尾。
  3. 以下是您的程序实际可行的版本:

    static void Main(string[] args)
    {
        Process proc = new Process();
    
        proc.StartInfo.FileName = @"ffmpeg.exe";
        proc.StartInfo.Arguments = String.Format("-f rawvideo -vcodec rawvideo -s {0}x{1} -pix_fmt rgb24 -r {2} -i - -an -codec:v libx264 -preset veryfast -f mp4 -movflags frag_keyframe+empty_moov -",
            16, 9, 30);
        proc.StartInfo.UseShellExecute = false;
        proc.StartInfo.RedirectStandardInput = true;
        proc.StartInfo.RedirectStandardOutput = true;
    
        FileStream fs = new FileStream(@"out.mp4", FileMode.Create, FileAccess.Write);
    
        proc.Start();
        var readTask = _ConsumeOutputAsync(fs, proc.StandardOutput.BaseStream);
    
        //Generate frames and write to stdin
        for (int i = 0; i < 30 * 60 * 60; i++)
        {
            byte[] myArray = Enumerable.Repeat((byte)Math.Min(i, 255), 9 * 16 * 3).ToArray();
            proc.StandardInput.BaseStream.Write(myArray, 0, myArray.Length);
        }
    
        proc.StandardInput.BaseStream.Close();
    
        readTask.Wait();
        fs.Close();
        Console.WriteLine("Done!");
        Console.ReadKey();
    }
    
    private static async Task _ConsumeOutputAsync(FileStream fs, Stream baseStream)
    {
        int cb;
        byte[] rgb = new byte[4096];
    
        while ((cb = await baseStream.ReadAsync(rgb, 0, rgb.Length)) > 0)
        {
            fs.Write(rgb, 0, cb);
        }
    }
    

    请注意,即使您动态生成数据,您的示例也会显示您将输出写入文件。如果你真的想要输出文件,那么我肯定使用 outfile 命令行参数来指定输出文件,而不是自己重定向StandardOutput。当您运行的外部进程将为您处理所有这些工作时,为什么要在处理数据时注入自己的代码?

    我仍然不知道你要对AutoResetEvent对象做什么,因为你从不等待对象,并且你的实现被打破了,因为你在你到达之前就把对象丢弃了使用它。因此,上述修订示例的任何内容都不会尝试复制该行为。没有它,它工作正常。