python - 如何使用 os.execv() 在 python 中继承 stdin 和 stdout

Question

首先我写了一段c++代码如下:

#include <cstdio>
int main()
{
    int a,b;
    while(scanf("%d %d",&a,&b) == 2)
        printf("%d\n",a+b);
    return 0;
}

我使用 g++ -o a.cpp 来编译它。

之后，我写了如下python代码:

import os,sys
sys.stdin = open("./data.in","r")
sys.stdout = open("./data.out","w")
pid = os.fork()
if pid == 0:
    cmd = ["./a","./a"]
    os.execv(cmd[0],cmd)

但是，data.out 文件不包含任何内容。也就是说，子进程并没有继承父进程的stdin和stdout。但是当我写了一个c++代码如下:

#include<unistd.h>
#include<cstdio>
int main()
{
    freopen("data.in","r",stdin);a
    freopen("data.out","w",stdout);
    int pid = fork();
    if(pid == 0)
    {
        char* cmd[]= {"./a","./a"};
        execv(cmd[0],cmd);
    }
    return 0;
}

我在 data.out 中得到了正确的答案，也就是说 execv 在 c++ 代码中工作。

那么，我应该怎么做才能让 execv 在 python 中也能正常工作？我真的需要这个功能才能工作，有人可以告诉我吗？非常感谢!

data.in 包含以下内容:

1 1

Answer 1

在您的 Python 代码中，您打开了新的 IO 流并设置了 sys.stdin 和 sys.stdout Python 名称 指向这些。他们将有自己的文件描述符(很可能是 3 和 4)。但是，标准文件描述符 0 和 1 将保持不变，并继续指向它们的原始位置。这些是当您在 C++ 程序中 exec 到 stdin 和 stdout 时继承的描述符。

您需要确保在执行之前调整了实际的文件描述符。

os.dup2(sys.stdin.fileno(), 0)
os.dup2(sys.stdout.fileno(), 1)