blob: 221c2dd2d230bca5f502c66361f17b4a1b27820a [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Module definition and import implementation */
3
Guido van Rossum79f25d91997-04-29 20:08:16 +00004#include "Python.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Guido van Rossumd8faa362007-04-27 19:54:29 +00007#undef Yield /* undefine macro conflicting with winbase.h */
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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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
Guido van Rossum45aecf42006-03-15 04:58:47 +000035 The magic numbers must be spaced apart at least 2 values, as the
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000036 -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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000064 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
65 Python 2.5b3: 62111 (fix wrong code: x += yield)
66 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
67 storing constants that should have been removed)
Thomas Wouters89f507f2006-12-13 04:49:30 +000068 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp)
69 Python 2.6a0: 62141 (peephole optimizations)
Guido van Rossum45aecf42006-03-15 04:58:47 +000070 Python 3000: 3000
Brett Cannone2e23ef2006-08-25 05:05:30 +000071 3010 (removed UNARY_CONVERT)
Guido van Rossum86e58e22006-08-28 15:27:34 +000072 3020 (added BUILD_SET)
Guido van Rossum4f72a782006-10-27 23:31:49 +000073 3030 (added keyword-only parameters)
Guido van Rossum3203bdc2006-12-28 21:03:31 +000074 3040 (added signature annotations)
Guido van Rossum452bf512007-02-09 05:32:43 +000075 3050 (print becomes a function)
Guido van Rossum52cc1d82007-03-18 15:41:51 +000076 3060 (PEP 3115 metaclass syntax)
Guido van Rossum00bc0e02007-10-15 02:52:41 +000077 3070 (PEP 3109 raise changes)
78 3080 (PEP 3137 make __file__ and __name__ unicode)
Guido van Rossum98297ee2007-11-06 21:34:58 +000079 3090 (kill str8 interning)
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000080.
Tim Peters36515e22001-11-18 04:06:29 +000081*/
Guido van Rossum98297ee2007-11-06 21:34:58 +000082#define MAGIC (3090 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000083
Guido van Rossum96774c12000-05-01 20:19:08 +000084/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000085 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000086 compiler works which are enabled by command line switches. */
87static long pyc_magic = MAGIC;
88
Guido van Rossum25ce5661997-08-02 03:10:38 +000089/* See _PyImport_FixupExtension() below */
90static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000091
Guido van Rossum771c6c81997-10-31 18:37:24 +000092/* This table is defined in config.c: */
93extern struct _inittab _PyImport_Inittab[];
94
Guido van Rossumce3a72a2007-10-19 23:16:50 +000095/* Method from Parser/tokenizer.c */
Guido van Rossum40d20bc2007-10-22 00:09:51 +000096extern char * PyTokenizer_FindEncoding(int);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000097
Guido van Rossum771c6c81997-10-31 18:37:24 +000098struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000099
Guido van Rossumed1170e1999-12-20 21:23:41 +0000100/* these tables define the module suffixes that Python recognizes */
101struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000102
Guido van Rossumed1170e1999-12-20 21:23:41 +0000103static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +0000104 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000105#ifdef MS_WINDOWS
Jack Jansenc88da1f2002-05-28 10:58:19 +0000106 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +0000107#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000108 {".pyc", "rb", PY_COMPILED},
109 {0, 0}
110};
111
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000112
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000113/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000114
115void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000116_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000117{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000118 const struct filedescr *scan;
119 struct filedescr *filetab;
120 int countD = 0;
121 int countS = 0;
122
123 /* prepare _PyImport_Filetab: copy entries from
124 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
125 */
Guido van Rossum04110fb2007-08-24 16:32:05 +0000126#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000127 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
128 ++countD;
Guido van Rossum04110fb2007-08-24 16:32:05 +0000129#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000130 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
131 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000132 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000133 if (filetab == NULL)
134 Py_FatalError("Can't initialize import file table.");
Guido van Rossum04110fb2007-08-24 16:32:05 +0000135#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000136 memcpy(filetab, _PyImport_DynLoadFiletab,
137 countD * sizeof(struct filedescr));
Guido van Rossum04110fb2007-08-24 16:32:05 +0000138#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000139 memcpy(filetab + countD, _PyImport_StandardFiletab,
140 countS * sizeof(struct filedescr));
141 filetab[countD + countS].suffix = NULL;
142
143 _PyImport_Filetab = filetab;
144
Guido van Rossum0824f631997-03-11 18:37:35 +0000145 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000146 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
147 for (; filetab->suffix != NULL; filetab++) {
148 if (strcmp(filetab->suffix, ".pyc") == 0)
149 filetab->suffix = ".pyo";
Guido van Rossum0824f631997-03-11 18:37:35 +0000150 }
151 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000152
Guido van Rossum572dbf82007-04-27 23:53:51 +0000153 {
Guido van Rossum96774c12000-05-01 20:19:08 +0000154 /* 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 */
Christian Heimes9cd17752007-11-18 19:35:23 +0000169 if (PyType_Ready(&PyNullImporter_Type) < 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +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 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000259 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000260 import_lock = PyThread_allocate_lock();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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();
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
Guido van Rossumd8faa362007-04-27 19:54:29 +0000347static void
348imp_modules_reloading_clear(void)
349{
350 PyInterpreterState *interp = PyThreadState_Get()->interp;
351 if (interp->modules_reloading != NULL)
352 PyDict_Clear(interp->modules_reloading);
353}
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[] = {
Collin Winter670e6922007-03-21 02:57:17 +0000369 "path", "argv", "ps1", "ps2",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000370 "last_type", "last_value", "last_traceback",
Just van Rossum52e14d62002-12-30 22:08:05 +0000371 "path_hooks", "path_importer_cache", "meta_path",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000372 NULL
373};
374
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000375static char* sys_files[] = {
376 "stdin", "__stdin__",
377 "stdout", "__stdout__",
378 "stderr", "__stderr__",
379 NULL
380};
381
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000382
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000383/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000384
Guido van Rossum3f5da241990-12-20 15:06:42 +0000385void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000386PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000387{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000388 Py_ssize_t pos, ndone;
Guido van Rossum758eec01998-01-19 21:58:26 +0000389 char *name;
390 PyObject *key, *value, *dict;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000391 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000392 PyObject *modules = interp->modules;
393
394 if (modules == NULL)
395 return; /* Already done */
396
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000397 /* Delete some special variables first. These are common
398 places where user values hide and people complain when their
399 destructors fail. Since the modules containing them are
400 deleted *last* of all, they would come too late in the normal
401 destruction order. Sigh. */
402
Georg Brandl1a3284e2007-12-02 09:40:06 +0000403 value = PyDict_GetItemString(modules, "builtins");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000404 if (value != NULL && PyModule_Check(value)) {
405 dict = PyModule_GetDict(value);
406 if (Py_VerboseFlag)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000407 PySys_WriteStderr("# clear builtins._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000408 PyDict_SetItemString(dict, "_", Py_None);
409 }
410 value = PyDict_GetItemString(modules, "sys");
411 if (value != NULL && PyModule_Check(value)) {
412 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000413 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000414 dict = PyModule_GetDict(value);
415 for (p = sys_deletes; *p != NULL; p++) {
416 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000417 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000418 PyDict_SetItemString(dict, *p, Py_None);
419 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000420 for (p = sys_files; *p != NULL; p+=2) {
421 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000422 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000423 v = PyDict_GetItemString(dict, *(p+1));
424 if (v == NULL)
425 v = Py_None;
426 PyDict_SetItemString(dict, *p, v);
427 }
428 }
429
430 /* First, delete __main__ */
431 value = PyDict_GetItemString(modules, "__main__");
432 if (value != NULL && PyModule_Check(value)) {
433 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000434 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000435 _PyModule_Clear(value);
436 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000437 }
438
Georg Brandl1a3284e2007-12-02 09:40:06 +0000439 /* The special treatment of "builtins" here is because even
Guido van Rossum758eec01998-01-19 21:58:26 +0000440 when it's not referenced as a module, its dictionary is
441 referenced by almost every module's __builtins__. Since
442 deleting a module clears its dictionary (even if there are
Georg Brandl1a3284e2007-12-02 09:40:06 +0000443 references left to it), we need to delete the "builtins"
Guido van Rossum758eec01998-01-19 21:58:26 +0000444 module last. Likewise, we don't delete sys until the very
445 end because it is implicitly referenced (e.g. by print).
446
447 Also note that we 'delete' modules by replacing their entry
448 in the modules dict with None, rather than really deleting
449 them; this avoids a rehash of the modules dictionary and
450 also marks them as "non existent" so they won't be
451 re-imported. */
452
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000453 /* Next, repeatedly delete modules with a reference count of
Georg Brandl1a3284e2007-12-02 09:40:06 +0000454 one (skipping builtins and sys) and delete them */
Guido van Rossum758eec01998-01-19 21:58:26 +0000455 do {
456 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000457 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000458 while (PyDict_Next(modules, &pos, &key, &value)) {
459 if (value->ob_refcnt != 1)
460 continue;
Neal Norwitz80e7f272007-08-26 06:45:23 +0000461 if (PyUnicode_Check(key) && PyModule_Check(value)) {
462 name = PyUnicode_AsString(key);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000463 if (strcmp(name, "builtins") == 0)
Guido van Rossum758eec01998-01-19 21:58:26 +0000464 continue;
465 if (strcmp(name, "sys") == 0)
466 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000467 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000468 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000469 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000470 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000471 PyDict_SetItem(modules, key, Py_None);
472 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000473 }
474 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000475 } while (ndone > 0);
476
Georg Brandl1a3284e2007-12-02 09:40:06 +0000477 /* Next, delete all modules (still skipping builtins and sys) */
Guido van Rossum758eec01998-01-19 21:58:26 +0000478 pos = 0;
479 while (PyDict_Next(modules, &pos, &key, &value)) {
Neal Norwitz80e7f272007-08-26 06:45:23 +0000480 if (PyUnicode_Check(key) && PyModule_Check(value)) {
481 name = PyUnicode_AsString(key);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000482 if (strcmp(name, "builtins") == 0)
Guido van Rossum758eec01998-01-19 21:58:26 +0000483 continue;
484 if (strcmp(name, "sys") == 0)
485 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000486 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000487 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000488 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000489 PyDict_SetItem(modules, key, Py_None);
490 }
491 }
492
Georg Brandl1a3284e2007-12-02 09:40:06 +0000493 /* Next, delete sys and builtins (in that order) */
Guido van Rossum758eec01998-01-19 21:58:26 +0000494 value = PyDict_GetItemString(modules, "sys");
495 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000496 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000497 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000498 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000499 PyDict_SetItemString(modules, "sys", Py_None);
500 }
Georg Brandl1a3284e2007-12-02 09:40:06 +0000501 value = PyDict_GetItemString(modules, "builtins");
Guido van Rossum758eec01998-01-19 21:58:26 +0000502 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000503 if (Py_VerboseFlag)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000504 PySys_WriteStderr("# cleanup builtins\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000505 _PyModule_Clear(value);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000506 PyDict_SetItemString(modules, "builtins", Py_None);
Guido van Rossum758eec01998-01-19 21:58:26 +0000507 }
508
509 /* Finally, clear and delete the modules directory */
510 PyDict_Clear(modules);
511 interp->modules = NULL;
512 Py_DECREF(modules);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000513 Py_CLEAR(interp->modules_reloading);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000514}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000515
516
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000517/* Helper for pythonrun.c -- return magic number */
518
519long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000520PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000521{
Guido van Rossum96774c12000-05-01 20:19:08 +0000522 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000523}
524
525
Guido van Rossum25ce5661997-08-02 03:10:38 +0000526/* Magic for extension modules (built-in as well as dynamically
527 loaded). To prevent initializing an extension module more than
528 once, we keep a static dictionary 'extensions' keyed by module name
529 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000530 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000531 dictionary is stored by calling _PyImport_FixupExtension()
532 immediately after the module initialization function succeeds. A
533 copy can be retrieved from there by calling
534 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000535
Guido van Rossum79f25d91997-04-29 20:08:16 +0000536PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000537_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000538{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000539 PyObject *modules, *mod, *dict, *copy;
540 if (extensions == NULL) {
541 extensions = PyDict_New();
542 if (extensions == NULL)
543 return NULL;
544 }
545 modules = PyImport_GetModuleDict();
546 mod = PyDict_GetItemString(modules, name);
547 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000548 PyErr_Format(PyExc_SystemError,
549 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000550 return NULL;
551 }
552 dict = PyModule_GetDict(mod);
553 if (dict == NULL)
554 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000555 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000556 if (copy == NULL)
557 return NULL;
558 PyDict_SetItemString(extensions, filename, copy);
559 Py_DECREF(copy);
560 return copy;
561}
562
563PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000564_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000565{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000566 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567 if (extensions == NULL)
568 return NULL;
569 dict = PyDict_GetItemString(extensions, filename);
570 if (dict == NULL)
571 return NULL;
572 mod = PyImport_AddModule(name);
573 if (mod == NULL)
574 return NULL;
575 mdict = PyModule_GetDict(mod);
576 if (mdict == NULL)
577 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000578 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000579 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000580 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000581 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000582 name, filename);
583 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000584}
585
586
587/* Get the module object corresponding to a module name.
588 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000589 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000590 Because the former action is most common, THIS DOES NOT RETURN A
591 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000592
Guido van Rossum79f25d91997-04-29 20:08:16 +0000593PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000594PyImport_AddModule(const char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000595{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000596 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000597 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000598
Guido van Rossum25ce5661997-08-02 03:10:38 +0000599 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000600 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000601 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000602 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000603 if (m == NULL)
604 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000605 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000606 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000607 return NULL;
608 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000609 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000610
611 return m;
612}
613
Tim Peters1cd70172004-08-02 03:52:12 +0000614/* Remove name from sys.modules, if it's there. */
615static void
616_RemoveModule(const char *name)
617{
618 PyObject *modules = PyImport_GetModuleDict();
619 if (PyDict_GetItemString(modules, name) == NULL)
620 return;
621 if (PyDict_DelItemString(modules, name) < 0)
622 Py_FatalError("import: deleting existing key in"
623 "sys.modules failed");
624}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000625
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000626/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000627 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
628 * removed from sys.modules, to avoid leaving damaged module objects
629 * in sys.modules. The caller may wish to restore the original
630 * module object (if any) in this case; PyImport_ReloadModule is an
631 * example.
632 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000633PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000634PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000635{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000636 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
637}
638
639PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000640PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000641{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000642 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000643 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000644
Guido van Rossum79f25d91997-04-29 20:08:16 +0000645 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000646 if (m == NULL)
647 return NULL;
Jeremy Hyltonecd91292004-01-02 23:25:32 +0000648 /* If the module is being reloaded, we get the old module back
649 and re-use its dict to exec the new code. */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000650 d = PyModule_GetDict(m);
651 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
652 if (PyDict_SetItemString(d, "__builtins__",
653 PyEval_GetBuiltins()) != 0)
Tim Peters1cd70172004-08-02 03:52:12 +0000654 goto error;
Guido van Rossum6135a871995-01-09 17:53:26 +0000655 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000656 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000657 v = NULL;
658 if (pathname != NULL) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +0000659 v = PyUnicode_DecodeFSDefault(pathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000660 if (v == NULL)
661 PyErr_Clear();
662 }
663 if (v == NULL) {
664 v = ((PyCodeObject *)co)->co_filename;
665 Py_INCREF(v);
666 }
667 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000668 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000669 Py_DECREF(v);
670
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000671 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000672 if (v == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000673 goto error;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000674 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000675
Guido van Rossum25ce5661997-08-02 03:10:38 +0000676 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000677 PyErr_Format(PyExc_ImportError,
678 "Loaded module %.200s not found in sys.modules",
679 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000680 return NULL;
681 }
682
Guido van Rossum79f25d91997-04-29 20:08:16 +0000683 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000684
685 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000686
687 error:
688 _RemoveModule(name);
689 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000690}
691
692
693/* Given a pathname for a Python source file, fill a buffer with the
694 pathname for the corresponding compiled file. Return the pathname
695 for the compiled file, or NULL if there's no space in the buffer.
696 Doesn't set an exception. */
697
698static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000699make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000700{
Tim Petersc1731372001-08-04 08:12:36 +0000701 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000702 if (len+2 > buflen)
703 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000704
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000705#ifdef MS_WINDOWS
Tim Petersc1731372001-08-04 08:12:36 +0000706 /* Treat .pyw as if it were .py. The case of ".pyw" must match
707 that used in _PyImport_StandardFiletab. */
708 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
709 --len; /* pretend 'w' isn't there */
710#endif
711 memcpy(buf, pathname, len);
712 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
713 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000714
715 return buf;
716}
717
718
719/* Given a pathname for a Python source file, its time of last
720 modification, and a pathname for a compiled file, check whether the
721 compiled file represents the same version of the source. If so,
722 return a FILE pointer for the compiled file, positioned just after
723 the header; if not, return NULL.
724 Doesn't set an exception. */
725
726static FILE *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000727check_compiled_module(char *pathname, time_t mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000728{
729 FILE *fp;
730 long magic;
731 long pyc_mtime;
732
733 fp = fopen(cpathname, "rb");
734 if (fp == NULL)
735 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000736 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000737 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000738 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000739 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000740 fclose(fp);
741 return NULL;
742 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000743 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000744 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000745 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000746 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000747 fclose(fp);
748 return NULL;
749 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000750 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000751 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000752 return fp;
753}
754
755
756/* Read a code object from a file and check it for validity */
757
Guido van Rossum79f25d91997-04-29 20:08:16 +0000758static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000759read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000760{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000761 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000762
Tim Petersd9b9ac82001-01-28 00:27:39 +0000763 co = PyMarshal_ReadLastObjectFromFile(fp);
Armin Rigo01ab2792004-03-26 15:09:27 +0000764 if (co == NULL)
765 return NULL;
766 if (!PyCode_Check(co)) {
767 PyErr_Format(PyExc_ImportError,
768 "Non-code object in %.200s", cpathname);
769 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000770 return NULL;
771 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000772 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000773}
774
775
776/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000777 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000778
Guido van Rossum79f25d91997-04-29 20:08:16 +0000779static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000780load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000781{
782 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000783 PyCodeObject *co;
784 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000785
Guido van Rossum79f25d91997-04-29 20:08:16 +0000786 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000787 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000788 PyErr_Format(PyExc_ImportError,
789 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000790 return NULL;
791 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000792 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000793 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000794 if (co == NULL)
795 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000796 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000797 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000798 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000799 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000800 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000801
802 return m;
803}
804
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000805/* Parse a source file and return the corresponding code object */
806
Guido van Rossum79f25d91997-04-29 20:08:16 +0000807static PyCodeObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000808parse_source_module(const char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000809{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000810 PyCodeObject *co = NULL;
811 mod_ty mod;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000812 PyArena *arena = PyArena_New();
813 if (arena == NULL)
814 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000815
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000816 mod = PyParser_ASTFromFile(fp, pathname, NULL,
817 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 }
Thomas Wouters89f507f2006-12-13 04:49:30 +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);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000953 if (cpathname)
954 write_compiled_module(co, cpathname, mtime);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000955 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000956 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000957 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000958
959 return m;
960}
961
962
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000963/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +0000964static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
965static struct filedescr *find_module(char *, char *, PyObject *,
966 char *, size_t, FILE **, PyObject **);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000967static struct _frozen *find_frozen(char *name);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000968
969/* Load a package and return its module object WITH INCREMENTED
970 REFERENCE COUNT */
971
972static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000973load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000974{
Tim Peters1cd70172004-08-02 03:52:12 +0000975 PyObject *m, *d;
976 PyObject *file = NULL;
977 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000978 int err;
979 char buf[MAXPATHLEN+1];
980 FILE *fp = NULL;
981 struct filedescr *fdp;
982
983 m = PyImport_AddModule(name);
984 if (m == NULL)
985 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000986 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000987 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000988 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000989 d = PyModule_GetDict(m);
Guido van Rossum00bc0e02007-10-15 02:52:41 +0000990 file = PyUnicode_DecodeFSDefault(pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000991 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000992 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000993 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +0000994 if (path == NULL)
995 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000996 err = PyDict_SetItemString(d, "__file__", file);
997 if (err == 0)
998 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +0000999 if (err != 0)
1000 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001001 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00001002 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001003 if (fdp == NULL) {
1004 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1005 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +00001006 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001007 }
1008 else
1009 m = NULL;
1010 goto cleanup;
1011 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001012 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001013 if (fp != NULL)
1014 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00001015 goto cleanup;
1016
1017 error:
1018 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001019 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001020 Py_XDECREF(path);
1021 Py_XDECREF(file);
1022 return m;
1023}
1024
1025
1026/* Helper to test for built-in module */
1027
1028static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001029is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001030{
1031 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001032 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1033 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1034 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001035 return -1;
1036 else
1037 return 1;
1038 }
1039 }
1040 return 0;
1041}
1042
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001043
Just van Rossum52e14d62002-12-30 22:08:05 +00001044/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1045 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +00001046 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +00001047 that can handle the path item. Return None if no hook could;
1048 this tells our caller it should fall back to the builtin
1049 import mechanism. Cache the result in path_importer_cache.
1050 Returns a borrowed reference. */
1051
1052static PyObject *
1053get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1054 PyObject *p)
1055{
1056 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001057 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001058
1059 /* These conditions are the caller's responsibility: */
1060 assert(PyList_Check(path_hooks));
1061 assert(PyDict_Check(path_importer_cache));
1062
1063 nhooks = PyList_Size(path_hooks);
1064 if (nhooks < 0)
1065 return NULL; /* Shouldn't happen */
1066
1067 importer = PyDict_GetItem(path_importer_cache, p);
1068 if (importer != NULL)
1069 return importer;
1070
1071 /* set path_importer_cache[p] to None to avoid recursion */
1072 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1073 return NULL;
1074
1075 for (j = 0; j < nhooks; j++) {
1076 PyObject *hook = PyList_GetItem(path_hooks, j);
1077 if (hook == NULL)
1078 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001079 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001080 if (importer != NULL)
1081 break;
1082
1083 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1084 return NULL;
1085 }
1086 PyErr_Clear();
1087 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001088 if (importer == NULL) {
1089 importer = PyObject_CallFunctionObjArgs(
Christian Heimes9cd17752007-11-18 19:35:23 +00001090 (PyObject *)&PyNullImporter_Type, p, NULL
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001091 );
1092 if (importer == NULL) {
1093 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1094 PyErr_Clear();
1095 return Py_None;
1096 }
1097 }
1098 }
1099 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001100 int err = PyDict_SetItem(path_importer_cache, p, importer);
1101 Py_DECREF(importer);
1102 if (err != 0)
1103 return NULL;
1104 }
1105 return importer;
1106}
1107
Christian Heimes9cd17752007-11-18 19:35:23 +00001108PyAPI_FUNC(PyObject *)
1109PyImport_GetImporter(PyObject *path) {
1110 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
1111
1112 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) {
1113 if ((path_hooks = PySys_GetObject("path_hooks"))) {
1114 importer = get_path_importer(path_importer_cache,
1115 path_hooks, path);
1116 }
1117 }
1118 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1119 return importer;
1120}
1121
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001122/* Search the path (default sys.path) for a module. Return the
1123 corresponding filedescr struct, and (via return arguments) the
1124 pathname and an open file. Return NULL if the module is not found. */
1125
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001126#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001127extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001128 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001129#endif
1130
Martin v. Löwis18e16552006-02-15 17:27:45 +00001131static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001132static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001133static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001134
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001135static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001136find_module(char *fullname, char *subname, PyObject *path, char *buf,
1137 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001138{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001139 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001140 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001141 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001142 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001143 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001144 PyObject *path_hooks, *path_importer_cache;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001145 struct stat statbuf;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001146 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1147 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1148 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001149 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001150#if defined(PYOS_OS2)
1151 size_t saved_len;
1152 size_t saved_namelen;
1153 char *saved_buf = NULL;
1154#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001155 if (p_loader != NULL)
1156 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001157
Just van Rossum52e14d62002-12-30 22:08:05 +00001158 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001159 PyErr_SetString(PyExc_OverflowError,
1160 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001161 return NULL;
1162 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001163 strcpy(name, subname);
1164
1165 /* sys.meta_path import hook */
1166 if (p_loader != NULL) {
1167 PyObject *meta_path;
1168
1169 meta_path = PySys_GetObject("meta_path");
1170 if (meta_path == NULL || !PyList_Check(meta_path)) {
1171 PyErr_SetString(PyExc_ImportError,
1172 "sys.meta_path must be a list of "
1173 "import hooks");
1174 return NULL;
1175 }
1176 Py_INCREF(meta_path); /* zap guard */
1177 npath = PyList_Size(meta_path);
1178 for (i = 0; i < npath; i++) {
1179 PyObject *loader;
1180 PyObject *hook = PyList_GetItem(meta_path, i);
1181 loader = PyObject_CallMethod(hook, "find_module",
1182 "sO", fullname,
1183 path != NULL ?
1184 path : Py_None);
1185 if (loader == NULL) {
1186 Py_DECREF(meta_path);
1187 return NULL; /* true error */
1188 }
1189 if (loader != Py_None) {
1190 /* a loader was found */
1191 *p_loader = loader;
1192 Py_DECREF(meta_path);
1193 return &importhookdescr;
1194 }
1195 Py_DECREF(loader);
1196 }
1197 Py_DECREF(meta_path);
1198 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001199
Guido van Rossum13d77992007-07-23 03:16:50 +00001200 if (path != NULL && PyUnicode_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001201 /* The only type of submodule allowed inside a "frozen"
1202 package are other frozen modules or packages. */
Guido van Rossum13d77992007-07-23 03:16:50 +00001203 char *p = PyUnicode_AsString(path);
1204 if (strlen(p) + 1 + strlen(name) >= (size_t)buflen) {
Guido van Rossum0506a431998-08-11 15:07:39 +00001205 PyErr_SetString(PyExc_ImportError,
1206 "full frozen module name too long");
1207 return NULL;
1208 }
Guido van Rossum13d77992007-07-23 03:16:50 +00001209 strcpy(buf, p);
Guido van Rossum0506a431998-08-11 15:07:39 +00001210 strcat(buf, ".");
1211 strcat(buf, name);
1212 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001213 if (find_frozen(name) != NULL) {
1214 strcpy(buf, name);
1215 return &fd_frozen;
1216 }
1217 PyErr_Format(PyExc_ImportError,
1218 "No frozen submodule named %.200s", name);
1219 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001220 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001221 if (path == NULL) {
1222 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001223 strcpy(buf, name);
1224 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001225 }
Greg Ward201baee2001-10-04 14:52:06 +00001226 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001227 strcpy(buf, name);
1228 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001229 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001230
Guido van Rossumac279101996-08-22 23:10:58 +00001231#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001232 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1233 if (fp != NULL) {
1234 *p_fp = fp;
1235 return fdp;
1236 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001237#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001238 path = PySys_GetObject("path");
1239 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001240 if (path == NULL || !PyList_Check(path)) {
1241 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001242 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001243 return NULL;
1244 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001245
1246 path_hooks = PySys_GetObject("path_hooks");
1247 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1248 PyErr_SetString(PyExc_ImportError,
1249 "sys.path_hooks must be a list of "
1250 "import hooks");
1251 return NULL;
1252 }
1253 path_importer_cache = PySys_GetObject("path_importer_cache");
1254 if (path_importer_cache == NULL ||
1255 !PyDict_Check(path_importer_cache)) {
1256 PyErr_SetString(PyExc_ImportError,
1257 "sys.path_importer_cache must be a dict");
1258 return NULL;
1259 }
1260
Guido van Rossum79f25d91997-04-29 20:08:16 +00001261 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001262 namelen = strlen(name);
1263 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001264 PyObject *v = PyList_GetItem(path, i);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001265 PyObject *origv = v;
Guido van Rossum6262cc72007-05-09 23:29:27 +00001266 const char *base;
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001267 Py_ssize_t size;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001268 if (!v)
1269 return NULL;
Guido van Rossuma4b8d1d2007-08-27 15:02:28 +00001270 if (PyUnicode_Check(v)) {
1271 v = _PyUnicode_AsDefaultEncodedString(v, NULL);
1272 if (v == NULL)
1273 return NULL;
1274 }
1275 if (!PyString_Check(v))
1276 continue;
1277 base = PyString_AS_STRING(v);
1278 size = PyString_GET_SIZE(v);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001279 len = size;
Walter Dörwald3430d702002-06-17 10:43:59 +00001280 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001281 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001282 }
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001283 strcpy(buf, base);
Walter Dörwald3430d702002-06-17 10:43:59 +00001284 if (strlen(buf) != len) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001285 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001286 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001287
1288 /* sys.path_hooks import hook */
1289 if (p_loader != NULL) {
1290 PyObject *importer;
1291
1292 importer = get_path_importer(path_importer_cache,
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001293 path_hooks, origv);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001294 if (importer == NULL) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001295 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001296 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001297 /* Note: importer is a borrowed reference */
1298 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001299 PyObject *loader;
1300 loader = PyObject_CallMethod(importer,
1301 "find_module",
1302 "s", fullname);
1303 if (loader == NULL)
1304 return NULL; /* error */
1305 if (loader != Py_None) {
1306 /* a loader was found */
1307 *p_loader = loader;
1308 return &importhookdescr;
1309 }
1310 Py_DECREF(loader);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001311 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001312 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001313 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001314 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001315
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001316 if (len > 0 && buf[len-1] != SEP
1317#ifdef ALTSEP
1318 && buf[len-1] != ALTSEP
1319#endif
1320 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001321 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001322 strcpy(buf+len, name);
1323 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001324
1325 /* Check for package import (buf holds a directory name,
1326 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001327#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001328 if (stat(buf, &statbuf) == 0 && /* it exists */
1329 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001330 case_ok(buf, len, namelen, name)) { /* case matches */
1331 if (find_init_module(buf)) { /* and has __init__.py */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001332 return &fd_package;
1333 }
1334 else {
1335 char warnstr[MAXPATHLEN+80];
1336 sprintf(warnstr, "Not importing directory "
1337 "'%.*s': missing __init__.py",
1338 MAXPATHLEN, buf);
Skip Montanaro46fc3372007-08-12 11:44:53 +00001339 if (PyErr_WarnEx(PyExc_ImportWarning,
1340 warnstr, 1)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001341 return NULL;
1342 }
1343 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001344 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001345#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001346#if defined(PYOS_OS2)
1347 /* take a snapshot of the module spec for restoration
1348 * after the 8 character DLL hackery
1349 */
1350 saved_buf = strdup(buf);
1351 saved_len = len;
1352 saved_namelen = namelen;
1353#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001354 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Guido van Rossum04110fb2007-08-24 16:32:05 +00001355#if defined(PYOS_OS2) && defined(HAVE_DYNAMIC_LOADING)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001356 /* OS/2 limits DLLs to 8 character names (w/o
1357 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001358 * so if the name is longer than that and its a
1359 * dynamically loaded module we're going to try,
1360 * truncate the name before trying
1361 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001362 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001363 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001364 const struct filedescr *scan;
1365 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001366 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001367 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001368 break;
1369 else
1370 scan++;
1371 }
1372 if (scan->suffix != NULL) {
1373 /* yes, so truncate the name */
1374 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001375 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001376 buf[len] = '\0';
1377 }
1378 }
1379#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001380 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001381 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001382 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001383 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001384 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001385 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001386 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001387 if (fp != NULL) {
1388 if (case_ok(buf, len, namelen, name))
1389 break;
1390 else { /* continue search */
1391 fclose(fp);
1392 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001393 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001394 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001395#if defined(PYOS_OS2)
1396 /* restore the saved snapshot */
1397 strcpy(buf, saved_buf);
1398 len = saved_len;
1399 namelen = saved_namelen;
1400#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001401 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001402#if defined(PYOS_OS2)
1403 /* don't need/want the module name snapshot anymore */
1404 if (saved_buf)
1405 {
1406 free(saved_buf);
1407 saved_buf = NULL;
1408 }
1409#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001410 if (fp != NULL)
1411 break;
1412 }
1413 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001414 PyErr_Format(PyExc_ImportError,
1415 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001416 return NULL;
1417 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001418 *p_fp = fp;
1419 return fdp;
1420}
1421
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001422/* Helpers for main.c
1423 * Find the source file corresponding to a named module
1424 */
1425struct filedescr *
1426_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1427 size_t buflen, FILE **p_fp, PyObject **p_loader)
1428{
1429 return find_module((char *) name, (char *) name, path,
1430 buf, buflen, p_fp, p_loader);
1431}
1432
1433PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1434{
1435 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1436}
1437
Martin v. Löwis18e16552006-02-15 17:27:45 +00001438/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001439 * The arguments here are tricky, best shown by example:
1440 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1441 * ^ ^ ^ ^
1442 * |--------------------- buf ---------------------|
1443 * |------------------- len ------------------|
1444 * |------ name -------|
1445 * |----- namelen -----|
1446 * buf is the full path, but len only counts up to (& exclusive of) the
1447 * extension. name is the module name, also exclusive of extension.
1448 *
1449 * We've already done a successful stat() or fopen() on buf, so know that
1450 * there's some match, possibly case-insensitive.
1451 *
Tim Peters50d8d372001-02-28 05:34:27 +00001452 * case_ok() is to return 1 if there's a case-sensitive match for
1453 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1454 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001455 *
Tim Peters50d8d372001-02-28 05:34:27 +00001456 * case_ok() is used to implement case-sensitive import semantics even
1457 * on platforms with case-insensitive filesystems. It's trivial to implement
1458 * for case-sensitive filesystems. It's pretty much a cross-platform
1459 * nightmare for systems with case-insensitive filesystems.
1460 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001461
Tim Peters50d8d372001-02-28 05:34:27 +00001462/* First we may need a pile of platform-specific header files; the sequence
1463 * of #if's here should match the sequence in the body of case_ok().
1464 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001465#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001466#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001467
Tim Peters50d8d372001-02-28 05:34:27 +00001468#elif defined(DJGPP)
1469#include <dir.h>
1470
Jason Tishler7961aa62005-05-20 00:56:54 +00001471#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001472#include <sys/types.h>
1473#include <dirent.h>
1474
Andrew MacIntyred9400542002-02-26 11:41:34 +00001475#elif defined(PYOS_OS2)
1476#define INCL_DOS
1477#define INCL_DOSERRORS
1478#define INCL_NOPMAPI
1479#include <os2.h>
Tim Peters50d8d372001-02-28 05:34:27 +00001480#endif
1481
Guido van Rossum0980bd91998-02-13 17:18:36 +00001482static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001483case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001484{
Tim Peters50d8d372001-02-28 05:34:27 +00001485/* Pick a platform-specific implementation; the sequence of #if's here should
1486 * match the sequence just above.
1487 */
1488
Jason Tishler7961aa62005-05-20 00:56:54 +00001489/* MS_WINDOWS */
1490#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001491 WIN32_FIND_DATA data;
1492 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001493
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001494 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001495 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001496
Guido van Rossum0980bd91998-02-13 17:18:36 +00001497 h = FindFirstFile(buf, &data);
1498 if (h == INVALID_HANDLE_VALUE) {
1499 PyErr_Format(PyExc_NameError,
1500 "Can't find file for module %.100s\n(filename %.300s)",
1501 name, buf);
1502 return 0;
1503 }
1504 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001505 return strncmp(data.cFileName, name, namelen) == 0;
1506
1507/* DJGPP */
1508#elif defined(DJGPP)
1509 struct ffblk ffblk;
1510 int done;
1511
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001512 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001513 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001514
1515 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1516 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001517 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001518 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001519 name, buf);
1520 return 0;
1521 }
Tim Peters50d8d372001-02-28 05:34:27 +00001522 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001523
Jason Tishler7961aa62005-05-20 00:56:54 +00001524/* new-fangled macintosh (macosx) or Cygwin */
1525#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001526 DIR *dirp;
1527 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001528 char dirname[MAXPATHLEN + 1];
1529 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001530
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001531 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001532 return 1;
1533
Tim Petersd1e87a82001-03-01 18:12:00 +00001534 /* Copy the dir component into dirname; substitute "." if empty */
1535 if (dirlen <= 0) {
1536 dirname[0] = '.';
1537 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001538 }
1539 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001540 assert(dirlen <= MAXPATHLEN);
1541 memcpy(dirname, buf, dirlen);
1542 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001543 }
1544 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001545 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001546 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001547 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001548 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001549 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001550#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001551 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001552#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001553 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001554#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001555 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001556 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001557 (void)closedir(dirp);
1558 return 1; /* Found */
1559 }
1560 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001561 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001562 }
Tim Peters430f5d42001-03-01 01:30:56 +00001563 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001564
Andrew MacIntyred9400542002-02-26 11:41:34 +00001565/* OS/2 */
1566#elif defined(PYOS_OS2)
1567 HDIR hdir = 1;
1568 ULONG srchcnt = 1;
1569 FILEFINDBUF3 ffbuf;
1570 APIRET rc;
1571
1572 if (getenv("PYTHONCASEOK") != NULL)
1573 return 1;
1574
1575 rc = DosFindFirst(buf,
1576 &hdir,
1577 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1578 &ffbuf, sizeof(ffbuf),
1579 &srchcnt,
1580 FIL_STANDARD);
1581 if (rc != NO_ERROR)
1582 return 0;
1583 return strncmp(ffbuf.achName, name, namelen) == 0;
1584
Tim Peters50d8d372001-02-28 05:34:27 +00001585/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1586#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001587 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001588
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001589#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001590}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001591
Guido van Rossum0980bd91998-02-13 17:18:36 +00001592
Guido van Rossum197346f1997-10-31 18:38:52 +00001593#ifdef HAVE_STAT
1594/* Helper to look for __init__.py or __init__.py[co] in potential package */
1595static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001596find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001597{
Tim Peters0f9431f2001-07-05 03:47:53 +00001598 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001599 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001600 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001601 struct stat statbuf;
1602
Tim Peters0f9431f2001-07-05 03:47:53 +00001603/* For calling case_ok(buf, len, namelen, name):
1604 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1605 * ^ ^ ^ ^
1606 * |--------------------- buf ---------------------|
1607 * |------------------- len ------------------|
1608 * |------ name -------|
1609 * |----- namelen -----|
1610 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001611 if (save_len + 13 >= MAXPATHLEN)
1612 return 0;
1613 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001614 pname = buf + i;
1615 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001616 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001617 if (case_ok(buf,
1618 save_len + 9, /* len("/__init__") */
1619 8, /* len("__init__") */
1620 pname)) {
1621 buf[save_len] = '\0';
1622 return 1;
1623 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001624 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001625 i += strlen(pname);
1626 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001627 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001628 if (case_ok(buf,
1629 save_len + 9, /* len("/__init__") */
1630 8, /* len("__init__") */
1631 pname)) {
1632 buf[save_len] = '\0';
1633 return 1;
1634 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001635 }
1636 buf[save_len] = '\0';
1637 return 0;
1638}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001639
Guido van Rossum197346f1997-10-31 18:38:52 +00001640#endif /* HAVE_STAT */
1641
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001642
Tim Petersdbd9ba62000-07-09 03:09:57 +00001643static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001644
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001645/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001646 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001647
Guido van Rossum79f25d91997-04-29 20:08:16 +00001648static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001649load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001650{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001651 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001652 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001653 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001654
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001655 /* First check that there's an open file (if we need one) */
1656 switch (type) {
1657 case PY_SOURCE:
1658 case PY_COMPILED:
1659 if (fp == NULL) {
1660 PyErr_Format(PyExc_ValueError,
1661 "file object required for import (type code %d)",
1662 type);
1663 return NULL;
1664 }
1665 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001666
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001667 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001668
1669 case PY_SOURCE:
1670 m = load_source_module(name, buf, fp);
1671 break;
1672
1673 case PY_COMPILED:
1674 m = load_compiled_module(name, buf, fp);
1675 break;
1676
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001677#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001678 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001679 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001680 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001681#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001682
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001683 case PKG_DIRECTORY:
1684 m = load_package(name, buf);
1685 break;
1686
1687 case C_BUILTIN:
1688 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001689 if (buf != NULL && buf[0] != '\0')
1690 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001691 if (type == C_BUILTIN)
1692 err = init_builtin(name);
1693 else
1694 err = PyImport_ImportFrozenModule(name);
1695 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001696 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001697 if (err == 0) {
1698 PyErr_Format(PyExc_ImportError,
1699 "Purported %s module %.200s not found",
1700 type == C_BUILTIN ?
1701 "builtin" : "frozen",
1702 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001703 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001704 }
1705 modules = PyImport_GetModuleDict();
1706 m = PyDict_GetItemString(modules, name);
1707 if (m == NULL) {
1708 PyErr_Format(
1709 PyExc_ImportError,
1710 "%s module %.200s not properly initialized",
1711 type == C_BUILTIN ?
1712 "builtin" : "frozen",
1713 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001714 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001715 }
1716 Py_INCREF(m);
1717 break;
1718
Just van Rossum52e14d62002-12-30 22:08:05 +00001719 case IMP_HOOK: {
1720 if (loader == NULL) {
1721 PyErr_SetString(PyExc_ImportError,
1722 "import hook without loader");
1723 return NULL;
1724 }
1725 m = PyObject_CallMethod(loader, "load_module", "s", name);
1726 break;
1727 }
1728
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001729 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001730 PyErr_Format(PyExc_ImportError,
1731 "Don't know how to import %.200s (type code %d)",
1732 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001733 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001734
1735 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001736
1737 return m;
1738}
1739
1740
1741/* Initialize a built-in module.
Thomas Wouters89f507f2006-12-13 04:49:30 +00001742 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001743 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001744
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001745static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001746init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001747{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001748 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001749
Greg Ward201baee2001-10-04 14:52:06 +00001750 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001751 return 1;
1752
Guido van Rossum771c6c81997-10-31 18:37:24 +00001753 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001754 if (strcmp(name, p->name) == 0) {
1755 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001756 PyErr_Format(PyExc_ImportError,
1757 "Cannot re-init internal module %.200s",
1758 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001759 return -1;
1760 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001761 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001762 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001763 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001764 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001765 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001766 if (_PyImport_FixupExtension(name, name) == NULL)
1767 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001768 return 1;
1769 }
1770 }
1771 return 0;
1772}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001773
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001774
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001775/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001776
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001777static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001778find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001779{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001780 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001781
Guido van Rossum79f25d91997-04-29 20:08:16 +00001782 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001783 if (p->name == NULL)
1784 return NULL;
1785 if (strcmp(p->name, name) == 0)
1786 break;
1787 }
1788 return p;
1789}
1790
Guido van Rossum79f25d91997-04-29 20:08:16 +00001791static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001792get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001793{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001794 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001795 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001796
1797 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001798 PyErr_Format(PyExc_ImportError,
1799 "No such frozen object named %.200s",
1800 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001801 return NULL;
1802 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001803 if (p->code == NULL) {
1804 PyErr_Format(PyExc_ImportError,
1805 "Excluded frozen object named %.200s",
1806 name);
1807 return NULL;
1808 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001809 size = p->size;
1810 if (size < 0)
1811 size = -size;
1812 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001813}
1814
1815/* Initialize a frozen module.
1816 Return 1 for succes, 0 if the module is not found, and -1 with
1817 an exception set if the initialization failed.
1818 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001819
1820int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001821PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001822{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001823 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001824 PyObject *co;
1825 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001826 int ispackage;
1827 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001828
1829 if (p == NULL)
1830 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001831 if (p->code == NULL) {
1832 PyErr_Format(PyExc_ImportError,
1833 "Excluded frozen object named %.200s",
1834 name);
1835 return -1;
1836 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001837 size = p->size;
1838 ispackage = (size < 0);
1839 if (ispackage)
1840 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001841 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001842 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001843 name, ispackage ? " package" : "");
1844 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001845 if (co == NULL)
1846 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001847 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001848 PyErr_Format(PyExc_TypeError,
1849 "frozen object %.200s is not a code object",
1850 name);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001851 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001852 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001853 if (ispackage) {
1854 /* Set __path__ to the package name */
1855 PyObject *d, *s;
1856 int err;
1857 m = PyImport_AddModule(name);
1858 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001859 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001860 d = PyModule_GetDict(m);
Martin v. Löwis5b222132007-06-10 09:51:05 +00001861 s = PyUnicode_InternFromString(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001862 if (s == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001863 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001864 err = PyDict_SetItemString(d, "__path__", s);
1865 Py_DECREF(s);
1866 if (err != 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001867 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001868 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001869 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001870 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001871 goto err_return;
1872 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001873 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001874 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001875err_return:
1876 Py_DECREF(co);
1877 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001878}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001879
1880
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001881/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001882 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001883
Guido van Rossum79f25d91997-04-29 20:08:16 +00001884PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001885PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001886{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001887 PyObject *pname;
1888 PyObject *result;
1889
Martin v. Löwis5b222132007-06-10 09:51:05 +00001890 pname = PyUnicode_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001891 if (pname == NULL)
1892 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001893 result = PyImport_Import(pname);
1894 Py_DECREF(pname);
1895 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001896}
1897
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001898/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001899static PyObject *get_parent(PyObject *globals, char *buf,
1900 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001901static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001902 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001903static int mark_miss(char *name);
1904static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001905 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001906static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001907
1908/* The Magnum Opus of dotted-name import :-) */
1909
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001910static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001911import_module_level(char *name, PyObject *globals, PyObject *locals,
1912 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001913{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001914 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00001915 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001916 PyObject *parent, *head, *next, *tail;
1917
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001918 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001919 if (parent == NULL)
1920 return NULL;
1921
1922 head = load_next(parent, Py_None, &name, buf, &buflen);
1923 if (head == NULL)
1924 return NULL;
1925
1926 tail = head;
1927 Py_INCREF(tail);
1928 while (name) {
1929 next = load_next(tail, tail, &name, buf, &buflen);
1930 Py_DECREF(tail);
1931 if (next == NULL) {
1932 Py_DECREF(head);
1933 return NULL;
1934 }
1935 tail = next;
1936 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001937 if (tail == Py_None) {
1938 /* If tail is Py_None, both get_parent and load_next found
1939 an empty module name: someone called __import__("") or
1940 doctored faulty bytecode */
1941 Py_DECREF(tail);
1942 Py_DECREF(head);
1943 PyErr_SetString(PyExc_ValueError,
1944 "Empty module name");
1945 return NULL;
1946 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001947
1948 if (fromlist != NULL) {
1949 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
1950 fromlist = NULL;
1951 }
1952
1953 if (fromlist == NULL) {
1954 Py_DECREF(tail);
1955 return head;
1956 }
1957
1958 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001959 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001960 Py_DECREF(tail);
1961 return NULL;
1962 }
1963
1964 return tail;
1965}
1966
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001967/* For DLL compatibility */
1968#undef PyImport_ImportModuleEx
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001969PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001970PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
1971 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001972{
1973 PyObject *result;
1974 lock_import();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001975 result = import_module_level(name, globals, locals, fromlist, -1);
1976 if (unlock_import() < 0) {
1977 Py_XDECREF(result);
1978 PyErr_SetString(PyExc_RuntimeError,
1979 "not holding the import lock");
1980 return NULL;
1981 }
1982 return result;
1983}
1984#define PyImport_ImportModuleEx(n, g, l, f) \
1985 PyImport_ImportModuleLevel(n, g, l, f, -1);
1986
1987PyObject *
1988PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
1989 PyObject *fromlist, int level)
1990{
1991 PyObject *result;
1992 lock_import();
1993 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00001994 if (unlock_import() < 0) {
1995 Py_XDECREF(result);
1996 PyErr_SetString(PyExc_RuntimeError,
1997 "not holding the import lock");
1998 return NULL;
1999 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002000 return result;
2001}
2002
Fred Drake87590902004-05-28 20:21:36 +00002003/* Return the package that an import is being performed in. If globals comes
2004 from the module foo.bar.bat (not itself a package), this returns the
2005 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002006 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002007
2008 The *name* of the returned package is returned in buf, with the length of
2009 the name in *p_buflen.
2010
2011 If globals doesn't come from a package or a module in a package, or a
2012 corresponding entry is not found in sys.modules, Py_None is returned.
2013*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002014static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002015get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002016{
2017 static PyObject *namestr = NULL;
2018 static PyObject *pathstr = NULL;
2019 PyObject *modname, *modpath, *modules, *parent;
2020
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002021 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002022 return Py_None;
2023
2024 if (namestr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002025 namestr = PyUnicode_InternFromString("__name__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002026 if (namestr == NULL)
2027 return NULL;
2028 }
2029 if (pathstr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002030 pathstr = PyUnicode_InternFromString("__path__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002031 if (pathstr == NULL)
2032 return NULL;
2033 }
2034
2035 *buf = '\0';
2036 *p_buflen = 0;
2037 modname = PyDict_GetItem(globals, namestr);
Neal Norwitz80e7f272007-08-26 06:45:23 +00002038 if (modname == NULL || !PyUnicode_Check(modname))
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002039 return Py_None;
2040
2041 modpath = PyDict_GetItem(globals, pathstr);
2042 if (modpath != NULL) {
Neal Norwitz80e7f272007-08-26 06:45:23 +00002043 Py_ssize_t len = PyUnicode_GET_SIZE(modname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002044 if (len > MAXPATHLEN) {
2045 PyErr_SetString(PyExc_ValueError,
2046 "Module name too long");
2047 return NULL;
2048 }
Neal Norwitz80e7f272007-08-26 06:45:23 +00002049 strcpy(buf, PyUnicode_AsString(modname));
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002050 }
2051 else {
Neal Norwitz80e7f272007-08-26 06:45:23 +00002052 char *start = PyUnicode_AsString(modname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002053 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002054 size_t len;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002055 if (lastdot == NULL && level > 0) {
2056 PyErr_SetString(PyExc_ValueError,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002057 "Attempted relative import in non-package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002058 return NULL;
2059 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002060 if (lastdot == NULL)
2061 return Py_None;
2062 len = lastdot - start;
2063 if (len >= MAXPATHLEN) {
2064 PyErr_SetString(PyExc_ValueError,
2065 "Module name too long");
2066 return NULL;
2067 }
2068 strncpy(buf, start, len);
2069 buf[len] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002070 }
2071
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002072 while (--level > 0) {
2073 char *dot = strrchr(buf, '.');
2074 if (dot == NULL) {
2075 PyErr_SetString(PyExc_ValueError,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002076 "Attempted relative import beyond "
2077 "toplevel package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002078 return NULL;
2079 }
2080 *dot = '\0';
2081 }
2082 *p_buflen = strlen(buf);
2083
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002084 modules = PyImport_GetModuleDict();
2085 parent = PyDict_GetItemString(modules, buf);
2086 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002087 PyErr_Format(PyExc_SystemError,
2088 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002089 return parent;
2090 /* We expect, but can't guarantee, if parent != None, that:
2091 - parent.__name__ == buf
2092 - parent.__dict__ is globals
2093 If this is violated... Who cares? */
2094}
2095
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002096/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002097static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002098load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002099 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002100{
2101 char *name = *p_name;
2102 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002103 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002104 char *p;
2105 PyObject *result;
2106
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002107 if (strlen(name) == 0) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002108 /* completely empty module name should only happen in
2109 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002110 Py_INCREF(mod);
2111 *p_name = NULL;
2112 return mod;
2113 }
2114
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002115 if (dot == NULL) {
2116 *p_name = NULL;
2117 len = strlen(name);
2118 }
2119 else {
2120 *p_name = dot+1;
2121 len = dot-name;
2122 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002123 if (len == 0) {
2124 PyErr_SetString(PyExc_ValueError,
2125 "Empty module name");
2126 return NULL;
2127 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002128
2129 p = buf + *p_buflen;
2130 if (p != buf)
2131 *p++ = '.';
2132 if (p+len-buf >= MAXPATHLEN) {
2133 PyErr_SetString(PyExc_ValueError,
2134 "Module name too long");
2135 return NULL;
2136 }
2137 strncpy(p, name, len);
2138 p[len] = '\0';
2139 *p_buflen = p+len-buf;
2140
2141 result = import_submodule(mod, p, buf);
2142 if (result == Py_None && altmod != mod) {
2143 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002144 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002145 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002146 if (result != NULL && result != Py_None) {
2147 if (mark_miss(buf) != 0) {
2148 Py_DECREF(result);
2149 return NULL;
2150 }
2151 strncpy(buf, name, len);
2152 buf[len] = '\0';
2153 *p_buflen = len;
2154 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002155 }
2156 if (result == NULL)
2157 return NULL;
2158
2159 if (result == Py_None) {
2160 Py_DECREF(result);
2161 PyErr_Format(PyExc_ImportError,
2162 "No module named %.200s", name);
2163 return NULL;
2164 }
2165
2166 return result;
2167}
2168
2169static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002170mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002171{
2172 PyObject *modules = PyImport_GetModuleDict();
2173 return PyDict_SetItemString(modules, name, Py_None);
2174}
2175
2176static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002177ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002178 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002179{
2180 int i;
2181
2182 if (!PyObject_HasAttrString(mod, "__path__"))
2183 return 1;
2184
2185 for (i = 0; ; i++) {
2186 PyObject *item = PySequence_GetItem(fromlist, i);
2187 int hasit;
2188 if (item == NULL) {
2189 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2190 PyErr_Clear();
2191 return 1;
2192 }
2193 return 0;
2194 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002195 if (!PyUnicode_Check(item)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002196 PyErr_SetString(PyExc_TypeError,
Neal Norwitz80e7f272007-08-26 06:45:23 +00002197 "Item in ``from list'' not a string");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002198 Py_DECREF(item);
2199 return 0;
2200 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002201 if (PyUnicode_AS_UNICODE(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002202 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002203 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002204 /* See if the package defines __all__ */
2205 if (recursive)
2206 continue; /* Avoid endless recursion */
2207 all = PyObject_GetAttrString(mod, "__all__");
2208 if (all == NULL)
2209 PyErr_Clear();
2210 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002211 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002212 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002213 if (!ret)
2214 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002215 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002216 continue;
2217 }
2218 hasit = PyObject_HasAttr(mod, item);
2219 if (!hasit) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002220 PyObject *item8;
2221 char *subname;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002222 PyObject *submod;
2223 char *p;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002224 if (!Py_FileSystemDefaultEncoding) {
2225 item8 = PyUnicode_EncodeASCII(PyUnicode_AsUnicode(item),
2226 PyUnicode_GetSize(item),
Martin v. Löwis427dbff2007-06-12 05:53:00 +00002227 NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002228 } else {
Guido van Rossum98297ee2007-11-06 21:34:58 +00002229 item8 = PyUnicode_AsEncodedString(item,
2230 Py_FileSystemDefaultEncoding, NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002231 }
2232 if (!item8) {
2233 PyErr_SetString(PyExc_ValueError, "Cannot encode path item");
2234 return 0;
2235 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00002236 subname = PyString_AS_STRING(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002237 if (buflen + strlen(subname) >= MAXPATHLEN) {
2238 PyErr_SetString(PyExc_ValueError,
2239 "Module name too long");
2240 Py_DECREF(item);
2241 return 0;
2242 }
2243 p = buf + buflen;
2244 *p++ = '.';
2245 strcpy(p, subname);
2246 submod = import_submodule(mod, subname, buf);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002247 Py_DECREF(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002248 Py_XDECREF(submod);
2249 if (submod == NULL) {
2250 Py_DECREF(item);
2251 return 0;
2252 }
2253 }
2254 Py_DECREF(item);
2255 }
2256
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002257 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002258}
2259
Neil Schemenauer00b09662003-06-16 21:03:07 +00002260static int
2261add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2262 PyObject *modules)
2263{
2264 if (mod == Py_None)
2265 return 1;
2266 /* Irrespective of the success of this load, make a
2267 reference to it in the parent package module. A copy gets
2268 saved in the modules dictionary under the full name, so get a
2269 reference from there, if need be. (The exception is when the
2270 load failed with a SyntaxError -- then there's no trace in
2271 sys.modules. In that case, of course, do nothing extra.) */
2272 if (submod == NULL) {
2273 submod = PyDict_GetItemString(modules, fullname);
2274 if (submod == NULL)
2275 return 1;
2276 }
2277 if (PyModule_Check(mod)) {
2278 /* We can't use setattr here since it can give a
2279 * spurious warning if the submodule name shadows a
2280 * builtin name */
2281 PyObject *dict = PyModule_GetDict(mod);
2282 if (!dict)
2283 return 0;
2284 if (PyDict_SetItemString(dict, subname, submod) < 0)
2285 return 0;
2286 }
2287 else {
2288 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2289 return 0;
2290 }
2291 return 1;
2292}
2293
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002294static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002295import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002296{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002297 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002298 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002299
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002300 /* Require:
2301 if mod == None: subname == fullname
2302 else: mod.__name__ + "." + subname == fullname
2303 */
2304
Tim Peters50d8d372001-02-28 05:34:27 +00002305 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002306 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002307 }
2308 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002309 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002310 char buf[MAXPATHLEN+1];
2311 struct filedescr *fdp;
2312 FILE *fp = NULL;
2313
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002314 if (mod == Py_None)
2315 path = NULL;
2316 else {
2317 path = PyObject_GetAttrString(mod, "__path__");
2318 if (path == NULL) {
2319 PyErr_Clear();
2320 Py_INCREF(Py_None);
2321 return Py_None;
2322 }
2323 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002324
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002325 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002326 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2327 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002328 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002329 if (fdp == NULL) {
2330 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2331 return NULL;
2332 PyErr_Clear();
2333 Py_INCREF(Py_None);
2334 return Py_None;
2335 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002336 m = load_module(fullname, fp, buf, fdp->type, loader);
2337 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002338 if (fp)
2339 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002340 if (!add_submodule(mod, m, fullname, subname, modules)) {
2341 Py_XDECREF(m);
2342 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002343 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002344 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002345
2346 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002347}
2348
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002349
2350/* Re-import a module of any kind and return its module object, WITH
2351 INCREMENTED REFERENCE COUNT */
2352
Guido van Rossum79f25d91997-04-29 20:08:16 +00002353PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002354PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002355{
Guido van Rossumd8faa362007-04-27 19:54:29 +00002356 PyInterpreterState *interp = PyThreadState_Get()->interp;
2357 PyObject *modules_reloading = interp->modules_reloading;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002358 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumd8faa362007-04-27 19:54:29 +00002359 PyObject *path = NULL, *loader = NULL, *existing_m = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002360 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002361 char buf[MAXPATHLEN+1];
2362 struct filedescr *fdp;
2363 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002364 PyObject *newm;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002365
2366 if (modules_reloading == NULL) {
2367 Py_FatalError("PyImport_ReloadModule: "
Guido van Rossume7ba4952007-06-06 23:52:48 +00002368 "no modules_reloading dictionary!");
Guido van Rossumd8faa362007-04-27 19:54:29 +00002369 return NULL;
2370 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002371
Guido van Rossum79f25d91997-04-29 20:08:16 +00002372 if (m == NULL || !PyModule_Check(m)) {
2373 PyErr_SetString(PyExc_TypeError,
2374 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002375 return NULL;
2376 }
Neal Norwitz5616c6d2007-08-26 05:32:41 +00002377 name = (char*)PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002378 if (name == NULL)
2379 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002380 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002381 PyErr_Format(PyExc_ImportError,
2382 "reload(): module %.200s not in sys.modules",
2383 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002384 return NULL;
2385 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002386 existing_m = PyDict_GetItemString(modules_reloading, name);
2387 if (existing_m != NULL) {
2388 /* Due to a recursive reload, this module is already
2389 being reloaded. */
2390 Py_INCREF(existing_m);
2391 return existing_m;
2392 }
2393 if (PyDict_SetItemString(modules_reloading, name, m) < 0)
2394 return NULL;
2395
Guido van Rossum222ef561997-09-06 19:41:09 +00002396 subname = strrchr(name, '.');
2397 if (subname == NULL)
2398 subname = name;
2399 else {
2400 PyObject *parentname, *parent;
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002401 parentname = PyUnicode_FromStringAndSize(name, (subname-name));
Guido van Rossumd8faa362007-04-27 19:54:29 +00002402 if (parentname == NULL) {
2403 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002404 return NULL;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002405 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002406 parent = PyDict_GetItem(modules, parentname);
2407 if (parent == NULL) {
2408 PyErr_Format(PyExc_ImportError,
2409 "reload(): parent %.200s not in sys.modules",
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002410 PyUnicode_AsString(parentname));
Georg Brandl0c55f292005-09-14 06:56:20 +00002411 Py_DECREF(parentname);
Guido van Rossume7ba4952007-06-06 23:52:48 +00002412 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002413 return NULL;
2414 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002415 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002416 subname++;
2417 path = PyObject_GetAttrString(parent, "__path__");
2418 if (path == NULL)
2419 PyErr_Clear();
2420 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002421 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002422 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002423 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002424
2425 if (fdp == NULL) {
2426 Py_XDECREF(loader);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002427 imp_modules_reloading_clear();
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002428 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002429 }
2430
2431 newm = load_module(name, fp, buf, fdp->type, loader);
2432 Py_XDECREF(loader);
2433
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002434 if (fp)
2435 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002436 if (newm == NULL) {
2437 /* load_module probably removed name from modules because of
2438 * the error. Put back the original module object. We're
2439 * going to return NULL in this case regardless of whether
2440 * replacing name succeeds, so the return value is ignored.
2441 */
2442 PyDict_SetItemString(modules, name, m);
2443 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002444 imp_modules_reloading_clear();
Tim Peters1cd70172004-08-02 03:52:12 +00002445 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002446}
2447
2448
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002449/* Higher-level import emulator which emulates the "import" statement
2450 more accurately -- it invokes the __import__() function from the
2451 builtins of the current globals. This means that the import is
2452 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002453 environment, e.g. by "rexec".
2454 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00002455 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00002456 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002457
2458PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002459PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002460{
2461 static PyObject *silly_list = NULL;
2462 static PyObject *builtins_str = NULL;
2463 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002464 PyObject *globals = NULL;
2465 PyObject *import = NULL;
2466 PyObject *builtins = NULL;
2467 PyObject *r = NULL;
2468
2469 /* Initialize constant string objects */
2470 if (silly_list == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002471 import_str = PyUnicode_InternFromString("__import__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002472 if (import_str == NULL)
2473 return NULL;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002474 builtins_str = PyUnicode_InternFromString("__builtins__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002475 if (builtins_str == NULL)
2476 return NULL;
2477 silly_list = Py_BuildValue("[s]", "__doc__");
2478 if (silly_list == NULL)
2479 return NULL;
2480 }
2481
2482 /* Get the builtins from current globals */
2483 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002484 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002485 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002486 builtins = PyObject_GetItem(globals, builtins_str);
2487 if (builtins == NULL)
2488 goto err;
2489 }
2490 else {
2491 /* No globals -- use standard builtins, and fake globals */
2492 PyErr_Clear();
2493
Georg Brandl1a3284e2007-12-02 09:40:06 +00002494 builtins = PyImport_ImportModuleLevel("builtins",
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002495 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002496 if (builtins == NULL)
2497 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002498 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2499 if (globals == NULL)
2500 goto err;
2501 }
2502
2503 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002504 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002505 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002506 if (import == NULL)
2507 PyErr_SetObject(PyExc_KeyError, import_str);
2508 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002509 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002510 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002511 if (import == NULL)
2512 goto err;
2513
Guido van Rossumd8faa362007-04-27 19:54:29 +00002514 /* Call the __import__ function with the proper argument list */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002515 r = PyObject_CallFunctionObjArgs(import, module_name, globals,
2516 globals, silly_list, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002517
2518 err:
2519 Py_XDECREF(globals);
2520 Py_XDECREF(builtins);
2521 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002522
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002523 return r;
2524}
2525
2526
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002527/* Module 'imp' provides Python access to the primitives used for
2528 importing modules.
2529*/
2530
Guido van Rossum79f25d91997-04-29 20:08:16 +00002531static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002532imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002533{
2534 char buf[4];
2535
Guido van Rossum96774c12000-05-01 20:19:08 +00002536 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2537 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2538 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2539 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002540
Guido van Rossumad8d3002007-08-03 18:40:49 +00002541 return PyBytes_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002542}
2543
Guido van Rossum79f25d91997-04-29 20:08:16 +00002544static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002545imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002546{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002547 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002548 struct filedescr *fdp;
2549
Guido van Rossum79f25d91997-04-29 20:08:16 +00002550 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002551 if (list == NULL)
2552 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002553 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2554 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002555 fdp->suffix, fdp->mode, fdp->type);
2556 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002557 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002558 return NULL;
2559 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002560 if (PyList_Append(list, item) < 0) {
2561 Py_DECREF(list);
2562 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002563 return NULL;
2564 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002565 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002566 }
2567 return list;
2568}
2569
Guido van Rossum79f25d91997-04-29 20:08:16 +00002570static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002571call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002572{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002573 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002574 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002575 struct filedescr *fdp;
2576 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002577 FILE *fp = NULL;
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002578 int fd = -1;
Brett Cannon3bb42d92007-10-20 03:43:15 +00002579 char *found_encoding = NULL;
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002580 char *encoding = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002581
2582 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002583 if (path == Py_None)
2584 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002585 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002586 if (fdp == NULL)
2587 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002588 if (fp != NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002589 fd = fileno(fp);
2590 if (fd != -1)
2591 fd = dup(fd);
2592 fclose(fp);
2593 fp = NULL;
2594 }
2595 if (fd != -1) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002596 if (strchr(fdp->mode, 'b') == NULL) {
Brett Cannon3bb42d92007-10-20 03:43:15 +00002597 /* PyTokenizer_FindEncoding() returns PyMem_MALLOC'ed
2598 memory. */
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002599 found_encoding = PyTokenizer_FindEncoding(fd);
2600 lseek(fd, 0, 0); /* Reset position */
Brett Cannon3bb42d92007-10-20 03:43:15 +00002601 encoding = (found_encoding != NULL) ? found_encoding :
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002602 (char*)PyUnicode_GetDefaultEncoding();
2603 }
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002604 fob = PyFile_FromFd(fd, pathname, fdp->mode, -1,
Guido van Rossum2dced8b2007-10-30 17:27:30 +00002605 (char*)encoding, NULL, 1);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002606 if (fob == NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002607 close(fd);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002608 PyMem_FREE(found_encoding);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002609 return NULL;
2610 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002611 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002612 else {
2613 fob = Py_None;
2614 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002615 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002616 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002617 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002618 Py_DECREF(fob);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002619 PyMem_FREE(found_encoding);
2620
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002621 return ret;
2622}
2623
Guido van Rossum79f25d91997-04-29 20:08:16 +00002624static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002625imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002626{
2627 char *name;
2628 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002629 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002630 return NULL;
2631 return call_find_module(name, path);
2632}
2633
2634static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002635imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002636{
2637 char *name;
2638 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002639 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002640 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002641 return NULL;
2642 ret = init_builtin(name);
2643 if (ret < 0)
2644 return NULL;
2645 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002646 Py_INCREF(Py_None);
2647 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002648 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002649 m = PyImport_AddModule(name);
2650 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002651 return m;
2652}
2653
Guido van Rossum79f25d91997-04-29 20:08:16 +00002654static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002655imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002656{
2657 char *name;
2658 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002659 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002660 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002661 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002662 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002663 if (ret < 0)
2664 return NULL;
2665 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002666 Py_INCREF(Py_None);
2667 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002668 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002669 m = PyImport_AddModule(name);
2670 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002671 return m;
2672}
2673
Guido van Rossum79f25d91997-04-29 20:08:16 +00002674static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002675imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002676{
2677 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002678
Guido van Rossum43713e52000-02-29 13:59:29 +00002679 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002680 return NULL;
2681 return get_frozen_object(name);
2682}
2683
Guido van Rossum79f25d91997-04-29 20:08:16 +00002684static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002685imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002686{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002687 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002688 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002689 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00002690 return PyLong_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002691}
2692
Guido van Rossum79f25d91997-04-29 20:08:16 +00002693static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002694imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002695{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002696 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002697 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002698 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002699 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002700 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002701 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002702}
2703
2704static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002705get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002706{
2707 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002708 if (mode[0] == 'U')
2709 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002710 if (fob == NULL) {
2711 fp = fopen(pathname, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002712 }
2713 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002714 int fd = PyObject_AsFileDescriptor(fob);
2715 if (fd == -1)
2716 return NULL;
2717 /* XXX This will leak a FILE struct. Fix this!!!!
2718 (But it doesn't leak a file descrioptor!) */
2719 fp = fdopen(fd, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002720 }
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002721 if (fp == NULL)
2722 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002723 return fp;
2724}
2725
Guido van Rossum79f25d91997-04-29 20:08:16 +00002726static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002727imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002728{
2729 char *name;
2730 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002731 PyObject *fob = NULL;
2732 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002733 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002734 if (!PyArg_ParseTuple(args, "ss|O:load_compiled",
2735 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002736 return NULL;
2737 fp = get_file(pathname, fob, "rb");
2738 if (fp == NULL)
2739 return NULL;
2740 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002741 if (fob == NULL)
2742 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002743 return m;
2744}
2745
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002746#ifdef HAVE_DYNAMIC_LOADING
2747
Guido van Rossum79f25d91997-04-29 20:08:16 +00002748static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002749imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002750{
2751 char *name;
2752 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002753 PyObject *fob = NULL;
2754 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002755 FILE *fp = NULL;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002756 if (!PyArg_ParseTuple(args, "ss|O:load_dynamic",
2757 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002758 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002759 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002760 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002761 if (fp == NULL)
2762 return NULL;
2763 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002764 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002765 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002766}
2767
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002768#endif /* HAVE_DYNAMIC_LOADING */
2769
Guido van Rossum79f25d91997-04-29 20:08:16 +00002770static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002771imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002772{
2773 char *name;
2774 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002775 PyObject *fob = NULL;
2776 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002777 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002778 if (!PyArg_ParseTuple(args, "ss|O:load_source",
2779 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002780 return NULL;
2781 fp = get_file(pathname, fob, "r");
2782 if (fp == NULL)
2783 return NULL;
2784 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002785 if (fob == NULL)
2786 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002787 return m;
2788}
2789
Guido van Rossum79f25d91997-04-29 20:08:16 +00002790static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002791imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002792{
2793 char *name;
2794 PyObject *fob;
2795 char *pathname;
2796 char *suffix; /* Unused */
2797 char *mode;
2798 int type;
2799 FILE *fp;
2800
Guido van Rossum43713e52000-02-29 13:59:29 +00002801 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002802 &name, &fob, &pathname,
2803 &suffix, &mode, &type))
2804 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002805 if (*mode) {
2806 /* Mode must start with 'r' or 'U' and must not contain '+'.
2807 Implicit in this test is the assumption that the mode
2808 may contain other modifiers like 'b' or 't'. */
2809
2810 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002811 PyErr_Format(PyExc_ValueError,
2812 "invalid file open mode %.200s", mode);
2813 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002814 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002815 }
2816 if (fob == Py_None)
2817 fp = NULL;
2818 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002819 fp = get_file(NULL, fob, mode);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002820 if (fp == NULL)
2821 return NULL;
2822 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002823 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002824}
2825
2826static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002827imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002828{
2829 char *name;
2830 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002831 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002832 return NULL;
2833 return load_package(name, pathname);
2834}
2835
2836static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002837imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002838{
2839 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002840 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002841 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002842 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002843}
2844
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002845/* Doc strings */
2846
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002847PyDoc_STRVAR(doc_imp,
2848"This module provides the components needed to build your own\n\
2849__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002850
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002851PyDoc_STRVAR(doc_find_module,
2852"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002853Search for a module. If path is omitted or None, search for a\n\
2854built-in, frozen or special module and continue search in sys.path.\n\
2855The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002856package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002857
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002858PyDoc_STRVAR(doc_load_module,
2859"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002860Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002861The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002862
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002863PyDoc_STRVAR(doc_get_magic,
2864"get_magic() -> string\n\
2865Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002866
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002867PyDoc_STRVAR(doc_get_suffixes,
2868"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002869Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002870that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002871
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002872PyDoc_STRVAR(doc_new_module,
2873"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002874Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002875The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002876
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002877PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00002878"lock_held() -> boolean\n\
2879Return True if the import lock is currently held, else False.\n\
2880On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00002881
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002882PyDoc_STRVAR(doc_acquire_lock,
2883"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00002884Acquires the interpreter's import lock for the current thread.\n\
2885This lock should be used by import hooks to ensure thread-safety\n\
2886when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002887On platforms without threads, this function does nothing.");
2888
2889PyDoc_STRVAR(doc_release_lock,
2890"release_lock() -> None\n\
2891Release the interpreter's import lock.\n\
2892On platforms without threads, this function does nothing.");
2893
Guido van Rossum79f25d91997-04-29 20:08:16 +00002894static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002895 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
2896 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
2897 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
2898 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
2899 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
2900 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
2901 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
2902 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002903 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00002904 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
2905 {"init_builtin", imp_init_builtin, METH_VARARGS},
2906 {"init_frozen", imp_init_frozen, METH_VARARGS},
2907 {"is_builtin", imp_is_builtin, METH_VARARGS},
2908 {"is_frozen", imp_is_frozen, METH_VARARGS},
2909 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002910#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00002911 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002912#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002913 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00002914 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002915 {NULL, NULL} /* sentinel */
2916};
2917
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002918static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002919setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002920{
2921 PyObject *v;
2922 int err;
2923
Christian Heimes217cfd12007-12-02 14:31:20 +00002924 v = PyLong_FromLong((long)value);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002925 err = PyDict_SetItemString(d, name, v);
2926 Py_XDECREF(v);
2927 return err;
2928}
2929
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002930typedef struct {
2931 PyObject_HEAD
2932} NullImporter;
2933
2934static int
2935NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
2936{
2937 char *path;
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00002938 Py_ssize_t pathlen;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002939
2940 if (!_PyArg_NoKeywords("NullImporter()", kwds))
2941 return -1;
2942
2943 if (!PyArg_ParseTuple(args, "s:NullImporter",
2944 &path))
2945 return -1;
2946
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00002947 pathlen = strlen(path);
2948 if (pathlen == 0) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002949 PyErr_SetString(PyExc_ImportError, "empty pathname");
2950 return -1;
2951 } else {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002952 struct stat statbuf;
2953 int rv;
2954
2955 rv = stat(path, &statbuf);
Guido van Rossumc1bdbc32007-11-07 17:46:34 +00002956#ifdef MS_WINDOWS
2957 /* MS Windows stat() chokes on paths like C:\path\. Try to
2958 * recover *one* time by stripping off a trailing slash or
2959 * backslash. http://bugs.python.org/issue1293
2960 */
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00002961 if (rv != 0 && pathlen <= MAXPATHLEN &&
2962 (path[pathlen-1] == '/' || path[pathlen-1] == '\\')) {
2963 char mangled[MAXPATHLEN+1];
2964
2965 strcpy(mangled, path);
2966 mangled[pathlen-1] = '\0';
2967 rv = stat(mangled, &statbuf);
2968 }
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00002969#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002970 if (rv == 0) {
2971 /* it exists */
2972 if (S_ISDIR(statbuf.st_mode)) {
2973 /* it's a directory */
2974 PyErr_SetString(PyExc_ImportError,
2975 "existing directory");
2976 return -1;
2977 }
2978 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002979 }
2980 return 0;
2981}
2982
2983static PyObject *
2984NullImporter_find_module(NullImporter *self, PyObject *args)
2985{
2986 Py_RETURN_NONE;
2987}
2988
2989static PyMethodDef NullImporter_methods[] = {
2990 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
2991 "Always return None"
2992 },
2993 {NULL} /* Sentinel */
2994};
2995
2996
Christian Heimes9cd17752007-11-18 19:35:23 +00002997PyTypeObject PyNullImporter_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002998 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002999 "imp.NullImporter", /*tp_name*/
3000 sizeof(NullImporter), /*tp_basicsize*/
3001 0, /*tp_itemsize*/
3002 0, /*tp_dealloc*/
3003 0, /*tp_print*/
3004 0, /*tp_getattr*/
3005 0, /*tp_setattr*/
3006 0, /*tp_compare*/
3007 0, /*tp_repr*/
3008 0, /*tp_as_number*/
3009 0, /*tp_as_sequence*/
3010 0, /*tp_as_mapping*/
3011 0, /*tp_hash */
3012 0, /*tp_call*/
3013 0, /*tp_str*/
3014 0, /*tp_getattro*/
3015 0, /*tp_setattro*/
3016 0, /*tp_as_buffer*/
3017 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3018 "Null importer object", /* tp_doc */
3019 0, /* tp_traverse */
3020 0, /* tp_clear */
3021 0, /* tp_richcompare */
3022 0, /* tp_weaklistoffset */
3023 0, /* tp_iter */
3024 0, /* tp_iternext */
3025 NullImporter_methods, /* tp_methods */
3026 0, /* tp_members */
3027 0, /* tp_getset */
3028 0, /* tp_base */
3029 0, /* tp_dict */
3030 0, /* tp_descr_get */
3031 0, /* tp_descr_set */
3032 0, /* tp_dictoffset */
3033 (initproc)NullImporter_init, /* tp_init */
3034 0, /* tp_alloc */
3035 PyType_GenericNew /* tp_new */
3036};
3037
3038
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003039PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003040initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003041{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003042 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003043
Christian Heimes9cd17752007-11-18 19:35:23 +00003044 if (PyType_Ready(&PyNullImporter_Type) < 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003045 goto failure;
3046
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003047 m = Py_InitModule4("imp", imp_methods, doc_imp,
3048 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003049 if (m == NULL)
3050 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003051 d = PyModule_GetDict(m);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003052 if (d == NULL)
3053 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003054
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003055 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3056 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3057 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3058 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3059 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3060 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3061 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3062 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003063 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003064 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003065
Christian Heimes9cd17752007-11-18 19:35:23 +00003066 Py_INCREF(&PyNullImporter_Type);
3067 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003068 failure:
3069 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003070}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003071
3072
Guido van Rossumb18618d2000-05-03 23:44:39 +00003073/* API for embedding applications that want to add their own entries
3074 to the table of built-in modules. This should normally be called
3075 *before* Py_Initialize(). When the table resize fails, -1 is
3076 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003077
3078 After a similar function by Just van Rossum. */
3079
3080int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003081PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003082{
3083 static struct _inittab *our_copy = NULL;
3084 struct _inittab *p;
3085 int i, n;
3086
3087 /* Count the number of entries in both tables */
3088 for (n = 0; newtab[n].name != NULL; n++)
3089 ;
3090 if (n == 0)
3091 return 0; /* Nothing to do */
3092 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3093 ;
3094
3095 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003096 p = our_copy;
3097 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003098 if (p == NULL)
3099 return -1;
3100
3101 /* Copy the tables into the new memory */
3102 if (our_copy != PyImport_Inittab)
3103 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3104 PyImport_Inittab = our_copy = p;
3105 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3106
3107 return 0;
3108}
3109
3110/* Shorthand to add a single entry given a name and a function */
3111
3112int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003113PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003114{
3115 struct _inittab newtab[2];
3116
3117 memset(newtab, '\0', sizeof newtab);
3118
3119 newtab[0].name = name;
3120 newtab[0].initfunc = initfunc;
3121
3122 return PyImport_ExtendInittab(newtab);
3123}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003124
3125#ifdef __cplusplus
3126}
3127#endif