blob: 5af365187ec5fc8ee6d4db9005a1036e64da9dee [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"
Neal Norwitzadb69fc2005-12-17 20:54:49 +00007#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00008#include "pythonrun.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00009#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000010#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000011#include "code.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000012#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000013#include "eval.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000014#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000015#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000016
Guido van Rossum55a83382000-09-20 20:31:38 +000017#ifdef HAVE_FCNTL_H
18#include <fcntl.h>
19#endif
Anthony Baxterac6bd462006-04-13 02:06:09 +000020#ifdef __cplusplus
21extern "C" {
22#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000023
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000024extern time_t PyOS_GetLastModificationTime(char *, FILE *);
25 /* In getmtime.c */
Guido van Rossum21d335e1993-10-15 13:01:11 +000026
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000027/* Magic word to reject .pyc files generated by other Python versions.
28 It should change for each incompatible change to the bytecode.
29
30 The value of CR and LF is incorporated so if you ever read or write
Guido van Rossum7faeab31995-07-07 22:50:36 +000031 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000032 Apple MPW compiler swaps their values, botching string constants.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000033
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000034 The magic numbers must be spaced apart atleast 2 values, as the
35 -U interpeter flag will cause MAGIC+1 being used. They have been
36 odd numbers for some time now.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000037
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000038 There were a variety of old schemes for setting the magic number.
39 The current working scheme is to increment the previous value by
40 10.
Guido van Rossumf6894922002-08-31 15:16:14 +000041
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000042 Known values:
43 Python 1.5: 20121
44 Python 1.5.1: 20121
45 Python 1.5.2: 20121
Skip Montanaro4ec3c262006-03-25 14:12:03 +000046 Python 1.6: 50428
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000047 Python 2.0: 50823
48 Python 2.0.1: 50823
49 Python 2.1: 60202
50 Python 2.1.1: 60202
51 Python 2.1.2: 60202
52 Python 2.2: 60717
Neal Norwitz7fdcb412002-06-14 01:07:39 +000053 Python 2.3a0: 62011
Michael W. Hudsondd32a912002-08-15 14:59:02 +000054 Python 2.3a0: 62021
Guido van Rossumf6894922002-08-31 15:16:14 +000055 Python 2.3a0: 62011 (!)
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000056 Python 2.4a0: 62041
Raymond Hettingerfd2d1f72004-08-23 23:37:48 +000057 Python 2.4a3: 62051
Raymond Hettinger2c31a052004-09-22 18:44:21 +000058 Python 2.4b1: 62061
Michael W. Hudsondf888462005-06-03 14:41:55 +000059 Python 2.5a0: 62071
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000060 Python 2.5a0: 62081 (ast-branch)
Guido van Rossumc2e20742006-02-27 22:32:47 +000061 Python 2.5a0: 62091 (with)
Guido van Rossumf6694362006-03-10 02:28:35 +000062 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode)
Neal Norwitz0d62a062006-07-30 06:53:31 +000063 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
64 Python 2.5b3: 62111 (fix wrong code: x += yield)
Neal Norwitz9a70f952006-08-04 05:12:19 +000065 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
66 storing constants that should have been removed)
Neal Norwitz3b3aae02006-09-05 03:56:01 +000067 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp)
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000068.
Tim Peters36515e22001-11-18 04:06:29 +000069*/
Neal Norwitz3b3aae02006-09-05 03:56:01 +000070#define MAGIC (62131 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000071
Guido van Rossum96774c12000-05-01 20:19:08 +000072/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000073 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000074 compiler works which are enabled by command line switches. */
75static long pyc_magic = MAGIC;
76
Guido van Rossum25ce5661997-08-02 03:10:38 +000077/* See _PyImport_FixupExtension() below */
78static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000079
Guido van Rossum771c6c81997-10-31 18:37:24 +000080/* This table is defined in config.c: */
81extern struct _inittab _PyImport_Inittab[];
82
83struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000084
Guido van Rossumed1170e1999-12-20 21:23:41 +000085/* these tables define the module suffixes that Python recognizes */
86struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +000087
88#ifdef RISCOS
89static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000090 {"/py", "U", PY_SOURCE},
Guido van Rossum48a680c2001-03-02 06:34:14 +000091 {"/pyc", "rb", PY_COMPILED},
92 {0, 0}
93};
94#else
Guido van Rossumed1170e1999-12-20 21:23:41 +000095static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000096 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000097#ifdef MS_WINDOWS
Jack Jansenc88da1f2002-05-28 10:58:19 +000098 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +000099#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000100 {".pyc", "rb", PY_COMPILED},
101 {0, 0}
102};
Guido van Rossum48a680c2001-03-02 06:34:14 +0000103#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000104
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000105static PyTypeObject NullImporterType; /* Forward reference */
106
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000107/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000108
109void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000110_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000111{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000112 const struct filedescr *scan;
113 struct filedescr *filetab;
114 int countD = 0;
115 int countS = 0;
116
117 /* prepare _PyImport_Filetab: copy entries from
118 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
119 */
120 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
121 ++countD;
122 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
123 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000124 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Neal Norwitze1fdb322006-07-21 05:32:28 +0000125 if (filetab == NULL)
Neal Norwitz33722ae2006-07-21 07:59:02 +0000126 Py_FatalError("Can't initialize import file table.");
Guido van Rossumed1170e1999-12-20 21:23:41 +0000127 memcpy(filetab, _PyImport_DynLoadFiletab,
128 countD * sizeof(struct filedescr));
129 memcpy(filetab + countD, _PyImport_StandardFiletab,
130 countS * sizeof(struct filedescr));
131 filetab[countD + countS].suffix = NULL;
132
133 _PyImport_Filetab = filetab;
134
Guido van Rossum0824f631997-03-11 18:37:35 +0000135 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000136 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
137 for (; filetab->suffix != NULL; filetab++) {
Guido van Rossum48a680c2001-03-02 06:34:14 +0000138#ifndef RISCOS
Guido van Rossumed1170e1999-12-20 21:23:41 +0000139 if (strcmp(filetab->suffix, ".pyc") == 0)
140 filetab->suffix = ".pyo";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000141#else
142 if (strcmp(filetab->suffix, "/pyc") == 0)
143 filetab->suffix = "/pyo";
144#endif
Guido van Rossum0824f631997-03-11 18:37:35 +0000145 }
146 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000147
148 if (Py_UnicodeFlag) {
149 /* Fix the pyc_magic so that byte compiled code created
150 using the all-Unicode method doesn't interfere with
151 code created in normal operation mode. */
152 pyc_magic = MAGIC + 1;
153 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000154}
155
Guido van Rossum25ce5661997-08-02 03:10:38 +0000156void
Just van Rossum52e14d62002-12-30 22:08:05 +0000157_PyImportHooks_Init(void)
158{
159 PyObject *v, *path_hooks = NULL, *zimpimport;
160 int err = 0;
161
162 /* adding sys.path_hooks and sys.path_importer_cache, setting up
163 zipimport */
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000164 if (PyType_Ready(&NullImporterType) < 0)
165 goto error;
Just van Rossum52e14d62002-12-30 22:08:05 +0000166
167 if (Py_VerboseFlag)
168 PySys_WriteStderr("# installing zipimport hook\n");
169
170 v = PyList_New(0);
171 if (v == NULL)
172 goto error;
173 err = PySys_SetObject("meta_path", v);
174 Py_DECREF(v);
175 if (err)
176 goto error;
177 v = PyDict_New();
178 if (v == NULL)
179 goto error;
180 err = PySys_SetObject("path_importer_cache", v);
181 Py_DECREF(v);
182 if (err)
183 goto error;
184 path_hooks = PyList_New(0);
185 if (path_hooks == NULL)
186 goto error;
187 err = PySys_SetObject("path_hooks", path_hooks);
188 if (err) {
189 error:
190 PyErr_Print();
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000191 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
192 "path_importer_cache, or NullImporter failed"
193 );
Just van Rossum52e14d62002-12-30 22:08:05 +0000194 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000195
Just van Rossum52e14d62002-12-30 22:08:05 +0000196 zimpimport = PyImport_ImportModule("zipimport");
197 if (zimpimport == NULL) {
198 PyErr_Clear(); /* No zip import module -- okay */
199 if (Py_VerboseFlag)
200 PySys_WriteStderr("# can't import zipimport\n");
201 }
202 else {
203 PyObject *zipimporter = PyObject_GetAttrString(zimpimport,
204 "zipimporter");
205 Py_DECREF(zimpimport);
206 if (zipimporter == NULL) {
207 PyErr_Clear(); /* No zipimporter object -- okay */
208 if (Py_VerboseFlag)
209 PySys_WriteStderr(
Fred Drake1e5fc552003-07-11 15:01:02 +0000210 "# can't import zipimport.zipimporter\n");
Just van Rossum52e14d62002-12-30 22:08:05 +0000211 }
212 else {
213 /* sys.path_hooks.append(zipimporter) */
214 err = PyList_Append(path_hooks, zipimporter);
215 Py_DECREF(zipimporter);
216 if (err)
217 goto error;
218 if (Py_VerboseFlag)
219 PySys_WriteStderr(
220 "# installed zipimport hook\n");
221 }
222 }
223 Py_DECREF(path_hooks);
224}
225
226void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000227_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000228{
229 Py_XDECREF(extensions);
230 extensions = NULL;
Barry Warsaw84294482000-10-03 16:02:05 +0000231 PyMem_DEL(_PyImport_Filetab);
232 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000233}
234
235
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000236/* Locking primitives to prevent parallel imports of the same module
237 in different threads to return with a partially loaded module.
238 These calls are serialized by the global interpreter lock. */
239
240#ifdef WITH_THREAD
241
Guido van Rossum49b56061998-10-01 20:42:43 +0000242#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000243
Guido van Rossum65d5b571998-12-21 19:32:43 +0000244static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000245static long import_lock_thread = -1;
246static int import_lock_level = 0;
247
248static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000249lock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000250{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000251 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000252 if (me == -1)
253 return; /* Too bad */
Neal Norwitze1fdb322006-07-21 05:32:28 +0000254 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000255 import_lock = PyThread_allocate_lock();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000256 if (import_lock == NULL)
257 return; /* Nothing much we can do. */
258 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000259 if (import_lock_thread == me) {
260 import_lock_level++;
261 return;
262 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000263 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
264 {
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000265 PyThreadState *tstate = PyEval_SaveThread();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000266 PyThread_acquire_lock(import_lock, 1);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000267 PyEval_RestoreThread(tstate);
268 }
269 import_lock_thread = me;
270 import_lock_level = 1;
271}
272
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000273static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000274unlock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000275{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000276 long me = PyThread_get_thread_ident();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000277 if (me == -1 || import_lock == NULL)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000278 return 0; /* Too bad */
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000279 if (import_lock_thread != me)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000280 return -1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000281 import_lock_level--;
282 if (import_lock_level == 0) {
283 import_lock_thread = -1;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000284 PyThread_release_lock(import_lock);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000285 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000286 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000287}
288
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000289/* This function is called from PyOS_AfterFork to ensure that newly
290 created child processes do not share locks with the parent. */
291
292void
293_PyImport_ReInitLock(void)
294{
295#ifdef _AIX
296 if (import_lock != NULL)
297 import_lock = PyThread_allocate_lock();
298#endif
299}
300
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000301#else
302
303#define lock_import()
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000304#define unlock_import() 0
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000305
306#endif
307
Tim Peters69232342001-08-30 05:16:13 +0000308static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000309imp_lock_held(PyObject *self, PyObject *noargs)
Tim Peters69232342001-08-30 05:16:13 +0000310{
Tim Peters69232342001-08-30 05:16:13 +0000311#ifdef WITH_THREAD
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000312 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000313#else
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000314 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000315#endif
316}
317
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000318static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000319imp_acquire_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000320{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000321#ifdef WITH_THREAD
322 lock_import();
323#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000324 Py_INCREF(Py_None);
325 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000326}
327
328static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000329imp_release_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000330{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000331#ifdef WITH_THREAD
332 if (unlock_import() < 0) {
333 PyErr_SetString(PyExc_RuntimeError,
334 "not holding the import lock");
335 return NULL;
336 }
337#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000338 Py_INCREF(Py_None);
339 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000340}
341
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342/* Helper for sys */
343
344PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000345PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000346{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000347 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000348 if (interp->modules == NULL)
349 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
350 return interp->modules;
351}
352
Guido van Rossum3f5da241990-12-20 15:06:42 +0000353
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000354/* List of names to clear in sys */
355static char* sys_deletes[] = {
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000356 "path", "argv", "ps1", "ps2", "exitfunc",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000357 "exc_type", "exc_value", "exc_traceback",
358 "last_type", "last_value", "last_traceback",
Just van Rossum52e14d62002-12-30 22:08:05 +0000359 "path_hooks", "path_importer_cache", "meta_path",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000360 NULL
361};
362
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000363static char* sys_files[] = {
364 "stdin", "__stdin__",
365 "stdout", "__stdout__",
366 "stderr", "__stderr__",
367 NULL
368};
369
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000370
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000371/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000372
Guido van Rossum3f5da241990-12-20 15:06:42 +0000373void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000374PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000375{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000376 Py_ssize_t pos, ndone;
Guido van Rossum758eec01998-01-19 21:58:26 +0000377 char *name;
378 PyObject *key, *value, *dict;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000379 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000380 PyObject *modules = interp->modules;
381
382 if (modules == NULL)
383 return; /* Already done */
384
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000385 /* Delete some special variables first. These are common
386 places where user values hide and people complain when their
387 destructors fail. Since the modules containing them are
388 deleted *last* of all, they would come too late in the normal
389 destruction order. Sigh. */
390
391 value = PyDict_GetItemString(modules, "__builtin__");
392 if (value != NULL && PyModule_Check(value)) {
393 dict = PyModule_GetDict(value);
394 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000395 PySys_WriteStderr("# clear __builtin__._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000396 PyDict_SetItemString(dict, "_", Py_None);
397 }
398 value = PyDict_GetItemString(modules, "sys");
399 if (value != NULL && PyModule_Check(value)) {
400 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000401 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000402 dict = PyModule_GetDict(value);
403 for (p = sys_deletes; *p != NULL; p++) {
404 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000405 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000406 PyDict_SetItemString(dict, *p, Py_None);
407 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000408 for (p = sys_files; *p != NULL; p+=2) {
409 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000410 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000411 v = PyDict_GetItemString(dict, *(p+1));
412 if (v == NULL)
413 v = Py_None;
414 PyDict_SetItemString(dict, *p, v);
415 }
416 }
417
418 /* First, delete __main__ */
419 value = PyDict_GetItemString(modules, "__main__");
420 if (value != NULL && PyModule_Check(value)) {
421 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000422 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000423 _PyModule_Clear(value);
424 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000425 }
426
Guido van Rossum758eec01998-01-19 21:58:26 +0000427 /* The special treatment of __builtin__ here is because even
428 when it's not referenced as a module, its dictionary is
429 referenced by almost every module's __builtins__. Since
430 deleting a module clears its dictionary (even if there are
431 references left to it), we need to delete the __builtin__
432 module last. Likewise, we don't delete sys until the very
433 end because it is implicitly referenced (e.g. by print).
434
435 Also note that we 'delete' modules by replacing their entry
436 in the modules dict with None, rather than really deleting
437 them; this avoids a rehash of the modules dictionary and
438 also marks them as "non existent" so they won't be
439 re-imported. */
440
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000441 /* Next, repeatedly delete modules with a reference count of
Guido van Rossum758eec01998-01-19 21:58:26 +0000442 one (skipping __builtin__ and sys) and delete them */
443 do {
444 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000445 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000446 while (PyDict_Next(modules, &pos, &key, &value)) {
447 if (value->ob_refcnt != 1)
448 continue;
Guido van Rossum566373e1998-10-01 15:24:50 +0000449 if (PyString_Check(key) && PyModule_Check(value)) {
450 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000451 if (strcmp(name, "__builtin__") == 0)
452 continue;
453 if (strcmp(name, "sys") == 0)
454 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000455 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000456 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000457 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000458 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000459 PyDict_SetItem(modules, key, Py_None);
460 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000461 }
462 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000463 } while (ndone > 0);
464
Guido van Rossum758eec01998-01-19 21:58:26 +0000465 /* Next, delete all modules (still skipping __builtin__ and sys) */
466 pos = 0;
467 while (PyDict_Next(modules, &pos, &key, &value)) {
Guido van Rossum566373e1998-10-01 15:24:50 +0000468 if (PyString_Check(key) && PyModule_Check(value)) {
469 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000470 if (strcmp(name, "__builtin__") == 0)
471 continue;
472 if (strcmp(name, "sys") == 0)
473 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000474 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000475 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000476 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000477 PyDict_SetItem(modules, key, Py_None);
478 }
479 }
480
481 /* Next, delete sys and __builtin__ (in that order) */
482 value = PyDict_GetItemString(modules, "sys");
483 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000484 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000485 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000486 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000487 PyDict_SetItemString(modules, "sys", Py_None);
488 }
489 value = PyDict_GetItemString(modules, "__builtin__");
490 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000491 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000492 PySys_WriteStderr("# cleanup __builtin__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000493 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000494 PyDict_SetItemString(modules, "__builtin__", Py_None);
495 }
496
497 /* Finally, clear and delete the modules directory */
498 PyDict_Clear(modules);
499 interp->modules = NULL;
500 Py_DECREF(modules);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000501}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000502
503
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000504/* Helper for pythonrun.c -- return magic number */
505
506long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000507PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000508{
Guido van Rossum96774c12000-05-01 20:19:08 +0000509 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000510}
511
512
Guido van Rossum25ce5661997-08-02 03:10:38 +0000513/* Magic for extension modules (built-in as well as dynamically
514 loaded). To prevent initializing an extension module more than
515 once, we keep a static dictionary 'extensions' keyed by module name
516 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000517 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000518 dictionary is stored by calling _PyImport_FixupExtension()
519 immediately after the module initialization function succeeds. A
520 copy can be retrieved from there by calling
521 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000522
Guido van Rossum79f25d91997-04-29 20:08:16 +0000523PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000524_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000525{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000526 PyObject *modules, *mod, *dict, *copy;
527 if (extensions == NULL) {
528 extensions = PyDict_New();
529 if (extensions == NULL)
530 return NULL;
531 }
532 modules = PyImport_GetModuleDict();
533 mod = PyDict_GetItemString(modules, name);
534 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000535 PyErr_Format(PyExc_SystemError,
536 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000537 return NULL;
538 }
539 dict = PyModule_GetDict(mod);
540 if (dict == NULL)
541 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000542 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000543 if (copy == NULL)
544 return NULL;
545 PyDict_SetItemString(extensions, filename, copy);
546 Py_DECREF(copy);
547 return copy;
548}
549
550PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000551_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000552{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000553 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000554 if (extensions == NULL)
555 return NULL;
556 dict = PyDict_GetItemString(extensions, filename);
557 if (dict == NULL)
558 return NULL;
559 mod = PyImport_AddModule(name);
560 if (mod == NULL)
561 return NULL;
562 mdict = PyModule_GetDict(mod);
563 if (mdict == NULL)
564 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000565 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000566 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000568 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000569 name, filename);
570 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000571}
572
573
574/* Get the module object corresponding to a module name.
575 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000576 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000577 Because the former action is most common, THIS DOES NOT RETURN A
578 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000579
Guido van Rossum79f25d91997-04-29 20:08:16 +0000580PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000581PyImport_AddModule(const char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000582{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000583 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000584 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000585
Guido van Rossum25ce5661997-08-02 03:10:38 +0000586 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000587 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000588 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000589 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000590 if (m == NULL)
591 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000592 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000593 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000594 return NULL;
595 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000596 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000597
598 return m;
599}
600
Tim Peters1cd70172004-08-02 03:52:12 +0000601/* Remove name from sys.modules, if it's there. */
602static void
603_RemoveModule(const char *name)
604{
605 PyObject *modules = PyImport_GetModuleDict();
606 if (PyDict_GetItemString(modules, name) == NULL)
607 return;
608 if (PyDict_DelItemString(modules, name) < 0)
609 Py_FatalError("import: deleting existing key in"
610 "sys.modules failed");
611}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000612
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000613/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000614 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
615 * removed from sys.modules, to avoid leaving damaged module objects
616 * in sys.modules. The caller may wish to restore the original
617 * module object (if any) in this case; PyImport_ReloadModule is an
618 * example.
619 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000620PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000621PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000622{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000623 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
624}
625
626PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000627PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000628{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000629 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000630 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000631
Guido van Rossum79f25d91997-04-29 20:08:16 +0000632 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000633 if (m == NULL)
634 return NULL;
Jeremy Hyltonecd91292004-01-02 23:25:32 +0000635 /* If the module is being reloaded, we get the old module back
636 and re-use its dict to exec the new code. */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000637 d = PyModule_GetDict(m);
638 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
639 if (PyDict_SetItemString(d, "__builtins__",
640 PyEval_GetBuiltins()) != 0)
Tim Peters1cd70172004-08-02 03:52:12 +0000641 goto error;
Guido van Rossum6135a871995-01-09 17:53:26 +0000642 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000643 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000644 v = NULL;
645 if (pathname != NULL) {
646 v = PyString_FromString(pathname);
647 if (v == NULL)
648 PyErr_Clear();
649 }
650 if (v == NULL) {
651 v = ((PyCodeObject *)co)->co_filename;
652 Py_INCREF(v);
653 }
654 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000655 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000656 Py_DECREF(v);
657
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000658 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000659 if (v == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000660 goto error;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000661 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000662
Guido van Rossum25ce5661997-08-02 03:10:38 +0000663 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000664 PyErr_Format(PyExc_ImportError,
665 "Loaded module %.200s not found in sys.modules",
666 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000667 return NULL;
668 }
669
Guido van Rossum79f25d91997-04-29 20:08:16 +0000670 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000671
672 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000673
674 error:
675 _RemoveModule(name);
676 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000677}
678
679
680/* Given a pathname for a Python source file, fill a buffer with the
681 pathname for the corresponding compiled file. Return the pathname
682 for the compiled file, or NULL if there's no space in the buffer.
683 Doesn't set an exception. */
684
685static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000686make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000687{
Tim Petersc1731372001-08-04 08:12:36 +0000688 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000689 if (len+2 > buflen)
690 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000691
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000692#ifdef MS_WINDOWS
Tim Petersc1731372001-08-04 08:12:36 +0000693 /* Treat .pyw as if it were .py. The case of ".pyw" must match
694 that used in _PyImport_StandardFiletab. */
695 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
696 --len; /* pretend 'w' isn't there */
697#endif
698 memcpy(buf, pathname, len);
699 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
700 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000701
702 return buf;
703}
704
705
706/* Given a pathname for a Python source file, its time of last
707 modification, and a pathname for a compiled file, check whether the
708 compiled file represents the same version of the source. If so,
709 return a FILE pointer for the compiled file, positioned just after
710 the header; if not, return NULL.
711 Doesn't set an exception. */
712
713static FILE *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000714check_compiled_module(char *pathname, time_t mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000715{
716 FILE *fp;
717 long magic;
718 long pyc_mtime;
719
720 fp = fopen(cpathname, "rb");
721 if (fp == NULL)
722 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000723 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000724 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000725 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000726 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000727 fclose(fp);
728 return NULL;
729 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000730 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000731 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000732 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000733 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000734 fclose(fp);
735 return NULL;
736 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000737 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000738 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000739 return fp;
740}
741
742
743/* Read a code object from a file and check it for validity */
744
Guido van Rossum79f25d91997-04-29 20:08:16 +0000745static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000746read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000747{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000748 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000749
Tim Petersd9b9ac82001-01-28 00:27:39 +0000750 co = PyMarshal_ReadLastObjectFromFile(fp);
Armin Rigo01ab2792004-03-26 15:09:27 +0000751 if (co == NULL)
752 return NULL;
753 if (!PyCode_Check(co)) {
754 PyErr_Format(PyExc_ImportError,
755 "Non-code object in %.200s", cpathname);
756 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000757 return NULL;
758 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000759 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000760}
761
762
763/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000764 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000765
Guido van Rossum79f25d91997-04-29 20:08:16 +0000766static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000767load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000768{
769 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000770 PyCodeObject *co;
771 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000772
Guido van Rossum79f25d91997-04-29 20:08:16 +0000773 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000774 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000775 PyErr_Format(PyExc_ImportError,
776 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000777 return NULL;
778 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000779 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000780 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000781 if (co == NULL)
782 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000783 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000784 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000785 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000786 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000787 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000788
789 return m;
790}
791
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000792/* Parse a source file and return the corresponding code object */
793
Guido van Rossum79f25d91997-04-29 20:08:16 +0000794static PyCodeObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000795parse_source_module(const char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000796{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000797 PyCodeObject *co = NULL;
798 mod_ty mod;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000799 PyArena *arena = PyArena_New();
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000800
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000801 mod = PyParser_ASTFromFile(fp, pathname, Py_file_input, 0, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000802 NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000803 if (mod) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000804 co = PyAST_Compile(mod, pathname, NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000805 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000806 PyArena_Free(arena);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000807 return co;
808}
809
810
Guido van Rossum55a83382000-09-20 20:31:38 +0000811/* Helper to open a bytecode file for writing in exclusive mode */
812
813static FILE *
814open_exclusive(char *filename)
815{
816#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
817 /* Use O_EXCL to avoid a race condition when another process tries to
818 write the same file. When that happens, our open() call fails,
819 which is just fine (since it's only a cache).
820 XXX If the file exists and is writable but the directory is not
821 writable, the file will never be written. Oh well.
822 */
823 int fd;
824 (void) unlink(filename);
Tim Peters42c83af2000-09-29 04:03:10 +0000825 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
826#ifdef O_BINARY
827 |O_BINARY /* necessary for Windows */
828#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000829#ifdef __VMS
Martin v. Löwis18e16552006-02-15 17:27:45 +0000830 , 0666, "ctxt=bin", "shr=nil"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000831#else
Martin v. Löwis18e16552006-02-15 17:27:45 +0000832 , 0666
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000833#endif
Martin v. Löwis18e16552006-02-15 17:27:45 +0000834 );
Guido van Rossum55a83382000-09-20 20:31:38 +0000835 if (fd < 0)
836 return NULL;
837 return fdopen(fd, "wb");
838#else
839 /* Best we can do -- on Windows this can't happen anyway */
840 return fopen(filename, "wb");
841#endif
842}
843
844
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000845/* Write a compiled module to a file, placing the time of last
846 modification of its source into the header.
847 Errors are ignored, if a write error occurs an attempt is made to
848 remove the file. */
849
850static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000851write_compiled_module(PyCodeObject *co, char *cpathname, time_t mtime)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000852{
853 FILE *fp;
854
Guido van Rossum55a83382000-09-20 20:31:38 +0000855 fp = open_exclusive(cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000856 if (fp == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000857 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000858 PySys_WriteStderr(
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000859 "# can't create %s\n", cpathname);
860 return;
861 }
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000862 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000863 /* First write a 0 for mtime */
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000864 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION);
865 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION);
Armin Rigo01ab2792004-03-26 15:09:27 +0000866 if (fflush(fp) != 0 || ferror(fp)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000867 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000868 PySys_WriteStderr("# can't write %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000869 /* Don't keep partial file */
870 fclose(fp);
871 (void) unlink(cpathname);
872 return;
873 }
874 /* Now write the true mtime */
875 fseek(fp, 4L, 0);
Martin v. Löwis18e16552006-02-15 17:27:45 +0000876 assert(mtime < LONG_MAX);
Martin v. Löwis725507b2006-03-07 12:08:51 +0000877 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000878 fflush(fp);
879 fclose(fp);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000880 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000881 PySys_WriteStderr("# wrote %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000882}
883
884
885/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000886 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
887 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000888
Guido van Rossum79f25d91997-04-29 20:08:16 +0000889static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000890load_source_module(char *name, char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000891{
Fred Drake4c82b232000-06-30 16:18:57 +0000892 time_t mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000893 FILE *fpc;
894 char buf[MAXPATHLEN+1];
895 char *cpathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000896 PyCodeObject *co;
897 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000898
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000899 mtime = PyOS_GetLastModificationTime(pathname, fp);
Neal Norwitz708e51a2005-10-03 04:48:15 +0000900 if (mtime == (time_t)(-1)) {
901 PyErr_Format(PyExc_RuntimeError,
902 "unable to get modification time from '%s'",
903 pathname);
Fred Drake4c82b232000-06-30 16:18:57 +0000904 return NULL;
Neal Norwitz708e51a2005-10-03 04:48:15 +0000905 }
Fred Drake4c82b232000-06-30 16:18:57 +0000906#if SIZEOF_TIME_T > 4
907 /* Python's .pyc timestamp handling presumes that the timestamp fits
908 in 4 bytes. This will be fine until sometime in the year 2038,
909 when a 4-byte signed time_t will overflow.
910 */
911 if (mtime >> 32) {
912 PyErr_SetString(PyExc_OverflowError,
Fred Drakec63d3e92001-03-01 06:33:32 +0000913 "modification time overflows a 4 byte field");
Fred Drake4c82b232000-06-30 16:18:57 +0000914 return NULL;
915 }
916#endif
Tim Peters36515e22001-11-18 04:06:29 +0000917 cpathname = make_compiled_pathname(pathname, buf,
Jeremy Hylton37832f02001-04-13 17:50:20 +0000918 (size_t)MAXPATHLEN + 1);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000919 if (cpathname != NULL &&
920 (fpc = check_compiled_module(pathname, mtime, cpathname))) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000921 co = read_compiled_module(cpathname, fpc);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000922 fclose(fpc);
923 if (co == NULL)
924 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000925 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000926 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000927 name, cpathname);
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000928 pathname = cpathname;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000929 }
930 else {
931 co = parse_source_module(pathname, fp);
932 if (co == NULL)
933 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000934 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000935 PySys_WriteStderr("import %s # from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000936 name, pathname);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +0000937 if (cpathname)
938 write_compiled_module(co, cpathname, mtime);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000939 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000940 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000941 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000942
943 return m;
944}
945
946
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000947/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +0000948static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
949static struct filedescr *find_module(char *, char *, PyObject *,
950 char *, size_t, FILE **, PyObject **);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000951static struct _frozen *find_frozen(char *name);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000952
953/* Load a package and return its module object WITH INCREMENTED
954 REFERENCE COUNT */
955
956static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000957load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000958{
Tim Peters1cd70172004-08-02 03:52:12 +0000959 PyObject *m, *d;
960 PyObject *file = NULL;
961 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000962 int err;
963 char buf[MAXPATHLEN+1];
964 FILE *fp = NULL;
965 struct filedescr *fdp;
966
967 m = PyImport_AddModule(name);
968 if (m == NULL)
969 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000970 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000971 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000972 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000973 d = PyModule_GetDict(m);
974 file = PyString_FromString(pathname);
975 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000976 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000977 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +0000978 if (path == NULL)
979 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000980 err = PyDict_SetItemString(d, "__file__", file);
981 if (err == 0)
982 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +0000983 if (err != 0)
984 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000985 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +0000986 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000987 if (fdp == NULL) {
988 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
989 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +0000990 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000991 }
992 else
993 m = NULL;
994 goto cleanup;
995 }
Just van Rossum52e14d62002-12-30 22:08:05 +0000996 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000997 if (fp != NULL)
998 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +0000999 goto cleanup;
1000
1001 error:
1002 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001003 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001004 Py_XDECREF(path);
1005 Py_XDECREF(file);
1006 return m;
1007}
1008
1009
1010/* Helper to test for built-in module */
1011
1012static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001013is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001014{
1015 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001016 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1017 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1018 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001019 return -1;
1020 else
1021 return 1;
1022 }
1023 }
1024 return 0;
1025}
1026
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001027
Just van Rossum52e14d62002-12-30 22:08:05 +00001028/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1029 possibly by fetching it from the path_importer_cache dict. If it
1030 wasn't yet cached, traverse path_hooks until it a hook is found
1031 that can handle the path item. Return None if no hook could;
1032 this tells our caller it should fall back to the builtin
1033 import mechanism. Cache the result in path_importer_cache.
1034 Returns a borrowed reference. */
1035
1036static PyObject *
1037get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1038 PyObject *p)
1039{
1040 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001041 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001042
1043 /* These conditions are the caller's responsibility: */
1044 assert(PyList_Check(path_hooks));
1045 assert(PyDict_Check(path_importer_cache));
1046
1047 nhooks = PyList_Size(path_hooks);
1048 if (nhooks < 0)
1049 return NULL; /* Shouldn't happen */
1050
1051 importer = PyDict_GetItem(path_importer_cache, p);
1052 if (importer != NULL)
1053 return importer;
1054
1055 /* set path_importer_cache[p] to None to avoid recursion */
1056 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1057 return NULL;
1058
1059 for (j = 0; j < nhooks; j++) {
1060 PyObject *hook = PyList_GetItem(path_hooks, j);
1061 if (hook == NULL)
1062 return NULL;
Georg Brandl684fd0c2006-05-25 19:15:31 +00001063 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001064 if (importer != NULL)
1065 break;
1066
1067 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1068 return NULL;
1069 }
1070 PyErr_Clear();
1071 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001072 if (importer == NULL) {
1073 importer = PyObject_CallFunctionObjArgs(
1074 (PyObject *)&NullImporterType, p, NULL
1075 );
1076 if (importer == NULL) {
1077 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1078 PyErr_Clear();
1079 return Py_None;
1080 }
1081 }
1082 }
1083 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001084 int err = PyDict_SetItem(path_importer_cache, p, importer);
1085 Py_DECREF(importer);
1086 if (err != 0)
1087 return NULL;
1088 }
1089 return importer;
1090}
1091
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001092/* Search the path (default sys.path) for a module. Return the
1093 corresponding filedescr struct, and (via return arguments) the
1094 pathname and an open file. Return NULL if the module is not found. */
1095
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001096#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001097extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001098 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001099#endif
1100
Martin v. Löwis18e16552006-02-15 17:27:45 +00001101static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001102static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001103static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001104
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001105static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001106find_module(char *fullname, char *subname, PyObject *path, char *buf,
1107 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001108{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001109 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001110 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001111 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001112 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001113 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001114 PyObject *path_hooks, *path_importer_cache;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001115#ifndef RISCOS
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001116 struct stat statbuf;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001117#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001118 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1119 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1120 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001121 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001122#if defined(PYOS_OS2)
1123 size_t saved_len;
1124 size_t saved_namelen;
1125 char *saved_buf = NULL;
1126#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001127 if (p_loader != NULL)
1128 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001129
Just van Rossum52e14d62002-12-30 22:08:05 +00001130 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001131 PyErr_SetString(PyExc_OverflowError,
1132 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001133 return NULL;
1134 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001135 strcpy(name, subname);
1136
1137 /* sys.meta_path import hook */
1138 if (p_loader != NULL) {
1139 PyObject *meta_path;
1140
1141 meta_path = PySys_GetObject("meta_path");
1142 if (meta_path == NULL || !PyList_Check(meta_path)) {
1143 PyErr_SetString(PyExc_ImportError,
1144 "sys.meta_path must be a list of "
1145 "import hooks");
1146 return NULL;
1147 }
1148 Py_INCREF(meta_path); /* zap guard */
1149 npath = PyList_Size(meta_path);
1150 for (i = 0; i < npath; i++) {
1151 PyObject *loader;
1152 PyObject *hook = PyList_GetItem(meta_path, i);
1153 loader = PyObject_CallMethod(hook, "find_module",
1154 "sO", fullname,
1155 path != NULL ?
1156 path : Py_None);
1157 if (loader == NULL) {
1158 Py_DECREF(meta_path);
1159 return NULL; /* true error */
1160 }
1161 if (loader != Py_None) {
1162 /* a loader was found */
1163 *p_loader = loader;
1164 Py_DECREF(meta_path);
1165 return &importhookdescr;
1166 }
1167 Py_DECREF(loader);
1168 }
1169 Py_DECREF(meta_path);
1170 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001171
1172 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001173 /* The only type of submodule allowed inside a "frozen"
1174 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +00001175 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
1176 PyErr_SetString(PyExc_ImportError,
1177 "full frozen module name too long");
1178 return NULL;
1179 }
1180 strcpy(buf, PyString_AsString(path));
1181 strcat(buf, ".");
1182 strcat(buf, name);
1183 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001184 if (find_frozen(name) != NULL) {
1185 strcpy(buf, name);
1186 return &fd_frozen;
1187 }
1188 PyErr_Format(PyExc_ImportError,
1189 "No frozen submodule named %.200s", name);
1190 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001191 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001192 if (path == NULL) {
1193 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001194 strcpy(buf, name);
1195 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001196 }
Greg Ward201baee2001-10-04 14:52:06 +00001197 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001198 strcpy(buf, name);
1199 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001200 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001201
Guido van Rossumac279101996-08-22 23:10:58 +00001202#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001203 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1204 if (fp != NULL) {
1205 *p_fp = fp;
1206 return fdp;
1207 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001208#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001209 path = PySys_GetObject("path");
1210 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001211 if (path == NULL || !PyList_Check(path)) {
1212 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001213 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001214 return NULL;
1215 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001216
1217 path_hooks = PySys_GetObject("path_hooks");
1218 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1219 PyErr_SetString(PyExc_ImportError,
1220 "sys.path_hooks must be a list of "
1221 "import hooks");
1222 return NULL;
1223 }
1224 path_importer_cache = PySys_GetObject("path_importer_cache");
1225 if (path_importer_cache == NULL ||
1226 !PyDict_Check(path_importer_cache)) {
1227 PyErr_SetString(PyExc_ImportError,
1228 "sys.path_importer_cache must be a dict");
1229 return NULL;
1230 }
1231
Guido van Rossum79f25d91997-04-29 20:08:16 +00001232 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001233 namelen = strlen(name);
1234 for (i = 0; i < npath; i++) {
Walter Dörwald3430d702002-06-17 10:43:59 +00001235 PyObject *copy = NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001236 PyObject *v = PyList_GetItem(path, i);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00001237 if (!v)
1238 return NULL;
Walter Dörwald3430d702002-06-17 10:43:59 +00001239#ifdef Py_USING_UNICODE
1240 if (PyUnicode_Check(v)) {
1241 copy = PyUnicode_Encode(PyUnicode_AS_UNICODE(v),
1242 PyUnicode_GET_SIZE(v), Py_FileSystemDefaultEncoding, NULL);
1243 if (copy == NULL)
1244 return NULL;
1245 v = copy;
1246 }
1247 else
1248#endif
Guido van Rossum79f25d91997-04-29 20:08:16 +00001249 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001250 continue;
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001251 len = PyString_GET_SIZE(v);
Walter Dörwald3430d702002-06-17 10:43:59 +00001252 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
1253 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001254 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001255 }
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001256 strcpy(buf, PyString_AS_STRING(v));
Walter Dörwald3430d702002-06-17 10:43:59 +00001257 if (strlen(buf) != len) {
1258 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001259 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001260 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001261
1262 /* sys.path_hooks import hook */
1263 if (p_loader != NULL) {
1264 PyObject *importer;
1265
1266 importer = get_path_importer(path_importer_cache,
1267 path_hooks, v);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001268 if (importer == NULL) {
1269 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001270 return NULL;
Neal Norwitza4df11d2006-07-06 04:28:59 +00001271 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001272 /* Note: importer is a borrowed reference */
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001273 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001274 PyObject *loader;
1275 loader = PyObject_CallMethod(importer,
1276 "find_module",
1277 "s", fullname);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001278 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001279 if (loader == NULL)
1280 return NULL; /* error */
1281 if (loader != Py_None) {
1282 /* a loader was found */
1283 *p_loader = loader;
1284 return &importhookdescr;
1285 }
1286 Py_DECREF(loader);
Georg Brandlf4ef1162006-05-26 18:03:31 +00001287 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001288 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001289 }
Georg Brandlf4ef1162006-05-26 18:03:31 +00001290 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001291
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001292 if (len > 0 && buf[len-1] != SEP
1293#ifdef ALTSEP
1294 && buf[len-1] != ALTSEP
1295#endif
1296 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001297 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001298 strcpy(buf+len, name);
1299 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001300
1301 /* Check for package import (buf holds a directory name,
1302 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001303#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001304 if (stat(buf, &statbuf) == 0 && /* it exists */
1305 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001306 case_ok(buf, len, namelen, name)) { /* case matches */
1307 if (find_init_module(buf)) { /* and has __init__.py */
1308 Py_XDECREF(copy);
1309 return &fd_package;
1310 }
1311 else {
1312 char warnstr[MAXPATHLEN+80];
1313 sprintf(warnstr, "Not importing directory "
1314 "'%.*s': missing __init__.py",
1315 MAXPATHLEN, buf);
1316 if (PyErr_Warn(PyExc_ImportWarning,
1317 warnstr)) {
1318 Py_XDECREF(copy);
1319 return NULL;
1320 }
1321 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001322 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001323#else
1324 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001325#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001326 if (isdir(buf) &&
Walter Dörwald3430d702002-06-17 10:43:59 +00001327 case_ok(buf, len, namelen, name)) {
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001328 if (find_init_module(buf)) {
1329 Py_XDECREF(copy);
1330 return &fd_package;
1331 }
1332 else {
1333 char warnstr[MAXPATHLEN+80];
1334 sprintf(warnstr, "Not importing directory "
1335 "'%.*s': missing __init__.py",
1336 MAXPATHLEN, buf);
1337 if (PyErr_Warn(PyExc_ImportWarning,
1338 warnstr)) {
1339 Py_XDECREF(copy);
1340 return NULL;
1341 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001342 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001343#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001344#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001345#if defined(PYOS_OS2)
1346 /* take a snapshot of the module spec for restoration
1347 * after the 8 character DLL hackery
1348 */
1349 saved_buf = strdup(buf);
1350 saved_len = len;
1351 saved_namelen = namelen;
1352#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001353 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001354#if defined(PYOS_OS2)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001355 /* OS/2 limits DLLs to 8 character names (w/o
1356 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001357 * so if the name is longer than that and its a
1358 * dynamically loaded module we're going to try,
1359 * truncate the name before trying
1360 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001361 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001362 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001363 const struct filedescr *scan;
1364 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001365 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001366 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001367 break;
1368 else
1369 scan++;
1370 }
1371 if (scan->suffix != NULL) {
1372 /* yes, so truncate the name */
1373 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001374 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001375 buf[len] = '\0';
1376 }
1377 }
1378#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001379 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001380 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001381 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001382 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001383 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001384 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001385 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001386 if (fp != NULL) {
1387 if (case_ok(buf, len, namelen, name))
1388 break;
1389 else { /* continue search */
1390 fclose(fp);
1391 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001392 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001393 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001394#if defined(PYOS_OS2)
1395 /* restore the saved snapshot */
1396 strcpy(buf, saved_buf);
1397 len = saved_len;
1398 namelen = saved_namelen;
1399#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001400 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001401#if defined(PYOS_OS2)
1402 /* don't need/want the module name snapshot anymore */
1403 if (saved_buf)
1404 {
1405 free(saved_buf);
1406 saved_buf = NULL;
1407 }
1408#endif
Walter Dörwald3430d702002-06-17 10:43:59 +00001409 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001410 if (fp != NULL)
1411 break;
1412 }
1413 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001414 PyErr_Format(PyExc_ImportError,
1415 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001416 return NULL;
1417 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001418 *p_fp = fp;
1419 return fdp;
1420}
1421
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001422/* Helpers for main.c
1423 * Find the source file corresponding to a named module
1424 */
1425struct filedescr *
1426_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1427 size_t buflen, FILE **p_fp, PyObject **p_loader)
1428{
1429 return find_module((char *) name, (char *) name, path,
1430 buf, buflen, p_fp, p_loader);
1431}
1432
1433PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1434{
1435 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1436}
1437
Martin v. Löwis18e16552006-02-15 17:27:45 +00001438/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001439 * The arguments here are tricky, best shown by example:
1440 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1441 * ^ ^ ^ ^
1442 * |--------------------- buf ---------------------|
1443 * |------------------- len ------------------|
1444 * |------ name -------|
1445 * |----- namelen -----|
1446 * buf is the full path, but len only counts up to (& exclusive of) the
1447 * extension. name is the module name, also exclusive of extension.
1448 *
1449 * We've already done a successful stat() or fopen() on buf, so know that
1450 * there's some match, possibly case-insensitive.
1451 *
Tim Peters50d8d372001-02-28 05:34:27 +00001452 * case_ok() is to return 1 if there's a case-sensitive match for
1453 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1454 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001455 *
Tim Peters50d8d372001-02-28 05:34:27 +00001456 * case_ok() is used to implement case-sensitive import semantics even
1457 * on platforms with case-insensitive filesystems. It's trivial to implement
1458 * for case-sensitive filesystems. It's pretty much a cross-platform
1459 * nightmare for systems with case-insensitive filesystems.
1460 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001461
Tim Peters50d8d372001-02-28 05:34:27 +00001462/* First we may need a pile of platform-specific header files; the sequence
1463 * of #if's here should match the sequence in the body of case_ok().
1464 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001465#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001466#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001467
Tim Peters50d8d372001-02-28 05:34:27 +00001468#elif defined(DJGPP)
1469#include <dir.h>
1470
Jason Tishler7961aa62005-05-20 00:56:54 +00001471#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001472#include <sys/types.h>
1473#include <dirent.h>
1474
Andrew MacIntyred9400542002-02-26 11:41:34 +00001475#elif defined(PYOS_OS2)
1476#define INCL_DOS
1477#define INCL_DOSERRORS
1478#define INCL_NOPMAPI
1479#include <os2.h>
1480
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001481#elif defined(RISCOS)
1482#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001483#endif
1484
Guido van Rossum0980bd91998-02-13 17:18:36 +00001485static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001486case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001487{
Tim Peters50d8d372001-02-28 05:34:27 +00001488/* Pick a platform-specific implementation; the sequence of #if's here should
1489 * match the sequence just above.
1490 */
1491
Jason Tishler7961aa62005-05-20 00:56:54 +00001492/* MS_WINDOWS */
1493#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001494 WIN32_FIND_DATA data;
1495 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001496
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001497 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001498 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001499
Guido van Rossum0980bd91998-02-13 17:18:36 +00001500 h = FindFirstFile(buf, &data);
1501 if (h == INVALID_HANDLE_VALUE) {
1502 PyErr_Format(PyExc_NameError,
1503 "Can't find file for module %.100s\n(filename %.300s)",
1504 name, buf);
1505 return 0;
1506 }
1507 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001508 return strncmp(data.cFileName, name, namelen) == 0;
1509
1510/* DJGPP */
1511#elif defined(DJGPP)
1512 struct ffblk ffblk;
1513 int done;
1514
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001515 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001516 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001517
1518 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1519 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001520 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001521 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001522 name, buf);
1523 return 0;
1524 }
Tim Peters50d8d372001-02-28 05:34:27 +00001525 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001526
Jason Tishler7961aa62005-05-20 00:56:54 +00001527/* new-fangled macintosh (macosx) or Cygwin */
1528#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001529 DIR *dirp;
1530 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001531 char dirname[MAXPATHLEN + 1];
1532 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001533
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001534 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001535 return 1;
1536
Tim Petersd1e87a82001-03-01 18:12:00 +00001537 /* Copy the dir component into dirname; substitute "." if empty */
1538 if (dirlen <= 0) {
1539 dirname[0] = '.';
1540 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001541 }
1542 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001543 assert(dirlen <= MAXPATHLEN);
1544 memcpy(dirname, buf, dirlen);
1545 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001546 }
1547 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001548 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001549 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001550 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001551 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001552 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001553#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001554 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001555#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001556 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001557#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001558 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001559 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001560 (void)closedir(dirp);
1561 return 1; /* Found */
1562 }
1563 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001564 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001565 }
Tim Peters430f5d42001-03-01 01:30:56 +00001566 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001567
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001568/* RISC OS */
1569#elif defined(RISCOS)
1570 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1571 char buf2[MAXPATHLEN+2];
1572 char *nameWithExt = buf+len-namelen;
1573 int canonlen;
1574 os_error *e;
1575
1576 if (Py_GETENV("PYTHONCASEOK") != NULL)
1577 return 1;
1578
1579 /* workaround:
1580 append wildcard, otherwise case of filename wouldn't be touched */
1581 strcpy(buf2, buf);
1582 strcat(buf2, "*");
1583
1584 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1585 canonlen = MAXPATHLEN+1-canonlen;
1586 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1587 return 0;
1588 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1589 return 1; /* match */
1590
1591 return 0;
1592
Andrew MacIntyred9400542002-02-26 11:41:34 +00001593/* OS/2 */
1594#elif defined(PYOS_OS2)
1595 HDIR hdir = 1;
1596 ULONG srchcnt = 1;
1597 FILEFINDBUF3 ffbuf;
1598 APIRET rc;
1599
1600 if (getenv("PYTHONCASEOK") != NULL)
1601 return 1;
1602
1603 rc = DosFindFirst(buf,
1604 &hdir,
1605 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1606 &ffbuf, sizeof(ffbuf),
1607 &srchcnt,
1608 FIL_STANDARD);
1609 if (rc != NO_ERROR)
1610 return 0;
1611 return strncmp(ffbuf.achName, name, namelen) == 0;
1612
Tim Peters50d8d372001-02-28 05:34:27 +00001613/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1614#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001615 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001616
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001617#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001618}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001619
Guido van Rossum0980bd91998-02-13 17:18:36 +00001620
Guido van Rossum197346f1997-10-31 18:38:52 +00001621#ifdef HAVE_STAT
1622/* Helper to look for __init__.py or __init__.py[co] in potential package */
1623static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001624find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001625{
Tim Peters0f9431f2001-07-05 03:47:53 +00001626 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001627 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001628 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001629 struct stat statbuf;
1630
Tim Peters0f9431f2001-07-05 03:47:53 +00001631/* For calling case_ok(buf, len, namelen, name):
1632 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1633 * ^ ^ ^ ^
1634 * |--------------------- buf ---------------------|
1635 * |------------------- len ------------------|
1636 * |------ name -------|
1637 * |----- namelen -----|
1638 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001639 if (save_len + 13 >= MAXPATHLEN)
1640 return 0;
1641 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001642 pname = buf + i;
1643 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001644 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001645 if (case_ok(buf,
1646 save_len + 9, /* len("/__init__") */
1647 8, /* len("__init__") */
1648 pname)) {
1649 buf[save_len] = '\0';
1650 return 1;
1651 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001652 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001653 i += strlen(pname);
1654 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001655 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001656 if (case_ok(buf,
1657 save_len + 9, /* len("/__init__") */
1658 8, /* len("__init__") */
1659 pname)) {
1660 buf[save_len] = '\0';
1661 return 1;
1662 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001663 }
1664 buf[save_len] = '\0';
1665 return 0;
1666}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001667
1668#else
1669
1670#ifdef RISCOS
1671static int
1672find_init_module(buf)
1673 char *buf;
1674{
1675 int save_len = strlen(buf);
1676 int i = save_len;
1677
1678 if (save_len + 13 >= MAXPATHLEN)
1679 return 0;
1680 buf[i++] = SEP;
1681 strcpy(buf+i, "__init__/py");
1682 if (isfile(buf)) {
1683 buf[save_len] = '\0';
1684 return 1;
1685 }
1686
1687 if (Py_OptimizeFlag)
1688 strcpy(buf+i, "o");
1689 else
1690 strcpy(buf+i, "c");
1691 if (isfile(buf)) {
1692 buf[save_len] = '\0';
1693 return 1;
1694 }
1695 buf[save_len] = '\0';
1696 return 0;
1697}
1698#endif /*RISCOS*/
1699
Guido van Rossum197346f1997-10-31 18:38:52 +00001700#endif /* HAVE_STAT */
1701
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001702
Tim Petersdbd9ba62000-07-09 03:09:57 +00001703static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001704
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001705/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001706 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001707
Guido van Rossum79f25d91997-04-29 20:08:16 +00001708static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001709load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001710{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001711 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001712 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001713 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001714
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001715 /* First check that there's an open file (if we need one) */
1716 switch (type) {
1717 case PY_SOURCE:
1718 case PY_COMPILED:
1719 if (fp == NULL) {
1720 PyErr_Format(PyExc_ValueError,
1721 "file object required for import (type code %d)",
1722 type);
1723 return NULL;
1724 }
1725 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001726
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001727 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001728
1729 case PY_SOURCE:
1730 m = load_source_module(name, buf, fp);
1731 break;
1732
1733 case PY_COMPILED:
1734 m = load_compiled_module(name, buf, fp);
1735 break;
1736
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001737#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001738 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001739 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001740 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001741#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001742
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001743 case PKG_DIRECTORY:
1744 m = load_package(name, buf);
1745 break;
1746
1747 case C_BUILTIN:
1748 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001749 if (buf != NULL && buf[0] != '\0')
1750 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001751 if (type == C_BUILTIN)
1752 err = init_builtin(name);
1753 else
1754 err = PyImport_ImportFrozenModule(name);
1755 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001756 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001757 if (err == 0) {
1758 PyErr_Format(PyExc_ImportError,
1759 "Purported %s module %.200s not found",
1760 type == C_BUILTIN ?
1761 "builtin" : "frozen",
1762 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001763 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001764 }
1765 modules = PyImport_GetModuleDict();
1766 m = PyDict_GetItemString(modules, name);
1767 if (m == NULL) {
1768 PyErr_Format(
1769 PyExc_ImportError,
1770 "%s module %.200s not properly initialized",
1771 type == C_BUILTIN ?
1772 "builtin" : "frozen",
1773 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001774 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001775 }
1776 Py_INCREF(m);
1777 break;
1778
Just van Rossum52e14d62002-12-30 22:08:05 +00001779 case IMP_HOOK: {
1780 if (loader == NULL) {
1781 PyErr_SetString(PyExc_ImportError,
1782 "import hook without loader");
1783 return NULL;
1784 }
1785 m = PyObject_CallMethod(loader, "load_module", "s", name);
1786 break;
1787 }
1788
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001789 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001790 PyErr_Format(PyExc_ImportError,
1791 "Don't know how to import %.200s (type code %d)",
1792 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001793 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001794
1795 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001796
1797 return m;
1798}
1799
1800
1801/* Initialize a built-in module.
1802 Return 1 for succes, 0 if the module is not found, and -1 with
1803 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001804
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001805static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001806init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001807{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001808 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001809
Greg Ward201baee2001-10-04 14:52:06 +00001810 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001811 return 1;
1812
Guido van Rossum771c6c81997-10-31 18:37:24 +00001813 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001814 if (strcmp(name, p->name) == 0) {
1815 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001816 PyErr_Format(PyExc_ImportError,
1817 "Cannot re-init internal module %.200s",
1818 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001819 return -1;
1820 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001821 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001822 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001823 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001824 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001825 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001826 if (_PyImport_FixupExtension(name, name) == NULL)
1827 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001828 return 1;
1829 }
1830 }
1831 return 0;
1832}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001833
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001834
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001835/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001836
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001837static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001838find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001839{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001840 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001841
Guido van Rossum79f25d91997-04-29 20:08:16 +00001842 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001843 if (p->name == NULL)
1844 return NULL;
1845 if (strcmp(p->name, name) == 0)
1846 break;
1847 }
1848 return p;
1849}
1850
Guido van Rossum79f25d91997-04-29 20:08:16 +00001851static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001852get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001853{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001854 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001855 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001856
1857 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001858 PyErr_Format(PyExc_ImportError,
1859 "No such frozen object named %.200s",
1860 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001861 return NULL;
1862 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001863 if (p->code == NULL) {
1864 PyErr_Format(PyExc_ImportError,
1865 "Excluded frozen object named %.200s",
1866 name);
1867 return NULL;
1868 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001869 size = p->size;
1870 if (size < 0)
1871 size = -size;
1872 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001873}
1874
1875/* Initialize a frozen module.
1876 Return 1 for succes, 0 if the module is not found, and -1 with
1877 an exception set if the initialization failed.
1878 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001879
1880int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001881PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001882{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001883 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001884 PyObject *co;
1885 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001886 int ispackage;
1887 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001888
1889 if (p == NULL)
1890 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001891 if (p->code == NULL) {
1892 PyErr_Format(PyExc_ImportError,
1893 "Excluded frozen object named %.200s",
1894 name);
1895 return -1;
1896 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001897 size = p->size;
1898 ispackage = (size < 0);
1899 if (ispackage)
1900 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001901 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001902 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001903 name, ispackage ? " package" : "");
1904 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001905 if (co == NULL)
1906 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001907 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001908 PyErr_Format(PyExc_TypeError,
1909 "frozen object %.200s is not a code object",
1910 name);
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001911 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001912 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001913 if (ispackage) {
1914 /* Set __path__ to the package name */
1915 PyObject *d, *s;
1916 int err;
1917 m = PyImport_AddModule(name);
1918 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001919 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001920 d = PyModule_GetDict(m);
1921 s = PyString_InternFromString(name);
1922 if (s == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001923 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001924 err = PyDict_SetItemString(d, "__path__", s);
1925 Py_DECREF(s);
1926 if (err != 0)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001927 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001928 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001929 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001930 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001931 goto err_return;
1932 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001933 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001934 return 1;
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001935err_return:
1936 Py_DECREF(co);
1937 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001938}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001939
1940
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001941/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001942 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001943
Guido van Rossum79f25d91997-04-29 20:08:16 +00001944PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001945PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001946{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001947 PyObject *pname;
1948 PyObject *result;
1949
1950 pname = PyString_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001951 if (pname == NULL)
1952 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001953 result = PyImport_Import(pname);
1954 Py_DECREF(pname);
1955 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001956}
1957
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001958/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001959static PyObject *get_parent(PyObject *globals, char *buf,
1960 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001961static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001962 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001963static int mark_miss(char *name);
1964static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001965 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001966static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001967
1968/* The Magnum Opus of dotted-name import :-) */
1969
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001970static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001971import_module_level(char *name, PyObject *globals, PyObject *locals,
1972 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001973{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001974 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00001975 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001976 PyObject *parent, *head, *next, *tail;
1977
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001978 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001979 if (parent == NULL)
1980 return NULL;
1981
1982 head = load_next(parent, Py_None, &name, buf, &buflen);
1983 if (head == NULL)
1984 return NULL;
1985
1986 tail = head;
1987 Py_INCREF(tail);
1988 while (name) {
1989 next = load_next(tail, tail, &name, buf, &buflen);
1990 Py_DECREF(tail);
1991 if (next == NULL) {
1992 Py_DECREF(head);
1993 return NULL;
1994 }
1995 tail = next;
1996 }
Thomas Wouters8ddab272006-04-04 16:17:02 +00001997 if (tail == Py_None) {
1998 /* If tail is Py_None, both get_parent and load_next found
1999 an empty module name: someone called __import__("") or
2000 doctored faulty bytecode */
Thomas Wouters4bdaa272006-04-05 13:39:37 +00002001 Py_DECREF(tail);
2002 Py_DECREF(head);
Thomas Wouters8ddab272006-04-04 16:17:02 +00002003 PyErr_SetString(PyExc_ValueError,
2004 "Empty module name");
2005 return NULL;
2006 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002007
2008 if (fromlist != NULL) {
2009 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2010 fromlist = NULL;
2011 }
2012
2013 if (fromlist == NULL) {
2014 Py_DECREF(tail);
2015 return head;
2016 }
2017
2018 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002019 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002020 Py_DECREF(tail);
2021 return NULL;
2022 }
2023
2024 return tail;
2025}
2026
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002027/* For DLL compatibility */
2028#undef PyImport_ImportModuleEx
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002029PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002030PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
2031 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002032{
2033 PyObject *result;
2034 lock_import();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002035 result = import_module_level(name, globals, locals, fromlist, -1);
2036 if (unlock_import() < 0) {
2037 Py_XDECREF(result);
2038 PyErr_SetString(PyExc_RuntimeError,
2039 "not holding the import lock");
2040 return NULL;
2041 }
2042 return result;
2043}
2044#define PyImport_ImportModuleEx(n, g, l, f) \
2045 PyImport_ImportModuleLevel(n, g, l, f, -1);
2046
2047PyObject *
2048PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
2049 PyObject *fromlist, int level)
2050{
2051 PyObject *result;
2052 lock_import();
2053 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002054 if (unlock_import() < 0) {
2055 Py_XDECREF(result);
2056 PyErr_SetString(PyExc_RuntimeError,
2057 "not holding the import lock");
2058 return NULL;
2059 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002060 return result;
2061}
2062
Fred Drake87590902004-05-28 20:21:36 +00002063/* Return the package that an import is being performed in. If globals comes
2064 from the module foo.bar.bat (not itself a package), this returns the
2065 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Georg Brandl5f6861d2006-05-28 21:57:35 +00002066 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002067
2068 The *name* of the returned package is returned in buf, with the length of
2069 the name in *p_buflen.
2070
2071 If globals doesn't come from a package or a module in a package, or a
2072 corresponding entry is not found in sys.modules, Py_None is returned.
2073*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002074static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002075get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002076{
2077 static PyObject *namestr = NULL;
2078 static PyObject *pathstr = NULL;
2079 PyObject *modname, *modpath, *modules, *parent;
2080
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002081 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002082 return Py_None;
2083
2084 if (namestr == NULL) {
2085 namestr = PyString_InternFromString("__name__");
2086 if (namestr == NULL)
2087 return NULL;
2088 }
2089 if (pathstr == NULL) {
2090 pathstr = PyString_InternFromString("__path__");
2091 if (pathstr == NULL)
2092 return NULL;
2093 }
2094
2095 *buf = '\0';
2096 *p_buflen = 0;
2097 modname = PyDict_GetItem(globals, namestr);
2098 if (modname == NULL || !PyString_Check(modname))
2099 return Py_None;
2100
2101 modpath = PyDict_GetItem(globals, pathstr);
2102 if (modpath != NULL) {
Martin v. Löwis725507b2006-03-07 12:08:51 +00002103 Py_ssize_t len = PyString_GET_SIZE(modname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002104 if (len > MAXPATHLEN) {
2105 PyErr_SetString(PyExc_ValueError,
2106 "Module name too long");
2107 return NULL;
2108 }
2109 strcpy(buf, PyString_AS_STRING(modname));
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002110 }
2111 else {
2112 char *start = PyString_AS_STRING(modname);
2113 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002114 size_t len;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002115 if (lastdot == NULL && level > 0) {
2116 PyErr_SetString(PyExc_ValueError,
Georg Brandl37a9e572006-09-06 06:09:34 +00002117 "Attempted relative import in non-package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002118 return NULL;
2119 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002120 if (lastdot == NULL)
2121 return Py_None;
2122 len = lastdot - start;
2123 if (len >= MAXPATHLEN) {
2124 PyErr_SetString(PyExc_ValueError,
2125 "Module name too long");
2126 return NULL;
2127 }
2128 strncpy(buf, start, len);
2129 buf[len] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002130 }
2131
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002132 while (--level > 0) {
2133 char *dot = strrchr(buf, '.');
2134 if (dot == NULL) {
2135 PyErr_SetString(PyExc_ValueError,
Georg Brandl37a9e572006-09-06 06:09:34 +00002136 "Attempted relative import beyond "
2137 "toplevel package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002138 return NULL;
2139 }
2140 *dot = '\0';
2141 }
2142 *p_buflen = strlen(buf);
2143
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002144 modules = PyImport_GetModuleDict();
2145 parent = PyDict_GetItemString(modules, buf);
2146 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002147 PyErr_Format(PyExc_SystemError,
2148 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002149 return parent;
2150 /* We expect, but can't guarantee, if parent != None, that:
2151 - parent.__name__ == buf
2152 - parent.__dict__ is globals
2153 If this is violated... Who cares? */
2154}
2155
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002156/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002157static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002158load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002159 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002160{
2161 char *name = *p_name;
2162 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002163 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002164 char *p;
2165 PyObject *result;
2166
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002167 if (strlen(name) == 0) {
Thomas Wouters8ddab272006-04-04 16:17:02 +00002168 /* completely empty module name should only happen in
2169 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002170 Py_INCREF(mod);
2171 *p_name = NULL;
2172 return mod;
2173 }
2174
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002175 if (dot == NULL) {
2176 *p_name = NULL;
2177 len = strlen(name);
2178 }
2179 else {
2180 *p_name = dot+1;
2181 len = dot-name;
2182 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002183 if (len == 0) {
2184 PyErr_SetString(PyExc_ValueError,
2185 "Empty module name");
2186 return NULL;
2187 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002188
2189 p = buf + *p_buflen;
2190 if (p != buf)
2191 *p++ = '.';
2192 if (p+len-buf >= MAXPATHLEN) {
2193 PyErr_SetString(PyExc_ValueError,
2194 "Module name too long");
2195 return NULL;
2196 }
2197 strncpy(p, name, len);
2198 p[len] = '\0';
2199 *p_buflen = p+len-buf;
2200
2201 result = import_submodule(mod, p, buf);
2202 if (result == Py_None && altmod != mod) {
2203 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002204 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002205 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002206 if (result != NULL && result != Py_None) {
2207 if (mark_miss(buf) != 0) {
2208 Py_DECREF(result);
2209 return NULL;
2210 }
2211 strncpy(buf, name, len);
2212 buf[len] = '\0';
2213 *p_buflen = len;
2214 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002215 }
2216 if (result == NULL)
2217 return NULL;
2218
2219 if (result == Py_None) {
2220 Py_DECREF(result);
2221 PyErr_Format(PyExc_ImportError,
2222 "No module named %.200s", name);
2223 return NULL;
2224 }
2225
2226 return result;
2227}
2228
2229static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002230mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002231{
2232 PyObject *modules = PyImport_GetModuleDict();
2233 return PyDict_SetItemString(modules, name, Py_None);
2234}
2235
2236static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002237ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002238 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002239{
2240 int i;
2241
2242 if (!PyObject_HasAttrString(mod, "__path__"))
2243 return 1;
2244
2245 for (i = 0; ; i++) {
2246 PyObject *item = PySequence_GetItem(fromlist, i);
2247 int hasit;
2248 if (item == NULL) {
2249 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2250 PyErr_Clear();
2251 return 1;
2252 }
2253 return 0;
2254 }
2255 if (!PyString_Check(item)) {
2256 PyErr_SetString(PyExc_TypeError,
2257 "Item in ``from list'' not a string");
2258 Py_DECREF(item);
2259 return 0;
2260 }
2261 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002262 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002263 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002264 /* See if the package defines __all__ */
2265 if (recursive)
2266 continue; /* Avoid endless recursion */
2267 all = PyObject_GetAttrString(mod, "__all__");
2268 if (all == NULL)
2269 PyErr_Clear();
2270 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002271 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002272 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002273 if (!ret)
2274 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002275 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002276 continue;
2277 }
2278 hasit = PyObject_HasAttr(mod, item);
2279 if (!hasit) {
2280 char *subname = PyString_AS_STRING(item);
2281 PyObject *submod;
2282 char *p;
2283 if (buflen + strlen(subname) >= MAXPATHLEN) {
2284 PyErr_SetString(PyExc_ValueError,
2285 "Module name too long");
2286 Py_DECREF(item);
2287 return 0;
2288 }
2289 p = buf + buflen;
2290 *p++ = '.';
2291 strcpy(p, subname);
2292 submod = import_submodule(mod, subname, buf);
2293 Py_XDECREF(submod);
2294 if (submod == NULL) {
2295 Py_DECREF(item);
2296 return 0;
2297 }
2298 }
2299 Py_DECREF(item);
2300 }
2301
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002302 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002303}
2304
Neil Schemenauer00b09662003-06-16 21:03:07 +00002305static int
2306add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2307 PyObject *modules)
2308{
2309 if (mod == Py_None)
2310 return 1;
2311 /* Irrespective of the success of this load, make a
2312 reference to it in the parent package module. A copy gets
2313 saved in the modules dictionary under the full name, so get a
2314 reference from there, if need be. (The exception is when the
2315 load failed with a SyntaxError -- then there's no trace in
2316 sys.modules. In that case, of course, do nothing extra.) */
2317 if (submod == NULL) {
2318 submod = PyDict_GetItemString(modules, fullname);
2319 if (submod == NULL)
2320 return 1;
2321 }
2322 if (PyModule_Check(mod)) {
2323 /* We can't use setattr here since it can give a
2324 * spurious warning if the submodule name shadows a
2325 * builtin name */
2326 PyObject *dict = PyModule_GetDict(mod);
2327 if (!dict)
2328 return 0;
2329 if (PyDict_SetItemString(dict, subname, submod) < 0)
2330 return 0;
2331 }
2332 else {
2333 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2334 return 0;
2335 }
2336 return 1;
2337}
2338
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002339static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002340import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002341{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002342 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002343 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002344
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002345 /* Require:
2346 if mod == None: subname == fullname
2347 else: mod.__name__ + "." + subname == fullname
2348 */
2349
Tim Peters50d8d372001-02-28 05:34:27 +00002350 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002351 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002352 }
2353 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002354 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002355 char buf[MAXPATHLEN+1];
2356 struct filedescr *fdp;
2357 FILE *fp = NULL;
2358
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002359 if (mod == Py_None)
2360 path = NULL;
2361 else {
2362 path = PyObject_GetAttrString(mod, "__path__");
2363 if (path == NULL) {
2364 PyErr_Clear();
2365 Py_INCREF(Py_None);
2366 return Py_None;
2367 }
2368 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002369
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002370 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002371 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2372 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002373 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002374 if (fdp == NULL) {
2375 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2376 return NULL;
2377 PyErr_Clear();
2378 Py_INCREF(Py_None);
2379 return Py_None;
2380 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002381 m = load_module(fullname, fp, buf, fdp->type, loader);
2382 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002383 if (fp)
2384 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002385 if (!add_submodule(mod, m, fullname, subname, modules)) {
2386 Py_XDECREF(m);
2387 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002388 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002389 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002390
2391 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002392}
2393
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002394
2395/* Re-import a module of any kind and return its module object, WITH
2396 INCREMENTED REFERENCE COUNT */
2397
Guido van Rossum79f25d91997-04-29 20:08:16 +00002398PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002399PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002400{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002401 PyObject *modules = PyImport_GetModuleDict();
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002402 PyObject *path = NULL, *loader = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002403 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002404 char buf[MAXPATHLEN+1];
2405 struct filedescr *fdp;
2406 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002407 PyObject *newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002408
Guido van Rossum79f25d91997-04-29 20:08:16 +00002409 if (m == NULL || !PyModule_Check(m)) {
2410 PyErr_SetString(PyExc_TypeError,
2411 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002412 return NULL;
2413 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002414 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002415 if (name == NULL)
2416 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002417 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002418 PyErr_Format(PyExc_ImportError,
2419 "reload(): module %.200s not in sys.modules",
2420 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002421 return NULL;
2422 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002423 subname = strrchr(name, '.');
2424 if (subname == NULL)
2425 subname = name;
2426 else {
2427 PyObject *parentname, *parent;
2428 parentname = PyString_FromStringAndSize(name, (subname-name));
2429 if (parentname == NULL)
2430 return NULL;
2431 parent = PyDict_GetItem(modules, parentname);
2432 if (parent == NULL) {
2433 PyErr_Format(PyExc_ImportError,
2434 "reload(): parent %.200s not in sys.modules",
Georg Brandl0c55f292005-09-14 06:56:20 +00002435 PyString_AS_STRING(parentname));
2436 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002437 return NULL;
2438 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002439 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002440 subname++;
2441 path = PyObject_GetAttrString(parent, "__path__");
2442 if (path == NULL)
2443 PyErr_Clear();
2444 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002445 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002446 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002447 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002448
2449 if (fdp == NULL) {
2450 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002451 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002452 }
2453
2454 newm = load_module(name, fp, buf, fdp->type, loader);
2455 Py_XDECREF(loader);
2456
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002457 if (fp)
2458 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002459 if (newm == NULL) {
2460 /* load_module probably removed name from modules because of
2461 * the error. Put back the original module object. We're
2462 * going to return NULL in this case regardless of whether
2463 * replacing name succeeds, so the return value is ignored.
2464 */
2465 PyDict_SetItemString(modules, name, m);
2466 }
2467 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002468}
2469
2470
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002471/* Higher-level import emulator which emulates the "import" statement
2472 more accurately -- it invokes the __import__() function from the
2473 builtins of the current globals. This means that the import is
2474 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002475 environment, e.g. by "rexec".
2476 A dummy list ["__doc__"] is passed as the 4th argument so that
2477 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2478 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002479
2480PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002481PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002482{
2483 static PyObject *silly_list = NULL;
2484 static PyObject *builtins_str = NULL;
2485 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002486 PyObject *globals = NULL;
2487 PyObject *import = NULL;
2488 PyObject *builtins = NULL;
2489 PyObject *r = NULL;
2490
2491 /* Initialize constant string objects */
2492 if (silly_list == NULL) {
2493 import_str = PyString_InternFromString("__import__");
2494 if (import_str == NULL)
2495 return NULL;
2496 builtins_str = PyString_InternFromString("__builtins__");
2497 if (builtins_str == NULL)
2498 return NULL;
2499 silly_list = Py_BuildValue("[s]", "__doc__");
2500 if (silly_list == NULL)
2501 return NULL;
2502 }
2503
2504 /* Get the builtins from current globals */
2505 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002506 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002507 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002508 builtins = PyObject_GetItem(globals, builtins_str);
2509 if (builtins == NULL)
2510 goto err;
2511 }
2512 else {
2513 /* No globals -- use standard builtins, and fake globals */
2514 PyErr_Clear();
2515
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002516 builtins = PyImport_ImportModuleLevel("__builtin__",
2517 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002518 if (builtins == NULL)
2519 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002520 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2521 if (globals == NULL)
2522 goto err;
2523 }
2524
2525 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002526 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002527 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002528 if (import == NULL)
2529 PyErr_SetObject(PyExc_KeyError, import_str);
2530 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002531 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002532 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002533 if (import == NULL)
2534 goto err;
2535
2536 /* Call the _import__ function with the proper argument list */
Georg Brandl684fd0c2006-05-25 19:15:31 +00002537 r = PyObject_CallFunctionObjArgs(import, module_name, globals,
2538 globals, silly_list, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002539
2540 err:
2541 Py_XDECREF(globals);
2542 Py_XDECREF(builtins);
2543 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002544
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002545 return r;
2546}
2547
2548
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002549/* Module 'imp' provides Python access to the primitives used for
2550 importing modules.
2551*/
2552
Guido van Rossum79f25d91997-04-29 20:08:16 +00002553static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002554imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002555{
2556 char buf[4];
2557
Guido van Rossum96774c12000-05-01 20:19:08 +00002558 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2559 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2560 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2561 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002562
Guido van Rossum79f25d91997-04-29 20:08:16 +00002563 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002564}
2565
Guido van Rossum79f25d91997-04-29 20:08:16 +00002566static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002567imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002568{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002569 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002570 struct filedescr *fdp;
2571
Guido van Rossum79f25d91997-04-29 20:08:16 +00002572 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002573 if (list == NULL)
2574 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002575 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2576 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002577 fdp->suffix, fdp->mode, fdp->type);
2578 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002579 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002580 return NULL;
2581 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002582 if (PyList_Append(list, item) < 0) {
2583 Py_DECREF(list);
2584 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002585 return NULL;
2586 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002587 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002588 }
2589 return list;
2590}
2591
Guido van Rossum79f25d91997-04-29 20:08:16 +00002592static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002593call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002594{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002595 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002596 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002597 struct filedescr *fdp;
2598 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002599 FILE *fp = NULL;
2600
2601 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002602 if (path == Py_None)
2603 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002604 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002605 if (fdp == NULL)
2606 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002607 if (fp != NULL) {
2608 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2609 if (fob == NULL) {
2610 fclose(fp);
2611 return NULL;
2612 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002613 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002614 else {
2615 fob = Py_None;
2616 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002617 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002618 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002619 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002620 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002621 return ret;
2622}
2623
Guido van Rossum79f25d91997-04-29 20:08:16 +00002624static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002625imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002626{
2627 char *name;
2628 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002629 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002630 return NULL;
2631 return call_find_module(name, path);
2632}
2633
2634static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002635imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002636{
2637 char *name;
2638 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002639 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002640 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002641 return NULL;
2642 ret = init_builtin(name);
2643 if (ret < 0)
2644 return NULL;
2645 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002646 Py_INCREF(Py_None);
2647 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002648 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002649 m = PyImport_AddModule(name);
2650 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002651 return m;
2652}
2653
Guido van Rossum79f25d91997-04-29 20:08:16 +00002654static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002655imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002656{
2657 char *name;
2658 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002659 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002660 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002661 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002662 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002663 if (ret < 0)
2664 return NULL;
2665 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002666 Py_INCREF(Py_None);
2667 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002668 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002669 m = PyImport_AddModule(name);
2670 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002671 return m;
2672}
2673
Guido van Rossum79f25d91997-04-29 20:08:16 +00002674static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002675imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002676{
2677 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002678
Guido van Rossum43713e52000-02-29 13:59:29 +00002679 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002680 return NULL;
2681 return get_frozen_object(name);
2682}
2683
Guido van Rossum79f25d91997-04-29 20:08:16 +00002684static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002685imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002686{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002687 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002688 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002689 return NULL;
Guido van Rossum50ee94f2002-04-09 18:00:58 +00002690 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002691}
2692
Guido van Rossum79f25d91997-04-29 20:08:16 +00002693static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002694imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002695{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002696 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002697 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002698 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002699 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002700 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002701 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002702}
2703
2704static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002705get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002706{
2707 FILE *fp;
2708 if (fob == NULL) {
Tim Peters86c7d2f2004-08-01 23:24:21 +00002709 if (mode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002710 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002711 fp = fopen(pathname, mode);
2712 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002713 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002714 }
2715 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002716 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002717 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002718 PyErr_SetString(PyExc_ValueError,
2719 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002720 }
2721 return fp;
2722}
2723
Guido van Rossum79f25d91997-04-29 20:08:16 +00002724static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002725imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002726{
2727 char *name;
2728 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002729 PyObject *fob = NULL;
2730 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002731 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002732 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002733 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002734 return NULL;
2735 fp = get_file(pathname, fob, "rb");
2736 if (fp == NULL)
2737 return NULL;
2738 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002739 if (fob == NULL)
2740 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002741 return m;
2742}
2743
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002744#ifdef HAVE_DYNAMIC_LOADING
2745
Guido van Rossum79f25d91997-04-29 20:08:16 +00002746static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002747imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002748{
2749 char *name;
2750 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002751 PyObject *fob = NULL;
2752 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002753 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002754 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002755 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002756 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002757 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002758 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002759 if (fp == NULL)
2760 return NULL;
2761 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002762 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002763 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002764}
2765
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002766#endif /* HAVE_DYNAMIC_LOADING */
2767
Guido van Rossum79f25d91997-04-29 20:08:16 +00002768static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002769imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002770{
2771 char *name;
2772 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002773 PyObject *fob = NULL;
2774 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002775 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002776 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002777 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002778 return NULL;
2779 fp = get_file(pathname, fob, "r");
2780 if (fp == NULL)
2781 return NULL;
2782 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002783 if (fob == NULL)
2784 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002785 return m;
2786}
2787
Guido van Rossum79f25d91997-04-29 20:08:16 +00002788static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002789imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002790{
2791 char *name;
2792 PyObject *fob;
2793 char *pathname;
2794 char *suffix; /* Unused */
2795 char *mode;
2796 int type;
2797 FILE *fp;
2798
Guido van Rossum43713e52000-02-29 13:59:29 +00002799 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002800 &name, &fob, &pathname,
2801 &suffix, &mode, &type))
2802 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002803 if (*mode) {
2804 /* Mode must start with 'r' or 'U' and must not contain '+'.
2805 Implicit in this test is the assumption that the mode
2806 may contain other modifiers like 'b' or 't'. */
2807
2808 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002809 PyErr_Format(PyExc_ValueError,
2810 "invalid file open mode %.200s", mode);
2811 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002812 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002813 }
2814 if (fob == Py_None)
2815 fp = NULL;
2816 else {
2817 if (!PyFile_Check(fob)) {
2818 PyErr_SetString(PyExc_ValueError,
2819 "load_module arg#2 should be a file or None");
2820 return NULL;
2821 }
2822 fp = get_file(pathname, fob, mode);
2823 if (fp == NULL)
2824 return NULL;
2825 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002826 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002827}
2828
2829static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002830imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002831{
2832 char *name;
2833 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002834 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002835 return NULL;
2836 return load_package(name, pathname);
2837}
2838
2839static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002840imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002841{
2842 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002843 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002844 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002845 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002846}
2847
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002848/* Doc strings */
2849
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002850PyDoc_STRVAR(doc_imp,
2851"This module provides the components needed to build your own\n\
2852__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002853
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002854PyDoc_STRVAR(doc_find_module,
2855"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002856Search for a module. If path is omitted or None, search for a\n\
2857built-in, frozen or special module and continue search in sys.path.\n\
2858The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002859package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002860
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002861PyDoc_STRVAR(doc_load_module,
2862"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002863Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002864The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002865
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002866PyDoc_STRVAR(doc_get_magic,
2867"get_magic() -> string\n\
2868Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002869
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002870PyDoc_STRVAR(doc_get_suffixes,
2871"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002872Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002873that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002874
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002875PyDoc_STRVAR(doc_new_module,
2876"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002877Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002878The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002879
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002880PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00002881"lock_held() -> boolean\n\
2882Return True if the import lock is currently held, else False.\n\
2883On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00002884
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002885PyDoc_STRVAR(doc_acquire_lock,
2886"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00002887Acquires the interpreter's import lock for the current thread.\n\
2888This lock should be used by import hooks to ensure thread-safety\n\
2889when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002890On platforms without threads, this function does nothing.");
2891
2892PyDoc_STRVAR(doc_release_lock,
2893"release_lock() -> None\n\
2894Release the interpreter's import lock.\n\
2895On platforms without threads, this function does nothing.");
2896
Guido van Rossum79f25d91997-04-29 20:08:16 +00002897static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002898 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
2899 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
2900 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
2901 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
2902 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
2903 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
2904 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
2905 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002906 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00002907 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
2908 {"init_builtin", imp_init_builtin, METH_VARARGS},
2909 {"init_frozen", imp_init_frozen, METH_VARARGS},
2910 {"is_builtin", imp_is_builtin, METH_VARARGS},
2911 {"is_frozen", imp_is_frozen, METH_VARARGS},
2912 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002913#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00002914 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002915#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002916 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00002917 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002918 {NULL, NULL} /* sentinel */
2919};
2920
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002921static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002922setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002923{
2924 PyObject *v;
2925 int err;
2926
2927 v = PyInt_FromLong((long)value);
2928 err = PyDict_SetItemString(d, name, v);
2929 Py_XDECREF(v);
2930 return err;
2931}
2932
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00002933typedef struct {
2934 PyObject_HEAD
2935} NullImporter;
2936
2937static int
2938NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
2939{
2940 char *path;
2941
2942 if (!_PyArg_NoKeywords("NullImporter()", kwds))
2943 return -1;
2944
2945 if (!PyArg_ParseTuple(args, "s:NullImporter",
2946 &path))
2947 return -1;
2948
2949 if (strlen(path) == 0) {
2950 PyErr_SetString(PyExc_ImportError, "empty pathname");
2951 return -1;
2952 } else {
2953#ifndef RISCOS
2954 struct stat statbuf;
2955 int rv;
2956
2957 rv = stat(path, &statbuf);
2958 if (rv == 0) {
2959 /* it exists */
2960 if (S_ISDIR(statbuf.st_mode)) {
2961 /* it's a directory */
2962 PyErr_SetString(PyExc_ImportError,
2963 "existing directory");
2964 return -1;
2965 }
2966 }
2967#else
2968 if (object_exists(path)) {
2969 /* it exists */
2970 if (isdir(path)) {
2971 /* it's a directory */
2972 PyErr_SetString(PyExc_ImportError,
2973 "existing directory");
2974 return -1;
2975 }
2976 }
2977#endif
2978 }
2979 return 0;
2980}
2981
2982static PyObject *
2983NullImporter_find_module(NullImporter *self, PyObject *args)
2984{
2985 Py_RETURN_NONE;
2986}
2987
2988static PyMethodDef NullImporter_methods[] = {
2989 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
2990 "Always return None"
2991 },
2992 {NULL} /* Sentinel */
2993};
2994
2995
2996static PyTypeObject NullImporterType = {
2997 PyObject_HEAD_INIT(NULL)
2998 0, /*ob_size*/
2999 "imp.NullImporter", /*tp_name*/
3000 sizeof(NullImporter), /*tp_basicsize*/
3001 0, /*tp_itemsize*/
3002 0, /*tp_dealloc*/
3003 0, /*tp_print*/
3004 0, /*tp_getattr*/
3005 0, /*tp_setattr*/
3006 0, /*tp_compare*/
3007 0, /*tp_repr*/
3008 0, /*tp_as_number*/
3009 0, /*tp_as_sequence*/
3010 0, /*tp_as_mapping*/
3011 0, /*tp_hash */
3012 0, /*tp_call*/
3013 0, /*tp_str*/
3014 0, /*tp_getattro*/
3015 0, /*tp_setattro*/
3016 0, /*tp_as_buffer*/
3017 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3018 "Null importer object", /* tp_doc */
3019 0, /* tp_traverse */
3020 0, /* tp_clear */
3021 0, /* tp_richcompare */
3022 0, /* tp_weaklistoffset */
3023 0, /* tp_iter */
3024 0, /* tp_iternext */
3025 NullImporter_methods, /* tp_methods */
3026 0, /* tp_members */
3027 0, /* tp_getset */
3028 0, /* tp_base */
3029 0, /* tp_dict */
3030 0, /* tp_descr_get */
3031 0, /* tp_descr_set */
3032 0, /* tp_dictoffset */
3033 (initproc)NullImporter_init, /* tp_init */
3034 0, /* tp_alloc */
3035 PyType_GenericNew /* tp_new */
3036};
3037
3038
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003039PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003040initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003041{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003042 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003043
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003044 if (PyType_Ready(&NullImporterType) < 0)
3045 goto failure;
3046
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003047 m = Py_InitModule4("imp", imp_methods, doc_imp,
3048 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003049 if (m == NULL)
3050 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003051 d = PyModule_GetDict(m);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00003052 if (d == NULL)
3053 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003054
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003055 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3056 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3057 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3058 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3059 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3060 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3061 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3062 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003063 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003064 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003065
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003066 Py_INCREF(&NullImporterType);
3067 PyModule_AddObject(m, "NullImporter", (PyObject *)&NullImporterType);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003068 failure:
3069 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003070}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003071
3072
Guido van Rossumb18618d2000-05-03 23:44:39 +00003073/* API for embedding applications that want to add their own entries
3074 to the table of built-in modules. This should normally be called
3075 *before* Py_Initialize(). When the table resize fails, -1 is
3076 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003077
3078 After a similar function by Just van Rossum. */
3079
3080int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003081PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003082{
3083 static struct _inittab *our_copy = NULL;
3084 struct _inittab *p;
3085 int i, n;
3086
3087 /* Count the number of entries in both tables */
3088 for (n = 0; newtab[n].name != NULL; n++)
3089 ;
3090 if (n == 0)
3091 return 0; /* Nothing to do */
3092 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3093 ;
3094
3095 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003096 p = our_copy;
3097 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003098 if (p == NULL)
3099 return -1;
3100
3101 /* Copy the tables into the new memory */
3102 if (our_copy != PyImport_Inittab)
3103 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3104 PyImport_Inittab = our_copy = p;
3105 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3106
3107 return 0;
3108}
3109
3110/* Shorthand to add a single entry given a name and a function */
3111
3112int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003113PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003114{
3115 struct _inittab newtab[2];
3116
3117 memset(newtab, '\0', sizeof newtab);
3118
3119 newtab[0].name = name;
3120 newtab[0].initfunc = initfunc;
3121
3122 return PyImport_ExtendInittab(newtab);
3123}
Anthony Baxterac6bd462006-04-13 02:06:09 +00003124
3125#ifdef __cplusplus
3126}
3127#endif