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:
Eric Snow 2023-03-15 18:43:54 -06:00 committed by GitHub
parent 675b97a6ab
commit 2a03ed034e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 28 additions and 28 deletions

View file

@ -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

View file

@ -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;

View file

@ -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;