3

我正在编写一个从 STDIN 上的匿名管道读取的 C++ (Windows) 客户端控制台应用程序。我希望能够按如下方式使用我的程序:

echo input text here | my_app.exe

并使用管道输入的文本在应用程序中执行某些操作

或者

my_app.exe

然后在应用程序内部使用一些默认文本,而不是来自管道的输入。

鉴于第一种情况,我目前有成功从 STDIN 上的管道读取的代码:

#include <Windows.h>
#include <iostream>
#include <string>

#define BUFSIZE 4096

int main(int argc, const char *argv[]) {
    char char_buffer[BUFSIZE]; 
    DWORD bytes_read;
    HANDLE stdin_handle;
    BOOL continue_reading;
    unsigned int required_size;
    bool read_successful = true;

    stdin_handle = GetStdHandle(STD_INPUT_HANDLE);

    if (stdin_handle == INVALID_HANDLE_VALUE) {
        std::cout << "Error: invalid handle value!\n\n";
    } else {
        continue_reading = true;

        while (continue_reading) { 
            continue_reading = ReadFile(stdin_handle, char_buffer, BUFSIZE,
                &bytes_read, NULL); 

            if (continue_reading) {
                if (bytes_read != 0) {
                    // Output what we have read so far
                    for (unsigned int i = 0; i < bytes_read; i++) {
                        std::cout << char_buffer[i];
                    }
                } else {
                    continue_reading = false;
                }
            }
        }
    }

    return 0;
}

我知道我对匿名管道的唯一选择是使用 ReadFile 进行阻塞读取。如果我理解正确,关于我如何调用它,ReadFile 将继续从 STDIN 上的缓冲区读取,直到它检测到管道另一端的写操作结束(也许读取某种“写结束”令牌??)。我想知道是否有某种“开始写入”令牌将在缓冲区中,如果正在传输某些东西,我可以在调用 ReadFile之前检查 STDIN。如果是这种情况,我可以跳过调用 ReadFile 并使用一些默认文本。

如果没有办法做到这一点,我总是可以传入一个命令行参数,表示我不应该检查管道而只使用默认文本(或相反),但我更愿意这样做我指定的方式。

4

3 回答 3

5

PeekNamedPipe()。尽管它的名字,它适用于命名和匿名管道。

int main(int argc, const char *argv[])
{
    char char_buffer[BUFSIZE]; 
    DWORD bytes_read;
    DWORD bytes_avail;
    DWORD dw;
    HANDLE stdin_handle;
    bool is_pipe;

    stdin_handle = GetStdHandle(STD_INPUT_HANDLE);
    is_pipe = !GetConsoleMode(stdin_handle, &dw);

    if (stdin_handle == INVALID_HANDLE_VALUE) {
        std::cout << "Error: invalid handle value!\n\n";
    } else {
        while (1) { 
            if (is_pipe) {
                if (PeekNamedPipe(stdin_handle, NULL, 0, NULL, &bytes_avail, NULL)) {
                    if (bytes_avail == 0) {
                        Sleep(100);
                        continue;
                    }
                }
            }

            if (!ReadFile(stdin_handle, char_buffer, min(bytes_avail, BUFSIZE), &bytes_read, NULL)) {
                break; 
            }

            if (bytes_read == 0) {
                break;
            }

            // Output what we have read so far
            for (unsigned int i = 0; i < bytes_read; i++) {
                std::cout << char_buffer[i];
            }
        }
    }

    return 0;
}
于 2013-01-24T01:16:41.203 回答
3

看起来您在这里真正想要做的是确定您是否有控制台输入(使用默认值)与管道输入(使用管道输入)。

建议直接测试而不是尝试检查是否有输入准备好:尝试嗅探管道中是否有数据的问题是,如果源应用程序生成输出的速度很慢,那么您的应用程序可能会做出不正确的假设,因为没有t 输入尚未可用。(也有可能,由于提前输入,在您的应用程序开始检查输入是否可用之前,用户可能已经输入了准备从控制台 STDIN 读取的区域的字符。)

另外,请记住,允许您的应用程序与文件重定向一起使用可能很有用,而不仅仅是管道 - 例如:

myapp.exe < some_input_file

在 unix 上执行此“交互模式,与使用重定向输入”测试的经典方法是使用 isatty(); 幸运的是,在 Windows CRT 中有一个等价物 - 请参阅函数 _isatty();或使用GetFileType()检查 GetStdHandle(STD_INPUT_HANDLE) 上的 FILE_TYPE_CHAR - 或像 Remy 那样使用说 GetConsoleMode,这只会在真正的控制台句柄上成功。

于 2013-01-24T08:25:59.240 回答
0

这在使用第二个线程时也可以在没有重叠 I/O 的情况下工作,该线程执行同步 ReadFile 调用。然后主线程等待任意时间并像上面一样行事......

希望这可以帮助...

于 2013-01-23T23:44:59.720 回答