Python中使用FD_CLOEXEC标志关闭文件描述符的手动和自动方式

在Python中，可以使用fcntl模块来设置FD_CLOEXEC标志以关闭文件描述符的手动方式。FD_CLOEXEC标志用于在执行exec或者fork时关闭文件描述符，并且不被子进程继承。在下面的例子中，我们将演示如何手动关闭文件描述符。

import os
import fcntl

def close_on_exec(file_descriptor):
    flags = fcntl.fcntl(file_descriptor, fcntl.F_GETFD)
    flags |= fcntl.FD_CLOEXEC
    fcntl.fcntl(file_descriptor, fcntl.F_SETFD, flags)

# 打开一个文件
file_descriptor = os.open("example.txt", os.O_RDWR)

# 手动设置FD_CLOEXEC标志
close_on_exec(file_descriptor)

# 执行某个命令，调用exec函数会导致文件描述符自动关闭
os.execv("/bin/ls", ["ls"])

上述代码中，我们首先打开了一个文件并获得了文件描述符。然后，使用fcntl.fcntl函数获取了当前文件描述符的标志。我们通过按位或操作将FD_CLOEXEC标志添加到标志中。最后，我们使用fcntl.fcntl函数将修改后的标志设置回文件描述符。

在 execv 调用之后，文件描述符将会自动关闭。

另一种自动关闭文件描述符的方式是使用 os.set_inheritable 函数。这个函数允许您设置一个文件描述符的继承属性。

import os

# 打开一个文件
file_descriptor = os.open("example.txt", os.O_RDWR)

# 设置继承属性为False（关闭文件描述符）
os.set_inheritable(file_descriptor, False)

# 执行某个命令，调用exec函数会导致文件描述符自动关闭
os.execv("/bin/ls", ["ls"])

在上述代码中，我们使用os.set_inheritable函数将文件描述符的继承属性设置为False，然后调用execv函数。在execv调用之后，文件描述符将自动关闭。

这两种方式都可以在exec或fork时关闭文件描述符，以防止它们被子进程继承。