mirror of https://github.com/python/cpython
Issue #18763: subprocess: The file descriptors are now closed after calling the
preexec_fn callback, which may open file descriptors.
This commit is contained in:
commit
d317f4c84e
|
@ -1972,6 +1972,23 @@ class POSIXProcessTestCase(BaseTestCase):
|
|||
self.assertRaises(OSError, os.waitpid, pid, 0)
|
||||
self.assertNotIn(ident, [id(o) for o in subprocess._active])
|
||||
|
||||
def test_close_fds_after_preexec(self):
|
||||
fd_status = support.findfile("fd_status.py", subdir="subprocessdata")
|
||||
|
||||
# this FD is used as dup2() target by preexec_fn, and should be closed
|
||||
# in the child process
|
||||
fd = os.dup(1)
|
||||
self.addCleanup(os.close, fd)
|
||||
|
||||
p = subprocess.Popen([sys.executable, fd_status],
|
||||
stdout=subprocess.PIPE, close_fds=True,
|
||||
preexec_fn=lambda: os.dup2(1, fd))
|
||||
output, ignored = p.communicate()
|
||||
|
||||
remaining_fds = set(map(int, output.split(b',')))
|
||||
|
||||
self.assertNotIn(fd, remaining_fds)
|
||||
|
||||
|
||||
@unittest.skipUnless(mswindows, "Windows specific tests")
|
||||
class Win32ProcessTestCase(BaseTestCase):
|
||||
|
|
|
@ -412,17 +412,6 @@ child_exec(char *const exec_array[],
|
|||
POSIX_CALL(close(errwrite));
|
||||
}
|
||||
|
||||
if (close_fds) {
|
||||
int local_max_fd = max_fd;
|
||||
#if defined(__NetBSD__)
|
||||
local_max_fd = fcntl(0, F_MAXFD);
|
||||
if (local_max_fd < 0)
|
||||
local_max_fd = max_fd;
|
||||
#endif
|
||||
/* TODO HP-UX could use pstat_getproc() if anyone cares about it. */
|
||||
_close_open_fd_range(3, local_max_fd, py_fds_to_keep);
|
||||
}
|
||||
|
||||
if (cwd)
|
||||
POSIX_CALL(chdir(cwd));
|
||||
|
||||
|
@ -451,6 +440,18 @@ child_exec(char *const exec_array[],
|
|||
/* Py_DECREF(result); - We're about to exec so why bother? */
|
||||
}
|
||||
|
||||
/* close FDs after executing preexec_fn, which might open FDs */
|
||||
if (close_fds) {
|
||||
int local_max_fd = max_fd;
|
||||
#if defined(__NetBSD__)
|
||||
local_max_fd = fcntl(0, F_MAXFD);
|
||||
if (local_max_fd < 0)
|
||||
local_max_fd = max_fd;
|
||||
#endif
|
||||
/* TODO HP-UX could use pstat_getproc() if anyone cares about it. */
|
||||
_close_open_fd_range(3, local_max_fd, py_fds_to_keep);
|
||||
}
|
||||
|
||||
/* This loop matches the Lib/os.py _execvpe()'s PATH search when */
|
||||
/* given the executable_list generated by Lib/subprocess.py. */
|
||||
saved_errno = 0;
|
||||
|
|
Loading…
Reference in New Issue