Issue #7732: Fix a crash on importing a module if a directory has the same name

than a Python module (e.g. "__init__.py"): don't close the file twice.

PyFile_FromFile() does also close the file if PyString_FromString() failed. It
did already close the file on fill_file_fields() error (e.g. if the file is a
directory).
This commit is contained in:
Victor Stinner 2011-09-23 19:37:03 +02:00
parent 51b719814e
commit 63c22fac72
5 changed files with 34 additions and 18 deletions

View File

@ -55,7 +55,8 @@ change in future releases of Python.
Create a new :ctype:`PyFileObject` from the already-open standard C file Create a new :ctype:`PyFileObject` from the already-open standard C file
pointer, *fp*. The function *close* will be called when the file should be pointer, *fp*. The function *close* will be called when the file should be
closed. Return *NULL* on failure. closed. Return *NULL* and close the file using *close* on failure.
*close* is optional and can be set to *NULL*.
.. cfunction:: FILE* PyFile_AsFile(PyObject \*p) .. cfunction:: FILE* PyFile_AsFile(PyObject \*p)

View File

@ -265,6 +265,14 @@ class ImportTests(unittest.TestCase):
""")) """))
script_helper.assert_python_ok(testfn) script_helper.assert_python_ok(testfn)
def test_bug7732(self):
source = TESTFN + '.py'
os.mkdir(source)
try:
self.assertRaises(IOError, imp.find_module, TESTFN, ["."])
finally:
os.rmdir(source)
class PycRewritingTests(unittest.TestCase): class PycRewritingTests(unittest.TestCase):
# Test that the `co_filename` attribute on code objects always points # Test that the `co_filename` attribute on code objects always points

View File

@ -9,6 +9,9 @@ What's New in Python 2.7.3?
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #7732: Fix a crash on importing a module if a directory has the same
name than a Python module (e.g. "__init__.py"): don't close the file twice.
- Issue #12973: Fix overflow checks that invoked undefined behaviour in - Issue #12973: Fix overflow checks that invoked undefined behaviour in
int.__pow__. These overflow checks were causing int.__pow__ to produce int.__pow__. These overflow checks were causing int.__pow__ to produce
incorrect results with recent versions of Clang, as a result of the incorrect results with recent versions of Clang, as a result of the

View File

@ -468,28 +468,34 @@ close_the_file(PyFileObject *f)
PyObject * PyObject *
PyFile_FromFile(FILE *fp, char *name, char *mode, int (*close)(FILE *)) PyFile_FromFile(FILE *fp, char *name, char *mode, int (*close)(FILE *))
{ {
PyFileObject *f = (PyFileObject *)PyFile_Type.tp_new(&PyFile_Type, PyFileObject *f;
NULL, NULL); PyObject *o_name;
if (f != NULL) {
PyObject *o_name = PyString_FromString(name); f = (PyFileObject *)PyFile_Type.tp_new(&PyFile_Type, NULL, NULL);
if (o_name == NULL) if (f == NULL)
return NULL; return NULL;
if (fill_file_fields(f, fp, o_name, mode, close) == NULL) { o_name = PyString_FromString(name);
Py_DECREF(f); if (o_name == NULL) {
f = NULL; if (close != NULL && fp != NULL)
} close(fp);
Py_DECREF(o_name); Py_DECREF(f);
return NULL;
} }
return (PyObject *) f; if (fill_file_fields(f, fp, o_name, mode, close) == NULL) {
Py_DECREF(f);
Py_DECREF(o_name);
return NULL;
}
Py_DECREF(o_name);
return (PyObject *)f;
} }
PyObject * PyObject *
PyFile_FromString(char *name, char *mode) PyFile_FromString(char *name, char *mode)
{ {
extern int fclose(FILE *);
PyFileObject *f; PyFileObject *f;
f = (PyFileObject *)PyFile_FromFile((FILE *)NULL, name, mode, fclose); f = (PyFileObject *)PyFile_FromFile((FILE *)NULL, name, mode, NULL);
if (f != NULL) { if (f != NULL) {
if (open_the_file(f, name, mode) == NULL) { if (open_the_file(f, name, mode) == NULL) {
Py_DECREF(f); Py_DECREF(f);

View File

@ -2845,10 +2845,8 @@ call_find_module(char *name, PyObject *path)
return NULL; return NULL;
if (fp != NULL) { if (fp != NULL) {
fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose); fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
if (fob == NULL) { if (fob == NULL)
fclose(fp);
return NULL; return NULL;
}
} }
else { else {
fob = Py_None; fob = Py_None;