blob: c1071c0de3eb25744b3ff9ae4b946eb2285a4f65 [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 */
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200323static const char * const 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
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200333static const char * const 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 Storchaka2d06e842015-12-25 19:53:18 +0200350 const char * const *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
Victor Stinner27ee0892011-03-04 12:57:09 +0000674 if ((m = PyDict_GetItem(modules, name)) != NULL &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000675 PyModule_Check(m))
676 return m;
Victor Stinner27ee0892011-03-04 12:57:09 +0000677 m = PyModule_NewObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000678 if (m == NULL)
679 return NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000680 if (PyDict_SetItem(modules, name, m) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 Py_DECREF(m);
682 return NULL;
683 }
684 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000687}
688
Victor Stinner27ee0892011-03-04 12:57:09 +0000689PyObject *
690PyImport_AddModule(const char *name)
691{
692 PyObject *nameobj, *module;
693 nameobj = PyUnicode_FromString(name);
694 if (nameobj == NULL)
695 return NULL;
696 module = PyImport_AddModuleObject(nameobj);
697 Py_DECREF(nameobj);
698 return module;
699}
700
701
Tim Peters1cd70172004-08-02 03:52:12 +0000702/* Remove name from sys.modules, if it's there. */
703static void
Victor Stinner27ee0892011-03-04 12:57:09 +0000704remove_module(PyObject *name)
Tim Peters1cd70172004-08-02 03:52:12 +0000705{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000706 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner27ee0892011-03-04 12:57:09 +0000707 if (PyDict_GetItem(modules, name) == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000708 return;
Victor Stinner27ee0892011-03-04 12:57:09 +0000709 if (PyDict_DelItem(modules, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 Py_FatalError("import: deleting existing key in"
711 "sys.modules failed");
Tim Peters1cd70172004-08-02 03:52:12 +0000712}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000713
Christian Heimes3b06e532008-01-07 20:12:44 +0000714
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000715/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000716 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
717 * removed from sys.modules, to avoid leaving damaged module objects
718 * in sys.modules. The caller may wish to restore the original
719 * module object (if any) in this case; PyImport_ReloadModule is an
720 * example.
Barry Warsaw28a691b2010-04-17 00:19:56 +0000721 *
722 * Note that PyImport_ExecCodeModuleWithPathnames() is the preferred, richer
723 * interface. The other two exist primarily for backward compatibility.
Tim Peters1cd70172004-08-02 03:52:12 +0000724 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000725PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300726PyImport_ExecCodeModule(const char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000727{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000728 return PyImport_ExecCodeModuleWithPathnames(
729 name, co, (char *)NULL, (char *)NULL);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000730}
731
732PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300733PyImport_ExecCodeModuleEx(const char *name, PyObject *co, const char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000734{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000735 return PyImport_ExecCodeModuleWithPathnames(
736 name, co, pathname, (char *)NULL);
Barry Warsaw28a691b2010-04-17 00:19:56 +0000737}
738
739PyObject *
Serhiy Storchakac6792272013-10-19 21:03:34 +0300740PyImport_ExecCodeModuleWithPathnames(const char *name, PyObject *co,
741 const char *pathname,
742 const char *cpathname)
Barry Warsaw28a691b2010-04-17 00:19:56 +0000743{
Victor Stinner27ee0892011-03-04 12:57:09 +0000744 PyObject *m = NULL;
Eric Snow32439d62015-05-02 19:15:18 -0600745 PyObject *nameobj, *pathobj = NULL, *cpathobj = NULL, *external= NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000746
747 nameobj = PyUnicode_FromString(name);
748 if (nameobj == NULL)
749 return NULL;
750
Victor Stinner27ee0892011-03-04 12:57:09 +0000751 if (cpathname != NULL) {
752 cpathobj = PyUnicode_DecodeFSDefault(cpathname);
753 if (cpathobj == NULL)
754 goto error;
Brett Cannona6473f92012-07-13 13:57:03 -0400755 }
756 else
Victor Stinner27ee0892011-03-04 12:57:09 +0000757 cpathobj = NULL;
Brett Cannona6473f92012-07-13 13:57:03 -0400758
759 if (pathname != NULL) {
760 pathobj = PyUnicode_DecodeFSDefault(pathname);
761 if (pathobj == NULL)
762 goto error;
763 }
764 else if (cpathobj != NULL) {
765 PyInterpreterState *interp = PyThreadState_GET()->interp;
766 _Py_IDENTIFIER(_get_sourcefile);
767
768 if (interp == NULL) {
769 Py_FatalError("PyImport_ExecCodeModuleWithPathnames: "
770 "no interpreter!");
771 }
772
Eric Snow32439d62015-05-02 19:15:18 -0600773 external= PyObject_GetAttrString(interp->importlib,
774 "_bootstrap_external");
775 if (external != NULL) {
776 pathobj = _PyObject_CallMethodIdObjArgs(external,
777 &PyId__get_sourcefile, cpathobj,
778 NULL);
779 Py_DECREF(external);
780 }
Brett Cannona6473f92012-07-13 13:57:03 -0400781 if (pathobj == NULL)
782 PyErr_Clear();
783 }
784 else
785 pathobj = NULL;
786
Victor Stinner27ee0892011-03-04 12:57:09 +0000787 m = PyImport_ExecCodeModuleObject(nameobj, co, pathobj, cpathobj);
788error:
789 Py_DECREF(nameobj);
790 Py_XDECREF(pathobj);
791 Py_XDECREF(cpathobj);
792 return m;
793}
794
Brett Cannon18fc4e72014-04-04 10:01:46 -0400795static PyObject *
796module_dict_for_exec(PyObject *name)
Victor Stinner27ee0892011-03-04 12:57:09 +0000797{
Brett Cannon18fc4e72014-04-04 10:01:46 -0400798 PyObject *m, *d = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000799
Victor Stinner27ee0892011-03-04 12:57:09 +0000800 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000801 if (m == NULL)
802 return NULL;
803 /* If the module is being reloaded, we get the old module back
804 and re-use its dict to exec the new code. */
805 d = PyModule_GetDict(m);
806 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
807 if (PyDict_SetItemString(d, "__builtins__",
Brett Cannon18fc4e72014-04-04 10:01:46 -0400808 PyEval_GetBuiltins()) != 0) {
809 remove_module(name);
810 return NULL;
811 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000812 }
Brett Cannon18fc4e72014-04-04 10:01:46 -0400813
Eric Snow08197a42014-05-12 17:54:55 -0600814 return d; /* Return a borrowed reference. */
Brett Cannon18fc4e72014-04-04 10:01:46 -0400815}
816
817static PyObject *
818exec_code_in_module(PyObject *name, PyObject *module_dict, PyObject *code_object)
819{
820 PyObject *modules = PyImport_GetModuleDict();
821 PyObject *v, *m;
822
823 v = PyEval_EvalCode(code_object, module_dict, module_dict);
824 if (v == NULL) {
825 remove_module(name);
826 return NULL;
827 }
828 Py_DECREF(v);
829
830 if ((m = PyDict_GetItem(modules, name)) == NULL) {
831 PyErr_Format(PyExc_ImportError,
832 "Loaded module %R not found in sys.modules",
833 name);
834 return NULL;
835 }
836
837 Py_INCREF(m);
838
839 return m;
840}
841
842PyObject*
843PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
844 PyObject *cpathname)
845{
Eric Snow32439d62015-05-02 19:15:18 -0600846 PyObject *d, *external, *res;
Eric Snow08197a42014-05-12 17:54:55 -0600847 PyInterpreterState *interp = PyThreadState_GET()->interp;
848 _Py_IDENTIFIER(_fix_up_module);
Brett Cannon18fc4e72014-04-04 10:01:46 -0400849
850 d = module_dict_for_exec(name);
851 if (d == NULL) {
852 return NULL;
853 }
854
Eric Snow08197a42014-05-12 17:54:55 -0600855 if (pathname == NULL) {
856 pathname = ((PyCodeObject *)co)->co_filename;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 }
Eric Snow32439d62015-05-02 19:15:18 -0600858 external = PyObject_GetAttrString(interp->importlib, "_bootstrap_external");
859 if (external == NULL)
860 return NULL;
861 res = _PyObject_CallMethodIdObjArgs(external,
Eric Snow08197a42014-05-12 17:54:55 -0600862 &PyId__fix_up_module,
863 d, name, pathname, cpathname, NULL);
Eric Snow32439d62015-05-02 19:15:18 -0600864 Py_DECREF(external);
Eric Snow08197a42014-05-12 17:54:55 -0600865 if (res != NULL) {
Eric Snow58cfdd82014-05-29 12:31:39 -0600866 Py_DECREF(res);
Eric Snow08197a42014-05-12 17:54:55 -0600867 res = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 }
Eric Snow08197a42014-05-12 17:54:55 -0600869 return res;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000870}
871
872
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000873static void
874update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
875{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 PyObject *constants, *tmp;
877 Py_ssize_t i, n;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000878
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000879 if (PyUnicode_Compare(co->co_filename, oldname))
880 return;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000881
Serhiy Storchaka576f1322016-01-05 21:27:54 +0200882 Py_INCREF(newname);
883 Py_SETREF(co->co_filename, newname);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000884
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000885 constants = co->co_consts;
886 n = PyTuple_GET_SIZE(constants);
887 for (i = 0; i < n; i++) {
888 tmp = PyTuple_GET_ITEM(constants, i);
889 if (PyCode_Check(tmp))
890 update_code_filenames((PyCodeObject *)tmp,
891 oldname, newname);
892 }
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000893}
894
Victor Stinner2f42ae52011-03-20 00:41:24 +0100895static void
896update_compiled_module(PyCodeObject *co, PyObject *newname)
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000897{
Victor Stinner2f42ae52011-03-20 00:41:24 +0100898 PyObject *oldname;
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000899
Victor Stinner2f42ae52011-03-20 00:41:24 +0100900 if (PyUnicode_Compare(co->co_filename, newname) == 0)
901 return;
Hirokazu Yamamoto4f447fb2009-03-04 01:52:10 +0000902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000903 oldname = co->co_filename;
904 Py_INCREF(oldname);
905 update_code_filenames(co, oldname, newname);
906 Py_DECREF(oldname);
Antoine Pitroud35cbf62009-01-06 19:02:24 +0000907}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000908
Brett Cannon4caa61d2014-01-09 19:03:32 -0500909/*[clinic input]
910_imp._fix_co_filename
911
912 code: object(type="PyCodeObject *", subclass_of="&PyCode_Type")
913 Code object to change.
914
915 path: unicode
916 File path to use.
917 /
918
919Changes code.co_filename to specify the passed-in file path.
920[clinic start generated code]*/
921
Brett Cannon4caa61d2014-01-09 19:03:32 -0500922static PyObject *
Larry Hastings89964c42015-04-14 18:07:59 -0400923_imp__fix_co_filename_impl(PyModuleDef *module, PyCodeObject *code,
924 PyObject *path)
925/*[clinic end generated code: output=f4db56aac0a1327f input=895ba50e78b82f05]*/
Brett Cannon442c9b92011-03-23 16:14:42 -0700926
Brett Cannon4caa61d2014-01-09 19:03:32 -0500927{
Brett Cannona6dec2e2014-01-10 07:43:55 -0500928 update_compiled_module(code, path);
Brett Cannon442c9b92011-03-23 16:14:42 -0700929
930 Py_RETURN_NONE;
931}
932
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000933
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000934/* Forward */
Benjamin Peterson6fba3db2013-03-18 23:13:31 -0700935static const struct _frozen * find_frozen(PyObject *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000936
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000937
938/* Helper to test for built-in module */
939
940static int
Victor Stinner95872862011-03-07 18:20:56 +0100941is_builtin(PyObject *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000942{
Victor Stinner95872862011-03-07 18:20:56 +0100943 int i, cmp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000944 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Victor Stinner95872862011-03-07 18:20:56 +0100945 cmp = PyUnicode_CompareWithASCIIString(name, PyImport_Inittab[i].name);
946 if (cmp == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000947 if (PyImport_Inittab[i].initfunc == NULL)
948 return -1;
949 else
950 return 1;
951 }
952 }
953 return 0;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000954}
955
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000956
Just van Rossum52e14d62002-12-30 22:08:05 +0000957/* Return an importer object for a sys.path/pkg.__path__ item 'p',
958 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +0000959 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +0000960 that can handle the path item. Return None if no hook could;
961 this tells our caller it should fall back to the builtin
962 import mechanism. Cache the result in path_importer_cache.
963 Returns a borrowed reference. */
964
965static PyObject *
966get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000967 PyObject *p)
Just van Rossum52e14d62002-12-30 22:08:05 +0000968{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000969 PyObject *importer;
970 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +0000971
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000972 /* These conditions are the caller's responsibility: */
973 assert(PyList_Check(path_hooks));
974 assert(PyDict_Check(path_importer_cache));
Just van Rossum52e14d62002-12-30 22:08:05 +0000975
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000976 nhooks = PyList_Size(path_hooks);
977 if (nhooks < 0)
978 return NULL; /* Shouldn't happen */
Just van Rossum52e14d62002-12-30 22:08:05 +0000979
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000980 importer = PyDict_GetItem(path_importer_cache, p);
981 if (importer != NULL)
982 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +0000983
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000984 /* set path_importer_cache[p] to None to avoid recursion */
985 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
986 return NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +0000987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000988 for (j = 0; j < nhooks; j++) {
989 PyObject *hook = PyList_GetItem(path_hooks, j);
990 if (hook == NULL)
991 return NULL;
992 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
993 if (importer != NULL)
994 break;
Just van Rossum52e14d62002-12-30 22:08:05 +0000995
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000996 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
997 return NULL;
998 }
999 PyErr_Clear();
1000 }
1001 if (importer == NULL) {
Brett Cannonaa936422012-04-27 15:30:58 -04001002 return Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 }
1004 if (importer != NULL) {
1005 int err = PyDict_SetItem(path_importer_cache, p, importer);
1006 Py_DECREF(importer);
1007 if (err != 0)
1008 return NULL;
1009 }
1010 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +00001011}
1012
Christian Heimes9cd17752007-11-18 19:35:23 +00001013PyAPI_FUNC(PyObject *)
1014PyImport_GetImporter(PyObject *path) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001015 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
Christian Heimes9cd17752007-11-18 19:35:23 +00001016
Victor Stinner1e53bba2013-07-16 22:26:05 +02001017 path_importer_cache = PySys_GetObject("path_importer_cache");
1018 path_hooks = PySys_GetObject("path_hooks");
1019 if (path_importer_cache != NULL && path_hooks != NULL) {
1020 importer = get_path_importer(path_importer_cache,
1021 path_hooks, path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 }
1023 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1024 return importer;
Christian Heimes9cd17752007-11-18 19:35:23 +00001025}
1026
Nick Coghland5cacbb2015-05-23 22:24:10 +10001027/*[clinic input]
1028_imp.create_builtin
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001029
Nick Coghland5cacbb2015-05-23 22:24:10 +10001030 spec: object
1031 /
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001032
Nick Coghland5cacbb2015-05-23 22:24:10 +10001033Create an extension module.
1034[clinic start generated code]*/
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001035
Nick Coghland5cacbb2015-05-23 22:24:10 +10001036static PyObject *
1037_imp_create_builtin(PyModuleDef *module, PyObject *spec)
1038/*[clinic end generated code: output=5038f467617226bd input=37f966f890384e47]*/
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001039{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001040 struct _inittab *p;
Nick Coghland5cacbb2015-05-23 22:24:10 +10001041 PyObject *name;
1042 char *namestr;
Victor Stinner5eb4f592013-11-14 22:38:52 +01001043 PyObject *mod;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001044
Nick Coghland5cacbb2015-05-23 22:24:10 +10001045 name = PyObject_GetAttrString(spec, "name");
1046 if (name == NULL) {
1047 return NULL;
1048 }
1049
Victor Stinner5eb4f592013-11-14 22:38:52 +01001050 mod = _PyImport_FindExtensionObject(name, name);
Nick Coghland5cacbb2015-05-23 22:24:10 +10001051 if (mod || PyErr_Occurred()) {
1052 Py_DECREF(name);
1053 Py_INCREF(mod);
1054 return mod;
1055 }
1056
1057 namestr = PyUnicode_AsUTF8(name);
1058 if (namestr == NULL) {
1059 Py_DECREF(name);
1060 return NULL;
1061 }
Guido van Rossum25ce5661997-08-02 03:10:38 +00001062
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 for (p = PyImport_Inittab; p->name != NULL; p++) {
Antoine Pitrou6c40eb72012-01-18 20:16:09 +01001064 PyModuleDef *def;
Victor Stinner95872862011-03-07 18:20:56 +01001065 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 if (p->initfunc == NULL) {
Nick Coghland5cacbb2015-05-23 22:24:10 +10001067 /* Cannot re-init internal module ("sys" or "builtins") */
1068 mod = PyImport_AddModule(namestr);
1069 Py_DECREF(name);
1070 return mod;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001071 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001072 mod = (*p->initfunc)();
Nick Coghland5cacbb2015-05-23 22:24:10 +10001073 if (mod == NULL) {
1074 Py_DECREF(name);
1075 return NULL;
1076 }
1077 if (PyObject_TypeCheck(mod, &PyModuleDef_Type)) {
1078 Py_DECREF(name);
1079 return PyModule_FromDefAndSpec((PyModuleDef*)mod, spec);
1080 } else {
1081 /* Remember pointer to module init function. */
1082 def = PyModule_GetDef(mod);
1083 def->m_base.m_init = p->initfunc;
1084 if (_PyImport_FixupExtensionObject(mod, name, name) < 0) {
1085 Py_DECREF(name);
1086 return NULL;
1087 }
1088 Py_DECREF(name);
1089 return mod;
1090 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001091 }
1092 }
Nick Coghland5cacbb2015-05-23 22:24:10 +10001093 Py_DECREF(name);
1094 Py_RETURN_NONE;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001095}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001096
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001097
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001098/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001099
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001100static const struct _frozen *
Victor Stinner53dc7352011-03-20 01:50:21 +01001101find_frozen(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001102{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001103 const struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001104
Victor Stinner53dc7352011-03-20 01:50:21 +01001105 if (name == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001106 return NULL;
Benjamin Petersond968e272008-11-05 22:48:33 +00001107
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001108 for (p = PyImport_FrozenModules; ; p++) {
1109 if (p->name == NULL)
1110 return NULL;
Victor Stinner53dc7352011-03-20 01:50:21 +01001111 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001112 break;
1113 }
1114 return p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001115}
1116
Guido van Rossum79f25d91997-04-29 20:08:16 +00001117static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001118get_frozen_object(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001119{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001120 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001121 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 if (p == NULL) {
1124 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001125 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001126 name);
1127 return NULL;
1128 }
1129 if (p->code == NULL) {
1130 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001131 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001132 name);
1133 return NULL;
1134 }
1135 size = p->size;
1136 if (size < 0)
1137 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001138 return PyMarshal_ReadObjectFromString((const char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001139}
1140
Brett Cannon8d110132009-03-15 02:20:16 +00001141static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01001142is_frozen_package(PyObject *name)
Brett Cannon8d110132009-03-15 02:20:16 +00001143{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001144 const struct _frozen *p = find_frozen(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001145 int size;
Brett Cannon8d110132009-03-15 02:20:16 +00001146
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001147 if (p == NULL) {
1148 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001149 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001150 name);
1151 return NULL;
1152 }
Brett Cannon8d110132009-03-15 02:20:16 +00001153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001154 size = p->size;
Brett Cannon8d110132009-03-15 02:20:16 +00001155
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001156 if (size < 0)
1157 Py_RETURN_TRUE;
1158 else
1159 Py_RETURN_FALSE;
Brett Cannon8d110132009-03-15 02:20:16 +00001160}
1161
1162
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001163/* Initialize a frozen module.
Brett Cannon3c2ac442009-03-08 20:49:47 +00001164 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001165 an exception set if the initialization failed.
1166 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001167
1168int
Victor Stinner53dc7352011-03-20 01:50:21 +01001169PyImport_ImportFrozenModuleObject(PyObject *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001170{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001171 const struct _frozen *p;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001172 PyObject *co, *m, *d;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001173 int ispackage;
1174 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001175
Victor Stinner53dc7352011-03-20 01:50:21 +01001176 p = find_frozen(name);
1177
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001178 if (p == NULL)
1179 return 0;
1180 if (p->code == NULL) {
1181 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001182 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001183 name);
1184 return -1;
1185 }
1186 size = p->size;
1187 ispackage = (size < 0);
1188 if (ispackage)
1189 size = -size;
Serhiy Storchakac6792272013-10-19 21:03:34 +03001190 co = PyMarshal_ReadObjectFromString((const char *)p->code, size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 if (co == NULL)
1192 return -1;
1193 if (!PyCode_Check(co)) {
1194 PyErr_Format(PyExc_TypeError,
Victor Stinner53dc7352011-03-20 01:50:21 +01001195 "frozen object %R is not a code object",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 name);
1197 goto err_return;
1198 }
1199 if (ispackage) {
Brett Cannon3e0651b2013-05-31 23:18:39 -04001200 /* Set __path__ to the empty list */
Brett Cannon18fc4e72014-04-04 10:01:46 -04001201 PyObject *l;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 int err;
Victor Stinner53dc7352011-03-20 01:50:21 +01001203 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001204 if (m == NULL)
1205 goto err_return;
1206 d = PyModule_GetDict(m);
Brett Cannon3e0651b2013-05-31 23:18:39 -04001207 l = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 if (l == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001209 goto err_return;
1210 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001211 err = PyDict_SetItemString(d, "__path__", l);
1212 Py_DECREF(l);
1213 if (err != 0)
1214 goto err_return;
1215 }
Brett Cannon18fc4e72014-04-04 10:01:46 -04001216 d = module_dict_for_exec(name);
1217 if (d == NULL) {
Victor Stinner53dc7352011-03-20 01:50:21 +01001218 goto err_return;
Brett Cannon18fc4e72014-04-04 10:01:46 -04001219 }
1220 m = exec_code_in_module(name, d, co);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001221 if (m == NULL)
1222 goto err_return;
1223 Py_DECREF(co);
1224 Py_DECREF(m);
1225 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001226err_return:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 Py_DECREF(co);
1228 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001229}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001230
Victor Stinner53dc7352011-03-20 01:50:21 +01001231int
Serhiy Storchakac6792272013-10-19 21:03:34 +03001232PyImport_ImportFrozenModule(const char *name)
Victor Stinner53dc7352011-03-20 01:50:21 +01001233{
1234 PyObject *nameobj;
1235 int ret;
1236 nameobj = PyUnicode_InternFromString(name);
1237 if (nameobj == NULL)
1238 return -1;
1239 ret = PyImport_ImportFrozenModuleObject(nameobj);
1240 Py_DECREF(nameobj);
1241 return ret;
1242}
1243
Guido van Rossum74e6a111994-08-29 12:54:38 +00001244
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001245/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001246 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001247
Guido van Rossum79f25d91997-04-29 20:08:16 +00001248PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001249PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001250{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 PyObject *pname;
1252 PyObject *result;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001253
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001254 pname = PyUnicode_FromString(name);
1255 if (pname == NULL)
1256 return NULL;
1257 result = PyImport_Import(pname);
1258 Py_DECREF(pname);
1259 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001260}
1261
Christian Heimes072c0f12008-01-03 23:01:04 +00001262/* Import a module without blocking
1263 *
1264 * At first it tries to fetch the module from sys.modules. If the module was
1265 * never loaded before it loads it with PyImport_ImportModule() unless another
1266 * thread holds the import lock. In the latter case the function raises an
1267 * ImportError instead of blocking.
1268 *
1269 * Returns the module object with incremented ref count.
1270 */
1271PyObject *
1272PyImport_ImportModuleNoBlock(const char *name)
1273{
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001274 return PyImport_ImportModule(name);
Christian Heimes072c0f12008-01-03 23:01:04 +00001275}
1276
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001277
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001278/* Remove importlib frames from the traceback,
1279 * except in Verbose mode. */
1280static void
1281remove_importlib_frames(void)
1282{
1283 const char *importlib_filename = "<frozen importlib._bootstrap>";
Eric Snow32439d62015-05-02 19:15:18 -06001284 const char *external_filename = "<frozen importlib._bootstrap_external>";
Nick Coghlan42c07662012-07-31 21:14:18 +10001285 const char *remove_frames = "_call_with_frames_removed";
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001286 int always_trim = 0;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001287 int in_importlib = 0;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001288 PyObject *exception, *value, *base_tb, *tb;
Benjamin Petersonfa873702012-07-09 13:43:53 -07001289 PyObject **prev_link, **outer_link = NULL;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001290
1291 /* Synopsis: if it's an ImportError, we trim all importlib chunks
Nick Coghlan42c07662012-07-31 21:14:18 +10001292 from the traceback. We always trim chunks
1293 which end with a call to "_call_with_frames_removed". */
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001294
1295 PyErr_Fetch(&exception, &value, &base_tb);
1296 if (!exception || Py_VerboseFlag)
1297 goto done;
1298 if (PyType_IsSubtype((PyTypeObject *) exception,
1299 (PyTypeObject *) PyExc_ImportError))
1300 always_trim = 1;
1301
1302 prev_link = &base_tb;
1303 tb = base_tb;
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001304 while (tb != NULL) {
1305 PyTracebackObject *traceback = (PyTracebackObject *)tb;
1306 PyObject *next = (PyObject *) traceback->tb_next;
1307 PyFrameObject *frame = traceback->tb_frame;
1308 PyCodeObject *code = frame->f_code;
1309 int now_in_importlib;
1310
1311 assert(PyTraceBack_Check(tb));
1312 now_in_importlib = (PyUnicode_CompareWithASCIIString(
1313 code->co_filename,
Eric Snow32439d62015-05-02 19:15:18 -06001314 importlib_filename) == 0) ||
1315 (PyUnicode_CompareWithASCIIString(
1316 code->co_filename,
1317 external_filename) == 0);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001318 if (now_in_importlib && !in_importlib) {
1319 /* This is the link to this chunk of importlib tracebacks */
1320 outer_link = prev_link;
1321 }
1322 in_importlib = now_in_importlib;
1323
1324 if (in_importlib &&
1325 (always_trim ||
Nick Coghlan42c07662012-07-31 21:14:18 +10001326 PyUnicode_CompareWithASCIIString(code->co_name,
1327 remove_frames) == 0)) {
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001328 Py_XINCREF(next);
Serhiy Storchaka576f1322016-01-05 21:27:54 +02001329 Py_SETREF(*outer_link, next);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001330 prev_link = outer_link;
1331 }
1332 else {
1333 prev_link = (PyObject **) &traceback->tb_next;
1334 }
1335 tb = next;
1336 }
1337done:
1338 PyErr_Restore(exception, value, base_tb);
1339}
1340
1341
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001342PyObject *
Brett Cannonfd074152012-04-14 14:10:13 -04001343PyImport_ImportModuleLevelObject(PyObject *name, PyObject *given_globals,
1344 PyObject *locals, PyObject *given_fromlist,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001345 int level)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001346{
Brett Cannonfd074152012-04-14 14:10:13 -04001347 _Py_IDENTIFIER(__import__);
Eric Snowb523f842013-11-22 09:05:39 -07001348 _Py_IDENTIFIER(__spec__);
1349 _Py_IDENTIFIER(_initializing);
Brett Cannonfd074152012-04-14 14:10:13 -04001350 _Py_IDENTIFIER(__package__);
1351 _Py_IDENTIFIER(__path__);
1352 _Py_IDENTIFIER(__name__);
1353 _Py_IDENTIFIER(_find_and_load);
1354 _Py_IDENTIFIER(_handle_fromlist);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001355 _Py_IDENTIFIER(_lock_unlock_module);
Brett Cannonfd074152012-04-14 14:10:13 -04001356 _Py_static_string(single_dot, ".");
1357 PyObject *abs_name = NULL;
1358 PyObject *builtins_import = NULL;
1359 PyObject *final_mod = NULL;
1360 PyObject *mod = NULL;
1361 PyObject *package = NULL;
Brett Cannon63b85052016-01-15 13:33:03 -08001362 PyObject *spec = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001363 PyObject *globals = NULL;
1364 PyObject *fromlist = NULL;
1365 PyInterpreterState *interp = PyThreadState_GET()->interp;
Serhiy Storchakafa494fd2015-05-30 17:45:22 +03001366 int has_from;
Brett Cannonfd074152012-04-14 14:10:13 -04001367
1368 /* Make sure to use default values so as to not have
1369 PyObject_CallMethodObjArgs() truncate the parameter list because of a
1370 NULL argument. */
1371 if (given_globals == NULL) {
1372 globals = PyDict_New();
1373 if (globals == NULL) {
1374 goto error;
1375 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001376 }
Brett Cannonfd074152012-04-14 14:10:13 -04001377 else {
1378 /* Only have to care what given_globals is if it will be used
Brett Cannonecfefb72012-08-05 19:24:57 -04001379 for something. */
Brett Cannonfd074152012-04-14 14:10:13 -04001380 if (level > 0 && !PyDict_Check(given_globals)) {
1381 PyErr_SetString(PyExc_TypeError, "globals must be a dict");
1382 goto error;
1383 }
1384 globals = given_globals;
1385 Py_INCREF(globals);
1386 }
1387
1388 if (given_fromlist == NULL) {
1389 fromlist = PyList_New(0);
1390 if (fromlist == NULL) {
1391 goto error;
1392 }
1393 }
1394 else {
1395 fromlist = given_fromlist;
1396 Py_INCREF(fromlist);
1397 }
1398 if (name == NULL) {
1399 PyErr_SetString(PyExc_ValueError, "Empty module name");
1400 goto error;
1401 }
1402
1403 /* The below code is importlib.__import__() & _gcd_import(), ported to C
1404 for added performance. */
1405
1406 if (!PyUnicode_Check(name)) {
1407 PyErr_SetString(PyExc_TypeError, "module name must be a string");
1408 goto error;
1409 }
1410 else if (PyUnicode_READY(name) < 0) {
1411 goto error;
1412 }
1413 if (level < 0) {
1414 PyErr_SetString(PyExc_ValueError, "level must be >= 0");
1415 goto error;
1416 }
1417 else if (level > 0) {
Brett Cannon63b85052016-01-15 13:33:03 -08001418 spec = _PyDict_GetItemId(globals, &PyId___spec__);
1419 if (spec != NULL) {
1420 package = PyObject_GetAttrString(spec, "parent");
1421 }
Brett Cannonfd074152012-04-14 14:10:13 -04001422 if (package != NULL && package != Py_None) {
Brett Cannonfd074152012-04-14 14:10:13 -04001423 if (!PyUnicode_Check(package)) {
Brett Cannon63b85052016-01-15 13:33:03 -08001424 PyErr_SetString(PyExc_TypeError,
1425 "__spec__.parent must be a string");
Brett Cannonfd074152012-04-14 14:10:13 -04001426 goto error;
1427 }
1428 }
1429 else {
Brett Cannon63b85052016-01-15 13:33:03 -08001430 package = _PyDict_GetItemId(globals, &PyId___package__);
1431 if (package != NULL && package != Py_None) {
Brett Cannonfd074152012-04-14 14:10:13 -04001432 Py_INCREF(package);
Brett Cannon63b85052016-01-15 13:33:03 -08001433 if (!PyUnicode_Check(package)) {
1434 PyErr_SetString(PyExc_TypeError, "package must be a string");
1435 goto error;
1436 }
1437 }
1438 else {
1439 if (PyErr_WarnEx(PyExc_ImportWarning,
1440 "can't resolve package from __spec__ or __package__, "
1441 "falling back on __name__ and __path__", 1) < 0) {
1442 goto error;
1443 }
1444
1445 package = _PyDict_GetItemId(globals, &PyId___name__);
1446 if (package == NULL) {
1447 PyErr_SetString(PyExc_KeyError, "'__name__' not in globals");
1448 goto error;
1449 }
1450
1451 Py_INCREF(package);
1452 if (!PyUnicode_Check(package)) {
1453 PyErr_SetString(PyExc_TypeError, "__name__ must be a string");
1454 goto error;
1455 }
1456
1457 if (_PyDict_GetItemId(globals, &PyId___path__) == NULL) {
1458 PyObject *partition = NULL;
1459 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1460 if (borrowed_dot == NULL) {
1461 goto error;
1462 }
1463 partition = PyUnicode_RPartition(package, borrowed_dot);
1464 Py_DECREF(package);
1465 if (partition == NULL) {
1466 goto error;
1467 }
1468 package = PyTuple_GET_ITEM(partition, 0);
1469 Py_INCREF(package);
1470 Py_DECREF(partition);
1471 }
Brett Cannonfd074152012-04-14 14:10:13 -04001472 }
1473 }
1474
1475 if (PyDict_GetItem(interp->modules, package) == NULL) {
1476 PyErr_Format(PyExc_SystemError,
1477 "Parent module %R not loaded, cannot perform relative "
1478 "import", package);
1479 goto error;
1480 }
1481 }
1482 else { /* level == 0 */
1483 if (PyUnicode_GET_LENGTH(name) == 0) {
1484 PyErr_SetString(PyExc_ValueError, "Empty module name");
1485 goto error;
1486 }
1487 package = Py_None;
1488 Py_INCREF(package);
1489 }
1490
1491 if (level > 0) {
1492 Py_ssize_t last_dot = PyUnicode_GET_LENGTH(package);
1493 PyObject *base = NULL;
1494 int level_up = 1;
1495
1496 for (level_up = 1; level_up < level; level_up += 1) {
1497 last_dot = PyUnicode_FindChar(package, '.', 0, last_dot, -1);
1498 if (last_dot == -2) {
1499 goto error;
1500 }
1501 else if (last_dot == -1) {
1502 PyErr_SetString(PyExc_ValueError,
1503 "attempted relative import beyond top-level "
1504 "package");
1505 goto error;
1506 }
1507 }
Victor Stinner22af2592013-11-13 12:11:36 +01001508
Brett Cannonfd074152012-04-14 14:10:13 -04001509 base = PyUnicode_Substring(package, 0, last_dot);
Victor Stinner22af2592013-11-13 12:11:36 +01001510 if (base == NULL)
1511 goto error;
1512
Brett Cannonfd074152012-04-14 14:10:13 -04001513 if (PyUnicode_GET_LENGTH(name) > 0) {
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001514 PyObject *borrowed_dot, *seq = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001515
1516 borrowed_dot = _PyUnicode_FromId(&single_dot);
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001517 seq = PyTuple_Pack(2, base, name);
1518 Py_DECREF(base);
Brett Cannonfd074152012-04-14 14:10:13 -04001519 if (borrowed_dot == NULL || seq == NULL) {
1520 goto error;
1521 }
1522
1523 abs_name = PyUnicode_Join(borrowed_dot, seq);
1524 Py_DECREF(seq);
1525 if (abs_name == NULL) {
1526 goto error;
1527 }
1528 }
1529 else {
1530 abs_name = base;
1531 }
1532 }
1533 else {
1534 abs_name = name;
1535 Py_INCREF(abs_name);
1536 }
1537
Brian Curtine6b299f2012-04-14 14:19:33 -05001538#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001539 _PyImport_AcquireLock();
1540#endif
1541 /* From this point forward, goto error_with_unlock! */
1542 if (PyDict_Check(globals)) {
1543 builtins_import = _PyDict_GetItemId(globals, &PyId___import__);
1544 }
1545 if (builtins_import == NULL) {
1546 builtins_import = _PyDict_GetItemId(interp->builtins, &PyId___import__);
1547 if (builtins_import == NULL) {
Benjamin Peterson7d110042013-04-29 09:08:14 -04001548 PyErr_SetString(PyExc_ImportError, "__import__ not found");
1549 goto error_with_unlock;
Brett Cannonfd074152012-04-14 14:10:13 -04001550 }
1551 }
1552 Py_INCREF(builtins_import);
1553
1554 mod = PyDict_GetItem(interp->modules, abs_name);
1555 if (mod == Py_None) {
Brett Cannon27fc5282012-04-15 14:15:31 -04001556 PyObject *msg = PyUnicode_FromFormat("import of %R halted; "
1557 "None in sys.modules", abs_name);
1558 if (msg != NULL) {
Brett Cannon82da8882013-07-04 17:48:16 -04001559 PyErr_SetImportError(msg, abs_name, NULL);
Brian Curtin09b86d12012-04-17 16:57:09 -05001560 Py_DECREF(msg);
Brett Cannon27fc5282012-04-15 14:15:31 -04001561 }
Antoine Pitrou71382cb2012-04-16 18:48:49 +02001562 mod = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001563 goto error_with_unlock;
1564 }
1565 else if (mod != NULL) {
Eric Snowb523f842013-11-22 09:05:39 -07001566 PyObject *value = NULL;
1567 PyObject *spec;
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001568 int initializing = 0;
1569
Brett Cannonfd074152012-04-14 14:10:13 -04001570 Py_INCREF(mod);
Antoine Pitrou03989852012-08-28 00:24:52 +02001571 /* Optimization: only call _bootstrap._lock_unlock_module() if
Eric Snowb523f842013-11-22 09:05:39 -07001572 __spec__._initializing is true.
1573 NOTE: because of this, initializing must be set *before*
Antoine Pitrou03989852012-08-28 00:24:52 +02001574 stuffing the new module in sys.modules.
1575 */
Eric Snowb523f842013-11-22 09:05:39 -07001576 spec = _PyObject_GetAttrId(mod, &PyId___spec__);
1577 if (spec != NULL) {
1578 value = _PyObject_GetAttrId(spec, &PyId__initializing);
1579 Py_DECREF(spec);
1580 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001581 if (value == NULL)
1582 PyErr_Clear();
1583 else {
1584 initializing = PyObject_IsTrue(value);
1585 Py_DECREF(value);
1586 if (initializing == -1)
1587 PyErr_Clear();
1588 }
1589 if (initializing > 0) {
1590 /* _bootstrap._lock_unlock_module() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001591 value = _PyObject_CallMethodIdObjArgs(interp->importlib,
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001592 &PyId__lock_unlock_module, abs_name,
1593 NULL);
1594 if (value == NULL)
1595 goto error;
1596 Py_DECREF(value);
1597 }
1598 else {
1599#ifdef WITH_THREAD
1600 if (_PyImport_ReleaseLock() < 0) {
1601 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1602 goto error;
1603 }
1604#endif
1605 }
Brett Cannonfd074152012-04-14 14:10:13 -04001606 }
1607 else {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001608 /* _bootstrap._find_and_load() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001609 mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001610 &PyId__find_and_load, abs_name,
1611 builtins_import, NULL);
1612 if (mod == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001613 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001614 }
1615 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001616 /* From now on we don't hold the import lock anymore. */
Brett Cannonfd074152012-04-14 14:10:13 -04001617
Serhiy Storchakafa494fd2015-05-30 17:45:22 +03001618 has_from = PyObject_IsTrue(fromlist);
1619 if (has_from < 0)
1620 goto error;
1621 if (!has_from) {
Brett Cannonfd074152012-04-14 14:10:13 -04001622 if (level == 0 || PyUnicode_GET_LENGTH(name) > 0) {
1623 PyObject *front = NULL;
Brian Curtine6b299f2012-04-14 14:19:33 -05001624 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001625 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1626
1627 if (borrowed_dot == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001628 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001629 }
1630
Brian Curtine6b299f2012-04-14 14:19:33 -05001631 partition = PyUnicode_Partition(name, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001632 if (partition == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001633 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001634 }
1635
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001636 if (PyUnicode_GET_LENGTH(PyTuple_GET_ITEM(partition, 1)) == 0) {
1637 /* No dot in module name, simple exit */
1638 Py_DECREF(partition);
1639 final_mod = mod;
1640 Py_INCREF(mod);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001641 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001642 }
1643
Brett Cannonfd074152012-04-14 14:10:13 -04001644 front = PyTuple_GET_ITEM(partition, 0);
1645 Py_INCREF(front);
1646 Py_DECREF(partition);
1647
1648 if (level == 0) {
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001649 final_mod = PyObject_CallFunctionObjArgs(builtins_import, front, NULL);
Antoine Pitroub78174c2012-05-06 17:15:23 +02001650 Py_DECREF(front);
Brett Cannonfd074152012-04-14 14:10:13 -04001651 }
1652 else {
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001653 Py_ssize_t cut_off = PyUnicode_GET_LENGTH(name) -
1654 PyUnicode_GET_LENGTH(front);
1655 Py_ssize_t abs_name_len = PyUnicode_GET_LENGTH(abs_name);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001656 PyObject *to_return = PyUnicode_Substring(abs_name, 0,
Brett Cannonfd074152012-04-14 14:10:13 -04001657 abs_name_len - cut_off);
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001658 Py_DECREF(front);
1659 if (to_return == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001660 goto error;
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001661 }
Brett Cannonfd074152012-04-14 14:10:13 -04001662
1663 final_mod = PyDict_GetItem(interp->modules, to_return);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001664 if (final_mod == NULL) {
1665 PyErr_Format(PyExc_KeyError,
1666 "%R not in sys.modules as expected",
1667 to_return);
1668 }
1669 else {
1670 Py_INCREF(final_mod);
1671 }
Antoine Pitroub78174c2012-05-06 17:15:23 +02001672 Py_DECREF(to_return);
Brett Cannonfd074152012-04-14 14:10:13 -04001673 }
1674 }
1675 else {
1676 final_mod = mod;
1677 Py_INCREF(mod);
1678 }
1679 }
1680 else {
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001681 final_mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001682 &PyId__handle_fromlist, mod,
1683 fromlist, builtins_import,
1684 NULL);
1685 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001686 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001687
Brett Cannonfd074152012-04-14 14:10:13 -04001688 error_with_unlock:
Brian Curtine6b299f2012-04-14 14:19:33 -05001689#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001690 if (_PyImport_ReleaseLock() < 0) {
1691 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1692 }
1693#endif
1694 error:
1695 Py_XDECREF(abs_name);
1696 Py_XDECREF(builtins_import);
1697 Py_XDECREF(mod);
1698 Py_XDECREF(package);
1699 Py_XDECREF(globals);
1700 Py_XDECREF(fromlist);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001701 if (final_mod == NULL)
1702 remove_importlib_frames();
Brett Cannonfd074152012-04-14 14:10:13 -04001703 return final_mod;
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001704}
1705
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001706PyObject *
Benjamin Peterson04778a82011-05-25 09:29:00 -05001707PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001708 PyObject *fromlist, int level)
1709{
1710 PyObject *nameobj, *mod;
1711 nameobj = PyUnicode_FromString(name);
1712 if (nameobj == NULL)
1713 return NULL;
1714 mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
1715 fromlist, level);
1716 Py_DECREF(nameobj);
1717 return mod;
1718}
1719
1720
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001721/* Re-import a module of any kind and return its module object, WITH
1722 INCREMENTED REFERENCE COUNT */
1723
Guido van Rossum79f25d91997-04-29 20:08:16 +00001724PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001725PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001726{
Brett Cannon62228db2012-04-29 14:38:11 -04001727 _Py_IDENTIFIER(reload);
1728 PyObject *reloaded_module = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001729 PyObject *modules = PyImport_GetModuleDict();
Brett Cannon62228db2012-04-29 14:38:11 -04001730 PyObject *imp = PyDict_GetItemString(modules, "imp");
1731 if (imp == NULL) {
1732 imp = PyImport_ImportModule("imp");
1733 if (imp == NULL) {
1734 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001735 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001736 }
Brett Cannon62228db2012-04-29 14:38:11 -04001737 else {
1738 Py_INCREF(imp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001739 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001740
Brett Cannon62228db2012-04-29 14:38:11 -04001741 reloaded_module = _PyObject_CallMethodId(imp, &PyId_reload, "O", m);
1742 Py_DECREF(imp);
1743 return reloaded_module;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001744}
1745
1746
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001747/* Higher-level import emulator which emulates the "import" statement
1748 more accurately -- it invokes the __import__() function from the
1749 builtins of the current globals. This means that the import is
1750 done using whatever import hooks are installed in the current
Brett Cannonbc2eff32010-09-19 21:39:02 +00001751 environment.
Guido van Rossum6058eb41998-12-21 19:51:00 +00001752 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00001753 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00001754 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001755
1756PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001757PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001759 static PyObject *silly_list = NULL;
1760 static PyObject *builtins_str = NULL;
1761 static PyObject *import_str = NULL;
1762 PyObject *globals = NULL;
1763 PyObject *import = NULL;
1764 PyObject *builtins = NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001765 PyObject *modules = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001766 PyObject *r = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001767
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001768 /* Initialize constant string objects */
1769 if (silly_list == NULL) {
1770 import_str = PyUnicode_InternFromString("__import__");
1771 if (import_str == NULL)
1772 return NULL;
1773 builtins_str = PyUnicode_InternFromString("__builtins__");
1774 if (builtins_str == NULL)
1775 return NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001776 silly_list = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001777 if (silly_list == NULL)
1778 return NULL;
1779 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001780
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001781 /* Get the builtins from current globals */
1782 globals = PyEval_GetGlobals();
1783 if (globals != NULL) {
1784 Py_INCREF(globals);
1785 builtins = PyObject_GetItem(globals, builtins_str);
1786 if (builtins == NULL)
1787 goto err;
1788 }
1789 else {
1790 /* No globals -- use standard builtins, and fake globals */
1791 builtins = PyImport_ImportModuleLevel("builtins",
1792 NULL, NULL, NULL, 0);
1793 if (builtins == NULL)
1794 return NULL;
1795 globals = Py_BuildValue("{OO}", builtins_str, builtins);
1796 if (globals == NULL)
1797 goto err;
1798 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001799
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001800 /* Get the __import__ function from the builtins */
1801 if (PyDict_Check(builtins)) {
1802 import = PyObject_GetItem(builtins, import_str);
1803 if (import == NULL)
1804 PyErr_SetObject(PyExc_KeyError, import_str);
1805 }
1806 else
1807 import = PyObject_GetAttr(builtins, import_str);
1808 if (import == NULL)
1809 goto err;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001810
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001811 /* Call the __import__ function with the proper argument list
Brett Cannonbc2eff32010-09-19 21:39:02 +00001812 Always use absolute import here.
1813 Calling for side-effect of import. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001814 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
1815 globals, silly_list, 0, NULL);
Brett Cannonbc2eff32010-09-19 21:39:02 +00001816 if (r == NULL)
1817 goto err;
1818 Py_DECREF(r);
1819
1820 modules = PyImport_GetModuleDict();
1821 r = PyDict_GetItem(modules, module_name);
1822 if (r != NULL)
1823 Py_INCREF(r);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001824
1825 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001826 Py_XDECREF(globals);
1827 Py_XDECREF(builtins);
1828 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00001829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001830 return r;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001831}
1832
Brett Cannon4caa61d2014-01-09 19:03:32 -05001833/*[clinic input]
1834_imp.extension_suffixes
1835
1836Returns the list of file suffixes used to identify extension modules.
1837[clinic start generated code]*/
1838
Brett Cannon4caa61d2014-01-09 19:03:32 -05001839static PyObject *
1840_imp_extension_suffixes_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001841/*[clinic end generated code: output=d44c1566ef362229 input=ecdeeecfcb6f839e]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001842{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001843 PyObject *list;
Brett Cannon2657df42012-05-04 15:20:40 -04001844 const char *suffix;
1845 unsigned int index = 0;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001846
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001847 list = PyList_New(0);
1848 if (list == NULL)
1849 return NULL;
Brett Cannon2657df42012-05-04 15:20:40 -04001850#ifdef HAVE_DYNAMIC_LOADING
1851 while ((suffix = _PyImport_DynLoadFiletab[index])) {
1852 PyObject *item = PyUnicode_FromString(suffix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001853 if (item == NULL) {
1854 Py_DECREF(list);
1855 return NULL;
1856 }
1857 if (PyList_Append(list, item) < 0) {
1858 Py_DECREF(list);
1859 Py_DECREF(item);
1860 return NULL;
1861 }
1862 Py_DECREF(item);
Brett Cannon2657df42012-05-04 15:20:40 -04001863 index += 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001864 }
Brett Cannon2657df42012-05-04 15:20:40 -04001865#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001866 return list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001867}
1868
Brett Cannon4caa61d2014-01-09 19:03:32 -05001869/*[clinic input]
Brett Cannon4caa61d2014-01-09 19:03:32 -05001870_imp.init_frozen
1871
1872 name: unicode
1873 /
1874
1875Initializes a frozen module.
1876[clinic start generated code]*/
1877
Brett Cannon4caa61d2014-01-09 19:03:32 -05001878static PyObject *
1879_imp_init_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001880/*[clinic end generated code: output=a9de493bdd711878 input=13019adfc04f3fb3]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001881{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001882 int ret;
1883 PyObject *m;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001884
Victor Stinner53dc7352011-03-20 01:50:21 +01001885 ret = PyImport_ImportFrozenModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001886 if (ret < 0)
1887 return NULL;
1888 if (ret == 0) {
1889 Py_INCREF(Py_None);
1890 return Py_None;
1891 }
Victor Stinner53dc7352011-03-20 01:50:21 +01001892 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001893 Py_XINCREF(m);
1894 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001895}
1896
Brett Cannon4caa61d2014-01-09 19:03:32 -05001897/*[clinic input]
1898_imp.get_frozen_object
1899
1900 name: unicode
1901 /
1902
1903Create a code object for a frozen module.
1904[clinic start generated code]*/
1905
Brett Cannon4caa61d2014-01-09 19:03:32 -05001906static PyObject *
1907_imp_get_frozen_object_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001908/*[clinic end generated code: output=3114c970a47f2e3c input=ed689bc05358fdbd]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001909{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001910 return get_frozen_object(name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001911}
1912
Brett Cannon4caa61d2014-01-09 19:03:32 -05001913/*[clinic input]
1914_imp.is_frozen_package
1915
1916 name: unicode
1917 /
1918
1919Returns True if the module name is of a frozen package.
1920[clinic start generated code]*/
1921
Brett Cannon4caa61d2014-01-09 19:03:32 -05001922static PyObject *
1923_imp_is_frozen_package_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001924/*[clinic end generated code: output=3e4cab802b56d649 input=81b6cdecd080fbb8]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001925{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001926 return is_frozen_package(name);
Brett Cannon8d110132009-03-15 02:20:16 +00001927}
1928
Brett Cannon4caa61d2014-01-09 19:03:32 -05001929/*[clinic input]
1930_imp.is_builtin
1931
1932 name: unicode
1933 /
1934
1935Returns True if the module name corresponds to a built-in module.
1936[clinic start generated code]*/
1937
Guido van Rossum79f25d91997-04-29 20:08:16 +00001938static PyObject *
Brett Cannon4caa61d2014-01-09 19:03:32 -05001939_imp_is_builtin_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001940/*[clinic end generated code: output=2deec9cac6fb9a7e input=86befdac021dd1c7]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001941{
Brett Cannon4caa61d2014-01-09 19:03:32 -05001942 return PyLong_FromLong(is_builtin(name));
1943}
1944
1945/*[clinic input]
1946_imp.is_frozen
1947
1948 name: unicode
1949 /
1950
1951Returns True if the module name corresponds to a frozen module.
1952[clinic start generated code]*/
1953
Brett Cannon4caa61d2014-01-09 19:03:32 -05001954static PyObject *
1955_imp_is_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001956/*[clinic end generated code: output=7de8e260c8e36aed input=7301dbca1897d66b]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001957{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001958 const struct _frozen *p;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001959
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001960 p = find_frozen(name);
1961 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001962}
1963
Larry Hastings1df0b352015-08-24 19:53:56 -07001964/* Common implementation for _imp.exec_dynamic and _imp.exec_builtin */
1965static int
1966exec_builtin_or_dynamic(PyObject *mod) {
1967 PyModuleDef *def;
1968 void *state;
1969
1970 if (!PyModule_Check(mod)) {
1971 return 0;
1972 }
1973
1974 def = PyModule_GetDef(mod);
1975 if (def == NULL) {
1976 if (PyErr_Occurred()) {
1977 return -1;
1978 }
1979 return 0;
1980 }
1981 state = PyModule_GetState(mod);
1982 if (PyErr_Occurred()) {
1983 return -1;
1984 }
1985 if (state) {
1986 /* Already initialized; skip reload */
1987 return 0;
1988 }
1989 return PyModule_ExecDef(mod, def);
1990}
1991
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001992#ifdef HAVE_DYNAMIC_LOADING
1993
Brett Cannon4caa61d2014-01-09 19:03:32 -05001994/*[clinic input]
Nick Coghland5cacbb2015-05-23 22:24:10 +10001995_imp.create_dynamic
Brett Cannon4caa61d2014-01-09 19:03:32 -05001996
Nick Coghland5cacbb2015-05-23 22:24:10 +10001997 spec: object
Brett Cannon4caa61d2014-01-09 19:03:32 -05001998 file: object = NULL
1999 /
2000
Nick Coghland5cacbb2015-05-23 22:24:10 +10002001Create an extension module.
Brett Cannon4caa61d2014-01-09 19:03:32 -05002002[clinic start generated code]*/
2003
Brett Cannon4caa61d2014-01-09 19:03:32 -05002004static PyObject *
Nick Coghland5cacbb2015-05-23 22:24:10 +10002005_imp_create_dynamic_impl(PyModuleDef *module, PyObject *spec, PyObject *file)
2006/*[clinic end generated code: output=935cde5b3872d56d input=c31b954f4cf4e09d]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05002007{
Nick Coghland5cacbb2015-05-23 22:24:10 +10002008 PyObject *mod, *name, *path;
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002009 FILE *fp;
2010
Nick Coghland5cacbb2015-05-23 22:24:10 +10002011 name = PyObject_GetAttrString(spec, "name");
2012 if (name == NULL) {
2013 return NULL;
2014 }
2015
2016 path = PyObject_GetAttrString(spec, "origin");
2017 if (path == NULL) {
2018 Py_DECREF(name);
2019 return NULL;
2020 }
2021
2022 mod = _PyImport_FindExtensionObject(name, path);
2023 if (mod != NULL) {
2024 Py_DECREF(name);
2025 Py_DECREF(path);
2026 Py_INCREF(mod);
2027 return mod;
2028 }
2029
Brett Cannon4caa61d2014-01-09 19:03:32 -05002030 if (file != NULL) {
2031 fp = _Py_fopen_obj(path, "r");
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002032 if (fp == NULL) {
Nick Coghland5cacbb2015-05-23 22:24:10 +10002033 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002034 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002035 return NULL;
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002036 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002037 }
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002038 else
2039 fp = NULL;
Nick Coghland5cacbb2015-05-23 22:24:10 +10002040
2041 mod = _PyImport_LoadDynamicModuleWithSpec(spec, fp);
2042
2043 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002044 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002045 if (fp)
2046 fclose(fp);
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002047 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002048}
2049
Nick Coghland5cacbb2015-05-23 22:24:10 +10002050/*[clinic input]
2051_imp.exec_dynamic -> int
2052
2053 mod: object
2054 /
2055
2056Initialize an extension module.
2057[clinic start generated code]*/
2058
2059static int
2060_imp_exec_dynamic_impl(PyModuleDef *module, PyObject *mod)
2061/*[clinic end generated code: output=4b84f1301b22d4bd input=9fdbfcb250280d3a]*/
2062{
Larry Hastings1df0b352015-08-24 19:53:56 -07002063 return exec_builtin_or_dynamic(mod);
Nick Coghland5cacbb2015-05-23 22:24:10 +10002064}
2065
2066
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002067#endif /* HAVE_DYNAMIC_LOADING */
2068
Larry Hastings7726ac92014-01-31 22:03:12 -08002069/*[clinic input]
Larry Hastings1df0b352015-08-24 19:53:56 -07002070_imp.exec_builtin -> int
2071
2072 mod: object
2073 /
2074
2075Initialize a built-in module.
2076[clinic start generated code]*/
2077
2078static int
2079_imp_exec_builtin_impl(PyModuleDef *module, PyObject *mod)
Larry Hastings01b1ff62015-08-24 20:23:27 -07002080/*[clinic end generated code: output=215e99876a27e284 input=7beed5a2f12a60ca]*/
Larry Hastings1df0b352015-08-24 19:53:56 -07002081{
2082 return exec_builtin_or_dynamic(mod);
2083}
2084
2085/*[clinic input]
Larry Hastings7726ac92014-01-31 22:03:12 -08002086dump buffer
2087[clinic start generated code]*/
Brett Cannonfd4d0502014-05-30 11:21:14 -04002088/*[clinic end generated code: output=da39a3ee5e6b4b0d input=524ce2e021e4eba6]*/
Larry Hastings7726ac92014-01-31 22:03:12 -08002089
Barry Warsaw28a691b2010-04-17 00:19:56 +00002090
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002091PyDoc_STRVAR(doc_imp,
Brett Cannon6f44d662012-04-15 16:08:47 -04002092"(Extremely) low-level import machinery bits as used by importlib and imp.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002093
Guido van Rossum79f25d91997-04-29 20:08:16 +00002094static PyMethodDef imp_methods[] = {
Brett Cannon4caa61d2014-01-09 19:03:32 -05002095 _IMP_EXTENSION_SUFFIXES_METHODDEF
2096 _IMP_LOCK_HELD_METHODDEF
2097 _IMP_ACQUIRE_LOCK_METHODDEF
2098 _IMP_RELEASE_LOCK_METHODDEF
2099 _IMP_GET_FROZEN_OBJECT_METHODDEF
2100 _IMP_IS_FROZEN_PACKAGE_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002101 _IMP_CREATE_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002102 _IMP_INIT_FROZEN_METHODDEF
2103 _IMP_IS_BUILTIN_METHODDEF
2104 _IMP_IS_FROZEN_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002105 _IMP_CREATE_DYNAMIC_METHODDEF
2106 _IMP_EXEC_DYNAMIC_METHODDEF
Larry Hastings1df0b352015-08-24 19:53:56 -07002107 _IMP_EXEC_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002108 _IMP__FIX_CO_FILENAME_METHODDEF
2109 {NULL, NULL} /* sentinel */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002110};
2111
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002112
Martin v. Löwis1a214512008-06-11 05:26:20 +00002113static struct PyModuleDef impmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002114 PyModuleDef_HEAD_INIT,
Brett Cannon6f44d662012-04-15 16:08:47 -04002115 "_imp",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002116 doc_imp,
2117 0,
2118 imp_methods,
2119 NULL,
2120 NULL,
2121 NULL,
2122 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00002123};
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002124
Jason Tishler6bc06ec2003-09-04 11:59:50 +00002125PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00002126PyInit_imp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002127{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002128 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002130 m = PyModule_Create(&impmodule);
2131 if (m == NULL)
2132 goto failure;
2133 d = PyModule_GetDict(m);
2134 if (d == NULL)
2135 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002136
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002137 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002138 failure:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002139 Py_XDECREF(m);
2140 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002141}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002142
2143
Guido van Rossumb18618d2000-05-03 23:44:39 +00002144/* API for embedding applications that want to add their own entries
2145 to the table of built-in modules. This should normally be called
2146 *before* Py_Initialize(). When the table resize fails, -1 is
2147 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002148
2149 After a similar function by Just van Rossum. */
2150
2151int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002152PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002153{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002154 static struct _inittab *our_copy = NULL;
2155 struct _inittab *p;
2156 int i, n;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002157
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002158 /* Count the number of entries in both tables */
2159 for (n = 0; newtab[n].name != NULL; n++)
2160 ;
2161 if (n == 0)
2162 return 0; /* Nothing to do */
2163 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2164 ;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002165
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002166 /* Allocate new memory for the combined table */
2167 p = our_copy;
2168 PyMem_RESIZE(p, struct _inittab, i+n+1);
2169 if (p == NULL)
2170 return -1;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002171
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002172 /* Copy the tables into the new memory */
2173 if (our_copy != PyImport_Inittab)
2174 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2175 PyImport_Inittab = our_copy = p;
2176 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002177
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002178 return 0;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002179}
2180
2181/* Shorthand to add a single entry given a name and a function */
2182
2183int
Brett Cannona826f322009-04-02 03:41:46 +00002184PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002185{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002186 struct _inittab newtab[2];
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002187
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002188 memset(newtab, '\0', sizeof newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002189
Serhiy Storchaka20b39b22014-09-28 11:27:24 +03002190 newtab[0].name = name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002191 newtab[0].initfunc = initfunc;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002193 return PyImport_ExtendInittab(newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002194}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002195
2196#ifdef __cplusplus
2197}
2198#endif