diff options
| author | orivej <[email protected]> | 2022-02-10 16:45:01 +0300 |
|---|---|---|
| committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:45:01 +0300 |
| commit | 2d37894b1b037cf24231090eda8589bbb44fb6fc (patch) | |
| tree | be835aa92c6248212e705f25388ebafcf84bc7a1 /contrib/tools/python3/src/Objects/moduleobject.c | |
| parent | 718c552901d703c502ccbefdfc3c9028d608b947 (diff) | |
Restoring authorship annotation for <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'contrib/tools/python3/src/Objects/moduleobject.c')
| -rw-r--r-- | contrib/tools/python3/src/Objects/moduleobject.c | 1554 |
1 files changed, 777 insertions, 777 deletions
diff --git a/contrib/tools/python3/src/Objects/moduleobject.c b/contrib/tools/python3/src/Objects/moduleobject.c index 24022fe7365..ee4ed97588e 100644 --- a/contrib/tools/python3/src/Objects/moduleobject.c +++ b/contrib/tools/python3/src/Objects/moduleobject.c @@ -1,694 +1,694 @@ - -/* Module object implementation */ - -#include "Python.h" + +/* Module object implementation */ + +#include "Python.h" #include "pycore_interp.h" // PyInterpreterState.importlib #include "pycore_pystate.h" // _PyInterpreterState_GET() #include "structmember.h" // PyMemberDef - -static Py_ssize_t max_module_number; - + +static Py_ssize_t max_module_number; + _Py_IDENTIFIER(__doc__); _Py_IDENTIFIER(__name__); _Py_IDENTIFIER(__spec__); -typedef struct { - PyObject_HEAD - PyObject *md_dict; - struct PyModuleDef *md_def; - void *md_state; - PyObject *md_weaklist; - PyObject *md_name; /* for logging purposes after md_dict is cleared */ -} PyModuleObject; - -static PyMemberDef module_members[] = { - {"__dict__", T_OBJECT, offsetof(PyModuleObject, md_dict), READONLY}, - {0} -}; - - -PyTypeObject PyModuleDef_Type = { - PyVarObject_HEAD_INIT(&PyType_Type, 0) - "moduledef", /* tp_name */ - sizeof(struct PyModuleDef), /* tp_basicsize */ - 0, /* tp_itemsize */ -}; - - -PyObject* -PyModuleDef_Init(struct PyModuleDef* def) -{ - if (PyType_Ready(&PyModuleDef_Type) < 0) - return NULL; - if (def->m_base.m_index == 0) { - max_module_number++; +typedef struct { + PyObject_HEAD + PyObject *md_dict; + struct PyModuleDef *md_def; + void *md_state; + PyObject *md_weaklist; + PyObject *md_name; /* for logging purposes after md_dict is cleared */ +} PyModuleObject; + +static PyMemberDef module_members[] = { + {"__dict__", T_OBJECT, offsetof(PyModuleObject, md_dict), READONLY}, + {0} +}; + + +PyTypeObject PyModuleDef_Type = { + PyVarObject_HEAD_INIT(&PyType_Type, 0) + "moduledef", /* tp_name */ + sizeof(struct PyModuleDef), /* tp_basicsize */ + 0, /* tp_itemsize */ +}; + + +PyObject* +PyModuleDef_Init(struct PyModuleDef* def) +{ + if (PyType_Ready(&PyModuleDef_Type) < 0) + return NULL; + if (def->m_base.m_index == 0) { + max_module_number++; Py_SET_REFCNT(def, 1); Py_SET_TYPE(def, &PyModuleDef_Type); - def->m_base.m_index = max_module_number; - } - return (PyObject*)def; -} - -static int -module_init_dict(PyModuleObject *mod, PyObject *md_dict, - PyObject *name, PyObject *doc) -{ - _Py_IDENTIFIER(__package__); - _Py_IDENTIFIER(__loader__); - - if (md_dict == NULL) - return -1; - if (doc == NULL) - doc = Py_None; - - if (_PyDict_SetItemId(md_dict, &PyId___name__, name) != 0) - return -1; - if (_PyDict_SetItemId(md_dict, &PyId___doc__, doc) != 0) - return -1; - if (_PyDict_SetItemId(md_dict, &PyId___package__, Py_None) != 0) - return -1; - if (_PyDict_SetItemId(md_dict, &PyId___loader__, Py_None) != 0) - return -1; - if (_PyDict_SetItemId(md_dict, &PyId___spec__, Py_None) != 0) - return -1; - if (PyUnicode_CheckExact(name)) { - Py_INCREF(name); - Py_XSETREF(mod->md_name, name); - } - - return 0; -} - - -PyObject * -PyModule_NewObject(PyObject *name) -{ - PyModuleObject *m; - m = PyObject_GC_New(PyModuleObject, &PyModule_Type); - if (m == NULL) - return NULL; - m->md_def = NULL; - m->md_state = NULL; - m->md_weaklist = NULL; - m->md_name = NULL; - m->md_dict = PyDict_New(); - if (module_init_dict(m, m->md_dict, name, NULL) != 0) - goto fail; - PyObject_GC_Track(m); - return (PyObject *)m; - - fail: - Py_DECREF(m); - return NULL; -} - -PyObject * -PyModule_New(const char *name) -{ - PyObject *nameobj, *module; - nameobj = PyUnicode_FromString(name); - if (nameobj == NULL) - return NULL; - module = PyModule_NewObject(nameobj); - Py_DECREF(nameobj); - return module; -} - -/* Check API/ABI version - * Issues a warning on mismatch, which is usually not fatal. - * Returns 0 if an exception is raised. - */ -static int -check_api_version(const char *name, int module_api_version) -{ - if (module_api_version != PYTHON_API_VERSION && module_api_version != PYTHON_ABI_VERSION) { - int err; - err = PyErr_WarnFormat(PyExc_RuntimeWarning, 1, - "Python C API version mismatch for module %.100s: " - "This Python has API version %d, module %.100s has version %d.", - name, - PYTHON_API_VERSION, name, module_api_version); - if (err) - return 0; - } - return 1; -} - -static int -_add_methods_to_object(PyObject *module, PyObject *name, PyMethodDef *functions) -{ - PyObject *func; - PyMethodDef *fdef; - - for (fdef = functions; fdef->ml_name != NULL; fdef++) { - if ((fdef->ml_flags & METH_CLASS) || - (fdef->ml_flags & METH_STATIC)) { - PyErr_SetString(PyExc_ValueError, - "module functions cannot set" - " METH_CLASS or METH_STATIC"); - return -1; - } - func = PyCFunction_NewEx(fdef, (PyObject*)module, name); - if (func == NULL) { - return -1; - } - if (PyObject_SetAttrString(module, fdef->ml_name, func) != 0) { - Py_DECREF(func); - return -1; - } - Py_DECREF(func); - } - - return 0; -} - -PyObject * -PyModule_Create2(struct PyModuleDef* module, int module_api_version) -{ + def->m_base.m_index = max_module_number; + } + return (PyObject*)def; +} + +static int +module_init_dict(PyModuleObject *mod, PyObject *md_dict, + PyObject *name, PyObject *doc) +{ + _Py_IDENTIFIER(__package__); + _Py_IDENTIFIER(__loader__); + + if (md_dict == NULL) + return -1; + if (doc == NULL) + doc = Py_None; + + if (_PyDict_SetItemId(md_dict, &PyId___name__, name) != 0) + return -1; + if (_PyDict_SetItemId(md_dict, &PyId___doc__, doc) != 0) + return -1; + if (_PyDict_SetItemId(md_dict, &PyId___package__, Py_None) != 0) + return -1; + if (_PyDict_SetItemId(md_dict, &PyId___loader__, Py_None) != 0) + return -1; + if (_PyDict_SetItemId(md_dict, &PyId___spec__, Py_None) != 0) + return -1; + if (PyUnicode_CheckExact(name)) { + Py_INCREF(name); + Py_XSETREF(mod->md_name, name); + } + + return 0; +} + + +PyObject * +PyModule_NewObject(PyObject *name) +{ + PyModuleObject *m; + m = PyObject_GC_New(PyModuleObject, &PyModule_Type); + if (m == NULL) + return NULL; + m->md_def = NULL; + m->md_state = NULL; + m->md_weaklist = NULL; + m->md_name = NULL; + m->md_dict = PyDict_New(); + if (module_init_dict(m, m->md_dict, name, NULL) != 0) + goto fail; + PyObject_GC_Track(m); + return (PyObject *)m; + + fail: + Py_DECREF(m); + return NULL; +} + +PyObject * +PyModule_New(const char *name) +{ + PyObject *nameobj, *module; + nameobj = PyUnicode_FromString(name); + if (nameobj == NULL) + return NULL; + module = PyModule_NewObject(nameobj); + Py_DECREF(nameobj); + return module; +} + +/* Check API/ABI version + * Issues a warning on mismatch, which is usually not fatal. + * Returns 0 if an exception is raised. + */ +static int +check_api_version(const char *name, int module_api_version) +{ + if (module_api_version != PYTHON_API_VERSION && module_api_version != PYTHON_ABI_VERSION) { + int err; + err = PyErr_WarnFormat(PyExc_RuntimeWarning, 1, + "Python C API version mismatch for module %.100s: " + "This Python has API version %d, module %.100s has version %d.", + name, + PYTHON_API_VERSION, name, module_api_version); + if (err) + return 0; + } + return 1; +} + +static int +_add_methods_to_object(PyObject *module, PyObject *name, PyMethodDef *functions) +{ + PyObject *func; + PyMethodDef *fdef; + + for (fdef = functions; fdef->ml_name != NULL; fdef++) { + if ((fdef->ml_flags & METH_CLASS) || + (fdef->ml_flags & METH_STATIC)) { + PyErr_SetString(PyExc_ValueError, + "module functions cannot set" + " METH_CLASS or METH_STATIC"); + return -1; + } + func = PyCFunction_NewEx(fdef, (PyObject*)module, name); + if (func == NULL) { + return -1; + } + if (PyObject_SetAttrString(module, fdef->ml_name, func) != 0) { + Py_DECREF(func); + return -1; + } + Py_DECREF(func); + } + + return 0; +} + +PyObject * +PyModule_Create2(struct PyModuleDef* module, int module_api_version) +{ if (!_PyImport_IsInitialized(_PyInterpreterState_GET())) { PyErr_SetString(PyExc_SystemError, "Python import machinery not initialized"); return NULL; } - return _PyModule_CreateInitialized(module, module_api_version); -} - -PyObject * -_PyModule_CreateInitialized(struct PyModuleDef* module, int module_api_version) -{ - const char* name; - PyModuleObject *m; - - if (!PyModuleDef_Init(module)) - return NULL; - name = module->m_name; - if (!check_api_version(name, module_api_version)) { - return NULL; - } - if (module->m_slots) { - PyErr_Format( - PyExc_SystemError, - "module %s: PyModule_Create is incompatible with m_slots", name); - return NULL; - } - /* Make sure name is fully qualified. - - This is a bit of a hack: when the shared library is loaded, - the module name is "package.module", but the module calls - PyModule_Create*() with just "module" for the name. The shared - library loader squirrels away the true name of the module in - _Py_PackageContext, and PyModule_Create*() will substitute this - (if the name actually matches). - */ - if (_Py_PackageContext != NULL) { - const char *p = strrchr(_Py_PackageContext, '.'); - if (p != NULL && strcmp(module->m_name, p+1) == 0) { - name = _Py_PackageContext; - _Py_PackageContext = NULL; - } - } - if ((m = (PyModuleObject*)PyModule_New(name)) == NULL) - return NULL; - - if (module->m_size > 0) { - m->md_state = PyMem_MALLOC(module->m_size); - if (!m->md_state) { - PyErr_NoMemory(); - Py_DECREF(m); - return NULL; - } - memset(m->md_state, 0, module->m_size); - } - - if (module->m_methods != NULL) { - if (PyModule_AddFunctions((PyObject *) m, module->m_methods) != 0) { - Py_DECREF(m); - return NULL; - } - } - if (module->m_doc != NULL) { - if (PyModule_SetDocString((PyObject *) m, module->m_doc) != 0) { - Py_DECREF(m); - return NULL; - } - } - m->md_def = module; - return (PyObject*)m; -} - -PyObject * -PyModule_FromDefAndSpec2(struct PyModuleDef* def, PyObject *spec, int module_api_version) -{ - PyModuleDef_Slot* cur_slot; - PyObject *(*create)(PyObject *, PyModuleDef*) = NULL; - PyObject *nameobj; - PyObject *m = NULL; - int has_execution_slots = 0; - const char *name; - int ret; - - PyModuleDef_Init(def); - - nameobj = PyObject_GetAttrString(spec, "name"); - if (nameobj == NULL) { - return NULL; - } - name = PyUnicode_AsUTF8(nameobj); - if (name == NULL) { - goto error; - } - - if (!check_api_version(name, module_api_version)) { - goto error; - } - - if (def->m_size < 0) { - PyErr_Format( - PyExc_SystemError, - "module %s: m_size may not be negative for multi-phase initialization", - name); - goto error; - } - - for (cur_slot = def->m_slots; cur_slot && cur_slot->slot; cur_slot++) { - if (cur_slot->slot == Py_mod_create) { - if (create) { - PyErr_Format( - PyExc_SystemError, - "module %s has multiple create slots", - name); - goto error; - } - create = cur_slot->value; - } else if (cur_slot->slot < 0 || cur_slot->slot > _Py_mod_LAST_SLOT) { - PyErr_Format( - PyExc_SystemError, - "module %s uses unknown slot ID %i", - name, cur_slot->slot); - goto error; - } else { - has_execution_slots = 1; - } - } - - if (create) { - m = create(spec, def); - if (m == NULL) { - if (!PyErr_Occurred()) { - PyErr_Format( - PyExc_SystemError, - "creation of module %s failed without setting an exception", - name); - } - goto error; - } else { - if (PyErr_Occurred()) { - PyErr_Format(PyExc_SystemError, - "creation of module %s raised unreported exception", - name); - goto error; - } - } - } else { - m = PyModule_NewObject(nameobj); - if (m == NULL) { - goto error; - } - } - - if (PyModule_Check(m)) { - ((PyModuleObject*)m)->md_state = NULL; - ((PyModuleObject*)m)->md_def = def; - } else { - if (def->m_size > 0 || def->m_traverse || def->m_clear || def->m_free) { - PyErr_Format( - PyExc_SystemError, - "module %s is not a module object, but requests module state", - name); - goto error; - } - if (has_execution_slots) { - PyErr_Format( - PyExc_SystemError, - "module %s specifies execution slots, but did not create " - "a ModuleType instance", - name); - goto error; - } - } - - if (def->m_methods != NULL) { - ret = _add_methods_to_object(m, nameobj, def->m_methods); - if (ret != 0) { - goto error; - } - } - - if (def->m_doc != NULL) { - ret = PyModule_SetDocString(m, def->m_doc); - if (ret != 0) { - goto error; - } - } - - Py_DECREF(nameobj); - return m; - -error: - Py_DECREF(nameobj); - Py_XDECREF(m); - return NULL; -} - -int -PyModule_ExecDef(PyObject *module, PyModuleDef *def) -{ - PyModuleDef_Slot *cur_slot; - const char *name; - int ret; - - name = PyModule_GetName(module); - if (name == NULL) { - return -1; - } - - if (def->m_size >= 0) { - PyModuleObject *md = (PyModuleObject*)module; - if (md->md_state == NULL) { - /* Always set a state pointer; this serves as a marker to skip - * multiple initialization (importlib.reload() is no-op) */ - md->md_state = PyMem_MALLOC(def->m_size); - if (!md->md_state) { - PyErr_NoMemory(); - return -1; - } - memset(md->md_state, 0, def->m_size); - } - } - - if (def->m_slots == NULL) { - return 0; - } - - for (cur_slot = def->m_slots; cur_slot && cur_slot->slot; cur_slot++) { - switch (cur_slot->slot) { - case Py_mod_create: - /* handled in PyModule_FromDefAndSpec2 */ - break; - case Py_mod_exec: - ret = ((int (*)(PyObject *))cur_slot->value)(module); - if (ret != 0) { - if (!PyErr_Occurred()) { - PyErr_Format( - PyExc_SystemError, - "execution of module %s failed without setting an exception", - name); - } - return -1; - } - if (PyErr_Occurred()) { - PyErr_Format( - PyExc_SystemError, - "execution of module %s raised unreported exception", - name); - return -1; - } - break; - default: - PyErr_Format( - PyExc_SystemError, - "module %s initialized with unknown slot %i", - name, cur_slot->slot); - return -1; - } - } - return 0; -} - -int -PyModule_AddFunctions(PyObject *m, PyMethodDef *functions) -{ - int res; - PyObject *name = PyModule_GetNameObject(m); - if (name == NULL) { - return -1; - } - - res = _add_methods_to_object(m, name, functions); - Py_DECREF(name); - return res; -} - -int -PyModule_SetDocString(PyObject *m, const char *doc) -{ - PyObject *v; - - v = PyUnicode_FromString(doc); - if (v == NULL || _PyObject_SetAttrId(m, &PyId___doc__, v) != 0) { - Py_XDECREF(v); - return -1; - } - Py_DECREF(v); - return 0; -} - -PyObject * -PyModule_GetDict(PyObject *m) -{ - PyObject *d; - if (!PyModule_Check(m)) { - PyErr_BadInternalCall(); - return NULL; - } - d = ((PyModuleObject *)m) -> md_dict; - assert(d != NULL); - return d; -} - -PyObject* -PyModule_GetNameObject(PyObject *m) -{ - PyObject *d; - PyObject *name; - if (!PyModule_Check(m)) { - PyErr_BadArgument(); - return NULL; - } - d = ((PyModuleObject *)m)->md_dict; - if (d == NULL || - (name = _PyDict_GetItemId(d, &PyId___name__)) == NULL || - !PyUnicode_Check(name)) - { - PyErr_SetString(PyExc_SystemError, "nameless module"); - return NULL; - } - Py_INCREF(name); - return name; -} - -const char * -PyModule_GetName(PyObject *m) -{ - PyObject *name = PyModule_GetNameObject(m); - if (name == NULL) - return NULL; - Py_DECREF(name); /* module dict has still a reference */ - return PyUnicode_AsUTF8(name); -} - -PyObject* -PyModule_GetFilenameObject(PyObject *m) -{ - _Py_IDENTIFIER(__file__); - PyObject *d; - PyObject *fileobj; - if (!PyModule_Check(m)) { - PyErr_BadArgument(); - return NULL; - } - d = ((PyModuleObject *)m)->md_dict; - if (d == NULL || - (fileobj = _PyDict_GetItemId(d, &PyId___file__)) == NULL || - !PyUnicode_Check(fileobj)) - { - PyErr_SetString(PyExc_SystemError, "module filename missing"); - return NULL; - } - Py_INCREF(fileobj); - return fileobj; -} - -const char * -PyModule_GetFilename(PyObject *m) -{ - PyObject *fileobj; - const char *utf8; - fileobj = PyModule_GetFilenameObject(m); - if (fileobj == NULL) - return NULL; - utf8 = PyUnicode_AsUTF8(fileobj); - Py_DECREF(fileobj); /* module dict has still a reference */ - return utf8; -} - -PyModuleDef* -PyModule_GetDef(PyObject* m) -{ - if (!PyModule_Check(m)) { - PyErr_BadArgument(); - return NULL; - } - return ((PyModuleObject *)m)->md_def; -} - -void* -PyModule_GetState(PyObject* m) -{ - if (!PyModule_Check(m)) { - PyErr_BadArgument(); - return NULL; - } - return ((PyModuleObject *)m)->md_state; -} - -void -_PyModule_Clear(PyObject *m) -{ - PyObject *d = ((PyModuleObject *)m)->md_dict; - if (d != NULL) - _PyModule_ClearDict(d); -} - -void -_PyModule_ClearDict(PyObject *d) -{ - /* To make the execution order of destructors for global - objects a bit more predictable, we first zap all objects - whose name starts with a single underscore, before we clear - the entire dictionary. We zap them by replacing them with - None, rather than deleting them from the dictionary, to - avoid rehashing the dictionary (to some extent). */ - - Py_ssize_t pos; - PyObject *key, *value; - + return _PyModule_CreateInitialized(module, module_api_version); +} + +PyObject * +_PyModule_CreateInitialized(struct PyModuleDef* module, int module_api_version) +{ + const char* name; + PyModuleObject *m; + + if (!PyModuleDef_Init(module)) + return NULL; + name = module->m_name; + if (!check_api_version(name, module_api_version)) { + return NULL; + } + if (module->m_slots) { + PyErr_Format( + PyExc_SystemError, + "module %s: PyModule_Create is incompatible with m_slots", name); + return NULL; + } + /* Make sure name is fully qualified. + + This is a bit of a hack: when the shared library is loaded, + the module name is "package.module", but the module calls + PyModule_Create*() with just "module" for the name. The shared + library loader squirrels away the true name of the module in + _Py_PackageContext, and PyModule_Create*() will substitute this + (if the name actually matches). + */ + if (_Py_PackageContext != NULL) { + const char *p = strrchr(_Py_PackageContext, '.'); + if (p != NULL && strcmp(module->m_name, p+1) == 0) { + name = _Py_PackageContext; + _Py_PackageContext = NULL; + } + } + if ((m = (PyModuleObject*)PyModule_New(name)) == NULL) + return NULL; + + if (module->m_size > 0) { + m->md_state = PyMem_MALLOC(module->m_size); + if (!m->md_state) { + PyErr_NoMemory(); + Py_DECREF(m); + return NULL; + } + memset(m->md_state, 0, module->m_size); + } + + if (module->m_methods != NULL) { + if (PyModule_AddFunctions((PyObject *) m, module->m_methods) != 0) { + Py_DECREF(m); + return NULL; + } + } + if (module->m_doc != NULL) { + if (PyModule_SetDocString((PyObject *) m, module->m_doc) != 0) { + Py_DECREF(m); + return NULL; + } + } + m->md_def = module; + return (PyObject*)m; +} + +PyObject * +PyModule_FromDefAndSpec2(struct PyModuleDef* def, PyObject *spec, int module_api_version) +{ + PyModuleDef_Slot* cur_slot; + PyObject *(*create)(PyObject *, PyModuleDef*) = NULL; + PyObject *nameobj; + PyObject *m = NULL; + int has_execution_slots = 0; + const char *name; + int ret; + + PyModuleDef_Init(def); + + nameobj = PyObject_GetAttrString(spec, "name"); + if (nameobj == NULL) { + return NULL; + } + name = PyUnicode_AsUTF8(nameobj); + if (name == NULL) { + goto error; + } + + if (!check_api_version(name, module_api_version)) { + goto error; + } + + if (def->m_size < 0) { + PyErr_Format( + PyExc_SystemError, + "module %s: m_size may not be negative for multi-phase initialization", + name); + goto error; + } + + for (cur_slot = def->m_slots; cur_slot && cur_slot->slot; cur_slot++) { + if (cur_slot->slot == Py_mod_create) { + if (create) { + PyErr_Format( + PyExc_SystemError, + "module %s has multiple create slots", + name); + goto error; + } + create = cur_slot->value; + } else if (cur_slot->slot < 0 || cur_slot->slot > _Py_mod_LAST_SLOT) { + PyErr_Format( + PyExc_SystemError, + "module %s uses unknown slot ID %i", + name, cur_slot->slot); + goto error; + } else { + has_execution_slots = 1; + } + } + + if (create) { + m = create(spec, def); + if (m == NULL) { + if (!PyErr_Occurred()) { + PyErr_Format( + PyExc_SystemError, + "creation of module %s failed without setting an exception", + name); + } + goto error; + } else { + if (PyErr_Occurred()) { + PyErr_Format(PyExc_SystemError, + "creation of module %s raised unreported exception", + name); + goto error; + } + } + } else { + m = PyModule_NewObject(nameobj); + if (m == NULL) { + goto error; + } + } + + if (PyModule_Check(m)) { + ((PyModuleObject*)m)->md_state = NULL; + ((PyModuleObject*)m)->md_def = def; + } else { + if (def->m_size > 0 || def->m_traverse || def->m_clear || def->m_free) { + PyErr_Format( + PyExc_SystemError, + "module %s is not a module object, but requests module state", + name); + goto error; + } + if (has_execution_slots) { + PyErr_Format( + PyExc_SystemError, + "module %s specifies execution slots, but did not create " + "a ModuleType instance", + name); + goto error; + } + } + + if (def->m_methods != NULL) { + ret = _add_methods_to_object(m, nameobj, def->m_methods); + if (ret != 0) { + goto error; + } + } + + if (def->m_doc != NULL) { + ret = PyModule_SetDocString(m, def->m_doc); + if (ret != 0) { + goto error; + } + } + + Py_DECREF(nameobj); + return m; + +error: + Py_DECREF(nameobj); + Py_XDECREF(m); + return NULL; +} + +int +PyModule_ExecDef(PyObject *module, PyModuleDef *def) +{ + PyModuleDef_Slot *cur_slot; + const char *name; + int ret; + + name = PyModule_GetName(module); + if (name == NULL) { + return -1; + } + + if (def->m_size >= 0) { + PyModuleObject *md = (PyModuleObject*)module; + if (md->md_state == NULL) { + /* Always set a state pointer; this serves as a marker to skip + * multiple initialization (importlib.reload() is no-op) */ + md->md_state = PyMem_MALLOC(def->m_size); + if (!md->md_state) { + PyErr_NoMemory(); + return -1; + } + memset(md->md_state, 0, def->m_size); + } + } + + if (def->m_slots == NULL) { + return 0; + } + + for (cur_slot = def->m_slots; cur_slot && cur_slot->slot; cur_slot++) { + switch (cur_slot->slot) { + case Py_mod_create: + /* handled in PyModule_FromDefAndSpec2 */ + break; + case Py_mod_exec: + ret = ((int (*)(PyObject *))cur_slot->value)(module); + if (ret != 0) { + if (!PyErr_Occurred()) { + PyErr_Format( + PyExc_SystemError, + "execution of module %s failed without setting an exception", + name); + } + return -1; + } + if (PyErr_Occurred()) { + PyErr_Format( + PyExc_SystemError, + "execution of module %s raised unreported exception", + name); + return -1; + } + break; + default: + PyErr_Format( + PyExc_SystemError, + "module %s initialized with unknown slot %i", + name, cur_slot->slot); + return -1; + } + } + return 0; +} + +int +PyModule_AddFunctions(PyObject *m, PyMethodDef *functions) +{ + int res; + PyObject *name = PyModule_GetNameObject(m); + if (name == NULL) { + return -1; + } + + res = _add_methods_to_object(m, name, functions); + Py_DECREF(name); + return res; +} + +int +PyModule_SetDocString(PyObject *m, const char *doc) +{ + PyObject *v; + + v = PyUnicode_FromString(doc); + if (v == NULL || _PyObject_SetAttrId(m, &PyId___doc__, v) != 0) { + Py_XDECREF(v); + return -1; + } + Py_DECREF(v); + return 0; +} + +PyObject * +PyModule_GetDict(PyObject *m) +{ + PyObject *d; + if (!PyModule_Check(m)) { + PyErr_BadInternalCall(); + return NULL; + } + d = ((PyModuleObject *)m) -> md_dict; + assert(d != NULL); + return d; +} + +PyObject* +PyModule_GetNameObject(PyObject *m) +{ + PyObject *d; + PyObject *name; + if (!PyModule_Check(m)) { + PyErr_BadArgument(); + return NULL; + } + d = ((PyModuleObject *)m)->md_dict; + if (d == NULL || + (name = _PyDict_GetItemId(d, &PyId___name__)) == NULL || + !PyUnicode_Check(name)) + { + PyErr_SetString(PyExc_SystemError, "nameless module"); + return NULL; + } + Py_INCREF(name); + return name; +} + +const char * +PyModule_GetName(PyObject *m) +{ + PyObject *name = PyModule_GetNameObject(m); + if (name == NULL) + return NULL; + Py_DECREF(name); /* module dict has still a reference */ + return PyUnicode_AsUTF8(name); +} + +PyObject* +PyModule_GetFilenameObject(PyObject *m) +{ + _Py_IDENTIFIER(__file__); + PyObject *d; + PyObject *fileobj; + if (!PyModule_Check(m)) { + PyErr_BadArgument(); + return NULL; + } + d = ((PyModuleObject *)m)->md_dict; + if (d == NULL || + (fileobj = _PyDict_GetItemId(d, &PyId___file__)) == NULL || + !PyUnicode_Check(fileobj)) + { + PyErr_SetString(PyExc_SystemError, "module filename missing"); + return NULL; + } + Py_INCREF(fileobj); + return fileobj; +} + +const char * +PyModule_GetFilename(PyObject *m) +{ + PyObject *fileobj; + const char *utf8; + fileobj = PyModule_GetFilenameObject(m); + if (fileobj == NULL) + return NULL; + utf8 = PyUnicode_AsUTF8(fileobj); + Py_DECREF(fileobj); /* module dict has still a reference */ + return utf8; +} + +PyModuleDef* +PyModule_GetDef(PyObject* m) +{ + if (!PyModule_Check(m)) { + PyErr_BadArgument(); + return NULL; + } + return ((PyModuleObject *)m)->md_def; +} + +void* +PyModule_GetState(PyObject* m) +{ + if (!PyModule_Check(m)) { + PyErr_BadArgument(); + return NULL; + } + return ((PyModuleObject *)m)->md_state; +} + +void +_PyModule_Clear(PyObject *m) +{ + PyObject *d = ((PyModuleObject *)m)->md_dict; + if (d != NULL) + _PyModule_ClearDict(d); +} + +void +_PyModule_ClearDict(PyObject *d) +{ + /* To make the execution order of destructors for global + objects a bit more predictable, we first zap all objects + whose name starts with a single underscore, before we clear + the entire dictionary. We zap them by replacing them with + None, rather than deleting them from the dictionary, to + avoid rehashing the dictionary (to some extent). */ + + Py_ssize_t pos; + PyObject *key, *value; + int verbose = _Py_GetConfig()->verbose; - /* First, clear only names starting with a single underscore */ - pos = 0; - while (PyDict_Next(d, &pos, &key, &value)) { - if (value != Py_None && PyUnicode_Check(key)) { - if (PyUnicode_READ_CHAR(key, 0) == '_' && - PyUnicode_READ_CHAR(key, 1) != '_') { + /* First, clear only names starting with a single underscore */ + pos = 0; + while (PyDict_Next(d, &pos, &key, &value)) { + if (value != Py_None && PyUnicode_Check(key)) { + if (PyUnicode_READ_CHAR(key, 0) == '_' && + PyUnicode_READ_CHAR(key, 1) != '_') { if (verbose > 1) { - const char *s = PyUnicode_AsUTF8(key); - if (s != NULL) - PySys_WriteStderr("# clear[1] %s\n", s); - else - PyErr_Clear(); - } + const char *s = PyUnicode_AsUTF8(key); + if (s != NULL) + PySys_WriteStderr("# clear[1] %s\n", s); + else + PyErr_Clear(); + } if (PyDict_SetItem(d, key, Py_None) != 0) { PyErr_WriteUnraisable(NULL); } - } - } - } - - /* Next, clear all names except for __builtins__ */ - pos = 0; - while (PyDict_Next(d, &pos, &key, &value)) { - if (value != Py_None && PyUnicode_Check(key)) { - if (PyUnicode_READ_CHAR(key, 0) != '_' || - !_PyUnicode_EqualToASCIIString(key, "__builtins__")) - { + } + } + } + + /* Next, clear all names except for __builtins__ */ + pos = 0; + while (PyDict_Next(d, &pos, &key, &value)) { + if (value != Py_None && PyUnicode_Check(key)) { + if (PyUnicode_READ_CHAR(key, 0) != '_' || + !_PyUnicode_EqualToASCIIString(key, "__builtins__")) + { if (verbose > 1) { - const char *s = PyUnicode_AsUTF8(key); - if (s != NULL) - PySys_WriteStderr("# clear[2] %s\n", s); - else - PyErr_Clear(); - } + const char *s = PyUnicode_AsUTF8(key); + if (s != NULL) + PySys_WriteStderr("# clear[2] %s\n", s); + else + PyErr_Clear(); + } if (PyDict_SetItem(d, key, Py_None) != 0) { PyErr_WriteUnraisable(NULL); } - } - } - } - - /* Note: we leave __builtins__ in place, so that destructors - of non-global objects defined in this module can still use - builtins, in particularly 'None'. */ - -} - -/*[clinic input] -class module "PyModuleObject *" "&PyModule_Type" -[clinic start generated code]*/ -/*[clinic end generated code: output=da39a3ee5e6b4b0d input=3e35d4f708ecb6af]*/ - -#include "clinic/moduleobject.c.h" - -/* Methods */ - -/*[clinic input] -module.__init__ - name: unicode - doc: object = None - -Create a module object. - -The name must be a string; the optional doc argument can have any type. -[clinic start generated code]*/ - -static int -module___init___impl(PyModuleObject *self, PyObject *name, PyObject *doc) -/*[clinic end generated code: output=e7e721c26ce7aad7 input=57f9e177401e5e1e]*/ -{ - PyObject *dict = self->md_dict; - if (dict == NULL) { - dict = PyDict_New(); - if (dict == NULL) - return -1; - self->md_dict = dict; - } - if (module_init_dict(self, dict, name, doc) < 0) - return -1; - return 0; -} - -static void -module_dealloc(PyModuleObject *m) -{ + } + } + } + + /* Note: we leave __builtins__ in place, so that destructors + of non-global objects defined in this module can still use + builtins, in particularly 'None'. */ + +} + +/*[clinic input] +class module "PyModuleObject *" "&PyModule_Type" +[clinic start generated code]*/ +/*[clinic end generated code: output=da39a3ee5e6b4b0d input=3e35d4f708ecb6af]*/ + +#include "clinic/moduleobject.c.h" + +/* Methods */ + +/*[clinic input] +module.__init__ + name: unicode + doc: object = None + +Create a module object. + +The name must be a string; the optional doc argument can have any type. +[clinic start generated code]*/ + +static int +module___init___impl(PyModuleObject *self, PyObject *name, PyObject *doc) +/*[clinic end generated code: output=e7e721c26ce7aad7 input=57f9e177401e5e1e]*/ +{ + PyObject *dict = self->md_dict; + if (dict == NULL) { + dict = PyDict_New(); + if (dict == NULL) + return -1; + self->md_dict = dict; + } + if (module_init_dict(self, dict, name, doc) < 0) + return -1; + return 0; +} + +static void +module_dealloc(PyModuleObject *m) +{ int verbose = _Py_GetConfig()->verbose; - PyObject_GC_UnTrack(m); + PyObject_GC_UnTrack(m); if (verbose && m->md_name) { PySys_FormatStderr("# destroy %U\n", m->md_name); - } - if (m->md_weaklist != NULL) - PyObject_ClearWeakRefs((PyObject *) m); + } + if (m->md_weaklist != NULL) + PyObject_ClearWeakRefs((PyObject *) m); /* bpo-39824: Don't call m_free() if m_size > 0 and md_state=NULL */ if (m->md_def && m->md_def->m_free && (m->md_def->m_size <= 0 || m->md_state != NULL)) { - m->md_def->m_free(m); + m->md_def->m_free(m); } - Py_XDECREF(m->md_dict); - Py_XDECREF(m->md_name); - if (m->md_state != NULL) - PyMem_FREE(m->md_state); - Py_TYPE(m)->tp_free((PyObject *)m); -} - -static PyObject * -module_repr(PyModuleObject *m) -{ + Py_XDECREF(m->md_dict); + Py_XDECREF(m->md_name); + if (m->md_state != NULL) + PyMem_FREE(m->md_state); + Py_TYPE(m)->tp_free((PyObject *)m); +} + +static PyObject * +module_repr(PyModuleObject *m) +{ PyInterpreterState *interp = _PyInterpreterState_GET(); - - return PyObject_CallMethod(interp->importlib, "_module_repr", "O", m); -} - + + return PyObject_CallMethod(interp->importlib, "_module_repr", "O", m); +} + /* Check if the "_initializing" attribute of the module spec is set to true. Clear the exception and return 0 if spec is NULL. */ @@ -710,23 +710,23 @@ _PyModuleSpec_IsInitializing(PyObject *spec) return 0; } -static PyObject* -module_getattro(PyModuleObject *m, PyObject *name) -{ - PyObject *attr, *mod_name, *getattr; - attr = PyObject_GenericGetAttr((PyObject *)m, name); - if (attr || !PyErr_ExceptionMatches(PyExc_AttributeError)) { - return attr; - } - PyErr_Clear(); - if (m->md_dict) { - _Py_IDENTIFIER(__getattr__); - getattr = _PyDict_GetItemId(m->md_dict, &PyId___getattr__); - if (getattr) { +static PyObject* +module_getattro(PyModuleObject *m, PyObject *name) +{ + PyObject *attr, *mod_name, *getattr; + attr = PyObject_GenericGetAttr((PyObject *)m, name); + if (attr || !PyErr_ExceptionMatches(PyExc_AttributeError)) { + return attr; + } + PyErr_Clear(); + if (m->md_dict) { + _Py_IDENTIFIER(__getattr__); + getattr = _PyDict_GetItemId(m->md_dict, &PyId___getattr__); + if (getattr) { return PyObject_CallOneArg(getattr, name); - } - mod_name = _PyDict_GetItemId(m->md_dict, &PyId___name__); - if (mod_name && PyUnicode_Check(mod_name)) { + } + mod_name = _PyDict_GetItemId(m->md_dict, &PyId___name__); + if (mod_name && PyUnicode_Check(mod_name)) { Py_INCREF(mod_name); PyObject *spec = _PyDict_GetItemId(m->md_dict, &PyId___spec__); Py_XINCREF(spec); @@ -744,126 +744,126 @@ module_getattro(PyModuleObject *m, PyObject *name) } Py_XDECREF(spec); Py_DECREF(mod_name); - return NULL; - } - } - PyErr_Format(PyExc_AttributeError, - "module has no attribute '%U'", name); - return NULL; -} - -static int -module_traverse(PyModuleObject *m, visitproc visit, void *arg) -{ + return NULL; + } + } + PyErr_Format(PyExc_AttributeError, + "module has no attribute '%U'", name); + return NULL; +} + +static int +module_traverse(PyModuleObject *m, visitproc visit, void *arg) +{ /* bpo-39824: Don't call m_traverse() if m_size > 0 and md_state=NULL */ if (m->md_def && m->md_def->m_traverse && (m->md_def->m_size <= 0 || m->md_state != NULL)) { - int res = m->md_def->m_traverse((PyObject*)m, visit, arg); - if (res) - return res; - } - Py_VISIT(m->md_dict); - return 0; -} - -static int -module_clear(PyModuleObject *m) -{ + int res = m->md_def->m_traverse((PyObject*)m, visit, arg); + if (res) + return res; + } + Py_VISIT(m->md_dict); + return 0; +} + +static int +module_clear(PyModuleObject *m) +{ /* bpo-39824: Don't call m_clear() if m_size > 0 and md_state=NULL */ if (m->md_def && m->md_def->m_clear && (m->md_def->m_size <= 0 || m->md_state != NULL)) { - int res = m->md_def->m_clear((PyObject*)m); + int res = m->md_def->m_clear((PyObject*)m); if (PyErr_Occurred()) { PySys_FormatStderr("Exception ignored in m_clear of module%s%V\n", m->md_name ? " " : "", m->md_name, ""); PyErr_WriteUnraisable(NULL); } - if (res) - return res; - } - Py_CLEAR(m->md_dict); - return 0; -} - -static PyObject * -module_dir(PyObject *self, PyObject *args) -{ - _Py_IDENTIFIER(__dict__); + if (res) + return res; + } + Py_CLEAR(m->md_dict); + return 0; +} + +static PyObject * +module_dir(PyObject *self, PyObject *args) +{ + _Py_IDENTIFIER(__dict__); _Py_IDENTIFIER(__dir__); - PyObject *result = NULL; - PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__); - - if (dict != NULL) { - if (PyDict_Check(dict)) { + PyObject *result = NULL; + PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__); + + if (dict != NULL) { + if (PyDict_Check(dict)) { PyObject *dirfunc = _PyDict_GetItemIdWithError(dict, &PyId___dir__); - if (dirfunc) { - result = _PyObject_CallNoArg(dirfunc); - } + if (dirfunc) { + result = _PyObject_CallNoArg(dirfunc); + } else if (!PyErr_Occurred()) { - result = PyDict_Keys(dict); - } - } - else { - const char *name = PyModule_GetName(self); - if (name) - PyErr_Format(PyExc_TypeError, - "%.200s.__dict__ is not a dictionary", - name); - } - } - - Py_XDECREF(dict); - return result; -} - -static PyMethodDef module_methods[] = { - {"__dir__", module_dir, METH_NOARGS, - PyDoc_STR("__dir__() -> list\nspecialized dir() implementation")}, - {0} -}; - -PyTypeObject PyModule_Type = { - PyVarObject_HEAD_INIT(&PyType_Type, 0) - "module", /* tp_name */ - sizeof(PyModuleObject), /* tp_basicsize */ - 0, /* tp_itemsize */ - (destructor)module_dealloc, /* tp_dealloc */ + result = PyDict_Keys(dict); + } + } + else { + const char *name = PyModule_GetName(self); + if (name) + PyErr_Format(PyExc_TypeError, + "%.200s.__dict__ is not a dictionary", + name); + } + } + + Py_XDECREF(dict); + return result; +} + +static PyMethodDef module_methods[] = { + {"__dir__", module_dir, METH_NOARGS, + PyDoc_STR("__dir__() -> list\nspecialized dir() implementation")}, + {0} +}; + +PyTypeObject PyModule_Type = { + PyVarObject_HEAD_INIT(&PyType_Type, 0) + "module", /* tp_name */ + sizeof(PyModuleObject), /* tp_basicsize */ + 0, /* tp_itemsize */ + (destructor)module_dealloc, /* tp_dealloc */ 0, /* tp_vectorcall_offset */ - 0, /* tp_getattr */ - 0, /* tp_setattr */ + 0, /* tp_getattr */ + 0, /* tp_setattr */ 0, /* tp_as_async */ - (reprfunc)module_repr, /* tp_repr */ - 0, /* tp_as_number */ - 0, /* tp_as_sequence */ - 0, /* tp_as_mapping */ - 0, /* tp_hash */ - 0, /* tp_call */ - 0, /* tp_str */ - (getattrofunc)module_getattro, /* tp_getattro */ - PyObject_GenericSetAttr, /* tp_setattro */ - 0, /* tp_as_buffer */ - Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | - Py_TPFLAGS_BASETYPE, /* tp_flags */ - module___init____doc__, /* tp_doc */ - (traverseproc)module_traverse, /* tp_traverse */ - (inquiry)module_clear, /* tp_clear */ - 0, /* tp_richcompare */ - offsetof(PyModuleObject, md_weaklist), /* tp_weaklistoffset */ - 0, /* tp_iter */ - 0, /* tp_iternext */ - module_methods, /* tp_methods */ - module_members, /* tp_members */ - 0, /* tp_getset */ - 0, /* tp_base */ - 0, /* tp_dict */ - 0, /* tp_descr_get */ - 0, /* tp_descr_set */ - offsetof(PyModuleObject, md_dict), /* tp_dictoffset */ - module___init__, /* tp_init */ - PyType_GenericAlloc, /* tp_alloc */ - PyType_GenericNew, /* tp_new */ - PyObject_GC_Del, /* tp_free */ -}; + (reprfunc)module_repr, /* tp_repr */ + 0, /* tp_as_number */ + 0, /* tp_as_sequence */ + 0, /* tp_as_mapping */ + 0, /* tp_hash */ + 0, /* tp_call */ + 0, /* tp_str */ + (getattrofunc)module_getattro, /* tp_getattro */ + PyObject_GenericSetAttr, /* tp_setattro */ + 0, /* tp_as_buffer */ + Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | + Py_TPFLAGS_BASETYPE, /* tp_flags */ + module___init____doc__, /* tp_doc */ + (traverseproc)module_traverse, /* tp_traverse */ + (inquiry)module_clear, /* tp_clear */ + 0, /* tp_richcompare */ + offsetof(PyModuleObject, md_weaklist), /* tp_weaklistoffset */ + 0, /* tp_iter */ + 0, /* tp_iternext */ + module_methods, /* tp_methods */ + module_members, /* tp_members */ + 0, /* tp_getset */ + 0, /* tp_base */ + 0, /* tp_dict */ + 0, /* tp_descr_get */ + 0, /* tp_descr_set */ + offsetof(PyModuleObject, md_dict), /* tp_dictoffset */ + module___init__, /* tp_init */ + PyType_GenericAlloc, /* tp_alloc */ + PyType_GenericNew, /* tp_new */ + PyObject_GC_Del, /* tp_free */ +}; |
