一个非常简单的 python subprocess.Popen 任务的无尽问题

Question

我希望 python 向用 C++ 编写的可执行文件发送大约 50 万个 0-255 范围内的整数。然后这个可执行文件将响应几千个整数。每个在一条线上。这似乎对子流程来说应该非常简单，但我遇到了无穷无尽的麻烦。现在我正在使用代码进行测试：

// main()
u32 num;
std::cin >> num;

u8* data = new u8[num];
for (u32 i = 0; i < num; ++i)
    std::cin >> data[i];

// test output / spit it back out
for (u32 i = 0; i < num; ++i)
    std::cout << data[i] << std::endl;

return 0;

在python中构建一个字符串数组（“data”），每个都像“255\n”，然后使用：

output = proc.communicate("".join(data))[0]

...不起作用（说标准输入已关闭，可能一次太多了）。使用 proc.stdin 和 proc.stdout 都没有工作。这应该非常简单，但是我经常遇到异常，和/或没有输出数据返回给我。我的 Popen 目前是：

proc = Popen('aux/test_cpp_program', stdin=PIPE, stdout=PIPE, bufsize=1)

在我拔头发之前给我建议。 ;)

Answer 1

这对我来说非常适合：

#include <iostream>

int main()
{
    int num;
    std::cin >> num;

    char* data = new char[num];
    for (int i = 0; i < num; ++i)
        std::cin >> data[i];

    // test output / spit it back out
    for (int i = 0; i < num; ++i)
        std::cout << data[i] << std::endl;

    return 0;
}

蟒蛇：

In [9]: from subprocess import Popen, PIPE
In [10]: a = Popen('./a.out', stdin=PIPE, stdout=PIPE)
In [11]: a.stdin.write("2\nab")
In [12]: a.stdout.read()
Out[12]: 'a\nb\n'

请注意，我在要写入的字节数之间添加了一个分隔符 (\n)，如果您不希望您的 c++ cin 在发送类似 3,1,2,3 的内容时失败，这是最安全的，它将连接到3123，期待这么多争论。

Answer 2

在 C++ 中，当您从 cin 读取 char 或 unsigned char 时，它会从 stdin 读取单个字节。但是，您希望它读取从 0 到 255 的数字的十进制表示。如果您读取的是 int，它应该正确读取它：

unsigned int n;
std::cin >> n;
data[i] = static_cast<u8>(n);

或者，您可以让 Python 代码使用 chr 函数将值写入字节序列而不是数字序列。