mirror of https://github.com/python/cpython
#10053: Don't close FDs when FileIO.__init__ fails
Loosely based on the work by Hirokazu Yamamoto.
This commit is contained in:
parent
4215d2738a
commit
9bd4bf2a3d
|
@ -421,6 +421,17 @@ class OtherFileTests(unittest.TestCase):
|
||||||
'IOError: [Errno 2] No such file or directory' not in out):
|
'IOError: [Errno 2] No such file or directory' not in out):
|
||||||
self.fail('Bad output: %r' % out)
|
self.fail('Bad output: %r' % out)
|
||||||
|
|
||||||
|
def testUnclosedFDOnException(self):
|
||||||
|
class MyException(Exception): pass
|
||||||
|
class MyFileIO(_FileIO):
|
||||||
|
def __setattr__(self, name, value):
|
||||||
|
if name == "name":
|
||||||
|
raise MyException("blocked setting name")
|
||||||
|
return super(MyFileIO, self).__setattr__(name, value)
|
||||||
|
fd = os.open(__file__, os.O_RDONLY)
|
||||||
|
self.assertRaises(MyException, MyFileIO, fd)
|
||||||
|
os.close(fd) # should not raise OSError(EBADF)
|
||||||
|
|
||||||
def test_main():
|
def test_main():
|
||||||
# Historically, these tests have been sloppy about removing TESTFN.
|
# Historically, these tests have been sloppy about removing TESTFN.
|
||||||
# So get rid of it no matter what.
|
# So get rid of it no matter what.
|
||||||
|
|
|
@ -9,6 +9,9 @@ What's New in Python 2.7.4
|
||||||
Core and Builtins
|
Core and Builtins
|
||||||
-----------------
|
-----------------
|
||||||
|
|
||||||
|
- Issue #10053: Don't close FDs when FileIO.__init__ fails. Loosely based on
|
||||||
|
the work by Hirokazu Yamamoto.
|
||||||
|
|
||||||
- Issue #14775: Fix a potential quadratic dict build-up due to the garbage
|
- Issue #14775: Fix a potential quadratic dict build-up due to the garbage
|
||||||
collector repeatedly trying to untrack dicts.
|
collector repeatedly trying to untrack dicts.
|
||||||
|
|
||||||
|
|
|
@ -195,6 +195,7 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
|
||||||
int flags = 0;
|
int flags = 0;
|
||||||
int fd = -1;
|
int fd = -1;
|
||||||
int closefd = 1;
|
int closefd = 1;
|
||||||
|
int fd_is_own = 0;
|
||||||
|
|
||||||
assert(PyFileIO_Check(oself));
|
assert(PyFileIO_Check(oself));
|
||||||
if (self->fd >= 0) {
|
if (self->fd >= 0) {
|
||||||
|
@ -345,6 +346,7 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
|
||||||
#endif
|
#endif
|
||||||
self->fd = open(name, flags, 0666);
|
self->fd = open(name, flags, 0666);
|
||||||
Py_END_ALLOW_THREADS
|
Py_END_ALLOW_THREADS
|
||||||
|
fd_is_own = 1;
|
||||||
if (self->fd < 0) {
|
if (self->fd < 0) {
|
||||||
#ifdef MS_WINDOWS
|
#ifdef MS_WINDOWS
|
||||||
if (widename != NULL)
|
if (widename != NULL)
|
||||||
|
@ -366,19 +368,17 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
|
||||||
end of file (otherwise, it might be done only on the
|
end of file (otherwise, it might be done only on the
|
||||||
first write()). */
|
first write()). */
|
||||||
PyObject *pos = portable_lseek(self->fd, NULL, 2);
|
PyObject *pos = portable_lseek(self->fd, NULL, 2);
|
||||||
if (pos == NULL) {
|
if (pos == NULL)
|
||||||
if (closefd) {
|
|
||||||
close(self->fd);
|
|
||||||
self->fd = -1;
|
|
||||||
}
|
|
||||||
goto error;
|
goto error;
|
||||||
}
|
|
||||||
Py_DECREF(pos);
|
Py_DECREF(pos);
|
||||||
}
|
}
|
||||||
|
|
||||||
goto done;
|
goto done;
|
||||||
|
|
||||||
error:
|
error:
|
||||||
|
if (!fd_is_own)
|
||||||
|
self->fd = -1;
|
||||||
|
|
||||||
ret = -1;
|
ret = -1;
|
||||||
|
|
||||||
done:
|
done:
|
||||||
|
|
Loading…
Reference in New Issue