blob: 46c6b53e7ab9cea01fa91e58d56815395b555324 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Module definition and import implementation */
3
Guido van Rossum79f25d91997-04-29 20:08:16 +00004#include "Python.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Guido van Rossumd8faa362007-04-27 19:54:29 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +00009#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000010#include "code.h"
Antoine Pitroubc07a5c2012-07-08 12:01:27 +020011#include "frameobject.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000012#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000013#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000014
Guido van Rossum55a83382000-09-20 20:31:38 +000015#ifdef HAVE_FCNTL_H
16#include <fcntl.h>
17#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000018#ifdef __cplusplus
Brett Cannon9a5b25a2010-03-01 02:09:17 +000019extern "C" {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000020#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000021
Barry Warsaw28a691b2010-04-17 00:19:56 +000022#define CACHEDIR "__pycache__"
Guido van Rossum96774c12000-05-01 20:19:08 +000023
Victor Stinner95872862011-03-07 18:20:56 +010024/* See _PyImport_FixupExtensionObject() below */
Guido van Rossum25ce5661997-08-02 03:10:38 +000025static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000026
Guido van Rossum771c6c81997-10-31 18:37:24 +000027/* This table is defined in config.c: */
28extern struct _inittab _PyImport_Inittab[];
29
30struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000031
Victor Stinnerd0296212011-03-14 14:04:10 -040032static PyObject *initstr = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000033
Brett Cannon4caa61d2014-01-09 19:03:32 -050034/*[clinic input]
35module _imp
36[clinic start generated code]*/
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030037/*[clinic end generated code: output=da39a3ee5e6b4b0d input=9c332475d8686284]*/
Brett Cannonfd4d0502014-05-30 11:21:14 -040038
39#include "clinic/import.c.h"
Brett Cannon4caa61d2014-01-09 19:03:32 -050040
41/*[python input]
42class fs_unicode_converter(CConverter):
43 type = 'PyObject *'
44 converter = 'PyUnicode_FSDecoder'
45
46[python start generated code]*/
Larry Hastings581ee362014-01-28 05:00:08 -080047/*[python end generated code: output=da39a3ee5e6b4b0d input=9d6786230166006e]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -050048
Guido van Rossum1ae940a1995-01-02 19:04:15 +000049/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000050
51void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000052_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000053{
Serhiy Storchaka013bb912014-02-10 18:21:34 +020054 PyInterpreterState *interp = PyThreadState_Get()->interp;
Victor Stinnerd0296212011-03-14 14:04:10 -040055 initstr = PyUnicode_InternFromString("__init__");
56 if (initstr == NULL)
57 Py_FatalError("Can't initialize import variables");
Serhiy Storchaka013bb912014-02-10 18:21:34 +020058 interp->builtins_copy = PyDict_Copy(interp->builtins);
59 if (interp->builtins_copy == NULL)
60 Py_FatalError("Can't backup builtins dict");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000061}
62
Guido van Rossum25ce5661997-08-02 03:10:38 +000063void
Just van Rossum52e14d62002-12-30 22:08:05 +000064_PyImportHooks_Init(void)
65{
Brett Cannonfd074152012-04-14 14:10:13 -040066 PyObject *v, *path_hooks = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000067 int err = 0;
Just van Rossum52e14d62002-12-30 22:08:05 +000068
Brett Cannonfd074152012-04-14 14:10:13 -040069 /* adding sys.path_hooks and sys.path_importer_cache */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000070 v = PyList_New(0);
71 if (v == NULL)
72 goto error;
73 err = PySys_SetObject("meta_path", v);
74 Py_DECREF(v);
75 if (err)
76 goto error;
77 v = PyDict_New();
78 if (v == NULL)
79 goto error;
80 err = PySys_SetObject("path_importer_cache", v);
81 Py_DECREF(v);
82 if (err)
83 goto error;
84 path_hooks = PyList_New(0);
85 if (path_hooks == NULL)
86 goto error;
87 err = PySys_SetObject("path_hooks", path_hooks);
88 if (err) {
Just van Rossum52e14d62002-12-30 22:08:05 +000089 error:
Brett Cannonfd074152012-04-14 14:10:13 -040090 PyErr_Print();
91 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
Nadeem Vawda8f46d652012-05-05 12:27:30 +020092 "or path_importer_cache failed");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000093 }
Brett Cannonfd074152012-04-14 14:10:13 -040094 Py_DECREF(path_hooks);
95}
96
97void
98_PyImportZip_Init(void)
99{
100 PyObject *path_hooks, *zimpimport;
101 int err = 0;
102
103 path_hooks = PySys_GetObject("path_hooks");
Victor Stinner1e53bba2013-07-16 22:26:05 +0200104 if (path_hooks == NULL) {
105 PyErr_SetString(PyExc_RuntimeError, "unable to get sys.path_hooks");
Brett Cannonfd074152012-04-14 14:10:13 -0400106 goto error;
Victor Stinner1e53bba2013-07-16 22:26:05 +0200107 }
Brett Cannonfd074152012-04-14 14:10:13 -0400108
109 if (Py_VerboseFlag)
110 PySys_WriteStderr("# installing zipimport hook\n");
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000111
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000112 zimpimport = PyImport_ImportModule("zipimport");
113 if (zimpimport == NULL) {
114 PyErr_Clear(); /* No zip import module -- okay */
115 if (Py_VerboseFlag)
116 PySys_WriteStderr("# can't import zipimport\n");
117 }
118 else {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200119 _Py_IDENTIFIER(zipimporter);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200120 PyObject *zipimporter = _PyObject_GetAttrId(zimpimport,
121 &PyId_zipimporter);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000122 Py_DECREF(zimpimport);
123 if (zipimporter == NULL) {
124 PyErr_Clear(); /* No zipimporter object -- okay */
125 if (Py_VerboseFlag)
126 PySys_WriteStderr(
127 "# can't import zipimport.zipimporter\n");
128 }
129 else {
Brett Cannon8923a4d2012-04-24 22:03:46 -0400130 /* sys.path_hooks.insert(0, zipimporter) */
131 err = PyList_Insert(path_hooks, 0, zipimporter);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000132 Py_DECREF(zipimporter);
Brett Cannonfd074152012-04-14 14:10:13 -0400133 if (err < 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000134 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -0400135 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000136 if (Py_VerboseFlag)
137 PySys_WriteStderr(
138 "# installed zipimport hook\n");
139 }
140 }
Brett Cannonfd074152012-04-14 14:10:13 -0400141
142 return;
143
144 error:
145 PyErr_Print();
Brett Cannonacf85cd2012-04-29 12:50:03 -0400146 Py_FatalError("initializing zipimport failed");
Just van Rossum52e14d62002-12-30 22:08:05 +0000147}
148
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000149/* Locking primitives to prevent parallel imports of the same module
150 in different threads to return with a partially loaded module.
151 These calls are serialized by the global interpreter lock. */
152
153#ifdef WITH_THREAD
154
Guido van Rossum49b56061998-10-01 20:42:43 +0000155#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000156
Guido van Rossum65d5b571998-12-21 19:32:43 +0000157static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000158static long import_lock_thread = -1;
159static int import_lock_level = 0;
160
Benjamin Peterson0df35a92009-10-04 20:32:25 +0000161void
162_PyImport_AcquireLock(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000163{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000164 long me = PyThread_get_thread_ident();
165 if (me == -1)
166 return; /* Too bad */
167 if (import_lock == NULL) {
168 import_lock = PyThread_allocate_lock();
169 if (import_lock == NULL)
170 return; /* Nothing much we can do. */
171 }
172 if (import_lock_thread == me) {
173 import_lock_level++;
174 return;
175 }
176 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
177 {
178 PyThreadState *tstate = PyEval_SaveThread();
179 PyThread_acquire_lock(import_lock, 1);
180 PyEval_RestoreThread(tstate);
181 }
Antoine Pitrou202b6062012-12-18 22:18:17 +0100182 assert(import_lock_level == 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000183 import_lock_thread = me;
184 import_lock_level = 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000185}
186
Benjamin Peterson0df35a92009-10-04 20:32:25 +0000187int
188_PyImport_ReleaseLock(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000189{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000190 long me = PyThread_get_thread_ident();
191 if (me == -1 || import_lock == NULL)
192 return 0; /* Too bad */
193 if (import_lock_thread != me)
194 return -1;
195 import_lock_level--;
Antoine Pitrou202b6062012-12-18 22:18:17 +0100196 assert(import_lock_level >= 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000197 if (import_lock_level == 0) {
198 import_lock_thread = -1;
199 PyThread_release_lock(import_lock);
200 }
201 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000202}
203
Gregory P. Smith24cec9f2010-03-01 06:18:41 +0000204/* This function is called from PyOS_AfterFork to ensure that newly
205 created child processes do not share locks with the parent.
206 We now acquire the import lock around fork() calls but on some platforms
207 (Solaris 9 and earlier? see isue7242) that still left us with problems. */
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000208
209void
210_PyImport_ReInitLock(void)
211{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000212 if (import_lock != NULL)
213 import_lock = PyThread_allocate_lock();
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000214 if (import_lock_level > 1) {
215 /* Forked as a side effect of import */
216 long me = PyThread_get_thread_ident();
Brett Cannone4710cf2012-11-15 21:39:36 -0500217 /* The following could fail if the lock is already held, but forking as
218 a side-effect of an import is a) rare, b) nuts, and c) difficult to
219 do thanks to the lock only being held when doing individual module
220 locks per import. */
221 PyThread_acquire_lock(import_lock, NOWAIT_LOCK);
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000222 import_lock_thread = me;
223 import_lock_level--;
224 } else {
225 import_lock_thread = -1;
226 import_lock_level = 0;
227 }
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000228}
229
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000230#endif
231
Brett Cannon4caa61d2014-01-09 19:03:32 -0500232/*[clinic input]
233_imp.lock_held
234
235Return True if the import lock is currently held, else False.
236
237On platforms without threads, return False.
238[clinic start generated code]*/
239
Brett Cannon4caa61d2014-01-09 19:03:32 -0500240static PyObject *
241_imp_lock_held_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400242/*[clinic end generated code: output=d7a8cc3a5169081a input=9b088f9b217d9bdf]*/
Tim Peters69232342001-08-30 05:16:13 +0000243{
Tim Peters69232342001-08-30 05:16:13 +0000244#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000246#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000247 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000248#endif
249}
250
Brett Cannon4caa61d2014-01-09 19:03:32 -0500251/*[clinic input]
252_imp.acquire_lock
253
254Acquires the interpreter's import lock for the current thread.
255
256This lock should be used by import hooks to ensure thread-safety when importing
257modules. On platforms without threads, this function does nothing.
258[clinic start generated code]*/
259
Brett Cannon4caa61d2014-01-09 19:03:32 -0500260static PyObject *
261_imp_acquire_lock_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400262/*[clinic end generated code: output=cc143b1d16422cae input=4a2d4381866d5fdc]*/
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000263{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000264#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000265 _PyImport_AcquireLock();
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000266#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000267 Py_INCREF(Py_None);
268 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000269}
270
Brett Cannon4caa61d2014-01-09 19:03:32 -0500271/*[clinic input]
272_imp.release_lock
273
274Release the interpreter's import lock.
275
276On platforms without threads, this function does nothing.
277[clinic start generated code]*/
278
Brett Cannon4caa61d2014-01-09 19:03:32 -0500279static PyObject *
280_imp_release_lock_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400281/*[clinic end generated code: output=74d28e38ebe2b224 input=934fb11516dd778b]*/
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000282{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000283#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000284 if (_PyImport_ReleaseLock() < 0) {
285 PyErr_SetString(PyExc_RuntimeError,
286 "not holding the import lock");
287 return NULL;
288 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000289#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000290 Py_INCREF(Py_None);
291 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000292}
293
Antoine Pitrou8db076c2011-10-30 19:13:55 +0100294void
295_PyImport_Fini(void)
296{
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200297 Py_CLEAR(extensions);
Antoine Pitrou8db076c2011-10-30 19:13:55 +0100298#ifdef WITH_THREAD
299 if (import_lock != NULL) {
300 PyThread_free_lock(import_lock);
301 import_lock = NULL;
302 }
303#endif
304}
305
Guido van Rossum25ce5661997-08-02 03:10:38 +0000306/* Helper for sys */
307
308PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000309PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000310{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000311 PyInterpreterState *interp = PyThreadState_GET()->interp;
312 if (interp->modules == NULL)
313 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
314 return interp->modules;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000315}
316
Guido van Rossum3f5da241990-12-20 15:06:42 +0000317
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000318/* List of names to clear in sys */
319static char* sys_deletes[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000320 "path", "argv", "ps1", "ps2",
321 "last_type", "last_value", "last_traceback",
322 "path_hooks", "path_importer_cache", "meta_path",
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200323 "__interactivehook__",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000324 /* misc stuff */
325 "flags", "float_info",
326 NULL
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000327};
328
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000329static char* sys_files[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000330 "stdin", "__stdin__",
331 "stdout", "__stdout__",
332 "stderr", "__stderr__",
333 NULL
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000334};
335
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000336/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000337
Guido van Rossum3f5da241990-12-20 15:06:42 +0000338void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000339PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000340{
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200341 Py_ssize_t pos;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000342 PyObject *key, *value, *dict;
343 PyInterpreterState *interp = PyThreadState_GET()->interp;
344 PyObject *modules = interp->modules;
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200345 PyObject *weaklist = NULL;
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200346 char **p;
Guido van Rossum758eec01998-01-19 21:58:26 +0000347
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000348 if (modules == NULL)
349 return; /* Already done */
Guido van Rossum758eec01998-01-19 21:58:26 +0000350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000351 /* Delete some special variables first. These are common
352 places where user values hide and people complain when their
353 destructors fail. Since the modules containing them are
354 deleted *last* of all, they would come too late in the normal
355 destruction order. Sigh. */
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000356
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200357 /* XXX Perhaps these precautions are obsolete. Who knows? */
358
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200359 if (Py_VerboseFlag)
360 PySys_WriteStderr("# clear builtins._\n");
361 PyDict_SetItemString(interp->builtins, "_", Py_None);
362
363 for (p = sys_deletes; *p != NULL; p++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000364 if (Py_VerboseFlag)
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200365 PySys_WriteStderr("# clear sys.%s\n", *p);
366 PyDict_SetItemString(interp->sysdict, *p, Py_None);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000367 }
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200368 for (p = sys_files; *p != NULL; p+=2) {
369 if (Py_VerboseFlag)
370 PySys_WriteStderr("# restore sys.%s\n", *p);
371 value = PyDict_GetItemString(interp->sysdict, *(p+1));
372 if (value == NULL)
373 value = Py_None;
374 PyDict_SetItemString(interp->sysdict, *p, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000375 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000376
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200377 /* We prepare a list which will receive (name, weakref) tuples of
378 modules when they are removed from sys.modules. The name is used
379 for diagnosis messages (in verbose mode), while the weakref helps
380 detect those modules which have been held alive. */
381 weaklist = PyList_New(0);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200382 if (weaklist == NULL)
383 PyErr_Clear();
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200384
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200385#define STORE_MODULE_WEAKREF(name, mod) \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200386 if (weaklist != NULL) { \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200387 PyObject *wr = PyWeakref_NewRef(mod, NULL); \
388 if (name && wr) { \
389 PyObject *tup = PyTuple_Pack(2, name, wr); \
390 PyList_Append(weaklist, tup); \
391 Py_XDECREF(tup); \
392 } \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200393 Py_XDECREF(wr); \
394 if (PyErr_Occurred()) \
395 PyErr_Clear(); \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000396 }
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000397
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200398 /* Remove all modules from sys.modules, hoping that garbage collection
399 can reclaim most of them. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 pos = 0;
401 while (PyDict_Next(modules, &pos, &key, &value)) {
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200402 if (PyModule_Check(value)) {
403 if (Py_VerboseFlag && PyUnicode_Check(key))
Victor Stinnerab826d12014-07-07 23:06:15 +0200404 PySys_FormatStderr("# cleanup[2] removing %U\n", key);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200405 STORE_MODULE_WEAKREF(key, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 PyDict_SetItem(modules, key, Py_None);
407 }
408 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000409
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200410 /* Clear the modules dict. */
411 PyDict_Clear(modules);
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200412 /* Restore the original builtins dict, to ensure that any
413 user data gets cleared. */
414 dict = PyDict_Copy(interp->builtins);
415 if (dict == NULL)
416 PyErr_Clear();
417 PyDict_Clear(interp->builtins);
418 if (PyDict_Update(interp->builtins, interp->builtins_copy))
419 PyErr_Clear();
420 Py_XDECREF(dict);
Antoine Pitrou40322e62013-08-11 00:30:09 +0200421 /* Clear module dict copies stored in the interpreter state */
422 _PyState_ClearModules();
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200423 /* Collect references */
424 _PyGC_CollectNoFail();
Antoine Pitrou5f454a02013-05-06 21:15:57 +0200425 /* Dump GC stats before it's too late, since it uses the warnings
426 machinery. */
427 _PyGC_DumpShutdownStats();
428
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200429 /* Now, if there are any modules left alive, clear their globals to
430 minimize potential leaks. All C extension modules actually end
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200431 up here, since they are kept alive in the interpreter state.
432
433 The special treatment of "builtins" here is because even
434 when it's not referenced as a module, its dictionary is
435 referenced by almost every module's __builtins__. Since
436 deleting a module clears its dictionary (even if there are
437 references left to it), we need to delete the "builtins"
438 module last. Likewise, we don't delete sys until the very
439 end because it is implicitly referenced (e.g. by print). */
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200440 if (weaklist != NULL) {
441 Py_ssize_t i, n;
442 n = PyList_GET_SIZE(weaklist);
443 for (i = 0; i < n; i++) {
444 PyObject *tup = PyList_GET_ITEM(weaklist, i);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200445 PyObject *name = PyTuple_GET_ITEM(tup, 0);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200446 PyObject *mod = PyWeakref_GET_OBJECT(PyTuple_GET_ITEM(tup, 1));
447 if (mod == Py_None)
448 continue;
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200449 assert(PyModule_Check(mod));
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200450 dict = PyModule_GetDict(mod);
451 if (dict == interp->builtins || dict == interp->sysdict)
452 continue;
453 Py_INCREF(mod);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200454 if (Py_VerboseFlag && PyUnicode_Check(name))
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200455 PySys_FormatStderr("# cleanup[3] wiping %U\n", name);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200456 _PyModule_Clear(mod);
457 Py_DECREF(mod);
Antoine Pitrou070cb3c2013-05-08 13:23:25 +0200458 }
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200459 Py_DECREF(weaklist);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000460 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000461
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200462 /* Next, delete sys and builtins (in that order) */
463 if (Py_VerboseFlag)
464 PySys_FormatStderr("# cleanup[3] wiping sys\n");
465 _PyModule_ClearDict(interp->sysdict);
466 if (Py_VerboseFlag)
467 PySys_FormatStderr("# cleanup[3] wiping builtins\n");
468 _PyModule_ClearDict(interp->builtins);
469
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200470 /* Clear and delete the modules directory. Actual modules will
471 still be there only if imported during the execution of some
472 destructor. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 interp->modules = NULL;
474 Py_DECREF(modules);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200475
476 /* Once more */
477 _PyGC_CollectNoFail();
478
479#undef STORE_MODULE_WEAKREF
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000480}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000481
482
Barry Warsaw28a691b2010-04-17 00:19:56 +0000483/* Helper for pythonrun.c -- return magic number and tag. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000484
485long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000486PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000487{
Antoine Pitrou44b4b6a2012-07-10 18:27:54 +0200488 long res;
Brett Cannon77b2abd2012-07-09 16:09:00 -0400489 PyInterpreterState *interp = PyThreadState_Get()->interp;
490 PyObject *pyc_magic = PyObject_GetAttrString(interp->importlib,
491 "_RAW_MAGIC_NUMBER");
492 if (pyc_magic == NULL)
493 return -1;
Antoine Pitrou44b4b6a2012-07-10 18:27:54 +0200494 res = PyLong_AsLong(pyc_magic);
Benjamin Peterson66f36592012-07-09 22:21:55 -0700495 Py_DECREF(pyc_magic);
496 return res;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000497}
498
499
Brett Cannon3adc7b72012-07-09 14:22:12 -0400500extern const char * _PySys_ImplCacheTag;
501
Barry Warsaw28a691b2010-04-17 00:19:56 +0000502const char *
503PyImport_GetMagicTag(void)
504{
Brett Cannon3adc7b72012-07-09 14:22:12 -0400505 return _PySys_ImplCacheTag;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000506}
507
Brett Cannon98979b82012-07-02 15:13:11 -0400508
Guido van Rossum25ce5661997-08-02 03:10:38 +0000509/* Magic for extension modules (built-in as well as dynamically
510 loaded). To prevent initializing an extension module more than
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200511 once, we keep a static dictionary 'extensions' keyed by the tuple
512 (module name, module name) (for built-in modules) or by
513 (filename, module name) (for dynamically loaded modules), containing these
514 modules. A copy of the module's dictionary is stored by calling
515 _PyImport_FixupExtensionObject() immediately after the module initialization
516 function succeeds. A copy can be retrieved from there by calling
Victor Stinner95872862011-03-07 18:20:56 +0100517 _PyImport_FindExtensionObject().
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000518
Barry Warsaw7c9627b2010-06-17 18:38:20 +0000519 Modules which do support multiple initialization set their m_size
520 field to a non-negative number (indicating the size of the
521 module-specific state). They are still recorded in the extensions
522 dictionary, to avoid loading shared libraries twice.
Martin v. Löwis1a214512008-06-11 05:26:20 +0000523*/
524
525int
Victor Stinner95872862011-03-07 18:20:56 +0100526_PyImport_FixupExtensionObject(PyObject *mod, PyObject *name,
527 PyObject *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000528{
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500529 PyObject *modules, *dict, *key;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 struct PyModuleDef *def;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500531 int res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000532 if (extensions == NULL) {
533 extensions = PyDict_New();
534 if (extensions == NULL)
535 return -1;
536 }
537 if (mod == NULL || !PyModule_Check(mod)) {
538 PyErr_BadInternalCall();
539 return -1;
540 }
541 def = PyModule_GetDef(mod);
542 if (!def) {
543 PyErr_BadInternalCall();
544 return -1;
545 }
546 modules = PyImport_GetModuleDict();
Victor Stinner95872862011-03-07 18:20:56 +0100547 if (PyDict_SetItem(modules, name, mod) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000548 return -1;
549 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100550 PyDict_DelItem(modules, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 return -1;
552 }
553 if (def->m_size == -1) {
554 if (def->m_base.m_copy) {
555 /* Somebody already imported the module,
556 likely under a different name.
557 XXX this should really not happen. */
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200558 Py_CLEAR(def->m_base.m_copy);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000559 }
560 dict = PyModule_GetDict(mod);
561 if (dict == NULL)
562 return -1;
563 def->m_base.m_copy = PyDict_Copy(dict);
564 if (def->m_base.m_copy == NULL)
565 return -1;
566 }
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500567 key = PyTuple_Pack(2, filename, name);
568 if (key == NULL)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200569 return -1;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500570 res = PyDict_SetItem(extensions, key, (PyObject *)def);
571 Py_DECREF(key);
572 if (res < 0)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200573 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 return 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000575}
576
Victor Stinner49d3f252010-10-17 01:24:53 +0000577int
Serhiy Storchakac6792272013-10-19 21:03:34 +0300578_PyImport_FixupBuiltin(PyObject *mod, const char *name)
Victor Stinner49d3f252010-10-17 01:24:53 +0000579{
580 int res;
Victor Stinner95872862011-03-07 18:20:56 +0100581 PyObject *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100582 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100583 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000584 return -1;
Victor Stinner95872862011-03-07 18:20:56 +0100585 res = _PyImport_FixupExtensionObject(mod, nameobj, nameobj);
586 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000587 return res;
588}
589
Guido van Rossum25ce5661997-08-02 03:10:38 +0000590PyObject *
Victor Stinner95872862011-03-07 18:20:56 +0100591_PyImport_FindExtensionObject(PyObject *name, PyObject *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000592{
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500593 PyObject *mod, *mdict, *key;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 PyModuleDef* def;
595 if (extensions == NULL)
596 return NULL;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500597 key = PyTuple_Pack(2, filename, name);
598 if (key == NULL)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200599 return NULL;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500600 def = (PyModuleDef *)PyDict_GetItem(extensions, key);
601 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000602 if (def == NULL)
603 return NULL;
604 if (def->m_size == -1) {
605 /* Module does not support repeated initialization */
606 if (def->m_base.m_copy == NULL)
607 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100608 mod = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000609 if (mod == NULL)
610 return NULL;
611 mdict = PyModule_GetDict(mod);
612 if (mdict == NULL)
613 return NULL;
614 if (PyDict_Update(mdict, def->m_base.m_copy))
615 return NULL;
616 }
617 else {
618 if (def->m_base.m_init == NULL)
619 return NULL;
620 mod = def->m_base.m_init();
621 if (mod == NULL)
622 return NULL;
Christian Heimes09ca7942013-07-20 14:51:53 +0200623 if (PyDict_SetItem(PyImport_GetModuleDict(), name, mod) == -1) {
624 Py_DECREF(mod);
625 return NULL;
626 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000627 Py_DECREF(mod);
628 }
629 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100630 PyDict_DelItem(PyImport_GetModuleDict(), name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 Py_DECREF(mod);
632 return NULL;
633 }
634 if (Py_VerboseFlag)
Victor Stinner95872862011-03-07 18:20:56 +0100635 PySys_FormatStderr("import %U # previously loaded (%R)\n",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 name, filename);
637 return mod;
Brett Cannon9a5b25a2010-03-01 02:09:17 +0000638
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000639}
640
Victor Stinner49d3f252010-10-17 01:24:53 +0000641PyObject *
Victor Stinner501c09a2011-02-23 00:02:00 +0000642_PyImport_FindBuiltin(const char *name)
Victor Stinner49d3f252010-10-17 01:24:53 +0000643{
Victor Stinner95872862011-03-07 18:20:56 +0100644 PyObject *res, *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100645 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100646 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000647 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100648 res = _PyImport_FindExtensionObject(nameobj, nameobj);
649 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000650 return res;
651}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000652
653/* Get the module object corresponding to a module name.
654 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000655 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000656 Because the former action is most common, THIS DOES NOT RETURN A
657 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000658
Guido van Rossum79f25d91997-04-29 20:08:16 +0000659PyObject *
Victor Stinner27ee0892011-03-04 12:57:09 +0000660PyImport_AddModuleObject(PyObject *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000661{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 PyObject *modules = PyImport_GetModuleDict();
663 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000664
Victor Stinner27ee0892011-03-04 12:57:09 +0000665 if ((m = PyDict_GetItem(modules, name)) != NULL &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000666 PyModule_Check(m))
667 return m;
Victor Stinner27ee0892011-03-04 12:57:09 +0000668 m = PyModule_NewObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000669 if (m == NULL)
670 return NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000671 if (PyDict_SetItem(modules, name, m) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000672 Py_DECREF(m);
673 return NULL;
674 }
675 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000677 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000678}
679
Victor Stinner27ee0892011-03-04 12:57:09 +0000680PyObject *
681PyImport_AddModule(const char *name)
682{
683 PyObject *nameobj, *module;
684 nameobj = PyUnicode_FromString(name);
685 if (nameobj == NULL)
686 return NULL;
687 module = PyImport_AddModuleObject(nameobj);
688 Py_DECREF(nameobj);
689 return module;
690}
691
692
Tim Peters1cd70172004-08-02 03:52:12 +0000693/* Remove name from sys.modules, if it's there. */
694static void
Victor Stinner27ee0892011-03-04 12:57:09 +0000695remove_module(PyObject *name)
Tim Peters1cd70172004-08-02 03:52:12 +0000696{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000697 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner27ee0892011-03-04 12:57:09 +0000698 if (PyDict_GetItem(modules, name) == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000699 return;
Victor Stinner27ee0892011-03-04 12:57:09 +0000700 if (PyDict_DelItem(modules, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 Py_FatalError("import: deleting existing key in"
702 "sys.modules failed");
Tim Peters1cd70172004-08-02 03:52:12 +0000703}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000704
Christian Heimes3b06e532008-01-07 20:12:44 +0000705
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000706/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000707 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
708 * removed from sys.modules, to avoid leaving damaged module objects
709 * in sys.modules. The caller may wish to restore the original
710 * module object (if any) in this case; PyImport_ReloadModule is an
711 * example.
Barry Warsaw28a691b2010-04-17 00:19:56 +0000712 *
713 * Note that PyImport_ExecCodeModuleWithPathnames() is the preferred, richer
714 * interface. The other two exist primarily for backward compatibility.
Tim Peters1cd70172004-08-02 03:52:12 +0000715 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000716PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300717PyImport_ExecCodeModule(const char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000718{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719 return PyImport_ExecCodeModuleWithPathnames(
720 name, co, (char *)NULL, (char *)NULL);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000721}
722
723PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300724PyImport_ExecCodeModuleEx(const char *name, PyObject *co, const char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000725{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000726 return PyImport_ExecCodeModuleWithPathnames(
727 name, co, pathname, (char *)NULL);
Barry Warsaw28a691b2010-04-17 00:19:56 +0000728}
729
730PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300731PyImport_ExecCodeModuleWithPathnames(const char *name, PyObject *co,
732 const char *pathname,
733 const char *cpathname)
Barry Warsaw28a691b2010-04-17 00:19:56 +0000734{
Victor Stinner27ee0892011-03-04 12:57:09 +0000735 PyObject *m = NULL;
736 PyObject *nameobj, *pathobj = NULL, *cpathobj = NULL;
737
738 nameobj = PyUnicode_FromString(name);
739 if (nameobj == NULL)
740 return NULL;
741
Victor Stinner27ee0892011-03-04 12:57:09 +0000742 if (cpathname != NULL) {
743 cpathobj = PyUnicode_DecodeFSDefault(cpathname);
744 if (cpathobj == NULL)
745 goto error;
Brett Cannona6473f92012-07-13 13:57:03 -0400746 }
747 else
Victor Stinner27ee0892011-03-04 12:57:09 +0000748 cpathobj = NULL;
Brett Cannona6473f92012-07-13 13:57:03 -0400749
750 if (pathname != NULL) {
751 pathobj = PyUnicode_DecodeFSDefault(pathname);
752 if (pathobj == NULL)
753 goto error;
754 }
755 else if (cpathobj != NULL) {
756 PyInterpreterState *interp = PyThreadState_GET()->interp;
757 _Py_IDENTIFIER(_get_sourcefile);
758
759 if (interp == NULL) {
760 Py_FatalError("PyImport_ExecCodeModuleWithPathnames: "
761 "no interpreter!");
762 }
763
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -0700764 pathobj = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannona6473f92012-07-13 13:57:03 -0400765 &PyId__get_sourcefile, cpathobj,
766 NULL);
767 if (pathobj == NULL)
768 PyErr_Clear();
769 }
770 else
771 pathobj = NULL;
772
Victor Stinner27ee0892011-03-04 12:57:09 +0000773 m = PyImport_ExecCodeModuleObject(nameobj, co, pathobj, cpathobj);
774error:
775 Py_DECREF(nameobj);
776 Py_XDECREF(pathobj);
777 Py_XDECREF(cpathobj);
778 return m;
779}
780
Brett Cannon18fc4e72014-04-04 10:01:46 -0400781static PyObject *
782module_dict_for_exec(PyObject *name)
Victor Stinner27ee0892011-03-04 12:57:09 +0000783{
Brett Cannon18fc4e72014-04-04 10:01:46 -0400784 PyObject *m, *d = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000785
Victor Stinner27ee0892011-03-04 12:57:09 +0000786 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000787 if (m == NULL)
788 return NULL;
789 /* If the module is being reloaded, we get the old module back
790 and re-use its dict to exec the new code. */
791 d = PyModule_GetDict(m);
792 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
793 if (PyDict_SetItemString(d, "__builtins__",
Brett Cannon18fc4e72014-04-04 10:01:46 -0400794 PyEval_GetBuiltins()) != 0) {
795 remove_module(name);
796 return NULL;
797 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000798 }
Brett Cannon18fc4e72014-04-04 10:01:46 -0400799
Eric Snow08197a42014-05-12 17:54:55 -0600800 return d; /* Return a borrowed reference. */
Brett Cannon18fc4e72014-04-04 10:01:46 -0400801}
802
803static PyObject *
804exec_code_in_module(PyObject *name, PyObject *module_dict, PyObject *code_object)
805{
806 PyObject *modules = PyImport_GetModuleDict();
807 PyObject *v, *m;
808
809 v = PyEval_EvalCode(code_object, module_dict, module_dict);
810 if (v == NULL) {
811 remove_module(name);
812 return NULL;
813 }
814 Py_DECREF(v);
815
816 if ((m = PyDict_GetItem(modules, name)) == NULL) {
817 PyErr_Format(PyExc_ImportError,
818 "Loaded module %R not found in sys.modules",
819 name);
820 return NULL;
821 }
822
823 Py_INCREF(m);
824
825 return m;
826}
827
828PyObject*
829PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
830 PyObject *cpathname)
831{
Eric Snow08197a42014-05-12 17:54:55 -0600832 PyObject *d, *res;
833 PyInterpreterState *interp = PyThreadState_GET()->interp;
834 _Py_IDENTIFIER(_fix_up_module);
Brett Cannon18fc4e72014-04-04 10:01:46 -0400835
836 d = module_dict_for_exec(name);
837 if (d == NULL) {
838 return NULL;
839 }
840
Eric Snow08197a42014-05-12 17:54:55 -0600841 if (pathname == NULL) {
842 pathname = ((PyCodeObject *)co)->co_filename;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000843 }
Eric Snow08197a42014-05-12 17:54:55 -0600844 res = _PyObject_CallMethodIdObjArgs(interp->importlib,
845 &PyId__fix_up_module,
846 d, name, pathname, cpathname, NULL);
847 if (res != NULL) {
Eric Snow58cfdd82014-05-29 12:31:39 -0600848 Py_DECREF(res);
Eric Snow08197a42014-05-12 17:54:55 -0600849 res = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000850 }
Eric Snow08197a42014-05-12 17:54:55 -0600851 return res;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000852}
853
854
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000855static void
856update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
857{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 PyObject *constants, *tmp;
859 Py_ssize_t i, n;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 if (PyUnicode_Compare(co->co_filename, oldname))
862 return;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000863
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000864 tmp = co->co_filename;
865 co->co_filename = newname;
866 Py_INCREF(co->co_filename);
867 Py_DECREF(tmp);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 constants = co->co_consts;
870 n = PyTuple_GET_SIZE(constants);
871 for (i = 0; i < n; i++) {
872 tmp = PyTuple_GET_ITEM(constants, i);
873 if (PyCode_Check(tmp))
874 update_code_filenames((PyCodeObject *)tmp,
875 oldname, newname);
876 }
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000877}
878
Victor Stinner2f42ae52011-03-20 00:41:24 +0100879static void
880update_compiled_module(PyCodeObject *co, PyObject *newname)
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000881{
Victor Stinner2f42ae52011-03-20 00:41:24 +0100882 PyObject *oldname;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000883
Victor Stinner2f42ae52011-03-20 00:41:24 +0100884 if (PyUnicode_Compare(co->co_filename, newname) == 0)
885 return;
Hirokazu Yamamoto4f447fb2009-03-04 01:52:10 +0000886
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000887 oldname = co->co_filename;
888 Py_INCREF(oldname);
889 update_code_filenames(co, oldname, newname);
890 Py_DECREF(oldname);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000891}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000892
Brett Cannon4caa61d2014-01-09 19:03:32 -0500893/*[clinic input]
894_imp._fix_co_filename
895
896 code: object(type="PyCodeObject *", subclass_of="&PyCode_Type")
897 Code object to change.
898
899 path: unicode
900 File path to use.
901 /
902
903Changes code.co_filename to specify the passed-in file path.
904[clinic start generated code]*/
905
Brett Cannon4caa61d2014-01-09 19:03:32 -0500906static PyObject *
Larry Hastings89964c42015-04-14 18:07:59 -0400907_imp__fix_co_filename_impl(PyModuleDef *module, PyCodeObject *code,
908 PyObject *path)
909/*[clinic end generated code: output=f4db56aac0a1327f input=895ba50e78b82f05]*/
Brett Cannon442c9b92011-03-23 16:14:42 -0700910
Brett Cannon4caa61d2014-01-09 19:03:32 -0500911{
Brett Cannona6dec2e2014-01-10 07:43:55 -0500912 update_compiled_module(code, path);
Brett Cannon442c9b92011-03-23 16:14:42 -0700913
914 Py_RETURN_NONE;
915}
916
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000917
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000918/* Forward */
Benjamin Peterson6fba3db2013-03-18 23:13:31 -0700919static const struct _frozen * find_frozen(PyObject *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000920
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000921
922/* Helper to test for built-in module */
923
924static int
Victor Stinner95872862011-03-07 18:20:56 +0100925is_builtin(PyObject *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000926{
Victor Stinner95872862011-03-07 18:20:56 +0100927 int i, cmp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000928 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Victor Stinner95872862011-03-07 18:20:56 +0100929 cmp = PyUnicode_CompareWithASCIIString(name, PyImport_Inittab[i].name);
930 if (cmp == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000931 if (PyImport_Inittab[i].initfunc == NULL)
932 return -1;
933 else
934 return 1;
935 }
936 }
937 return 0;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000938}
939
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000940
Just van Rossum52e14d62002-12-30 22:08:05 +0000941/* Return an importer object for a sys.path/pkg.__path__ item 'p',
942 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +0000943 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +0000944 that can handle the path item. Return None if no hook could;
945 this tells our caller it should fall back to the builtin
946 import mechanism. Cache the result in path_importer_cache.
947 Returns a borrowed reference. */
948
949static PyObject *
950get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000951 PyObject *p)
Just van Rossum52e14d62002-12-30 22:08:05 +0000952{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000953 PyObject *importer;
954 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +0000955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000956 /* These conditions are the caller's responsibility: */
957 assert(PyList_Check(path_hooks));
958 assert(PyDict_Check(path_importer_cache));
Just van Rossum52e14d62002-12-30 22:08:05 +0000959
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000960 nhooks = PyList_Size(path_hooks);
961 if (nhooks < 0)
962 return NULL; /* Shouldn't happen */
Just van Rossum52e14d62002-12-30 22:08:05 +0000963
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000964 importer = PyDict_GetItem(path_importer_cache, p);
965 if (importer != NULL)
966 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +0000967
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000968 /* set path_importer_cache[p] to None to avoid recursion */
969 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
970 return NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +0000971
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000972 for (j = 0; j < nhooks; j++) {
973 PyObject *hook = PyList_GetItem(path_hooks, j);
974 if (hook == NULL)
975 return NULL;
976 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
977 if (importer != NULL)
978 break;
Just van Rossum52e14d62002-12-30 22:08:05 +0000979
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000980 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
981 return NULL;
982 }
983 PyErr_Clear();
984 }
985 if (importer == NULL) {
Brett Cannonaa936422012-04-27 15:30:58 -0400986 return Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000987 }
988 if (importer != NULL) {
989 int err = PyDict_SetItem(path_importer_cache, p, importer);
990 Py_DECREF(importer);
991 if (err != 0)
992 return NULL;
993 }
994 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +0000995}
996
Christian Heimes9cd17752007-11-18 19:35:23 +0000997PyAPI_FUNC(PyObject *)
998PyImport_GetImporter(PyObject *path) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000999 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
Christian Heimes9cd17752007-11-18 19:35:23 +00001000
Victor Stinner1e53bba2013-07-16 22:26:05 +02001001 path_importer_cache = PySys_GetObject("path_importer_cache");
1002 path_hooks = PySys_GetObject("path_hooks");
1003 if (path_importer_cache != NULL && path_hooks != NULL) {
1004 importer = get_path_importer(path_importer_cache,
1005 path_hooks, path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001006 }
1007 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1008 return importer;
Christian Heimes9cd17752007-11-18 19:35:23 +00001009}
1010
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001011
Victor Stinner95872862011-03-07 18:20:56 +01001012static int init_builtin(PyObject *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001013
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001014/* Initialize a built-in module.
Thomas Wouters89f507f2006-12-13 04:49:30 +00001015 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001016 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001017
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001018static int
Victor Stinner95872862011-03-07 18:20:56 +01001019init_builtin(PyObject *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001020{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 struct _inittab *p;
Victor Stinner5eb4f592013-11-14 22:38:52 +01001022 PyObject *mod;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001023
Victor Stinner5eb4f592013-11-14 22:38:52 +01001024 mod = _PyImport_FindExtensionObject(name, name);
1025 if (PyErr_Occurred())
1026 return -1;
1027 if (mod != NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028 return 1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001029
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001030 for (p = PyImport_Inittab; p->name != NULL; p++) {
1031 PyObject *mod;
Antoine Pitrou6c40eb72012-01-18 20:16:09 +01001032 PyModuleDef *def;
Victor Stinner95872862011-03-07 18:20:56 +01001033 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001034 if (p->initfunc == NULL) {
1035 PyErr_Format(PyExc_ImportError,
Victor Stinner95872862011-03-07 18:20:56 +01001036 "Cannot re-init internal module %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001037 name);
1038 return -1;
1039 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001040 mod = (*p->initfunc)();
1041 if (mod == 0)
1042 return -1;
Antoine Pitrou6c40eb72012-01-18 20:16:09 +01001043 /* Remember pointer to module init function. */
1044 def = PyModule_GetDef(mod);
1045 def->m_base.m_init = p->initfunc;
Victor Stinner95872862011-03-07 18:20:56 +01001046 if (_PyImport_FixupExtensionObject(mod, name, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001047 return -1;
1048 /* FixupExtension has put the module into sys.modules,
1049 so we can release our own reference. */
1050 Py_DECREF(mod);
1051 return 1;
1052 }
1053 }
1054 return 0;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001055}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001056
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001057
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001058/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001059
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001060static const struct _frozen *
Victor Stinner53dc7352011-03-20 01:50:21 +01001061find_frozen(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001062{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001063 const struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001064
Victor Stinner53dc7352011-03-20 01:50:21 +01001065 if (name == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 return NULL;
Benjamin Petersond968e272008-11-05 22:48:33 +00001067
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001068 for (p = PyImport_FrozenModules; ; p++) {
1069 if (p->name == NULL)
1070 return NULL;
Victor Stinner53dc7352011-03-20 01:50:21 +01001071 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001072 break;
1073 }
1074 return p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001075}
1076
Guido van Rossum79f25d91997-04-29 20:08:16 +00001077static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001078get_frozen_object(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001079{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001080 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001081 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001083 if (p == NULL) {
1084 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001085 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001086 name);
1087 return NULL;
1088 }
1089 if (p->code == NULL) {
1090 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001091 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001092 name);
1093 return NULL;
1094 }
1095 size = p->size;
1096 if (size < 0)
1097 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001098 return PyMarshal_ReadObjectFromString((const char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001099}
1100
Brett Cannon8d110132009-03-15 02:20:16 +00001101static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001102is_frozen_package(PyObject *name)
Brett Cannon8d110132009-03-15 02:20:16 +00001103{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001104 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001105 int size;
Brett Cannon8d110132009-03-15 02:20:16 +00001106
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001107 if (p == NULL) {
1108 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001109 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001110 name);
1111 return NULL;
1112 }
Brett Cannon8d110132009-03-15 02:20:16 +00001113
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001114 size = p->size;
Brett Cannon8d110132009-03-15 02:20:16 +00001115
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001116 if (size < 0)
1117 Py_RETURN_TRUE;
1118 else
1119 Py_RETURN_FALSE;
Brett Cannon8d110132009-03-15 02:20:16 +00001120}
1121
1122
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001123/* Initialize a frozen module.
Brett Cannon3c2ac442009-03-08 20:49:47 +00001124 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001125 an exception set if the initialization failed.
1126 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001127
1128int
Victor Stinner53dc7352011-03-20 01:50:21 +01001129PyImport_ImportFrozenModuleObject(PyObject *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001130{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001131 const struct _frozen *p;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001132 PyObject *co, *m, *d;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001133 int ispackage;
1134 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001135
Victor Stinner53dc7352011-03-20 01:50:21 +01001136 p = find_frozen(name);
1137
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001138 if (p == NULL)
1139 return 0;
1140 if (p->code == NULL) {
1141 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001142 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001143 name);
1144 return -1;
1145 }
1146 size = p->size;
1147 ispackage = (size < 0);
1148 if (ispackage)
1149 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001150 co = PyMarshal_ReadObjectFromString((const char *)p->code, size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001151 if (co == NULL)
1152 return -1;
1153 if (!PyCode_Check(co)) {
1154 PyErr_Format(PyExc_TypeError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001155 "frozen object %R is not a code object",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001156 name);
1157 goto err_return;
1158 }
1159 if (ispackage) {
Brett Cannon3e0651b2013-05-31 23:18:39 -04001160 /* Set __path__ to the empty list */
Brett Cannon18fc4e72014-04-04 10:01:46 -04001161 PyObject *l;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001162 int err;
Victor Stinner53dc7352011-03-20 01:50:21 +01001163 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 if (m == NULL)
1165 goto err_return;
1166 d = PyModule_GetDict(m);
Brett Cannon3e0651b2013-05-31 23:18:39 -04001167 l = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001168 if (l == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001169 goto err_return;
1170 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001171 err = PyDict_SetItemString(d, "__path__", l);
1172 Py_DECREF(l);
1173 if (err != 0)
1174 goto err_return;
1175 }
Brett Cannon18fc4e72014-04-04 10:01:46 -04001176 d = module_dict_for_exec(name);
1177 if (d == NULL) {
Victor Stinner53dc7352011-03-20 01:50:21 +01001178 goto err_return;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001179 }
1180 m = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001181 if (m == NULL)
1182 goto err_return;
1183 Py_DECREF(co);
1184 Py_DECREF(m);
1185 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001186err_return:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001187 Py_DECREF(co);
1188 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001189}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001190
Victor Stinner53dc7352011-03-20 01:50:21 +01001191int
Serhiy Storchakac6792272013-10-19 21:03:34 +03001192PyImport_ImportFrozenModule(const char *name)
Victor Stinner53dc7352011-03-20 01:50:21 +01001193{
1194 PyObject *nameobj;
1195 int ret;
1196 nameobj = PyUnicode_InternFromString(name);
1197 if (nameobj == NULL)
1198 return -1;
1199 ret = PyImport_ImportFrozenModuleObject(nameobj);
1200 Py_DECREF(nameobj);
1201 return ret;
1202}
1203
Guido van Rossum74e6a111994-08-29 12:54:38 +00001204
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001205/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001206 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001207
Guido van Rossum79f25d91997-04-29 20:08:16 +00001208PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001209PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001210{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001211 PyObject *pname;
1212 PyObject *result;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 pname = PyUnicode_FromString(name);
1215 if (pname == NULL)
1216 return NULL;
1217 result = PyImport_Import(pname);
1218 Py_DECREF(pname);
1219 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001220}
1221
Christian Heimes072c0f12008-01-03 23:01:04 +00001222/* Import a module without blocking
1223 *
1224 * At first it tries to fetch the module from sys.modules. If the module was
1225 * never loaded before it loads it with PyImport_ImportModule() unless another
1226 * thread holds the import lock. In the latter case the function raises an
1227 * ImportError instead of blocking.
1228 *
1229 * Returns the module object with incremented ref count.
1230 */
1231PyObject *
1232PyImport_ImportModuleNoBlock(const char *name)
1233{
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001234 return PyImport_ImportModule(name);
Christian Heimes072c0f12008-01-03 23:01:04 +00001235}
1236
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001237
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001238/* Remove importlib frames from the traceback,
1239 * except in Verbose mode. */
1240static void
1241remove_importlib_frames(void)
1242{
1243 const char *importlib_filename = "<frozen importlib._bootstrap>";
Nick Coghlan42c07662012-07-31 21:14:18 +10001244 const char *remove_frames = "_call_with_frames_removed";
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001245 int always_trim = 0;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001246 int in_importlib = 0;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001247 PyObject *exception, *value, *base_tb, *tb;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001248 PyObject **prev_link, **outer_link = NULL;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001249
1250 /* Synopsis: if it's an ImportError, we trim all importlib chunks
Nick Coghlan42c07662012-07-31 21:14:18 +10001251 from the traceback. We always trim chunks
1252 which end with a call to "_call_with_frames_removed". */
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001253
1254 PyErr_Fetch(&exception, &value, &base_tb);
1255 if (!exception || Py_VerboseFlag)
1256 goto done;
1257 if (PyType_IsSubtype((PyTypeObject *) exception,
1258 (PyTypeObject *) PyExc_ImportError))
1259 always_trim = 1;
1260
1261 prev_link = &base_tb;
1262 tb = base_tb;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001263 while (tb != NULL) {
1264 PyTracebackObject *traceback = (PyTracebackObject *)tb;
1265 PyObject *next = (PyObject *) traceback->tb_next;
1266 PyFrameObject *frame = traceback->tb_frame;
1267 PyCodeObject *code = frame->f_code;
1268 int now_in_importlib;
1269
1270 assert(PyTraceBack_Check(tb));
1271 now_in_importlib = (PyUnicode_CompareWithASCIIString(
1272 code->co_filename,
1273 importlib_filename) == 0);
1274 if (now_in_importlib && !in_importlib) {
1275 /* This is the link to this chunk of importlib tracebacks */
1276 outer_link = prev_link;
1277 }
1278 in_importlib = now_in_importlib;
1279
1280 if (in_importlib &&
1281 (always_trim ||
Nick Coghlan42c07662012-07-31 21:14:18 +10001282 PyUnicode_CompareWithASCIIString(code->co_name,
1283 remove_frames) == 0)) {
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001284 PyObject *tmp = *outer_link;
1285 *outer_link = next;
1286 Py_XINCREF(next);
1287 Py_DECREF(tmp);
1288 prev_link = outer_link;
1289 }
1290 else {
1291 prev_link = (PyObject **) &traceback->tb_next;
1292 }
1293 tb = next;
1294 }
1295done:
1296 PyErr_Restore(exception, value, base_tb);
1297}
1298
1299
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001300PyObject *
Brett Cannonfd074152012-04-14 14:10:13 -04001301PyImport_ImportModuleLevelObject(PyObject *name, PyObject *given_globals,
1302 PyObject *locals, PyObject *given_fromlist,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001303 int level)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001304{
Brett Cannonfd074152012-04-14 14:10:13 -04001305 _Py_IDENTIFIER(__import__);
Eric Snowb523f842013-11-22 09:05:39 -07001306 _Py_IDENTIFIER(__spec__);
1307 _Py_IDENTIFIER(_initializing);
Brett Cannonfd074152012-04-14 14:10:13 -04001308 _Py_IDENTIFIER(__package__);
1309 _Py_IDENTIFIER(__path__);
1310 _Py_IDENTIFIER(__name__);
1311 _Py_IDENTIFIER(_find_and_load);
1312 _Py_IDENTIFIER(_handle_fromlist);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001313 _Py_IDENTIFIER(_lock_unlock_module);
Brett Cannonfd074152012-04-14 14:10:13 -04001314 _Py_static_string(single_dot, ".");
1315 PyObject *abs_name = NULL;
1316 PyObject *builtins_import = NULL;
1317 PyObject *final_mod = NULL;
1318 PyObject *mod = NULL;
1319 PyObject *package = NULL;
1320 PyObject *globals = NULL;
1321 PyObject *fromlist = NULL;
1322 PyInterpreterState *interp = PyThreadState_GET()->interp;
1323
1324 /* Make sure to use default values so as to not have
1325 PyObject_CallMethodObjArgs() truncate the parameter list because of a
1326 NULL argument. */
1327 if (given_globals == NULL) {
1328 globals = PyDict_New();
1329 if (globals == NULL) {
1330 goto error;
1331 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001332 }
Brett Cannonfd074152012-04-14 14:10:13 -04001333 else {
1334 /* Only have to care what given_globals is if it will be used
Brett Cannonecfefb72012-08-05 19:24:57 -04001335 for something. */
Brett Cannonfd074152012-04-14 14:10:13 -04001336 if (level > 0 && !PyDict_Check(given_globals)) {
1337 PyErr_SetString(PyExc_TypeError, "globals must be a dict");
1338 goto error;
1339 }
1340 globals = given_globals;
1341 Py_INCREF(globals);
1342 }
1343
1344 if (given_fromlist == NULL) {
1345 fromlist = PyList_New(0);
1346 if (fromlist == NULL) {
1347 goto error;
1348 }
1349 }
1350 else {
1351 fromlist = given_fromlist;
1352 Py_INCREF(fromlist);
1353 }
1354 if (name == NULL) {
1355 PyErr_SetString(PyExc_ValueError, "Empty module name");
1356 goto error;
1357 }
1358
1359 /* The below code is importlib.__import__() & _gcd_import(), ported to C
1360 for added performance. */
1361
1362 if (!PyUnicode_Check(name)) {
1363 PyErr_SetString(PyExc_TypeError, "module name must be a string");
1364 goto error;
1365 }
1366 else if (PyUnicode_READY(name) < 0) {
1367 goto error;
1368 }
1369 if (level < 0) {
1370 PyErr_SetString(PyExc_ValueError, "level must be >= 0");
1371 goto error;
1372 }
1373 else if (level > 0) {
1374 package = _PyDict_GetItemId(globals, &PyId___package__);
1375 if (package != NULL && package != Py_None) {
1376 Py_INCREF(package);
1377 if (!PyUnicode_Check(package)) {
1378 PyErr_SetString(PyExc_TypeError, "package must be a string");
1379 goto error;
1380 }
1381 }
1382 else {
Brett Cannon273323c2012-04-17 19:05:11 -04001383 package = _PyDict_GetItemId(globals, &PyId___name__);
Brett Cannonfd074152012-04-14 14:10:13 -04001384 if (package == NULL) {
Brett Cannon273323c2012-04-17 19:05:11 -04001385 PyErr_SetString(PyExc_KeyError, "'__name__' not in globals");
Brett Cannonfd074152012-04-14 14:10:13 -04001386 goto error;
1387 }
1388 else if (!PyUnicode_Check(package)) {
1389 PyErr_SetString(PyExc_TypeError, "__name__ must be a string");
1390 }
1391 Py_INCREF(package);
1392
1393 if (_PyDict_GetItemId(globals, &PyId___path__) == NULL) {
Brett Cannon740fce02012-04-14 14:23:49 -04001394 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001395 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1396 if (borrowed_dot == NULL) {
1397 goto error;
1398 }
Brett Cannon740fce02012-04-14 14:23:49 -04001399 partition = PyUnicode_RPartition(package, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001400 Py_DECREF(package);
1401 if (partition == NULL) {
1402 goto error;
1403 }
1404 package = PyTuple_GET_ITEM(partition, 0);
1405 Py_INCREF(package);
1406 Py_DECREF(partition);
1407 }
1408 }
1409
1410 if (PyDict_GetItem(interp->modules, package) == NULL) {
1411 PyErr_Format(PyExc_SystemError,
1412 "Parent module %R not loaded, cannot perform relative "
1413 "import", package);
1414 goto error;
1415 }
1416 }
1417 else { /* level == 0 */
1418 if (PyUnicode_GET_LENGTH(name) == 0) {
1419 PyErr_SetString(PyExc_ValueError, "Empty module name");
1420 goto error;
1421 }
1422 package = Py_None;
1423 Py_INCREF(package);
1424 }
1425
1426 if (level > 0) {
1427 Py_ssize_t last_dot = PyUnicode_GET_LENGTH(package);
1428 PyObject *base = NULL;
1429 int level_up = 1;
1430
1431 for (level_up = 1; level_up < level; level_up += 1) {
1432 last_dot = PyUnicode_FindChar(package, '.', 0, last_dot, -1);
1433 if (last_dot == -2) {
1434 goto error;
1435 }
1436 else if (last_dot == -1) {
1437 PyErr_SetString(PyExc_ValueError,
1438 "attempted relative import beyond top-level "
1439 "package");
1440 goto error;
1441 }
1442 }
Victor Stinner22af2592013-11-13 12:11:36 +01001443
Brett Cannonfd074152012-04-14 14:10:13 -04001444 base = PyUnicode_Substring(package, 0, last_dot);
Victor Stinner22af2592013-11-13 12:11:36 +01001445 if (base == NULL)
1446 goto error;
1447
Brett Cannonfd074152012-04-14 14:10:13 -04001448 if (PyUnicode_GET_LENGTH(name) > 0) {
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001449 PyObject *borrowed_dot, *seq = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001450
1451 borrowed_dot = _PyUnicode_FromId(&single_dot);
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001452 seq = PyTuple_Pack(2, base, name);
1453 Py_DECREF(base);
Brett Cannonfd074152012-04-14 14:10:13 -04001454 if (borrowed_dot == NULL || seq == NULL) {
1455 goto error;
1456 }
1457
1458 abs_name = PyUnicode_Join(borrowed_dot, seq);
1459 Py_DECREF(seq);
1460 if (abs_name == NULL) {
1461 goto error;
1462 }
1463 }
1464 else {
1465 abs_name = base;
1466 }
1467 }
1468 else {
1469 abs_name = name;
1470 Py_INCREF(abs_name);
1471 }
1472
Brian Curtine6b299f2012-04-14 14:19:33 -05001473#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001474 _PyImport_AcquireLock();
1475#endif
1476 /* From this point forward, goto error_with_unlock! */
1477 if (PyDict_Check(globals)) {
1478 builtins_import = _PyDict_GetItemId(globals, &PyId___import__);
1479 }
1480 if (builtins_import == NULL) {
1481 builtins_import = _PyDict_GetItemId(interp->builtins, &PyId___import__);
1482 if (builtins_import == NULL) {
Benjamin Peterson7d110042013-04-29 09:08:14 -04001483 PyErr_SetString(PyExc_ImportError, "__import__ not found");
1484 goto error_with_unlock;
Brett Cannonfd074152012-04-14 14:10:13 -04001485 }
1486 }
1487 Py_INCREF(builtins_import);
1488
1489 mod = PyDict_GetItem(interp->modules, abs_name);
1490 if (mod == Py_None) {
Brett Cannon27fc5282012-04-15 14:15:31 -04001491 PyObject *msg = PyUnicode_FromFormat("import of %R halted; "
1492 "None in sys.modules", abs_name);
1493 if (msg != NULL) {
Brett Cannon82da8882013-07-04 17:48:16 -04001494 PyErr_SetImportError(msg, abs_name, NULL);
Brian Curtin09b86d12012-04-17 16:57:09 -05001495 Py_DECREF(msg);
Brett Cannon27fc5282012-04-15 14:15:31 -04001496 }
Antoine Pitrou71382cb2012-04-16 18:48:49 +02001497 mod = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001498 goto error_with_unlock;
1499 }
1500 else if (mod != NULL) {
Eric Snowb523f842013-11-22 09:05:39 -07001501 PyObject *value = NULL;
1502 PyObject *spec;
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001503 int initializing = 0;
1504
Brett Cannonfd074152012-04-14 14:10:13 -04001505 Py_INCREF(mod);
Antoine Pitrou03989852012-08-28 00:24:52 +02001506 /* Optimization: only call _bootstrap._lock_unlock_module() if
Eric Snowb523f842013-11-22 09:05:39 -07001507 __spec__._initializing is true.
1508 NOTE: because of this, initializing must be set *before*
Antoine Pitrou03989852012-08-28 00:24:52 +02001509 stuffing the new module in sys.modules.
1510 */
Eric Snowb523f842013-11-22 09:05:39 -07001511 spec = _PyObject_GetAttrId(mod, &PyId___spec__);
1512 if (spec != NULL) {
1513 value = _PyObject_GetAttrId(spec, &PyId__initializing);
1514 Py_DECREF(spec);
1515 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001516 if (value == NULL)
1517 PyErr_Clear();
1518 else {
1519 initializing = PyObject_IsTrue(value);
1520 Py_DECREF(value);
1521 if (initializing == -1)
1522 PyErr_Clear();
1523 }
1524 if (initializing > 0) {
1525 /* _bootstrap._lock_unlock_module() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001526 value = _PyObject_CallMethodIdObjArgs(interp->importlib,
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001527 &PyId__lock_unlock_module, abs_name,
1528 NULL);
1529 if (value == NULL)
1530 goto error;
1531 Py_DECREF(value);
1532 }
1533 else {
1534#ifdef WITH_THREAD
1535 if (_PyImport_ReleaseLock() < 0) {
1536 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1537 goto error;
1538 }
1539#endif
1540 }
Brett Cannonfd074152012-04-14 14:10:13 -04001541 }
1542 else {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001543 /* _bootstrap._find_and_load() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001544 mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001545 &PyId__find_and_load, abs_name,
1546 builtins_import, NULL);
1547 if (mod == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001548 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001549 }
1550 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001551 /* From now on we don't hold the import lock anymore. */
Brett Cannonfd074152012-04-14 14:10:13 -04001552
1553 if (PyObject_Not(fromlist)) {
1554 if (level == 0 || PyUnicode_GET_LENGTH(name) > 0) {
1555 PyObject *front = NULL;
Brian Curtine6b299f2012-04-14 14:19:33 -05001556 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001557 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1558
1559 if (borrowed_dot == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001560 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001561 }
1562
Brian Curtine6b299f2012-04-14 14:19:33 -05001563 partition = PyUnicode_Partition(name, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001564 if (partition == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001565 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001566 }
1567
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001568 if (PyUnicode_GET_LENGTH(PyTuple_GET_ITEM(partition, 1)) == 0) {
1569 /* No dot in module name, simple exit */
1570 Py_DECREF(partition);
1571 final_mod = mod;
1572 Py_INCREF(mod);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001573 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001574 }
1575
Brett Cannonfd074152012-04-14 14:10:13 -04001576 front = PyTuple_GET_ITEM(partition, 0);
1577 Py_INCREF(front);
1578 Py_DECREF(partition);
1579
1580 if (level == 0) {
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001581 final_mod = PyObject_CallFunctionObjArgs(builtins_import, front, NULL);
Antoine Pitroub78174c2012-05-06 17:15:23 +02001582 Py_DECREF(front);
Brett Cannonfd074152012-04-14 14:10:13 -04001583 }
1584 else {
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001585 Py_ssize_t cut_off = PyUnicode_GET_LENGTH(name) -
1586 PyUnicode_GET_LENGTH(front);
1587 Py_ssize_t abs_name_len = PyUnicode_GET_LENGTH(abs_name);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001588 PyObject *to_return = PyUnicode_Substring(abs_name, 0,
Brett Cannonfd074152012-04-14 14:10:13 -04001589 abs_name_len - cut_off);
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001590 Py_DECREF(front);
1591 if (to_return == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001592 goto error;
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001593 }
Brett Cannonfd074152012-04-14 14:10:13 -04001594
1595 final_mod = PyDict_GetItem(interp->modules, to_return);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001596 if (final_mod == NULL) {
1597 PyErr_Format(PyExc_KeyError,
1598 "%R not in sys.modules as expected",
1599 to_return);
1600 }
1601 else {
1602 Py_INCREF(final_mod);
1603 }
Antoine Pitroub78174c2012-05-06 17:15:23 +02001604 Py_DECREF(to_return);
Brett Cannonfd074152012-04-14 14:10:13 -04001605 }
1606 }
1607 else {
1608 final_mod = mod;
1609 Py_INCREF(mod);
1610 }
1611 }
1612 else {
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001613 final_mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001614 &PyId__handle_fromlist, mod,
1615 fromlist, builtins_import,
1616 NULL);
1617 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001618 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001619
Brett Cannonfd074152012-04-14 14:10:13 -04001620 error_with_unlock:
Brian Curtine6b299f2012-04-14 14:19:33 -05001621#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001622 if (_PyImport_ReleaseLock() < 0) {
1623 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1624 }
1625#endif
1626 error:
1627 Py_XDECREF(abs_name);
1628 Py_XDECREF(builtins_import);
1629 Py_XDECREF(mod);
1630 Py_XDECREF(package);
1631 Py_XDECREF(globals);
1632 Py_XDECREF(fromlist);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001633 if (final_mod == NULL)
1634 remove_importlib_frames();
Brett Cannonfd074152012-04-14 14:10:13 -04001635 return final_mod;
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001636}
1637
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001638PyObject *
Benjamin Peterson04778a82011-05-25 09:29:00 -05001639PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001640 PyObject *fromlist, int level)
1641{
1642 PyObject *nameobj, *mod;
1643 nameobj = PyUnicode_FromString(name);
1644 if (nameobj == NULL)
1645 return NULL;
1646 mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
1647 fromlist, level);
1648 Py_DECREF(nameobj);
1649 return mod;
1650}
1651
1652
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001653/* Re-import a module of any kind and return its module object, WITH
1654 INCREMENTED REFERENCE COUNT */
1655
Guido van Rossum79f25d91997-04-29 20:08:16 +00001656PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001657PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001658{
Brett Cannon62228db2012-04-29 14:38:11 -04001659 _Py_IDENTIFIER(reload);
1660 PyObject *reloaded_module = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001661 PyObject *modules = PyImport_GetModuleDict();
Brett Cannon62228db2012-04-29 14:38:11 -04001662 PyObject *imp = PyDict_GetItemString(modules, "imp");
1663 if (imp == NULL) {
1664 imp = PyImport_ImportModule("imp");
1665 if (imp == NULL) {
1666 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001667 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001668 }
Brett Cannon62228db2012-04-29 14:38:11 -04001669 else {
1670 Py_INCREF(imp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001671 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001672
Brett Cannon62228db2012-04-29 14:38:11 -04001673 reloaded_module = _PyObject_CallMethodId(imp, &PyId_reload, "O", m);
1674 Py_DECREF(imp);
1675 return reloaded_module;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001676}
1677
1678
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001679/* Higher-level import emulator which emulates the "import" statement
1680 more accurately -- it invokes the __import__() function from the
1681 builtins of the current globals. This means that the import is
1682 done using whatever import hooks are installed in the current
Brett Cannonbc2eff32010-09-19 21:39:02 +00001683 environment.
Guido van Rossum6058eb41998-12-21 19:51:00 +00001684 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00001685 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00001686 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001687
1688PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001689PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001690{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001691 static PyObject *silly_list = NULL;
1692 static PyObject *builtins_str = NULL;
1693 static PyObject *import_str = NULL;
1694 PyObject *globals = NULL;
1695 PyObject *import = NULL;
1696 PyObject *builtins = NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001697 PyObject *modules = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001698 PyObject *r = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001699
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001700 /* Initialize constant string objects */
1701 if (silly_list == NULL) {
1702 import_str = PyUnicode_InternFromString("__import__");
1703 if (import_str == NULL)
1704 return NULL;
1705 builtins_str = PyUnicode_InternFromString("__builtins__");
1706 if (builtins_str == NULL)
1707 return NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001708 silly_list = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001709 if (silly_list == NULL)
1710 return NULL;
1711 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001712
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001713 /* Get the builtins from current globals */
1714 globals = PyEval_GetGlobals();
1715 if (globals != NULL) {
1716 Py_INCREF(globals);
1717 builtins = PyObject_GetItem(globals, builtins_str);
1718 if (builtins == NULL)
1719 goto err;
1720 }
1721 else {
1722 /* No globals -- use standard builtins, and fake globals */
1723 builtins = PyImport_ImportModuleLevel("builtins",
1724 NULL, NULL, NULL, 0);
1725 if (builtins == NULL)
1726 return NULL;
1727 globals = Py_BuildValue("{OO}", builtins_str, builtins);
1728 if (globals == NULL)
1729 goto err;
1730 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001731
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001732 /* Get the __import__ function from the builtins */
1733 if (PyDict_Check(builtins)) {
1734 import = PyObject_GetItem(builtins, import_str);
1735 if (import == NULL)
1736 PyErr_SetObject(PyExc_KeyError, import_str);
1737 }
1738 else
1739 import = PyObject_GetAttr(builtins, import_str);
1740 if (import == NULL)
1741 goto err;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001742
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001743 /* Call the __import__ function with the proper argument list
Brett Cannonbc2eff32010-09-19 21:39:02 +00001744 Always use absolute import here.
1745 Calling for side-effect of import. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001746 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
1747 globals, silly_list, 0, NULL);
Brett Cannonbc2eff32010-09-19 21:39:02 +00001748 if (r == NULL)
1749 goto err;
1750 Py_DECREF(r);
1751
1752 modules = PyImport_GetModuleDict();
1753 r = PyDict_GetItem(modules, module_name);
1754 if (r != NULL)
1755 Py_INCREF(r);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001756
1757 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001758 Py_XDECREF(globals);
1759 Py_XDECREF(builtins);
1760 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00001761
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001762 return r;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001763}
1764
Brett Cannon4caa61d2014-01-09 19:03:32 -05001765/*[clinic input]
1766_imp.extension_suffixes
1767
1768Returns the list of file suffixes used to identify extension modules.
1769[clinic start generated code]*/
1770
Brett Cannon4caa61d2014-01-09 19:03:32 -05001771static PyObject *
1772_imp_extension_suffixes_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001773/*[clinic end generated code: output=d44c1566ef362229 input=ecdeeecfcb6f839e]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001774{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001775 PyObject *list;
Brett Cannon2657df42012-05-04 15:20:40 -04001776 const char *suffix;
1777 unsigned int index = 0;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001779 list = PyList_New(0);
1780 if (list == NULL)
1781 return NULL;
Brett Cannon2657df42012-05-04 15:20:40 -04001782#ifdef HAVE_DYNAMIC_LOADING
1783 while ((suffix = _PyImport_DynLoadFiletab[index])) {
1784 PyObject *item = PyUnicode_FromString(suffix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001785 if (item == NULL) {
1786 Py_DECREF(list);
1787 return NULL;
1788 }
1789 if (PyList_Append(list, item) < 0) {
1790 Py_DECREF(list);
1791 Py_DECREF(item);
1792 return NULL;
1793 }
1794 Py_DECREF(item);
Brett Cannon2657df42012-05-04 15:20:40 -04001795 index += 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001796 }
Brett Cannon2657df42012-05-04 15:20:40 -04001797#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001798 return list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001799}
1800
Brett Cannon4caa61d2014-01-09 19:03:32 -05001801/*[clinic input]
1802_imp.init_builtin
1803
1804 name: unicode
1805 /
1806
1807Initializes a built-in module.
1808[clinic start generated code]*/
1809
Brett Cannon4caa61d2014-01-09 19:03:32 -05001810static PyObject *
1811_imp_init_builtin_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001812/*[clinic end generated code: output=1868f473685f6d67 input=f934d2231ec52a2e]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001813{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001814 int ret;
1815 PyObject *m;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001817 ret = init_builtin(name);
1818 if (ret < 0)
1819 return NULL;
1820 if (ret == 0) {
1821 Py_INCREF(Py_None);
1822 return Py_None;
1823 }
Victor Stinner95872862011-03-07 18:20:56 +01001824 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001825 Py_XINCREF(m);
1826 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001827}
1828
Brett Cannon4caa61d2014-01-09 19:03:32 -05001829/*[clinic input]
1830_imp.init_frozen
1831
1832 name: unicode
1833 /
1834
1835Initializes a frozen module.
1836[clinic start generated code]*/
1837
Brett Cannon4caa61d2014-01-09 19:03:32 -05001838static PyObject *
1839_imp_init_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001840/*[clinic end generated code: output=a9de493bdd711878 input=13019adfc04f3fb3]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001841{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001842 int ret;
1843 PyObject *m;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001844
Victor Stinner53dc7352011-03-20 01:50:21 +01001845 ret = PyImport_ImportFrozenModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001846 if (ret < 0)
1847 return NULL;
1848 if (ret == 0) {
1849 Py_INCREF(Py_None);
1850 return Py_None;
1851 }
Victor Stinner53dc7352011-03-20 01:50:21 +01001852 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001853 Py_XINCREF(m);
1854 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001855}
1856
Brett Cannon4caa61d2014-01-09 19:03:32 -05001857/*[clinic input]
1858_imp.get_frozen_object
1859
1860 name: unicode
1861 /
1862
1863Create a code object for a frozen module.
1864[clinic start generated code]*/
1865
Brett Cannon4caa61d2014-01-09 19:03:32 -05001866static PyObject *
1867_imp_get_frozen_object_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001868/*[clinic end generated code: output=3114c970a47f2e3c input=ed689bc05358fdbd]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001869{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001870 return get_frozen_object(name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001871}
1872
Brett Cannon4caa61d2014-01-09 19:03:32 -05001873/*[clinic input]
1874_imp.is_frozen_package
1875
1876 name: unicode
1877 /
1878
1879Returns True if the module name is of a frozen package.
1880[clinic start generated code]*/
1881
Brett Cannon4caa61d2014-01-09 19:03:32 -05001882static PyObject *
1883_imp_is_frozen_package_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001884/*[clinic end generated code: output=3e4cab802b56d649 input=81b6cdecd080fbb8]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001885{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001886 return is_frozen_package(name);
Brett Cannon8d110132009-03-15 02:20:16 +00001887}
1888
Brett Cannon4caa61d2014-01-09 19:03:32 -05001889/*[clinic input]
1890_imp.is_builtin
1891
1892 name: unicode
1893 /
1894
1895Returns True if the module name corresponds to a built-in module.
1896[clinic start generated code]*/
1897
Guido van Rossum79f25d91997-04-29 20:08:16 +00001898static PyObject *
Brett Cannon4caa61d2014-01-09 19:03:32 -05001899_imp_is_builtin_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001900/*[clinic end generated code: output=2deec9cac6fb9a7e input=86befdac021dd1c7]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001901{
Brett Cannon4caa61d2014-01-09 19:03:32 -05001902 return PyLong_FromLong(is_builtin(name));
1903}
1904
1905/*[clinic input]
1906_imp.is_frozen
1907
1908 name: unicode
1909 /
1910
1911Returns True if the module name corresponds to a frozen module.
1912[clinic start generated code]*/
1913
Brett Cannon4caa61d2014-01-09 19:03:32 -05001914static PyObject *
1915_imp_is_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001916/*[clinic end generated code: output=7de8e260c8e36aed input=7301dbca1897d66b]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001917{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001918 const struct _frozen *p;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001919
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001920 p = find_frozen(name);
1921 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001922}
1923
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001924#ifdef HAVE_DYNAMIC_LOADING
1925
Brett Cannon4caa61d2014-01-09 19:03:32 -05001926/*[clinic input]
1927_imp.load_dynamic
1928
1929 name: unicode
1930 path: fs_unicode
1931 file: object = NULL
1932 /
1933
1934Loads an extension module.
1935[clinic start generated code]*/
1936
Brett Cannon4caa61d2014-01-09 19:03:32 -05001937static PyObject *
Larry Hastings89964c42015-04-14 18:07:59 -04001938_imp_load_dynamic_impl(PyModuleDef *module, PyObject *name, PyObject *path,
1939 PyObject *file)
1940/*[clinic end generated code: output=e84e5f7f0f39bc54 input=af64f06e4bad3526]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001941{
1942 PyObject *mod;
Victor Stinnerfefd70c2011-03-14 15:54:07 -04001943 FILE *fp;
1944
Brett Cannon4caa61d2014-01-09 19:03:32 -05001945 if (file != NULL) {
1946 fp = _Py_fopen_obj(path, "r");
Victor Stinnere9ddbf62011-03-22 10:46:35 +01001947 if (fp == NULL) {
Brett Cannon4caa61d2014-01-09 19:03:32 -05001948 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001949 return NULL;
Victor Stinnere9ddbf62011-03-22 10:46:35 +01001950 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001951 }
Victor Stinnerfefd70c2011-03-14 15:54:07 -04001952 else
1953 fp = NULL;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001954 mod = _PyImport_LoadDynamicModule(name, path, fp);
1955 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001956 if (fp)
1957 fclose(fp);
Victor Stinnerfefd70c2011-03-14 15:54:07 -04001958 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001959}
1960
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001961#endif /* HAVE_DYNAMIC_LOADING */
1962
Larry Hastings7726ac92014-01-31 22:03:12 -08001963/*[clinic input]
1964dump buffer
1965[clinic start generated code]*/
Brett Cannonfd4d0502014-05-30 11:21:14 -04001966/*[clinic end generated code: output=da39a3ee5e6b4b0d input=524ce2e021e4eba6]*/
Larry Hastings7726ac92014-01-31 22:03:12 -08001967
Barry Warsaw28a691b2010-04-17 00:19:56 +00001968
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001969PyDoc_STRVAR(doc_imp,
Brett Cannon6f44d662012-04-15 16:08:47 -04001970"(Extremely) low-level import machinery bits as used by importlib and imp.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00001971
Guido van Rossum79f25d91997-04-29 20:08:16 +00001972static PyMethodDef imp_methods[] = {
Brett Cannon4caa61d2014-01-09 19:03:32 -05001973 _IMP_EXTENSION_SUFFIXES_METHODDEF
1974 _IMP_LOCK_HELD_METHODDEF
1975 _IMP_ACQUIRE_LOCK_METHODDEF
1976 _IMP_RELEASE_LOCK_METHODDEF
1977 _IMP_GET_FROZEN_OBJECT_METHODDEF
1978 _IMP_IS_FROZEN_PACKAGE_METHODDEF
1979 _IMP_INIT_BUILTIN_METHODDEF
1980 _IMP_INIT_FROZEN_METHODDEF
1981 _IMP_IS_BUILTIN_METHODDEF
1982 _IMP_IS_FROZEN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05001983 _IMP_LOAD_DYNAMIC_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05001984 _IMP__FIX_CO_FILENAME_METHODDEF
1985 {NULL, NULL} /* sentinel */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001986};
1987
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001988
Martin v. Löwis1a214512008-06-11 05:26:20 +00001989static struct PyModuleDef impmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001990 PyModuleDef_HEAD_INIT,
Brett Cannon6f44d662012-04-15 16:08:47 -04001991 "_imp",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001992 doc_imp,
1993 0,
1994 imp_methods,
1995 NULL,
1996 NULL,
1997 NULL,
1998 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001999};
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002000
Jason Tishler6bc06ec2003-09-04 11:59:50 +00002001PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00002002PyInit_imp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002003{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002004 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002006 m = PyModule_Create(&impmodule);
2007 if (m == NULL)
2008 goto failure;
2009 d = PyModule_GetDict(m);
2010 if (d == NULL)
2011 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002012
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002013 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002014 failure:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002015 Py_XDECREF(m);
2016 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002017}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002018
2019
Guido van Rossumb18618d2000-05-03 23:44:39 +00002020/* API for embedding applications that want to add their own entries
2021 to the table of built-in modules. This should normally be called
2022 *before* Py_Initialize(). When the table resize fails, -1 is
2023 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002024
2025 After a similar function by Just van Rossum. */
2026
2027int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002028PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002029{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002030 static struct _inittab *our_copy = NULL;
2031 struct _inittab *p;
2032 int i, n;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002034 /* Count the number of entries in both tables */
2035 for (n = 0; newtab[n].name != NULL; n++)
2036 ;
2037 if (n == 0)
2038 return 0; /* Nothing to do */
2039 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2040 ;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002042 /* Allocate new memory for the combined table */
2043 p = our_copy;
2044 PyMem_RESIZE(p, struct _inittab, i+n+1);
2045 if (p == NULL)
2046 return -1;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002048 /* Copy the tables into the new memory */
2049 if (our_copy != PyImport_Inittab)
2050 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2051 PyImport_Inittab = our_copy = p;
2052 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002053
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002054 return 0;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002055}
2056
2057/* Shorthand to add a single entry given a name and a function */
2058
2059int
Brett Cannona826f322009-04-02 03:41:46 +00002060PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002061{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002062 struct _inittab newtab[2];
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002063
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002064 memset(newtab, '\0', sizeof newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002065
Serhiy Storchaka20b39b22014-09-28 11:27:24 +03002066 newtab[0].name = name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002067 newtab[0].initfunc = initfunc;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002068
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002069 return PyImport_ExtendInittab(newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002070}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002071
2072#ifdef __cplusplus
2073}
2074#endif