From: Neil Schemenauer <nascheme@enme.ucalgary.ca>
Date: Sun, 24 Mar 2002 22:21:48 +0000 (+0000)
Subject: If possible, set FD_CLOEXEC flag on file descriptors opened using
X-Git-Tag: v2.3c1~6356
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=153cc0fddc33415c35a071d351442e5cb889b5d7;p=python

If possible, set FD_CLOEXEC flag on file descriptors opened using
TemporaryFile.  This flag causes the fd to be closed on exec().
---

diff --git a/Lib/tempfile.py b/Lib/tempfile.py
index b981084b8b..74922405a1 100644
--- a/Lib/tempfile.py
+++ b/Lib/tempfile.py
@@ -180,6 +180,17 @@ class TemporaryFileWrapper:
             setattr(self, name, a)
         return a
 
+try:
+    import fcntl as _fcntl
+    def _set_cloexec(fd, flag=_fcntl.FD_CLOEXEC):
+        flags = _fcntl.fcntl(fd, _fcntl.F_GETFD, 0)
+        if flags >= 0:
+            # flags read successfully, modify
+            flags |= flag
+            _fcntl.fcntl(fd, _fcntl.F_SETFD, flags)
+except (ImportError, AttributeError):
+    def _set_cloexec(fd):
+        pass
 
 def TemporaryFile(mode='w+b', bufsize=-1, suffix=""):
     """Create and return a temporary file (opened read-write by default)."""
@@ -187,6 +198,7 @@ def TemporaryFile(mode='w+b', bufsize=-1, suffix=""):
     if os.name == 'posix':
         # Unix -- be very careful
         fd = os.open(name, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0700)
+        _set_cloexec(fd)
         try:
             os.unlink(name)
             return os.fdopen(fd, mode, bufsize)