blob: a05078e2b72ff2694b3792558778c38bff9e18dc [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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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
Guido van Rossum45aecf42006-03-15 04:58:47 +000034 The magic numbers must be spaced apart at least 2 values, as the
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000035 -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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000063 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
64 Python 2.5b3: 62111 (fix wrong code: x += yield)
65 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
66 storing constants that should have been removed)
Guido van Rossum45aecf42006-03-15 04:58:47 +000067 Python 3000: 3000
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000068.
Tim Peters36515e22001-11-18 04:06:29 +000069*/
Guido van Rossum45aecf42006-03-15 04:58:47 +000070#define MAGIC (3000 | ((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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000125 if (filetab == NULL)
126 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 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +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();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +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 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000254 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000255 import_lock = PyThread_allocate_lock();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +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;
Thomas Wouters477c8d52006-05-27 19:21:47 +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 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +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);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +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;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001268 if (importer == NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001269 Py_XDECREF(copy);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001270 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001271 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001272 /* Note: importer is a borrowed reference */
1273 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);
Thomas Wouters0e3f5912006-08-11 14:57:12 +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);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001287 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001288 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001289 }
Thomas Wouters477c8d52006-05-27 19:21:47 +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 Wouters477c8d52006-05-27 19:21:47 +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 Wouters477c8d52006-05-27 19:21:47 +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);
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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;
Thomas Wouters0e3f5912006-08-11 14:57:12 +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 Wouters49fd7fa2006-04-21 10:40:58 +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 */
2001 Py_DECREF(tail);
2002 Py_DECREF(head);
2003 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,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +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,
2117 "Relative importpath too deep");
2118 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,
2136 "Relative importpath too deep");
2137 return NULL;
2138 }
2139 *dot = '\0';
2140 }
2141 *p_buflen = strlen(buf);
2142
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002143 modules = PyImport_GetModuleDict();
2144 parent = PyDict_GetItemString(modules, buf);
2145 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002146 PyErr_Format(PyExc_SystemError,
2147 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002148 return parent;
2149 /* We expect, but can't guarantee, if parent != None, that:
2150 - parent.__name__ == buf
2151 - parent.__dict__ is globals
2152 If this is violated... Who cares? */
2153}
2154
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002155/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002156static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002157load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002158 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002159{
2160 char *name = *p_name;
2161 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002162 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002163 char *p;
2164 PyObject *result;
2165
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002166 if (strlen(name) == 0) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002167 /* completely empty module name should only happen in
2168 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002169 Py_INCREF(mod);
2170 *p_name = NULL;
2171 return mod;
2172 }
2173
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002174 if (dot == NULL) {
2175 *p_name = NULL;
2176 len = strlen(name);
2177 }
2178 else {
2179 *p_name = dot+1;
2180 len = dot-name;
2181 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002182 if (len == 0) {
2183 PyErr_SetString(PyExc_ValueError,
2184 "Empty module name");
2185 return NULL;
2186 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002187
2188 p = buf + *p_buflen;
2189 if (p != buf)
2190 *p++ = '.';
2191 if (p+len-buf >= MAXPATHLEN) {
2192 PyErr_SetString(PyExc_ValueError,
2193 "Module name too long");
2194 return NULL;
2195 }
2196 strncpy(p, name, len);
2197 p[len] = '\0';
2198 *p_buflen = p+len-buf;
2199
2200 result = import_submodule(mod, p, buf);
2201 if (result == Py_None && altmod != mod) {
2202 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002203 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002204 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002205 if (result != NULL && result != Py_None) {
2206 if (mark_miss(buf) != 0) {
2207 Py_DECREF(result);
2208 return NULL;
2209 }
2210 strncpy(buf, name, len);
2211 buf[len] = '\0';
2212 *p_buflen = len;
2213 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002214 }
2215 if (result == NULL)
2216 return NULL;
2217
2218 if (result == Py_None) {
2219 Py_DECREF(result);
2220 PyErr_Format(PyExc_ImportError,
2221 "No module named %.200s", name);
2222 return NULL;
2223 }
2224
2225 return result;
2226}
2227
2228static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002229mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002230{
2231 PyObject *modules = PyImport_GetModuleDict();
2232 return PyDict_SetItemString(modules, name, Py_None);
2233}
2234
2235static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002236ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002237 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002238{
2239 int i;
2240
2241 if (!PyObject_HasAttrString(mod, "__path__"))
2242 return 1;
2243
2244 for (i = 0; ; i++) {
2245 PyObject *item = PySequence_GetItem(fromlist, i);
2246 int hasit;
2247 if (item == NULL) {
2248 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2249 PyErr_Clear();
2250 return 1;
2251 }
2252 return 0;
2253 }
2254 if (!PyString_Check(item)) {
2255 PyErr_SetString(PyExc_TypeError,
2256 "Item in ``from list'' not a string");
2257 Py_DECREF(item);
2258 return 0;
2259 }
2260 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002261 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002262 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002263 /* See if the package defines __all__ */
2264 if (recursive)
2265 continue; /* Avoid endless recursion */
2266 all = PyObject_GetAttrString(mod, "__all__");
2267 if (all == NULL)
2268 PyErr_Clear();
2269 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002270 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002271 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002272 if (!ret)
2273 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002274 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002275 continue;
2276 }
2277 hasit = PyObject_HasAttr(mod, item);
2278 if (!hasit) {
2279 char *subname = PyString_AS_STRING(item);
2280 PyObject *submod;
2281 char *p;
2282 if (buflen + strlen(subname) >= MAXPATHLEN) {
2283 PyErr_SetString(PyExc_ValueError,
2284 "Module name too long");
2285 Py_DECREF(item);
2286 return 0;
2287 }
2288 p = buf + buflen;
2289 *p++ = '.';
2290 strcpy(p, subname);
2291 submod = import_submodule(mod, subname, buf);
2292 Py_XDECREF(submod);
2293 if (submod == NULL) {
2294 Py_DECREF(item);
2295 return 0;
2296 }
2297 }
2298 Py_DECREF(item);
2299 }
2300
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002301 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002302}
2303
Neil Schemenauer00b09662003-06-16 21:03:07 +00002304static int
2305add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2306 PyObject *modules)
2307{
2308 if (mod == Py_None)
2309 return 1;
2310 /* Irrespective of the success of this load, make a
2311 reference to it in the parent package module. A copy gets
2312 saved in the modules dictionary under the full name, so get a
2313 reference from there, if need be. (The exception is when the
2314 load failed with a SyntaxError -- then there's no trace in
2315 sys.modules. In that case, of course, do nothing extra.) */
2316 if (submod == NULL) {
2317 submod = PyDict_GetItemString(modules, fullname);
2318 if (submod == NULL)
2319 return 1;
2320 }
2321 if (PyModule_Check(mod)) {
2322 /* We can't use setattr here since it can give a
2323 * spurious warning if the submodule name shadows a
2324 * builtin name */
2325 PyObject *dict = PyModule_GetDict(mod);
2326 if (!dict)
2327 return 0;
2328 if (PyDict_SetItemString(dict, subname, submod) < 0)
2329 return 0;
2330 }
2331 else {
2332 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2333 return 0;
2334 }
2335 return 1;
2336}
2337
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002338static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002339import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002340{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002341 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002342 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002343
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002344 /* Require:
2345 if mod == None: subname == fullname
2346 else: mod.__name__ + "." + subname == fullname
2347 */
2348
Tim Peters50d8d372001-02-28 05:34:27 +00002349 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002350 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002351 }
2352 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002353 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002354 char buf[MAXPATHLEN+1];
2355 struct filedescr *fdp;
2356 FILE *fp = NULL;
2357
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002358 if (mod == Py_None)
2359 path = NULL;
2360 else {
2361 path = PyObject_GetAttrString(mod, "__path__");
2362 if (path == NULL) {
2363 PyErr_Clear();
2364 Py_INCREF(Py_None);
2365 return Py_None;
2366 }
2367 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002368
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002369 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002370 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2371 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002372 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002373 if (fdp == NULL) {
2374 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2375 return NULL;
2376 PyErr_Clear();
2377 Py_INCREF(Py_None);
2378 return Py_None;
2379 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002380 m = load_module(fullname, fp, buf, fdp->type, loader);
2381 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002382 if (fp)
2383 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002384 if (!add_submodule(mod, m, fullname, subname, modules)) {
2385 Py_XDECREF(m);
2386 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002387 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002388 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002389
2390 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002391}
2392
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002393
2394/* Re-import a module of any kind and return its module object, WITH
2395 INCREMENTED REFERENCE COUNT */
2396
Guido van Rossum79f25d91997-04-29 20:08:16 +00002397PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002398PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002399{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002400 PyObject *modules = PyImport_GetModuleDict();
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002401 PyObject *path = NULL, *loader = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002402 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002403 char buf[MAXPATHLEN+1];
2404 struct filedescr *fdp;
2405 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002406 PyObject *newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002407
Guido van Rossum79f25d91997-04-29 20:08:16 +00002408 if (m == NULL || !PyModule_Check(m)) {
2409 PyErr_SetString(PyExc_TypeError,
2410 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002411 return NULL;
2412 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002413 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002414 if (name == NULL)
2415 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002416 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002417 PyErr_Format(PyExc_ImportError,
2418 "reload(): module %.200s not in sys.modules",
2419 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002420 return NULL;
2421 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002422 subname = strrchr(name, '.');
2423 if (subname == NULL)
2424 subname = name;
2425 else {
2426 PyObject *parentname, *parent;
2427 parentname = PyString_FromStringAndSize(name, (subname-name));
2428 if (parentname == NULL)
2429 return NULL;
2430 parent = PyDict_GetItem(modules, parentname);
2431 if (parent == NULL) {
2432 PyErr_Format(PyExc_ImportError,
2433 "reload(): parent %.200s not in sys.modules",
Georg Brandl0c55f292005-09-14 06:56:20 +00002434 PyString_AS_STRING(parentname));
2435 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002436 return NULL;
2437 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002438 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002439 subname++;
2440 path = PyObject_GetAttrString(parent, "__path__");
2441 if (path == NULL)
2442 PyErr_Clear();
2443 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002444 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002445 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002446 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002447
2448 if (fdp == NULL) {
2449 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002450 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002451 }
2452
2453 newm = load_module(name, fp, buf, fdp->type, loader);
2454 Py_XDECREF(loader);
2455
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002456 if (fp)
2457 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002458 if (newm == NULL) {
2459 /* load_module probably removed name from modules because of
2460 * the error. Put back the original module object. We're
2461 * going to return NULL in this case regardless of whether
2462 * replacing name succeeds, so the return value is ignored.
2463 */
2464 PyDict_SetItemString(modules, name, m);
2465 }
2466 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002467}
2468
2469
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002470/* Higher-level import emulator which emulates the "import" statement
2471 more accurately -- it invokes the __import__() function from the
2472 builtins of the current globals. This means that the import is
2473 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002474 environment, e.g. by "rexec".
2475 A dummy list ["__doc__"] is passed as the 4th argument so that
2476 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2477 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002478
2479PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002480PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002481{
2482 static PyObject *silly_list = NULL;
2483 static PyObject *builtins_str = NULL;
2484 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002485 PyObject *globals = NULL;
2486 PyObject *import = NULL;
2487 PyObject *builtins = NULL;
2488 PyObject *r = NULL;
2489
2490 /* Initialize constant string objects */
2491 if (silly_list == NULL) {
2492 import_str = PyString_InternFromString("__import__");
2493 if (import_str == NULL)
2494 return NULL;
2495 builtins_str = PyString_InternFromString("__builtins__");
2496 if (builtins_str == NULL)
2497 return NULL;
2498 silly_list = Py_BuildValue("[s]", "__doc__");
2499 if (silly_list == NULL)
2500 return NULL;
2501 }
2502
2503 /* Get the builtins from current globals */
2504 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002505 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002506 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002507 builtins = PyObject_GetItem(globals, builtins_str);
2508 if (builtins == NULL)
2509 goto err;
2510 }
2511 else {
2512 /* No globals -- use standard builtins, and fake globals */
2513 PyErr_Clear();
2514
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002515 builtins = PyImport_ImportModuleLevel("__builtin__",
2516 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002517 if (builtins == NULL)
2518 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002519 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2520 if (globals == NULL)
2521 goto err;
2522 }
2523
2524 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002525 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002526 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002527 if (import == NULL)
2528 PyErr_SetObject(PyExc_KeyError, import_str);
2529 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002530 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002531 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002532 if (import == NULL)
2533 goto err;
2534
2535 /* Call the _import__ function with the proper argument list */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002536 r = PyObject_CallFunctionObjArgs(import, module_name, globals,
2537 globals, silly_list, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002538
2539 err:
2540 Py_XDECREF(globals);
2541 Py_XDECREF(builtins);
2542 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002543
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002544 return r;
2545}
2546
2547
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002548/* Module 'imp' provides Python access to the primitives used for
2549 importing modules.
2550*/
2551
Guido van Rossum79f25d91997-04-29 20:08:16 +00002552static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002553imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002554{
2555 char buf[4];
2556
Guido van Rossum96774c12000-05-01 20:19:08 +00002557 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2558 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2559 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2560 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002561
Guido van Rossum79f25d91997-04-29 20:08:16 +00002562 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002563}
2564
Guido van Rossum79f25d91997-04-29 20:08:16 +00002565static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002566imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002567{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002568 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002569 struct filedescr *fdp;
2570
Guido van Rossum79f25d91997-04-29 20:08:16 +00002571 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002572 if (list == NULL)
2573 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002574 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2575 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002576 fdp->suffix, fdp->mode, fdp->type);
2577 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002578 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002579 return NULL;
2580 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002581 if (PyList_Append(list, item) < 0) {
2582 Py_DECREF(list);
2583 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002584 return NULL;
2585 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002586 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002587 }
2588 return list;
2589}
2590
Guido van Rossum79f25d91997-04-29 20:08:16 +00002591static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002592call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002593{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002594 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002595 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002596 struct filedescr *fdp;
2597 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002598 FILE *fp = NULL;
2599
2600 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002601 if (path == Py_None)
2602 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002603 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002604 if (fdp == NULL)
2605 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002606 if (fp != NULL) {
2607 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2608 if (fob == NULL) {
2609 fclose(fp);
2610 return NULL;
2611 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002612 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002613 else {
2614 fob = Py_None;
2615 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002616 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002617 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002618 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002619 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002620 return ret;
2621}
2622
Guido van Rossum79f25d91997-04-29 20:08:16 +00002623static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002624imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002625{
2626 char *name;
2627 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002628 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002629 return NULL;
2630 return call_find_module(name, path);
2631}
2632
2633static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002634imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002635{
2636 char *name;
2637 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002638 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002639 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002640 return NULL;
2641 ret = init_builtin(name);
2642 if (ret < 0)
2643 return NULL;
2644 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002645 Py_INCREF(Py_None);
2646 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002647 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002648 m = PyImport_AddModule(name);
2649 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002650 return m;
2651}
2652
Guido van Rossum79f25d91997-04-29 20:08:16 +00002653static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002654imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002655{
2656 char *name;
2657 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002658 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002659 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002660 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002661 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002662 if (ret < 0)
2663 return NULL;
2664 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002665 Py_INCREF(Py_None);
2666 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002667 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002668 m = PyImport_AddModule(name);
2669 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002670 return m;
2671}
2672
Guido van Rossum79f25d91997-04-29 20:08:16 +00002673static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002674imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002675{
2676 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002677
Guido van Rossum43713e52000-02-29 13:59:29 +00002678 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002679 return NULL;
2680 return get_frozen_object(name);
2681}
2682
Guido van Rossum79f25d91997-04-29 20:08:16 +00002683static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002684imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002685{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002686 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002687 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002688 return NULL;
Guido van Rossum50ee94f2002-04-09 18:00:58 +00002689 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002690}
2691
Guido van Rossum79f25d91997-04-29 20:08:16 +00002692static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002693imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002694{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002695 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002696 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002697 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002698 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002699 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002700 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002701}
2702
2703static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002704get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002705{
2706 FILE *fp;
2707 if (fob == NULL) {
Tim Peters86c7d2f2004-08-01 23:24:21 +00002708 if (mode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002709 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002710 fp = fopen(pathname, mode);
2711 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002712 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002713 }
2714 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002715 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002716 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002717 PyErr_SetString(PyExc_ValueError,
2718 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002719 }
2720 return fp;
2721}
2722
Guido van Rossum79f25d91997-04-29 20:08:16 +00002723static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002724imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002725{
2726 char *name;
2727 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002728 PyObject *fob = NULL;
2729 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002730 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002731 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002732 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002733 return NULL;
2734 fp = get_file(pathname, fob, "rb");
2735 if (fp == NULL)
2736 return NULL;
2737 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002738 if (fob == NULL)
2739 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002740 return m;
2741}
2742
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002743#ifdef HAVE_DYNAMIC_LOADING
2744
Guido van Rossum79f25d91997-04-29 20:08:16 +00002745static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002746imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002747{
2748 char *name;
2749 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002750 PyObject *fob = NULL;
2751 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002752 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002753 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002754 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002755 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002756 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002757 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002758 if (fp == NULL)
2759 return NULL;
2760 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002761 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002762 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002763}
2764
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002765#endif /* HAVE_DYNAMIC_LOADING */
2766
Guido van Rossum79f25d91997-04-29 20:08:16 +00002767static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002768imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002769{
2770 char *name;
2771 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002772 PyObject *fob = NULL;
2773 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002774 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002775 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002776 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002777 return NULL;
2778 fp = get_file(pathname, fob, "r");
2779 if (fp == NULL)
2780 return NULL;
2781 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002782 if (fob == NULL)
2783 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002784 return m;
2785}
2786
Guido van Rossum79f25d91997-04-29 20:08:16 +00002787static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002788imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002789{
2790 char *name;
2791 PyObject *fob;
2792 char *pathname;
2793 char *suffix; /* Unused */
2794 char *mode;
2795 int type;
2796 FILE *fp;
2797
Guido van Rossum43713e52000-02-29 13:59:29 +00002798 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002799 &name, &fob, &pathname,
2800 &suffix, &mode, &type))
2801 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002802 if (*mode) {
2803 /* Mode must start with 'r' or 'U' and must not contain '+'.
2804 Implicit in this test is the assumption that the mode
2805 may contain other modifiers like 'b' or 't'. */
2806
2807 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002808 PyErr_Format(PyExc_ValueError,
2809 "invalid file open mode %.200s", mode);
2810 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002811 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002812 }
2813 if (fob == Py_None)
2814 fp = NULL;
2815 else {
2816 if (!PyFile_Check(fob)) {
2817 PyErr_SetString(PyExc_ValueError,
2818 "load_module arg#2 should be a file or None");
2819 return NULL;
2820 }
2821 fp = get_file(pathname, fob, mode);
2822 if (fp == NULL)
2823 return NULL;
2824 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002825 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002826}
2827
2828static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002829imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002830{
2831 char *name;
2832 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002833 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002834 return NULL;
2835 return load_package(name, pathname);
2836}
2837
2838static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002839imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002840{
2841 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002842 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002843 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002844 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002845}
2846
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002847/* Doc strings */
2848
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002849PyDoc_STRVAR(doc_imp,
2850"This module provides the components needed to build your own\n\
2851__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002852
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002853PyDoc_STRVAR(doc_find_module,
2854"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002855Search for a module. If path is omitted or None, search for a\n\
2856built-in, frozen or special module and continue search in sys.path.\n\
2857The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002858package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002859
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002860PyDoc_STRVAR(doc_load_module,
2861"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002862Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002863The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002864
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002865PyDoc_STRVAR(doc_get_magic,
2866"get_magic() -> string\n\
2867Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002868
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002869PyDoc_STRVAR(doc_get_suffixes,
2870"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002871Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002872that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002873
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002874PyDoc_STRVAR(doc_new_module,
2875"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002876Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002877The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002878
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002879PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00002880"lock_held() -> boolean\n\
2881Return True if the import lock is currently held, else False.\n\
2882On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00002883
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002884PyDoc_STRVAR(doc_acquire_lock,
2885"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00002886Acquires the interpreter's import lock for the current thread.\n\
2887This lock should be used by import hooks to ensure thread-safety\n\
2888when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002889On platforms without threads, this function does nothing.");
2890
2891PyDoc_STRVAR(doc_release_lock,
2892"release_lock() -> None\n\
2893Release the interpreter's import lock.\n\
2894On platforms without threads, this function does nothing.");
2895
Guido van Rossum79f25d91997-04-29 20:08:16 +00002896static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002897 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
2898 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
2899 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
2900 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
2901 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
2902 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
2903 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
2904 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002905 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00002906 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
2907 {"init_builtin", imp_init_builtin, METH_VARARGS},
2908 {"init_frozen", imp_init_frozen, METH_VARARGS},
2909 {"is_builtin", imp_is_builtin, METH_VARARGS},
2910 {"is_frozen", imp_is_frozen, METH_VARARGS},
2911 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002912#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00002913 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002914#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002915 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00002916 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002917 {NULL, NULL} /* sentinel */
2918};
2919
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002920static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002921setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002922{
2923 PyObject *v;
2924 int err;
2925
2926 v = PyInt_FromLong((long)value);
2927 err = PyDict_SetItemString(d, name, v);
2928 Py_XDECREF(v);
2929 return err;
2930}
2931
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002932typedef struct {
2933 PyObject_HEAD
2934} NullImporter;
2935
2936static int
2937NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
2938{
2939 char *path;
2940
2941 if (!_PyArg_NoKeywords("NullImporter()", kwds))
2942 return -1;
2943
2944 if (!PyArg_ParseTuple(args, "s:NullImporter",
2945 &path))
2946 return -1;
2947
2948 if (strlen(path) == 0) {
2949 PyErr_SetString(PyExc_ImportError, "empty pathname");
2950 return -1;
2951 } else {
2952#ifndef RISCOS
2953 struct stat statbuf;
2954 int rv;
2955
2956 rv = stat(path, &statbuf);
2957 if (rv == 0) {
2958 /* it exists */
2959 if (S_ISDIR(statbuf.st_mode)) {
2960 /* it's a directory */
2961 PyErr_SetString(PyExc_ImportError,
2962 "existing directory");
2963 return -1;
2964 }
2965 }
2966#else
2967 if (object_exists(path)) {
2968 /* it exists */
2969 if (isdir(path)) {
2970 /* it's a directory */
2971 PyErr_SetString(PyExc_ImportError,
2972 "existing directory");
2973 return -1;
2974 }
2975 }
2976#endif
2977 }
2978 return 0;
2979}
2980
2981static PyObject *
2982NullImporter_find_module(NullImporter *self, PyObject *args)
2983{
2984 Py_RETURN_NONE;
2985}
2986
2987static PyMethodDef NullImporter_methods[] = {
2988 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
2989 "Always return None"
2990 },
2991 {NULL} /* Sentinel */
2992};
2993
2994
2995static PyTypeObject NullImporterType = {
2996 PyObject_HEAD_INIT(NULL)
2997 0, /*ob_size*/
2998 "imp.NullImporter", /*tp_name*/
2999 sizeof(NullImporter), /*tp_basicsize*/
3000 0, /*tp_itemsize*/
3001 0, /*tp_dealloc*/
3002 0, /*tp_print*/
3003 0, /*tp_getattr*/
3004 0, /*tp_setattr*/
3005 0, /*tp_compare*/
3006 0, /*tp_repr*/
3007 0, /*tp_as_number*/
3008 0, /*tp_as_sequence*/
3009 0, /*tp_as_mapping*/
3010 0, /*tp_hash */
3011 0, /*tp_call*/
3012 0, /*tp_str*/
3013 0, /*tp_getattro*/
3014 0, /*tp_setattro*/
3015 0, /*tp_as_buffer*/
3016 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3017 "Null importer object", /* tp_doc */
3018 0, /* tp_traverse */
3019 0, /* tp_clear */
3020 0, /* tp_richcompare */
3021 0, /* tp_weaklistoffset */
3022 0, /* tp_iter */
3023 0, /* tp_iternext */
3024 NullImporter_methods, /* tp_methods */
3025 0, /* tp_members */
3026 0, /* tp_getset */
3027 0, /* tp_base */
3028 0, /* tp_dict */
3029 0, /* tp_descr_get */
3030 0, /* tp_descr_set */
3031 0, /* tp_dictoffset */
3032 (initproc)NullImporter_init, /* tp_init */
3033 0, /* tp_alloc */
3034 PyType_GenericNew /* tp_new */
3035};
3036
3037
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003038PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003039initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003040{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003041 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003042
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003043 if (PyType_Ready(&NullImporterType) < 0)
3044 goto failure;
3045
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003046 m = Py_InitModule4("imp", imp_methods, doc_imp,
3047 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003048 if (m == NULL)
3049 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003050 d = PyModule_GetDict(m);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003051 if (d == NULL)
3052 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003053
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003054 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3055 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3056 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3057 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3058 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3059 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3060 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3061 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003062 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003063 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003064
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003065 Py_INCREF(&NullImporterType);
3066 PyModule_AddObject(m, "NullImporter", (PyObject *)&NullImporterType);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003067 failure:
3068 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003069}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003070
3071
Guido van Rossumb18618d2000-05-03 23:44:39 +00003072/* API for embedding applications that want to add their own entries
3073 to the table of built-in modules. This should normally be called
3074 *before* Py_Initialize(). When the table resize fails, -1 is
3075 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003076
3077 After a similar function by Just van Rossum. */
3078
3079int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003080PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003081{
3082 static struct _inittab *our_copy = NULL;
3083 struct _inittab *p;
3084 int i, n;
3085
3086 /* Count the number of entries in both tables */
3087 for (n = 0; newtab[n].name != NULL; n++)
3088 ;
3089 if (n == 0)
3090 return 0; /* Nothing to do */
3091 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3092 ;
3093
3094 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003095 p = our_copy;
3096 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003097 if (p == NULL)
3098 return -1;
3099
3100 /* Copy the tables into the new memory */
3101 if (our_copy != PyImport_Inittab)
3102 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3103 PyImport_Inittab = our_copy = p;
3104 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3105
3106 return 0;
3107}
3108
3109/* Shorthand to add a single entry given a name and a function */
3110
3111int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003112PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003113{
3114 struct _inittab newtab[2];
3115
3116 memset(newtab, '\0', sizeof newtab);
3117
3118 newtab[0].name = name;
3119 newtab[0].initfunc = initfunc;
3120
3121 return PyImport_ExtendInittab(newtab);
3122}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003123
3124#ifdef __cplusplus
3125}
3126#endif