mirror of
https://github.com/python/cpython.git
synced 2025-08-03 16:39:00 +00:00
gh-102660: Fix Refleaks in import.c (#102744)
gh-102661 introduced some leaks. This fixes them. https://github.com/python/cpython/issues/102660
This commit is contained in:
parent
675b97a6ab
commit
2a03ed034e
3 changed files with 28 additions and 28 deletions
|
@ -3098,9 +3098,6 @@ _PyBuiltin_Init(PyInterpreterState *interp)
|
||||||
}
|
}
|
||||||
Py_DECREF(debug);
|
Py_DECREF(debug);
|
||||||
|
|
||||||
/* m_copy of Py_None means it is copied some other way. */
|
|
||||||
builtinsmodule.m_base.m_copy = Py_NewRef(Py_None);
|
|
||||||
|
|
||||||
return mod;
|
return mod;
|
||||||
#undef ADD_TO_ALL
|
#undef ADD_TO_ALL
|
||||||
#undef SETBUILTIN
|
#undef SETBUILTIN
|
||||||
|
|
|
@ -978,14 +978,29 @@ _PyImport_CheckSubinterpIncompatibleExtensionAllowed(const char *name)
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int
|
static PyObject *
|
||||||
match_mod_name(PyObject *actual, const char *expected)
|
get_core_module_dict(PyInterpreterState *interp,
|
||||||
|
PyObject *name, PyObject *filename)
|
||||||
{
|
{
|
||||||
if (PyUnicode_CompareWithASCIIString(actual, expected) == 0) {
|
/* Only builtin modules are core. */
|
||||||
return 1;
|
if (filename == name) {
|
||||||
|
assert(!PyErr_Occurred());
|
||||||
|
if (PyUnicode_CompareWithASCIIString(name, "sys") == 0) {
|
||||||
|
return interp->sysdict_copy;
|
||||||
|
}
|
||||||
|
assert(!PyErr_Occurred());
|
||||||
|
if (PyUnicode_CompareWithASCIIString(name, "builtins") == 0) {
|
||||||
|
return interp->builtins_copy;
|
||||||
|
}
|
||||||
|
assert(!PyErr_Occurred());
|
||||||
}
|
}
|
||||||
assert(!PyErr_Occurred());
|
return NULL;
|
||||||
return 0;
|
}
|
||||||
|
|
||||||
|
static inline int
|
||||||
|
is_core_module(PyInterpreterState *interp, PyObject *name, PyObject *filename)
|
||||||
|
{
|
||||||
|
return get_core_module_dict(interp, name, filename) != NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int
|
static int
|
||||||
|
@ -1009,10 +1024,8 @@ fix_up_extension(PyObject *mod, PyObject *name, PyObject *filename)
|
||||||
|
|
||||||
// bpo-44050: Extensions and def->m_base.m_copy can be updated
|
// bpo-44050: Extensions and def->m_base.m_copy can be updated
|
||||||
// when the extension module doesn't support sub-interpreters.
|
// when the extension module doesn't support sub-interpreters.
|
||||||
// XXX Why special-case the main interpreter?
|
if (def->m_size == -1) {
|
||||||
if (_Py_IsMainInterpreter(tstate->interp) || def->m_size == -1) {
|
if (!is_core_module(tstate->interp, name, filename)) {
|
||||||
/* m_copy of Py_None means it is copied some other way. */
|
|
||||||
if (def->m_size == -1 && def->m_base.m_copy != Py_None) {
|
|
||||||
if (def->m_base.m_copy) {
|
if (def->m_base.m_copy) {
|
||||||
/* Somebody already imported the module,
|
/* Somebody already imported the module,
|
||||||
likely under a different name.
|
likely under a different name.
|
||||||
|
@ -1028,7 +1041,10 @@ fix_up_extension(PyObject *mod, PyObject *name, PyObject *filename)
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// XXX Why special-case the main interpreter?
|
||||||
|
if (_Py_IsMainInterpreter(tstate->interp) || def->m_size == -1) {
|
||||||
if (_extensions_cache_set(filename, name, def) < 0) {
|
if (_extensions_cache_set(filename, name, def) < 0) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
@ -1069,21 +1085,11 @@ import_find_extension(PyThreadState *tstate, PyObject *name,
|
||||||
PyObject *m_copy = def->m_base.m_copy;
|
PyObject *m_copy = def->m_base.m_copy;
|
||||||
/* Module does not support repeated initialization */
|
/* Module does not support repeated initialization */
|
||||||
if (m_copy == NULL) {
|
if (m_copy == NULL) {
|
||||||
return NULL;
|
m_copy = get_core_module_dict(tstate->interp, name, filename);
|
||||||
}
|
if (m_copy == NULL) {
|
||||||
else if (m_copy == Py_None) {
|
|
||||||
if (match_mod_name(name, "sys")) {
|
|
||||||
m_copy = tstate->interp->sysdict_copy;
|
|
||||||
}
|
|
||||||
else if (match_mod_name(name, "builtins")) {
|
|
||||||
m_copy = tstate->interp->builtins_copy;
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
_PyErr_SetString(tstate, PyExc_ImportError, "missing m_copy");
|
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
/* m_copy of Py_None means it is copied some other way. */
|
|
||||||
mod = import_add_module(tstate, name);
|
mod = import_add_module(tstate, name);
|
||||||
if (mod == NULL) {
|
if (mod == NULL) {
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
|
@ -3425,9 +3425,6 @@ _PySys_Create(PyThreadState *tstate, PyObject **sysmod_p)
|
||||||
return _PyStatus_ERR("failed to create a module object");
|
return _PyStatus_ERR("failed to create a module object");
|
||||||
}
|
}
|
||||||
|
|
||||||
/* m_copy of Py_None means it is copied some other way. */
|
|
||||||
sysmodule.m_base.m_copy = Py_NewRef(Py_None);
|
|
||||||
|
|
||||||
PyObject *sysdict = PyModule_GetDict(sysmod);
|
PyObject *sysdict = PyModule_GetDict(sysmod);
|
||||||
if (sysdict == NULL) {
|
if (sysdict == NULL) {
|
||||||
goto error;
|
goto error;
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue