bpo-32270: Don't close stdin/out/err in pass_fds (GH-6242)
When subprocess.Popen() stdin= stdout= or stderr= handles are specified
and appear in pass_fds=, don't close the original fds after dup'ing them.
This implementation and unittest primarily came from @izbyshev (see the PR)
See also b89b52f284
This also removes the old manual p2cread, c2pwrite, and errwrite closing logic
as inheritable flags and _close_open_fds takes care of that properly today without special treatment.
This code is within child_exec() where it is the only thread so there is no
race condition between the dup and _Py_set_inheritable_async_safe call.
This commit is contained in:
parent
880d42a3b2
commit
ce34410b8b
|
@ -2529,6 +2529,36 @@ class POSIXProcessTestCase(BaseTestCase):
|
||||||
self.assertEqual(os.get_inheritable(inheritable), True)
|
self.assertEqual(os.get_inheritable(inheritable), True)
|
||||||
self.assertEqual(os.get_inheritable(non_inheritable), False)
|
self.assertEqual(os.get_inheritable(non_inheritable), False)
|
||||||
|
|
||||||
|
|
||||||
|
# bpo-32270: Ensure that descriptors specified in pass_fds
|
||||||
|
# are inherited even if they are used in redirections.
|
||||||
|
# Contributed by @izbyshev.
|
||||||
|
def test_pass_fds_redirected(self):
|
||||||
|
"""Regression test for https://bugs.python.org/issue32270."""
|
||||||
|
fd_status = support.findfile("fd_status.py", subdir="subprocessdata")
|
||||||
|
pass_fds = []
|
||||||
|
for _ in range(2):
|
||||||
|
fd = os.open(os.devnull, os.O_RDWR)
|
||||||
|
self.addCleanup(os.close, fd)
|
||||||
|
pass_fds.append(fd)
|
||||||
|
|
||||||
|
stdout_r, stdout_w = os.pipe()
|
||||||
|
self.addCleanup(os.close, stdout_r)
|
||||||
|
self.addCleanup(os.close, stdout_w)
|
||||||
|
pass_fds.insert(1, stdout_w)
|
||||||
|
|
||||||
|
with subprocess.Popen([sys.executable, fd_status],
|
||||||
|
stdin=pass_fds[0],
|
||||||
|
stdout=pass_fds[1],
|
||||||
|
stderr=pass_fds[2],
|
||||||
|
close_fds=True,
|
||||||
|
pass_fds=pass_fds):
|
||||||
|
output = os.read(stdout_r, 1024)
|
||||||
|
fds = {int(num) for num in output.split(b',')}
|
||||||
|
|
||||||
|
self.assertEqual(fds, {0, 1, 2} | frozenset(pass_fds), f"output={output!a}")
|
||||||
|
|
||||||
|
|
||||||
def test_stdout_stdin_are_single_inout_fd(self):
|
def test_stdout_stdin_are_single_inout_fd(self):
|
||||||
with io.open(os.devnull, "r+") as inout:
|
with io.open(os.devnull, "r+") as inout:
|
||||||
p = subprocess.Popen([sys.executable, "-c", "import sys; sys.exit(0)"],
|
p = subprocess.Popen([sys.executable, "-c", "import sys; sys.exit(0)"],
|
||||||
|
|
|
@ -0,0 +1,2 @@
|
||||||
|
The subprocess module no longer mistakenly closes redirected fds even when
|
||||||
|
they were in pass_fds when outside of the default {0, 1, 2} set.
|
|
@ -422,10 +422,20 @@ child_exec(char *const exec_array[],
|
||||||
|
|
||||||
/* When duping fds, if there arises a situation where one of the fds is
|
/* When duping fds, if there arises a situation where one of the fds is
|
||||||
either 0, 1 or 2, it is possible that it is overwritten (#12607). */
|
either 0, 1 or 2, it is possible that it is overwritten (#12607). */
|
||||||
if (c2pwrite == 0)
|
if (c2pwrite == 0) {
|
||||||
POSIX_CALL(c2pwrite = dup(c2pwrite));
|
POSIX_CALL(c2pwrite = dup(c2pwrite));
|
||||||
while (errwrite == 0 || errwrite == 1)
|
/* issue32270 */
|
||||||
|
if (_Py_set_inheritable_async_safe(c2pwrite, 0, NULL) < 0) {
|
||||||
|
goto error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
while (errwrite == 0 || errwrite == 1) {
|
||||||
POSIX_CALL(errwrite = dup(errwrite));
|
POSIX_CALL(errwrite = dup(errwrite));
|
||||||
|
/* issue32270 */
|
||||||
|
if (_Py_set_inheritable_async_safe(errwrite, 0, NULL) < 0) {
|
||||||
|
goto error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/* Dup fds for child.
|
/* Dup fds for child.
|
||||||
dup2() removes the CLOEXEC flag but we must do it ourselves if dup2()
|
dup2() removes the CLOEXEC flag but we must do it ourselves if dup2()
|
||||||
|
@ -451,14 +461,8 @@ child_exec(char *const exec_array[],
|
||||||
else if (errwrite != -1)
|
else if (errwrite != -1)
|
||||||
POSIX_CALL(dup2(errwrite, 2)); /* stderr */
|
POSIX_CALL(dup2(errwrite, 2)); /* stderr */
|
||||||
|
|
||||||
/* Close pipe fds. Make sure we don't close the same fd more than */
|
/* We no longer manually close p2cread, c2pwrite, and errwrite here as
|
||||||
/* once, or standard fds. */
|
* _close_open_fds takes care when it is not already non-inheritable. */
|
||||||
if (p2cread > 2)
|
|
||||||
POSIX_CALL(close(p2cread));
|
|
||||||
if (c2pwrite > 2 && c2pwrite != p2cread)
|
|
||||||
POSIX_CALL(close(c2pwrite));
|
|
||||||
if (errwrite != c2pwrite && errwrite != p2cread && errwrite > 2)
|
|
||||||
POSIX_CALL(close(errwrite));
|
|
||||||
|
|
||||||
if (cwd)
|
if (cwd)
|
||||||
POSIX_CALL(chdir(cwd));
|
POSIX_CALL(chdir(cwd));
|
||||||
|
|
Loading…
Reference in New Issue