mirror of
https://git.freebsd.org/ports.git
synced 2025-05-13 07:41:50 -04:00
02966fbc51
Worked out over BPO-40422 and BPO-40423, this is the culmination of months of work to coordinate with Linux and get close_range(2) added to FreeBSD, then the usage accepted into CPython. It has landed for Python 3.10 and here I've backported it locally to all the supported Python 3 versions we have. Note that this does include and supercede our previous closefrom(2) patches. There was a lot of intersection between the work done, so this patch against the ports tree does remove those patches from each of the ports in favor of this patch. All the patches involved have been accepted and merged upstream. This patch will bring a performance boost in some more situations on 12.2 and 13.0, as close_range exists there. There is one additional patch sitting in an upstream PR that shuffles the _Py_closerange implementation into a different file -- this is not important for the backport, and the absence of that patch here will not realistically cause any issues. PR: 250322 Approved by: lwhsu (python)
199 lines
6.3 KiB
Text
199 lines
6.3 KiB
Text
diff --git a/Modules/_posixsubprocess.c b/Modules/_posixsubprocess.c
|
|
index e693e53206..a26ef96e91 100644
|
|
--- Modules/_posixsubprocess.c
|
|
+++ Modules/_posixsubprocess.c
|
|
@@ -21,6 +21,8 @@
|
|
#include <dirent.h>
|
|
#endif
|
|
|
|
+#include "posixmodule.h"
|
|
+
|
|
#ifdef _Py_MEMORY_SANITIZER
|
|
# include <sanitizer/msan_interface.h>
|
|
#endif
|
|
@@ -222,7 +222,6 @@ _close_fds_by_brute_force(long start_fd, PyObject *py_fds_to_keep)
|
|
long end_fd = safe_get_max_fd();
|
|
Py_ssize_t num_fds_to_keep = PyTuple_GET_SIZE(py_fds_to_keep);
|
|
Py_ssize_t keep_seq_idx;
|
|
- int fd_num;
|
|
/* As py_fds_to_keep is sorted we can loop through the list closing
|
|
* fds in between any in the keep list falling within our range. */
|
|
for (keep_seq_idx = 0; keep_seq_idx < num_fds_to_keep; ++keep_seq_idx) {
|
|
@@ -230,15 +229,11 @@ _close_fds_by_brute_force(long start_fd, PyObject *py_fds_to_keep)
|
|
int keep_fd = PyLong_AsLong(py_keep_fd);
|
|
if (keep_fd < start_fd)
|
|
continue;
|
|
- for (fd_num = start_fd; fd_num < keep_fd; ++fd_num) {
|
|
- close(fd_num);
|
|
- }
|
|
+ _Py_closerange(start_fd, keep_fd - 1);
|
|
start_fd = keep_fd + 1;
|
|
}
|
|
if (start_fd <= end_fd) {
|
|
- for (fd_num = start_fd; fd_num < end_fd; ++fd_num) {
|
|
- close(fd_num);
|
|
- }
|
|
+ _Py_closerange(start_fd, end_fd);
|
|
}
|
|
}
|
|
|
|
diff --git a/Modules/posixmodule.c b/Modules/posixmodule.c
|
|
index 726e3723f9..7ef2dbc854 100644
|
|
--- Modules/posixmodule.c
|
|
+++ Modules/posixmodule.c
|
|
@@ -8555,8 +8555,26 @@ os_close_impl(PyObject *module, int fd)
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
+/* Our selection logic for which function to use is as follows:
|
|
+ * 1. If close_range(2) is available, always prefer that; it's better for
|
|
+ * contiguous ranges like this than fdwalk(3) which entails iterating over
|
|
+ * the entire fd space and simply doing nothing for those outside the range.
|
|
+ * 2. If closefrom(2) is available, we'll attempt to use that next if we're
|
|
+ * closing up to sysconf(_SC_OPEN_MAX).
|
|
+ * 2a. Fallback to fdwalk(3) if we're not closing up to sysconf(_SC_OPEN_MAX),
|
|
+ * as that will be more performant if the range happens to have any chunk of
|
|
+ * non-opened fd in the middle.
|
|
+ * 2b. If fdwalk(3) isn't available, just do a plain close(2) loop.
|
|
+ */
|
|
+#ifdef __FreeBSD__
|
|
+#define USE_CLOSEFROM
|
|
+#endif /* __FreeBSD__ */
|
|
|
|
#ifdef HAVE_FDWALK
|
|
+#define USE_FDWALK
|
|
+#endif /* HAVE_FDWALK */
|
|
+
|
|
+#ifdef USE_FDWALK
|
|
static int
|
|
_fdwalk_close_func(void *lohi, int fd)
|
|
{
|
|
@@ -8569,7 +8587,46 @@ _fdwalk_close_func(void *lohi, int fd)
|
|
close(fd);
|
|
return 0;
|
|
}
|
|
-#endif /* HAVE_FDWALK */
|
|
+#endif /* USE_FDWALK */
|
|
+
|
|
+/* Closes all file descriptors in [first, last], ignoring errors. */
|
|
+void
|
|
+_Py_closerange(int first, int last)
|
|
+{
|
|
+ first = Py_MAX(first, 0);
|
|
+ _Py_BEGIN_SUPPRESS_IPH
|
|
+#ifdef HAVE_CLOSE_RANGE
|
|
+ if (close_range(first, last, 0) == 0 || errno != ENOSYS) {
|
|
+ /* Any errors encountered while closing file descriptors are ignored;
|
|
+ * ENOSYS means no kernel support, though,
|
|
+ * so we'll fallback to the other methods. */
|
|
+ }
|
|
+ else
|
|
+#endif /* HAVE_CLOSE_RANGE */
|
|
+#ifdef USE_CLOSEFROM
|
|
+ if (last >= sysconf(_SC_OPEN_MAX)) {
|
|
+ /* Any errors encountered while closing file descriptors are ignored */
|
|
+ closefrom(first);
|
|
+ }
|
|
+ else
|
|
+#endif /* USE_CLOSEFROM */
|
|
+#ifdef USE_FDWALK
|
|
+ {
|
|
+ int lohi[2];
|
|
+ lohi[0] = first;
|
|
+ lohi[1] = last + 1;
|
|
+ fdwalk(_fdwalk_close_func, lohi);
|
|
+ }
|
|
+#else
|
|
+ {
|
|
+ for (int i = first; i <= last; i++) {
|
|
+ /* Ignore errors */
|
|
+ (void)close(i);
|
|
+ }
|
|
+ }
|
|
+#endif /* USE_FDWALK */
|
|
+ _Py_END_SUPPRESS_IPH
|
|
+}
|
|
|
|
/*[clinic input]
|
|
os.closerange
|
|
@@ -8585,22 +8642,8 @@ static PyObject *
|
|
os_closerange_impl(PyObject *module, int fd_low, int fd_high)
|
|
/*[clinic end generated code: output=0ce5c20fcda681c2 input=5855a3d053ebd4ec]*/
|
|
{
|
|
-#ifdef HAVE_FDWALK
|
|
- int lohi[2];
|
|
-#else
|
|
- int i;
|
|
-#endif
|
|
Py_BEGIN_ALLOW_THREADS
|
|
- _Py_BEGIN_SUPPRESS_IPH
|
|
-#ifdef HAVE_FDWALK
|
|
- lohi[0] = Py_MAX(fd_low, 0);
|
|
- lohi[1] = fd_high;
|
|
- fdwalk(_fdwalk_close_func, lohi);
|
|
-#else
|
|
- for (i = Py_MAX(fd_low, 0); i < fd_high; i++)
|
|
- close(i);
|
|
-#endif
|
|
- _Py_END_SUPPRESS_IPH
|
|
+ _Py_closerange(fd_low, fd_high - 1);
|
|
Py_END_ALLOW_THREADS
|
|
Py_RETURN_NONE;
|
|
}
|
|
diff --git a/Modules/posixmodule.h b/Modules/posixmodule.h
|
|
index 1e00562abc..749833f71c 100644
|
|
--- Modules/posixmodule.h
|
|
+++ Modules/posixmodule.h
|
|
@@ -28,6 +28,8 @@ PyAPI_FUNC(int) _Py_Sigset_Converter(PyObject *, void *);
|
|
#endif /* HAVE_SIGSET_T */
|
|
#endif /* Py_LIMITED_API */
|
|
|
|
+PyAPI_FUNC(void) _Py_closerange(int first, int last);
|
|
+
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
diff --git a/configure b/configure
|
|
index 96dcd0dcd5..90b61f9f8f 100755
|
|
--- configure
|
|
+++ configure
|
|
@@ -11503,8 +11503,8 @@ fi
|
|
|
|
# checks for library functions
|
|
for ac_func in alarm accept4 setitimer getitimer bind_textdomain_codeset chown \
|
|
- clock confstr copy_file_range ctermid dup3 execv explicit_bzero explicit_memset \
|
|
- faccessat fchmod fchmodat fchown fchownat \
|
|
+ clock confstr close_range copy_file_range ctermid dup3 execv explicit_bzero \
|
|
+ explicit_memset faccessat fchmod fchmodat fchown fchownat \
|
|
fdwalk fexecve fdopendir fork fpathconf fstatat ftime ftruncate futimesat \
|
|
futimens futimes gai_strerror getentropy \
|
|
getgrgid_r getgrnam_r \
|
|
diff --git a/configure.ac b/configure.ac
|
|
index 18a044629a..fbb051d49c 100644
|
|
--- configure.ac
|
|
+++ configure.ac
|
|
@@ -3560,8 +3560,8 @@ fi
|
|
|
|
# checks for library functions
|
|
AC_CHECK_FUNCS(alarm accept4 setitimer getitimer bind_textdomain_codeset chown \
|
|
- clock confstr copy_file_range ctermid dup3 execv explicit_bzero explicit_memset \
|
|
- faccessat fchmod fchmodat fchown fchownat \
|
|
+ clock confstr close_range copy_file_range ctermid dup3 execv explicit_bzero \
|
|
+ explicit_memset faccessat fchmod fchmodat fchown fchownat \
|
|
fdwalk fexecve fdopendir fork fpathconf fstatat ftime ftruncate futimesat \
|
|
futimens futimes gai_strerror getentropy \
|
|
getgrgid_r getgrnam_r \
|
|
diff --git a/pyconfig.h.in b/pyconfig.h.in
|
|
index 4263a712c3..166d278f7c 100644
|
|
--- pyconfig.h.in
|
|
+++ pyconfig.h.in
|
|
@@ -136,6 +136,9 @@
|
|
/* Define to 1 if you have the `clock_settime' function. */
|
|
#undef HAVE_CLOCK_SETTIME
|
|
|
|
+/* Define to 1 if you have the `close_range' function. */
|
|
+#undef HAVE_CLOSE_RANGE
|
|
+
|
|
/* Define if the C compiler supports computed gotos. */
|
|
#undef HAVE_COMPUTED_GOTOS
|
|
|