3

使用 C#，我想自动化第三方 Windows 命令行程序。通常，它是一个交互式控制台，您发送命令，它可能会提示详细信息，发回结果并显示提示以询问更多命令。通常：

c:\>console_access.exe
Prompt> version
2.03g.2321
Prompt>

我使用了 .NET 类 Process 和 ProcessStartInfo 以及 stdin/stdout/stderr 的重定向。

    public ConsoleAccess()
    {
        if (!File.Exists(consoleAccessPath)) throw new FileNotFoundException(consoleAccessPath + " not found");

        myProcess = new Process();
        ProcessStartInfo myProcessStartInfo = new ProcessStartInfo(consoleAccessPath, ""); // even "2>&1" as argument does not work; my code still hangs
        myProcessStartInfo.CreateNoWindow = true; 
        myProcessStartInfo.UseShellExecute = false; 
        myProcessStartInfo.RedirectStandardOutput = true;
        myProcessStartInfo.RedirectStandardError = true;
        myProcessStartInfo.RedirectStandardInput = true;
        //myProcessStartInfo.ErrorDialog = true; // I tried, to no avail.
        myProcess.StartInfo = myProcessStartInfo;

        outputQueue = new ConcurrentQueue<string>(); // thread-safe queue
        errorQueue = new ConcurrentQueue<string>();

        myProcess.Start();
        myStandardOutput = myProcess.StandardOutput;
        myStandardError = myProcess.StandardError;
        myStandardInput = myProcess.StandardInput;

        stdOutPumper = new Thread(new ThreadStart(PumpStdOutLoop));
        stdOutPumper.Start();
        stdErrPumper = new Thread(new ThreadStart(PumpStdErrLoop));
        stdErrPumper.Start();

        string empty = getResponse(); // check for prompt
        string version = getVersion(); // one simple command
    }
    // [...]
    private void PumpStdErrLoop()
    {
        while (true)
        {
            string message = myStandardError.ReadLine();
            errorQueue.Enqueue(message);
        }
    }

    private void PumpStdOutLoop()
    {
        while (true)
        {
            bool done = false;
            string buffer = "";
            //int blocksize = 1024;
            string prompt = "Prompt> ";
            while (!done)
            {
                //char[] intermediaire = new char[blocksize];
                //int res = myStandardOutput.Read(intermediaire, 0, blocksize);
                //buffer += new string(intermediaire).Substring(0, res);
                byte b = (byte)myStandardOutput.Read(); // I go byte per byte, just in case the char[] above is the source of the problem. To no avail.
                buffer += (char)b;
                done = buffer.EndsWith(prompt);
            }
            buffer = buffer.Substring(0, buffer.Length - prompt.Length);
            outputQueue.Enqueue(buffer);
        }
    }

由于该程序在等待命令时返回“Prompt>”（重要：末尾没有“\n”），所以我不能使用 myProcess.BeginOutputReadLine();

但是，我必须使用线程，因为我必须同时收听 stdout 和 stderr。

这就是为什么我将线程和线程安全队列用于类生产者/消费者模式。

“您可以使用异步读取操作来避免这些依赖关系及其潜在的死锁。或者，您可以通过创建两个线程并在单独的线程上读取每个流的输出来避免死锁情况。” 来源：http: //msdn.microsoft.com/en-us/library/system.diagnostics.process.standardoutput%28v=vs.100%29.aspx

使用这种设计，所有序列如 * cmd -> 结果没有错误（stdout 上的东西，stderr 上没有东西）* cmd -> 错误（stderr 上的东西，stderr 上没有东西）按预期工作。没问题。

  • cmd -> 带有警告的结果（stderr 和 stdout 上的东西）应该可以工作（我正在尝试重现这种情况）

但是，特别是对于一个命令——在执行过程中提示输入密码的命令——不起作用：

  • 主线程主循环永远开启 if (errorQueue.Count == 0 && outputQueue.Count == 0) { System.Threading.Thread.Sleep(500); }
  • 线程泵送标准输出永远等待 byte b = (byte)myStandardOutput.Read();
  • 线程泵送标准输出永远等待一行 string message = myStandardError.ReadLine();

我不明白为什么byte b = (byte)myStandardOutput.Read();不发送消息“密码：”。什么都没发生。我从来没有得到第一个'p'。

我觉得我遇到了僵局，但我不明白为什么。

怎么了？

（我认为这不是很相关，但我在 Windows 7 32 位上使用 MS Visual Studio 2010 在 .NET 4.0 上尝试了上述内容。）

4

1 回答 1

3

对于这类交互式控制台模式程序，这是一种非常常见的故障模式。当 C 运行时库检测到输出被重定向时，它会自动将 stderr 和 stdout 流切换到缓冲模式。对提高吞吐量很重要。因此输出进入该缓冲区，而不是直接写入控制台。让您的程序看到输出需要刷新缓冲区。

缓冲区被刷新的三种情况。当缓冲区已满时发生刷新，通常约为 2 KB。或者当程序写入行终止符 (\n) 时。或者当程序显式调用 fflush() 时。前两种情况没有发生，没有足够的输出并且程序没有使用\n。问题的所在，原来的程序员忘记调用 fflush()。忘记这一点很常见，程序员根本不打算以交互方式使用程序。

对此无能为力，您需要要求程序的所有者或作者添加 fflush()。也许您可以通过假设正在编写提示来跛行。

于 2013-02-26T13:35:01.003 回答