blob: da9de8b64fd3e7372708d7abc52de2811be67853 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Module definition and import implementation */
3
Guido van Rossum79f25d91997-04-29 20:08:16 +00004#include "Python.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +00007#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00008#include "pythonrun.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00009#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000010#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000011#include "code.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000012#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000013#include "eval.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000014#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000015#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000016
Guido van Rossum55a83382000-09-20 20:31:38 +000017#ifdef HAVE_FCNTL_H
18#include <fcntl.h>
19#endif
Anthony Baxterac6bd462006-04-13 02:06:09 +000020#ifdef __cplusplus
21extern "C" {
22#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000023
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000024extern time_t PyOS_GetLastModificationTime(char *, FILE *);
25 /* In getmtime.c */
Guido van Rossum21d335e1993-10-15 13:01:11 +000026
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000027/* Magic word to reject .pyc files generated by other Python versions.
28 It should change for each incompatible change to the bytecode.
29
30 The value of CR and LF is incorporated so if you ever read or write
Guido van Rossum7faeab31995-07-07 22:50:36 +000031 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000032 Apple MPW compiler swaps their values, botching string constants.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000033
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000034 The magic numbers must be spaced apart atleast 2 values, as the
35 -U interpeter flag will cause MAGIC+1 being used. They have been
36 odd numbers for some time now.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000037
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000038 There were a variety of old schemes for setting the magic number.
39 The current working scheme is to increment the previous value by
40 10.
Guido van Rossumf6894922002-08-31 15:16:14 +000041
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000042 Known values:
43 Python 1.5: 20121
44 Python 1.5.1: 20121
45 Python 1.5.2: 20121
Skip Montanaro4ec3c262006-03-25 14:12:03 +000046 Python 1.6: 50428
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000047 Python 2.0: 50823
48 Python 2.0.1: 50823
49 Python 2.1: 60202
50 Python 2.1.1: 60202
51 Python 2.1.2: 60202
52 Python 2.2: 60717
Neal Norwitz7fdcb412002-06-14 01:07:39 +000053 Python 2.3a0: 62011
Michael W. Hudsondd32a912002-08-15 14:59:02 +000054 Python 2.3a0: 62021
Guido van Rossumf6894922002-08-31 15:16:14 +000055 Python 2.3a0: 62011 (!)
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000056 Python 2.4a0: 62041
Raymond Hettingerfd2d1f72004-08-23 23:37:48 +000057 Python 2.4a3: 62051
Raymond Hettinger2c31a052004-09-22 18:44:21 +000058 Python 2.4b1: 62061
Michael W. Hudsondf888462005-06-03 14:41:55 +000059 Python 2.5a0: 62071
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000060 Python 2.5a0: 62081 (ast-branch)
Guido van Rossumc2e20742006-02-27 22:32:47 +000061 Python 2.5a0: 62091 (with)
Guido van Rossumf6694362006-03-10 02:28:35 +000062 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode)
Neal Norwitz0d62a062006-07-30 06:53:31 +000063 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
64 Python 2.5b3: 62111 (fix wrong code: x += yield)
Neal Norwitz9a70f952006-08-04 05:12:19 +000065 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
66 storing constants that should have been removed)
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000067.
Tim Peters36515e22001-11-18 04:06:29 +000068*/
Neal Norwitz4ffedad2006-08-04 04:58:47 +000069#define MAGIC (62121 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000070
Guido van Rossum96774c12000-05-01 20:19:08 +000071/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000072 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000073 compiler works which are enabled by command line switches. */
74static long pyc_magic = MAGIC;
75
Guido van Rossum25ce5661997-08-02 03:10:38 +000076/* See _PyImport_FixupExtension() below */
77static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000078
Guido van Rossum771c6c81997-10-31 18:37:24 +000079/* This table is defined in config.c: */
80extern struct _inittab _PyImport_Inittab[];
81
82struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000083
Guido van Rossumed1170e1999-12-20 21:23:41 +000084/* these tables define the module suffixes that Python recognizes */
85struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +000086
87#ifdef RISCOS
88static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000089 {"/py", "U", PY_SOURCE},
Guido van Rossum48a680c2001-03-02 06:34:14 +000090 {"/pyc", "rb", PY_COMPILED},
91 {0, 0}
92};
93#else
Guido van Rossumed1170e1999-12-20 21:23:41 +000094static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000095 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000096#ifdef MS_WINDOWS
Jack Jansenc88da1f2002-05-28 10:58:19 +000097 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +000098#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +000099 {".pyc", "rb", PY_COMPILED},
100 {0, 0}
101};
Guido van Rossum48a680c2001-03-02 06:34:14 +0000102#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000103
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000104static PyTypeObject NullImporterType; /* Forward reference */
105
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000106/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000107
108void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000109_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000110{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000111 const struct filedescr *scan;
112 struct filedescr *filetab;
113 int countD = 0;
114 int countS = 0;
115
116 /* prepare _PyImport_Filetab: copy entries from
117 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
118 */
119 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
120 ++countD;
121 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
122 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000123 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Neal Norwitze1fdb322006-07-21 05:32:28 +0000124 if (filetab == NULL)
Neal Norwitz33722ae2006-07-21 07:59:02 +0000125 Py_FatalError("Can't initialize import file table.");
Guido van Rossumed1170e1999-12-20 21:23:41 +0000126 memcpy(filetab, _PyImport_DynLoadFiletab,
127 countD * sizeof(struct filedescr));
128 memcpy(filetab + countD, _PyImport_StandardFiletab,
129 countS * sizeof(struct filedescr));
130 filetab[countD + countS].suffix = NULL;
131
132 _PyImport_Filetab = filetab;
133
Guido van Rossum0824f631997-03-11 18:37:35 +0000134 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000135 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
136 for (; filetab->suffix != NULL; filetab++) {
Guido van Rossum48a680c2001-03-02 06:34:14 +0000137#ifndef RISCOS
Guido van Rossumed1170e1999-12-20 21:23:41 +0000138 if (strcmp(filetab->suffix, ".pyc") == 0)
139 filetab->suffix = ".pyo";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000140#else
141 if (strcmp(filetab->suffix, "/pyc") == 0)
142 filetab->suffix = "/pyo";
143#endif
Guido van Rossum0824f631997-03-11 18:37:35 +0000144 }
145 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000146
147 if (Py_UnicodeFlag) {
148 /* Fix the pyc_magic so that byte compiled code created
149 using the all-Unicode method doesn't interfere with
150 code created in normal operation mode. */
151 pyc_magic = MAGIC + 1;
152 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000153}
154
Guido van Rossum25ce5661997-08-02 03:10:38 +0000155void
Just van Rossum52e14d62002-12-30 22:08:05 +0000156_PyImportHooks_Init(void)
157{
158 PyObject *v, *path_hooks = NULL, *zimpimport;
159 int err = 0;
160
161 /* adding sys.path_hooks and sys.path_importer_cache, setting up
162 zipimport */
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000163 if (PyType_Ready(&NullImporterType) < 0)
164 goto error;
Just van Rossum52e14d62002-12-30 22:08:05 +0000165
166 if (Py_VerboseFlag)
167 PySys_WriteStderr("# installing zipimport hook\n");
168
169 v = PyList_New(0);
170 if (v == NULL)
171 goto error;
172 err = PySys_SetObject("meta_path", v);
173 Py_DECREF(v);
174 if (err)
175 goto error;
176 v = PyDict_New();
177 if (v == NULL)
178 goto error;
179 err = PySys_SetObject("path_importer_cache", v);
180 Py_DECREF(v);
181 if (err)
182 goto error;
183 path_hooks = PyList_New(0);
184 if (path_hooks == NULL)
185 goto error;
186 err = PySys_SetObject("path_hooks", path_hooks);
187 if (err) {
188 error:
189 PyErr_Print();
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000190 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
191 "path_importer_cache, or NullImporter failed"
192 );
Just van Rossum52e14d62002-12-30 22:08:05 +0000193 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000194
Just van Rossum52e14d62002-12-30 22:08:05 +0000195 zimpimport = PyImport_ImportModule("zipimport");
196 if (zimpimport == NULL) {
197 PyErr_Clear(); /* No zip import module -- okay */
198 if (Py_VerboseFlag)
199 PySys_WriteStderr("# can't import zipimport\n");
200 }
201 else {
202 PyObject *zipimporter = PyObject_GetAttrString(zimpimport,
203 "zipimporter");
204 Py_DECREF(zimpimport);
205 if (zipimporter == NULL) {
206 PyErr_Clear(); /* No zipimporter object -- okay */
207 if (Py_VerboseFlag)
208 PySys_WriteStderr(
Fred Drake1e5fc552003-07-11 15:01:02 +0000209 "# can't import zipimport.zipimporter\n");
Just van Rossum52e14d62002-12-30 22:08:05 +0000210 }
211 else {
212 /* sys.path_hooks.append(zipimporter) */
213 err = PyList_Append(path_hooks, zipimporter);
214 Py_DECREF(zipimporter);
215 if (err)
216 goto error;
217 if (Py_VerboseFlag)
218 PySys_WriteStderr(
219 "# installed zipimport hook\n");
220 }
221 }
222 Py_DECREF(path_hooks);
223}
224
225void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000226_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000227{
228 Py_XDECREF(extensions);
229 extensions = NULL;
Barry Warsaw84294482000-10-03 16:02:05 +0000230 PyMem_DEL(_PyImport_Filetab);
231 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000232}
233
234
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000235/* Locking primitives to prevent parallel imports of the same module
236 in different threads to return with a partially loaded module.
237 These calls are serialized by the global interpreter lock. */
238
239#ifdef WITH_THREAD
240
Guido van Rossum49b56061998-10-01 20:42:43 +0000241#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000242
Guido van Rossum65d5b571998-12-21 19:32:43 +0000243static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000244static long import_lock_thread = -1;
245static int import_lock_level = 0;
246
247static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000248lock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000249{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000250 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000251 if (me == -1)
252 return; /* Too bad */
Neal Norwitze1fdb322006-07-21 05:32:28 +0000253 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000254 import_lock = PyThread_allocate_lock();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000255 if (import_lock == NULL)
256 return; /* Nothing much we can do. */
257 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000258 if (import_lock_thread == me) {
259 import_lock_level++;
260 return;
261 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000262 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
263 {
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000264 PyThreadState *tstate = PyEval_SaveThread();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000265 PyThread_acquire_lock(import_lock, 1);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000266 PyEval_RestoreThread(tstate);
267 }
268 import_lock_thread = me;
269 import_lock_level = 1;
270}
271
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000272static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000273unlock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000274{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000275 long me = PyThread_get_thread_ident();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000276 if (me == -1 || import_lock == NULL)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000277 return 0; /* Too bad */
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000278 if (import_lock_thread != me)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000279 return -1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000280 import_lock_level--;
281 if (import_lock_level == 0) {
282 import_lock_thread = -1;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000283 PyThread_release_lock(import_lock);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000284 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000285 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000286}
287
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000288/* This function is called from PyOS_AfterFork to ensure that newly
289 created child processes do not share locks with the parent. */
290
291void
292_PyImport_ReInitLock(void)
293{
294#ifdef _AIX
295 if (import_lock != NULL)
296 import_lock = PyThread_allocate_lock();
297#endif
298}
299
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000300#else
301
302#define lock_import()
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000303#define unlock_import() 0
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000304
305#endif
306
Tim Peters69232342001-08-30 05:16:13 +0000307static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000308imp_lock_held(PyObject *self, PyObject *noargs)
Tim Peters69232342001-08-30 05:16:13 +0000309{
Tim Peters69232342001-08-30 05:16:13 +0000310#ifdef WITH_THREAD
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000311 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000312#else
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000313 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000314#endif
315}
316
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000317static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000318imp_acquire_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000319{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000320#ifdef WITH_THREAD
321 lock_import();
322#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000323 Py_INCREF(Py_None);
324 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000325}
326
327static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000328imp_release_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000329{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000330#ifdef WITH_THREAD
331 if (unlock_import() < 0) {
332 PyErr_SetString(PyExc_RuntimeError,
333 "not holding the import lock");
334 return NULL;
335 }
336#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000337 Py_INCREF(Py_None);
338 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000339}
340
Guido van Rossum25ce5661997-08-02 03:10:38 +0000341/* Helper for sys */
342
343PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000344PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000345{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000346 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000347 if (interp->modules == NULL)
348 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
349 return interp->modules;
350}
351
Guido van Rossum3f5da241990-12-20 15:06:42 +0000352
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000353/* List of names to clear in sys */
354static char* sys_deletes[] = {
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000355 "path", "argv", "ps1", "ps2", "exitfunc",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000356 "exc_type", "exc_value", "exc_traceback",
357 "last_type", "last_value", "last_traceback",
Just van Rossum52e14d62002-12-30 22:08:05 +0000358 "path_hooks", "path_importer_cache", "meta_path",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000359 NULL
360};
361
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000362static char* sys_files[] = {
363 "stdin", "__stdin__",
364 "stdout", "__stdout__",
365 "stderr", "__stderr__",
366 NULL
367};
368
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000369
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000370/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000371
Guido van Rossum3f5da241990-12-20 15:06:42 +0000372void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000373PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000374{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000375 Py_ssize_t pos, ndone;
Guido van Rossum758eec01998-01-19 21:58:26 +0000376 char *name;
377 PyObject *key, *value, *dict;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000378 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000379 PyObject *modules = interp->modules;
380
381 if (modules == NULL)
382 return; /* Already done */
383
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000384 /* Delete some special variables first. These are common
385 places where user values hide and people complain when their
386 destructors fail. Since the modules containing them are
387 deleted *last* of all, they would come too late in the normal
388 destruction order. Sigh. */
389
390 value = PyDict_GetItemString(modules, "__builtin__");
391 if (value != NULL && PyModule_Check(value)) {
392 dict = PyModule_GetDict(value);
393 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000394 PySys_WriteStderr("# clear __builtin__._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000395 PyDict_SetItemString(dict, "_", Py_None);
396 }
397 value = PyDict_GetItemString(modules, "sys");
398 if (value != NULL && PyModule_Check(value)) {
399 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000400 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000401 dict = PyModule_GetDict(value);
402 for (p = sys_deletes; *p != NULL; p++) {
403 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000404 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000405 PyDict_SetItemString(dict, *p, Py_None);
406 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000407 for (p = sys_files; *p != NULL; p+=2) {
408 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000409 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000410 v = PyDict_GetItemString(dict, *(p+1));
411 if (v == NULL)
412 v = Py_None;
413 PyDict_SetItemString(dict, *p, v);
414 }
415 }
416
417 /* First, delete __main__ */
418 value = PyDict_GetItemString(modules, "__main__");
419 if (value != NULL && PyModule_Check(value)) {
420 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000421 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000422 _PyModule_Clear(value);
423 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000424 }
425
Guido van Rossum758eec01998-01-19 21:58:26 +0000426 /* The special treatment of __builtin__ here is because even
427 when it's not referenced as a module, its dictionary is
428 referenced by almost every module's __builtins__. Since
429 deleting a module clears its dictionary (even if there are
430 references left to it), we need to delete the __builtin__
431 module last. Likewise, we don't delete sys until the very
432 end because it is implicitly referenced (e.g. by print).
433
434 Also note that we 'delete' modules by replacing their entry
435 in the modules dict with None, rather than really deleting
436 them; this avoids a rehash of the modules dictionary and
437 also marks them as "non existent" so they won't be
438 re-imported. */
439
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000440 /* Next, repeatedly delete modules with a reference count of
Guido van Rossum758eec01998-01-19 21:58:26 +0000441 one (skipping __builtin__ and sys) and delete them */
442 do {
443 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000444 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000445 while (PyDict_Next(modules, &pos, &key, &value)) {
446 if (value->ob_refcnt != 1)
447 continue;
Guido van Rossum566373e1998-10-01 15:24:50 +0000448 if (PyString_Check(key) && PyModule_Check(value)) {
449 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000450 if (strcmp(name, "__builtin__") == 0)
451 continue;
452 if (strcmp(name, "sys") == 0)
453 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000454 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000455 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000456 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000457 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000458 PyDict_SetItem(modules, key, Py_None);
459 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000460 }
461 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000462 } while (ndone > 0);
463
Guido van Rossum758eec01998-01-19 21:58:26 +0000464 /* Next, delete all modules (still skipping __builtin__ and sys) */
465 pos = 0;
466 while (PyDict_Next(modules, &pos, &key, &value)) {
Guido van Rossum566373e1998-10-01 15:24:50 +0000467 if (PyString_Check(key) && PyModule_Check(value)) {
468 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000469 if (strcmp(name, "__builtin__") == 0)
470 continue;
471 if (strcmp(name, "sys") == 0)
472 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000473 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000474 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000475 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000476 PyDict_SetItem(modules, key, Py_None);
477 }
478 }
479
480 /* Next, delete sys and __builtin__ (in that order) */
481 value = PyDict_GetItemString(modules, "sys");
482 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000483 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000484 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000485 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000486 PyDict_SetItemString(modules, "sys", Py_None);
487 }
488 value = PyDict_GetItemString(modules, "__builtin__");
489 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000490 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000491 PySys_WriteStderr("# cleanup __builtin__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000492 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000493 PyDict_SetItemString(modules, "__builtin__", Py_None);
494 }
495
496 /* Finally, clear and delete the modules directory */
497 PyDict_Clear(modules);
498 interp->modules = NULL;
499 Py_DECREF(modules);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000500}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000501
502
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000503/* Helper for pythonrun.c -- return magic number */
504
505long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000506PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000507{
Guido van Rossum96774c12000-05-01 20:19:08 +0000508 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000509}
510
511
Guido van Rossum25ce5661997-08-02 03:10:38 +0000512/* Magic for extension modules (built-in as well as dynamically
513 loaded). To prevent initializing an extension module more than
514 once, we keep a static dictionary 'extensions' keyed by module name
515 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000516 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000517 dictionary is stored by calling _PyImport_FixupExtension()
518 immediately after the module initialization function succeeds. A
519 copy can be retrieved from there by calling
520 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000521
Guido van Rossum79f25d91997-04-29 20:08:16 +0000522PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000523_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000524{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000525 PyObject *modules, *mod, *dict, *copy;
526 if (extensions == NULL) {
527 extensions = PyDict_New();
528 if (extensions == NULL)
529 return NULL;
530 }
531 modules = PyImport_GetModuleDict();
532 mod = PyDict_GetItemString(modules, name);
533 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000534 PyErr_Format(PyExc_SystemError,
535 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000536 return NULL;
537 }
538 dict = PyModule_GetDict(mod);
539 if (dict == NULL)
540 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000541 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000542 if (copy == NULL)
543 return NULL;
544 PyDict_SetItemString(extensions, filename, copy);
545 Py_DECREF(copy);
546 return copy;
547}
548
549PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000550_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000552 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000553 if (extensions == NULL)
554 return NULL;
555 dict = PyDict_GetItemString(extensions, filename);
556 if (dict == NULL)
557 return NULL;
558 mod = PyImport_AddModule(name);
559 if (mod == NULL)
560 return NULL;
561 mdict = PyModule_GetDict(mod);
562 if (mdict == NULL)
563 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000564 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000565 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000566 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000567 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000568 name, filename);
569 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000570}
571
572
573/* Get the module object corresponding to a module name.
574 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000575 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000576 Because the former action is most common, THIS DOES NOT RETURN A
577 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000578
Guido van Rossum79f25d91997-04-29 20:08:16 +0000579PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000580PyImport_AddModule(const char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000581{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000582 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000583 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000584
Guido van Rossum25ce5661997-08-02 03:10:38 +0000585 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000586 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000587 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000588 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000589 if (m == NULL)
590 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000591 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000592 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000593 return NULL;
594 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000595 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000596
597 return m;
598}
599
Tim Peters1cd70172004-08-02 03:52:12 +0000600/* Remove name from sys.modules, if it's there. */
601static void
602_RemoveModule(const char *name)
603{
604 PyObject *modules = PyImport_GetModuleDict();
605 if (PyDict_GetItemString(modules, name) == NULL)
606 return;
607 if (PyDict_DelItemString(modules, name) < 0)
608 Py_FatalError("import: deleting existing key in"
609 "sys.modules failed");
610}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000611
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000612/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000613 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
614 * removed from sys.modules, to avoid leaving damaged module objects
615 * in sys.modules. The caller may wish to restore the original
616 * module object (if any) in this case; PyImport_ReloadModule is an
617 * example.
618 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000619PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000620PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000621{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000622 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
623}
624
625PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000626PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000627{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000628 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000629 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000630
Guido van Rossum79f25d91997-04-29 20:08:16 +0000631 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000632 if (m == NULL)
633 return NULL;
Jeremy Hyltonecd91292004-01-02 23:25:32 +0000634 /* If the module is being reloaded, we get the old module back
635 and re-use its dict to exec the new code. */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000636 d = PyModule_GetDict(m);
637 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
638 if (PyDict_SetItemString(d, "__builtins__",
639 PyEval_GetBuiltins()) != 0)
Tim Peters1cd70172004-08-02 03:52:12 +0000640 goto error;
Guido van Rossum6135a871995-01-09 17:53:26 +0000641 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000642 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000643 v = NULL;
644 if (pathname != NULL) {
645 v = PyString_FromString(pathname);
646 if (v == NULL)
647 PyErr_Clear();
648 }
649 if (v == NULL) {
650 v = ((PyCodeObject *)co)->co_filename;
651 Py_INCREF(v);
652 }
653 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000654 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000655 Py_DECREF(v);
656
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000657 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000658 if (v == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000659 goto error;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000660 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000661
Guido van Rossum25ce5661997-08-02 03:10:38 +0000662 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000663 PyErr_Format(PyExc_ImportError,
664 "Loaded module %.200s not found in sys.modules",
665 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000666 return NULL;
667 }
668
Guido van Rossum79f25d91997-04-29 20:08:16 +0000669 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000670
671 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000672
673 error:
674 _RemoveModule(name);
675 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000676}
677
678
679/* Given a pathname for a Python source file, fill a buffer with the
680 pathname for the corresponding compiled file. Return the pathname
681 for the compiled file, or NULL if there's no space in the buffer.
682 Doesn't set an exception. */
683
684static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000685make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000686{
Tim Petersc1731372001-08-04 08:12:36 +0000687 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000688 if (len+2 > buflen)
689 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000690
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000691#ifdef MS_WINDOWS
Tim Petersc1731372001-08-04 08:12:36 +0000692 /* Treat .pyw as if it were .py. The case of ".pyw" must match
693 that used in _PyImport_StandardFiletab. */
694 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
695 --len; /* pretend 'w' isn't there */
696#endif
697 memcpy(buf, pathname, len);
698 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
699 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000700
701 return buf;
702}
703
704
705/* Given a pathname for a Python source file, its time of last
706 modification, and a pathname for a compiled file, check whether the
707 compiled file represents the same version of the source. If so,
708 return a FILE pointer for the compiled file, positioned just after
709 the header; if not, return NULL.
710 Doesn't set an exception. */
711
712static FILE *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000713check_compiled_module(char *pathname, time_t mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000714{
715 FILE *fp;
716 long magic;
717 long pyc_mtime;
718
719 fp = fopen(cpathname, "rb");
720 if (fp == NULL)
721 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000722 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000723 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000724 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000725 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000726 fclose(fp);
727 return NULL;
728 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000729 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000730 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000731 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000732 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000733 fclose(fp);
734 return NULL;
735 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000736 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000737 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000738 return fp;
739}
740
741
742/* Read a code object from a file and check it for validity */
743
Guido van Rossum79f25d91997-04-29 20:08:16 +0000744static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000745read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000746{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000747 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000748
Tim Petersd9b9ac82001-01-28 00:27:39 +0000749 co = PyMarshal_ReadLastObjectFromFile(fp);
Armin Rigo01ab2792004-03-26 15:09:27 +0000750 if (co == NULL)
751 return NULL;
752 if (!PyCode_Check(co)) {
753 PyErr_Format(PyExc_ImportError,
754 "Non-code object in %.200s", cpathname);
755 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000756 return NULL;
757 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000758 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000759}
760
761
762/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000763 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000764
Guido van Rossum79f25d91997-04-29 20:08:16 +0000765static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000766load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000767{
768 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000769 PyCodeObject *co;
770 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000771
Guido van Rossum79f25d91997-04-29 20:08:16 +0000772 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000773 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000774 PyErr_Format(PyExc_ImportError,
775 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000776 return NULL;
777 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000778 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000779 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000780 if (co == NULL)
781 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000782 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000783 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000784 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000785 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000786 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000787
788 return m;
789}
790
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000791/* Parse a source file and return the corresponding code object */
792
Guido van Rossum79f25d91997-04-29 20:08:16 +0000793static PyCodeObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000794parse_source_module(const char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000795{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000796 PyCodeObject *co = NULL;
797 mod_ty mod;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000798 PyArena *arena = PyArena_New();
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000799
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000800 mod = PyParser_ASTFromFile(fp, pathname, Py_file_input, 0, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000801 NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000802 if (mod) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000803 co = PyAST_Compile(mod, pathname, NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000804 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000805 PyArena_Free(arena);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000806 return co;
807}
808
809
Guido van Rossum55a83382000-09-20 20:31:38 +0000810/* Helper to open a bytecode file for writing in exclusive mode */
811
812static FILE *
813open_exclusive(char *filename)
814{
815#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
816 /* Use O_EXCL to avoid a race condition when another process tries to
817 write the same file. When that happens, our open() call fails,
818 which is just fine (since it's only a cache).
819 XXX If the file exists and is writable but the directory is not
820 writable, the file will never be written. Oh well.
821 */
822 int fd;
823 (void) unlink(filename);
Tim Peters42c83af2000-09-29 04:03:10 +0000824 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
825#ifdef O_BINARY
826 |O_BINARY /* necessary for Windows */
827#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000828#ifdef __VMS
Martin v. Löwis18e16552006-02-15 17:27:45 +0000829 , 0666, "ctxt=bin", "shr=nil"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000830#else
Martin v. Löwis18e16552006-02-15 17:27:45 +0000831 , 0666
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000832#endif
Martin v. Löwis18e16552006-02-15 17:27:45 +0000833 );
Guido van Rossum55a83382000-09-20 20:31:38 +0000834 if (fd < 0)
835 return NULL;
836 return fdopen(fd, "wb");
837#else
838 /* Best we can do -- on Windows this can't happen anyway */
839 return fopen(filename, "wb");
840#endif
841}
842
843
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000844/* Write a compiled module to a file, placing the time of last
845 modification of its source into the header.
846 Errors are ignored, if a write error occurs an attempt is made to
847 remove the file. */
848
849static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000850write_compiled_module(PyCodeObject *co, char *cpathname, time_t mtime)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000851{
852 FILE *fp;
853
Guido van Rossum55a83382000-09-20 20:31:38 +0000854 fp = open_exclusive(cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000855 if (fp == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000856 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000857 PySys_WriteStderr(
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000858 "# can't create %s\n", cpathname);
859 return;
860 }
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000861 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000862 /* First write a 0 for mtime */
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000863 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION);
864 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION);
Armin Rigo01ab2792004-03-26 15:09:27 +0000865 if (fflush(fp) != 0 || ferror(fp)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000866 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000867 PySys_WriteStderr("# can't write %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000868 /* Don't keep partial file */
869 fclose(fp);
870 (void) unlink(cpathname);
871 return;
872 }
873 /* Now write the true mtime */
874 fseek(fp, 4L, 0);
Martin v. Löwis18e16552006-02-15 17:27:45 +0000875 assert(mtime < LONG_MAX);
Martin v. Löwis725507b2006-03-07 12:08:51 +0000876 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000877 fflush(fp);
878 fclose(fp);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000879 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000880 PySys_WriteStderr("# wrote %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000881}
882
883
884/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000885 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
886 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000887
Guido van Rossum79f25d91997-04-29 20:08:16 +0000888static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000889load_source_module(char *name, char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000890{
Fred Drake4c82b232000-06-30 16:18:57 +0000891 time_t mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000892 FILE *fpc;
893 char buf[MAXPATHLEN+1];
894 char *cpathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000895 PyCodeObject *co;
896 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000897
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000898 mtime = PyOS_GetLastModificationTime(pathname, fp);
Neal Norwitz708e51a2005-10-03 04:48:15 +0000899 if (mtime == (time_t)(-1)) {
900 PyErr_Format(PyExc_RuntimeError,
901 "unable to get modification time from '%s'",
902 pathname);
Fred Drake4c82b232000-06-30 16:18:57 +0000903 return NULL;
Neal Norwitz708e51a2005-10-03 04:48:15 +0000904 }
Fred Drake4c82b232000-06-30 16:18:57 +0000905#if SIZEOF_TIME_T > 4
906 /* Python's .pyc timestamp handling presumes that the timestamp fits
907 in 4 bytes. This will be fine until sometime in the year 2038,
908 when a 4-byte signed time_t will overflow.
909 */
910 if (mtime >> 32) {
911 PyErr_SetString(PyExc_OverflowError,
Fred Drakec63d3e92001-03-01 06:33:32 +0000912 "modification time overflows a 4 byte field");
Fred Drake4c82b232000-06-30 16:18:57 +0000913 return NULL;
914 }
915#endif
Tim Peters36515e22001-11-18 04:06:29 +0000916 cpathname = make_compiled_pathname(pathname, buf,
Jeremy Hylton37832f02001-04-13 17:50:20 +0000917 (size_t)MAXPATHLEN + 1);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000918 if (cpathname != NULL &&
919 (fpc = check_compiled_module(pathname, mtime, cpathname))) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000920 co = read_compiled_module(cpathname, fpc);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000921 fclose(fpc);
922 if (co == NULL)
923 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000924 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000925 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000926 name, cpathname);
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000927 pathname = cpathname;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000928 }
929 else {
930 co = parse_source_module(pathname, fp);
931 if (co == NULL)
932 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000933 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000934 PySys_WriteStderr("import %s # from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000935 name, pathname);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +0000936 if (cpathname)
937 write_compiled_module(co, cpathname, mtime);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000938 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000939 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000940 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000941
942 return m;
943}
944
945
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000946/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +0000947static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
948static struct filedescr *find_module(char *, char *, PyObject *,
949 char *, size_t, FILE **, PyObject **);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000950static struct _frozen *find_frozen(char *name);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000951
952/* Load a package and return its module object WITH INCREMENTED
953 REFERENCE COUNT */
954
955static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000956load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000957{
Tim Peters1cd70172004-08-02 03:52:12 +0000958 PyObject *m, *d;
959 PyObject *file = NULL;
960 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000961 int err;
962 char buf[MAXPATHLEN+1];
963 FILE *fp = NULL;
964 struct filedescr *fdp;
965
966 m = PyImport_AddModule(name);
967 if (m == NULL)
968 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000969 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000970 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000971 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000972 d = PyModule_GetDict(m);
973 file = PyString_FromString(pathname);
974 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000975 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000976 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +0000977 if (path == NULL)
978 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000979 err = PyDict_SetItemString(d, "__file__", file);
980 if (err == 0)
981 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +0000982 if (err != 0)
983 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000984 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +0000985 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000986 if (fdp == NULL) {
987 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
988 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +0000989 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000990 }
991 else
992 m = NULL;
993 goto cleanup;
994 }
Just van Rossum52e14d62002-12-30 22:08:05 +0000995 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000996 if (fp != NULL)
997 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +0000998 goto cleanup;
999
1000 error:
1001 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001002 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001003 Py_XDECREF(path);
1004 Py_XDECREF(file);
1005 return m;
1006}
1007
1008
1009/* Helper to test for built-in module */
1010
1011static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001012is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001013{
1014 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001015 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1016 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1017 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001018 return -1;
1019 else
1020 return 1;
1021 }
1022 }
1023 return 0;
1024}
1025
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001026
Just van Rossum52e14d62002-12-30 22:08:05 +00001027/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1028 possibly by fetching it from the path_importer_cache dict. If it
1029 wasn't yet cached, traverse path_hooks until it a hook is found
1030 that can handle the path item. Return None if no hook could;
1031 this tells our caller it should fall back to the builtin
1032 import mechanism. Cache the result in path_importer_cache.
1033 Returns a borrowed reference. */
1034
1035static PyObject *
1036get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1037 PyObject *p)
1038{
1039 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001040 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001041
1042 /* These conditions are the caller's responsibility: */
1043 assert(PyList_Check(path_hooks));
1044 assert(PyDict_Check(path_importer_cache));
1045
1046 nhooks = PyList_Size(path_hooks);
1047 if (nhooks < 0)
1048 return NULL; /* Shouldn't happen */
1049
1050 importer = PyDict_GetItem(path_importer_cache, p);
1051 if (importer != NULL)
1052 return importer;
1053
1054 /* set path_importer_cache[p] to None to avoid recursion */
1055 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1056 return NULL;
1057
1058 for (j = 0; j < nhooks; j++) {
1059 PyObject *hook = PyList_GetItem(path_hooks, j);
1060 if (hook == NULL)
1061 return NULL;
Georg Brandl684fd0c2006-05-25 19:15:31 +00001062 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001063 if (importer != NULL)
1064 break;
1065
1066 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1067 return NULL;
1068 }
1069 PyErr_Clear();
1070 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001071 if (importer == NULL) {
1072 importer = PyObject_CallFunctionObjArgs(
1073 (PyObject *)&NullImporterType, p, NULL
1074 );
1075 if (importer == NULL) {
1076 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1077 PyErr_Clear();
1078 return Py_None;
1079 }
1080 }
1081 }
1082 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001083 int err = PyDict_SetItem(path_importer_cache, p, importer);
1084 Py_DECREF(importer);
1085 if (err != 0)
1086 return NULL;
1087 }
1088 return importer;
1089}
1090
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001091/* Search the path (default sys.path) for a module. Return the
1092 corresponding filedescr struct, and (via return arguments) the
1093 pathname and an open file. Return NULL if the module is not found. */
1094
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001095#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001096extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001097 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001098#endif
1099
Martin v. Löwis18e16552006-02-15 17:27:45 +00001100static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001101static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001102static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001103
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001104static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001105find_module(char *fullname, char *subname, PyObject *path, char *buf,
1106 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001107{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001108 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001109 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001110 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001111 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001112 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001113 PyObject *path_hooks, *path_importer_cache;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001114#ifndef RISCOS
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001115 struct stat statbuf;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001116#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001117 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1118 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1119 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001120 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001121#if defined(PYOS_OS2)
1122 size_t saved_len;
1123 size_t saved_namelen;
1124 char *saved_buf = NULL;
1125#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001126 if (p_loader != NULL)
1127 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001128
Just van Rossum52e14d62002-12-30 22:08:05 +00001129 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001130 PyErr_SetString(PyExc_OverflowError,
1131 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001132 return NULL;
1133 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001134 strcpy(name, subname);
1135
1136 /* sys.meta_path import hook */
1137 if (p_loader != NULL) {
1138 PyObject *meta_path;
1139
1140 meta_path = PySys_GetObject("meta_path");
1141 if (meta_path == NULL || !PyList_Check(meta_path)) {
1142 PyErr_SetString(PyExc_ImportError,
1143 "sys.meta_path must be a list of "
1144 "import hooks");
1145 return NULL;
1146 }
1147 Py_INCREF(meta_path); /* zap guard */
1148 npath = PyList_Size(meta_path);
1149 for (i = 0; i < npath; i++) {
1150 PyObject *loader;
1151 PyObject *hook = PyList_GetItem(meta_path, i);
1152 loader = PyObject_CallMethod(hook, "find_module",
1153 "sO", fullname,
1154 path != NULL ?
1155 path : Py_None);
1156 if (loader == NULL) {
1157 Py_DECREF(meta_path);
1158 return NULL; /* true error */
1159 }
1160 if (loader != Py_None) {
1161 /* a loader was found */
1162 *p_loader = loader;
1163 Py_DECREF(meta_path);
1164 return &importhookdescr;
1165 }
1166 Py_DECREF(loader);
1167 }
1168 Py_DECREF(meta_path);
1169 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001170
1171 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001172 /* The only type of submodule allowed inside a "frozen"
1173 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +00001174 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
1175 PyErr_SetString(PyExc_ImportError,
1176 "full frozen module name too long");
1177 return NULL;
1178 }
1179 strcpy(buf, PyString_AsString(path));
1180 strcat(buf, ".");
1181 strcat(buf, name);
1182 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001183 if (find_frozen(name) != NULL) {
1184 strcpy(buf, name);
1185 return &fd_frozen;
1186 }
1187 PyErr_Format(PyExc_ImportError,
1188 "No frozen submodule named %.200s", name);
1189 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001190 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001191 if (path == NULL) {
1192 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001193 strcpy(buf, name);
1194 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001195 }
Greg Ward201baee2001-10-04 14:52:06 +00001196 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001197 strcpy(buf, name);
1198 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001199 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001200
Guido van Rossumac279101996-08-22 23:10:58 +00001201#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001202 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1203 if (fp != NULL) {
1204 *p_fp = fp;
1205 return fdp;
1206 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001207#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001208 path = PySys_GetObject("path");
1209 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001210 if (path == NULL || !PyList_Check(path)) {
1211 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001212 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001213 return NULL;
1214 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001215
1216 path_hooks = PySys_GetObject("path_hooks");
1217 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1218 PyErr_SetString(PyExc_ImportError,
1219 "sys.path_hooks must be a list of "
1220 "import hooks");
1221 return NULL;
1222 }
1223 path_importer_cache = PySys_GetObject("path_importer_cache");
1224 if (path_importer_cache == NULL ||
1225 !PyDict_Check(path_importer_cache)) {
1226 PyErr_SetString(PyExc_ImportError,
1227 "sys.path_importer_cache must be a dict");
1228 return NULL;
1229 }
1230
Guido van Rossum79f25d91997-04-29 20:08:16 +00001231 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001232 namelen = strlen(name);
1233 for (i = 0; i < npath; i++) {
Walter Dörwald3430d702002-06-17 10:43:59 +00001234 PyObject *copy = NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001235 PyObject *v = PyList_GetItem(path, i);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00001236 if (!v)
1237 return NULL;
Walter Dörwald3430d702002-06-17 10:43:59 +00001238#ifdef Py_USING_UNICODE
1239 if (PyUnicode_Check(v)) {
1240 copy = PyUnicode_Encode(PyUnicode_AS_UNICODE(v),
1241 PyUnicode_GET_SIZE(v), Py_FileSystemDefaultEncoding, NULL);
1242 if (copy == NULL)
1243 return NULL;
1244 v = copy;
1245 }
1246 else
1247#endif
Guido van Rossum79f25d91997-04-29 20:08:16 +00001248 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001249 continue;
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001250 len = PyString_GET_SIZE(v);
Walter Dörwald3430d702002-06-17 10:43:59 +00001251 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
1252 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001253 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001254 }
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001255 strcpy(buf, PyString_AS_STRING(v));
Walter Dörwald3430d702002-06-17 10:43:59 +00001256 if (strlen(buf) != len) {
1257 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001258 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001259 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001260
1261 /* sys.path_hooks import hook */
1262 if (p_loader != NULL) {
1263 PyObject *importer;
1264
1265 importer = get_path_importer(path_importer_cache,
1266 path_hooks, v);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001267 if (importer == NULL) {
1268 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001269 return NULL;
Neal Norwitza4df11d2006-07-06 04:28:59 +00001270 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001271 /* Note: importer is a borrowed reference */
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001272 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001273 PyObject *loader;
1274 loader = PyObject_CallMethod(importer,
1275 "find_module",
1276 "s", fullname);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001277 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001278 if (loader == NULL)
1279 return NULL; /* error */
1280 if (loader != Py_None) {
1281 /* a loader was found */
1282 *p_loader = loader;
1283 return &importhookdescr;
1284 }
1285 Py_DECREF(loader);
Georg Brandlf4ef1162006-05-26 18:03:31 +00001286 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001287 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001288 }
Georg Brandlf4ef1162006-05-26 18:03:31 +00001289 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001290
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001291 if (len > 0 && buf[len-1] != SEP
1292#ifdef ALTSEP
1293 && buf[len-1] != ALTSEP
1294#endif
1295 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001296 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001297 strcpy(buf+len, name);
1298 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001299
1300 /* Check for package import (buf holds a directory name,
1301 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001302#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001303 if (stat(buf, &statbuf) == 0 && /* it exists */
1304 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001305 case_ok(buf, len, namelen, name)) { /* case matches */
1306 if (find_init_module(buf)) { /* and has __init__.py */
1307 Py_XDECREF(copy);
1308 return &fd_package;
1309 }
1310 else {
1311 char warnstr[MAXPATHLEN+80];
1312 sprintf(warnstr, "Not importing directory "
1313 "'%.*s': missing __init__.py",
1314 MAXPATHLEN, buf);
1315 if (PyErr_Warn(PyExc_ImportWarning,
1316 warnstr)) {
1317 Py_XDECREF(copy);
1318 return NULL;
1319 }
1320 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001321 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001322#else
1323 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001324#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001325 if (isdir(buf) &&
Walter Dörwald3430d702002-06-17 10:43:59 +00001326 case_ok(buf, len, namelen, name)) {
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001327 if (find_init_module(buf)) {
1328 Py_XDECREF(copy);
1329 return &fd_package;
1330 }
1331 else {
1332 char warnstr[MAXPATHLEN+80];
1333 sprintf(warnstr, "Not importing directory "
1334 "'%.*s': missing __init__.py",
1335 MAXPATHLEN, buf);
1336 if (PyErr_Warn(PyExc_ImportWarning,
1337 warnstr)) {
1338 Py_XDECREF(copy);
1339 return NULL;
1340 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001341 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001342#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001343#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001344#if defined(PYOS_OS2)
1345 /* take a snapshot of the module spec for restoration
1346 * after the 8 character DLL hackery
1347 */
1348 saved_buf = strdup(buf);
1349 saved_len = len;
1350 saved_namelen = namelen;
1351#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001352 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001353#if defined(PYOS_OS2)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001354 /* OS/2 limits DLLs to 8 character names (w/o
1355 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001356 * so if the name is longer than that and its a
1357 * dynamically loaded module we're going to try,
1358 * truncate the name before trying
1359 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001360 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001361 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001362 const struct filedescr *scan;
1363 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001364 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001365 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001366 break;
1367 else
1368 scan++;
1369 }
1370 if (scan->suffix != NULL) {
1371 /* yes, so truncate the name */
1372 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001373 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001374 buf[len] = '\0';
1375 }
1376 }
1377#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001378 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001379 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001380 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001381 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001382 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001383 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001384 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001385 if (fp != NULL) {
1386 if (case_ok(buf, len, namelen, name))
1387 break;
1388 else { /* continue search */
1389 fclose(fp);
1390 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001391 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001392 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001393#if defined(PYOS_OS2)
1394 /* restore the saved snapshot */
1395 strcpy(buf, saved_buf);
1396 len = saved_len;
1397 namelen = saved_namelen;
1398#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001399 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001400#if defined(PYOS_OS2)
1401 /* don't need/want the module name snapshot anymore */
1402 if (saved_buf)
1403 {
1404 free(saved_buf);
1405 saved_buf = NULL;
1406 }
1407#endif
Walter Dörwald3430d702002-06-17 10:43:59 +00001408 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001409 if (fp != NULL)
1410 break;
1411 }
1412 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001413 PyErr_Format(PyExc_ImportError,
1414 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001415 return NULL;
1416 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001417 *p_fp = fp;
1418 return fdp;
1419}
1420
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001421/* Helpers for main.c
1422 * Find the source file corresponding to a named module
1423 */
1424struct filedescr *
1425_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1426 size_t buflen, FILE **p_fp, PyObject **p_loader)
1427{
1428 return find_module((char *) name, (char *) name, path,
1429 buf, buflen, p_fp, p_loader);
1430}
1431
1432PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1433{
1434 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1435}
1436
Martin v. Löwis18e16552006-02-15 17:27:45 +00001437/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001438 * The arguments here are tricky, best shown by example:
1439 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1440 * ^ ^ ^ ^
1441 * |--------------------- buf ---------------------|
1442 * |------------------- len ------------------|
1443 * |------ name -------|
1444 * |----- namelen -----|
1445 * buf is the full path, but len only counts up to (& exclusive of) the
1446 * extension. name is the module name, also exclusive of extension.
1447 *
1448 * We've already done a successful stat() or fopen() on buf, so know that
1449 * there's some match, possibly case-insensitive.
1450 *
Tim Peters50d8d372001-02-28 05:34:27 +00001451 * case_ok() is to return 1 if there's a case-sensitive match for
1452 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1453 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001454 *
Tim Peters50d8d372001-02-28 05:34:27 +00001455 * case_ok() is used to implement case-sensitive import semantics even
1456 * on platforms with case-insensitive filesystems. It's trivial to implement
1457 * for case-sensitive filesystems. It's pretty much a cross-platform
1458 * nightmare for systems with case-insensitive filesystems.
1459 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001460
Tim Peters50d8d372001-02-28 05:34:27 +00001461/* First we may need a pile of platform-specific header files; the sequence
1462 * of #if's here should match the sequence in the body of case_ok().
1463 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001464#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001465#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001466
Tim Peters50d8d372001-02-28 05:34:27 +00001467#elif defined(DJGPP)
1468#include <dir.h>
1469
Jason Tishler7961aa62005-05-20 00:56:54 +00001470#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001471#include <sys/types.h>
1472#include <dirent.h>
1473
Andrew MacIntyred9400542002-02-26 11:41:34 +00001474#elif defined(PYOS_OS2)
1475#define INCL_DOS
1476#define INCL_DOSERRORS
1477#define INCL_NOPMAPI
1478#include <os2.h>
1479
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001480#elif defined(RISCOS)
1481#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001482#endif
1483
Guido van Rossum0980bd91998-02-13 17:18:36 +00001484static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001485case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001486{
Tim Peters50d8d372001-02-28 05:34:27 +00001487/* Pick a platform-specific implementation; the sequence of #if's here should
1488 * match the sequence just above.
1489 */
1490
Jason Tishler7961aa62005-05-20 00:56:54 +00001491/* MS_WINDOWS */
1492#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001493 WIN32_FIND_DATA data;
1494 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001495
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001496 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001497 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001498
Guido van Rossum0980bd91998-02-13 17:18:36 +00001499 h = FindFirstFile(buf, &data);
1500 if (h == INVALID_HANDLE_VALUE) {
1501 PyErr_Format(PyExc_NameError,
1502 "Can't find file for module %.100s\n(filename %.300s)",
1503 name, buf);
1504 return 0;
1505 }
1506 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001507 return strncmp(data.cFileName, name, namelen) == 0;
1508
1509/* DJGPP */
1510#elif defined(DJGPP)
1511 struct ffblk ffblk;
1512 int done;
1513
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001514 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001515 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001516
1517 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1518 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001519 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001520 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001521 name, buf);
1522 return 0;
1523 }
Tim Peters50d8d372001-02-28 05:34:27 +00001524 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001525
Jason Tishler7961aa62005-05-20 00:56:54 +00001526/* new-fangled macintosh (macosx) or Cygwin */
1527#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001528 DIR *dirp;
1529 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001530 char dirname[MAXPATHLEN + 1];
1531 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001532
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001533 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001534 return 1;
1535
Tim Petersd1e87a82001-03-01 18:12:00 +00001536 /* Copy the dir component into dirname; substitute "." if empty */
1537 if (dirlen <= 0) {
1538 dirname[0] = '.';
1539 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001540 }
1541 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001542 assert(dirlen <= MAXPATHLEN);
1543 memcpy(dirname, buf, dirlen);
1544 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001545 }
1546 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001547 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001548 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001549 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001550 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001551 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001552#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001553 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001554#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001555 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001556#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001557 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001558 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001559 (void)closedir(dirp);
1560 return 1; /* Found */
1561 }
1562 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001563 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001564 }
Tim Peters430f5d42001-03-01 01:30:56 +00001565 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001566
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001567/* RISC OS */
1568#elif defined(RISCOS)
1569 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1570 char buf2[MAXPATHLEN+2];
1571 char *nameWithExt = buf+len-namelen;
1572 int canonlen;
1573 os_error *e;
1574
1575 if (Py_GETENV("PYTHONCASEOK") != NULL)
1576 return 1;
1577
1578 /* workaround:
1579 append wildcard, otherwise case of filename wouldn't be touched */
1580 strcpy(buf2, buf);
1581 strcat(buf2, "*");
1582
1583 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1584 canonlen = MAXPATHLEN+1-canonlen;
1585 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1586 return 0;
1587 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1588 return 1; /* match */
1589
1590 return 0;
1591
Andrew MacIntyred9400542002-02-26 11:41:34 +00001592/* OS/2 */
1593#elif defined(PYOS_OS2)
1594 HDIR hdir = 1;
1595 ULONG srchcnt = 1;
1596 FILEFINDBUF3 ffbuf;
1597 APIRET rc;
1598
1599 if (getenv("PYTHONCASEOK") != NULL)
1600 return 1;
1601
1602 rc = DosFindFirst(buf,
1603 &hdir,
1604 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1605 &ffbuf, sizeof(ffbuf),
1606 &srchcnt,
1607 FIL_STANDARD);
1608 if (rc != NO_ERROR)
1609 return 0;
1610 return strncmp(ffbuf.achName, name, namelen) == 0;
1611
Tim Peters50d8d372001-02-28 05:34:27 +00001612/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1613#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001614 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001615
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001616#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001617}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001618
Guido van Rossum0980bd91998-02-13 17:18:36 +00001619
Guido van Rossum197346f1997-10-31 18:38:52 +00001620#ifdef HAVE_STAT
1621/* Helper to look for __init__.py or __init__.py[co] in potential package */
1622static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001623find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001624{
Tim Peters0f9431f2001-07-05 03:47:53 +00001625 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001626 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001627 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001628 struct stat statbuf;
1629
Tim Peters0f9431f2001-07-05 03:47:53 +00001630/* For calling case_ok(buf, len, namelen, name):
1631 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1632 * ^ ^ ^ ^
1633 * |--------------------- buf ---------------------|
1634 * |------------------- len ------------------|
1635 * |------ name -------|
1636 * |----- namelen -----|
1637 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001638 if (save_len + 13 >= MAXPATHLEN)
1639 return 0;
1640 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001641 pname = buf + i;
1642 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001643 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001644 if (case_ok(buf,
1645 save_len + 9, /* len("/__init__") */
1646 8, /* len("__init__") */
1647 pname)) {
1648 buf[save_len] = '\0';
1649 return 1;
1650 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001651 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001652 i += strlen(pname);
1653 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001654 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001655 if (case_ok(buf,
1656 save_len + 9, /* len("/__init__") */
1657 8, /* len("__init__") */
1658 pname)) {
1659 buf[save_len] = '\0';
1660 return 1;
1661 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001662 }
1663 buf[save_len] = '\0';
1664 return 0;
1665}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001666
1667#else
1668
1669#ifdef RISCOS
1670static int
1671find_init_module(buf)
1672 char *buf;
1673{
1674 int save_len = strlen(buf);
1675 int i = save_len;
1676
1677 if (save_len + 13 >= MAXPATHLEN)
1678 return 0;
1679 buf[i++] = SEP;
1680 strcpy(buf+i, "__init__/py");
1681 if (isfile(buf)) {
1682 buf[save_len] = '\0';
1683 return 1;
1684 }
1685
1686 if (Py_OptimizeFlag)
1687 strcpy(buf+i, "o");
1688 else
1689 strcpy(buf+i, "c");
1690 if (isfile(buf)) {
1691 buf[save_len] = '\0';
1692 return 1;
1693 }
1694 buf[save_len] = '\0';
1695 return 0;
1696}
1697#endif /*RISCOS*/
1698
Guido van Rossum197346f1997-10-31 18:38:52 +00001699#endif /* HAVE_STAT */
1700
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001701
Tim Petersdbd9ba62000-07-09 03:09:57 +00001702static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001703
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001704/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001705 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001706
Guido van Rossum79f25d91997-04-29 20:08:16 +00001707static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001708load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001709{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001710 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001711 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001712 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001713
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001714 /* First check that there's an open file (if we need one) */
1715 switch (type) {
1716 case PY_SOURCE:
1717 case PY_COMPILED:
1718 if (fp == NULL) {
1719 PyErr_Format(PyExc_ValueError,
1720 "file object required for import (type code %d)",
1721 type);
1722 return NULL;
1723 }
1724 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001725
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001726 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001727
1728 case PY_SOURCE:
1729 m = load_source_module(name, buf, fp);
1730 break;
1731
1732 case PY_COMPILED:
1733 m = load_compiled_module(name, buf, fp);
1734 break;
1735
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001736#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001737 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001738 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001739 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001740#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001741
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001742 case PKG_DIRECTORY:
1743 m = load_package(name, buf);
1744 break;
1745
1746 case C_BUILTIN:
1747 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001748 if (buf != NULL && buf[0] != '\0')
1749 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001750 if (type == C_BUILTIN)
1751 err = init_builtin(name);
1752 else
1753 err = PyImport_ImportFrozenModule(name);
1754 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001755 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001756 if (err == 0) {
1757 PyErr_Format(PyExc_ImportError,
1758 "Purported %s module %.200s not found",
1759 type == C_BUILTIN ?
1760 "builtin" : "frozen",
1761 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001762 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001763 }
1764 modules = PyImport_GetModuleDict();
1765 m = PyDict_GetItemString(modules, name);
1766 if (m == NULL) {
1767 PyErr_Format(
1768 PyExc_ImportError,
1769 "%s module %.200s not properly initialized",
1770 type == C_BUILTIN ?
1771 "builtin" : "frozen",
1772 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001773 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001774 }
1775 Py_INCREF(m);
1776 break;
1777
Just van Rossum52e14d62002-12-30 22:08:05 +00001778 case IMP_HOOK: {
1779 if (loader == NULL) {
1780 PyErr_SetString(PyExc_ImportError,
1781 "import hook without loader");
1782 return NULL;
1783 }
1784 m = PyObject_CallMethod(loader, "load_module", "s", name);
1785 break;
1786 }
1787
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001788 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001789 PyErr_Format(PyExc_ImportError,
1790 "Don't know how to import %.200s (type code %d)",
1791 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001792 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001793
1794 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001795
1796 return m;
1797}
1798
1799
1800/* Initialize a built-in module.
1801 Return 1 for succes, 0 if the module is not found, and -1 with
1802 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001803
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001804static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001805init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001806{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001807 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001808
Greg Ward201baee2001-10-04 14:52:06 +00001809 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001810 return 1;
1811
Guido van Rossum771c6c81997-10-31 18:37:24 +00001812 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001813 if (strcmp(name, p->name) == 0) {
1814 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001815 PyErr_Format(PyExc_ImportError,
1816 "Cannot re-init internal module %.200s",
1817 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001818 return -1;
1819 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001820 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001821 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001822 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001823 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001824 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001825 if (_PyImport_FixupExtension(name, name) == NULL)
1826 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001827 return 1;
1828 }
1829 }
1830 return 0;
1831}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001832
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001833
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001834/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001835
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001836static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001837find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001838{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001839 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001840
Guido van Rossum79f25d91997-04-29 20:08:16 +00001841 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001842 if (p->name == NULL)
1843 return NULL;
1844 if (strcmp(p->name, name) == 0)
1845 break;
1846 }
1847 return p;
1848}
1849
Guido van Rossum79f25d91997-04-29 20:08:16 +00001850static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001851get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001852{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001853 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001854 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001855
1856 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001857 PyErr_Format(PyExc_ImportError,
1858 "No such frozen object named %.200s",
1859 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001860 return NULL;
1861 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001862 if (p->code == NULL) {
1863 PyErr_Format(PyExc_ImportError,
1864 "Excluded frozen object named %.200s",
1865 name);
1866 return NULL;
1867 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001868 size = p->size;
1869 if (size < 0)
1870 size = -size;
1871 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001872}
1873
1874/* Initialize a frozen module.
1875 Return 1 for succes, 0 if the module is not found, and -1 with
1876 an exception set if the initialization failed.
1877 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001878
1879int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001880PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001881{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001882 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001883 PyObject *co;
1884 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001885 int ispackage;
1886 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001887
1888 if (p == NULL)
1889 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001890 if (p->code == NULL) {
1891 PyErr_Format(PyExc_ImportError,
1892 "Excluded frozen object named %.200s",
1893 name);
1894 return -1;
1895 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001896 size = p->size;
1897 ispackage = (size < 0);
1898 if (ispackage)
1899 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001900 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001901 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001902 name, ispackage ? " package" : "");
1903 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001904 if (co == NULL)
1905 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001906 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001907 PyErr_Format(PyExc_TypeError,
1908 "frozen object %.200s is not a code object",
1909 name);
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001910 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001911 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001912 if (ispackage) {
1913 /* Set __path__ to the package name */
1914 PyObject *d, *s;
1915 int err;
1916 m = PyImport_AddModule(name);
1917 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001918 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001919 d = PyModule_GetDict(m);
1920 s = PyString_InternFromString(name);
1921 if (s == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001922 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001923 err = PyDict_SetItemString(d, "__path__", s);
1924 Py_DECREF(s);
1925 if (err != 0)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001926 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001927 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001928 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001929 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001930 goto err_return;
1931 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001932 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001933 return 1;
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001934err_return:
1935 Py_DECREF(co);
1936 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001937}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001938
1939
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001940/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001941 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001942
Guido van Rossum79f25d91997-04-29 20:08:16 +00001943PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001944PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001945{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001946 PyObject *pname;
1947 PyObject *result;
1948
1949 pname = PyString_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001950 if (pname == NULL)
1951 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001952 result = PyImport_Import(pname);
1953 Py_DECREF(pname);
1954 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001955}
1956
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001957/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001958static PyObject *get_parent(PyObject *globals, char *buf,
1959 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001960static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001961 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001962static int mark_miss(char *name);
1963static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001964 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001965static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001966
1967/* The Magnum Opus of dotted-name import :-) */
1968
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001969static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001970import_module_level(char *name, PyObject *globals, PyObject *locals,
1971 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001972{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001973 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00001974 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001975 PyObject *parent, *head, *next, *tail;
1976
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001977 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001978 if (parent == NULL)
1979 return NULL;
1980
1981 head = load_next(parent, Py_None, &name, buf, &buflen);
1982 if (head == NULL)
1983 return NULL;
1984
1985 tail = head;
1986 Py_INCREF(tail);
1987 while (name) {
1988 next = load_next(tail, tail, &name, buf, &buflen);
1989 Py_DECREF(tail);
1990 if (next == NULL) {
1991 Py_DECREF(head);
1992 return NULL;
1993 }
1994 tail = next;
1995 }
Thomas Wouters8ddab272006-04-04 16:17:02 +00001996 if (tail == Py_None) {
1997 /* If tail is Py_None, both get_parent and load_next found
1998 an empty module name: someone called __import__("") or
1999 doctored faulty bytecode */
Thomas Wouters4bdaa272006-04-05 13:39:37 +00002000 Py_DECREF(tail);
2001 Py_DECREF(head);
Thomas Wouters8ddab272006-04-04 16:17:02 +00002002 PyErr_SetString(PyExc_ValueError,
2003 "Empty module name");
2004 return NULL;
2005 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002006
2007 if (fromlist != NULL) {
2008 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2009 fromlist = NULL;
2010 }
2011
2012 if (fromlist == NULL) {
2013 Py_DECREF(tail);
2014 return head;
2015 }
2016
2017 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002018 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002019 Py_DECREF(tail);
2020 return NULL;
2021 }
2022
2023 return tail;
2024}
2025
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002026/* For DLL compatibility */
2027#undef PyImport_ImportModuleEx
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002028PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002029PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
2030 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002031{
2032 PyObject *result;
2033 lock_import();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002034 result = import_module_level(name, globals, locals, fromlist, -1);
2035 if (unlock_import() < 0) {
2036 Py_XDECREF(result);
2037 PyErr_SetString(PyExc_RuntimeError,
2038 "not holding the import lock");
2039 return NULL;
2040 }
2041 return result;
2042}
2043#define PyImport_ImportModuleEx(n, g, l, f) \
2044 PyImport_ImportModuleLevel(n, g, l, f, -1);
2045
2046PyObject *
2047PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
2048 PyObject *fromlist, int level)
2049{
2050 PyObject *result;
2051 lock_import();
2052 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002053 if (unlock_import() < 0) {
2054 Py_XDECREF(result);
2055 PyErr_SetString(PyExc_RuntimeError,
2056 "not holding the import lock");
2057 return NULL;
2058 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002059 return result;
2060}
2061
Fred Drake87590902004-05-28 20:21:36 +00002062/* Return the package that an import is being performed in. If globals comes
2063 from the module foo.bar.bat (not itself a package), this returns the
2064 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Georg Brandl5f6861d2006-05-28 21:57:35 +00002065 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002066
2067 The *name* of the returned package is returned in buf, with the length of
2068 the name in *p_buflen.
2069
2070 If globals doesn't come from a package or a module in a package, or a
2071 corresponding entry is not found in sys.modules, Py_None is returned.
2072*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002073static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002074get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002075{
2076 static PyObject *namestr = NULL;
2077 static PyObject *pathstr = NULL;
2078 PyObject *modname, *modpath, *modules, *parent;
2079
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002080 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002081 return Py_None;
2082
2083 if (namestr == NULL) {
2084 namestr = PyString_InternFromString("__name__");
2085 if (namestr == NULL)
2086 return NULL;
2087 }
2088 if (pathstr == NULL) {
2089 pathstr = PyString_InternFromString("__path__");
2090 if (pathstr == NULL)
2091 return NULL;
2092 }
2093
2094 *buf = '\0';
2095 *p_buflen = 0;
2096 modname = PyDict_GetItem(globals, namestr);
2097 if (modname == NULL || !PyString_Check(modname))
2098 return Py_None;
2099
2100 modpath = PyDict_GetItem(globals, pathstr);
2101 if (modpath != NULL) {
Martin v. Löwis725507b2006-03-07 12:08:51 +00002102 Py_ssize_t len = PyString_GET_SIZE(modname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002103 if (len > MAXPATHLEN) {
2104 PyErr_SetString(PyExc_ValueError,
2105 "Module name too long");
2106 return NULL;
2107 }
2108 strcpy(buf, PyString_AS_STRING(modname));
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002109 }
2110 else {
2111 char *start = PyString_AS_STRING(modname);
2112 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002113 size_t len;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002114 if (lastdot == NULL && level > 0) {
2115 PyErr_SetString(PyExc_ValueError,
2116 "Relative importpath too deep");
2117 return NULL;
2118 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002119 if (lastdot == NULL)
2120 return Py_None;
2121 len = lastdot - start;
2122 if (len >= MAXPATHLEN) {
2123 PyErr_SetString(PyExc_ValueError,
2124 "Module name too long");
2125 return NULL;
2126 }
2127 strncpy(buf, start, len);
2128 buf[len] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002129 }
2130
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002131 while (--level > 0) {
2132 char *dot = strrchr(buf, '.');
2133 if (dot == NULL) {
2134 PyErr_SetString(PyExc_ValueError,
2135 "Relative importpath too deep");
2136 return NULL;
2137 }
2138 *dot = '\0';
2139 }
2140 *p_buflen = strlen(buf);
2141
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002142 modules = PyImport_GetModuleDict();
2143 parent = PyDict_GetItemString(modules, buf);
2144 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002145 PyErr_Format(PyExc_SystemError,
2146 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002147 return parent;
2148 /* We expect, but can't guarantee, if parent != None, that:
2149 - parent.__name__ == buf
2150 - parent.__dict__ is globals
2151 If this is violated... Who cares? */
2152}
2153
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002154/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002155static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002156load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002157 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002158{
2159 char *name = *p_name;
2160 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002161 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002162 char *p;
2163 PyObject *result;
2164
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002165 if (strlen(name) == 0) {
Thomas Wouters8ddab272006-04-04 16:17:02 +00002166 /* completely empty module name should only happen in
2167 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002168 Py_INCREF(mod);
2169 *p_name = NULL;
2170 return mod;
2171 }
2172
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002173 if (dot == NULL) {
2174 *p_name = NULL;
2175 len = strlen(name);
2176 }
2177 else {
2178 *p_name = dot+1;
2179 len = dot-name;
2180 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002181 if (len == 0) {
2182 PyErr_SetString(PyExc_ValueError,
2183 "Empty module name");
2184 return NULL;
2185 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002186
2187 p = buf + *p_buflen;
2188 if (p != buf)
2189 *p++ = '.';
2190 if (p+len-buf >= MAXPATHLEN) {
2191 PyErr_SetString(PyExc_ValueError,
2192 "Module name too long");
2193 return NULL;
2194 }
2195 strncpy(p, name, len);
2196 p[len] = '\0';
2197 *p_buflen = p+len-buf;
2198
2199 result = import_submodule(mod, p, buf);
2200 if (result == Py_None && altmod != mod) {
2201 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002202 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002203 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002204 if (result != NULL && result != Py_None) {
2205 if (mark_miss(buf) != 0) {
2206 Py_DECREF(result);
2207 return NULL;
2208 }
2209 strncpy(buf, name, len);
2210 buf[len] = '\0';
2211 *p_buflen = len;
2212 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002213 }
2214 if (result == NULL)
2215 return NULL;
2216
2217 if (result == Py_None) {
2218 Py_DECREF(result);
2219 PyErr_Format(PyExc_ImportError,
2220 "No module named %.200s", name);
2221 return NULL;
2222 }
2223
2224 return result;
2225}
2226
2227static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002228mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002229{
2230 PyObject *modules = PyImport_GetModuleDict();
2231 return PyDict_SetItemString(modules, name, Py_None);
2232}
2233
2234static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002235ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002236 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002237{
2238 int i;
2239
2240 if (!PyObject_HasAttrString(mod, "__path__"))
2241 return 1;
2242
2243 for (i = 0; ; i++) {
2244 PyObject *item = PySequence_GetItem(fromlist, i);
2245 int hasit;
2246 if (item == NULL) {
2247 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2248 PyErr_Clear();
2249 return 1;
2250 }
2251 return 0;
2252 }
2253 if (!PyString_Check(item)) {
2254 PyErr_SetString(PyExc_TypeError,
2255 "Item in ``from list'' not a string");
2256 Py_DECREF(item);
2257 return 0;
2258 }
2259 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002260 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002261 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002262 /* See if the package defines __all__ */
2263 if (recursive)
2264 continue; /* Avoid endless recursion */
2265 all = PyObject_GetAttrString(mod, "__all__");
2266 if (all == NULL)
2267 PyErr_Clear();
2268 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002269 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002270 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002271 if (!ret)
2272 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002273 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002274 continue;
2275 }
2276 hasit = PyObject_HasAttr(mod, item);
2277 if (!hasit) {
2278 char *subname = PyString_AS_STRING(item);
2279 PyObject *submod;
2280 char *p;
2281 if (buflen + strlen(subname) >= MAXPATHLEN) {
2282 PyErr_SetString(PyExc_ValueError,
2283 "Module name too long");
2284 Py_DECREF(item);
2285 return 0;
2286 }
2287 p = buf + buflen;
2288 *p++ = '.';
2289 strcpy(p, subname);
2290 submod = import_submodule(mod, subname, buf);
2291 Py_XDECREF(submod);
2292 if (submod == NULL) {
2293 Py_DECREF(item);
2294 return 0;
2295 }
2296 }
2297 Py_DECREF(item);
2298 }
2299
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002300 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002301}
2302
Neil Schemenauer00b09662003-06-16 21:03:07 +00002303static int
2304add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2305 PyObject *modules)
2306{
2307 if (mod == Py_None)
2308 return 1;
2309 /* Irrespective of the success of this load, make a
2310 reference to it in the parent package module. A copy gets
2311 saved in the modules dictionary under the full name, so get a
2312 reference from there, if need be. (The exception is when the
2313 load failed with a SyntaxError -- then there's no trace in
2314 sys.modules. In that case, of course, do nothing extra.) */
2315 if (submod == NULL) {
2316 submod = PyDict_GetItemString(modules, fullname);
2317 if (submod == NULL)
2318 return 1;
2319 }
2320 if (PyModule_Check(mod)) {
2321 /* We can't use setattr here since it can give a
2322 * spurious warning if the submodule name shadows a
2323 * builtin name */
2324 PyObject *dict = PyModule_GetDict(mod);
2325 if (!dict)
2326 return 0;
2327 if (PyDict_SetItemString(dict, subname, submod) < 0)
2328 return 0;
2329 }
2330 else {
2331 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2332 return 0;
2333 }
2334 return 1;
2335}
2336
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002337static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002338import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002339{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002340 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002341 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002342
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002343 /* Require:
2344 if mod == None: subname == fullname
2345 else: mod.__name__ + "." + subname == fullname
2346 */
2347
Tim Peters50d8d372001-02-28 05:34:27 +00002348 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002349 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002350 }
2351 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002352 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002353 char buf[MAXPATHLEN+1];
2354 struct filedescr *fdp;
2355 FILE *fp = NULL;
2356
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002357 if (mod == Py_None)
2358 path = NULL;
2359 else {
2360 path = PyObject_GetAttrString(mod, "__path__");
2361 if (path == NULL) {
2362 PyErr_Clear();
2363 Py_INCREF(Py_None);
2364 return Py_None;
2365 }
2366 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002367
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002368 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002369 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2370 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002371 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002372 if (fdp == NULL) {
2373 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2374 return NULL;
2375 PyErr_Clear();
2376 Py_INCREF(Py_None);
2377 return Py_None;
2378 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002379 m = load_module(fullname, fp, buf, fdp->type, loader);
2380 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002381 if (fp)
2382 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002383 if (!add_submodule(mod, m, fullname, subname, modules)) {
2384 Py_XDECREF(m);
2385 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002386 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002387 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002388
2389 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002390}
2391
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002392
2393/* Re-import a module of any kind and return its module object, WITH
2394 INCREMENTED REFERENCE COUNT */
2395
Guido van Rossum79f25d91997-04-29 20:08:16 +00002396PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002397PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002398{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002399 PyObject *modules = PyImport_GetModuleDict();
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002400 PyObject *path = NULL, *loader = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002401 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002402 char buf[MAXPATHLEN+1];
2403 struct filedescr *fdp;
2404 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002405 PyObject *newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002406
Guido van Rossum79f25d91997-04-29 20:08:16 +00002407 if (m == NULL || !PyModule_Check(m)) {
2408 PyErr_SetString(PyExc_TypeError,
2409 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002410 return NULL;
2411 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002412 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002413 if (name == NULL)
2414 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002415 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002416 PyErr_Format(PyExc_ImportError,
2417 "reload(): module %.200s not in sys.modules",
2418 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002419 return NULL;
2420 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002421 subname = strrchr(name, '.');
2422 if (subname == NULL)
2423 subname = name;
2424 else {
2425 PyObject *parentname, *parent;
2426 parentname = PyString_FromStringAndSize(name, (subname-name));
2427 if (parentname == NULL)
2428 return NULL;
2429 parent = PyDict_GetItem(modules, parentname);
2430 if (parent == NULL) {
2431 PyErr_Format(PyExc_ImportError,
2432 "reload(): parent %.200s not in sys.modules",
Georg Brandl0c55f292005-09-14 06:56:20 +00002433 PyString_AS_STRING(parentname));
2434 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002435 return NULL;
2436 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002437 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002438 subname++;
2439 path = PyObject_GetAttrString(parent, "__path__");
2440 if (path == NULL)
2441 PyErr_Clear();
2442 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002443 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002444 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002445 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002446
2447 if (fdp == NULL) {
2448 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002449 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002450 }
2451
2452 newm = load_module(name, fp, buf, fdp->type, loader);
2453 Py_XDECREF(loader);
2454
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002455 if (fp)
2456 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002457 if (newm == NULL) {
2458 /* load_module probably removed name from modules because of
2459 * the error. Put back the original module object. We're
2460 * going to return NULL in this case regardless of whether
2461 * replacing name succeeds, so the return value is ignored.
2462 */
2463 PyDict_SetItemString(modules, name, m);
2464 }
2465 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002466}
2467
2468
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002469/* Higher-level import emulator which emulates the "import" statement
2470 more accurately -- it invokes the __import__() function from the
2471 builtins of the current globals. This means that the import is
2472 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002473 environment, e.g. by "rexec".
2474 A dummy list ["__doc__"] is passed as the 4th argument so that
2475 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2476 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002477
2478PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002479PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002480{
2481 static PyObject *silly_list = NULL;
2482 static PyObject *builtins_str = NULL;
2483 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002484 PyObject *globals = NULL;
2485 PyObject *import = NULL;
2486 PyObject *builtins = NULL;
2487 PyObject *r = NULL;
2488
2489 /* Initialize constant string objects */
2490 if (silly_list == NULL) {
2491 import_str = PyString_InternFromString("__import__");
2492 if (import_str == NULL)
2493 return NULL;
2494 builtins_str = PyString_InternFromString("__builtins__");
2495 if (builtins_str == NULL)
2496 return NULL;
2497 silly_list = Py_BuildValue("[s]", "__doc__");
2498 if (silly_list == NULL)
2499 return NULL;
2500 }
2501
2502 /* Get the builtins from current globals */
2503 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002504 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002505 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002506 builtins = PyObject_GetItem(globals, builtins_str);
2507 if (builtins == NULL)
2508 goto err;
2509 }
2510 else {
2511 /* No globals -- use standard builtins, and fake globals */
2512 PyErr_Clear();
2513
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002514 builtins = PyImport_ImportModuleLevel("__builtin__",
2515 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002516 if (builtins == NULL)
2517 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002518 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2519 if (globals == NULL)
2520 goto err;
2521 }
2522
2523 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002524 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002525 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002526 if (import == NULL)
2527 PyErr_SetObject(PyExc_KeyError, import_str);
2528 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002529 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002530 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002531 if (import == NULL)
2532 goto err;
2533
2534 /* Call the _import__ function with the proper argument list */
Georg Brandl684fd0c2006-05-25 19:15:31 +00002535 r = PyObject_CallFunctionObjArgs(import, module_name, globals,
2536 globals, silly_list, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002537
2538 err:
2539 Py_XDECREF(globals);
2540 Py_XDECREF(builtins);
2541 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002542
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002543 return r;
2544}
2545
2546
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002547/* Module 'imp' provides Python access to the primitives used for
2548 importing modules.
2549*/
2550
Guido van Rossum79f25d91997-04-29 20:08:16 +00002551static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002552imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002553{
2554 char buf[4];
2555
Guido van Rossum96774c12000-05-01 20:19:08 +00002556 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2557 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2558 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2559 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002560
Guido van Rossum79f25d91997-04-29 20:08:16 +00002561 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002562}
2563
Guido van Rossum79f25d91997-04-29 20:08:16 +00002564static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002565imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002566{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002567 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002568 struct filedescr *fdp;
2569
Guido van Rossum79f25d91997-04-29 20:08:16 +00002570 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002571 if (list == NULL)
2572 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002573 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2574 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002575 fdp->suffix, fdp->mode, fdp->type);
2576 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002577 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002578 return NULL;
2579 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002580 if (PyList_Append(list, item) < 0) {
2581 Py_DECREF(list);
2582 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002583 return NULL;
2584 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002585 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002586 }
2587 return list;
2588}
2589
Guido van Rossum79f25d91997-04-29 20:08:16 +00002590static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002591call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002592{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002593 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002594 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002595 struct filedescr *fdp;
2596 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002597 FILE *fp = NULL;
2598
2599 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002600 if (path == Py_None)
2601 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002602 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002603 if (fdp == NULL)
2604 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002605 if (fp != NULL) {
2606 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2607 if (fob == NULL) {
2608 fclose(fp);
2609 return NULL;
2610 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002611 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002612 else {
2613 fob = Py_None;
2614 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002615 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002616 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002617 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002618 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002619 return ret;
2620}
2621
Guido van Rossum79f25d91997-04-29 20:08:16 +00002622static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002623imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002624{
2625 char *name;
2626 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002627 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002628 return NULL;
2629 return call_find_module(name, path);
2630}
2631
2632static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002633imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002634{
2635 char *name;
2636 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002637 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002638 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002639 return NULL;
2640 ret = init_builtin(name);
2641 if (ret < 0)
2642 return NULL;
2643 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002644 Py_INCREF(Py_None);
2645 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002646 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002647 m = PyImport_AddModule(name);
2648 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002649 return m;
2650}
2651
Guido van Rossum79f25d91997-04-29 20:08:16 +00002652static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002653imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002654{
2655 char *name;
2656 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002657 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002658 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002659 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002660 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002661 if (ret < 0)
2662 return NULL;
2663 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002664 Py_INCREF(Py_None);
2665 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002666 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002667 m = PyImport_AddModule(name);
2668 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002669 return m;
2670}
2671
Guido van Rossum79f25d91997-04-29 20:08:16 +00002672static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002673imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002674{
2675 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002676
Guido van Rossum43713e52000-02-29 13:59:29 +00002677 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002678 return NULL;
2679 return get_frozen_object(name);
2680}
2681
Guido van Rossum79f25d91997-04-29 20:08:16 +00002682static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002683imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002684{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002685 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002686 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002687 return NULL;
Guido van Rossum50ee94f2002-04-09 18:00:58 +00002688 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002689}
2690
Guido van Rossum79f25d91997-04-29 20:08:16 +00002691static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002692imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002693{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002694 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002695 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002696 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002697 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002698 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002699 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002700}
2701
2702static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002703get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002704{
2705 FILE *fp;
2706 if (fob == NULL) {
Tim Peters86c7d2f2004-08-01 23:24:21 +00002707 if (mode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002708 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002709 fp = fopen(pathname, mode);
2710 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002711 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002712 }
2713 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002714 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002715 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002716 PyErr_SetString(PyExc_ValueError,
2717 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002718 }
2719 return fp;
2720}
2721
Guido van Rossum79f25d91997-04-29 20:08:16 +00002722static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002723imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002724{
2725 char *name;
2726 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002727 PyObject *fob = NULL;
2728 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002729 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002730 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002731 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002732 return NULL;
2733 fp = get_file(pathname, fob, "rb");
2734 if (fp == NULL)
2735 return NULL;
2736 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002737 if (fob == NULL)
2738 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002739 return m;
2740}
2741
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002742#ifdef HAVE_DYNAMIC_LOADING
2743
Guido van Rossum79f25d91997-04-29 20:08:16 +00002744static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002745imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002746{
2747 char *name;
2748 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002749 PyObject *fob = NULL;
2750 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002751 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002752 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002753 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002754 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002755 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002756 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002757 if (fp == NULL)
2758 return NULL;
2759 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002760 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002761 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002762}
2763
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002764#endif /* HAVE_DYNAMIC_LOADING */
2765
Guido van Rossum79f25d91997-04-29 20:08:16 +00002766static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002767imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002768{
2769 char *name;
2770 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002771 PyObject *fob = NULL;
2772 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002773 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002774 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002775 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002776 return NULL;
2777 fp = get_file(pathname, fob, "r");
2778 if (fp == NULL)
2779 return NULL;
2780 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002781 if (fob == NULL)
2782 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002783 return m;
2784}
2785
Guido van Rossum79f25d91997-04-29 20:08:16 +00002786static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002787imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002788{
2789 char *name;
2790 PyObject *fob;
2791 char *pathname;
2792 char *suffix; /* Unused */
2793 char *mode;
2794 int type;
2795 FILE *fp;
2796
Guido van Rossum43713e52000-02-29 13:59:29 +00002797 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002798 &name, &fob, &pathname,
2799 &suffix, &mode, &type))
2800 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002801 if (*mode) {
2802 /* Mode must start with 'r' or 'U' and must not contain '+'.
2803 Implicit in this test is the assumption that the mode
2804 may contain other modifiers like 'b' or 't'. */
2805
2806 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002807 PyErr_Format(PyExc_ValueError,
2808 "invalid file open mode %.200s", mode);
2809 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002810 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002811 }
2812 if (fob == Py_None)
2813 fp = NULL;
2814 else {
2815 if (!PyFile_Check(fob)) {
2816 PyErr_SetString(PyExc_ValueError,
2817 "load_module arg#2 should be a file or None");
2818 return NULL;
2819 }
2820 fp = get_file(pathname, fob, mode);
2821 if (fp == NULL)
2822 return NULL;
2823 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002824 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002825}
2826
2827static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002828imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002829{
2830 char *name;
2831 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002832 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002833 return NULL;
2834 return load_package(name, pathname);
2835}
2836
2837static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002838imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002839{
2840 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002841 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002842 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002843 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002844}
2845
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002846/* Doc strings */
2847
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002848PyDoc_STRVAR(doc_imp,
2849"This module provides the components needed to build your own\n\
2850__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002851
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002852PyDoc_STRVAR(doc_find_module,
2853"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002854Search for a module. If path is omitted or None, search for a\n\
2855built-in, frozen or special module and continue search in sys.path.\n\
2856The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002857package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002858
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002859PyDoc_STRVAR(doc_load_module,
2860"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002861Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002862The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002863
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002864PyDoc_STRVAR(doc_get_magic,
2865"get_magic() -> string\n\
2866Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002867
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002868PyDoc_STRVAR(doc_get_suffixes,
2869"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002870Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002871that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002872
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002873PyDoc_STRVAR(doc_new_module,
2874"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002875Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002876The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002877
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002878PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00002879"lock_held() -> boolean\n\
2880Return True if the import lock is currently held, else False.\n\
2881On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00002882
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002883PyDoc_STRVAR(doc_acquire_lock,
2884"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00002885Acquires the interpreter's import lock for the current thread.\n\
2886This lock should be used by import hooks to ensure thread-safety\n\
2887when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002888On platforms without threads, this function does nothing.");
2889
2890PyDoc_STRVAR(doc_release_lock,
2891"release_lock() -> None\n\
2892Release the interpreter's import lock.\n\
2893On platforms without threads, this function does nothing.");
2894
Guido van Rossum79f25d91997-04-29 20:08:16 +00002895static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002896 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
2897 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
2898 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
2899 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
2900 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
2901 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
2902 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
2903 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002904 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00002905 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
2906 {"init_builtin", imp_init_builtin, METH_VARARGS},
2907 {"init_frozen", imp_init_frozen, METH_VARARGS},
2908 {"is_builtin", imp_is_builtin, METH_VARARGS},
2909 {"is_frozen", imp_is_frozen, METH_VARARGS},
2910 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002911#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00002912 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002913#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002914 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00002915 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002916 {NULL, NULL} /* sentinel */
2917};
2918
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002919static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002920setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002921{
2922 PyObject *v;
2923 int err;
2924
2925 v = PyInt_FromLong((long)value);
2926 err = PyDict_SetItemString(d, name, v);
2927 Py_XDECREF(v);
2928 return err;
2929}
2930
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00002931typedef struct {
2932 PyObject_HEAD
2933} NullImporter;
2934
2935static int
2936NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
2937{
2938 char *path;
2939
2940 if (!_PyArg_NoKeywords("NullImporter()", kwds))
2941 return -1;
2942
2943 if (!PyArg_ParseTuple(args, "s:NullImporter",
2944 &path))
2945 return -1;
2946
2947 if (strlen(path) == 0) {
2948 PyErr_SetString(PyExc_ImportError, "empty pathname");
2949 return -1;
2950 } else {
2951#ifndef RISCOS
2952 struct stat statbuf;
2953 int rv;
2954
2955 rv = stat(path, &statbuf);
2956 if (rv == 0) {
2957 /* it exists */
2958 if (S_ISDIR(statbuf.st_mode)) {
2959 /* it's a directory */
2960 PyErr_SetString(PyExc_ImportError,
2961 "existing directory");
2962 return -1;
2963 }
2964 }
2965#else
2966 if (object_exists(path)) {
2967 /* it exists */
2968 if (isdir(path)) {
2969 /* it's a directory */
2970 PyErr_SetString(PyExc_ImportError,
2971 "existing directory");
2972 return -1;
2973 }
2974 }
2975#endif
2976 }
2977 return 0;
2978}
2979
2980static PyObject *
2981NullImporter_find_module(NullImporter *self, PyObject *args)
2982{
2983 Py_RETURN_NONE;
2984}
2985
2986static PyMethodDef NullImporter_methods[] = {
2987 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
2988 "Always return None"
2989 },
2990 {NULL} /* Sentinel */
2991};
2992
2993
2994static PyTypeObject NullImporterType = {
2995 PyObject_HEAD_INIT(NULL)
2996 0, /*ob_size*/
2997 "imp.NullImporter", /*tp_name*/
2998 sizeof(NullImporter), /*tp_basicsize*/
2999 0, /*tp_itemsize*/
3000 0, /*tp_dealloc*/
3001 0, /*tp_print*/
3002 0, /*tp_getattr*/
3003 0, /*tp_setattr*/
3004 0, /*tp_compare*/
3005 0, /*tp_repr*/
3006 0, /*tp_as_number*/
3007 0, /*tp_as_sequence*/
3008 0, /*tp_as_mapping*/
3009 0, /*tp_hash */
3010 0, /*tp_call*/
3011 0, /*tp_str*/
3012 0, /*tp_getattro*/
3013 0, /*tp_setattro*/
3014 0, /*tp_as_buffer*/
3015 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3016 "Null importer object", /* tp_doc */
3017 0, /* tp_traverse */
3018 0, /* tp_clear */
3019 0, /* tp_richcompare */
3020 0, /* tp_weaklistoffset */
3021 0, /* tp_iter */
3022 0, /* tp_iternext */
3023 NullImporter_methods, /* tp_methods */
3024 0, /* tp_members */
3025 0, /* tp_getset */
3026 0, /* tp_base */
3027 0, /* tp_dict */
3028 0, /* tp_descr_get */
3029 0, /* tp_descr_set */
3030 0, /* tp_dictoffset */
3031 (initproc)NullImporter_init, /* tp_init */
3032 0, /* tp_alloc */
3033 PyType_GenericNew /* tp_new */
3034};
3035
3036
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003037PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003038initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003039{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003040 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003041
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003042 if (PyType_Ready(&NullImporterType) < 0)
3043 goto failure;
3044
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003045 m = Py_InitModule4("imp", imp_methods, doc_imp,
3046 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003047 if (m == NULL)
3048 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003049 d = PyModule_GetDict(m);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00003050 if (d == NULL)
3051 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003052
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003053 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3054 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3055 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3056 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3057 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3058 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3059 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3060 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003061 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003062 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003063
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003064 Py_INCREF(&NullImporterType);
3065 PyModule_AddObject(m, "NullImporter", (PyObject *)&NullImporterType);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003066 failure:
3067 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003068}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003069
3070
Guido van Rossumb18618d2000-05-03 23:44:39 +00003071/* API for embedding applications that want to add their own entries
3072 to the table of built-in modules. This should normally be called
3073 *before* Py_Initialize(). When the table resize fails, -1 is
3074 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003075
3076 After a similar function by Just van Rossum. */
3077
3078int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003079PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003080{
3081 static struct _inittab *our_copy = NULL;
3082 struct _inittab *p;
3083 int i, n;
3084
3085 /* Count the number of entries in both tables */
3086 for (n = 0; newtab[n].name != NULL; n++)
3087 ;
3088 if (n == 0)
3089 return 0; /* Nothing to do */
3090 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3091 ;
3092
3093 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003094 p = our_copy;
3095 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003096 if (p == NULL)
3097 return -1;
3098
3099 /* Copy the tables into the new memory */
3100 if (our_copy != PyImport_Inittab)
3101 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3102 PyImport_Inittab = our_copy = p;
3103 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3104
3105 return 0;
3106}
3107
3108/* Shorthand to add a single entry given a name and a function */
3109
3110int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003111PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003112{
3113 struct _inittab newtab[2];
3114
3115 memset(newtab, '\0', sizeof newtab);
3116
3117 newtab[0].name = name;
3118 newtab[0].initfunc = initfunc;
3119
3120 return PyImport_ExtendInittab(newtab);
3121}
Anthony Baxterac6bd462006-04-13 02:06:09 +00003122
3123#ifdef __cplusplus
3124}
3125#endif