blob: 71ec20a252e40363305bd0434eb07663d5369fb8 [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"
Kristján Valur Jónsson67387fb2007-04-25 00:17:39 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Neal Norwitzadb69fc2005-12-17 20:54:49 +00008#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00009#include "pythonrun.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000011#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000012#include "code.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000013#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000014#include "eval.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000015#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000016#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000017
Guido van Rossum55a83382000-09-20 20:31:38 +000018#ifdef HAVE_FCNTL_H
19#include <fcntl.h>
20#endif
Anthony Baxterac6bd462006-04-13 02:06:09 +000021#ifdef __cplusplus
22extern "C" {
23#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000024
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000025extern time_t PyOS_GetLastModificationTime(char *, FILE *);
26 /* In getmtime.c */
Guido van Rossum21d335e1993-10-15 13:01:11 +000027
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000028/* Magic word to reject .pyc files generated by other Python versions.
29 It should change for each incompatible change to the bytecode.
30
31 The value of CR and LF is incorporated so if you ever read or write
Guido van Rossum7faeab31995-07-07 22:50:36 +000032 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000033 Apple MPW compiler swaps their values, botching string constants.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000034
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000035 The magic numbers must be spaced apart atleast 2 values, as the
36 -U interpeter flag will cause MAGIC+1 being used. They have been
37 odd numbers for some time now.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000038
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000039 There were a variety of old schemes for setting the magic number.
40 The current working scheme is to increment the previous value by
41 10.
Guido van Rossumf6894922002-08-31 15:16:14 +000042
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000043 Known values:
44 Python 1.5: 20121
45 Python 1.5.1: 20121
46 Python 1.5.2: 20121
Skip Montanaro4ec3c262006-03-25 14:12:03 +000047 Python 1.6: 50428
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000048 Python 2.0: 50823
49 Python 2.0.1: 50823
50 Python 2.1: 60202
51 Python 2.1.1: 60202
52 Python 2.1.2: 60202
53 Python 2.2: 60717
Neal Norwitz7fdcb412002-06-14 01:07:39 +000054 Python 2.3a0: 62011
Michael W. Hudsondd32a912002-08-15 14:59:02 +000055 Python 2.3a0: 62021
Guido van Rossumf6894922002-08-31 15:16:14 +000056 Python 2.3a0: 62011 (!)
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000057 Python 2.4a0: 62041
Raymond Hettingerfd2d1f72004-08-23 23:37:48 +000058 Python 2.4a3: 62051
Raymond Hettinger2c31a052004-09-22 18:44:21 +000059 Python 2.4b1: 62061
Michael W. Hudsondf888462005-06-03 14:41:55 +000060 Python 2.5a0: 62071
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000061 Python 2.5a0: 62081 (ast-branch)
Guido van Rossumc2e20742006-02-27 22:32:47 +000062 Python 2.5a0: 62091 (with)
Guido van Rossumf6694362006-03-10 02:28:35 +000063 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode)
Neal Norwitz0d62a062006-07-30 06:53:31 +000064 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
65 Python 2.5b3: 62111 (fix wrong code: x += yield)
Neal Norwitz9a70f952006-08-04 05:12:19 +000066 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
67 storing constants that should have been removed)
Neal Norwitzdac090d2006-09-05 03:53:08 +000068 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp)
Raymond Hettingereffde122007-12-18 18:26:18 +000069 Python 2.6a0: 62151 (peephole optimizations and STORE_MAP opcode)
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000070.
Tim Peters36515e22001-11-18 04:06:29 +000071*/
Raymond Hettingereffde122007-12-18 18:26:18 +000072#define MAGIC (62151 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000073
Guido van Rossum96774c12000-05-01 20:19:08 +000074/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000075 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000076 compiler works which are enabled by command line switches. */
77static long pyc_magic = MAGIC;
78
Guido van Rossum25ce5661997-08-02 03:10:38 +000079/* See _PyImport_FixupExtension() below */
80static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000081
Guido van Rossum771c6c81997-10-31 18:37:24 +000082/* This table is defined in config.c: */
83extern struct _inittab _PyImport_Inittab[];
84
85struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000086
Guido van Rossumed1170e1999-12-20 21:23:41 +000087/* these tables define the module suffixes that Python recognizes */
88struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +000089
90#ifdef RISCOS
91static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000092 {"/py", "U", PY_SOURCE},
Guido van Rossum48a680c2001-03-02 06:34:14 +000093 {"/pyc", "rb", PY_COMPILED},
94 {0, 0}
95};
96#else
Guido van Rossumed1170e1999-12-20 21:23:41 +000097static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000098 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000099#ifdef MS_WINDOWS
Jack Jansenc88da1f2002-05-28 10:58:19 +0000100 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +0000101#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000102 {".pyc", "rb", PY_COMPILED},
103 {0, 0}
104};
Guido van Rossum48a680c2001-03-02 06:34:14 +0000105#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000106
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000107
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000108/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000109
110void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000111_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000112{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000113 const struct filedescr *scan;
114 struct filedescr *filetab;
115 int countD = 0;
116 int countS = 0;
117
118 /* prepare _PyImport_Filetab: copy entries from
119 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
120 */
Georg Brandladd36e52007-08-23 18:08:06 +0000121#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000122 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
123 ++countD;
Georg Brandladd36e52007-08-23 18:08:06 +0000124#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000125 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
126 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000127 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Neal Norwitze1fdb322006-07-21 05:32:28 +0000128 if (filetab == NULL)
Neal Norwitz33722ae2006-07-21 07:59:02 +0000129 Py_FatalError("Can't initialize import file table.");
Georg Brandladd36e52007-08-23 18:08:06 +0000130#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000131 memcpy(filetab, _PyImport_DynLoadFiletab,
132 countD * sizeof(struct filedescr));
Georg Brandladd36e52007-08-23 18:08:06 +0000133#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000134 memcpy(filetab + countD, _PyImport_StandardFiletab,
135 countS * sizeof(struct filedescr));
136 filetab[countD + countS].suffix = NULL;
137
138 _PyImport_Filetab = filetab;
139
Guido van Rossum0824f631997-03-11 18:37:35 +0000140 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000141 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
142 for (; filetab->suffix != NULL; filetab++) {
Guido van Rossum48a680c2001-03-02 06:34:14 +0000143#ifndef RISCOS
Guido van Rossumed1170e1999-12-20 21:23:41 +0000144 if (strcmp(filetab->suffix, ".pyc") == 0)
145 filetab->suffix = ".pyo";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000146#else
147 if (strcmp(filetab->suffix, "/pyc") == 0)
148 filetab->suffix = "/pyo";
149#endif
Guido van Rossum0824f631997-03-11 18:37:35 +0000150 }
151 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000152
153 if (Py_UnicodeFlag) {
154 /* Fix the pyc_magic so that byte compiled code created
155 using the all-Unicode method doesn't interfere with
156 code created in normal operation mode. */
157 pyc_magic = MAGIC + 1;
158 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000159}
160
Guido van Rossum25ce5661997-08-02 03:10:38 +0000161void
Just van Rossum52e14d62002-12-30 22:08:05 +0000162_PyImportHooks_Init(void)
163{
164 PyObject *v, *path_hooks = NULL, *zimpimport;
165 int err = 0;
166
167 /* adding sys.path_hooks and sys.path_importer_cache, setting up
168 zipimport */
Nick Coghlan327a39b2007-11-18 11:56:28 +0000169 if (PyType_Ready(&PyNullImporter_Type) < 0)
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000170 goto error;
Just van Rossum52e14d62002-12-30 22:08:05 +0000171
172 if (Py_VerboseFlag)
173 PySys_WriteStderr("# installing zipimport hook\n");
174
175 v = PyList_New(0);
176 if (v == NULL)
177 goto error;
178 err = PySys_SetObject("meta_path", v);
179 Py_DECREF(v);
180 if (err)
181 goto error;
182 v = PyDict_New();
183 if (v == NULL)
184 goto error;
185 err = PySys_SetObject("path_importer_cache", v);
186 Py_DECREF(v);
187 if (err)
188 goto error;
189 path_hooks = PyList_New(0);
190 if (path_hooks == NULL)
191 goto error;
192 err = PySys_SetObject("path_hooks", path_hooks);
193 if (err) {
194 error:
195 PyErr_Print();
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000196 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
197 "path_importer_cache, or NullImporter failed"
198 );
Just van Rossum52e14d62002-12-30 22:08:05 +0000199 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +0000200
Just van Rossum52e14d62002-12-30 22:08:05 +0000201 zimpimport = PyImport_ImportModule("zipimport");
202 if (zimpimport == NULL) {
203 PyErr_Clear(); /* No zip import module -- okay */
204 if (Py_VerboseFlag)
205 PySys_WriteStderr("# can't import zipimport\n");
206 }
207 else {
208 PyObject *zipimporter = PyObject_GetAttrString(zimpimport,
209 "zipimporter");
210 Py_DECREF(zimpimport);
211 if (zipimporter == NULL) {
212 PyErr_Clear(); /* No zipimporter object -- okay */
213 if (Py_VerboseFlag)
214 PySys_WriteStderr(
Fred Drake1e5fc552003-07-11 15:01:02 +0000215 "# can't import zipimport.zipimporter\n");
Just van Rossum52e14d62002-12-30 22:08:05 +0000216 }
217 else {
218 /* sys.path_hooks.append(zipimporter) */
219 err = PyList_Append(path_hooks, zipimporter);
220 Py_DECREF(zipimporter);
221 if (err)
222 goto error;
223 if (Py_VerboseFlag)
224 PySys_WriteStderr(
225 "# installed zipimport hook\n");
226 }
227 }
228 Py_DECREF(path_hooks);
229}
230
231void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000232_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000233{
234 Py_XDECREF(extensions);
235 extensions = NULL;
Barry Warsaw84294482000-10-03 16:02:05 +0000236 PyMem_DEL(_PyImport_Filetab);
237 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000238}
239
240
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000241/* Locking primitives to prevent parallel imports of the same module
242 in different threads to return with a partially loaded module.
243 These calls are serialized by the global interpreter lock. */
244
245#ifdef WITH_THREAD
246
Guido van Rossum49b56061998-10-01 20:42:43 +0000247#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000248
Guido van Rossum65d5b571998-12-21 19:32:43 +0000249static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000250static long import_lock_thread = -1;
251static int import_lock_level = 0;
252
253static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000254lock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000255{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000256 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000257 if (me == -1)
258 return; /* Too bad */
Neal Norwitze1fdb322006-07-21 05:32:28 +0000259 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000260 import_lock = PyThread_allocate_lock();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000261 if (import_lock == NULL)
262 return; /* Nothing much we can do. */
263 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000264 if (import_lock_thread == me) {
265 import_lock_level++;
266 return;
267 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000268 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
269 {
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000270 PyThreadState *tstate = PyEval_SaveThread();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000271 PyThread_acquire_lock(import_lock, 1);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000272 PyEval_RestoreThread(tstate);
273 }
274 import_lock_thread = me;
275 import_lock_level = 1;
276}
277
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000278static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000279unlock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000280{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000281 long me = PyThread_get_thread_ident();
Neal Norwitze1fdb322006-07-21 05:32:28 +0000282 if (me == -1 || import_lock == NULL)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000283 return 0; /* Too bad */
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000284 if (import_lock_thread != me)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000285 return -1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000286 import_lock_level--;
287 if (import_lock_level == 0) {
288 import_lock_thread = -1;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000289 PyThread_release_lock(import_lock);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000290 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000291 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000292}
293
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000294/* This function is called from PyOS_AfterFork to ensure that newly
295 created child processes do not share locks with the parent. */
296
297void
298_PyImport_ReInitLock(void)
299{
300#ifdef _AIX
301 if (import_lock != NULL)
302 import_lock = PyThread_allocate_lock();
303#endif
304}
305
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000306#else
307
308#define lock_import()
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000309#define unlock_import() 0
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000310
311#endif
312
Tim Peters69232342001-08-30 05:16:13 +0000313static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000314imp_lock_held(PyObject *self, PyObject *noargs)
Tim Peters69232342001-08-30 05:16:13 +0000315{
Tim Peters69232342001-08-30 05:16:13 +0000316#ifdef WITH_THREAD
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000317 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000318#else
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000319 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000320#endif
321}
322
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000323static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000324imp_acquire_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000325{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000326#ifdef WITH_THREAD
327 lock_import();
328#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000329 Py_INCREF(Py_None);
330 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000331}
332
333static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000334imp_release_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000335{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000336#ifdef WITH_THREAD
337 if (unlock_import() < 0) {
338 PyErr_SetString(PyExc_RuntimeError,
339 "not holding the import lock");
340 return NULL;
341 }
342#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000343 Py_INCREF(Py_None);
344 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000345}
346
Collin Winter276887b2007-03-12 16:11:39 +0000347static void
Neal Norwitz75c7c802007-03-13 05:31:38 +0000348imp_modules_reloading_clear(void)
Collin Winter276887b2007-03-12 16:11:39 +0000349{
350 PyInterpreterState *interp = PyThreadState_Get()->interp;
Neal Norwitz75c7c802007-03-13 05:31:38 +0000351 if (interp->modules_reloading != NULL)
352 PyDict_Clear(interp->modules_reloading);
Collin Winter276887b2007-03-12 16:11:39 +0000353}
354
Guido van Rossum25ce5661997-08-02 03:10:38 +0000355/* Helper for sys */
356
357PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000358PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000359{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000360 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000361 if (interp->modules == NULL)
362 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
363 return interp->modules;
364}
365
Guido van Rossum3f5da241990-12-20 15:06:42 +0000366
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000367/* List of names to clear in sys */
368static char* sys_deletes[] = {
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000369 "path", "argv", "ps1", "ps2", "exitfunc",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000370 "exc_type", "exc_value", "exc_traceback",
371 "last_type", "last_value", "last_traceback",
Just van Rossum52e14d62002-12-30 22:08:05 +0000372 "path_hooks", "path_importer_cache", "meta_path",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000373 NULL
374};
375
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000376static char* sys_files[] = {
377 "stdin", "__stdin__",
378 "stdout", "__stdout__",
379 "stderr", "__stderr__",
380 NULL
381};
382
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000383
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000384/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000385
Guido van Rossum3f5da241990-12-20 15:06:42 +0000386void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000387PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000388{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000389 Py_ssize_t pos, ndone;
Guido van Rossum758eec01998-01-19 21:58:26 +0000390 char *name;
391 PyObject *key, *value, *dict;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000392 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000393 PyObject *modules = interp->modules;
394
395 if (modules == NULL)
396 return; /* Already done */
397
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000398 /* Delete some special variables first. These are common
399 places where user values hide and people complain when their
400 destructors fail. Since the modules containing them are
401 deleted *last* of all, they would come too late in the normal
402 destruction order. Sigh. */
403
404 value = PyDict_GetItemString(modules, "__builtin__");
405 if (value != NULL && PyModule_Check(value)) {
406 dict = PyModule_GetDict(value);
407 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000408 PySys_WriteStderr("# clear __builtin__._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000409 PyDict_SetItemString(dict, "_", Py_None);
410 }
411 value = PyDict_GetItemString(modules, "sys");
412 if (value != NULL && PyModule_Check(value)) {
413 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000414 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000415 dict = PyModule_GetDict(value);
416 for (p = sys_deletes; *p != NULL; p++) {
417 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000418 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000419 PyDict_SetItemString(dict, *p, Py_None);
420 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000421 for (p = sys_files; *p != NULL; p+=2) {
422 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000423 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000424 v = PyDict_GetItemString(dict, *(p+1));
425 if (v == NULL)
426 v = Py_None;
427 PyDict_SetItemString(dict, *p, v);
428 }
429 }
430
431 /* First, delete __main__ */
432 value = PyDict_GetItemString(modules, "__main__");
433 if (value != NULL && PyModule_Check(value)) {
434 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000435 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000436 _PyModule_Clear(value);
437 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000438 }
439
Guido van Rossum758eec01998-01-19 21:58:26 +0000440 /* The special treatment of __builtin__ here is because even
441 when it's not referenced as a module, its dictionary is
442 referenced by almost every module's __builtins__. Since
443 deleting a module clears its dictionary (even if there are
444 references left to it), we need to delete the __builtin__
445 module last. Likewise, we don't delete sys until the very
446 end because it is implicitly referenced (e.g. by print).
447
448 Also note that we 'delete' modules by replacing their entry
449 in the modules dict with None, rather than really deleting
450 them; this avoids a rehash of the modules dictionary and
451 also marks them as "non existent" so they won't be
452 re-imported. */
453
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000454 /* Next, repeatedly delete modules with a reference count of
Guido van Rossum758eec01998-01-19 21:58:26 +0000455 one (skipping __builtin__ and sys) and delete them */
456 do {
457 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000458 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000459 while (PyDict_Next(modules, &pos, &key, &value)) {
460 if (value->ob_refcnt != 1)
461 continue;
Guido van Rossum566373e1998-10-01 15:24:50 +0000462 if (PyString_Check(key) && PyModule_Check(value)) {
463 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000464 if (strcmp(name, "__builtin__") == 0)
465 continue;
466 if (strcmp(name, "sys") == 0)
467 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000468 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000469 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000470 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000471 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000472 PyDict_SetItem(modules, key, Py_None);
473 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000474 }
475 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000476 } while (ndone > 0);
477
Guido van Rossum758eec01998-01-19 21:58:26 +0000478 /* Next, delete all modules (still skipping __builtin__ and sys) */
479 pos = 0;
480 while (PyDict_Next(modules, &pos, &key, &value)) {
Guido van Rossum566373e1998-10-01 15:24:50 +0000481 if (PyString_Check(key) && PyModule_Check(value)) {
482 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000483 if (strcmp(name, "__builtin__") == 0)
484 continue;
485 if (strcmp(name, "sys") == 0)
486 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000487 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000488 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000489 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000490 PyDict_SetItem(modules, key, Py_None);
491 }
492 }
493
494 /* Next, delete sys and __builtin__ (in that order) */
495 value = PyDict_GetItemString(modules, "sys");
496 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000497 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000498 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000499 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000500 PyDict_SetItemString(modules, "sys", Py_None);
501 }
502 value = PyDict_GetItemString(modules, "__builtin__");
503 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000504 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000505 PySys_WriteStderr("# cleanup __builtin__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000506 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000507 PyDict_SetItemString(modules, "__builtin__", Py_None);
508 }
509
510 /* Finally, clear and delete the modules directory */
511 PyDict_Clear(modules);
512 interp->modules = NULL;
513 Py_DECREF(modules);
Collin Winter276887b2007-03-12 16:11:39 +0000514 Py_CLEAR(interp->modules_reloading);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000515}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000516
517
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000518/* Helper for pythonrun.c -- return magic number */
519
520long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000521PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000522{
Guido van Rossum96774c12000-05-01 20:19:08 +0000523 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000524}
525
526
Guido van Rossum25ce5661997-08-02 03:10:38 +0000527/* Magic for extension modules (built-in as well as dynamically
528 loaded). To prevent initializing an extension module more than
529 once, we keep a static dictionary 'extensions' keyed by module name
530 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000531 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000532 dictionary is stored by calling _PyImport_FixupExtension()
533 immediately after the module initialization function succeeds. A
534 copy can be retrieved from there by calling
535 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000536
Guido van Rossum79f25d91997-04-29 20:08:16 +0000537PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000538_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000539{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000540 PyObject *modules, *mod, *dict, *copy;
541 if (extensions == NULL) {
542 extensions = PyDict_New();
543 if (extensions == NULL)
544 return NULL;
545 }
546 modules = PyImport_GetModuleDict();
547 mod = PyDict_GetItemString(modules, name);
548 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000549 PyErr_Format(PyExc_SystemError,
550 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551 return NULL;
552 }
553 dict = PyModule_GetDict(mod);
554 if (dict == NULL)
555 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000556 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000557 if (copy == NULL)
558 return NULL;
559 PyDict_SetItemString(extensions, filename, copy);
560 Py_DECREF(copy);
561 return copy;
562}
563
564PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000565_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000566{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000567 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000568 if (extensions == NULL)
569 return NULL;
570 dict = PyDict_GetItemString(extensions, filename);
571 if (dict == NULL)
572 return NULL;
573 mod = PyImport_AddModule(name);
574 if (mod == NULL)
575 return NULL;
576 mdict = PyModule_GetDict(mod);
577 if (mdict == NULL)
578 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000579 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000580 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000581 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000582 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000583 name, filename);
584 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000585}
586
587
588/* Get the module object corresponding to a module name.
589 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000590 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000591 Because the former action is most common, THIS DOES NOT RETURN A
592 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000593
Guido van Rossum79f25d91997-04-29 20:08:16 +0000594PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000595PyImport_AddModule(const char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000596{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000597 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000598 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000599
Guido van Rossum25ce5661997-08-02 03:10:38 +0000600 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000601 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000602 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000603 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000604 if (m == NULL)
605 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000606 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000607 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000608 return NULL;
609 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000610 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000611
612 return m;
613}
614
Tim Peters1cd70172004-08-02 03:52:12 +0000615/* Remove name from sys.modules, if it's there. */
616static void
617_RemoveModule(const char *name)
618{
619 PyObject *modules = PyImport_GetModuleDict();
620 if (PyDict_GetItemString(modules, name) == NULL)
621 return;
622 if (PyDict_DelItemString(modules, name) < 0)
623 Py_FatalError("import: deleting existing key in"
624 "sys.modules failed");
625}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000626
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000627/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000628 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
629 * removed from sys.modules, to avoid leaving damaged module objects
630 * in sys.modules. The caller may wish to restore the original
631 * module object (if any) in this case; PyImport_ReloadModule is an
632 * example.
633 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000634PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000635PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000636{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000637 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
638}
639
640PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000641PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000642{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000643 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000644 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000645
Guido van Rossum79f25d91997-04-29 20:08:16 +0000646 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000647 if (m == NULL)
648 return NULL;
Jeremy Hyltonecd91292004-01-02 23:25:32 +0000649 /* If the module is being reloaded, we get the old module back
650 and re-use its dict to exec the new code. */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000651 d = PyModule_GetDict(m);
652 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
653 if (PyDict_SetItemString(d, "__builtins__",
654 PyEval_GetBuiltins()) != 0)
Tim Peters1cd70172004-08-02 03:52:12 +0000655 goto error;
Guido van Rossum6135a871995-01-09 17:53:26 +0000656 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000657 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000658 v = NULL;
659 if (pathname != NULL) {
660 v = PyString_FromString(pathname);
661 if (v == NULL)
662 PyErr_Clear();
663 }
664 if (v == NULL) {
665 v = ((PyCodeObject *)co)->co_filename;
666 Py_INCREF(v);
667 }
668 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000669 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000670 Py_DECREF(v);
671
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000672 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000673 if (v == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000674 goto error;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000675 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000676
Guido van Rossum25ce5661997-08-02 03:10:38 +0000677 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000678 PyErr_Format(PyExc_ImportError,
679 "Loaded module %.200s not found in sys.modules",
680 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000681 return NULL;
682 }
683
Guido van Rossum79f25d91997-04-29 20:08:16 +0000684 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000685
686 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000687
688 error:
689 _RemoveModule(name);
690 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000691}
692
693
694/* Given a pathname for a Python source file, fill a buffer with the
695 pathname for the corresponding compiled file. Return the pathname
696 for the compiled file, or NULL if there's no space in the buffer.
697 Doesn't set an exception. */
698
699static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000700make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000701{
Tim Petersc1731372001-08-04 08:12:36 +0000702 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000703 if (len+2 > buflen)
704 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000705
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000706#ifdef MS_WINDOWS
Tim Petersc1731372001-08-04 08:12:36 +0000707 /* Treat .pyw as if it were .py. The case of ".pyw" must match
708 that used in _PyImport_StandardFiletab. */
709 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
710 --len; /* pretend 'w' isn't there */
711#endif
712 memcpy(buf, pathname, len);
713 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
714 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000715
716 return buf;
717}
718
719
720/* Given a pathname for a Python source file, its time of last
721 modification, and a pathname for a compiled file, check whether the
722 compiled file represents the same version of the source. If so,
723 return a FILE pointer for the compiled file, positioned just after
724 the header; if not, return NULL.
725 Doesn't set an exception. */
726
727static FILE *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000728check_compiled_module(char *pathname, time_t mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000729{
730 FILE *fp;
731 long magic;
732 long pyc_mtime;
733
734 fp = fopen(cpathname, "rb");
735 if (fp == NULL)
736 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000737 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000738 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000739 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000740 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000741 fclose(fp);
742 return NULL;
743 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000744 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000745 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000746 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000747 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000748 fclose(fp);
749 return NULL;
750 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000751 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000752 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000753 return fp;
754}
755
756
757/* Read a code object from a file and check it for validity */
758
Guido van Rossum79f25d91997-04-29 20:08:16 +0000759static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000760read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000761{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000762 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000763
Tim Petersd9b9ac82001-01-28 00:27:39 +0000764 co = PyMarshal_ReadLastObjectFromFile(fp);
Armin Rigo01ab2792004-03-26 15:09:27 +0000765 if (co == NULL)
766 return NULL;
767 if (!PyCode_Check(co)) {
768 PyErr_Format(PyExc_ImportError,
769 "Non-code object in %.200s", cpathname);
770 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000771 return NULL;
772 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000773 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000774}
775
776
777/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000778 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000779
Guido van Rossum79f25d91997-04-29 20:08:16 +0000780static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000781load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000782{
783 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000784 PyCodeObject *co;
785 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000786
Guido van Rossum79f25d91997-04-29 20:08:16 +0000787 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000788 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000789 PyErr_Format(PyExc_ImportError,
790 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000791 return NULL;
792 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000793 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000794 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000795 if (co == NULL)
796 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000797 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000798 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000799 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000800 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000801 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000802
803 return m;
804}
805
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000806/* Parse a source file and return the corresponding code object */
807
Guido van Rossum79f25d91997-04-29 20:08:16 +0000808static PyCodeObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000809parse_source_module(const char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000810{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000811 PyCodeObject *co = NULL;
812 mod_ty mod;
Neal Norwitz2a399b02006-09-11 04:28:16 +0000813 PyArena *arena = PyArena_New();
814 if (arena == NULL)
815 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000816
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000817 mod = PyParser_ASTFromFile(fp, pathname, Py_file_input, 0, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000818 NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000819 if (mod) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000820 co = PyAST_Compile(mod, pathname, NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000821 }
Neal Norwitz2a399b02006-09-11 04:28:16 +0000822 PyArena_Free(arena);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000823 return co;
824}
825
826
Guido van Rossum55a83382000-09-20 20:31:38 +0000827/* Helper to open a bytecode file for writing in exclusive mode */
828
829static FILE *
830open_exclusive(char *filename)
831{
832#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
833 /* Use O_EXCL to avoid a race condition when another process tries to
834 write the same file. When that happens, our open() call fails,
835 which is just fine (since it's only a cache).
836 XXX If the file exists and is writable but the directory is not
837 writable, the file will never be written. Oh well.
838 */
839 int fd;
840 (void) unlink(filename);
Tim Peters42c83af2000-09-29 04:03:10 +0000841 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
842#ifdef O_BINARY
843 |O_BINARY /* necessary for Windows */
844#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000845#ifdef __VMS
Martin v. Löwis18e16552006-02-15 17:27:45 +0000846 , 0666, "ctxt=bin", "shr=nil"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000847#else
Martin v. Löwis18e16552006-02-15 17:27:45 +0000848 , 0666
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000849#endif
Martin v. Löwis18e16552006-02-15 17:27:45 +0000850 );
Guido van Rossum55a83382000-09-20 20:31:38 +0000851 if (fd < 0)
852 return NULL;
853 return fdopen(fd, "wb");
854#else
855 /* Best we can do -- on Windows this can't happen anyway */
856 return fopen(filename, "wb");
857#endif
858}
859
860
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000861/* Write a compiled module to a file, placing the time of last
862 modification of its source into the header.
863 Errors are ignored, if a write error occurs an attempt is made to
864 remove the file. */
865
866static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000867write_compiled_module(PyCodeObject *co, char *cpathname, time_t mtime)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000868{
869 FILE *fp;
870
Guido van Rossum55a83382000-09-20 20:31:38 +0000871 fp = open_exclusive(cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000872 if (fp == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000873 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000874 PySys_WriteStderr(
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000875 "# can't create %s\n", cpathname);
876 return;
877 }
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000878 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000879 /* First write a 0 for mtime */
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000880 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION);
881 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION);
Armin Rigo01ab2792004-03-26 15:09:27 +0000882 if (fflush(fp) != 0 || ferror(fp)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000883 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000884 PySys_WriteStderr("# can't write %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000885 /* Don't keep partial file */
886 fclose(fp);
887 (void) unlink(cpathname);
888 return;
889 }
890 /* Now write the true mtime */
891 fseek(fp, 4L, 0);
Martin v. Löwis18e16552006-02-15 17:27:45 +0000892 assert(mtime < LONG_MAX);
Martin v. Löwis725507b2006-03-07 12:08:51 +0000893 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000894 fflush(fp);
895 fclose(fp);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000896 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000897 PySys_WriteStderr("# wrote %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000898}
899
900
901/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000902 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
903 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000904
Guido van Rossum79f25d91997-04-29 20:08:16 +0000905static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000906load_source_module(char *name, char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000907{
Fred Drake4c82b232000-06-30 16:18:57 +0000908 time_t mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000909 FILE *fpc;
910 char buf[MAXPATHLEN+1];
911 char *cpathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000912 PyCodeObject *co;
913 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000914
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000915 mtime = PyOS_GetLastModificationTime(pathname, fp);
Neal Norwitz708e51a2005-10-03 04:48:15 +0000916 if (mtime == (time_t)(-1)) {
917 PyErr_Format(PyExc_RuntimeError,
918 "unable to get modification time from '%s'",
919 pathname);
Fred Drake4c82b232000-06-30 16:18:57 +0000920 return NULL;
Neal Norwitz708e51a2005-10-03 04:48:15 +0000921 }
Fred Drake4c82b232000-06-30 16:18:57 +0000922#if SIZEOF_TIME_T > 4
923 /* Python's .pyc timestamp handling presumes that the timestamp fits
924 in 4 bytes. This will be fine until sometime in the year 2038,
925 when a 4-byte signed time_t will overflow.
926 */
927 if (mtime >> 32) {
928 PyErr_SetString(PyExc_OverflowError,
Fred Drakec63d3e92001-03-01 06:33:32 +0000929 "modification time overflows a 4 byte field");
Fred Drake4c82b232000-06-30 16:18:57 +0000930 return NULL;
931 }
932#endif
Tim Peters36515e22001-11-18 04:06:29 +0000933 cpathname = make_compiled_pathname(pathname, buf,
Jeremy Hylton37832f02001-04-13 17:50:20 +0000934 (size_t)MAXPATHLEN + 1);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000935 if (cpathname != NULL &&
936 (fpc = check_compiled_module(pathname, mtime, cpathname))) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000937 co = read_compiled_module(cpathname, fpc);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000938 fclose(fpc);
939 if (co == NULL)
940 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000941 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000942 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000943 name, cpathname);
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000944 pathname = cpathname;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000945 }
946 else {
947 co = parse_source_module(pathname, fp);
948 if (co == NULL)
949 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000950 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000951 PySys_WriteStderr("import %s # from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000952 name, pathname);
Georg Brandl2da0fce2008-01-07 17:09:35 +0000953 if (cpathname) {
954 PyObject *ro = PySys_GetObject("dont_write_bytecode");
955 if (ro == NULL || !PyObject_IsTrue(ro))
956 write_compiled_module(co, cpathname, mtime);
957 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000958 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000959 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000960 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000961
962 return m;
963}
964
965
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000966/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +0000967static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
968static struct filedescr *find_module(char *, char *, PyObject *,
969 char *, size_t, FILE **, PyObject **);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000970static struct _frozen *find_frozen(char *name);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000971
972/* Load a package and return its module object WITH INCREMENTED
973 REFERENCE COUNT */
974
975static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000976load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000977{
Tim Peters1cd70172004-08-02 03:52:12 +0000978 PyObject *m, *d;
979 PyObject *file = NULL;
980 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000981 int err;
982 char buf[MAXPATHLEN+1];
983 FILE *fp = NULL;
984 struct filedescr *fdp;
985
986 m = PyImport_AddModule(name);
987 if (m == NULL)
988 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000989 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000990 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000991 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000992 d = PyModule_GetDict(m);
993 file = PyString_FromString(pathname);
994 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000995 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000996 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +0000997 if (path == NULL)
998 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000999 err = PyDict_SetItemString(d, "__file__", file);
1000 if (err == 0)
1001 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +00001002 if (err != 0)
1003 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001004 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00001005 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001006 if (fdp == NULL) {
1007 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1008 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +00001009 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001010 }
1011 else
1012 m = NULL;
1013 goto cleanup;
1014 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001015 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001016 if (fp != NULL)
1017 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00001018 goto cleanup;
1019
1020 error:
1021 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001022 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001023 Py_XDECREF(path);
1024 Py_XDECREF(file);
1025 return m;
1026}
1027
1028
1029/* Helper to test for built-in module */
1030
1031static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001032is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001033{
1034 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001035 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1036 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1037 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001038 return -1;
1039 else
1040 return 1;
1041 }
1042 }
1043 return 0;
1044}
1045
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001046
Just van Rossum52e14d62002-12-30 22:08:05 +00001047/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1048 possibly by fetching it from the path_importer_cache dict. If it
Brett Cannon94b69f62006-09-28 22:10:14 +00001049 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +00001050 that can handle the path item. Return None if no hook could;
1051 this tells our caller it should fall back to the builtin
1052 import mechanism. Cache the result in path_importer_cache.
1053 Returns a borrowed reference. */
1054
1055static PyObject *
1056get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1057 PyObject *p)
1058{
1059 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001060 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001061
1062 /* These conditions are the caller's responsibility: */
1063 assert(PyList_Check(path_hooks));
1064 assert(PyDict_Check(path_importer_cache));
1065
1066 nhooks = PyList_Size(path_hooks);
1067 if (nhooks < 0)
1068 return NULL; /* Shouldn't happen */
1069
1070 importer = PyDict_GetItem(path_importer_cache, p);
1071 if (importer != NULL)
1072 return importer;
1073
1074 /* set path_importer_cache[p] to None to avoid recursion */
1075 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1076 return NULL;
1077
1078 for (j = 0; j < nhooks; j++) {
1079 PyObject *hook = PyList_GetItem(path_hooks, j);
1080 if (hook == NULL)
1081 return NULL;
Georg Brandl684fd0c2006-05-25 19:15:31 +00001082 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001083 if (importer != NULL)
1084 break;
1085
1086 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1087 return NULL;
1088 }
1089 PyErr_Clear();
1090 }
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001091 if (importer == NULL) {
1092 importer = PyObject_CallFunctionObjArgs(
Nick Coghlan327a39b2007-11-18 11:56:28 +00001093 (PyObject *)&PyNullImporter_Type, p, NULL
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001094 );
1095 if (importer == NULL) {
1096 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1097 PyErr_Clear();
1098 return Py_None;
1099 }
1100 }
1101 }
1102 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001103 int err = PyDict_SetItem(path_importer_cache, p, importer);
1104 Py_DECREF(importer);
1105 if (err != 0)
1106 return NULL;
1107 }
1108 return importer;
1109}
1110
Nick Coghlan327a39b2007-11-18 11:56:28 +00001111PyAPI_FUNC(PyObject *)
1112PyImport_GetImporter(PyObject *path) {
1113 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
1114
1115 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) {
1116 if ((path_hooks = PySys_GetObject("path_hooks"))) {
1117 importer = get_path_importer(path_importer_cache,
1118 path_hooks, path);
1119 }
1120 }
1121 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1122 return importer;
1123}
1124
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001125/* Search the path (default sys.path) for a module. Return the
1126 corresponding filedescr struct, and (via return arguments) the
1127 pathname and an open file. Return NULL if the module is not found. */
1128
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001129#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001130extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001131 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001132#endif
1133
Martin v. Löwis18e16552006-02-15 17:27:45 +00001134static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001135static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001136static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001137
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001138static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001139find_module(char *fullname, char *subname, PyObject *path, char *buf,
1140 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001141{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001142 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001143 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001144 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001145 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001146 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001147 PyObject *path_hooks, *path_importer_cache;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001148#ifndef RISCOS
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001149 struct stat statbuf;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001150#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001151 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1152 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1153 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001154 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001155#if defined(PYOS_OS2)
1156 size_t saved_len;
1157 size_t saved_namelen;
1158 char *saved_buf = NULL;
1159#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001160 if (p_loader != NULL)
1161 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001162
Just van Rossum52e14d62002-12-30 22:08:05 +00001163 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001164 PyErr_SetString(PyExc_OverflowError,
1165 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001166 return NULL;
1167 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001168 strcpy(name, subname);
1169
1170 /* sys.meta_path import hook */
1171 if (p_loader != NULL) {
1172 PyObject *meta_path;
1173
1174 meta_path = PySys_GetObject("meta_path");
1175 if (meta_path == NULL || !PyList_Check(meta_path)) {
1176 PyErr_SetString(PyExc_ImportError,
1177 "sys.meta_path must be a list of "
1178 "import hooks");
1179 return NULL;
1180 }
1181 Py_INCREF(meta_path); /* zap guard */
1182 npath = PyList_Size(meta_path);
1183 for (i = 0; i < npath; i++) {
1184 PyObject *loader;
1185 PyObject *hook = PyList_GetItem(meta_path, i);
1186 loader = PyObject_CallMethod(hook, "find_module",
1187 "sO", fullname,
1188 path != NULL ?
1189 path : Py_None);
1190 if (loader == NULL) {
1191 Py_DECREF(meta_path);
1192 return NULL; /* true error */
1193 }
1194 if (loader != Py_None) {
1195 /* a loader was found */
1196 *p_loader = loader;
1197 Py_DECREF(meta_path);
1198 return &importhookdescr;
1199 }
1200 Py_DECREF(loader);
1201 }
1202 Py_DECREF(meta_path);
1203 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001204
1205 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001206 /* The only type of submodule allowed inside a "frozen"
1207 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +00001208 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
1209 PyErr_SetString(PyExc_ImportError,
1210 "full frozen module name too long");
1211 return NULL;
1212 }
1213 strcpy(buf, PyString_AsString(path));
1214 strcat(buf, ".");
1215 strcat(buf, name);
1216 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001217 if (find_frozen(name) != NULL) {
1218 strcpy(buf, name);
1219 return &fd_frozen;
1220 }
1221 PyErr_Format(PyExc_ImportError,
1222 "No frozen submodule named %.200s", name);
1223 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001224 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001225 if (path == NULL) {
1226 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001227 strcpy(buf, name);
1228 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001229 }
Greg Ward201baee2001-10-04 14:52:06 +00001230 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001231 strcpy(buf, name);
1232 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001233 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001234
Guido van Rossumac279101996-08-22 23:10:58 +00001235#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001236 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1237 if (fp != NULL) {
1238 *p_fp = fp;
1239 return fdp;
1240 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001241#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001242 path = PySys_GetObject("path");
1243 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001244 if (path == NULL || !PyList_Check(path)) {
1245 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001246 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001247 return NULL;
1248 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001249
1250 path_hooks = PySys_GetObject("path_hooks");
1251 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1252 PyErr_SetString(PyExc_ImportError,
1253 "sys.path_hooks must be a list of "
1254 "import hooks");
1255 return NULL;
1256 }
1257 path_importer_cache = PySys_GetObject("path_importer_cache");
1258 if (path_importer_cache == NULL ||
1259 !PyDict_Check(path_importer_cache)) {
1260 PyErr_SetString(PyExc_ImportError,
1261 "sys.path_importer_cache must be a dict");
1262 return NULL;
1263 }
1264
Guido van Rossum79f25d91997-04-29 20:08:16 +00001265 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001266 namelen = strlen(name);
1267 for (i = 0; i < npath; i++) {
Walter Dörwald3430d702002-06-17 10:43:59 +00001268 PyObject *copy = NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001269 PyObject *v = PyList_GetItem(path, i);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00001270 if (!v)
1271 return NULL;
Walter Dörwald3430d702002-06-17 10:43:59 +00001272#ifdef Py_USING_UNICODE
1273 if (PyUnicode_Check(v)) {
1274 copy = PyUnicode_Encode(PyUnicode_AS_UNICODE(v),
1275 PyUnicode_GET_SIZE(v), Py_FileSystemDefaultEncoding, NULL);
1276 if (copy == NULL)
1277 return NULL;
1278 v = copy;
1279 }
1280 else
1281#endif
Guido van Rossum79f25d91997-04-29 20:08:16 +00001282 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001283 continue;
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001284 len = PyString_GET_SIZE(v);
Walter Dörwald3430d702002-06-17 10:43:59 +00001285 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
1286 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001287 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001288 }
Neal Norwitz2aa9a5d2006-03-20 01:53:23 +00001289 strcpy(buf, PyString_AS_STRING(v));
Walter Dörwald3430d702002-06-17 10:43:59 +00001290 if (strlen(buf) != len) {
1291 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001292 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001293 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001294
1295 /* sys.path_hooks import hook */
1296 if (p_loader != NULL) {
1297 PyObject *importer;
1298
1299 importer = get_path_importer(path_importer_cache,
1300 path_hooks, v);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001301 if (importer == NULL) {
1302 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001303 return NULL;
Neal Norwitza4df11d2006-07-06 04:28:59 +00001304 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001305 /* Note: importer is a borrowed reference */
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00001306 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001307 PyObject *loader;
1308 loader = PyObject_CallMethod(importer,
1309 "find_module",
1310 "s", fullname);
Neal Norwitza4df11d2006-07-06 04:28:59 +00001311 Py_XDECREF(copy);
Just van Rossum52e14d62002-12-30 22:08:05 +00001312 if (loader == NULL)
1313 return NULL; /* error */
1314 if (loader != Py_None) {
1315 /* a loader was found */
1316 *p_loader = loader;
1317 return &importhookdescr;
1318 }
1319 Py_DECREF(loader);
Georg Brandlf4ef1162006-05-26 18:03:31 +00001320 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001321 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001322 }
Georg Brandlf4ef1162006-05-26 18:03:31 +00001323 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001324
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001325 if (len > 0 && buf[len-1] != SEP
1326#ifdef ALTSEP
1327 && buf[len-1] != ALTSEP
1328#endif
1329 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001330 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001331 strcpy(buf+len, name);
1332 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001333
1334 /* Check for package import (buf holds a directory name,
1335 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001336#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001337 if (stat(buf, &statbuf) == 0 && /* it exists */
1338 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001339 case_ok(buf, len, namelen, name)) { /* case matches */
1340 if (find_init_module(buf)) { /* and has __init__.py */
1341 Py_XDECREF(copy);
1342 return &fd_package;
1343 }
1344 else {
1345 char warnstr[MAXPATHLEN+80];
1346 sprintf(warnstr, "Not importing directory "
1347 "'%.*s': missing __init__.py",
1348 MAXPATHLEN, buf);
1349 if (PyErr_Warn(PyExc_ImportWarning,
1350 warnstr)) {
1351 Py_XDECREF(copy);
1352 return NULL;
1353 }
1354 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001355 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001356#else
1357 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001358#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001359 if (isdir(buf) &&
Walter Dörwald3430d702002-06-17 10:43:59 +00001360 case_ok(buf, len, namelen, name)) {
Thomas Wouters9df4e6f2006-04-27 23:13:20 +00001361 if (find_init_module(buf)) {
1362 Py_XDECREF(copy);
1363 return &fd_package;
1364 }
1365 else {
1366 char warnstr[MAXPATHLEN+80];
1367 sprintf(warnstr, "Not importing directory "
1368 "'%.*s': missing __init__.py",
1369 MAXPATHLEN, buf);
1370 if (PyErr_Warn(PyExc_ImportWarning,
1371 warnstr)) {
1372 Py_XDECREF(copy);
1373 return NULL;
1374 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001375 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001376#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001377#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001378#if defined(PYOS_OS2)
1379 /* take a snapshot of the module spec for restoration
1380 * after the 8 character DLL hackery
1381 */
1382 saved_buf = strdup(buf);
1383 saved_len = len;
1384 saved_namelen = namelen;
1385#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001386 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Georg Brandladd36e52007-08-23 18:08:06 +00001387#if defined(PYOS_OS2) && defined(HAVE_DYNAMIC_LOADING)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001388 /* OS/2 limits DLLs to 8 character names (w/o
1389 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001390 * so if the name is longer than that and its a
1391 * dynamically loaded module we're going to try,
1392 * truncate the name before trying
1393 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001394 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001395 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001396 const struct filedescr *scan;
1397 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001398 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001399 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001400 break;
1401 else
1402 scan++;
1403 }
1404 if (scan->suffix != NULL) {
1405 /* yes, so truncate the name */
1406 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001407 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001408 buf[len] = '\0';
1409 }
1410 }
1411#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001412 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001413 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001414 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001415 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001416 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001417 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001418 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001419 if (fp != NULL) {
1420 if (case_ok(buf, len, namelen, name))
1421 break;
1422 else { /* continue search */
1423 fclose(fp);
1424 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001425 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001426 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001427#if defined(PYOS_OS2)
1428 /* restore the saved snapshot */
1429 strcpy(buf, saved_buf);
1430 len = saved_len;
1431 namelen = saved_namelen;
1432#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001433 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001434#if defined(PYOS_OS2)
1435 /* don't need/want the module name snapshot anymore */
1436 if (saved_buf)
1437 {
1438 free(saved_buf);
1439 saved_buf = NULL;
1440 }
1441#endif
Walter Dörwald3430d702002-06-17 10:43:59 +00001442 Py_XDECREF(copy);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001443 if (fp != NULL)
1444 break;
1445 }
1446 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001447 PyErr_Format(PyExc_ImportError,
1448 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001449 return NULL;
1450 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001451 *p_fp = fp;
1452 return fdp;
1453}
1454
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001455/* Helpers for main.c
1456 * Find the source file corresponding to a named module
1457 */
1458struct filedescr *
1459_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1460 size_t buflen, FILE **p_fp, PyObject **p_loader)
1461{
1462 return find_module((char *) name, (char *) name, path,
1463 buf, buflen, p_fp, p_loader);
1464}
1465
1466PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1467{
1468 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1469}
1470
Martin v. Löwis18e16552006-02-15 17:27:45 +00001471/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001472 * The arguments here are tricky, best shown by example:
1473 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1474 * ^ ^ ^ ^
1475 * |--------------------- buf ---------------------|
1476 * |------------------- len ------------------|
1477 * |------ name -------|
1478 * |----- namelen -----|
1479 * buf is the full path, but len only counts up to (& exclusive of) the
1480 * extension. name is the module name, also exclusive of extension.
1481 *
1482 * We've already done a successful stat() or fopen() on buf, so know that
1483 * there's some match, possibly case-insensitive.
1484 *
Tim Peters50d8d372001-02-28 05:34:27 +00001485 * case_ok() is to return 1 if there's a case-sensitive match for
1486 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1487 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001488 *
Tim Peters50d8d372001-02-28 05:34:27 +00001489 * case_ok() is used to implement case-sensitive import semantics even
1490 * on platforms with case-insensitive filesystems. It's trivial to implement
1491 * for case-sensitive filesystems. It's pretty much a cross-platform
1492 * nightmare for systems with case-insensitive filesystems.
1493 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001494
Tim Peters50d8d372001-02-28 05:34:27 +00001495/* First we may need a pile of platform-specific header files; the sequence
1496 * of #if's here should match the sequence in the body of case_ok().
1497 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001498#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001499#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001500
Tim Peters50d8d372001-02-28 05:34:27 +00001501#elif defined(DJGPP)
1502#include <dir.h>
1503
Jason Tishler7961aa62005-05-20 00:56:54 +00001504#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001505#include <sys/types.h>
1506#include <dirent.h>
1507
Andrew MacIntyred9400542002-02-26 11:41:34 +00001508#elif defined(PYOS_OS2)
1509#define INCL_DOS
1510#define INCL_DOSERRORS
1511#define INCL_NOPMAPI
1512#include <os2.h>
1513
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001514#elif defined(RISCOS)
1515#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001516#endif
1517
Guido van Rossum0980bd91998-02-13 17:18:36 +00001518static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001519case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001520{
Tim Peters50d8d372001-02-28 05:34:27 +00001521/* Pick a platform-specific implementation; the sequence of #if's here should
1522 * match the sequence just above.
1523 */
1524
Jason Tishler7961aa62005-05-20 00:56:54 +00001525/* MS_WINDOWS */
1526#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001527 WIN32_FIND_DATA data;
1528 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001529
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001530 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001531 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001532
Guido van Rossum0980bd91998-02-13 17:18:36 +00001533 h = FindFirstFile(buf, &data);
1534 if (h == INVALID_HANDLE_VALUE) {
1535 PyErr_Format(PyExc_NameError,
1536 "Can't find file for module %.100s\n(filename %.300s)",
1537 name, buf);
1538 return 0;
1539 }
1540 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001541 return strncmp(data.cFileName, name, namelen) == 0;
1542
1543/* DJGPP */
1544#elif defined(DJGPP)
1545 struct ffblk ffblk;
1546 int done;
1547
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001548 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001549 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001550
1551 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1552 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001553 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001554 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001555 name, buf);
1556 return 0;
1557 }
Tim Peters50d8d372001-02-28 05:34:27 +00001558 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001559
Jason Tishler7961aa62005-05-20 00:56:54 +00001560/* new-fangled macintosh (macosx) or Cygwin */
1561#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001562 DIR *dirp;
1563 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001564 char dirname[MAXPATHLEN + 1];
1565 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001566
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001567 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001568 return 1;
1569
Tim Petersd1e87a82001-03-01 18:12:00 +00001570 /* Copy the dir component into dirname; substitute "." if empty */
1571 if (dirlen <= 0) {
1572 dirname[0] = '.';
1573 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001574 }
1575 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001576 assert(dirlen <= MAXPATHLEN);
1577 memcpy(dirname, buf, dirlen);
1578 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001579 }
1580 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001581 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001582 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001583 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001584 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001585 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001586#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001587 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001588#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001589 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001590#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001591 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001592 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001593 (void)closedir(dirp);
1594 return 1; /* Found */
1595 }
1596 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001597 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001598 }
Tim Peters430f5d42001-03-01 01:30:56 +00001599 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001600
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001601/* RISC OS */
1602#elif defined(RISCOS)
1603 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1604 char buf2[MAXPATHLEN+2];
1605 char *nameWithExt = buf+len-namelen;
1606 int canonlen;
1607 os_error *e;
1608
1609 if (Py_GETENV("PYTHONCASEOK") != NULL)
1610 return 1;
1611
1612 /* workaround:
1613 append wildcard, otherwise case of filename wouldn't be touched */
1614 strcpy(buf2, buf);
1615 strcat(buf2, "*");
1616
1617 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1618 canonlen = MAXPATHLEN+1-canonlen;
1619 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1620 return 0;
1621 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1622 return 1; /* match */
1623
1624 return 0;
1625
Andrew MacIntyred9400542002-02-26 11:41:34 +00001626/* OS/2 */
1627#elif defined(PYOS_OS2)
1628 HDIR hdir = 1;
1629 ULONG srchcnt = 1;
1630 FILEFINDBUF3 ffbuf;
1631 APIRET rc;
1632
Georg Brandlaed6c662008-01-07 17:25:53 +00001633 if (Py_GETENV("PYTHONCASEOK") != NULL)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001634 return 1;
1635
1636 rc = DosFindFirst(buf,
1637 &hdir,
1638 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1639 &ffbuf, sizeof(ffbuf),
1640 &srchcnt,
1641 FIL_STANDARD);
1642 if (rc != NO_ERROR)
1643 return 0;
1644 return strncmp(ffbuf.achName, name, namelen) == 0;
1645
Tim Peters50d8d372001-02-28 05:34:27 +00001646/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1647#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001648 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001649
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001650#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001651}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001652
Guido van Rossum0980bd91998-02-13 17:18:36 +00001653
Guido van Rossum197346f1997-10-31 18:38:52 +00001654#ifdef HAVE_STAT
1655/* Helper to look for __init__.py or __init__.py[co] in potential package */
1656static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001657find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001658{
Tim Peters0f9431f2001-07-05 03:47:53 +00001659 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001660 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001661 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001662 struct stat statbuf;
1663
Tim Peters0f9431f2001-07-05 03:47:53 +00001664/* For calling case_ok(buf, len, namelen, name):
1665 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1666 * ^ ^ ^ ^
1667 * |--------------------- buf ---------------------|
1668 * |------------------- len ------------------|
1669 * |------ name -------|
1670 * |----- namelen -----|
1671 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001672 if (save_len + 13 >= MAXPATHLEN)
1673 return 0;
1674 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001675 pname = buf + i;
1676 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001677 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001678 if (case_ok(buf,
1679 save_len + 9, /* len("/__init__") */
1680 8, /* len("__init__") */
1681 pname)) {
1682 buf[save_len] = '\0';
1683 return 1;
1684 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001685 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001686 i += strlen(pname);
1687 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001688 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001689 if (case_ok(buf,
1690 save_len + 9, /* len("/__init__") */
1691 8, /* len("__init__") */
1692 pname)) {
1693 buf[save_len] = '\0';
1694 return 1;
1695 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001696 }
1697 buf[save_len] = '\0';
1698 return 0;
1699}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001700
1701#else
1702
1703#ifdef RISCOS
1704static int
1705find_init_module(buf)
1706 char *buf;
1707{
1708 int save_len = strlen(buf);
1709 int i = save_len;
1710
1711 if (save_len + 13 >= MAXPATHLEN)
1712 return 0;
1713 buf[i++] = SEP;
1714 strcpy(buf+i, "__init__/py");
1715 if (isfile(buf)) {
1716 buf[save_len] = '\0';
1717 return 1;
1718 }
1719
1720 if (Py_OptimizeFlag)
1721 strcpy(buf+i, "o");
1722 else
1723 strcpy(buf+i, "c");
1724 if (isfile(buf)) {
1725 buf[save_len] = '\0';
1726 return 1;
1727 }
1728 buf[save_len] = '\0';
1729 return 0;
1730}
1731#endif /*RISCOS*/
1732
Guido van Rossum197346f1997-10-31 18:38:52 +00001733#endif /* HAVE_STAT */
1734
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001735
Tim Petersdbd9ba62000-07-09 03:09:57 +00001736static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001737
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001738/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001739 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001740
Guido van Rossum79f25d91997-04-29 20:08:16 +00001741static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001742load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001743{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001744 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001745 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001746 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001747
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001748 /* First check that there's an open file (if we need one) */
1749 switch (type) {
1750 case PY_SOURCE:
1751 case PY_COMPILED:
1752 if (fp == NULL) {
1753 PyErr_Format(PyExc_ValueError,
1754 "file object required for import (type code %d)",
1755 type);
1756 return NULL;
1757 }
1758 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001759
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001760 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001761
1762 case PY_SOURCE:
1763 m = load_source_module(name, buf, fp);
1764 break;
1765
1766 case PY_COMPILED:
1767 m = load_compiled_module(name, buf, fp);
1768 break;
1769
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001770#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001771 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001772 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001773 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001774#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001775
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001776 case PKG_DIRECTORY:
1777 m = load_package(name, buf);
1778 break;
1779
1780 case C_BUILTIN:
1781 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001782 if (buf != NULL && buf[0] != '\0')
1783 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001784 if (type == C_BUILTIN)
1785 err = init_builtin(name);
1786 else
1787 err = PyImport_ImportFrozenModule(name);
1788 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001789 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001790 if (err == 0) {
1791 PyErr_Format(PyExc_ImportError,
1792 "Purported %s module %.200s not found",
1793 type == C_BUILTIN ?
1794 "builtin" : "frozen",
1795 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001796 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001797 }
1798 modules = PyImport_GetModuleDict();
1799 m = PyDict_GetItemString(modules, name);
1800 if (m == NULL) {
1801 PyErr_Format(
1802 PyExc_ImportError,
1803 "%s module %.200s not properly initialized",
1804 type == C_BUILTIN ?
1805 "builtin" : "frozen",
1806 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001807 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001808 }
1809 Py_INCREF(m);
1810 break;
1811
Just van Rossum52e14d62002-12-30 22:08:05 +00001812 case IMP_HOOK: {
1813 if (loader == NULL) {
1814 PyErr_SetString(PyExc_ImportError,
1815 "import hook without loader");
1816 return NULL;
1817 }
1818 m = PyObject_CallMethod(loader, "load_module", "s", name);
1819 break;
1820 }
1821
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001822 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001823 PyErr_Format(PyExc_ImportError,
1824 "Don't know how to import %.200s (type code %d)",
1825 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001826 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001827
1828 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001829
1830 return m;
1831}
1832
1833
1834/* Initialize a built-in module.
Brett Cannon5a9aa4f2006-10-03 21:58:55 +00001835 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001836 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001837
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001838static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001839init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001840{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001841 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001842
Greg Ward201baee2001-10-04 14:52:06 +00001843 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001844 return 1;
1845
Guido van Rossum771c6c81997-10-31 18:37:24 +00001846 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001847 if (strcmp(name, p->name) == 0) {
1848 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001849 PyErr_Format(PyExc_ImportError,
1850 "Cannot re-init internal module %.200s",
1851 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001852 return -1;
1853 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001854 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001855 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001856 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001857 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001858 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001859 if (_PyImport_FixupExtension(name, name) == NULL)
1860 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001861 return 1;
1862 }
1863 }
1864 return 0;
1865}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001866
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001867
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001868/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001869
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001870static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001871find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001872{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001873 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001874
Guido van Rossum79f25d91997-04-29 20:08:16 +00001875 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001876 if (p->name == NULL)
1877 return NULL;
1878 if (strcmp(p->name, name) == 0)
1879 break;
1880 }
1881 return p;
1882}
1883
Guido van Rossum79f25d91997-04-29 20:08:16 +00001884static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001885get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001886{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001887 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001888 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001889
1890 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001891 PyErr_Format(PyExc_ImportError,
1892 "No such frozen object named %.200s",
1893 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001894 return NULL;
1895 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001896 if (p->code == NULL) {
1897 PyErr_Format(PyExc_ImportError,
1898 "Excluded frozen object named %.200s",
1899 name);
1900 return NULL;
1901 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001902 size = p->size;
1903 if (size < 0)
1904 size = -size;
1905 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001906}
1907
1908/* Initialize a frozen module.
1909 Return 1 for succes, 0 if the module is not found, and -1 with
1910 an exception set if the initialization failed.
1911 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001912
1913int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001914PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001915{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001916 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001917 PyObject *co;
1918 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001919 int ispackage;
1920 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001921
1922 if (p == NULL)
1923 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001924 if (p->code == NULL) {
1925 PyErr_Format(PyExc_ImportError,
1926 "Excluded frozen object named %.200s",
1927 name);
1928 return -1;
1929 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001930 size = p->size;
1931 ispackage = (size < 0);
1932 if (ispackage)
1933 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001934 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001935 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001936 name, ispackage ? " package" : "");
1937 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001938 if (co == NULL)
1939 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001940 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001941 PyErr_Format(PyExc_TypeError,
1942 "frozen object %.200s is not a code object",
1943 name);
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001944 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001945 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001946 if (ispackage) {
1947 /* Set __path__ to the package name */
1948 PyObject *d, *s;
1949 int err;
1950 m = PyImport_AddModule(name);
1951 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001952 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001953 d = PyModule_GetDict(m);
1954 s = PyString_InternFromString(name);
1955 if (s == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001956 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001957 err = PyDict_SetItemString(d, "__path__", s);
1958 Py_DECREF(s);
1959 if (err != 0)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001960 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001961 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001962 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001963 if (m == NULL)
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001964 goto err_return;
1965 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001966 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001967 return 1;
Neal Norwitzc0cde4d2006-07-16 02:17:36 +00001968err_return:
1969 Py_DECREF(co);
1970 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001971}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001972
1973
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001974/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001975 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001976
Guido van Rossum79f25d91997-04-29 20:08:16 +00001977PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001978PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001979{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001980 PyObject *pname;
1981 PyObject *result;
1982
1983 pname = PyString_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001984 if (pname == NULL)
1985 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001986 result = PyImport_Import(pname);
1987 Py_DECREF(pname);
1988 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001989}
1990
Christian Heimes000a0742008-01-03 22:16:32 +00001991/* Import a module without blocking
1992 *
1993 * At first it tries to fetch the module from sys.modules. If the module was
1994 * never loaded before it loads it with PyImport_ImportModule() unless another
1995 * thread holds the import lock. In the latter case the function raises an
1996 * ImportError instead of blocking.
1997 *
1998 * Returns the module object with incremented ref count.
1999 */
2000PyObject *
2001PyImport_ImportModuleNoBlock(const char *name)
2002{
2003 PyObject *result;
2004 PyObject *modules;
2005 long me;
2006
2007 /* Try to get the module from sys.modules[name] */
2008 modules = PyImport_GetModuleDict();
2009 if (modules == NULL)
2010 return NULL;
2011
2012 result = PyDict_GetItemString(modules, name);
2013 if (result != NULL) {
2014 Py_INCREF(result);
2015 return result;
2016 }
2017 else {
2018 PyErr_Clear();
2019 }
2020
2021 /* check the import lock
2022 * me might be -1 but I ignore the error here, the lock function
2023 * takes care of the problem */
2024 me = PyThread_get_thread_ident();
2025 if (import_lock_thread == -1 || import_lock_thread == me) {
2026 /* no thread or me is holding the lock */
2027 return PyImport_ImportModule(name);
2028 }
2029 else {
2030 PyErr_Format(PyExc_ImportError,
2031 "Failed to import %.200s because the import lock"
2032 "is held by another thread.",
2033 name);
2034 return NULL;
2035 }
2036}
2037
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002038/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002039static PyObject *get_parent(PyObject *globals, char *buf,
2040 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00002041static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002042 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00002043static int mark_miss(char *name);
2044static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002045 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00002046static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002047
2048/* The Magnum Opus of dotted-name import :-) */
2049
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002050static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002051import_module_level(char *name, PyObject *globals, PyObject *locals,
2052 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002053{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002054 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00002055 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002056 PyObject *parent, *head, *next, *tail;
2057
Christian Heimes3403f152008-01-09 19:56:33 +00002058 if (strchr(name, '/') != NULL
2059#ifdef MS_WINDOWS
2060 || strchr(name, '\\') != NULL
2061#endif
2062 ) {
2063 PyErr_SetString(PyExc_ImportError,
2064 "Import by filename is not supported.");
2065 return NULL;
2066 }
2067
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002068 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002069 if (parent == NULL)
2070 return NULL;
2071
2072 head = load_next(parent, Py_None, &name, buf, &buflen);
2073 if (head == NULL)
2074 return NULL;
2075
2076 tail = head;
2077 Py_INCREF(tail);
2078 while (name) {
2079 next = load_next(tail, tail, &name, buf, &buflen);
2080 Py_DECREF(tail);
2081 if (next == NULL) {
2082 Py_DECREF(head);
2083 return NULL;
2084 }
2085 tail = next;
2086 }
Thomas Wouters8ddab272006-04-04 16:17:02 +00002087 if (tail == Py_None) {
2088 /* If tail is Py_None, both get_parent and load_next found
2089 an empty module name: someone called __import__("") or
2090 doctored faulty bytecode */
Thomas Wouters4bdaa272006-04-05 13:39:37 +00002091 Py_DECREF(tail);
2092 Py_DECREF(head);
Thomas Wouters8ddab272006-04-04 16:17:02 +00002093 PyErr_SetString(PyExc_ValueError,
2094 "Empty module name");
2095 return NULL;
2096 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002097
2098 if (fromlist != NULL) {
2099 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2100 fromlist = NULL;
2101 }
2102
2103 if (fromlist == NULL) {
2104 Py_DECREF(tail);
2105 return head;
2106 }
2107
2108 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002109 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002110 Py_DECREF(tail);
2111 return NULL;
2112 }
2113
2114 return tail;
2115}
2116
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002117PyObject *
2118PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
2119 PyObject *fromlist, int level)
2120{
2121 PyObject *result;
2122 lock_import();
2123 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002124 if (unlock_import() < 0) {
2125 Py_XDECREF(result);
2126 PyErr_SetString(PyExc_RuntimeError,
2127 "not holding the import lock");
2128 return NULL;
2129 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002130 return result;
2131}
2132
Fred Drake87590902004-05-28 20:21:36 +00002133/* Return the package that an import is being performed in. If globals comes
2134 from the module foo.bar.bat (not itself a package), this returns the
2135 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Georg Brandl5f6861d2006-05-28 21:57:35 +00002136 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002137
2138 The *name* of the returned package is returned in buf, with the length of
2139 the name in *p_buflen.
2140
2141 If globals doesn't come from a package or a module in a package, or a
2142 corresponding entry is not found in sys.modules, Py_None is returned.
2143*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002144static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002145get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002146{
2147 static PyObject *namestr = NULL;
2148 static PyObject *pathstr = NULL;
Nick Coghlanef01d822007-12-03 12:55:17 +00002149 static PyObject *pkgstr = NULL;
2150 PyObject *pkgname, *modname, *modpath, *modules, *parent;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002151
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002152 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002153 return Py_None;
2154
2155 if (namestr == NULL) {
2156 namestr = PyString_InternFromString("__name__");
2157 if (namestr == NULL)
2158 return NULL;
2159 }
2160 if (pathstr == NULL) {
2161 pathstr = PyString_InternFromString("__path__");
2162 if (pathstr == NULL)
2163 return NULL;
2164 }
Nick Coghlanef01d822007-12-03 12:55:17 +00002165 if (pkgstr == NULL) {
2166 pkgstr = PyString_InternFromString("__package__");
2167 if (pkgstr == NULL)
2168 return NULL;
2169 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002170
2171 *buf = '\0';
2172 *p_buflen = 0;
Nick Coghlanef01d822007-12-03 12:55:17 +00002173 pkgname = PyDict_GetItem(globals, pkgstr);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002174
Nick Coghlanef01d822007-12-03 12:55:17 +00002175 if ((pkgname != NULL) && (pkgname != Py_None)) {
2176 /* __package__ is set, so use it */
2177 Py_ssize_t len;
2178 if (!PyString_Check(pkgname)) {
2179 PyErr_SetString(PyExc_ValueError,
2180 "__package__ set to non-string");
2181 return NULL;
2182 }
2183 len = PyString_GET_SIZE(pkgname);
2184 if (len == 0) {
2185 if (level > 0) {
2186 PyErr_SetString(PyExc_ValueError,
2187 "Attempted relative import in non-package");
2188 return NULL;
2189 }
2190 return Py_None;
2191 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002192 if (len > MAXPATHLEN) {
2193 PyErr_SetString(PyExc_ValueError,
Nick Coghlanef01d822007-12-03 12:55:17 +00002194 "Package name too long");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002195 return NULL;
2196 }
Nick Coghlanef01d822007-12-03 12:55:17 +00002197 strcpy(buf, PyString_AS_STRING(pkgname));
2198 } else {
2199 /* __package__ not set, so figure it out and set it */
2200 modname = PyDict_GetItem(globals, namestr);
2201 if (modname == NULL || !PyString_Check(modname))
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002202 return Py_None;
Nick Coghlanef01d822007-12-03 12:55:17 +00002203
2204 modpath = PyDict_GetItem(globals, pathstr);
2205 if (modpath != NULL) {
2206 /* __path__ is set, so modname is already the package name */
2207 Py_ssize_t len = PyString_GET_SIZE(modname);
2208 int error;
2209 if (len > MAXPATHLEN) {
2210 PyErr_SetString(PyExc_ValueError,
2211 "Module name too long");
2212 return NULL;
2213 }
2214 strcpy(buf, PyString_AS_STRING(modname));
2215 error = PyDict_SetItem(globals, pkgstr, modname);
2216 if (error) {
2217 PyErr_SetString(PyExc_ValueError,
2218 "Could not set __package__");
2219 return NULL;
2220 }
2221 } else {
2222 /* Normal module, so work out the package name if any */
2223 char *start = PyString_AS_STRING(modname);
2224 char *lastdot = strrchr(start, '.');
2225 size_t len;
2226 int error;
2227 if (lastdot == NULL && level > 0) {
2228 PyErr_SetString(PyExc_ValueError,
2229 "Attempted relative import in non-package");
2230 return NULL;
2231 }
2232 if (lastdot == NULL) {
2233 error = PyDict_SetItem(globals, pkgstr, Py_None);
2234 if (error) {
2235 PyErr_SetString(PyExc_ValueError,
2236 "Could not set __package__");
2237 return NULL;
2238 }
2239 return Py_None;
2240 }
2241 len = lastdot - start;
2242 if (len >= MAXPATHLEN) {
2243 PyErr_SetString(PyExc_ValueError,
2244 "Module name too long");
2245 return NULL;
2246 }
2247 strncpy(buf, start, len);
2248 buf[len] = '\0';
2249 pkgname = PyString_FromString(buf);
2250 if (pkgname == NULL) {
2251 return NULL;
2252 }
2253 error = PyDict_SetItem(globals, pkgstr, pkgname);
2254 Py_DECREF(pkgname);
2255 if (error) {
2256 PyErr_SetString(PyExc_ValueError,
2257 "Could not set __package__");
2258 return NULL;
2259 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002260 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002261 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002262 while (--level > 0) {
2263 char *dot = strrchr(buf, '.');
2264 if (dot == NULL) {
2265 PyErr_SetString(PyExc_ValueError,
Georg Brandl98775df2006-09-06 06:09:31 +00002266 "Attempted relative import beyond "
2267 "toplevel package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002268 return NULL;
2269 }
2270 *dot = '\0';
2271 }
2272 *p_buflen = strlen(buf);
2273
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002274 modules = PyImport_GetModuleDict();
2275 parent = PyDict_GetItemString(modules, buf);
2276 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002277 PyErr_Format(PyExc_SystemError,
2278 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002279 return parent;
2280 /* We expect, but can't guarantee, if parent != None, that:
2281 - parent.__name__ == buf
2282 - parent.__dict__ is globals
2283 If this is violated... Who cares? */
2284}
2285
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002286/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002287static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002288load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002289 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002290{
2291 char *name = *p_name;
2292 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002293 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002294 char *p;
2295 PyObject *result;
2296
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002297 if (strlen(name) == 0) {
Thomas Wouters8ddab272006-04-04 16:17:02 +00002298 /* completely empty module name should only happen in
2299 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002300 Py_INCREF(mod);
2301 *p_name = NULL;
2302 return mod;
2303 }
2304
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002305 if (dot == NULL) {
2306 *p_name = NULL;
2307 len = strlen(name);
2308 }
2309 else {
2310 *p_name = dot+1;
2311 len = dot-name;
2312 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002313 if (len == 0) {
2314 PyErr_SetString(PyExc_ValueError,
2315 "Empty module name");
2316 return NULL;
2317 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002318
2319 p = buf + *p_buflen;
2320 if (p != buf)
2321 *p++ = '.';
2322 if (p+len-buf >= MAXPATHLEN) {
2323 PyErr_SetString(PyExc_ValueError,
2324 "Module name too long");
2325 return NULL;
2326 }
2327 strncpy(p, name, len);
2328 p[len] = '\0';
2329 *p_buflen = p+len-buf;
2330
2331 result = import_submodule(mod, p, buf);
2332 if (result == Py_None && altmod != mod) {
2333 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002334 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002335 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002336 if (result != NULL && result != Py_None) {
2337 if (mark_miss(buf) != 0) {
2338 Py_DECREF(result);
2339 return NULL;
2340 }
2341 strncpy(buf, name, len);
2342 buf[len] = '\0';
2343 *p_buflen = len;
2344 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002345 }
2346 if (result == NULL)
2347 return NULL;
2348
2349 if (result == Py_None) {
2350 Py_DECREF(result);
2351 PyErr_Format(PyExc_ImportError,
2352 "No module named %.200s", name);
2353 return NULL;
2354 }
2355
2356 return result;
2357}
2358
2359static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002360mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002361{
2362 PyObject *modules = PyImport_GetModuleDict();
2363 return PyDict_SetItemString(modules, name, Py_None);
2364}
2365
2366static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002367ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002368 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002369{
2370 int i;
2371
2372 if (!PyObject_HasAttrString(mod, "__path__"))
2373 return 1;
2374
2375 for (i = 0; ; i++) {
2376 PyObject *item = PySequence_GetItem(fromlist, i);
2377 int hasit;
2378 if (item == NULL) {
2379 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2380 PyErr_Clear();
2381 return 1;
2382 }
2383 return 0;
2384 }
2385 if (!PyString_Check(item)) {
2386 PyErr_SetString(PyExc_TypeError,
2387 "Item in ``from list'' not a string");
2388 Py_DECREF(item);
2389 return 0;
2390 }
2391 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002392 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002393 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002394 /* See if the package defines __all__ */
2395 if (recursive)
2396 continue; /* Avoid endless recursion */
2397 all = PyObject_GetAttrString(mod, "__all__");
2398 if (all == NULL)
2399 PyErr_Clear();
2400 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002401 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002402 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002403 if (!ret)
2404 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002405 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002406 continue;
2407 }
2408 hasit = PyObject_HasAttr(mod, item);
2409 if (!hasit) {
2410 char *subname = PyString_AS_STRING(item);
2411 PyObject *submod;
2412 char *p;
2413 if (buflen + strlen(subname) >= MAXPATHLEN) {
2414 PyErr_SetString(PyExc_ValueError,
2415 "Module name too long");
2416 Py_DECREF(item);
2417 return 0;
2418 }
2419 p = buf + buflen;
2420 *p++ = '.';
2421 strcpy(p, subname);
2422 submod = import_submodule(mod, subname, buf);
2423 Py_XDECREF(submod);
2424 if (submod == NULL) {
2425 Py_DECREF(item);
2426 return 0;
2427 }
2428 }
2429 Py_DECREF(item);
2430 }
2431
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002432 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002433}
2434
Neil Schemenauer00b09662003-06-16 21:03:07 +00002435static int
2436add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2437 PyObject *modules)
2438{
2439 if (mod == Py_None)
2440 return 1;
2441 /* Irrespective of the success of this load, make a
2442 reference to it in the parent package module. A copy gets
2443 saved in the modules dictionary under the full name, so get a
2444 reference from there, if need be. (The exception is when the
2445 load failed with a SyntaxError -- then there's no trace in
2446 sys.modules. In that case, of course, do nothing extra.) */
2447 if (submod == NULL) {
2448 submod = PyDict_GetItemString(modules, fullname);
2449 if (submod == NULL)
2450 return 1;
2451 }
2452 if (PyModule_Check(mod)) {
2453 /* We can't use setattr here since it can give a
2454 * spurious warning if the submodule name shadows a
2455 * builtin name */
2456 PyObject *dict = PyModule_GetDict(mod);
2457 if (!dict)
2458 return 0;
2459 if (PyDict_SetItemString(dict, subname, submod) < 0)
2460 return 0;
2461 }
2462 else {
2463 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2464 return 0;
2465 }
2466 return 1;
2467}
2468
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002469static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002470import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002471{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002472 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002473 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002474
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002475 /* Require:
2476 if mod == None: subname == fullname
2477 else: mod.__name__ + "." + subname == fullname
2478 */
2479
Tim Peters50d8d372001-02-28 05:34:27 +00002480 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002481 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002482 }
2483 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002484 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002485 char buf[MAXPATHLEN+1];
2486 struct filedescr *fdp;
2487 FILE *fp = NULL;
2488
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002489 if (mod == Py_None)
2490 path = NULL;
2491 else {
2492 path = PyObject_GetAttrString(mod, "__path__");
2493 if (path == NULL) {
2494 PyErr_Clear();
2495 Py_INCREF(Py_None);
2496 return Py_None;
2497 }
2498 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002499
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002500 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002501 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2502 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002503 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002504 if (fdp == NULL) {
2505 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2506 return NULL;
2507 PyErr_Clear();
2508 Py_INCREF(Py_None);
2509 return Py_None;
2510 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002511 m = load_module(fullname, fp, buf, fdp->type, loader);
2512 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002513 if (fp)
2514 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002515 if (!add_submodule(mod, m, fullname, subname, modules)) {
2516 Py_XDECREF(m);
2517 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002518 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002519 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002520
2521 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002522}
2523
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002524
2525/* Re-import a module of any kind and return its module object, WITH
2526 INCREMENTED REFERENCE COUNT */
2527
Guido van Rossum79f25d91997-04-29 20:08:16 +00002528PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002529PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002530{
Collin Winter47c52a82007-03-13 23:02:15 +00002531 PyInterpreterState *interp = PyThreadState_Get()->interp;
2532 PyObject *modules_reloading = interp->modules_reloading;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002533 PyObject *modules = PyImport_GetModuleDict();
Collin Winter276887b2007-03-12 16:11:39 +00002534 PyObject *path = NULL, *loader = NULL, *existing_m = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002535 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002536 char buf[MAXPATHLEN+1];
2537 struct filedescr *fdp;
2538 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002539 PyObject *newm;
Collin Winter47c52a82007-03-13 23:02:15 +00002540
2541 if (modules_reloading == NULL) {
2542 Py_FatalError("PyImport_ReloadModule: "
Neal Norwitz2fca81c2007-05-30 04:53:41 +00002543 "no modules_reloading dictionary!");
Collin Winter47c52a82007-03-13 23:02:15 +00002544 return NULL;
2545 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002546
Guido van Rossum79f25d91997-04-29 20:08:16 +00002547 if (m == NULL || !PyModule_Check(m)) {
2548 PyErr_SetString(PyExc_TypeError,
2549 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002550 return NULL;
2551 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002552 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002553 if (name == NULL)
2554 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002555 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002556 PyErr_Format(PyExc_ImportError,
2557 "reload(): module %.200s not in sys.modules",
2558 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002559 return NULL;
2560 }
Neal Norwitz75c7c802007-03-13 05:31:38 +00002561 existing_m = PyDict_GetItemString(modules_reloading, name);
2562 if (existing_m != NULL) {
2563 /* Due to a recursive reload, this module is already
2564 being reloaded. */
2565 Py_INCREF(existing_m);
2566 return existing_m;
Collin Winter276887b2007-03-12 16:11:39 +00002567 }
Neal Norwitz75c7c802007-03-13 05:31:38 +00002568 if (PyDict_SetItemString(modules_reloading, name, m) < 0)
2569 return NULL;
Collin Winter276887b2007-03-12 16:11:39 +00002570
Guido van Rossum222ef561997-09-06 19:41:09 +00002571 subname = strrchr(name, '.');
2572 if (subname == NULL)
2573 subname = name;
2574 else {
2575 PyObject *parentname, *parent;
2576 parentname = PyString_FromStringAndSize(name, (subname-name));
Collin Winter276887b2007-03-12 16:11:39 +00002577 if (parentname == NULL) {
2578 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002579 return NULL;
Neal Norwitz75c7c802007-03-13 05:31:38 +00002580 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002581 parent = PyDict_GetItem(modules, parentname);
2582 if (parent == NULL) {
2583 PyErr_Format(PyExc_ImportError,
2584 "reload(): parent %.200s not in sys.modules",
Georg Brandl0c55f292005-09-14 06:56:20 +00002585 PyString_AS_STRING(parentname));
2586 Py_DECREF(parentname);
Neal Norwitz2fca81c2007-05-30 04:53:41 +00002587 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002588 return NULL;
2589 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002590 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002591 subname++;
2592 path = PyObject_GetAttrString(parent, "__path__");
2593 if (path == NULL)
2594 PyErr_Clear();
2595 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002596 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002597 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002598 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002599
2600 if (fdp == NULL) {
2601 Py_XDECREF(loader);
Collin Winter276887b2007-03-12 16:11:39 +00002602 imp_modules_reloading_clear();
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002603 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002604 }
2605
2606 newm = load_module(name, fp, buf, fdp->type, loader);
2607 Py_XDECREF(loader);
2608
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002609 if (fp)
2610 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002611 if (newm == NULL) {
2612 /* load_module probably removed name from modules because of
2613 * the error. Put back the original module object. We're
2614 * going to return NULL in this case regardless of whether
2615 * replacing name succeeds, so the return value is ignored.
2616 */
2617 PyDict_SetItemString(modules, name, m);
2618 }
Neal Norwitz75c7c802007-03-13 05:31:38 +00002619 imp_modules_reloading_clear();
Tim Peters1cd70172004-08-02 03:52:12 +00002620 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002621}
2622
2623
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002624/* Higher-level import emulator which emulates the "import" statement
2625 more accurately -- it invokes the __import__() function from the
2626 builtins of the current globals. This means that the import is
2627 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002628 environment, e.g. by "rexec".
2629 A dummy list ["__doc__"] is passed as the 4th argument so that
2630 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2631 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002632
2633PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002634PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002635{
2636 static PyObject *silly_list = NULL;
2637 static PyObject *builtins_str = NULL;
2638 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002639 PyObject *globals = NULL;
2640 PyObject *import = NULL;
2641 PyObject *builtins = NULL;
2642 PyObject *r = NULL;
2643
2644 /* Initialize constant string objects */
2645 if (silly_list == NULL) {
2646 import_str = PyString_InternFromString("__import__");
2647 if (import_str == NULL)
2648 return NULL;
2649 builtins_str = PyString_InternFromString("__builtins__");
2650 if (builtins_str == NULL)
2651 return NULL;
2652 silly_list = Py_BuildValue("[s]", "__doc__");
2653 if (silly_list == NULL)
2654 return NULL;
2655 }
2656
2657 /* Get the builtins from current globals */
2658 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002659 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002660 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002661 builtins = PyObject_GetItem(globals, builtins_str);
2662 if (builtins == NULL)
2663 goto err;
2664 }
2665 else {
2666 /* No globals -- use standard builtins, and fake globals */
2667 PyErr_Clear();
2668
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002669 builtins = PyImport_ImportModuleLevel("__builtin__",
2670 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002671 if (builtins == NULL)
2672 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002673 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2674 if (globals == NULL)
2675 goto err;
2676 }
2677
2678 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002679 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002680 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002681 if (import == NULL)
2682 PyErr_SetObject(PyExc_KeyError, import_str);
2683 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002684 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002685 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002686 if (import == NULL)
2687 goto err;
2688
Christian Heimes000a0742008-01-03 22:16:32 +00002689 /* Call the __import__ function with the proper argument list
2690 * Always use absolute import here. */
2691 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
2692 globals, silly_list, 0, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002693
2694 err:
2695 Py_XDECREF(globals);
2696 Py_XDECREF(builtins);
2697 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002698
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002699 return r;
2700}
2701
2702
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002703/* Module 'imp' provides Python access to the primitives used for
2704 importing modules.
2705*/
2706
Guido van Rossum79f25d91997-04-29 20:08:16 +00002707static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002708imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002709{
2710 char buf[4];
2711
Guido van Rossum96774c12000-05-01 20:19:08 +00002712 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2713 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2714 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2715 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002716
Guido van Rossum79f25d91997-04-29 20:08:16 +00002717 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002718}
2719
Guido van Rossum79f25d91997-04-29 20:08:16 +00002720static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002721imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002722{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002723 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002724 struct filedescr *fdp;
2725
Guido van Rossum79f25d91997-04-29 20:08:16 +00002726 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002727 if (list == NULL)
2728 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002729 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2730 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002731 fdp->suffix, fdp->mode, fdp->type);
2732 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002733 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002734 return NULL;
2735 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002736 if (PyList_Append(list, item) < 0) {
2737 Py_DECREF(list);
2738 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002739 return NULL;
2740 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002741 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002742 }
2743 return list;
2744}
2745
Guido van Rossum79f25d91997-04-29 20:08:16 +00002746static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002747call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002748{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002749 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002750 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002751 struct filedescr *fdp;
2752 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002753 FILE *fp = NULL;
2754
2755 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002756 if (path == Py_None)
2757 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002758 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002759 if (fdp == NULL)
2760 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002761 if (fp != NULL) {
2762 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2763 if (fob == NULL) {
2764 fclose(fp);
2765 return NULL;
2766 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002767 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002768 else {
2769 fob = Py_None;
2770 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002771 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002772 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002773 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002774 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002775 return ret;
2776}
2777
Guido van Rossum79f25d91997-04-29 20:08:16 +00002778static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002779imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002780{
2781 char *name;
2782 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002783 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002784 return NULL;
2785 return call_find_module(name, path);
2786}
2787
2788static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002789imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002790{
2791 char *name;
2792 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002793 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002794 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002795 return NULL;
2796 ret = init_builtin(name);
2797 if (ret < 0)
2798 return NULL;
2799 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002800 Py_INCREF(Py_None);
2801 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002802 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002803 m = PyImport_AddModule(name);
2804 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002805 return m;
2806}
2807
Guido van Rossum79f25d91997-04-29 20:08:16 +00002808static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002809imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002810{
2811 char *name;
2812 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002813 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002814 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002815 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002816 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002817 if (ret < 0)
2818 return NULL;
2819 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002820 Py_INCREF(Py_None);
2821 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002822 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002823 m = PyImport_AddModule(name);
2824 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002825 return m;
2826}
2827
Guido van Rossum79f25d91997-04-29 20:08:16 +00002828static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002829imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002830{
2831 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002832
Guido van Rossum43713e52000-02-29 13:59:29 +00002833 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002834 return NULL;
2835 return get_frozen_object(name);
2836}
2837
Guido van Rossum79f25d91997-04-29 20:08:16 +00002838static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002839imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002840{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002841 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002842 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002843 return NULL;
Guido van Rossum50ee94f2002-04-09 18:00:58 +00002844 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002845}
2846
Guido van Rossum79f25d91997-04-29 20:08:16 +00002847static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002848imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002849{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002850 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002851 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002852 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002853 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002854 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002855 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002856}
2857
2858static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002859get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002860{
2861 FILE *fp;
2862 if (fob == NULL) {
Tim Peters86c7d2f2004-08-01 23:24:21 +00002863 if (mode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002864 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002865 fp = fopen(pathname, mode);
2866 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002867 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002868 }
2869 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002870 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002871 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002872 PyErr_SetString(PyExc_ValueError,
2873 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002874 }
2875 return fp;
2876}
2877
Guido van Rossum79f25d91997-04-29 20:08:16 +00002878static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002879imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002880{
2881 char *name;
2882 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002883 PyObject *fob = NULL;
2884 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002885 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002886 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002887 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002888 return NULL;
2889 fp = get_file(pathname, fob, "rb");
2890 if (fp == NULL)
2891 return NULL;
2892 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002893 if (fob == NULL)
2894 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002895 return m;
2896}
2897
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002898#ifdef HAVE_DYNAMIC_LOADING
2899
Guido van Rossum79f25d91997-04-29 20:08:16 +00002900static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002901imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002902{
2903 char *name;
2904 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002905 PyObject *fob = NULL;
2906 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002907 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002908 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002909 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002910 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002911 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002912 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002913 if (fp == NULL)
2914 return NULL;
2915 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002916 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002917 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002918}
2919
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002920#endif /* HAVE_DYNAMIC_LOADING */
2921
Guido van Rossum79f25d91997-04-29 20:08:16 +00002922static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002923imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002924{
2925 char *name;
2926 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002927 PyObject *fob = NULL;
2928 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002929 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002930 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002931 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002932 return NULL;
2933 fp = get_file(pathname, fob, "r");
2934 if (fp == NULL)
2935 return NULL;
2936 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002937 if (fob == NULL)
2938 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002939 return m;
2940}
2941
Guido van Rossum79f25d91997-04-29 20:08:16 +00002942static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002943imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002944{
2945 char *name;
2946 PyObject *fob;
2947 char *pathname;
2948 char *suffix; /* Unused */
2949 char *mode;
2950 int type;
2951 FILE *fp;
2952
Guido van Rossum43713e52000-02-29 13:59:29 +00002953 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002954 &name, &fob, &pathname,
2955 &suffix, &mode, &type))
2956 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002957 if (*mode) {
2958 /* Mode must start with 'r' or 'U' and must not contain '+'.
2959 Implicit in this test is the assumption that the mode
2960 may contain other modifiers like 'b' or 't'. */
2961
2962 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002963 PyErr_Format(PyExc_ValueError,
2964 "invalid file open mode %.200s", mode);
2965 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002966 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002967 }
2968 if (fob == Py_None)
2969 fp = NULL;
2970 else {
2971 if (!PyFile_Check(fob)) {
2972 PyErr_SetString(PyExc_ValueError,
2973 "load_module arg#2 should be a file or None");
2974 return NULL;
2975 }
2976 fp = get_file(pathname, fob, mode);
2977 if (fp == NULL)
2978 return NULL;
2979 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002980 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002981}
2982
2983static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002984imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002985{
2986 char *name;
2987 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002988 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002989 return NULL;
2990 return load_package(name, pathname);
2991}
2992
2993static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002994imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002995{
2996 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002997 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002998 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002999 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003000}
3001
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003002/* Doc strings */
3003
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003004PyDoc_STRVAR(doc_imp,
3005"This module provides the components needed to build your own\n\
3006__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003007
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003008PyDoc_STRVAR(doc_find_module,
3009"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003010Search for a module. If path is omitted or None, search for a\n\
3011built-in, frozen or special module and continue search in sys.path.\n\
3012The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003013package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003014
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003015PyDoc_STRVAR(doc_load_module,
3016"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003017Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003018The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003019
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003020PyDoc_STRVAR(doc_get_magic,
3021"get_magic() -> string\n\
3022Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003023
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003024PyDoc_STRVAR(doc_get_suffixes,
3025"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003026Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003027that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003028
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003029PyDoc_STRVAR(doc_new_module,
3030"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003031Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003032The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003033
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003034PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00003035"lock_held() -> boolean\n\
3036Return True if the import lock is currently held, else False.\n\
3037On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00003038
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003039PyDoc_STRVAR(doc_acquire_lock,
3040"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00003041Acquires the interpreter's import lock for the current thread.\n\
3042This lock should be used by import hooks to ensure thread-safety\n\
3043when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003044On platforms without threads, this function does nothing.");
3045
3046PyDoc_STRVAR(doc_release_lock,
3047"release_lock() -> None\n\
3048Release the interpreter's import lock.\n\
3049On platforms without threads, this function does nothing.");
3050
Guido van Rossum79f25d91997-04-29 20:08:16 +00003051static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00003052 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
3053 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
3054 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
3055 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
3056 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
3057 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
3058 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
3059 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003060 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00003061 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
3062 {"init_builtin", imp_init_builtin, METH_VARARGS},
3063 {"init_frozen", imp_init_frozen, METH_VARARGS},
3064 {"is_builtin", imp_is_builtin, METH_VARARGS},
3065 {"is_frozen", imp_is_frozen, METH_VARARGS},
3066 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003067#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00003068 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003069#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00003070 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00003071 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003072 {NULL, NULL} /* sentinel */
3073};
3074
Guido van Rossum1a8791e1998-08-04 22:46:29 +00003075static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003076setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003077{
3078 PyObject *v;
3079 int err;
3080
3081 v = PyInt_FromLong((long)value);
3082 err = PyDict_SetItemString(d, name, v);
3083 Py_XDECREF(v);
3084 return err;
3085}
3086
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003087typedef struct {
3088 PyObject_HEAD
3089} NullImporter;
3090
3091static int
3092NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
3093{
3094 char *path;
Christian Heimescea681b2007-11-07 17:50:54 +00003095 Py_ssize_t pathlen;
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003096
3097 if (!_PyArg_NoKeywords("NullImporter()", kwds))
3098 return -1;
3099
3100 if (!PyArg_ParseTuple(args, "s:NullImporter",
3101 &path))
3102 return -1;
3103
Christian Heimescea681b2007-11-07 17:50:54 +00003104 pathlen = strlen(path);
3105 if (pathlen == 0) {
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003106 PyErr_SetString(PyExc_ImportError, "empty pathname");
3107 return -1;
3108 } else {
3109#ifndef RISCOS
3110 struct stat statbuf;
3111 int rv;
3112
3113 rv = stat(path, &statbuf);
Christian Heimes004c1c12007-11-07 18:30:22 +00003114#ifdef MS_WINDOWS
3115 /* MS Windows stat() chokes on paths like C:\path\. Try to
3116 * recover *one* time by stripping off a trailing slash or
3117 * backslash. http://bugs.python.org/issue1293
3118 */
Christian Heimescea681b2007-11-07 17:50:54 +00003119 if (rv != 0 && pathlen <= MAXPATHLEN &&
3120 (path[pathlen-1] == '/' || path[pathlen-1] == '\\')) {
3121 char mangled[MAXPATHLEN+1];
3122
3123 strcpy(mangled, path);
3124 mangled[pathlen-1] = '\0';
3125 rv = stat(mangled, &statbuf);
3126 }
Christian Heimescea681b2007-11-07 17:50:54 +00003127#endif
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003128 if (rv == 0) {
3129 /* it exists */
3130 if (S_ISDIR(statbuf.st_mode)) {
3131 /* it's a directory */
3132 PyErr_SetString(PyExc_ImportError,
3133 "existing directory");
3134 return -1;
3135 }
3136 }
3137#else
3138 if (object_exists(path)) {
3139 /* it exists */
3140 if (isdir(path)) {
3141 /* it's a directory */
3142 PyErr_SetString(PyExc_ImportError,
3143 "existing directory");
3144 return -1;
3145 }
3146 }
3147#endif
3148 }
3149 return 0;
3150}
3151
3152static PyObject *
3153NullImporter_find_module(NullImporter *self, PyObject *args)
3154{
3155 Py_RETURN_NONE;
3156}
3157
3158static PyMethodDef NullImporter_methods[] = {
3159 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
3160 "Always return None"
3161 },
3162 {NULL} /* Sentinel */
3163};
3164
3165
Nick Coghlan327a39b2007-11-18 11:56:28 +00003166PyTypeObject PyNullImporter_Type = {
Martin v. Löwis68192102007-07-21 06:55:02 +00003167 PyVarObject_HEAD_INIT(NULL, 0)
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003168 "imp.NullImporter", /*tp_name*/
3169 sizeof(NullImporter), /*tp_basicsize*/
3170 0, /*tp_itemsize*/
3171 0, /*tp_dealloc*/
3172 0, /*tp_print*/
3173 0, /*tp_getattr*/
3174 0, /*tp_setattr*/
3175 0, /*tp_compare*/
3176 0, /*tp_repr*/
3177 0, /*tp_as_number*/
3178 0, /*tp_as_sequence*/
3179 0, /*tp_as_mapping*/
3180 0, /*tp_hash */
3181 0, /*tp_call*/
3182 0, /*tp_str*/
3183 0, /*tp_getattro*/
3184 0, /*tp_setattro*/
3185 0, /*tp_as_buffer*/
3186 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3187 "Null importer object", /* tp_doc */
3188 0, /* tp_traverse */
3189 0, /* tp_clear */
3190 0, /* tp_richcompare */
3191 0, /* tp_weaklistoffset */
3192 0, /* tp_iter */
3193 0, /* tp_iternext */
3194 NullImporter_methods, /* tp_methods */
3195 0, /* tp_members */
3196 0, /* tp_getset */
3197 0, /* tp_base */
3198 0, /* tp_dict */
3199 0, /* tp_descr_get */
3200 0, /* tp_descr_set */
3201 0, /* tp_dictoffset */
3202 (initproc)NullImporter_init, /* tp_init */
3203 0, /* tp_alloc */
3204 PyType_GenericNew /* tp_new */
3205};
3206
3207
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003208PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003209initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003210{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003211 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003212
Nick Coghlan327a39b2007-11-18 11:56:28 +00003213 if (PyType_Ready(&PyNullImporter_Type) < 0)
Phillip J. Ebyf7575d02006-07-28 21:12:07 +00003214 goto failure;
3215
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003216 m = Py_InitModule4("imp", imp_methods, doc_imp,
3217 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003218 if (m == NULL)
3219 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003220 d = PyModule_GetDict(m);
Neal Norwitz3cb31ac2006-08-13 18:10:47 +00003221 if (d == NULL)
3222 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003223
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003224 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3225 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3226 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3227 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3228 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3229 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3230 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3231 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003232 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003233 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003234
Nick Coghlan327a39b2007-11-18 11:56:28 +00003235 Py_INCREF(&PyNullImporter_Type);
3236 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003237 failure:
3238 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003239}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003240
3241
Guido van Rossumb18618d2000-05-03 23:44:39 +00003242/* API for embedding applications that want to add their own entries
3243 to the table of built-in modules. This should normally be called
3244 *before* Py_Initialize(). When the table resize fails, -1 is
3245 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003246
3247 After a similar function by Just van Rossum. */
3248
3249int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003250PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003251{
3252 static struct _inittab *our_copy = NULL;
3253 struct _inittab *p;
3254 int i, n;
3255
3256 /* Count the number of entries in both tables */
3257 for (n = 0; newtab[n].name != NULL; n++)
3258 ;
3259 if (n == 0)
3260 return 0; /* Nothing to do */
3261 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3262 ;
3263
3264 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003265 p = our_copy;
3266 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003267 if (p == NULL)
3268 return -1;
3269
3270 /* Copy the tables into the new memory */
3271 if (our_copy != PyImport_Inittab)
3272 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3273 PyImport_Inittab = our_copy = p;
3274 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3275
3276 return 0;
3277}
3278
3279/* Shorthand to add a single entry given a name and a function */
3280
3281int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003282PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003283{
3284 struct _inittab newtab[2];
3285
3286 memset(newtab, '\0', sizeof newtab);
3287
3288 newtab[0].name = name;
3289 newtab[0].initfunc = initfunc;
3290
3291 return PyImport_ExtendInittab(newtab);
3292}
Anthony Baxterac6bd462006-04-13 02:06:09 +00003293
3294#ifdef __cplusplus
3295}
3296#endif