blob: 0b843dafd3df21d2d04877e77d0b9de0025931b3 [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{
Christian Heimes418fd742015-04-19 21:08:42 +0200212 if (import_lock != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000213 import_lock = PyThread_allocate_lock();
Christian Heimes418fd742015-04-19 21:08:42 +0200214 if (import_lock == NULL) {
215 Py_FatalError("PyImport_ReInitLock failed to create a new lock");
216 }
217 }
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000218 if (import_lock_level > 1) {
219 /* Forked as a side effect of import */
220 long me = PyThread_get_thread_ident();
Brett Cannone4710cf2012-11-15 21:39:36 -0500221 /* The following could fail if the lock is already held, but forking as
222 a side-effect of an import is a) rare, b) nuts, and c) difficult to
223 do thanks to the lock only being held when doing individual module
224 locks per import. */
225 PyThread_acquire_lock(import_lock, NOWAIT_LOCK);
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000226 import_lock_thread = me;
227 import_lock_level--;
228 } else {
229 import_lock_thread = -1;
230 import_lock_level = 0;
231 }
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000232}
233
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000234#endif
235
Brett Cannon4caa61d2014-01-09 19:03:32 -0500236/*[clinic input]
237_imp.lock_held
238
239Return True if the import lock is currently held, else False.
240
241On platforms without threads, return False.
242[clinic start generated code]*/
243
Brett Cannon4caa61d2014-01-09 19:03:32 -0500244static PyObject *
245_imp_lock_held_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400246/*[clinic end generated code: output=d7a8cc3a5169081a input=9b088f9b217d9bdf]*/
Tim Peters69232342001-08-30 05:16:13 +0000247{
Tim Peters69232342001-08-30 05:16:13 +0000248#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000249 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000250#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000251 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000252#endif
253}
254
Brett Cannon4caa61d2014-01-09 19:03:32 -0500255/*[clinic input]
256_imp.acquire_lock
257
258Acquires the interpreter's import lock for the current thread.
259
260This lock should be used by import hooks to ensure thread-safety when importing
261modules. On platforms without threads, this function does nothing.
262[clinic start generated code]*/
263
Brett Cannon4caa61d2014-01-09 19:03:32 -0500264static PyObject *
265_imp_acquire_lock_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400266/*[clinic end generated code: output=cc143b1d16422cae input=4a2d4381866d5fdc]*/
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000267{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000268#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000269 _PyImport_AcquireLock();
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000270#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000271 Py_INCREF(Py_None);
272 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000273}
274
Brett Cannon4caa61d2014-01-09 19:03:32 -0500275/*[clinic input]
276_imp.release_lock
277
278Release the interpreter's import lock.
279
280On platforms without threads, this function does nothing.
281[clinic start generated code]*/
282
Brett Cannon4caa61d2014-01-09 19:03:32 -0500283static PyObject *
284_imp_release_lock_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -0400285/*[clinic end generated code: output=74d28e38ebe2b224 input=934fb11516dd778b]*/
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000286{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000287#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000288 if (_PyImport_ReleaseLock() < 0) {
289 PyErr_SetString(PyExc_RuntimeError,
290 "not holding the import lock");
291 return NULL;
292 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000293#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000294 Py_INCREF(Py_None);
295 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000296}
297
Antoine Pitrou8db076c2011-10-30 19:13:55 +0100298void
299_PyImport_Fini(void)
300{
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200301 Py_CLEAR(extensions);
Antoine Pitrou8db076c2011-10-30 19:13:55 +0100302#ifdef WITH_THREAD
303 if (import_lock != NULL) {
304 PyThread_free_lock(import_lock);
305 import_lock = NULL;
306 }
307#endif
308}
309
Guido van Rossum25ce5661997-08-02 03:10:38 +0000310/* Helper for sys */
311
312PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000313PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000314{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000315 PyInterpreterState *interp = PyThreadState_GET()->interp;
316 if (interp->modules == NULL)
317 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
318 return interp->modules;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000319}
320
Guido van Rossum3f5da241990-12-20 15:06:42 +0000321
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000322/* List of names to clear in sys */
323static char* sys_deletes[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000324 "path", "argv", "ps1", "ps2",
325 "last_type", "last_value", "last_traceback",
326 "path_hooks", "path_importer_cache", "meta_path",
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200327 "__interactivehook__",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000328 /* misc stuff */
329 "flags", "float_info",
330 NULL
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000331};
332
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000333static char* sys_files[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000334 "stdin", "__stdin__",
335 "stdout", "__stdout__",
336 "stderr", "__stderr__",
337 NULL
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000338};
339
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000340/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000341
Guido van Rossum3f5da241990-12-20 15:06:42 +0000342void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000343PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000344{
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200345 Py_ssize_t pos;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000346 PyObject *key, *value, *dict;
347 PyInterpreterState *interp = PyThreadState_GET()->interp;
348 PyObject *modules = interp->modules;
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200349 PyObject *weaklist = NULL;
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200350 char **p;
Guido van Rossum758eec01998-01-19 21:58:26 +0000351
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000352 if (modules == NULL)
353 return; /* Already done */
Guido van Rossum758eec01998-01-19 21:58:26 +0000354
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000355 /* Delete some special variables first. These are common
356 places where user values hide and people complain when their
357 destructors fail. Since the modules containing them are
358 deleted *last* of all, they would come too late in the normal
359 destruction order. Sigh. */
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000360
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200361 /* XXX Perhaps these precautions are obsolete. Who knows? */
362
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200363 if (Py_VerboseFlag)
364 PySys_WriteStderr("# clear builtins._\n");
365 PyDict_SetItemString(interp->builtins, "_", Py_None);
366
367 for (p = sys_deletes; *p != NULL; p++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000368 if (Py_VerboseFlag)
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200369 PySys_WriteStderr("# clear sys.%s\n", *p);
370 PyDict_SetItemString(interp->sysdict, *p, Py_None);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000371 }
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200372 for (p = sys_files; *p != NULL; p+=2) {
373 if (Py_VerboseFlag)
374 PySys_WriteStderr("# restore sys.%s\n", *p);
375 value = PyDict_GetItemString(interp->sysdict, *(p+1));
376 if (value == NULL)
377 value = Py_None;
378 PyDict_SetItemString(interp->sysdict, *p, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000379 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000380
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200381 /* We prepare a list which will receive (name, weakref) tuples of
382 modules when they are removed from sys.modules. The name is used
383 for diagnosis messages (in verbose mode), while the weakref helps
384 detect those modules which have been held alive. */
385 weaklist = PyList_New(0);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200386 if (weaklist == NULL)
387 PyErr_Clear();
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200388
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200389#define STORE_MODULE_WEAKREF(name, mod) \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200390 if (weaklist != NULL) { \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200391 PyObject *wr = PyWeakref_NewRef(mod, NULL); \
392 if (name && wr) { \
393 PyObject *tup = PyTuple_Pack(2, name, wr); \
394 PyList_Append(weaklist, tup); \
395 Py_XDECREF(tup); \
396 } \
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200397 Py_XDECREF(wr); \
398 if (PyErr_Occurred()) \
399 PyErr_Clear(); \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 }
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000401
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200402 /* Remove all modules from sys.modules, hoping that garbage collection
403 can reclaim most of them. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000404 pos = 0;
405 while (PyDict_Next(modules, &pos, &key, &value)) {
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200406 if (PyModule_Check(value)) {
407 if (Py_VerboseFlag && PyUnicode_Check(key))
Victor Stinnerab826d12014-07-07 23:06:15 +0200408 PySys_FormatStderr("# cleanup[2] removing %U\n", key);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200409 STORE_MODULE_WEAKREF(key, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000410 PyDict_SetItem(modules, key, Py_None);
411 }
412 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000413
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200414 /* Clear the modules dict. */
415 PyDict_Clear(modules);
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200416 /* Restore the original builtins dict, to ensure that any
417 user data gets cleared. */
418 dict = PyDict_Copy(interp->builtins);
419 if (dict == NULL)
420 PyErr_Clear();
421 PyDict_Clear(interp->builtins);
422 if (PyDict_Update(interp->builtins, interp->builtins_copy))
423 PyErr_Clear();
424 Py_XDECREF(dict);
Antoine Pitrou40322e62013-08-11 00:30:09 +0200425 /* Clear module dict copies stored in the interpreter state */
426 _PyState_ClearModules();
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200427 /* Collect references */
428 _PyGC_CollectNoFail();
Antoine Pitrou5f454a02013-05-06 21:15:57 +0200429 /* Dump GC stats before it's too late, since it uses the warnings
430 machinery. */
431 _PyGC_DumpShutdownStats();
432
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200433 /* Now, if there are any modules left alive, clear their globals to
434 minimize potential leaks. All C extension modules actually end
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200435 up here, since they are kept alive in the interpreter state.
436
437 The special treatment of "builtins" here is because even
438 when it's not referenced as a module, its dictionary is
439 referenced by almost every module's __builtins__. Since
440 deleting a module clears its dictionary (even if there are
441 references left to it), we need to delete the "builtins"
442 module last. Likewise, we don't delete sys until the very
443 end because it is implicitly referenced (e.g. by print). */
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200444 if (weaklist != NULL) {
445 Py_ssize_t i, n;
446 n = PyList_GET_SIZE(weaklist);
447 for (i = 0; i < n; i++) {
448 PyObject *tup = PyList_GET_ITEM(weaklist, i);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200449 PyObject *name = PyTuple_GET_ITEM(tup, 0);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200450 PyObject *mod = PyWeakref_GET_OBJECT(PyTuple_GET_ITEM(tup, 1));
451 if (mod == Py_None)
452 continue;
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200453 assert(PyModule_Check(mod));
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200454 dict = PyModule_GetDict(mod);
455 if (dict == interp->builtins || dict == interp->sysdict)
456 continue;
457 Py_INCREF(mod);
Antoine Pitrou79ba3882013-08-06 22:50:15 +0200458 if (Py_VerboseFlag && PyUnicode_Check(name))
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200459 PySys_FormatStderr("# cleanup[3] wiping %U\n", name);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200460 _PyModule_Clear(mod);
461 Py_DECREF(mod);
Antoine Pitrou070cb3c2013-05-08 13:23:25 +0200462 }
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200463 Py_DECREF(weaklist);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000464 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000465
Serhiy Storchaka013bb912014-02-10 18:21:34 +0200466 /* Next, delete sys and builtins (in that order) */
467 if (Py_VerboseFlag)
468 PySys_FormatStderr("# cleanup[3] wiping sys\n");
469 _PyModule_ClearDict(interp->sysdict);
470 if (Py_VerboseFlag)
471 PySys_FormatStderr("# cleanup[3] wiping builtins\n");
472 _PyModule_ClearDict(interp->builtins);
473
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200474 /* Clear and delete the modules directory. Actual modules will
475 still be there only if imported during the execution of some
476 destructor. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000477 interp->modules = NULL;
478 Py_DECREF(modules);
Antoine Pitroudcedaf62013-07-31 23:14:08 +0200479
480 /* Once more */
481 _PyGC_CollectNoFail();
482
483#undef STORE_MODULE_WEAKREF
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000484}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000485
486
Barry Warsaw28a691b2010-04-17 00:19:56 +0000487/* Helper for pythonrun.c -- return magic number and tag. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000488
489long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000490PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000491{
Antoine Pitrou44b4b6a2012-07-10 18:27:54 +0200492 long res;
Brett Cannon77b2abd2012-07-09 16:09:00 -0400493 PyInterpreterState *interp = PyThreadState_Get()->interp;
Eric Snow32439d62015-05-02 19:15:18 -0600494 PyObject *external, *pyc_magic;
495
496 external = PyObject_GetAttrString(interp->importlib, "_bootstrap_external");
497 if (external == NULL)
498 return -1;
499 pyc_magic = PyObject_GetAttrString(external, "_RAW_MAGIC_NUMBER");
500 Py_DECREF(external);
Brett Cannon77b2abd2012-07-09 16:09:00 -0400501 if (pyc_magic == NULL)
502 return -1;
Antoine Pitrou44b4b6a2012-07-10 18:27:54 +0200503 res = PyLong_AsLong(pyc_magic);
Benjamin Peterson66f36592012-07-09 22:21:55 -0700504 Py_DECREF(pyc_magic);
505 return res;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000506}
507
508
Brett Cannon3adc7b72012-07-09 14:22:12 -0400509extern const char * _PySys_ImplCacheTag;
510
Barry Warsaw28a691b2010-04-17 00:19:56 +0000511const char *
512PyImport_GetMagicTag(void)
513{
Brett Cannon3adc7b72012-07-09 14:22:12 -0400514 return _PySys_ImplCacheTag;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000515}
516
Brett Cannon98979b82012-07-02 15:13:11 -0400517
Guido van Rossum25ce5661997-08-02 03:10:38 +0000518/* Magic for extension modules (built-in as well as dynamically
519 loaded). To prevent initializing an extension module more than
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200520 once, we keep a static dictionary 'extensions' keyed by the tuple
521 (module name, module name) (for built-in modules) or by
522 (filename, module name) (for dynamically loaded modules), containing these
523 modules. A copy of the module's dictionary is stored by calling
524 _PyImport_FixupExtensionObject() immediately after the module initialization
525 function succeeds. A copy can be retrieved from there by calling
Victor Stinner95872862011-03-07 18:20:56 +0100526 _PyImport_FindExtensionObject().
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000527
Barry Warsaw7c9627b2010-06-17 18:38:20 +0000528 Modules which do support multiple initialization set their m_size
529 field to a non-negative number (indicating the size of the
530 module-specific state). They are still recorded in the extensions
531 dictionary, to avoid loading shared libraries twice.
Martin v. Löwis1a214512008-06-11 05:26:20 +0000532*/
533
534int
Victor Stinner95872862011-03-07 18:20:56 +0100535_PyImport_FixupExtensionObject(PyObject *mod, PyObject *name,
536 PyObject *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000537{
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500538 PyObject *modules, *dict, *key;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000539 struct PyModuleDef *def;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500540 int res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 if (extensions == NULL) {
542 extensions = PyDict_New();
543 if (extensions == NULL)
544 return -1;
545 }
546 if (mod == NULL || !PyModule_Check(mod)) {
547 PyErr_BadInternalCall();
548 return -1;
549 }
550 def = PyModule_GetDef(mod);
551 if (!def) {
552 PyErr_BadInternalCall();
553 return -1;
554 }
555 modules = PyImport_GetModuleDict();
Victor Stinner95872862011-03-07 18:20:56 +0100556 if (PyDict_SetItem(modules, name, mod) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 return -1;
558 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100559 PyDict_DelItem(modules, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 return -1;
561 }
562 if (def->m_size == -1) {
563 if (def->m_base.m_copy) {
564 /* Somebody already imported the module,
565 likely under a different name.
566 XXX this should really not happen. */
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200567 Py_CLEAR(def->m_base.m_copy);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 }
569 dict = PyModule_GetDict(mod);
570 if (dict == NULL)
571 return -1;
572 def->m_base.m_copy = PyDict_Copy(dict);
573 if (def->m_base.m_copy == NULL)
574 return -1;
575 }
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500576 key = PyTuple_Pack(2, filename, name);
577 if (key == NULL)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200578 return -1;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500579 res = PyDict_SetItem(extensions, key, (PyObject *)def);
580 Py_DECREF(key);
581 if (res < 0)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200582 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 return 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000584}
585
Victor Stinner49d3f252010-10-17 01:24:53 +0000586int
Serhiy Storchakac6792272013-10-19 21:03:34 +0300587_PyImport_FixupBuiltin(PyObject *mod, const char *name)
Victor Stinner49d3f252010-10-17 01:24:53 +0000588{
589 int res;
Victor Stinner95872862011-03-07 18:20:56 +0100590 PyObject *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100591 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100592 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000593 return -1;
Victor Stinner95872862011-03-07 18:20:56 +0100594 res = _PyImport_FixupExtensionObject(mod, nameobj, nameobj);
595 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000596 return res;
597}
598
Guido van Rossum25ce5661997-08-02 03:10:38 +0000599PyObject *
Victor Stinner95872862011-03-07 18:20:56 +0100600_PyImport_FindExtensionObject(PyObject *name, PyObject *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000601{
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500602 PyObject *mod, *mdict, *key;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000603 PyModuleDef* def;
604 if (extensions == NULL)
605 return NULL;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500606 key = PyTuple_Pack(2, filename, name);
607 if (key == NULL)
Andrew Svetlov6b2cbeb2012-12-14 17:04:59 +0200608 return NULL;
Benjamin Peterson5cb8a312012-12-15 00:05:16 -0500609 def = (PyModuleDef *)PyDict_GetItem(extensions, key);
610 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 if (def == NULL)
612 return NULL;
613 if (def->m_size == -1) {
614 /* Module does not support repeated initialization */
615 if (def->m_base.m_copy == NULL)
616 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100617 mod = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 if (mod == NULL)
619 return NULL;
620 mdict = PyModule_GetDict(mod);
621 if (mdict == NULL)
622 return NULL;
623 if (PyDict_Update(mdict, def->m_base.m_copy))
624 return NULL;
625 }
626 else {
627 if (def->m_base.m_init == NULL)
628 return NULL;
629 mod = def->m_base.m_init();
630 if (mod == NULL)
631 return NULL;
Christian Heimes09ca7942013-07-20 14:51:53 +0200632 if (PyDict_SetItem(PyImport_GetModuleDict(), name, mod) == -1) {
633 Py_DECREF(mod);
634 return NULL;
635 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 Py_DECREF(mod);
637 }
638 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100639 PyDict_DelItem(PyImport_GetModuleDict(), name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000640 Py_DECREF(mod);
641 return NULL;
642 }
643 if (Py_VerboseFlag)
Victor Stinner95872862011-03-07 18:20:56 +0100644 PySys_FormatStderr("import %U # previously loaded (%R)\n",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 name, filename);
646 return mod;
Brett Cannon9a5b25a2010-03-01 02:09:17 +0000647
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000648}
649
Victor Stinner49d3f252010-10-17 01:24:53 +0000650PyObject *
Victor Stinner501c09a2011-02-23 00:02:00 +0000651_PyImport_FindBuiltin(const char *name)
Victor Stinner49d3f252010-10-17 01:24:53 +0000652{
Victor Stinner95872862011-03-07 18:20:56 +0100653 PyObject *res, *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100654 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100655 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000656 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100657 res = _PyImport_FindExtensionObject(nameobj, nameobj);
658 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000659 return res;
660}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000661
662/* Get the module object corresponding to a module name.
663 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000664 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000665 Because the former action is most common, THIS DOES NOT RETURN A
666 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000667
Guido van Rossum79f25d91997-04-29 20:08:16 +0000668PyObject *
Victor Stinner27ee0892011-03-04 12:57:09 +0000669PyImport_AddModuleObject(PyObject *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000670{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000671 PyObject *modules = PyImport_GetModuleDict();
672 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000673
Serhiy Storchaka48a583b2016-02-10 10:31:20 +0200674 if ((m = PyDict_GetItemWithError(modules, name)) != NULL &&
675 PyModule_Check(m)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 return m;
Serhiy Storchaka48a583b2016-02-10 10:31:20 +0200677 }
678 if (PyErr_Occurred()) {
679 return NULL;
680 }
Victor Stinner27ee0892011-03-04 12:57:09 +0000681 m = PyModule_NewObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000682 if (m == NULL)
683 return NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000684 if (PyDict_SetItem(modules, name, m) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 Py_DECREF(m);
686 return NULL;
687 }
688 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000691}
692
Victor Stinner27ee0892011-03-04 12:57:09 +0000693PyObject *
694PyImport_AddModule(const char *name)
695{
696 PyObject *nameobj, *module;
697 nameobj = PyUnicode_FromString(name);
698 if (nameobj == NULL)
699 return NULL;
700 module = PyImport_AddModuleObject(nameobj);
701 Py_DECREF(nameobj);
702 return module;
703}
704
705
Tim Peters1cd70172004-08-02 03:52:12 +0000706/* Remove name from sys.modules, if it's there. */
707static void
Victor Stinner27ee0892011-03-04 12:57:09 +0000708remove_module(PyObject *name)
Tim Peters1cd70172004-08-02 03:52:12 +0000709{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner27ee0892011-03-04 12:57:09 +0000711 if (PyDict_GetItem(modules, name) == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000712 return;
Victor Stinner27ee0892011-03-04 12:57:09 +0000713 if (PyDict_DelItem(modules, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000714 Py_FatalError("import: deleting existing key in"
715 "sys.modules failed");
Tim Peters1cd70172004-08-02 03:52:12 +0000716}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000717
Christian Heimes3b06e532008-01-07 20:12:44 +0000718
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000719/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000720 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
721 * removed from sys.modules, to avoid leaving damaged module objects
722 * in sys.modules. The caller may wish to restore the original
723 * module object (if any) in this case; PyImport_ReloadModule is an
724 * example.
Barry Warsaw28a691b2010-04-17 00:19:56 +0000725 *
726 * Note that PyImport_ExecCodeModuleWithPathnames() is the preferred, richer
727 * interface. The other two exist primarily for backward compatibility.
Tim Peters1cd70172004-08-02 03:52:12 +0000728 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000729PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300730PyImport_ExecCodeModule(const char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000731{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000732 return PyImport_ExecCodeModuleWithPathnames(
733 name, co, (char *)NULL, (char *)NULL);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000734}
735
736PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300737PyImport_ExecCodeModuleEx(const char *name, PyObject *co, const char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000738{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000739 return PyImport_ExecCodeModuleWithPathnames(
740 name, co, pathname, (char *)NULL);
Barry Warsaw28a691b2010-04-17 00:19:56 +0000741}
742
743PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300744PyImport_ExecCodeModuleWithPathnames(const char *name, PyObject *co,
745 const char *pathname,
746 const char *cpathname)
Barry Warsaw28a691b2010-04-17 00:19:56 +0000747{
Victor Stinner27ee0892011-03-04 12:57:09 +0000748 PyObject *m = NULL;
Eric Snow32439d62015-05-02 19:15:18 -0600749 PyObject *nameobj, *pathobj = NULL, *cpathobj = NULL, *external= NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000750
751 nameobj = PyUnicode_FromString(name);
752 if (nameobj == NULL)
753 return NULL;
754
Victor Stinner27ee0892011-03-04 12:57:09 +0000755 if (cpathname != NULL) {
756 cpathobj = PyUnicode_DecodeFSDefault(cpathname);
757 if (cpathobj == NULL)
758 goto error;
Brett Cannona6473f92012-07-13 13:57:03 -0400759 }
760 else
Victor Stinner27ee0892011-03-04 12:57:09 +0000761 cpathobj = NULL;
Brett Cannona6473f92012-07-13 13:57:03 -0400762
763 if (pathname != NULL) {
764 pathobj = PyUnicode_DecodeFSDefault(pathname);
765 if (pathobj == NULL)
766 goto error;
767 }
768 else if (cpathobj != NULL) {
769 PyInterpreterState *interp = PyThreadState_GET()->interp;
770 _Py_IDENTIFIER(_get_sourcefile);
771
772 if (interp == NULL) {
773 Py_FatalError("PyImport_ExecCodeModuleWithPathnames: "
774 "no interpreter!");
775 }
776
Eric Snow32439d62015-05-02 19:15:18 -0600777 external= PyObject_GetAttrString(interp->importlib,
778 "_bootstrap_external");
779 if (external != NULL) {
780 pathobj = _PyObject_CallMethodIdObjArgs(external,
781 &PyId__get_sourcefile, cpathobj,
782 NULL);
783 Py_DECREF(external);
784 }
Brett Cannona6473f92012-07-13 13:57:03 -0400785 if (pathobj == NULL)
786 PyErr_Clear();
787 }
788 else
789 pathobj = NULL;
790
Victor Stinner27ee0892011-03-04 12:57:09 +0000791 m = PyImport_ExecCodeModuleObject(nameobj, co, pathobj, cpathobj);
792error:
793 Py_DECREF(nameobj);
794 Py_XDECREF(pathobj);
795 Py_XDECREF(cpathobj);
796 return m;
797}
798
Brett Cannon18fc4e72014-04-04 10:01:46 -0400799static PyObject *
800module_dict_for_exec(PyObject *name)
Victor Stinner27ee0892011-03-04 12:57:09 +0000801{
Brett Cannon18fc4e72014-04-04 10:01:46 -0400802 PyObject *m, *d = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000803
Victor Stinner27ee0892011-03-04 12:57:09 +0000804 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 if (m == NULL)
806 return NULL;
807 /* If the module is being reloaded, we get the old module back
808 and re-use its dict to exec the new code. */
809 d = PyModule_GetDict(m);
810 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
811 if (PyDict_SetItemString(d, "__builtins__",
Brett Cannon18fc4e72014-04-04 10:01:46 -0400812 PyEval_GetBuiltins()) != 0) {
813 remove_module(name);
814 return NULL;
815 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 }
Brett Cannon18fc4e72014-04-04 10:01:46 -0400817
Eric Snow08197a42014-05-12 17:54:55 -0600818 return d; /* Return a borrowed reference. */
Brett Cannon18fc4e72014-04-04 10:01:46 -0400819}
820
821static PyObject *
822exec_code_in_module(PyObject *name, PyObject *module_dict, PyObject *code_object)
823{
824 PyObject *modules = PyImport_GetModuleDict();
825 PyObject *v, *m;
826
827 v = PyEval_EvalCode(code_object, module_dict, module_dict);
828 if (v == NULL) {
829 remove_module(name);
830 return NULL;
831 }
832 Py_DECREF(v);
833
834 if ((m = PyDict_GetItem(modules, name)) == NULL) {
835 PyErr_Format(PyExc_ImportError,
836 "Loaded module %R not found in sys.modules",
837 name);
838 return NULL;
839 }
840
841 Py_INCREF(m);
842
843 return m;
844}
845
846PyObject*
847PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
848 PyObject *cpathname)
849{
Eric Snow32439d62015-05-02 19:15:18 -0600850 PyObject *d, *external, *res;
Eric Snow08197a42014-05-12 17:54:55 -0600851 PyInterpreterState *interp = PyThreadState_GET()->interp;
852 _Py_IDENTIFIER(_fix_up_module);
Brett Cannon18fc4e72014-04-04 10:01:46 -0400853
854 d = module_dict_for_exec(name);
855 if (d == NULL) {
856 return NULL;
857 }
858
Eric Snow08197a42014-05-12 17:54:55 -0600859 if (pathname == NULL) {
860 pathname = ((PyCodeObject *)co)->co_filename;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 }
Eric Snow32439d62015-05-02 19:15:18 -0600862 external = PyObject_GetAttrString(interp->importlib, "_bootstrap_external");
863 if (external == NULL)
864 return NULL;
865 res = _PyObject_CallMethodIdObjArgs(external,
Eric Snow08197a42014-05-12 17:54:55 -0600866 &PyId__fix_up_module,
867 d, name, pathname, cpathname, NULL);
Eric Snow32439d62015-05-02 19:15:18 -0600868 Py_DECREF(external);
Eric Snow08197a42014-05-12 17:54:55 -0600869 if (res != NULL) {
Eric Snow58cfdd82014-05-29 12:31:39 -0600870 Py_DECREF(res);
Eric Snow08197a42014-05-12 17:54:55 -0600871 res = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000872 }
Eric Snow08197a42014-05-12 17:54:55 -0600873 return res;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000874}
875
876
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000877static void
878update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
879{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000880 PyObject *constants, *tmp;
881 Py_ssize_t i, n;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000882
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000883 if (PyUnicode_Compare(co->co_filename, oldname))
884 return;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000885
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000886 tmp = co->co_filename;
887 co->co_filename = newname;
888 Py_INCREF(co->co_filename);
889 Py_DECREF(tmp);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000890
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000891 constants = co->co_consts;
892 n = PyTuple_GET_SIZE(constants);
893 for (i = 0; i < n; i++) {
894 tmp = PyTuple_GET_ITEM(constants, i);
895 if (PyCode_Check(tmp))
896 update_code_filenames((PyCodeObject *)tmp,
897 oldname, newname);
898 }
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000899}
900
Victor Stinner2f42ae52011-03-20 00:41:24 +0100901static void
902update_compiled_module(PyCodeObject *co, PyObject *newname)
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000903{
Victor Stinner2f42ae52011-03-20 00:41:24 +0100904 PyObject *oldname;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000905
Victor Stinner2f42ae52011-03-20 00:41:24 +0100906 if (PyUnicode_Compare(co->co_filename, newname) == 0)
907 return;
Hirokazu Yamamoto4f447fb2009-03-04 01:52:10 +0000908
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000909 oldname = co->co_filename;
910 Py_INCREF(oldname);
911 update_code_filenames(co, oldname, newname);
912 Py_DECREF(oldname);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000913}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000914
Brett Cannon4caa61d2014-01-09 19:03:32 -0500915/*[clinic input]
916_imp._fix_co_filename
917
918 code: object(type="PyCodeObject *", subclass_of="&PyCode_Type")
919 Code object to change.
920
921 path: unicode
922 File path to use.
923 /
924
925Changes code.co_filename to specify the passed-in file path.
926[clinic start generated code]*/
927
Brett Cannon4caa61d2014-01-09 19:03:32 -0500928static PyObject *
Larry Hastings89964c42015-04-14 18:07:59 -0400929_imp__fix_co_filename_impl(PyModuleDef *module, PyCodeObject *code,
930 PyObject *path)
931/*[clinic end generated code: output=f4db56aac0a1327f input=895ba50e78b82f05]*/
Brett Cannon442c9b92011-03-23 16:14:42 -0700932
Brett Cannon4caa61d2014-01-09 19:03:32 -0500933{
Brett Cannona6dec2e2014-01-10 07:43:55 -0500934 update_compiled_module(code, path);
Brett Cannon442c9b92011-03-23 16:14:42 -0700935
936 Py_RETURN_NONE;
937}
938
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000939
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000940/* Forward */
Benjamin Peterson6fba3db2013-03-18 23:13:31 -0700941static const struct _frozen * find_frozen(PyObject *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000942
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000943
944/* Helper to test for built-in module */
945
946static int
Victor Stinner95872862011-03-07 18:20:56 +0100947is_builtin(PyObject *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000948{
Victor Stinner95872862011-03-07 18:20:56 +0100949 int i, cmp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Victor Stinner95872862011-03-07 18:20:56 +0100951 cmp = PyUnicode_CompareWithASCIIString(name, PyImport_Inittab[i].name);
952 if (cmp == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000953 if (PyImport_Inittab[i].initfunc == NULL)
954 return -1;
955 else
956 return 1;
957 }
958 }
959 return 0;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000960}
961
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000962
Just van Rossum52e14d62002-12-30 22:08:05 +0000963/* Return an importer object for a sys.path/pkg.__path__ item 'p',
964 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +0000965 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +0000966 that can handle the path item. Return None if no hook could;
967 this tells our caller it should fall back to the builtin
968 import mechanism. Cache the result in path_importer_cache.
969 Returns a borrowed reference. */
970
971static PyObject *
972get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000973 PyObject *p)
Just van Rossum52e14d62002-12-30 22:08:05 +0000974{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000975 PyObject *importer;
976 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +0000977
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000978 /* These conditions are the caller's responsibility: */
979 assert(PyList_Check(path_hooks));
980 assert(PyDict_Check(path_importer_cache));
Just van Rossum52e14d62002-12-30 22:08:05 +0000981
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000982 nhooks = PyList_Size(path_hooks);
983 if (nhooks < 0)
984 return NULL; /* Shouldn't happen */
Just van Rossum52e14d62002-12-30 22:08:05 +0000985
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000986 importer = PyDict_GetItem(path_importer_cache, p);
987 if (importer != NULL)
988 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +0000989
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 /* set path_importer_cache[p] to None to avoid recursion */
991 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
992 return NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +0000993
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000994 for (j = 0; j < nhooks; j++) {
995 PyObject *hook = PyList_GetItem(path_hooks, j);
996 if (hook == NULL)
997 return NULL;
998 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
999 if (importer != NULL)
1000 break;
Just van Rossum52e14d62002-12-30 22:08:05 +00001001
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001002 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1003 return NULL;
1004 }
1005 PyErr_Clear();
1006 }
1007 if (importer == NULL) {
Brett Cannonaa936422012-04-27 15:30:58 -04001008 return Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001009 }
1010 if (importer != NULL) {
1011 int err = PyDict_SetItem(path_importer_cache, p, importer);
1012 Py_DECREF(importer);
1013 if (err != 0)
1014 return NULL;
1015 }
1016 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +00001017}
1018
Christian Heimes9cd17752007-11-18 19:35:23 +00001019PyAPI_FUNC(PyObject *)
1020PyImport_GetImporter(PyObject *path) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
Christian Heimes9cd17752007-11-18 19:35:23 +00001022
Victor Stinner1e53bba2013-07-16 22:26:05 +02001023 path_importer_cache = PySys_GetObject("path_importer_cache");
1024 path_hooks = PySys_GetObject("path_hooks");
1025 if (path_importer_cache != NULL && path_hooks != NULL) {
1026 importer = get_path_importer(path_importer_cache,
1027 path_hooks, path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028 }
1029 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1030 return importer;
Christian Heimes9cd17752007-11-18 19:35:23 +00001031}
1032
Nick Coghland5cacbb2015-05-23 22:24:10 +10001033/*[clinic input]
1034_imp.create_builtin
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001035
Nick Coghland5cacbb2015-05-23 22:24:10 +10001036 spec: object
1037 /
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001038
Nick Coghland5cacbb2015-05-23 22:24:10 +10001039Create an extension module.
1040[clinic start generated code]*/
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001041
Nick Coghland5cacbb2015-05-23 22:24:10 +10001042static PyObject *
1043_imp_create_builtin(PyModuleDef *module, PyObject *spec)
1044/*[clinic end generated code: output=5038f467617226bd input=37f966f890384e47]*/
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001045{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001046 struct _inittab *p;
Nick Coghland5cacbb2015-05-23 22:24:10 +10001047 PyObject *name;
1048 char *namestr;
Victor Stinner5eb4f592013-11-14 22:38:52 +01001049 PyObject *mod;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001050
Nick Coghland5cacbb2015-05-23 22:24:10 +10001051 name = PyObject_GetAttrString(spec, "name");
1052 if (name == NULL) {
1053 return NULL;
1054 }
1055
Victor Stinner5eb4f592013-11-14 22:38:52 +01001056 mod = _PyImport_FindExtensionObject(name, name);
Nick Coghland5cacbb2015-05-23 22:24:10 +10001057 if (mod || PyErr_Occurred()) {
1058 Py_DECREF(name);
1059 Py_INCREF(mod);
1060 return mod;
1061 }
1062
1063 namestr = PyUnicode_AsUTF8(name);
1064 if (namestr == NULL) {
1065 Py_DECREF(name);
1066 return NULL;
1067 }
Guido van Rossum25ce5661997-08-02 03:10:38 +00001068
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001069 for (p = PyImport_Inittab; p->name != NULL; p++) {
Antoine Pitrou6c40eb72012-01-18 20:16:09 +01001070 PyModuleDef *def;
Victor Stinner95872862011-03-07 18:20:56 +01001071 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001072 if (p->initfunc == NULL) {
Nick Coghland5cacbb2015-05-23 22:24:10 +10001073 /* Cannot re-init internal module ("sys" or "builtins") */
1074 mod = PyImport_AddModule(namestr);
1075 Py_DECREF(name);
1076 return mod;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001077 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001078 mod = (*p->initfunc)();
Nick Coghland5cacbb2015-05-23 22:24:10 +10001079 if (mod == NULL) {
1080 Py_DECREF(name);
1081 return NULL;
1082 }
1083 if (PyObject_TypeCheck(mod, &PyModuleDef_Type)) {
1084 Py_DECREF(name);
1085 return PyModule_FromDefAndSpec((PyModuleDef*)mod, spec);
1086 } else {
1087 /* Remember pointer to module init function. */
1088 def = PyModule_GetDef(mod);
1089 def->m_base.m_init = p->initfunc;
1090 if (_PyImport_FixupExtensionObject(mod, name, name) < 0) {
1091 Py_DECREF(name);
1092 return NULL;
1093 }
1094 Py_DECREF(name);
1095 return mod;
1096 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001097 }
1098 }
Nick Coghland5cacbb2015-05-23 22:24:10 +10001099 Py_DECREF(name);
1100 Py_RETURN_NONE;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001101}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001102
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001103
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001104/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001105
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001106static const struct _frozen *
Victor Stinner53dc7352011-03-20 01:50:21 +01001107find_frozen(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001108{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001109 const struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001110
Victor Stinner53dc7352011-03-20 01:50:21 +01001111 if (name == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001112 return NULL;
Benjamin Petersond968e272008-11-05 22:48:33 +00001113
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001114 for (p = PyImport_FrozenModules; ; p++) {
1115 if (p->name == NULL)
1116 return NULL;
Victor Stinner53dc7352011-03-20 01:50:21 +01001117 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001118 break;
1119 }
1120 return p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001121}
1122
Guido van Rossum79f25d91997-04-29 20:08:16 +00001123static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001124get_frozen_object(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001125{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001126 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001127 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001129 if (p == NULL) {
1130 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001131 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001132 name);
1133 return NULL;
1134 }
1135 if (p->code == NULL) {
1136 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001137 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001138 name);
1139 return NULL;
1140 }
1141 size = p->size;
1142 if (size < 0)
1143 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001144 return PyMarshal_ReadObjectFromString((const char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001145}
1146
Brett Cannon8d110132009-03-15 02:20:16 +00001147static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001148is_frozen_package(PyObject *name)
Brett Cannon8d110132009-03-15 02:20:16 +00001149{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001150 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001151 int size;
Brett Cannon8d110132009-03-15 02:20:16 +00001152
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001153 if (p == NULL) {
1154 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001155 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001156 name);
1157 return NULL;
1158 }
Brett Cannon8d110132009-03-15 02:20:16 +00001159
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001160 size = p->size;
Brett Cannon8d110132009-03-15 02:20:16 +00001161
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001162 if (size < 0)
1163 Py_RETURN_TRUE;
1164 else
1165 Py_RETURN_FALSE;
Brett Cannon8d110132009-03-15 02:20:16 +00001166}
1167
1168
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001169/* Initialize a frozen module.
Brett Cannon3c2ac442009-03-08 20:49:47 +00001170 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001171 an exception set if the initialization failed.
1172 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001173
1174int
Victor Stinner53dc7352011-03-20 01:50:21 +01001175PyImport_ImportFrozenModuleObject(PyObject *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001176{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001177 const struct _frozen *p;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001178 PyObject *co, *m, *d;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001179 int ispackage;
1180 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001181
Victor Stinner53dc7352011-03-20 01:50:21 +01001182 p = find_frozen(name);
1183
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001184 if (p == NULL)
1185 return 0;
1186 if (p->code == NULL) {
1187 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001188 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001189 name);
1190 return -1;
1191 }
1192 size = p->size;
1193 ispackage = (size < 0);
1194 if (ispackage)
1195 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001196 co = PyMarshal_ReadObjectFromString((const char *)p->code, size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001197 if (co == NULL)
1198 return -1;
1199 if (!PyCode_Check(co)) {
1200 PyErr_Format(PyExc_TypeError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001201 "frozen object %R is not a code object",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 name);
1203 goto err_return;
1204 }
1205 if (ispackage) {
Brett Cannon3e0651b2013-05-31 23:18:39 -04001206 /* Set __path__ to the empty list */
Brett Cannon18fc4e72014-04-04 10:01:46 -04001207 PyObject *l;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 int err;
Victor Stinner53dc7352011-03-20 01:50:21 +01001209 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001210 if (m == NULL)
1211 goto err_return;
1212 d = PyModule_GetDict(m);
Brett Cannon3e0651b2013-05-31 23:18:39 -04001213 l = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 if (l == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001215 goto err_return;
1216 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001217 err = PyDict_SetItemString(d, "__path__", l);
1218 Py_DECREF(l);
1219 if (err != 0)
1220 goto err_return;
1221 }
Brett Cannon18fc4e72014-04-04 10:01:46 -04001222 d = module_dict_for_exec(name);
1223 if (d == NULL) {
Victor Stinner53dc7352011-03-20 01:50:21 +01001224 goto err_return;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001225 }
1226 m = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 if (m == NULL)
1228 goto err_return;
1229 Py_DECREF(co);
1230 Py_DECREF(m);
1231 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001232err_return:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 Py_DECREF(co);
1234 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001235}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001236
Victor Stinner53dc7352011-03-20 01:50:21 +01001237int
Serhiy Storchakac6792272013-10-19 21:03:34 +03001238PyImport_ImportFrozenModule(const char *name)
Victor Stinner53dc7352011-03-20 01:50:21 +01001239{
1240 PyObject *nameobj;
1241 int ret;
1242 nameobj = PyUnicode_InternFromString(name);
1243 if (nameobj == NULL)
1244 return -1;
1245 ret = PyImport_ImportFrozenModuleObject(nameobj);
1246 Py_DECREF(nameobj);
1247 return ret;
1248}
1249
Guido van Rossum74e6a111994-08-29 12:54:38 +00001250
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001251/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001252 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001253
Guido van Rossum79f25d91997-04-29 20:08:16 +00001254PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001255PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001256{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001257 PyObject *pname;
1258 PyObject *result;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001260 pname = PyUnicode_FromString(name);
1261 if (pname == NULL)
1262 return NULL;
1263 result = PyImport_Import(pname);
1264 Py_DECREF(pname);
1265 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001266}
1267
Christian Heimes072c0f12008-01-03 23:01:04 +00001268/* Import a module without blocking
1269 *
1270 * At first it tries to fetch the module from sys.modules. If the module was
1271 * never loaded before it loads it with PyImport_ImportModule() unless another
1272 * thread holds the import lock. In the latter case the function raises an
1273 * ImportError instead of blocking.
1274 *
1275 * Returns the module object with incremented ref count.
1276 */
1277PyObject *
1278PyImport_ImportModuleNoBlock(const char *name)
1279{
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001280 return PyImport_ImportModule(name);
Christian Heimes072c0f12008-01-03 23:01:04 +00001281}
1282
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001283
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001284/* Remove importlib frames from the traceback,
1285 * except in Verbose mode. */
1286static void
1287remove_importlib_frames(void)
1288{
1289 const char *importlib_filename = "<frozen importlib._bootstrap>";
Eric Snow32439d62015-05-02 19:15:18 -06001290 const char *external_filename = "<frozen importlib._bootstrap_external>";
Nick Coghlan42c07662012-07-31 21:14:18 +10001291 const char *remove_frames = "_call_with_frames_removed";
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001292 int always_trim = 0;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001293 int in_importlib = 0;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001294 PyObject *exception, *value, *base_tb, *tb;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001295 PyObject **prev_link, **outer_link = NULL;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001296
1297 /* Synopsis: if it's an ImportError, we trim all importlib chunks
Nick Coghlan42c07662012-07-31 21:14:18 +10001298 from the traceback. We always trim chunks
1299 which end with a call to "_call_with_frames_removed". */
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001300
1301 PyErr_Fetch(&exception, &value, &base_tb);
1302 if (!exception || Py_VerboseFlag)
1303 goto done;
1304 if (PyType_IsSubtype((PyTypeObject *) exception,
1305 (PyTypeObject *) PyExc_ImportError))
1306 always_trim = 1;
1307
1308 prev_link = &base_tb;
1309 tb = base_tb;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001310 while (tb != NULL) {
1311 PyTracebackObject *traceback = (PyTracebackObject *)tb;
1312 PyObject *next = (PyObject *) traceback->tb_next;
1313 PyFrameObject *frame = traceback->tb_frame;
1314 PyCodeObject *code = frame->f_code;
1315 int now_in_importlib;
1316
1317 assert(PyTraceBack_Check(tb));
1318 now_in_importlib = (PyUnicode_CompareWithASCIIString(
1319 code->co_filename,
Eric Snow32439d62015-05-02 19:15:18 -06001320 importlib_filename) == 0) ||
1321 (PyUnicode_CompareWithASCIIString(
1322 code->co_filename,
1323 external_filename) == 0);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001324 if (now_in_importlib && !in_importlib) {
1325 /* This is the link to this chunk of importlib tracebacks */
1326 outer_link = prev_link;
1327 }
1328 in_importlib = now_in_importlib;
1329
1330 if (in_importlib &&
1331 (always_trim ||
Nick Coghlan42c07662012-07-31 21:14:18 +10001332 PyUnicode_CompareWithASCIIString(code->co_name,
1333 remove_frames) == 0)) {
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001334 PyObject *tmp = *outer_link;
1335 *outer_link = next;
1336 Py_XINCREF(next);
1337 Py_DECREF(tmp);
1338 prev_link = outer_link;
1339 }
1340 else {
1341 prev_link = (PyObject **) &traceback->tb_next;
1342 }
1343 tb = next;
1344 }
1345done:
1346 PyErr_Restore(exception, value, base_tb);
1347}
1348
1349
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001350PyObject *
Brett Cannonfd074152012-04-14 14:10:13 -04001351PyImport_ImportModuleLevelObject(PyObject *name, PyObject *given_globals,
1352 PyObject *locals, PyObject *given_fromlist,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001353 int level)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001354{
Brett Cannonfd074152012-04-14 14:10:13 -04001355 _Py_IDENTIFIER(__import__);
Eric Snowb523f842013-11-22 09:05:39 -07001356 _Py_IDENTIFIER(__spec__);
1357 _Py_IDENTIFIER(_initializing);
Brett Cannonfd074152012-04-14 14:10:13 -04001358 _Py_IDENTIFIER(__package__);
1359 _Py_IDENTIFIER(__path__);
1360 _Py_IDENTIFIER(__name__);
1361 _Py_IDENTIFIER(_find_and_load);
1362 _Py_IDENTIFIER(_handle_fromlist);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001363 _Py_IDENTIFIER(_lock_unlock_module);
Brett Cannonfd074152012-04-14 14:10:13 -04001364 _Py_static_string(single_dot, ".");
1365 PyObject *abs_name = NULL;
1366 PyObject *builtins_import = NULL;
1367 PyObject *final_mod = NULL;
1368 PyObject *mod = NULL;
1369 PyObject *package = NULL;
1370 PyObject *globals = NULL;
1371 PyObject *fromlist = NULL;
1372 PyInterpreterState *interp = PyThreadState_GET()->interp;
Serhiy Storchakafa494fd2015-05-30 17:45:22 +03001373 int has_from;
Brett Cannonfd074152012-04-14 14:10:13 -04001374
1375 /* Make sure to use default values so as to not have
1376 PyObject_CallMethodObjArgs() truncate the parameter list because of a
1377 NULL argument. */
1378 if (given_globals == NULL) {
1379 globals = PyDict_New();
1380 if (globals == NULL) {
1381 goto error;
1382 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 }
Brett Cannonfd074152012-04-14 14:10:13 -04001384 else {
1385 /* Only have to care what given_globals is if it will be used
Brett Cannonecfefb72012-08-05 19:24:57 -04001386 for something. */
Brett Cannonfd074152012-04-14 14:10:13 -04001387 if (level > 0 && !PyDict_Check(given_globals)) {
1388 PyErr_SetString(PyExc_TypeError, "globals must be a dict");
1389 goto error;
1390 }
1391 globals = given_globals;
1392 Py_INCREF(globals);
1393 }
1394
1395 if (given_fromlist == NULL) {
1396 fromlist = PyList_New(0);
1397 if (fromlist == NULL) {
1398 goto error;
1399 }
1400 }
1401 else {
1402 fromlist = given_fromlist;
1403 Py_INCREF(fromlist);
1404 }
1405 if (name == NULL) {
1406 PyErr_SetString(PyExc_ValueError, "Empty module name");
1407 goto error;
1408 }
1409
1410 /* The below code is importlib.__import__() & _gcd_import(), ported to C
1411 for added performance. */
1412
1413 if (!PyUnicode_Check(name)) {
1414 PyErr_SetString(PyExc_TypeError, "module name must be a string");
1415 goto error;
1416 }
1417 else if (PyUnicode_READY(name) < 0) {
1418 goto error;
1419 }
1420 if (level < 0) {
1421 PyErr_SetString(PyExc_ValueError, "level must be >= 0");
1422 goto error;
1423 }
1424 else if (level > 0) {
1425 package = _PyDict_GetItemId(globals, &PyId___package__);
1426 if (package != NULL && package != Py_None) {
1427 Py_INCREF(package);
1428 if (!PyUnicode_Check(package)) {
1429 PyErr_SetString(PyExc_TypeError, "package must be a string");
1430 goto error;
1431 }
1432 }
1433 else {
Brett Cannon273323c2012-04-17 19:05:11 -04001434 package = _PyDict_GetItemId(globals, &PyId___name__);
Brett Cannonfd074152012-04-14 14:10:13 -04001435 if (package == NULL) {
Brett Cannon273323c2012-04-17 19:05:11 -04001436 PyErr_SetString(PyExc_KeyError, "'__name__' not in globals");
Brett Cannonfd074152012-04-14 14:10:13 -04001437 goto error;
1438 }
1439 else if (!PyUnicode_Check(package)) {
1440 PyErr_SetString(PyExc_TypeError, "__name__ must be a string");
1441 }
1442 Py_INCREF(package);
1443
1444 if (_PyDict_GetItemId(globals, &PyId___path__) == NULL) {
Brett Cannon740fce02012-04-14 14:23:49 -04001445 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001446 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1447 if (borrowed_dot == NULL) {
1448 goto error;
1449 }
Brett Cannon740fce02012-04-14 14:23:49 -04001450 partition = PyUnicode_RPartition(package, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001451 Py_DECREF(package);
1452 if (partition == NULL) {
1453 goto error;
1454 }
1455 package = PyTuple_GET_ITEM(partition, 0);
1456 Py_INCREF(package);
1457 Py_DECREF(partition);
1458 }
1459 }
1460
1461 if (PyDict_GetItem(interp->modules, package) == NULL) {
1462 PyErr_Format(PyExc_SystemError,
1463 "Parent module %R not loaded, cannot perform relative "
1464 "import", package);
1465 goto error;
1466 }
1467 }
1468 else { /* level == 0 */
1469 if (PyUnicode_GET_LENGTH(name) == 0) {
1470 PyErr_SetString(PyExc_ValueError, "Empty module name");
1471 goto error;
1472 }
1473 package = Py_None;
1474 Py_INCREF(package);
1475 }
1476
1477 if (level > 0) {
1478 Py_ssize_t last_dot = PyUnicode_GET_LENGTH(package);
1479 PyObject *base = NULL;
1480 int level_up = 1;
1481
1482 for (level_up = 1; level_up < level; level_up += 1) {
1483 last_dot = PyUnicode_FindChar(package, '.', 0, last_dot, -1);
1484 if (last_dot == -2) {
1485 goto error;
1486 }
1487 else if (last_dot == -1) {
1488 PyErr_SetString(PyExc_ValueError,
1489 "attempted relative import beyond top-level "
1490 "package");
1491 goto error;
1492 }
1493 }
Victor Stinner22af2592013-11-13 12:11:36 +01001494
Brett Cannonfd074152012-04-14 14:10:13 -04001495 base = PyUnicode_Substring(package, 0, last_dot);
Victor Stinner22af2592013-11-13 12:11:36 +01001496 if (base == NULL)
1497 goto error;
1498
Brett Cannonfd074152012-04-14 14:10:13 -04001499 if (PyUnicode_GET_LENGTH(name) > 0) {
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001500 PyObject *borrowed_dot, *seq = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001501
1502 borrowed_dot = _PyUnicode_FromId(&single_dot);
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001503 seq = PyTuple_Pack(2, base, name);
1504 Py_DECREF(base);
Brett Cannonfd074152012-04-14 14:10:13 -04001505 if (borrowed_dot == NULL || seq == NULL) {
1506 goto error;
1507 }
1508
1509 abs_name = PyUnicode_Join(borrowed_dot, seq);
1510 Py_DECREF(seq);
1511 if (abs_name == NULL) {
1512 goto error;
1513 }
1514 }
1515 else {
1516 abs_name = base;
1517 }
1518 }
1519 else {
1520 abs_name = name;
1521 Py_INCREF(abs_name);
1522 }
1523
Brian Curtine6b299f2012-04-14 14:19:33 -05001524#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001525 _PyImport_AcquireLock();
1526#endif
1527 /* From this point forward, goto error_with_unlock! */
1528 if (PyDict_Check(globals)) {
1529 builtins_import = _PyDict_GetItemId(globals, &PyId___import__);
1530 }
1531 if (builtins_import == NULL) {
1532 builtins_import = _PyDict_GetItemId(interp->builtins, &PyId___import__);
1533 if (builtins_import == NULL) {
Benjamin Peterson7d110042013-04-29 09:08:14 -04001534 PyErr_SetString(PyExc_ImportError, "__import__ not found");
1535 goto error_with_unlock;
Brett Cannonfd074152012-04-14 14:10:13 -04001536 }
1537 }
1538 Py_INCREF(builtins_import);
1539
1540 mod = PyDict_GetItem(interp->modules, abs_name);
1541 if (mod == Py_None) {
Brett Cannon27fc5282012-04-15 14:15:31 -04001542 PyObject *msg = PyUnicode_FromFormat("import of %R halted; "
1543 "None in sys.modules", abs_name);
1544 if (msg != NULL) {
Brett Cannon82da8882013-07-04 17:48:16 -04001545 PyErr_SetImportError(msg, abs_name, NULL);
Brian Curtin09b86d12012-04-17 16:57:09 -05001546 Py_DECREF(msg);
Brett Cannon27fc5282012-04-15 14:15:31 -04001547 }
Antoine Pitrou71382cb2012-04-16 18:48:49 +02001548 mod = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001549 goto error_with_unlock;
1550 }
1551 else if (mod != NULL) {
Eric Snowb523f842013-11-22 09:05:39 -07001552 PyObject *value = NULL;
1553 PyObject *spec;
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001554 int initializing = 0;
1555
Brett Cannonfd074152012-04-14 14:10:13 -04001556 Py_INCREF(mod);
Antoine Pitrou03989852012-08-28 00:24:52 +02001557 /* Optimization: only call _bootstrap._lock_unlock_module() if
Eric Snowb523f842013-11-22 09:05:39 -07001558 __spec__._initializing is true.
1559 NOTE: because of this, initializing must be set *before*
Antoine Pitrou03989852012-08-28 00:24:52 +02001560 stuffing the new module in sys.modules.
1561 */
Eric Snowb523f842013-11-22 09:05:39 -07001562 spec = _PyObject_GetAttrId(mod, &PyId___spec__);
1563 if (spec != NULL) {
1564 value = _PyObject_GetAttrId(spec, &PyId__initializing);
1565 Py_DECREF(spec);
1566 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001567 if (value == NULL)
1568 PyErr_Clear();
1569 else {
1570 initializing = PyObject_IsTrue(value);
1571 Py_DECREF(value);
1572 if (initializing == -1)
1573 PyErr_Clear();
1574 }
1575 if (initializing > 0) {
1576 /* _bootstrap._lock_unlock_module() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001577 value = _PyObject_CallMethodIdObjArgs(interp->importlib,
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001578 &PyId__lock_unlock_module, abs_name,
1579 NULL);
1580 if (value == NULL)
1581 goto error;
1582 Py_DECREF(value);
1583 }
1584 else {
1585#ifdef WITH_THREAD
1586 if (_PyImport_ReleaseLock() < 0) {
1587 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1588 goto error;
1589 }
1590#endif
1591 }
Brett Cannonfd074152012-04-14 14:10:13 -04001592 }
1593 else {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001594 /* _bootstrap._find_and_load() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001595 mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001596 &PyId__find_and_load, abs_name,
1597 builtins_import, NULL);
1598 if (mod == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001599 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001600 }
1601 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001602 /* From now on we don't hold the import lock anymore. */
Brett Cannonfd074152012-04-14 14:10:13 -04001603
Serhiy Storchakafa494fd2015-05-30 17:45:22 +03001604 has_from = PyObject_IsTrue(fromlist);
1605 if (has_from < 0)
1606 goto error;
1607 if (!has_from) {
Brett Cannonfd074152012-04-14 14:10:13 -04001608 if (level == 0 || PyUnicode_GET_LENGTH(name) > 0) {
1609 PyObject *front = NULL;
Brian Curtine6b299f2012-04-14 14:19:33 -05001610 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001611 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1612
1613 if (borrowed_dot == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001614 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001615 }
1616
Brian Curtine6b299f2012-04-14 14:19:33 -05001617 partition = PyUnicode_Partition(name, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001618 if (partition == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001619 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001620 }
1621
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001622 if (PyUnicode_GET_LENGTH(PyTuple_GET_ITEM(partition, 1)) == 0) {
1623 /* No dot in module name, simple exit */
1624 Py_DECREF(partition);
1625 final_mod = mod;
1626 Py_INCREF(mod);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001627 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001628 }
1629
Brett Cannonfd074152012-04-14 14:10:13 -04001630 front = PyTuple_GET_ITEM(partition, 0);
1631 Py_INCREF(front);
1632 Py_DECREF(partition);
1633
1634 if (level == 0) {
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001635 final_mod = PyObject_CallFunctionObjArgs(builtins_import, front, NULL);
Antoine Pitroub78174c2012-05-06 17:15:23 +02001636 Py_DECREF(front);
Brett Cannonfd074152012-04-14 14:10:13 -04001637 }
1638 else {
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001639 Py_ssize_t cut_off = PyUnicode_GET_LENGTH(name) -
1640 PyUnicode_GET_LENGTH(front);
1641 Py_ssize_t abs_name_len = PyUnicode_GET_LENGTH(abs_name);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001642 PyObject *to_return = PyUnicode_Substring(abs_name, 0,
Brett Cannonfd074152012-04-14 14:10:13 -04001643 abs_name_len - cut_off);
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001644 Py_DECREF(front);
1645 if (to_return == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001646 goto error;
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001647 }
Brett Cannonfd074152012-04-14 14:10:13 -04001648
1649 final_mod = PyDict_GetItem(interp->modules, to_return);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001650 if (final_mod == NULL) {
1651 PyErr_Format(PyExc_KeyError,
1652 "%R not in sys.modules as expected",
1653 to_return);
1654 }
1655 else {
1656 Py_INCREF(final_mod);
1657 }
Antoine Pitroub78174c2012-05-06 17:15:23 +02001658 Py_DECREF(to_return);
Brett Cannonfd074152012-04-14 14:10:13 -04001659 }
1660 }
1661 else {
1662 final_mod = mod;
1663 Py_INCREF(mod);
1664 }
1665 }
1666 else {
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001667 final_mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001668 &PyId__handle_fromlist, mod,
1669 fromlist, builtins_import,
1670 NULL);
1671 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001672 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001673
Brett Cannonfd074152012-04-14 14:10:13 -04001674 error_with_unlock:
Brian Curtine6b299f2012-04-14 14:19:33 -05001675#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001676 if (_PyImport_ReleaseLock() < 0) {
1677 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1678 }
1679#endif
1680 error:
1681 Py_XDECREF(abs_name);
1682 Py_XDECREF(builtins_import);
1683 Py_XDECREF(mod);
1684 Py_XDECREF(package);
1685 Py_XDECREF(globals);
1686 Py_XDECREF(fromlist);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001687 if (final_mod == NULL)
1688 remove_importlib_frames();
Brett Cannonfd074152012-04-14 14:10:13 -04001689 return final_mod;
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001690}
1691
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001692PyObject *
Benjamin Peterson04778a82011-05-25 09:29:00 -05001693PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001694 PyObject *fromlist, int level)
1695{
1696 PyObject *nameobj, *mod;
1697 nameobj = PyUnicode_FromString(name);
1698 if (nameobj == NULL)
1699 return NULL;
1700 mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
1701 fromlist, level);
1702 Py_DECREF(nameobj);
1703 return mod;
1704}
1705
1706
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001707/* Re-import a module of any kind and return its module object, WITH
1708 INCREMENTED REFERENCE COUNT */
1709
Guido van Rossum79f25d91997-04-29 20:08:16 +00001710PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001711PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001712{
Brett Cannon62228db2012-04-29 14:38:11 -04001713 _Py_IDENTIFIER(reload);
1714 PyObject *reloaded_module = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001715 PyObject *modules = PyImport_GetModuleDict();
Brett Cannon62228db2012-04-29 14:38:11 -04001716 PyObject *imp = PyDict_GetItemString(modules, "imp");
1717 if (imp == NULL) {
1718 imp = PyImport_ImportModule("imp");
1719 if (imp == NULL) {
1720 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001721 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001722 }
Brett Cannon62228db2012-04-29 14:38:11 -04001723 else {
1724 Py_INCREF(imp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001725 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001726
Brett Cannon62228db2012-04-29 14:38:11 -04001727 reloaded_module = _PyObject_CallMethodId(imp, &PyId_reload, "O", m);
1728 Py_DECREF(imp);
1729 return reloaded_module;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001730}
1731
1732
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001733/* Higher-level import emulator which emulates the "import" statement
1734 more accurately -- it invokes the __import__() function from the
1735 builtins of the current globals. This means that the import is
1736 done using whatever import hooks are installed in the current
Brett Cannonbc2eff32010-09-19 21:39:02 +00001737 environment.
Guido van Rossum6058eb41998-12-21 19:51:00 +00001738 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00001739 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00001740 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001741
1742PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001743PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001744{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001745 static PyObject *silly_list = NULL;
1746 static PyObject *builtins_str = NULL;
1747 static PyObject *import_str = NULL;
1748 PyObject *globals = NULL;
1749 PyObject *import = NULL;
1750 PyObject *builtins = NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001751 PyObject *modules = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001752 PyObject *r = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001753
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001754 /* Initialize constant string objects */
1755 if (silly_list == NULL) {
1756 import_str = PyUnicode_InternFromString("__import__");
1757 if (import_str == NULL)
1758 return NULL;
1759 builtins_str = PyUnicode_InternFromString("__builtins__");
1760 if (builtins_str == NULL)
1761 return NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001762 silly_list = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001763 if (silly_list == NULL)
1764 return NULL;
1765 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001766
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001767 /* Get the builtins from current globals */
1768 globals = PyEval_GetGlobals();
1769 if (globals != NULL) {
1770 Py_INCREF(globals);
1771 builtins = PyObject_GetItem(globals, builtins_str);
1772 if (builtins == NULL)
1773 goto err;
1774 }
1775 else {
1776 /* No globals -- use standard builtins, and fake globals */
1777 builtins = PyImport_ImportModuleLevel("builtins",
1778 NULL, NULL, NULL, 0);
1779 if (builtins == NULL)
1780 return NULL;
1781 globals = Py_BuildValue("{OO}", builtins_str, builtins);
1782 if (globals == NULL)
1783 goto err;
1784 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001785
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001786 /* Get the __import__ function from the builtins */
1787 if (PyDict_Check(builtins)) {
1788 import = PyObject_GetItem(builtins, import_str);
1789 if (import == NULL)
1790 PyErr_SetObject(PyExc_KeyError, import_str);
1791 }
1792 else
1793 import = PyObject_GetAttr(builtins, import_str);
1794 if (import == NULL)
1795 goto err;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001796
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001797 /* Call the __import__ function with the proper argument list
Brett Cannonbc2eff32010-09-19 21:39:02 +00001798 Always use absolute import here.
1799 Calling for side-effect of import. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001800 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
1801 globals, silly_list, 0, NULL);
Brett Cannonbc2eff32010-09-19 21:39:02 +00001802 if (r == NULL)
1803 goto err;
1804 Py_DECREF(r);
1805
1806 modules = PyImport_GetModuleDict();
1807 r = PyDict_GetItem(modules, module_name);
1808 if (r != NULL)
1809 Py_INCREF(r);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001810
1811 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001812 Py_XDECREF(globals);
1813 Py_XDECREF(builtins);
1814 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00001815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001816 return r;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001817}
1818
Brett Cannon4caa61d2014-01-09 19:03:32 -05001819/*[clinic input]
1820_imp.extension_suffixes
1821
1822Returns the list of file suffixes used to identify extension modules.
1823[clinic start generated code]*/
1824
Brett Cannon4caa61d2014-01-09 19:03:32 -05001825static PyObject *
1826_imp_extension_suffixes_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001827/*[clinic end generated code: output=d44c1566ef362229 input=ecdeeecfcb6f839e]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001828{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001829 PyObject *list;
Brett Cannon2657df42012-05-04 15:20:40 -04001830 const char *suffix;
1831 unsigned int index = 0;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001832
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001833 list = PyList_New(0);
1834 if (list == NULL)
1835 return NULL;
Brett Cannon2657df42012-05-04 15:20:40 -04001836#ifdef HAVE_DYNAMIC_LOADING
1837 while ((suffix = _PyImport_DynLoadFiletab[index])) {
1838 PyObject *item = PyUnicode_FromString(suffix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001839 if (item == NULL) {
1840 Py_DECREF(list);
1841 return NULL;
1842 }
1843 if (PyList_Append(list, item) < 0) {
1844 Py_DECREF(list);
1845 Py_DECREF(item);
1846 return NULL;
1847 }
1848 Py_DECREF(item);
Brett Cannon2657df42012-05-04 15:20:40 -04001849 index += 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001850 }
Brett Cannon2657df42012-05-04 15:20:40 -04001851#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001852 return list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001853}
1854
Brett Cannon4caa61d2014-01-09 19:03:32 -05001855/*[clinic input]
Brett Cannon4caa61d2014-01-09 19:03:32 -05001856_imp.init_frozen
1857
1858 name: unicode
1859 /
1860
1861Initializes a frozen module.
1862[clinic start generated code]*/
1863
Brett Cannon4caa61d2014-01-09 19:03:32 -05001864static PyObject *
1865_imp_init_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001866/*[clinic end generated code: output=a9de493bdd711878 input=13019adfc04f3fb3]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001867{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001868 int ret;
1869 PyObject *m;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001870
Victor Stinner53dc7352011-03-20 01:50:21 +01001871 ret = PyImport_ImportFrozenModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001872 if (ret < 0)
1873 return NULL;
1874 if (ret == 0) {
1875 Py_INCREF(Py_None);
1876 return Py_None;
1877 }
Victor Stinner53dc7352011-03-20 01:50:21 +01001878 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001879 Py_XINCREF(m);
1880 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001881}
1882
Brett Cannon4caa61d2014-01-09 19:03:32 -05001883/*[clinic input]
1884_imp.get_frozen_object
1885
1886 name: unicode
1887 /
1888
1889Create a code object for a frozen module.
1890[clinic start generated code]*/
1891
Brett Cannon4caa61d2014-01-09 19:03:32 -05001892static PyObject *
1893_imp_get_frozen_object_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001894/*[clinic end generated code: output=3114c970a47f2e3c input=ed689bc05358fdbd]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001895{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001896 return get_frozen_object(name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001897}
1898
Brett Cannon4caa61d2014-01-09 19:03:32 -05001899/*[clinic input]
1900_imp.is_frozen_package
1901
1902 name: unicode
1903 /
1904
1905Returns True if the module name is of a frozen package.
1906[clinic start generated code]*/
1907
Brett Cannon4caa61d2014-01-09 19:03:32 -05001908static PyObject *
1909_imp_is_frozen_package_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001910/*[clinic end generated code: output=3e4cab802b56d649 input=81b6cdecd080fbb8]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001911{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001912 return is_frozen_package(name);
Brett Cannon8d110132009-03-15 02:20:16 +00001913}
1914
Brett Cannon4caa61d2014-01-09 19:03:32 -05001915/*[clinic input]
1916_imp.is_builtin
1917
1918 name: unicode
1919 /
1920
1921Returns True if the module name corresponds to a built-in module.
1922[clinic start generated code]*/
1923
Guido van Rossum79f25d91997-04-29 20:08:16 +00001924static PyObject *
Brett Cannon4caa61d2014-01-09 19:03:32 -05001925_imp_is_builtin_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001926/*[clinic end generated code: output=2deec9cac6fb9a7e input=86befdac021dd1c7]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001927{
Brett Cannon4caa61d2014-01-09 19:03:32 -05001928 return PyLong_FromLong(is_builtin(name));
1929}
1930
1931/*[clinic input]
1932_imp.is_frozen
1933
1934 name: unicode
1935 /
1936
1937Returns True if the module name corresponds to a frozen module.
1938[clinic start generated code]*/
1939
Brett Cannon4caa61d2014-01-09 19:03:32 -05001940static PyObject *
1941_imp_is_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001942/*[clinic end generated code: output=7de8e260c8e36aed input=7301dbca1897d66b]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001943{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001944 const struct _frozen *p;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001945
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001946 p = find_frozen(name);
1947 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001948}
1949
Larry Hastings1df0b352015-08-24 19:53:56 -07001950/* Common implementation for _imp.exec_dynamic and _imp.exec_builtin */
1951static int
1952exec_builtin_or_dynamic(PyObject *mod) {
1953 PyModuleDef *def;
1954 void *state;
1955
1956 if (!PyModule_Check(mod)) {
1957 return 0;
1958 }
1959
1960 def = PyModule_GetDef(mod);
1961 if (def == NULL) {
1962 if (PyErr_Occurred()) {
1963 return -1;
1964 }
1965 return 0;
1966 }
1967 state = PyModule_GetState(mod);
1968 if (PyErr_Occurred()) {
1969 return -1;
1970 }
1971 if (state) {
1972 /* Already initialized; skip reload */
1973 return 0;
1974 }
1975 return PyModule_ExecDef(mod, def);
1976}
1977
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001978#ifdef HAVE_DYNAMIC_LOADING
1979
Brett Cannon4caa61d2014-01-09 19:03:32 -05001980/*[clinic input]
Nick Coghland5cacbb2015-05-23 22:24:10 +10001981_imp.create_dynamic
Brett Cannon4caa61d2014-01-09 19:03:32 -05001982
Nick Coghland5cacbb2015-05-23 22:24:10 +10001983 spec: object
Brett Cannon4caa61d2014-01-09 19:03:32 -05001984 file: object = NULL
1985 /
1986
Nick Coghland5cacbb2015-05-23 22:24:10 +10001987Create an extension module.
Brett Cannon4caa61d2014-01-09 19:03:32 -05001988[clinic start generated code]*/
1989
Brett Cannon4caa61d2014-01-09 19:03:32 -05001990static PyObject *
Nick Coghland5cacbb2015-05-23 22:24:10 +10001991_imp_create_dynamic_impl(PyModuleDef *module, PyObject *spec, PyObject *file)
1992/*[clinic end generated code: output=935cde5b3872d56d input=c31b954f4cf4e09d]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001993{
Nick Coghland5cacbb2015-05-23 22:24:10 +10001994 PyObject *mod, *name, *path;
Victor Stinnerfefd70c2011-03-14 15:54:07 -04001995 FILE *fp;
1996
Nick Coghland5cacbb2015-05-23 22:24:10 +10001997 name = PyObject_GetAttrString(spec, "name");
1998 if (name == NULL) {
1999 return NULL;
2000 }
2001
2002 path = PyObject_GetAttrString(spec, "origin");
2003 if (path == NULL) {
2004 Py_DECREF(name);
2005 return NULL;
2006 }
2007
2008 mod = _PyImport_FindExtensionObject(name, path);
2009 if (mod != NULL) {
2010 Py_DECREF(name);
2011 Py_DECREF(path);
2012 Py_INCREF(mod);
2013 return mod;
2014 }
2015
Brett Cannon4caa61d2014-01-09 19:03:32 -05002016 if (file != NULL) {
2017 fp = _Py_fopen_obj(path, "r");
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002018 if (fp == NULL) {
Nick Coghland5cacbb2015-05-23 22:24:10 +10002019 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002020 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002021 return NULL;
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002022 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002023 }
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002024 else
2025 fp = NULL;
Nick Coghland5cacbb2015-05-23 22:24:10 +10002026
2027 mod = _PyImport_LoadDynamicModuleWithSpec(spec, fp);
2028
2029 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002030 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002031 if (fp)
2032 fclose(fp);
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002033 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002034}
2035
Nick Coghland5cacbb2015-05-23 22:24:10 +10002036/*[clinic input]
2037_imp.exec_dynamic -> int
2038
2039 mod: object
2040 /
2041
2042Initialize an extension module.
2043[clinic start generated code]*/
2044
2045static int
2046_imp_exec_dynamic_impl(PyModuleDef *module, PyObject *mod)
2047/*[clinic end generated code: output=4b84f1301b22d4bd input=9fdbfcb250280d3a]*/
2048{
Larry Hastings1df0b352015-08-24 19:53:56 -07002049 return exec_builtin_or_dynamic(mod);
Nick Coghland5cacbb2015-05-23 22:24:10 +10002050}
2051
2052
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002053#endif /* HAVE_DYNAMIC_LOADING */
2054
Larry Hastings7726ac92014-01-31 22:03:12 -08002055/*[clinic input]
Larry Hastings1df0b352015-08-24 19:53:56 -07002056_imp.exec_builtin -> int
2057
2058 mod: object
2059 /
2060
2061Initialize a built-in module.
2062[clinic start generated code]*/
2063
2064static int
2065_imp_exec_builtin_impl(PyModuleDef *module, PyObject *mod)
Larry Hastings01b1ff62015-08-24 20:23:27 -07002066/*[clinic end generated code: output=215e99876a27e284 input=7beed5a2f12a60ca]*/
Larry Hastings1df0b352015-08-24 19:53:56 -07002067{
2068 return exec_builtin_or_dynamic(mod);
2069}
2070
2071/*[clinic input]
Larry Hastings7726ac92014-01-31 22:03:12 -08002072dump buffer
2073[clinic start generated code]*/
Brett Cannonfd4d0502014-05-30 11:21:14 -04002074/*[clinic end generated code: output=da39a3ee5e6b4b0d input=524ce2e021e4eba6]*/
Larry Hastings7726ac92014-01-31 22:03:12 -08002075
Barry Warsaw28a691b2010-04-17 00:19:56 +00002076
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002077PyDoc_STRVAR(doc_imp,
Brett Cannon6f44d662012-04-15 16:08:47 -04002078"(Extremely) low-level import machinery bits as used by importlib and imp.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002079
Guido van Rossum79f25d91997-04-29 20:08:16 +00002080static PyMethodDef imp_methods[] = {
Brett Cannon4caa61d2014-01-09 19:03:32 -05002081 _IMP_EXTENSION_SUFFIXES_METHODDEF
2082 _IMP_LOCK_HELD_METHODDEF
2083 _IMP_ACQUIRE_LOCK_METHODDEF
2084 _IMP_RELEASE_LOCK_METHODDEF
2085 _IMP_GET_FROZEN_OBJECT_METHODDEF
2086 _IMP_IS_FROZEN_PACKAGE_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002087 _IMP_CREATE_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002088 _IMP_INIT_FROZEN_METHODDEF
2089 _IMP_IS_BUILTIN_METHODDEF
2090 _IMP_IS_FROZEN_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002091 _IMP_CREATE_DYNAMIC_METHODDEF
2092 _IMP_EXEC_DYNAMIC_METHODDEF
Larry Hastings1df0b352015-08-24 19:53:56 -07002093 _IMP_EXEC_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002094 _IMP__FIX_CO_FILENAME_METHODDEF
2095 {NULL, NULL} /* sentinel */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002096};
2097
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002098
Martin v. Löwis1a214512008-06-11 05:26:20 +00002099static struct PyModuleDef impmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002100 PyModuleDef_HEAD_INIT,
Brett Cannon6f44d662012-04-15 16:08:47 -04002101 "_imp",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002102 doc_imp,
2103 0,
2104 imp_methods,
2105 NULL,
2106 NULL,
2107 NULL,
2108 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00002109};
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002110
Jason Tishler6bc06ec2003-09-04 11:59:50 +00002111PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00002112PyInit_imp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002113{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002114 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002115
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002116 m = PyModule_Create(&impmodule);
2117 if (m == NULL)
2118 goto failure;
2119 d = PyModule_GetDict(m);
2120 if (d == NULL)
2121 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002123 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002124 failure:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002125 Py_XDECREF(m);
2126 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002127}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002128
2129
Guido van Rossumb18618d2000-05-03 23:44:39 +00002130/* API for embedding applications that want to add their own entries
2131 to the table of built-in modules. This should normally be called
2132 *before* Py_Initialize(). When the table resize fails, -1 is
2133 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002134
2135 After a similar function by Just van Rossum. */
2136
2137int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002138PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002139{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002140 static struct _inittab *our_copy = NULL;
2141 struct _inittab *p;
2142 int i, n;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002143
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002144 /* Count the number of entries in both tables */
2145 for (n = 0; newtab[n].name != NULL; n++)
2146 ;
2147 if (n == 0)
2148 return 0; /* Nothing to do */
2149 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2150 ;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002151
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002152 /* Allocate new memory for the combined table */
2153 p = our_copy;
2154 PyMem_RESIZE(p, struct _inittab, i+n+1);
2155 if (p == NULL)
2156 return -1;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002157
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002158 /* Copy the tables into the new memory */
2159 if (our_copy != PyImport_Inittab)
2160 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2161 PyImport_Inittab = our_copy = p;
2162 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002164 return 0;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002165}
2166
2167/* Shorthand to add a single entry given a name and a function */
2168
2169int
Brett Cannona826f322009-04-02 03:41:46 +00002170PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002171{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002172 struct _inittab newtab[2];
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002174 memset(newtab, '\0', sizeof newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002175
Serhiy Storchaka20b39b22014-09-28 11:27:24 +03002176 newtab[0].name = name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002177 newtab[0].initfunc = initfunc;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002178
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002179 return PyImport_ExtendInittab(newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002180}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002181
2182#ifdef __cplusplus
2183}
2184#endif