2

我正在尝试在与主线程不同的线程中从标准输入读取一些数据。主线程应该能够通过写入标准输入与这个等待线程进行通信,但是当我运行测试代码(包括在下面)时,除了消息(我的测试代码中的'do_some_work')直接打印在终端上而不是从等待线程输出。

我尝试了 SO 上列出的几个解决方案,但没有成功。我的代码模仿了以下 SO question 的解决方案之一,它本身工作得很好,但是当与我的 read_stdin_thread 结合使用时,它就不行了。

是否可以在 Linux 中将数据写入自己的标准输入

#include <unistd.h>
#include <string>
#include <iostream>
#include <sstream>
#include <thread>

bool terminate_read = true;

void readStdin() {

    static const int INPUT_BUF_SIZE = 1024;
    char buf[INPUT_BUF_SIZE];

    while (terminate_read) {
        fd_set readfds;
        struct timeval tv;
        int data;

        FD_ZERO(&readfds);
        FD_SET(STDIN_FILENO, &readfds);
        tv.tv_sec=2;
        tv.tv_usec=0;
        int ret = select(16, &readfds, 0, 0, &tv);
        if (ret == 0) {
            continue;
        } else if (ret == -1) {
            perror("select");
            continue;
        }
        data=FD_ISSET(STDIN_FILENO, &readfds);
        if (data>0) {
            int bytes = read(STDIN_FILENO,buf,INPUT_BUF_SIZE);
            if (bytes == -1) {
                perror("input poll: read");
                continue;
            }
            if (bytes) {
                std::cout << "Execute: " << buf << std::endl;
                if (strncmp(buf, "quit", 4)==0) {
                    std::cout << "quitting reading from stdin." << std::endl;
                    break;
                }
                else {
                    continue;
                }
            }
        }
    }
}

int main() {
    std::thread threadReadStdin([] () {
        readStdin();
    });

    usleep(1000000);
    std::stringstream msg;
    msg << "do_some_work" << std::endl;
    auto s = msg.str();
    write(STDIN_FILENO, s.c_str(), s.size());
    usleep(1000000);

    terminate_read = false;
    threadReadStdin.join();

    return 0;
}

说明如何写入 stdin 的代码片段将由 threadReadStdin 读取,这将非常有帮助。

提前非常感谢!

编辑:

我在这里忘记提到的一件事是 readStdin() 中的代码是第三方代码,并且发生的任何类型的通信都必须符合其条款。

此外,我很容易将 std::cin 和 std::cout 重定向到 fstream 或 stringstream。问题是,当我写入重定向的 cin 缓冲区时,读取线程上并没有真正出现。

编辑2:

这是一个单进程应用程序,不能选择生成。

4

2 回答 2

1

在@Jorge Perez、@Remy Lebeau 和@Kamil Cuk 的非常有建设性的回应的帮助下,我似乎偶然发现了答案。该解决方案基于@Jorge Perez 非常有用的代码。为简洁起见,我没有包括整个代码,但部分来自我发布的代码,很大一部分来自@Jorge Perez 的代码。

我所做的是采用他的方法使用管道并用管道替换 STDIN_FILENO 使用 dup 读取 fd。以下链接真的很有帮助:

https://en.wikipedia.org/wiki/Dup_(system_call)

考虑到我在生产环境代码中的限制,我非常感谢您就这是否是一个黑客或足够好的方法/解决方案提供意见。

int main() {
    int read_fd;
    int write_fd;

    open_channel(read_fd, write_fd); 

    close(STDIN_FILENO);
    if(dup(read_fd) == -1)
        return -1;

    std::thread write_thread(write_message, write_fd, "Whatsup?"); 
    std::thread threadReadStdin([] () {
        readStdin();
    });

    write_thread.join(); 
    threadReadStdin.join();

    return 0;
}
于 2019-04-10T11:36:24.343 回答
1

如果您想使用管道在同一程序中的不同线程之间进行通信,则不应尝试使用stdinor stdout。相反,只需使用该pipe函数来创建您自己的管道。我将逐步指导您完成此操作!

开通渠道

让我们创建一个辅助函数来使用pipe. 此函数通过引用获取两个整数 - 读取端和写入端。它尝试打开管道,如果不能,则打印错误。

#include <unistd.h>
#include <cstdio>
#include <thread>
#include <string>

void open_channel(int& read_fd, int& write_fd) {
    int vals[2];
    int errc = pipe(vals); 
    if(errc) {
        fputs("Bad pipe", stderr); 
        read_fd = -1;
        write_fd = -1; 
    } else {
        read_fd = vals[0];
        write_fd = vals[1]; 
    }
}

写消息

接下来,我们定义一个函数来编写消息。该函数以 lambda 形式给出,因此我们可以将其直接传递给线程。

auto write_message = [](int write_fd, std::string message) {
    ssize_t amnt_written = write(write_fd, message.data(), message.size());
    if(amnt_written != message.size()) {
        fputs("Bad write", stderr); 
    }
    close(write_fd); 
}; 

阅读消息

我们还应该创建一个函数来读取消息。阅读消息将在不同的线程上完成。此 lambda 以一种类型读取消息1000字节,并将其打印到标准输出。

auto read_message = [](int read_fd) {
    constexpr int buffer_size = 1000; 
    char buffer[buffer_size + 1]; 
    ssize_t amnt_read; 
    do {
        amnt_read = read(read_fd, &buffer[0], buffer_size);
        buffer[amnt_read] = 0; 
        fwrite(buffer, 1, amnt_read, stdout); 
    } while(amnt_read > 0); 
};

主要方法

最后,我们可以编写main方法。它打开通道,在一个线程上写入消息,并在另一个线程上读取它。

int main() {
    int read_fd;
    int write_fd;
    open_channel(read_fd, write_fd); 

    std::thread write_thread(
        write_message, write_fd, "Hello, world!"); 
    std::thread read_thread(
        read_message, read_fd); 
    write_thread.join(); 
    read_thread.join(); 
}
于 2019-04-09T23:31:08.147 回答