]> granicus.if.org Git - python/commitdiff
#4841: Fix FileIO constructor to honor closefd when called repeatedly
authorHynek Schlawack <hs@ox.cx>
Fri, 25 May 2012 08:05:53 +0000 (10:05 +0200)
committerHynek Schlawack <hs@ox.cx>
Fri, 25 May 2012 08:05:53 +0000 (10:05 +0200)
Patch by Victor Stinner.

Lib/test/test_io.py
Modules/_io/fileio.c

index ea82cea17ae688570c299ad9b62539c419430d59..95ecd90a8ecf8fcceb510bd767e4d1d1b656cc49 100644 (file)
@@ -634,6 +634,19 @@ class IOTest(unittest.TestCase):
         for obj in test:
             self.assertTrue(hasattr(obj, "__dict__"))
 
+    def test_fileio_closefd(self):
+        # Issue #4841
+        with self.open(__file__, 'rb') as f1, \
+             self.open(__file__, 'rb') as f2:
+            fileio = self.FileIO(f1.fileno(), closefd=False)
+            # .__init__() must not close f1
+            fileio.__init__(f2.fileno(), closefd=False)
+            f1.readline()
+            # .close() must not close f2
+            fileio.close()
+            f2.readline()
+
+
 class CIOTest(IOTest):
 
     def test_IOBase_finalize(self):
index d5b03eec26aa0d676c9301401c76d60ed1ee6a5c..c51d69798bdbf073b0c0bf7daefcdb09a4a58ad0 100644 (file)
@@ -227,9 +227,13 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
 
     assert(PyFileIO_Check(oself));
     if (self->fd >= 0) {
-        /* Have to close the existing file first. */
-        if (internal_close(self) < 0)
-            return -1;
+        if (self->closefd) {
+            /* Have to close the existing file first. */
+            if (internal_close(self) < 0)
+                return -1;
+        }
+        else
+            self->fd = -1;
     }
 
     if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|si:fileio",