blob: 8ad5b4c1674327d7d6eb7571ca539dde59b756cd [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 Cannon849113a2016-01-22 15:25:50 -08001418 package = _PyDict_GetItemId(globals, &PyId___package__);
Brett Cannon63b85052016-01-15 13:33:03 -08001419 spec = _PyDict_GetItemId(globals, &PyId___spec__);
Brett Cannon849113a2016-01-22 15:25:50 -08001420
Brett Cannonfd074152012-04-14 14:10:13 -04001421 if (package != NULL && package != Py_None) {
Brett Cannon849113a2016-01-22 15:25:50 -08001422 Py_INCREF(package);
Brett Cannonfd074152012-04-14 14:10:13 -04001423 if (!PyUnicode_Check(package)) {
Brett Cannon849113a2016-01-22 15:25:50 -08001424 PyErr_SetString(PyExc_TypeError, "package must be a string");
1425 goto error;
1426 }
Brett Cannon9fa81262016-01-22 16:39:02 -08001427 else if (spec != NULL && spec != Py_None) {
Brett Cannon849113a2016-01-22 15:25:50 -08001428 int equal;
1429 PyObject *parent = PyObject_GetAttrString(spec, "parent");
1430 if (parent == NULL) {
1431 goto error;
1432 }
1433
1434 equal = PyObject_RichCompareBool(package, parent, Py_EQ);
1435 Py_DECREF(parent);
1436 if (equal < 0) {
1437 goto error;
1438 }
1439 else if (equal == 0) {
1440 if (PyErr_WarnEx(PyExc_ImportWarning,
1441 "__package__ != __spec__.parent", 1) < 0) {
1442 goto error;
1443 }
1444 }
1445 }
1446 }
Brett Cannon9fa81262016-01-22 16:39:02 -08001447 else if (spec != NULL && spec != Py_None) {
Brett Cannon849113a2016-01-22 15:25:50 -08001448 package = PyObject_GetAttrString(spec, "parent");
1449 if (package == NULL) {
1450 goto error;
1451 }
1452 else if (!PyUnicode_Check(package)) {
Brett Cannon63b85052016-01-15 13:33:03 -08001453 PyErr_SetString(PyExc_TypeError,
1454 "__spec__.parent must be a string");
Brett Cannonfd074152012-04-14 14:10:13 -04001455 goto error;
1456 }
1457 }
1458 else {
Brett Cannon849113a2016-01-22 15:25:50 -08001459 if (PyErr_WarnEx(PyExc_ImportWarning,
1460 "can't resolve package from __spec__ or __package__, "
1461 "falling back on __name__ and __path__", 1) < 0) {
1462 goto error;
Brett Cannon63b85052016-01-15 13:33:03 -08001463 }
Brett Cannon849113a2016-01-22 15:25:50 -08001464
1465 package = _PyDict_GetItemId(globals, &PyId___name__);
1466 if (package == NULL) {
1467 PyErr_SetString(PyExc_KeyError, "'__name__' not in globals");
1468 goto error;
1469 }
1470
1471 Py_INCREF(package);
1472 if (!PyUnicode_Check(package)) {
1473 PyErr_SetString(PyExc_TypeError, "__name__ must be a string");
1474 goto error;
1475 }
1476
1477 if (_PyDict_GetItemId(globals, &PyId___path__) == NULL) {
1478 PyObject *partition = NULL;
1479 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1480 if (borrowed_dot == NULL) {
Brett Cannon63b85052016-01-15 13:33:03 -08001481 goto error;
1482 }
Brett Cannon849113a2016-01-22 15:25:50 -08001483 partition = PyUnicode_RPartition(package, borrowed_dot);
1484 Py_DECREF(package);
1485 if (partition == NULL) {
Brett Cannon63b85052016-01-15 13:33:03 -08001486 goto error;
1487 }
Brett Cannon849113a2016-01-22 15:25:50 -08001488 package = PyTuple_GET_ITEM(partition, 0);
Brett Cannon63b85052016-01-15 13:33:03 -08001489 Py_INCREF(package);
Brett Cannon849113a2016-01-22 15:25:50 -08001490 Py_DECREF(partition);
Brett Cannonfd074152012-04-14 14:10:13 -04001491 }
1492 }
1493
Brett Cannon9fa81262016-01-22 16:39:02 -08001494 if (PyUnicode_CompareWithASCIIString(package, "") == 0) {
1495 PyErr_SetString(PyExc_ImportError,
1496 "attempted relative import with no known parent package");
1497 goto error;
1498 }
1499 else if (PyDict_GetItem(interp->modules, package) == NULL) {
Brett Cannonfd074152012-04-14 14:10:13 -04001500 PyErr_Format(PyExc_SystemError,
1501 "Parent module %R not loaded, cannot perform relative "
1502 "import", package);
1503 goto error;
1504 }
1505 }
1506 else { /* level == 0 */
1507 if (PyUnicode_GET_LENGTH(name) == 0) {
1508 PyErr_SetString(PyExc_ValueError, "Empty module name");
1509 goto error;
1510 }
1511 package = Py_None;
1512 Py_INCREF(package);
1513 }
1514
1515 if (level > 0) {
1516 Py_ssize_t last_dot = PyUnicode_GET_LENGTH(package);
1517 PyObject *base = NULL;
1518 int level_up = 1;
1519
1520 for (level_up = 1; level_up < level; level_up += 1) {
1521 last_dot = PyUnicode_FindChar(package, '.', 0, last_dot, -1);
1522 if (last_dot == -2) {
1523 goto error;
1524 }
1525 else if (last_dot == -1) {
1526 PyErr_SetString(PyExc_ValueError,
1527 "attempted relative import beyond top-level "
1528 "package");
1529 goto error;
1530 }
1531 }
Victor Stinner22af2592013-11-13 12:11:36 +01001532
Brett Cannonfd074152012-04-14 14:10:13 -04001533 base = PyUnicode_Substring(package, 0, last_dot);
Victor Stinner22af2592013-11-13 12:11:36 +01001534 if (base == NULL)
1535 goto error;
1536
Brett Cannonfd074152012-04-14 14:10:13 -04001537 if (PyUnicode_GET_LENGTH(name) > 0) {
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001538 PyObject *borrowed_dot, *seq = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001539
1540 borrowed_dot = _PyUnicode_FromId(&single_dot);
Antoine Pitrou538ba2a2012-04-16 21:52:45 +02001541 seq = PyTuple_Pack(2, base, name);
1542 Py_DECREF(base);
Brett Cannonfd074152012-04-14 14:10:13 -04001543 if (borrowed_dot == NULL || seq == NULL) {
1544 goto error;
1545 }
1546
1547 abs_name = PyUnicode_Join(borrowed_dot, seq);
1548 Py_DECREF(seq);
1549 if (abs_name == NULL) {
1550 goto error;
1551 }
1552 }
1553 else {
1554 abs_name = base;
1555 }
1556 }
1557 else {
1558 abs_name = name;
1559 Py_INCREF(abs_name);
1560 }
1561
Brian Curtine6b299f2012-04-14 14:19:33 -05001562#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001563 _PyImport_AcquireLock();
1564#endif
1565 /* From this point forward, goto error_with_unlock! */
1566 if (PyDict_Check(globals)) {
1567 builtins_import = _PyDict_GetItemId(globals, &PyId___import__);
1568 }
1569 if (builtins_import == NULL) {
1570 builtins_import = _PyDict_GetItemId(interp->builtins, &PyId___import__);
1571 if (builtins_import == NULL) {
Benjamin Peterson7d110042013-04-29 09:08:14 -04001572 PyErr_SetString(PyExc_ImportError, "__import__ not found");
1573 goto error_with_unlock;
Brett Cannonfd074152012-04-14 14:10:13 -04001574 }
1575 }
1576 Py_INCREF(builtins_import);
1577
1578 mod = PyDict_GetItem(interp->modules, abs_name);
1579 if (mod == Py_None) {
Brett Cannon27fc5282012-04-15 14:15:31 -04001580 PyObject *msg = PyUnicode_FromFormat("import of %R halted; "
1581 "None in sys.modules", abs_name);
1582 if (msg != NULL) {
Brett Cannon82da8882013-07-04 17:48:16 -04001583 PyErr_SetImportError(msg, abs_name, NULL);
Brian Curtin09b86d12012-04-17 16:57:09 -05001584 Py_DECREF(msg);
Brett Cannon27fc5282012-04-15 14:15:31 -04001585 }
Antoine Pitrou71382cb2012-04-16 18:48:49 +02001586 mod = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001587 goto error_with_unlock;
1588 }
1589 else if (mod != NULL) {
Eric Snowb523f842013-11-22 09:05:39 -07001590 PyObject *value = NULL;
1591 PyObject *spec;
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001592 int initializing = 0;
1593
Brett Cannonfd074152012-04-14 14:10:13 -04001594 Py_INCREF(mod);
Antoine Pitrou03989852012-08-28 00:24:52 +02001595 /* Optimization: only call _bootstrap._lock_unlock_module() if
Eric Snowb523f842013-11-22 09:05:39 -07001596 __spec__._initializing is true.
1597 NOTE: because of this, initializing must be set *before*
Antoine Pitrou03989852012-08-28 00:24:52 +02001598 stuffing the new module in sys.modules.
1599 */
Eric Snowb523f842013-11-22 09:05:39 -07001600 spec = _PyObject_GetAttrId(mod, &PyId___spec__);
1601 if (spec != NULL) {
1602 value = _PyObject_GetAttrId(spec, &PyId__initializing);
1603 Py_DECREF(spec);
1604 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001605 if (value == NULL)
1606 PyErr_Clear();
1607 else {
1608 initializing = PyObject_IsTrue(value);
1609 Py_DECREF(value);
1610 if (initializing == -1)
1611 PyErr_Clear();
1612 }
1613 if (initializing > 0) {
1614 /* _bootstrap._lock_unlock_module() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001615 value = _PyObject_CallMethodIdObjArgs(interp->importlib,
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001616 &PyId__lock_unlock_module, abs_name,
1617 NULL);
1618 if (value == NULL)
1619 goto error;
1620 Py_DECREF(value);
1621 }
1622 else {
1623#ifdef WITH_THREAD
1624 if (_PyImport_ReleaseLock() < 0) {
1625 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1626 goto error;
1627 }
1628#endif
1629 }
Brett Cannonfd074152012-04-14 14:10:13 -04001630 }
1631 else {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001632 /* _bootstrap._find_and_load() releases the import lock */
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001633 mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001634 &PyId__find_and_load, abs_name,
1635 builtins_import, NULL);
1636 if (mod == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001637 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001638 }
1639 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001640 /* From now on we don't hold the import lock anymore. */
Brett Cannonfd074152012-04-14 14:10:13 -04001641
Serhiy Storchakafa494fd2015-05-30 17:45:22 +03001642 has_from = PyObject_IsTrue(fromlist);
1643 if (has_from < 0)
1644 goto error;
1645 if (!has_from) {
Brett Cannonfd074152012-04-14 14:10:13 -04001646 if (level == 0 || PyUnicode_GET_LENGTH(name) > 0) {
1647 PyObject *front = NULL;
Brian Curtine6b299f2012-04-14 14:19:33 -05001648 PyObject *partition = NULL;
Brett Cannonfd074152012-04-14 14:10:13 -04001649 PyObject *borrowed_dot = _PyUnicode_FromId(&single_dot);
1650
1651 if (borrowed_dot == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001652 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001653 }
1654
Brian Curtine6b299f2012-04-14 14:19:33 -05001655 partition = PyUnicode_Partition(name, borrowed_dot);
Brett Cannonfd074152012-04-14 14:10:13 -04001656 if (partition == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001657 goto error;
Brett Cannonfd074152012-04-14 14:10:13 -04001658 }
1659
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001660 if (PyUnicode_GET_LENGTH(PyTuple_GET_ITEM(partition, 1)) == 0) {
1661 /* No dot in module name, simple exit */
1662 Py_DECREF(partition);
1663 final_mod = mod;
1664 Py_INCREF(mod);
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001665 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001666 }
1667
Brett Cannonfd074152012-04-14 14:10:13 -04001668 front = PyTuple_GET_ITEM(partition, 0);
1669 Py_INCREF(front);
1670 Py_DECREF(partition);
1671
1672 if (level == 0) {
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001673 final_mod = PyObject_CallFunctionObjArgs(builtins_import, front, NULL);
Antoine Pitroub78174c2012-05-06 17:15:23 +02001674 Py_DECREF(front);
Brett Cannonfd074152012-04-14 14:10:13 -04001675 }
1676 else {
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001677 Py_ssize_t cut_off = PyUnicode_GET_LENGTH(name) -
1678 PyUnicode_GET_LENGTH(front);
1679 Py_ssize_t abs_name_len = PyUnicode_GET_LENGTH(abs_name);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001680 PyObject *to_return = PyUnicode_Substring(abs_name, 0,
Brett Cannonfd074152012-04-14 14:10:13 -04001681 abs_name_len - cut_off);
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001682 Py_DECREF(front);
1683 if (to_return == NULL) {
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001684 goto error;
Antoine Pitrou22a1d172012-04-16 22:06:21 +02001685 }
Brett Cannonfd074152012-04-14 14:10:13 -04001686
1687 final_mod = PyDict_GetItem(interp->modules, to_return);
Brett Cannon49f8d8b2012-04-14 21:50:00 -04001688 if (final_mod == NULL) {
1689 PyErr_Format(PyExc_KeyError,
1690 "%R not in sys.modules as expected",
1691 to_return);
1692 }
1693 else {
1694 Py_INCREF(final_mod);
1695 }
Antoine Pitroub78174c2012-05-06 17:15:23 +02001696 Py_DECREF(to_return);
Brett Cannonfd074152012-04-14 14:10:13 -04001697 }
1698 }
1699 else {
1700 final_mod = mod;
1701 Py_INCREF(mod);
1702 }
1703 }
1704 else {
Alexandre Vassalotti865eaa12013-05-02 10:44:04 -07001705 final_mod = _PyObject_CallMethodIdObjArgs(interp->importlib,
Brett Cannonfd074152012-04-14 14:10:13 -04001706 &PyId__handle_fromlist, mod,
1707 fromlist, builtins_import,
1708 NULL);
1709 }
Antoine Pitrouea3eb882012-05-17 18:55:59 +02001710 goto error;
Antoine Pitrou6efa50a2012-05-07 21:41:59 +02001711
Brett Cannonfd074152012-04-14 14:10:13 -04001712 error_with_unlock:
Brian Curtine6b299f2012-04-14 14:19:33 -05001713#ifdef WITH_THREAD
Brett Cannonfd074152012-04-14 14:10:13 -04001714 if (_PyImport_ReleaseLock() < 0) {
1715 PyErr_SetString(PyExc_RuntimeError, "not holding the import lock");
1716 }
1717#endif
1718 error:
1719 Py_XDECREF(abs_name);
1720 Py_XDECREF(builtins_import);
1721 Py_XDECREF(mod);
1722 Py_XDECREF(package);
1723 Py_XDECREF(globals);
1724 Py_XDECREF(fromlist);
Antoine Pitroubc07a5c2012-07-08 12:01:27 +02001725 if (final_mod == NULL)
1726 remove_importlib_frames();
Brett Cannonfd074152012-04-14 14:10:13 -04001727 return final_mod;
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001728}
1729
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001730PyObject *
Benjamin Peterson04778a82011-05-25 09:29:00 -05001731PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04001732 PyObject *fromlist, int level)
1733{
1734 PyObject *nameobj, *mod;
1735 nameobj = PyUnicode_FromString(name);
1736 if (nameobj == NULL)
1737 return NULL;
1738 mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
1739 fromlist, level);
1740 Py_DECREF(nameobj);
1741 return mod;
1742}
1743
1744
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001745/* Re-import a module of any kind and return its module object, WITH
1746 INCREMENTED REFERENCE COUNT */
1747
Guido van Rossum79f25d91997-04-29 20:08:16 +00001748PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001749PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001750{
Brett Cannon62228db2012-04-29 14:38:11 -04001751 _Py_IDENTIFIER(reload);
1752 PyObject *reloaded_module = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001753 PyObject *modules = PyImport_GetModuleDict();
Brett Cannon62228db2012-04-29 14:38:11 -04001754 PyObject *imp = PyDict_GetItemString(modules, "imp");
1755 if (imp == NULL) {
1756 imp = PyImport_ImportModule("imp");
1757 if (imp == NULL) {
1758 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001759 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001760 }
Brett Cannon62228db2012-04-29 14:38:11 -04001761 else {
1762 Py_INCREF(imp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001763 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001764
Brett Cannon62228db2012-04-29 14:38:11 -04001765 reloaded_module = _PyObject_CallMethodId(imp, &PyId_reload, "O", m);
1766 Py_DECREF(imp);
1767 return reloaded_module;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001768}
1769
1770
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001771/* Higher-level import emulator which emulates the "import" statement
1772 more accurately -- it invokes the __import__() function from the
1773 builtins of the current globals. This means that the import is
1774 done using whatever import hooks are installed in the current
Brett Cannonbc2eff32010-09-19 21:39:02 +00001775 environment.
Guido van Rossum6058eb41998-12-21 19:51:00 +00001776 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00001777 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00001778 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001779
1780PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001781PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001782{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001783 static PyObject *silly_list = NULL;
1784 static PyObject *builtins_str = NULL;
1785 static PyObject *import_str = NULL;
1786 PyObject *globals = NULL;
1787 PyObject *import = NULL;
1788 PyObject *builtins = NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001789 PyObject *modules = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001790 PyObject *r = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001792 /* Initialize constant string objects */
1793 if (silly_list == NULL) {
1794 import_str = PyUnicode_InternFromString("__import__");
1795 if (import_str == NULL)
1796 return NULL;
1797 builtins_str = PyUnicode_InternFromString("__builtins__");
1798 if (builtins_str == NULL)
1799 return NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00001800 silly_list = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001801 if (silly_list == NULL)
1802 return NULL;
1803 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001804
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001805 /* Get the builtins from current globals */
1806 globals = PyEval_GetGlobals();
1807 if (globals != NULL) {
1808 Py_INCREF(globals);
1809 builtins = PyObject_GetItem(globals, builtins_str);
1810 if (builtins == NULL)
1811 goto err;
1812 }
1813 else {
1814 /* No globals -- use standard builtins, and fake globals */
1815 builtins = PyImport_ImportModuleLevel("builtins",
1816 NULL, NULL, NULL, 0);
1817 if (builtins == NULL)
1818 return NULL;
1819 globals = Py_BuildValue("{OO}", builtins_str, builtins);
1820 if (globals == NULL)
1821 goto err;
1822 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001823
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001824 /* Get the __import__ function from the builtins */
1825 if (PyDict_Check(builtins)) {
1826 import = PyObject_GetItem(builtins, import_str);
1827 if (import == NULL)
1828 PyErr_SetObject(PyExc_KeyError, import_str);
1829 }
1830 else
1831 import = PyObject_GetAttr(builtins, import_str);
1832 if (import == NULL)
1833 goto err;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001834
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001835 /* Call the __import__ function with the proper argument list
Brett Cannonbc2eff32010-09-19 21:39:02 +00001836 Always use absolute import here.
1837 Calling for side-effect of import. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001838 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
1839 globals, silly_list, 0, NULL);
Brett Cannonbc2eff32010-09-19 21:39:02 +00001840 if (r == NULL)
1841 goto err;
1842 Py_DECREF(r);
1843
1844 modules = PyImport_GetModuleDict();
1845 r = PyDict_GetItem(modules, module_name);
1846 if (r != NULL)
1847 Py_INCREF(r);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001848
1849 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001850 Py_XDECREF(globals);
1851 Py_XDECREF(builtins);
1852 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00001853
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001854 return r;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00001855}
1856
Brett Cannon4caa61d2014-01-09 19:03:32 -05001857/*[clinic input]
1858_imp.extension_suffixes
1859
1860Returns the list of file suffixes used to identify extension modules.
1861[clinic start generated code]*/
1862
Brett Cannon4caa61d2014-01-09 19:03:32 -05001863static PyObject *
1864_imp_extension_suffixes_impl(PyModuleDef *module)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001865/*[clinic end generated code: output=d44c1566ef362229 input=ecdeeecfcb6f839e]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001866{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001867 PyObject *list;
Brett Cannon2657df42012-05-04 15:20:40 -04001868 const char *suffix;
1869 unsigned int index = 0;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001870
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001871 list = PyList_New(0);
1872 if (list == NULL)
1873 return NULL;
Brett Cannon2657df42012-05-04 15:20:40 -04001874#ifdef HAVE_DYNAMIC_LOADING
1875 while ((suffix = _PyImport_DynLoadFiletab[index])) {
1876 PyObject *item = PyUnicode_FromString(suffix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 if (item == NULL) {
1878 Py_DECREF(list);
1879 return NULL;
1880 }
1881 if (PyList_Append(list, item) < 0) {
1882 Py_DECREF(list);
1883 Py_DECREF(item);
1884 return NULL;
1885 }
1886 Py_DECREF(item);
Brett Cannon2657df42012-05-04 15:20:40 -04001887 index += 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001888 }
Brett Cannon2657df42012-05-04 15:20:40 -04001889#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 return list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001891}
1892
Brett Cannon4caa61d2014-01-09 19:03:32 -05001893/*[clinic input]
Brett Cannon4caa61d2014-01-09 19:03:32 -05001894_imp.init_frozen
1895
1896 name: unicode
1897 /
1898
1899Initializes a frozen module.
1900[clinic start generated code]*/
1901
Brett Cannon4caa61d2014-01-09 19:03:32 -05001902static PyObject *
1903_imp_init_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001904/*[clinic end generated code: output=a9de493bdd711878 input=13019adfc04f3fb3]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001905{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001906 int ret;
1907 PyObject *m;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001908
Victor Stinner53dc7352011-03-20 01:50:21 +01001909 ret = PyImport_ImportFrozenModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001910 if (ret < 0)
1911 return NULL;
1912 if (ret == 0) {
1913 Py_INCREF(Py_None);
1914 return Py_None;
1915 }
Victor Stinner53dc7352011-03-20 01:50:21 +01001916 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001917 Py_XINCREF(m);
1918 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001919}
1920
Brett Cannon4caa61d2014-01-09 19:03:32 -05001921/*[clinic input]
1922_imp.get_frozen_object
1923
1924 name: unicode
1925 /
1926
1927Create a code object for a frozen module.
1928[clinic start generated code]*/
1929
Brett Cannon4caa61d2014-01-09 19:03:32 -05001930static PyObject *
1931_imp_get_frozen_object_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001932/*[clinic end generated code: output=3114c970a47f2e3c input=ed689bc05358fdbd]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001933{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001934 return get_frozen_object(name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001935}
1936
Brett Cannon4caa61d2014-01-09 19:03:32 -05001937/*[clinic input]
1938_imp.is_frozen_package
1939
1940 name: unicode
1941 /
1942
1943Returns True if the module name is of a frozen package.
1944[clinic start generated code]*/
1945
Brett Cannon4caa61d2014-01-09 19:03:32 -05001946static PyObject *
1947_imp_is_frozen_package_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001948/*[clinic end generated code: output=3e4cab802b56d649 input=81b6cdecd080fbb8]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001950 return is_frozen_package(name);
Brett Cannon8d110132009-03-15 02:20:16 +00001951}
1952
Brett Cannon4caa61d2014-01-09 19:03:32 -05001953/*[clinic input]
1954_imp.is_builtin
1955
1956 name: unicode
1957 /
1958
1959Returns True if the module name corresponds to a built-in module.
1960[clinic start generated code]*/
1961
Guido van Rossum79f25d91997-04-29 20:08:16 +00001962static PyObject *
Brett Cannon4caa61d2014-01-09 19:03:32 -05001963_imp_is_builtin_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001964/*[clinic end generated code: output=2deec9cac6fb9a7e input=86befdac021dd1c7]*/
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001965{
Brett Cannon4caa61d2014-01-09 19:03:32 -05001966 return PyLong_FromLong(is_builtin(name));
1967}
1968
1969/*[clinic input]
1970_imp.is_frozen
1971
1972 name: unicode
1973 /
1974
1975Returns True if the module name corresponds to a frozen module.
1976[clinic start generated code]*/
1977
Brett Cannon4caa61d2014-01-09 19:03:32 -05001978static PyObject *
1979_imp_is_frozen_impl(PyModuleDef *module, PyObject *name)
Brett Cannonfd4d0502014-05-30 11:21:14 -04001980/*[clinic end generated code: output=7de8e260c8e36aed input=7301dbca1897d66b]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05001981{
Benjamin Peterson6fba3db2013-03-18 23:13:31 -07001982 const struct _frozen *p;
Brett Cannon4caa61d2014-01-09 19:03:32 -05001983
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001984 p = find_frozen(name);
1985 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001986}
1987
Larry Hastings1df0b352015-08-24 19:53:56 -07001988/* Common implementation for _imp.exec_dynamic and _imp.exec_builtin */
1989static int
1990exec_builtin_or_dynamic(PyObject *mod) {
1991 PyModuleDef *def;
1992 void *state;
1993
1994 if (!PyModule_Check(mod)) {
1995 return 0;
1996 }
1997
1998 def = PyModule_GetDef(mod);
1999 if (def == NULL) {
2000 if (PyErr_Occurred()) {
2001 return -1;
2002 }
2003 return 0;
2004 }
2005 state = PyModule_GetState(mod);
2006 if (PyErr_Occurred()) {
2007 return -1;
2008 }
2009 if (state) {
2010 /* Already initialized; skip reload */
2011 return 0;
2012 }
2013 return PyModule_ExecDef(mod, def);
2014}
2015
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002016#ifdef HAVE_DYNAMIC_LOADING
2017
Brett Cannon4caa61d2014-01-09 19:03:32 -05002018/*[clinic input]
Nick Coghland5cacbb2015-05-23 22:24:10 +10002019_imp.create_dynamic
Brett Cannon4caa61d2014-01-09 19:03:32 -05002020
Nick Coghland5cacbb2015-05-23 22:24:10 +10002021 spec: object
Brett Cannon4caa61d2014-01-09 19:03:32 -05002022 file: object = NULL
2023 /
2024
Nick Coghland5cacbb2015-05-23 22:24:10 +10002025Create an extension module.
Brett Cannon4caa61d2014-01-09 19:03:32 -05002026[clinic start generated code]*/
2027
Brett Cannon4caa61d2014-01-09 19:03:32 -05002028static PyObject *
Nick Coghland5cacbb2015-05-23 22:24:10 +10002029_imp_create_dynamic_impl(PyModuleDef *module, PyObject *spec, PyObject *file)
2030/*[clinic end generated code: output=935cde5b3872d56d input=c31b954f4cf4e09d]*/
Brett Cannon4caa61d2014-01-09 19:03:32 -05002031{
Nick Coghland5cacbb2015-05-23 22:24:10 +10002032 PyObject *mod, *name, *path;
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002033 FILE *fp;
2034
Nick Coghland5cacbb2015-05-23 22:24:10 +10002035 name = PyObject_GetAttrString(spec, "name");
2036 if (name == NULL) {
2037 return NULL;
2038 }
2039
2040 path = PyObject_GetAttrString(spec, "origin");
2041 if (path == NULL) {
2042 Py_DECREF(name);
2043 return NULL;
2044 }
2045
2046 mod = _PyImport_FindExtensionObject(name, path);
2047 if (mod != NULL) {
2048 Py_DECREF(name);
2049 Py_DECREF(path);
2050 Py_INCREF(mod);
2051 return mod;
2052 }
2053
Brett Cannon4caa61d2014-01-09 19:03:32 -05002054 if (file != NULL) {
2055 fp = _Py_fopen_obj(path, "r");
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002056 if (fp == NULL) {
Nick Coghland5cacbb2015-05-23 22:24:10 +10002057 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002058 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002059 return NULL;
Victor Stinnere9ddbf62011-03-22 10:46:35 +01002060 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002061 }
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002062 else
2063 fp = NULL;
Nick Coghland5cacbb2015-05-23 22:24:10 +10002064
2065 mod = _PyImport_LoadDynamicModuleWithSpec(spec, fp);
2066
2067 Py_DECREF(name);
Brett Cannon4caa61d2014-01-09 19:03:32 -05002068 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002069 if (fp)
2070 fclose(fp);
Victor Stinnerfefd70c2011-03-14 15:54:07 -04002071 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002072}
2073
Nick Coghland5cacbb2015-05-23 22:24:10 +10002074/*[clinic input]
2075_imp.exec_dynamic -> int
2076
2077 mod: object
2078 /
2079
2080Initialize an extension module.
2081[clinic start generated code]*/
2082
2083static int
2084_imp_exec_dynamic_impl(PyModuleDef *module, PyObject *mod)
2085/*[clinic end generated code: output=4b84f1301b22d4bd input=9fdbfcb250280d3a]*/
2086{
Larry Hastings1df0b352015-08-24 19:53:56 -07002087 return exec_builtin_or_dynamic(mod);
Nick Coghland5cacbb2015-05-23 22:24:10 +10002088}
2089
2090
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002091#endif /* HAVE_DYNAMIC_LOADING */
2092
Larry Hastings7726ac92014-01-31 22:03:12 -08002093/*[clinic input]
Larry Hastings1df0b352015-08-24 19:53:56 -07002094_imp.exec_builtin -> int
2095
2096 mod: object
2097 /
2098
2099Initialize a built-in module.
2100[clinic start generated code]*/
2101
2102static int
2103_imp_exec_builtin_impl(PyModuleDef *module, PyObject *mod)
Larry Hastings01b1ff62015-08-24 20:23:27 -07002104/*[clinic end generated code: output=215e99876a27e284 input=7beed5a2f12a60ca]*/
Larry Hastings1df0b352015-08-24 19:53:56 -07002105{
2106 return exec_builtin_or_dynamic(mod);
2107}
2108
2109/*[clinic input]
Larry Hastings7726ac92014-01-31 22:03:12 -08002110dump buffer
2111[clinic start generated code]*/
Brett Cannonfd4d0502014-05-30 11:21:14 -04002112/*[clinic end generated code: output=da39a3ee5e6b4b0d input=524ce2e021e4eba6]*/
Larry Hastings7726ac92014-01-31 22:03:12 -08002113
Barry Warsaw28a691b2010-04-17 00:19:56 +00002114
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002115PyDoc_STRVAR(doc_imp,
Brett Cannon6f44d662012-04-15 16:08:47 -04002116"(Extremely) low-level import machinery bits as used by importlib and imp.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002117
Guido van Rossum79f25d91997-04-29 20:08:16 +00002118static PyMethodDef imp_methods[] = {
Brett Cannon4caa61d2014-01-09 19:03:32 -05002119 _IMP_EXTENSION_SUFFIXES_METHODDEF
2120 _IMP_LOCK_HELD_METHODDEF
2121 _IMP_ACQUIRE_LOCK_METHODDEF
2122 _IMP_RELEASE_LOCK_METHODDEF
2123 _IMP_GET_FROZEN_OBJECT_METHODDEF
2124 _IMP_IS_FROZEN_PACKAGE_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002125 _IMP_CREATE_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002126 _IMP_INIT_FROZEN_METHODDEF
2127 _IMP_IS_BUILTIN_METHODDEF
2128 _IMP_IS_FROZEN_METHODDEF
Nick Coghland5cacbb2015-05-23 22:24:10 +10002129 _IMP_CREATE_DYNAMIC_METHODDEF
2130 _IMP_EXEC_DYNAMIC_METHODDEF
Larry Hastings1df0b352015-08-24 19:53:56 -07002131 _IMP_EXEC_BUILTIN_METHODDEF
Brett Cannon4caa61d2014-01-09 19:03:32 -05002132 _IMP__FIX_CO_FILENAME_METHODDEF
2133 {NULL, NULL} /* sentinel */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002134};
2135
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002136
Martin v. Löwis1a214512008-06-11 05:26:20 +00002137static struct PyModuleDef impmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002138 PyModuleDef_HEAD_INIT,
Brett Cannon6f44d662012-04-15 16:08:47 -04002139 "_imp",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002140 doc_imp,
2141 0,
2142 imp_methods,
2143 NULL,
2144 NULL,
2145 NULL,
2146 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00002147};
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002148
Jason Tishler6bc06ec2003-09-04 11:59:50 +00002149PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00002150PyInit_imp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002151{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002152 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002154 m = PyModule_Create(&impmodule);
2155 if (m == NULL)
2156 goto failure;
2157 d = PyModule_GetDict(m);
2158 if (d == NULL)
2159 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002161 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002162 failure:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002163 Py_XDECREF(m);
2164 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002165}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002166
2167
Guido van Rossumb18618d2000-05-03 23:44:39 +00002168/* API for embedding applications that want to add their own entries
2169 to the table of built-in modules. This should normally be called
2170 *before* Py_Initialize(). When the table resize fails, -1 is
2171 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002172
2173 After a similar function by Just van Rossum. */
2174
2175int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002176PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002177{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002178 static struct _inittab *our_copy = NULL;
2179 struct _inittab *p;
2180 int i, n;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002181
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002182 /* Count the number of entries in both tables */
2183 for (n = 0; newtab[n].name != NULL; n++)
2184 ;
2185 if (n == 0)
2186 return 0; /* Nothing to do */
2187 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2188 ;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002189
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002190 /* Allocate new memory for the combined table */
2191 p = our_copy;
2192 PyMem_RESIZE(p, struct _inittab, i+n+1);
2193 if (p == NULL)
2194 return -1;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002196 /* Copy the tables into the new memory */
2197 if (our_copy != PyImport_Inittab)
2198 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2199 PyImport_Inittab = our_copy = p;
2200 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002202 return 0;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002203}
2204
2205/* Shorthand to add a single entry given a name and a function */
2206
2207int
Brett Cannona826f322009-04-02 03:41:46 +00002208PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002209{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002210 struct _inittab newtab[2];
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002211
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002212 memset(newtab, '\0', sizeof newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002213
Serhiy Storchaka20b39b22014-09-28 11:27:24 +03002214 newtab[0].name = name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002215 newtab[0].initfunc = initfunc;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002216
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002217 return PyImport_ExtendInittab(newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002218}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002219
2220#ifdef __cplusplus
2221}
2222#endif