blob: e657b70b8ef6effc496e43414bae6db6a8ffd7ca [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
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006#include "node.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00007#include "token.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +00009#include "marshal.h"
10#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000011#include "eval.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000012#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000013#include "importdl.h"
Jack Jansen9c96a921995-02-15 22:57:06 +000014#ifdef macintosh
15#include "macglue.h"
16#endif
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
21
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000022extern time_t PyOS_GetLastModificationTime(char *, FILE *);
23 /* In getmtime.c */
Guido van Rossum21d335e1993-10-15 13:01:11 +000024
Guido van Rossum6c849691994-09-26 15:47:17 +000025/* Magic word to reject .pyc files generated by other Python versions */
Guido van Rossum7faeab31995-07-07 22:50:36 +000026/* Change for each incompatible change */
27/* The value of CR and LF is incorporated so if you ever read or write
28 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000029 Apple MPW compiler swaps their values, botching string constants.
30 XXX That probably isn't important anymore.
31*/
Guido van Rossum7faeab31995-07-07 22:50:36 +000032/* XXX Perhaps the magic number should be frozen and a version field
33 added to the .pyc file header? */
Tim Peters36515e22001-11-18 04:06:29 +000034/* New way to come up with the low 16 bits of the magic number:
35 (YEAR-1995) * 10000 + MONTH * 100 + DAY
36 where MONTH and DAY are 1-based.
37 XXX Whatever the "old way" may have been isn't documented.
38 XXX This scheme breaks in 2002, as (2002-1995)*10000 = 70000 doesn't
39 fit in 16 bits.
40 XXX Later, sometimes 1 gets added to MAGIC in order to record that
41 the Unicode -U option is in use. IMO (Tim's), that's a Bad Idea
42 (quite apart from that the -U option doesn't work so isn't used
43 anyway).
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000044
45 XXX MAL, 2002-02-07: I had to modify the MAGIC due to a fix of the
46 UTF-8 encoder (it previously produced invalid UTF-8 for unpaired
47 high surrogates), so I simply bumped the month value to 20 (invalid
48 month) and set the day to 1. This should be recognizable by any
49 algorithm relying on the above scheme. Perhaps we should simply
50 start counting in increments of 10 from now on ?!
51
52 Known values:
53 Python 1.5: 20121
54 Python 1.5.1: 20121
55 Python 1.5.2: 20121
56 Python 2.0: 50823
57 Python 2.0.1: 50823
58 Python 2.1: 60202
59 Python 2.1.1: 60202
60 Python 2.1.2: 60202
61 Python 2.2: 60717
62 Python 2.3a0: 62001
Tim Peters36515e22001-11-18 04:06:29 +000063*/
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000064#define MAGIC (62001 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000065
Guido van Rossum96774c12000-05-01 20:19:08 +000066/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000067 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000068 compiler works which are enabled by command line switches. */
69static long pyc_magic = MAGIC;
70
Guido van Rossum25ce5661997-08-02 03:10:38 +000071/* See _PyImport_FixupExtension() below */
72static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000073
Guido van Rossum771c6c81997-10-31 18:37:24 +000074/* This table is defined in config.c: */
75extern struct _inittab _PyImport_Inittab[];
76
77struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +000078
Guido van Rossumed1170e1999-12-20 21:23:41 +000079/* these tables define the module suffixes that Python recognizes */
80struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +000081
82#ifdef RISCOS
83static const struct filedescr _PyImport_StandardFiletab[] = {
84 {"/py", "r", PY_SOURCE},
85 {"/pyc", "rb", PY_COMPILED},
86 {0, 0}
87};
88#else
Guido van Rossumed1170e1999-12-20 21:23:41 +000089static const struct filedescr _PyImport_StandardFiletab[] = {
90 {".py", "r", PY_SOURCE},
Tim Petersc1731372001-08-04 08:12:36 +000091#ifdef MS_WIN32
92 {".pyw", "r", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +000093#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +000094 {".pyc", "rb", PY_COMPILED},
95 {0, 0}
96};
Guido van Rossum48a680c2001-03-02 06:34:14 +000097#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +000098
Guido van Rossum1ae940a1995-01-02 19:04:15 +000099/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000100
101void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000102_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000103{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000104 const struct filedescr *scan;
105 struct filedescr *filetab;
106 int countD = 0;
107 int countS = 0;
108
109 /* prepare _PyImport_Filetab: copy entries from
110 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
111 */
112 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
113 ++countD;
114 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
115 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000116 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Guido van Rossumed1170e1999-12-20 21:23:41 +0000117 memcpy(filetab, _PyImport_DynLoadFiletab,
118 countD * sizeof(struct filedescr));
119 memcpy(filetab + countD, _PyImport_StandardFiletab,
120 countS * sizeof(struct filedescr));
121 filetab[countD + countS].suffix = NULL;
122
123 _PyImport_Filetab = filetab;
124
Guido van Rossum0824f631997-03-11 18:37:35 +0000125 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000126 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
127 for (; filetab->suffix != NULL; filetab++) {
Guido van Rossum48a680c2001-03-02 06:34:14 +0000128#ifndef RISCOS
Guido van Rossumed1170e1999-12-20 21:23:41 +0000129 if (strcmp(filetab->suffix, ".pyc") == 0)
130 filetab->suffix = ".pyo";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000131#else
132 if (strcmp(filetab->suffix, "/pyc") == 0)
133 filetab->suffix = "/pyo";
134#endif
Guido van Rossum0824f631997-03-11 18:37:35 +0000135 }
136 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000137
138 if (Py_UnicodeFlag) {
139 /* Fix the pyc_magic so that byte compiled code created
140 using the all-Unicode method doesn't interfere with
141 code created in normal operation mode. */
142 pyc_magic = MAGIC + 1;
143 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000144}
145
Guido van Rossum25ce5661997-08-02 03:10:38 +0000146void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000147_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000148{
149 Py_XDECREF(extensions);
150 extensions = NULL;
Barry Warsaw84294482000-10-03 16:02:05 +0000151 PyMem_DEL(_PyImport_Filetab);
152 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000153}
154
155
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000156/* Locking primitives to prevent parallel imports of the same module
157 in different threads to return with a partially loaded module.
158 These calls are serialized by the global interpreter lock. */
159
160#ifdef WITH_THREAD
161
Guido van Rossum49b56061998-10-01 20:42:43 +0000162#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000163
Guido van Rossum65d5b571998-12-21 19:32:43 +0000164static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000165static long import_lock_thread = -1;
166static int import_lock_level = 0;
167
168static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000169lock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000170{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000171 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000172 if (me == -1)
173 return; /* Too bad */
174 if (import_lock == NULL)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000175 import_lock = PyThread_allocate_lock();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000176 if (import_lock_thread == me) {
177 import_lock_level++;
178 return;
179 }
Guido van Rossum65d5b571998-12-21 19:32:43 +0000180 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0)) {
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000181 PyThreadState *tstate = PyEval_SaveThread();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000182 PyThread_acquire_lock(import_lock, 1);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000183 PyEval_RestoreThread(tstate);
184 }
185 import_lock_thread = me;
186 import_lock_level = 1;
187}
188
189static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000190unlock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000191{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000192 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000193 if (me == -1)
194 return; /* Too bad */
195 if (import_lock_thread != me)
196 Py_FatalError("unlock_import: not holding the import lock");
197 import_lock_level--;
198 if (import_lock_level == 0) {
199 import_lock_thread = -1;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000200 PyThread_release_lock(import_lock);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000201 }
202}
203
204#else
205
206#define lock_import()
207#define unlock_import()
208
209#endif
210
Tim Peters69232342001-08-30 05:16:13 +0000211static PyObject *
212imp_lock_held(PyObject *self, PyObject *args)
213{
214 if (!PyArg_ParseTuple(args, ":lock_held"))
215 return NULL;
216#ifdef WITH_THREAD
217 return PyInt_FromLong(import_lock_thread != -1);
218#else
219 return PyInt_FromLong(0);
220#endif
221}
222
Guido van Rossum25ce5661997-08-02 03:10:38 +0000223/* Helper for sys */
224
225PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000226PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000227{
228 PyInterpreterState *interp = PyThreadState_Get()->interp;
229 if (interp->modules == NULL)
230 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
231 return interp->modules;
232}
233
Guido van Rossum3f5da241990-12-20 15:06:42 +0000234
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000235/* List of names to clear in sys */
236static char* sys_deletes[] = {
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000237 "path", "argv", "ps1", "ps2", "exitfunc",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000238 "exc_type", "exc_value", "exc_traceback",
239 "last_type", "last_value", "last_traceback",
240 NULL
241};
242
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000243static char* sys_files[] = {
244 "stdin", "__stdin__",
245 "stdout", "__stdout__",
246 "stderr", "__stderr__",
247 NULL
248};
249
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000250
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000251/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000252
Guido van Rossum3f5da241990-12-20 15:06:42 +0000253void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000254PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000255{
Guido van Rossum758eec01998-01-19 21:58:26 +0000256 int pos, ndone;
257 char *name;
258 PyObject *key, *value, *dict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000259 PyInterpreterState *interp = PyThreadState_Get()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000260 PyObject *modules = interp->modules;
261
262 if (modules == NULL)
263 return; /* Already done */
264
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000265 /* Delete some special variables first. These are common
266 places where user values hide and people complain when their
267 destructors fail. Since the modules containing them are
268 deleted *last* of all, they would come too late in the normal
269 destruction order. Sigh. */
270
271 value = PyDict_GetItemString(modules, "__builtin__");
272 if (value != NULL && PyModule_Check(value)) {
273 dict = PyModule_GetDict(value);
274 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000275 PySys_WriteStderr("# clear __builtin__._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000276 PyDict_SetItemString(dict, "_", Py_None);
277 }
278 value = PyDict_GetItemString(modules, "sys");
279 if (value != NULL && PyModule_Check(value)) {
280 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000281 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000282 dict = PyModule_GetDict(value);
283 for (p = sys_deletes; *p != NULL; p++) {
284 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000285 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000286 PyDict_SetItemString(dict, *p, Py_None);
287 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000288 for (p = sys_files; *p != NULL; p+=2) {
289 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000290 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000291 v = PyDict_GetItemString(dict, *(p+1));
292 if (v == NULL)
293 v = Py_None;
294 PyDict_SetItemString(dict, *p, v);
295 }
296 }
297
298 /* First, delete __main__ */
299 value = PyDict_GetItemString(modules, "__main__");
300 if (value != NULL && PyModule_Check(value)) {
301 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000302 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000303 _PyModule_Clear(value);
304 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000305 }
306
Guido van Rossum758eec01998-01-19 21:58:26 +0000307 /* The special treatment of __builtin__ here is because even
308 when it's not referenced as a module, its dictionary is
309 referenced by almost every module's __builtins__. Since
310 deleting a module clears its dictionary (even if there are
311 references left to it), we need to delete the __builtin__
312 module last. Likewise, we don't delete sys until the very
313 end because it is implicitly referenced (e.g. by print).
314
315 Also note that we 'delete' modules by replacing their entry
316 in the modules dict with None, rather than really deleting
317 them; this avoids a rehash of the modules dictionary and
318 also marks them as "non existent" so they won't be
319 re-imported. */
320
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000321 /* Next, repeatedly delete modules with a reference count of
Guido van Rossum758eec01998-01-19 21:58:26 +0000322 one (skipping __builtin__ and sys) and delete them */
323 do {
324 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000325 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000326 while (PyDict_Next(modules, &pos, &key, &value)) {
327 if (value->ob_refcnt != 1)
328 continue;
Guido van Rossum566373e1998-10-01 15:24:50 +0000329 if (PyString_Check(key) && PyModule_Check(value)) {
330 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000331 if (strcmp(name, "__builtin__") == 0)
332 continue;
333 if (strcmp(name, "sys") == 0)
334 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000335 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000336 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000337 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000338 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000339 PyDict_SetItem(modules, key, Py_None);
340 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000341 }
342 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000343 } while (ndone > 0);
344
Guido van Rossum758eec01998-01-19 21:58:26 +0000345 /* Next, delete all modules (still skipping __builtin__ and sys) */
346 pos = 0;
347 while (PyDict_Next(modules, &pos, &key, &value)) {
Guido van Rossum566373e1998-10-01 15:24:50 +0000348 if (PyString_Check(key) && PyModule_Check(value)) {
349 name = PyString_AS_STRING(key);
Guido van Rossum758eec01998-01-19 21:58:26 +0000350 if (strcmp(name, "__builtin__") == 0)
351 continue;
352 if (strcmp(name, "sys") == 0)
353 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000354 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000355 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000356 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000357 PyDict_SetItem(modules, key, Py_None);
358 }
359 }
360
361 /* Next, delete sys and __builtin__ (in that order) */
362 value = PyDict_GetItemString(modules, "sys");
363 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000364 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000365 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000366 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000367 PyDict_SetItemString(modules, "sys", Py_None);
368 }
369 value = PyDict_GetItemString(modules, "__builtin__");
370 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000371 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000372 PySys_WriteStderr("# cleanup __builtin__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000373 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000374 PyDict_SetItemString(modules, "__builtin__", Py_None);
375 }
376
377 /* Finally, clear and delete the modules directory */
378 PyDict_Clear(modules);
379 interp->modules = NULL;
380 Py_DECREF(modules);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000381}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000382
383
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000384/* Helper for pythonrun.c -- return magic number */
385
386long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000387PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000388{
Guido van Rossum96774c12000-05-01 20:19:08 +0000389 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000390}
391
392
Guido van Rossum25ce5661997-08-02 03:10:38 +0000393/* Magic for extension modules (built-in as well as dynamically
394 loaded). To prevent initializing an extension module more than
395 once, we keep a static dictionary 'extensions' keyed by module name
396 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000397 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000398 dictionary is stored by calling _PyImport_FixupExtension()
399 immediately after the module initialization function succeeds. A
400 copy can be retrieved from there by calling
401 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000402
Guido van Rossum79f25d91997-04-29 20:08:16 +0000403PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000404_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000405{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000406 PyObject *modules, *mod, *dict, *copy;
407 if (extensions == NULL) {
408 extensions = PyDict_New();
409 if (extensions == NULL)
410 return NULL;
411 }
412 modules = PyImport_GetModuleDict();
413 mod = PyDict_GetItemString(modules, name);
414 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000415 PyErr_Format(PyExc_SystemError,
416 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000417 return NULL;
418 }
419 dict = PyModule_GetDict(mod);
420 if (dict == NULL)
421 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000422 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000423 if (copy == NULL)
424 return NULL;
425 PyDict_SetItemString(extensions, filename, copy);
426 Py_DECREF(copy);
427 return copy;
428}
429
430PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000431_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000432{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000433 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000434 if (extensions == NULL)
435 return NULL;
436 dict = PyDict_GetItemString(extensions, filename);
437 if (dict == NULL)
438 return NULL;
439 mod = PyImport_AddModule(name);
440 if (mod == NULL)
441 return NULL;
442 mdict = PyModule_GetDict(mod);
443 if (mdict == NULL)
444 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000445 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000446 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000447 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000448 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000449 name, filename);
450 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000451}
452
453
454/* Get the module object corresponding to a module name.
455 First check the modules dictionary if there's one there,
456 if not, create a new one and insert in in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000457 Because the former action is most common, THIS DOES NOT RETURN A
458 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000459
Guido van Rossum79f25d91997-04-29 20:08:16 +0000460PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000461PyImport_AddModule(char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000462{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000463 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000464 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000465
Guido van Rossum25ce5661997-08-02 03:10:38 +0000466 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000467 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000468 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000469 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000470 if (m == NULL)
471 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000472 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000473 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000474 return NULL;
475 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000476 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000477
478 return m;
479}
480
481
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000482/* Execute a code object in a module and return the module object
483 WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000484
Guido van Rossum79f25d91997-04-29 20:08:16 +0000485PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000486PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000487{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000488 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
489}
490
491PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000492PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000493{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000494 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000495 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000496
Guido van Rossum79f25d91997-04-29 20:08:16 +0000497 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000498 if (m == NULL)
499 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000500 d = PyModule_GetDict(m);
501 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
502 if (PyDict_SetItemString(d, "__builtins__",
503 PyEval_GetBuiltins()) != 0)
Guido van Rossum6135a871995-01-09 17:53:26 +0000504 return NULL;
505 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000506 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000507 v = NULL;
508 if (pathname != NULL) {
509 v = PyString_FromString(pathname);
510 if (v == NULL)
511 PyErr_Clear();
512 }
513 if (v == NULL) {
514 v = ((PyCodeObject *)co)->co_filename;
515 Py_INCREF(v);
516 }
517 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000518 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000519 Py_DECREF(v);
520
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000521 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000522 if (v == NULL)
523 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000524 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000525
Guido van Rossum25ce5661997-08-02 03:10:38 +0000526 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000527 PyErr_Format(PyExc_ImportError,
528 "Loaded module %.200s not found in sys.modules",
529 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000530 return NULL;
531 }
532
Guido van Rossum79f25d91997-04-29 20:08:16 +0000533 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000534
535 return m;
536}
537
538
539/* Given a pathname for a Python source file, fill a buffer with the
540 pathname for the corresponding compiled file. Return the pathname
541 for the compiled file, or NULL if there's no space in the buffer.
542 Doesn't set an exception. */
543
544static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000545make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000546{
Tim Petersc1731372001-08-04 08:12:36 +0000547 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000548 if (len+2 > buflen)
549 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000550
551#ifdef MS_WIN32
552 /* Treat .pyw as if it were .py. The case of ".pyw" must match
553 that used in _PyImport_StandardFiletab. */
554 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
555 --len; /* pretend 'w' isn't there */
556#endif
557 memcpy(buf, pathname, len);
558 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
559 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000560
561 return buf;
562}
563
564
565/* Given a pathname for a Python source file, its time of last
566 modification, and a pathname for a compiled file, check whether the
567 compiled file represents the same version of the source. If so,
568 return a FILE pointer for the compiled file, positioned just after
569 the header; if not, return NULL.
570 Doesn't set an exception. */
571
572static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000573check_compiled_module(char *pathname, long mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000574{
575 FILE *fp;
576 long magic;
577 long pyc_mtime;
578
579 fp = fopen(cpathname, "rb");
580 if (fp == NULL)
581 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000582 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000583 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000584 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000585 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000586 fclose(fp);
587 return NULL;
588 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000589 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000590 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000591 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000592 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000593 fclose(fp);
594 return NULL;
595 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000596 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000597 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000598 return fp;
599}
600
601
602/* Read a code object from a file and check it for validity */
603
Guido van Rossum79f25d91997-04-29 20:08:16 +0000604static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000605read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000606{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000607 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000608
Tim Petersd9b9ac82001-01-28 00:27:39 +0000609 co = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000610 /* Ugly: rd_object() may return NULL with or without error */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000611 if (co == NULL || !PyCode_Check(co)) {
612 if (!PyErr_Occurred())
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000613 PyErr_Format(PyExc_ImportError,
614 "Non-code object in %.200s", cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000615 Py_XDECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000616 return NULL;
617 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000618 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000619}
620
621
622/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000623 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000624
Guido van Rossum79f25d91997-04-29 20:08:16 +0000625static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000626load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000627{
628 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000629 PyCodeObject *co;
630 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000631
Guido van Rossum79f25d91997-04-29 20:08:16 +0000632 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000633 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000634 PyErr_Format(PyExc_ImportError,
635 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000636 return NULL;
637 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000638 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000639 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000640 if (co == NULL)
641 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000642 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000643 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000644 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000645 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000646 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000647
648 return m;
649}
650
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000651/* Parse a source file and return the corresponding code object */
652
Guido van Rossum79f25d91997-04-29 20:08:16 +0000653static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000654parse_source_module(char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000655{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000656 PyCodeObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000657 node *n;
658
Guido van Rossumb05a5c71997-05-07 17:46:13 +0000659 n = PyParser_SimpleParseFile(fp, pathname, Py_file_input);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000660 if (n == NULL)
661 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000662 co = PyNode_Compile(n, pathname);
663 PyNode_Free(n);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000664
665 return co;
666}
667
668
Guido van Rossum55a83382000-09-20 20:31:38 +0000669/* Helper to open a bytecode file for writing in exclusive mode */
670
671static FILE *
672open_exclusive(char *filename)
673{
674#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
675 /* Use O_EXCL to avoid a race condition when another process tries to
676 write the same file. When that happens, our open() call fails,
677 which is just fine (since it's only a cache).
678 XXX If the file exists and is writable but the directory is not
679 writable, the file will never be written. Oh well.
680 */
681 int fd;
682 (void) unlink(filename);
Tim Peters42c83af2000-09-29 04:03:10 +0000683 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
684#ifdef O_BINARY
685 |O_BINARY /* necessary for Windows */
686#endif
Tim Peters50d8d372001-02-28 05:34:27 +0000687
Tim Peters42c83af2000-09-29 04:03:10 +0000688 , 0666);
Guido van Rossum55a83382000-09-20 20:31:38 +0000689 if (fd < 0)
690 return NULL;
691 return fdopen(fd, "wb");
692#else
693 /* Best we can do -- on Windows this can't happen anyway */
694 return fopen(filename, "wb");
695#endif
696}
697
698
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000699/* Write a compiled module to a file, placing the time of last
700 modification of its source into the header.
701 Errors are ignored, if a write error occurs an attempt is made to
702 remove the file. */
703
704static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000705write_compiled_module(PyCodeObject *co, char *cpathname, long mtime)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000706{
707 FILE *fp;
708
Guido van Rossum55a83382000-09-20 20:31:38 +0000709 fp = open_exclusive(cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000710 if (fp == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000711 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000712 PySys_WriteStderr(
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000713 "# can't create %s\n", cpathname);
714 return;
715 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000716 PyMarshal_WriteLongToFile(pyc_magic, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000717 /* First write a 0 for mtime */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000718 PyMarshal_WriteLongToFile(0L, fp);
719 PyMarshal_WriteObjectToFile((PyObject *)co, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000720 if (ferror(fp)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000721 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000722 PySys_WriteStderr("# can't write %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000723 /* Don't keep partial file */
724 fclose(fp);
725 (void) unlink(cpathname);
726 return;
727 }
728 /* Now write the true mtime */
729 fseek(fp, 4L, 0);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000730 PyMarshal_WriteLongToFile(mtime, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000731 fflush(fp);
732 fclose(fp);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000733 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000734 PySys_WriteStderr("# wrote %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000735#ifdef macintosh
Jack Jansencbf630f2000-07-11 21:59:16 +0000736 PyMac_setfiletype(cpathname, 'Pyth', 'PYC ');
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000737#endif
738}
739
740
741/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000742 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
743 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000744
Guido van Rossum79f25d91997-04-29 20:08:16 +0000745static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000746load_source_module(char *name, char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000747{
Fred Drake4c82b232000-06-30 16:18:57 +0000748 time_t mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000749 FILE *fpc;
750 char buf[MAXPATHLEN+1];
751 char *cpathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000752 PyCodeObject *co;
753 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000754
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000755 mtime = PyOS_GetLastModificationTime(pathname, fp);
Fred Drakec63d3e92001-03-01 06:33:32 +0000756 if (mtime == (time_t)(-1))
Fred Drake4c82b232000-06-30 16:18:57 +0000757 return NULL;
758#if SIZEOF_TIME_T > 4
759 /* Python's .pyc timestamp handling presumes that the timestamp fits
760 in 4 bytes. This will be fine until sometime in the year 2038,
761 when a 4-byte signed time_t will overflow.
762 */
763 if (mtime >> 32) {
764 PyErr_SetString(PyExc_OverflowError,
Fred Drakec63d3e92001-03-01 06:33:32 +0000765 "modification time overflows a 4 byte field");
Fred Drake4c82b232000-06-30 16:18:57 +0000766 return NULL;
767 }
768#endif
Tim Peters36515e22001-11-18 04:06:29 +0000769 cpathname = make_compiled_pathname(pathname, buf,
Jeremy Hylton37832f02001-04-13 17:50:20 +0000770 (size_t)MAXPATHLEN + 1);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000771 if (cpathname != NULL &&
772 (fpc = check_compiled_module(pathname, mtime, cpathname))) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000773 co = read_compiled_module(cpathname, fpc);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000774 fclose(fpc);
775 if (co == NULL)
776 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000777 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000778 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000779 name, cpathname);
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000780 pathname = cpathname;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000781 }
782 else {
783 co = parse_source_module(pathname, fp);
784 if (co == NULL)
785 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000786 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000787 PySys_WriteStderr("import %s # from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000788 name, pathname);
789 write_compiled_module(co, cpathname, mtime);
790 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000791 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000792 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000793
794 return m;
795}
796
797
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000798/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000799static PyObject *load_module(char *, FILE *, char *, int);
800static struct filedescr *find_module(char *, PyObject *,
801 char *, size_t, FILE **);
802static struct _frozen *find_frozen(char *name);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000803
804/* Load a package and return its module object WITH INCREMENTED
805 REFERENCE COUNT */
806
807static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000808load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000809{
810 PyObject *m, *d, *file, *path;
811 int err;
812 char buf[MAXPATHLEN+1];
813 FILE *fp = NULL;
814 struct filedescr *fdp;
815
816 m = PyImport_AddModule(name);
817 if (m == NULL)
818 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000819 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000820 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +0000821 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000822 d = PyModule_GetDict(m);
823 file = PyString_FromString(pathname);
824 if (file == NULL)
825 return NULL;
826 path = Py_BuildValue("[O]", file);
827 if (path == NULL) {
828 Py_DECREF(file);
829 return NULL;
830 }
831 err = PyDict_SetItemString(d, "__file__", file);
832 if (err == 0)
833 err = PyDict_SetItemString(d, "__path__", path);
834 if (err != 0) {
835 m = NULL;
836 goto cleanup;
837 }
838 buf[0] = '\0';
839 fdp = find_module("__init__", path, buf, sizeof(buf), &fp);
840 if (fdp == NULL) {
841 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
842 PyErr_Clear();
843 }
844 else
845 m = NULL;
846 goto cleanup;
847 }
848 m = load_module(name, fp, buf, fdp->type);
849 if (fp != NULL)
850 fclose(fp);
851 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000852 Py_XDECREF(path);
853 Py_XDECREF(file);
854 return m;
855}
856
857
858/* Helper to test for built-in module */
859
860static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000861is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000862{
863 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +0000864 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
865 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
866 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000867 return -1;
868 else
869 return 1;
870 }
871 }
872 return 0;
873}
874
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000875
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000876/* Search the path (default sys.path) for a module. Return the
877 corresponding filedescr struct, and (via return arguments) the
878 pathname and an open file. Return NULL if the module is not found. */
879
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000880#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +0000881extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
882 char *, int);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000883#endif
884
Tim Peters50d8d372001-02-28 05:34:27 +0000885static int case_ok(char *, int, int, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000886static int find_init_module(char *); /* Forward */
Guido van Rossum197346f1997-10-31 18:38:52 +0000887
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000888static struct filedescr *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000889find_module(char *realname, PyObject *path, char *buf, size_t buflen,
890 FILE **p_fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000891{
Fred Drake4c82b232000-06-30 16:18:57 +0000892 int i, npath;
893 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +0000894 struct filedescr *fdp = NULL;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +0000895 FILE *fp = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000896#ifndef RISCOS
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000897 struct stat statbuf;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000898#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +0000899 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
900 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
901 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +0000902 char name[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000903
Fred Drake4c82b232000-06-30 16:18:57 +0000904 if (strlen(realname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000905 PyErr_SetString(PyExc_OverflowError,
906 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +0000907 return NULL;
908 }
Guido van Rossum0506a431998-08-11 15:07:39 +0000909 strcpy(name, realname);
910
911 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000912 /* The only type of submodule allowed inside a "frozen"
913 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +0000914 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
915 PyErr_SetString(PyExc_ImportError,
916 "full frozen module name too long");
917 return NULL;
918 }
919 strcpy(buf, PyString_AsString(path));
920 strcat(buf, ".");
921 strcat(buf, name);
922 strcpy(name, buf);
Jack Jansen550fdae2001-10-30 13:08:39 +0000923#ifdef macintosh
924 /* Freezing on the mac works different, and the modules are
925 ** actually on sys.path. So we don't take the quick exit but
926 ** continue with the normal flow.
927 */
928 path = NULL;
929#else
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000930 if (find_frozen(name) != NULL) {
931 strcpy(buf, name);
932 return &fd_frozen;
933 }
934 PyErr_Format(PyExc_ImportError,
935 "No frozen submodule named %.200s", name);
936 return NULL;
Jack Jansen550fdae2001-10-30 13:08:39 +0000937#endif
Guido van Rossum0506a431998-08-11 15:07:39 +0000938 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000939 if (path == NULL) {
940 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000941 strcpy(buf, name);
942 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000943 }
Greg Ward201baee2001-10-04 14:52:06 +0000944 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000945 strcpy(buf, name);
946 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000947 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000948
Guido van Rossumac279101996-08-22 23:10:58 +0000949#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000950 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
951 if (fp != NULL) {
952 *p_fp = fp;
953 return fdp;
954 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +0000955#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +0000956 path = PySys_GetObject("path");
957 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000958 if (path == NULL || !PyList_Check(path)) {
959 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +0000960 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000961 return NULL;
962 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000963 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000964 namelen = strlen(name);
965 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000966 PyObject *v = PyList_GetItem(path, i);
967 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000968 continue;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000969 len = PyString_Size(v);
Guido van Rossumef3d02e1997-07-21 14:54:36 +0000970 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000971 continue; /* Too long */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000972 strcpy(buf, PyString_AsString(v));
Fred Drake4c82b232000-06-30 16:18:57 +0000973 if (strlen(buf) != len)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000974 continue; /* v contains '\0' */
Jack Jansen9c96a921995-02-15 22:57:06 +0000975#ifdef macintosh
Guido van Rossum741689d1997-08-12 14:53:39 +0000976#ifdef INTERN_STRINGS
Tim Peters50d8d372001-02-28 05:34:27 +0000977 /*
Guido van Rossum741689d1997-08-12 14:53:39 +0000978 ** Speedup: each sys.path item is interned, and
979 ** FindResourceModule remembers which items refer to
980 ** folders (so we don't have to bother trying to look
Tim Peters50d8d372001-02-28 05:34:27 +0000981 ** into them for resources).
Guido van Rossum741689d1997-08-12 14:53:39 +0000982 */
983 PyString_InternInPlace(&PyList_GET_ITEM(path, i));
984 v = PyList_GET_ITEM(path, i);
985#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000986 if (PyMac_FindResourceModule((PyStringObject *)v, name, buf)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000987 static struct filedescr resfiledescr =
988 {"", "", PY_RESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +0000989
Jack Jansen9c96a921995-02-15 22:57:06 +0000990 return &resfiledescr;
991 }
Guido van Rossum0f84a341998-08-06 13:36:01 +0000992 if (PyMac_FindCodeResourceModule((PyStringObject *)v, name, buf)) {
993 static struct filedescr resfiledescr =
994 {"", "", PY_CODERESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +0000995
Guido van Rossum0f84a341998-08-06 13:36:01 +0000996 return &resfiledescr;
997 }
Jack Jansen9c96a921995-02-15 22:57:06 +0000998#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000999 if (len > 0 && buf[len-1] != SEP
1000#ifdef ALTSEP
1001 && buf[len-1] != ALTSEP
1002#endif
1003 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001004 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001005 strcpy(buf+len, name);
1006 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001007
1008 /* Check for package import (buf holds a directory name,
1009 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001010#ifdef HAVE_STAT
Tim Peterscab3f682001-04-29 22:21:25 +00001011 if (stat(buf, &statbuf) == 0 && /* it exists */
1012 S_ISDIR(statbuf.st_mode) && /* it's a directory */
1013 find_init_module(buf) && /* it has __init__.py */
1014 case_ok(buf, len, namelen, name)) /* and case matches */
1015 return &fd_package;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001016#else
1017 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001018#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001019 if (isdir(buf) &&
1020 find_init_module(buf) &&
1021 case_ok(buf, len, namelen, name))
1022 return &fd_package;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001023#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001024#endif
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001025#ifdef macintosh
1026 fdp = PyMac_FindModuleExtension(buf, &len, name);
Jack Jansen4df3c522001-03-20 23:09:54 +00001027 if (fdp) {
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001028#else
Guido van Rossum79f25d91997-04-29 20:08:16 +00001029 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001030 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001031 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001032 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansen4df3c522001-03-20 23:09:54 +00001033#endif /* !macintosh */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001034 fp = fopen(buf, fdp->mode);
Tim Peters50d8d372001-02-28 05:34:27 +00001035 if (fp != NULL) {
1036 if (case_ok(buf, len, namelen, name))
1037 break;
1038 else { /* continue search */
1039 fclose(fp);
1040 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001041 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001042 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001043 }
1044 if (fp != NULL)
1045 break;
1046 }
1047 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001048 PyErr_Format(PyExc_ImportError,
1049 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001050 return NULL;
1051 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001052 *p_fp = fp;
1053 return fdp;
1054}
1055
Tim Petersd1e87a82001-03-01 18:12:00 +00001056/* case_ok(char* buf, int len, int namelen, char* name)
1057 * The arguments here are tricky, best shown by example:
1058 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1059 * ^ ^ ^ ^
1060 * |--------------------- buf ---------------------|
1061 * |------------------- len ------------------|
1062 * |------ name -------|
1063 * |----- namelen -----|
1064 * buf is the full path, but len only counts up to (& exclusive of) the
1065 * extension. name is the module name, also exclusive of extension.
1066 *
1067 * We've already done a successful stat() or fopen() on buf, so know that
1068 * there's some match, possibly case-insensitive.
1069 *
Tim Peters50d8d372001-02-28 05:34:27 +00001070 * case_ok() is to return 1 if there's a case-sensitive match for
1071 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1072 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001073 *
Tim Peters50d8d372001-02-28 05:34:27 +00001074 * case_ok() is used to implement case-sensitive import semantics even
1075 * on platforms with case-insensitive filesystems. It's trivial to implement
1076 * for case-sensitive filesystems. It's pretty much a cross-platform
1077 * nightmare for systems with case-insensitive filesystems.
1078 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001079
Tim Peters50d8d372001-02-28 05:34:27 +00001080/* First we may need a pile of platform-specific header files; the sequence
1081 * of #if's here should match the sequence in the body of case_ok().
1082 */
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001083#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001084#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001085#ifdef __CYGWIN__
1086#include <sys/cygwin.h>
1087#endif
1088
Tim Peters50d8d372001-02-28 05:34:27 +00001089#elif defined(DJGPP)
1090#include <dir.h>
1091
1092#elif defined(macintosh)
1093#include <TextUtils.h>
1094#ifdef USE_GUSI1
1095#include "TFileSpec.h" /* for Path2FSSpec() */
1096#endif
1097
Tim Petersd1e87a82001-03-01 18:12:00 +00001098#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001099#include <sys/types.h>
1100#include <dirent.h>
1101
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001102#elif defined(RISCOS)
1103#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001104#endif
1105
Guido van Rossum0980bd91998-02-13 17:18:36 +00001106static int
Tim Peters50d8d372001-02-28 05:34:27 +00001107case_ok(char *buf, int len, int namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001108{
Tim Peters50d8d372001-02-28 05:34:27 +00001109/* Pick a platform-specific implementation; the sequence of #if's here should
1110 * match the sequence just above.
1111 */
1112
1113/* MS_WIN32 || __CYGWIN__ */
1114#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001115 WIN32_FIND_DATA data;
1116 HANDLE h;
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001117#ifdef __CYGWIN__
1118 char tempbuf[MAX_PATH];
1119#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001120
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001121 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001122 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001123
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001124#ifdef __CYGWIN__
1125 cygwin32_conv_to_win32_path(buf, tempbuf);
1126 h = FindFirstFile(tempbuf, &data);
1127#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001128 h = FindFirstFile(buf, &data);
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001129#endif
Guido van Rossum0980bd91998-02-13 17:18:36 +00001130 if (h == INVALID_HANDLE_VALUE) {
1131 PyErr_Format(PyExc_NameError,
1132 "Can't find file for module %.100s\n(filename %.300s)",
1133 name, buf);
1134 return 0;
1135 }
1136 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001137 return strncmp(data.cFileName, name, namelen) == 0;
1138
1139/* DJGPP */
1140#elif defined(DJGPP)
1141 struct ffblk ffblk;
1142 int done;
1143
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001144 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001145 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001146
1147 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1148 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001149 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001150 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001151 name, buf);
1152 return 0;
1153 }
Tim Peters50d8d372001-02-28 05:34:27 +00001154 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001155
Tim Peters50d8d372001-02-28 05:34:27 +00001156/* macintosh */
1157#elif defined(macintosh)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001158 FSSpec fss;
1159 OSErr err;
Tim Peters50d8d372001-02-28 05:34:27 +00001160
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001161 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Peters50d8d372001-02-28 05:34:27 +00001162 return 1;
1163
Guido van Rossumb33aa1a2000-04-24 15:08:18 +00001164#ifndef USE_GUSI1
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001165 err = FSMakeFSSpec(0, 0, Pstring(buf), &fss);
1166#else
1167 /* GUSI's Path2FSSpec() resolves all possible aliases nicely on
1168 the way, which is fine for all directories, but here we need
1169 the original name of the alias file (say, Dlg.ppc.slb, not
1170 toolboxmodules.ppc.slb). */
1171 char *colon;
1172 err = Path2FSSpec(buf, &fss);
1173 if (err == noErr) {
1174 colon = strrchr(buf, ':'); /* find filename */
1175 if (colon != NULL)
1176 err = FSMakeFSSpec(fss.vRefNum, fss.parID,
1177 Pstring(colon+1), &fss);
1178 else
1179 err = FSMakeFSSpec(fss.vRefNum, fss.parID,
1180 fss.name, &fss);
1181 }
1182#endif
Guido van Rossum0980bd91998-02-13 17:18:36 +00001183 if (err) {
1184 PyErr_Format(PyExc_NameError,
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001185 "Can't find file for module %.100s\n(filename %.300s)",
1186 name, buf);
Guido van Rossum0980bd91998-02-13 17:18:36 +00001187 return 0;
1188 }
Tim Peters50d8d372001-02-28 05:34:27 +00001189 return fss.name[0] >= namelen &&
1190 strncmp(name, (char *)fss.name+1, namelen) == 0;
1191
Tim Peters677898a2001-03-02 03:28:03 +00001192/* new-fangled macintosh (macosx) */
Tim Petersd1e87a82001-03-01 18:12:00 +00001193#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001194 DIR *dirp;
1195 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001196 char dirname[MAXPATHLEN + 1];
1197 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001198
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001199 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001200 return 1;
1201
Tim Petersd1e87a82001-03-01 18:12:00 +00001202 /* Copy the dir component into dirname; substitute "." if empty */
1203 if (dirlen <= 0) {
1204 dirname[0] = '.';
1205 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001206 }
1207 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001208 assert(dirlen <= MAXPATHLEN);
1209 memcpy(dirname, buf, dirlen);
1210 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001211 }
1212 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001213 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001214 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001215 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001216 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001217 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001218#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001219 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001220#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001221 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001222#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001223 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001224 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001225 (void)closedir(dirp);
1226 return 1; /* Found */
1227 }
1228 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001229 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001230 }
Tim Peters430f5d42001-03-01 01:30:56 +00001231 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001232
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001233/* RISC OS */
1234#elif defined(RISCOS)
1235 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1236 char buf2[MAXPATHLEN+2];
1237 char *nameWithExt = buf+len-namelen;
1238 int canonlen;
1239 os_error *e;
1240
1241 if (Py_GETENV("PYTHONCASEOK") != NULL)
1242 return 1;
1243
1244 /* workaround:
1245 append wildcard, otherwise case of filename wouldn't be touched */
1246 strcpy(buf2, buf);
1247 strcat(buf2, "*");
1248
1249 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1250 canonlen = MAXPATHLEN+1-canonlen;
1251 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1252 return 0;
1253 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1254 return 1; /* match */
1255
1256 return 0;
1257
Tim Peters50d8d372001-02-28 05:34:27 +00001258/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1259#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001260 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001261
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001262#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001263}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001264
Guido van Rossum0980bd91998-02-13 17:18:36 +00001265
Guido van Rossum197346f1997-10-31 18:38:52 +00001266#ifdef HAVE_STAT
1267/* Helper to look for __init__.py or __init__.py[co] in potential package */
1268static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001269find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001270{
Tim Peters0f9431f2001-07-05 03:47:53 +00001271 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001272 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001273 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001274 struct stat statbuf;
1275
Tim Peters0f9431f2001-07-05 03:47:53 +00001276/* For calling case_ok(buf, len, namelen, name):
1277 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1278 * ^ ^ ^ ^
1279 * |--------------------- buf ---------------------|
1280 * |------------------- len ------------------|
1281 * |------ name -------|
1282 * |----- namelen -----|
1283 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001284 if (save_len + 13 >= MAXPATHLEN)
1285 return 0;
1286 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001287 pname = buf + i;
1288 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001289 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001290 if (case_ok(buf,
1291 save_len + 9, /* len("/__init__") */
1292 8, /* len("__init__") */
1293 pname)) {
1294 buf[save_len] = '\0';
1295 return 1;
1296 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001297 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001298 i += strlen(pname);
1299 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001300 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001301 if (case_ok(buf,
1302 save_len + 9, /* len("/__init__") */
1303 8, /* len("__init__") */
1304 pname)) {
1305 buf[save_len] = '\0';
1306 return 1;
1307 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001308 }
1309 buf[save_len] = '\0';
1310 return 0;
1311}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001312
1313#else
1314
1315#ifdef RISCOS
1316static int
1317find_init_module(buf)
1318 char *buf;
1319{
1320 int save_len = strlen(buf);
1321 int i = save_len;
1322
1323 if (save_len + 13 >= MAXPATHLEN)
1324 return 0;
1325 buf[i++] = SEP;
1326 strcpy(buf+i, "__init__/py");
1327 if (isfile(buf)) {
1328 buf[save_len] = '\0';
1329 return 1;
1330 }
1331
1332 if (Py_OptimizeFlag)
1333 strcpy(buf+i, "o");
1334 else
1335 strcpy(buf+i, "c");
1336 if (isfile(buf)) {
1337 buf[save_len] = '\0';
1338 return 1;
1339 }
1340 buf[save_len] = '\0';
1341 return 0;
1342}
1343#endif /*RISCOS*/
1344
Guido van Rossum197346f1997-10-31 18:38:52 +00001345#endif /* HAVE_STAT */
1346
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001347
Tim Petersdbd9ba62000-07-09 03:09:57 +00001348static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001349
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001350/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001351 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001352
Guido van Rossum79f25d91997-04-29 20:08:16 +00001353static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001354load_module(char *name, FILE *fp, char *buf, int type)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001355{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001356 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001357 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001358 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001359
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001360 /* First check that there's an open file (if we need one) */
1361 switch (type) {
1362 case PY_SOURCE:
1363 case PY_COMPILED:
1364 if (fp == NULL) {
1365 PyErr_Format(PyExc_ValueError,
1366 "file object required for import (type code %d)",
1367 type);
1368 return NULL;
1369 }
1370 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001371
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001372 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001373
1374 case PY_SOURCE:
1375 m = load_source_module(name, buf, fp);
1376 break;
1377
1378 case PY_COMPILED:
1379 m = load_compiled_module(name, buf, fp);
1380 break;
1381
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001382#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001383 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001384 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001385 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001386#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001387
Jack Jansen9c96a921995-02-15 22:57:06 +00001388#ifdef macintosh
1389 case PY_RESOURCE:
1390 m = PyMac_LoadResourceModule(name, buf);
1391 break;
Guido van Rossum0f84a341998-08-06 13:36:01 +00001392 case PY_CODERESOURCE:
1393 m = PyMac_LoadCodeResourceModule(name, buf);
1394 break;
Jack Jansen9c96a921995-02-15 22:57:06 +00001395#endif
1396
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001397 case PKG_DIRECTORY:
1398 m = load_package(name, buf);
1399 break;
1400
1401 case C_BUILTIN:
1402 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001403 if (buf != NULL && buf[0] != '\0')
1404 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001405 if (type == C_BUILTIN)
1406 err = init_builtin(name);
1407 else
1408 err = PyImport_ImportFrozenModule(name);
1409 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001410 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001411 if (err == 0) {
1412 PyErr_Format(PyExc_ImportError,
1413 "Purported %s module %.200s not found",
1414 type == C_BUILTIN ?
1415 "builtin" : "frozen",
1416 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001417 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001418 }
1419 modules = PyImport_GetModuleDict();
1420 m = PyDict_GetItemString(modules, name);
1421 if (m == NULL) {
1422 PyErr_Format(
1423 PyExc_ImportError,
1424 "%s module %.200s not properly initialized",
1425 type == C_BUILTIN ?
1426 "builtin" : "frozen",
1427 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001428 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001429 }
1430 Py_INCREF(m);
1431 break;
1432
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001433 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001434 PyErr_Format(PyExc_ImportError,
1435 "Don't know how to import %.200s (type code %d)",
1436 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001437 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001438
1439 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001440
1441 return m;
1442}
1443
1444
1445/* Initialize a built-in module.
1446 Return 1 for succes, 0 if the module is not found, and -1 with
1447 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001448
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001449static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001450init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001451{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001452 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001453
Greg Ward201baee2001-10-04 14:52:06 +00001454 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001455 return 1;
1456
Guido van Rossum771c6c81997-10-31 18:37:24 +00001457 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001458 if (strcmp(name, p->name) == 0) {
1459 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001460 PyErr_Format(PyExc_ImportError,
1461 "Cannot re-init internal module %.200s",
1462 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001463 return -1;
1464 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001465 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001466 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001467 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001468 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001469 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001470 if (_PyImport_FixupExtension(name, name) == NULL)
1471 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001472 return 1;
1473 }
1474 }
1475 return 0;
1476}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001477
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001478
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001479/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001480
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001481static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001482find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001483{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001484 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001485
Guido van Rossum79f25d91997-04-29 20:08:16 +00001486 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001487 if (p->name == NULL)
1488 return NULL;
1489 if (strcmp(p->name, name) == 0)
1490 break;
1491 }
1492 return p;
1493}
1494
Guido van Rossum79f25d91997-04-29 20:08:16 +00001495static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001496get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001497{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001498 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001499 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001500
1501 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001502 PyErr_Format(PyExc_ImportError,
1503 "No such frozen object named %.200s",
1504 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001505 return NULL;
1506 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001507 if (p->code == NULL) {
1508 PyErr_Format(PyExc_ImportError,
1509 "Excluded frozen object named %.200s",
1510 name);
1511 return NULL;
1512 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001513 size = p->size;
1514 if (size < 0)
1515 size = -size;
1516 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001517}
1518
1519/* Initialize a frozen module.
1520 Return 1 for succes, 0 if the module is not found, and -1 with
1521 an exception set if the initialization failed.
1522 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001523
1524int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001525PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001526{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001527 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001528 PyObject *co;
1529 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001530 int ispackage;
1531 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001532
1533 if (p == NULL)
1534 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001535 if (p->code == NULL) {
1536 PyErr_Format(PyExc_ImportError,
1537 "Excluded frozen object named %.200s",
1538 name);
1539 return -1;
1540 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001541 size = p->size;
1542 ispackage = (size < 0);
1543 if (ispackage)
1544 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001545 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001546 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001547 name, ispackage ? " package" : "");
1548 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001549 if (co == NULL)
1550 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001551 if (!PyCode_Check(co)) {
1552 Py_DECREF(co);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001553 PyErr_Format(PyExc_TypeError,
1554 "frozen object %.200s is not a code object",
1555 name);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001556 return -1;
1557 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001558 if (ispackage) {
1559 /* Set __path__ to the package name */
1560 PyObject *d, *s;
1561 int err;
1562 m = PyImport_AddModule(name);
1563 if (m == NULL)
1564 return -1;
1565 d = PyModule_GetDict(m);
1566 s = PyString_InternFromString(name);
1567 if (s == NULL)
1568 return -1;
1569 err = PyDict_SetItemString(d, "__path__", s);
1570 Py_DECREF(s);
1571 if (err != 0)
1572 return err;
1573 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001574 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum79f25d91997-04-29 20:08:16 +00001575 Py_DECREF(co);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001576 if (m == NULL)
1577 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001578 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001579 return 1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001580}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001581
1582
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001583/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001584 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001585
Guido van Rossum79f25d91997-04-29 20:08:16 +00001586PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001587PyImport_ImportModule(char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001588{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001589 PyObject *pname;
1590 PyObject *result;
1591
1592 pname = PyString_FromString(name);
1593 result = PyImport_Import(pname);
1594 Py_DECREF(pname);
1595 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001596}
1597
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001598/* Forward declarations for helper routines */
Tim Petersdbd9ba62000-07-09 03:09:57 +00001599static PyObject *get_parent(PyObject *globals, char *buf, int *p_buflen);
1600static PyObject *load_next(PyObject *mod, PyObject *altmod,
1601 char **p_name, char *buf, int *p_buflen);
1602static int mark_miss(char *name);
1603static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
1604 char *buf, int buflen, int recursive);
1605static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001606
1607/* The Magnum Opus of dotted-name import :-) */
1608
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001609static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001610import_module_ex(char *name, PyObject *globals, PyObject *locals,
1611 PyObject *fromlist)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001612{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001613 char buf[MAXPATHLEN+1];
1614 int buflen = 0;
1615 PyObject *parent, *head, *next, *tail;
1616
1617 parent = get_parent(globals, buf, &buflen);
1618 if (parent == NULL)
1619 return NULL;
1620
1621 head = load_next(parent, Py_None, &name, buf, &buflen);
1622 if (head == NULL)
1623 return NULL;
1624
1625 tail = head;
1626 Py_INCREF(tail);
1627 while (name) {
1628 next = load_next(tail, tail, &name, buf, &buflen);
1629 Py_DECREF(tail);
1630 if (next == NULL) {
1631 Py_DECREF(head);
1632 return NULL;
1633 }
1634 tail = next;
1635 }
1636
1637 if (fromlist != NULL) {
1638 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
1639 fromlist = NULL;
1640 }
1641
1642 if (fromlist == NULL) {
1643 Py_DECREF(tail);
1644 return head;
1645 }
1646
1647 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001648 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001649 Py_DECREF(tail);
1650 return NULL;
1651 }
1652
1653 return tail;
1654}
1655
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001656PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001657PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
1658 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001659{
1660 PyObject *result;
1661 lock_import();
Guido van Rossumd65911b1998-03-03 22:33:27 +00001662 result = import_module_ex(name, globals, locals, fromlist);
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001663 unlock_import();
1664 return result;
1665}
1666
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001667static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001668get_parent(PyObject *globals, char *buf, int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001669{
1670 static PyObject *namestr = NULL;
1671 static PyObject *pathstr = NULL;
1672 PyObject *modname, *modpath, *modules, *parent;
1673
1674 if (globals == NULL || !PyDict_Check(globals))
1675 return Py_None;
1676
1677 if (namestr == NULL) {
1678 namestr = PyString_InternFromString("__name__");
1679 if (namestr == NULL)
1680 return NULL;
1681 }
1682 if (pathstr == NULL) {
1683 pathstr = PyString_InternFromString("__path__");
1684 if (pathstr == NULL)
1685 return NULL;
1686 }
1687
1688 *buf = '\0';
1689 *p_buflen = 0;
1690 modname = PyDict_GetItem(globals, namestr);
1691 if (modname == NULL || !PyString_Check(modname))
1692 return Py_None;
1693
1694 modpath = PyDict_GetItem(globals, pathstr);
1695 if (modpath != NULL) {
1696 int len = PyString_GET_SIZE(modname);
1697 if (len > MAXPATHLEN) {
1698 PyErr_SetString(PyExc_ValueError,
1699 "Module name too long");
1700 return NULL;
1701 }
1702 strcpy(buf, PyString_AS_STRING(modname));
1703 *p_buflen = len;
1704 }
1705 else {
1706 char *start = PyString_AS_STRING(modname);
1707 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001708 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001709 if (lastdot == NULL)
1710 return Py_None;
1711 len = lastdot - start;
1712 if (len >= MAXPATHLEN) {
1713 PyErr_SetString(PyExc_ValueError,
1714 "Module name too long");
1715 return NULL;
1716 }
1717 strncpy(buf, start, len);
1718 buf[len] = '\0';
1719 *p_buflen = len;
1720 }
1721
1722 modules = PyImport_GetModuleDict();
1723 parent = PyDict_GetItemString(modules, buf);
1724 if (parent == NULL)
1725 parent = Py_None;
1726 return parent;
1727 /* We expect, but can't guarantee, if parent != None, that:
1728 - parent.__name__ == buf
1729 - parent.__dict__ is globals
1730 If this is violated... Who cares? */
1731}
1732
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001733/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001734static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001735load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
1736 int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001737{
1738 char *name = *p_name;
1739 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001740 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001741 char *p;
1742 PyObject *result;
1743
1744 if (dot == NULL) {
1745 *p_name = NULL;
1746 len = strlen(name);
1747 }
1748 else {
1749 *p_name = dot+1;
1750 len = dot-name;
1751 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00001752 if (len == 0) {
1753 PyErr_SetString(PyExc_ValueError,
1754 "Empty module name");
1755 return NULL;
1756 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001757
1758 p = buf + *p_buflen;
1759 if (p != buf)
1760 *p++ = '.';
1761 if (p+len-buf >= MAXPATHLEN) {
1762 PyErr_SetString(PyExc_ValueError,
1763 "Module name too long");
1764 return NULL;
1765 }
1766 strncpy(p, name, len);
1767 p[len] = '\0';
1768 *p_buflen = p+len-buf;
1769
1770 result = import_submodule(mod, p, buf);
1771 if (result == Py_None && altmod != mod) {
1772 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001773 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00001774 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001775 if (result != NULL && result != Py_None) {
1776 if (mark_miss(buf) != 0) {
1777 Py_DECREF(result);
1778 return NULL;
1779 }
1780 strncpy(buf, name, len);
1781 buf[len] = '\0';
1782 *p_buflen = len;
1783 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001784 }
1785 if (result == NULL)
1786 return NULL;
1787
1788 if (result == Py_None) {
1789 Py_DECREF(result);
1790 PyErr_Format(PyExc_ImportError,
1791 "No module named %.200s", name);
1792 return NULL;
1793 }
1794
1795 return result;
1796}
1797
1798static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001799mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001800{
1801 PyObject *modules = PyImport_GetModuleDict();
1802 return PyDict_SetItemString(modules, name, Py_None);
1803}
1804
1805static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001806ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, int buflen,
1807 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001808{
1809 int i;
1810
1811 if (!PyObject_HasAttrString(mod, "__path__"))
1812 return 1;
1813
1814 for (i = 0; ; i++) {
1815 PyObject *item = PySequence_GetItem(fromlist, i);
1816 int hasit;
1817 if (item == NULL) {
1818 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
1819 PyErr_Clear();
1820 return 1;
1821 }
1822 return 0;
1823 }
1824 if (!PyString_Check(item)) {
1825 PyErr_SetString(PyExc_TypeError,
1826 "Item in ``from list'' not a string");
1827 Py_DECREF(item);
1828 return 0;
1829 }
1830 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00001831 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001832 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001833 /* See if the package defines __all__ */
1834 if (recursive)
1835 continue; /* Avoid endless recursion */
1836 all = PyObject_GetAttrString(mod, "__all__");
1837 if (all == NULL)
1838 PyErr_Clear();
1839 else {
1840 if (!ensure_fromlist(mod, all, buf, buflen, 1))
1841 return 0;
1842 Py_DECREF(all);
1843 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001844 continue;
1845 }
1846 hasit = PyObject_HasAttr(mod, item);
1847 if (!hasit) {
1848 char *subname = PyString_AS_STRING(item);
1849 PyObject *submod;
1850 char *p;
1851 if (buflen + strlen(subname) >= MAXPATHLEN) {
1852 PyErr_SetString(PyExc_ValueError,
1853 "Module name too long");
1854 Py_DECREF(item);
1855 return 0;
1856 }
1857 p = buf + buflen;
1858 *p++ = '.';
1859 strcpy(p, subname);
1860 submod = import_submodule(mod, subname, buf);
1861 Py_XDECREF(submod);
1862 if (submod == NULL) {
1863 Py_DECREF(item);
1864 return 0;
1865 }
1866 }
1867 Py_DECREF(item);
1868 }
1869
Guido van Rossuma7f2e811997-10-03 15:33:32 +00001870 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001871}
1872
1873static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001874import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001875{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001876 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001877 PyObject *m, *res = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00001878
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001879 /* Require:
1880 if mod == None: subname == fullname
1881 else: mod.__name__ + "." + subname == fullname
1882 */
1883
Tim Peters50d8d372001-02-28 05:34:27 +00001884 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001885 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001886 }
1887 else {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001888 PyObject *path;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001889 char buf[MAXPATHLEN+1];
1890 struct filedescr *fdp;
1891 FILE *fp = NULL;
1892
Guido van Rossum9c0afe51998-05-19 15:09:05 +00001893 if (mod == Py_None)
1894 path = NULL;
1895 else {
1896 path = PyObject_GetAttrString(mod, "__path__");
1897 if (path == NULL) {
1898 PyErr_Clear();
1899 Py_INCREF(Py_None);
1900 return Py_None;
1901 }
1902 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001903
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001904 buf[0] = '\0';
Guido van Rossumb68cd421998-07-01 17:36:26 +00001905 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
1906 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001907 if (fdp == NULL) {
1908 if (!PyErr_ExceptionMatches(PyExc_ImportError))
1909 return NULL;
1910 PyErr_Clear();
1911 Py_INCREF(Py_None);
1912 return Py_None;
1913 }
1914 m = load_module(fullname, fp, buf, fdp->type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001915 if (fp)
1916 fclose(fp);
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001917 if (mod != Py_None) {
1918 /* Irrespective of the success of this load, make a
1919 reference to it in the parent package module.
1920 A copy gets saved in the modules dictionary
1921 under the full name, so get a reference from
1922 there, if need be. (The exception is when
1923 the load failed with a SyntaxError -- then
1924 there's no trace in sys.modules. In that case,
1925 of course, do nothing extra.) */
1926 res = m;
1927 if (res == NULL)
1928 res = PyDict_GetItemString(modules, fullname);
1929 if (res != NULL &&
1930 PyObject_SetAttrString(mod, subname, res) < 0) {
1931 Py_XDECREF(m);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001932 m = NULL;
1933 }
1934 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00001935 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001936
1937 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00001938}
1939
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001940
1941/* Re-import a module of any kind and return its module object, WITH
1942 INCREMENTED REFERENCE COUNT */
1943
Guido van Rossum79f25d91997-04-29 20:08:16 +00001944PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001945PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001946{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001947 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum222ef561997-09-06 19:41:09 +00001948 PyObject *path = NULL;
1949 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001950 char buf[MAXPATHLEN+1];
1951 struct filedescr *fdp;
1952 FILE *fp = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001953
Guido van Rossum79f25d91997-04-29 20:08:16 +00001954 if (m == NULL || !PyModule_Check(m)) {
1955 PyErr_SetString(PyExc_TypeError,
1956 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001957 return NULL;
1958 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001959 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001960 if (name == NULL)
1961 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001962 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001963 PyErr_Format(PyExc_ImportError,
1964 "reload(): module %.200s not in sys.modules",
1965 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001966 return NULL;
1967 }
Guido van Rossum222ef561997-09-06 19:41:09 +00001968 subname = strrchr(name, '.');
1969 if (subname == NULL)
1970 subname = name;
1971 else {
1972 PyObject *parentname, *parent;
1973 parentname = PyString_FromStringAndSize(name, (subname-name));
1974 if (parentname == NULL)
1975 return NULL;
1976 parent = PyDict_GetItem(modules, parentname);
Barry Warsaw38793331999-01-27 17:54:20 +00001977 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00001978 if (parent == NULL) {
1979 PyErr_Format(PyExc_ImportError,
1980 "reload(): parent %.200s not in sys.modules",
1981 name);
1982 return NULL;
1983 }
1984 subname++;
1985 path = PyObject_GetAttrString(parent, "__path__");
1986 if (path == NULL)
1987 PyErr_Clear();
1988 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001989 buf[0] = '\0';
Guido van Rossum222ef561997-09-06 19:41:09 +00001990 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
1991 Py_XDECREF(path);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001992 if (fdp == NULL)
1993 return NULL;
1994 m = load_module(name, fp, buf, fdp->type);
1995 if (fp)
1996 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001997 return m;
1998}
1999
2000
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002001/* Higher-level import emulator which emulates the "import" statement
2002 more accurately -- it invokes the __import__() function from the
2003 builtins of the current globals. This means that the import is
2004 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002005 environment, e.g. by "rexec".
2006 A dummy list ["__doc__"] is passed as the 4th argument so that
2007 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2008 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002009
2010PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002011PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002012{
2013 static PyObject *silly_list = NULL;
2014 static PyObject *builtins_str = NULL;
2015 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002016 PyObject *globals = NULL;
2017 PyObject *import = NULL;
2018 PyObject *builtins = NULL;
2019 PyObject *r = NULL;
2020
2021 /* Initialize constant string objects */
2022 if (silly_list == NULL) {
2023 import_str = PyString_InternFromString("__import__");
2024 if (import_str == NULL)
2025 return NULL;
2026 builtins_str = PyString_InternFromString("__builtins__");
2027 if (builtins_str == NULL)
2028 return NULL;
2029 silly_list = Py_BuildValue("[s]", "__doc__");
2030 if (silly_list == NULL)
2031 return NULL;
2032 }
2033
2034 /* Get the builtins from current globals */
2035 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002036 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002037 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002038 builtins = PyObject_GetItem(globals, builtins_str);
2039 if (builtins == NULL)
2040 goto err;
2041 }
2042 else {
2043 /* No globals -- use standard builtins, and fake globals */
2044 PyErr_Clear();
2045
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002046 builtins = PyImport_ImportModuleEx("__builtin__",
2047 NULL, NULL, NULL);
2048 if (builtins == NULL)
2049 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002050 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2051 if (globals == NULL)
2052 goto err;
2053 }
2054
2055 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002056 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002057 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002058 if (import == NULL)
2059 PyErr_SetObject(PyExc_KeyError, import_str);
2060 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002061 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002062 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002063 if (import == NULL)
2064 goto err;
2065
2066 /* Call the _import__ function with the proper argument list */
2067 r = PyObject_CallFunction(import, "OOOO",
2068 module_name, globals, globals, silly_list);
2069
2070 err:
2071 Py_XDECREF(globals);
2072 Py_XDECREF(builtins);
2073 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002074
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002075 return r;
2076}
2077
2078
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002079/* Module 'imp' provides Python access to the primitives used for
2080 importing modules.
2081*/
2082
Guido van Rossum79f25d91997-04-29 20:08:16 +00002083static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002084imp_get_magic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002085{
2086 char buf[4];
2087
Guido van Rossum43713e52000-02-29 13:59:29 +00002088 if (!PyArg_ParseTuple(args, ":get_magic"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002089 return NULL;
Guido van Rossum96774c12000-05-01 20:19:08 +00002090 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2091 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2092 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2093 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002094
Guido van Rossum79f25d91997-04-29 20:08:16 +00002095 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002096}
2097
Guido van Rossum79f25d91997-04-29 20:08:16 +00002098static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002099imp_get_suffixes(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002100{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002101 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002102 struct filedescr *fdp;
2103
Guido van Rossum43713e52000-02-29 13:59:29 +00002104 if (!PyArg_ParseTuple(args, ":get_suffixes"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002105 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002106 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002107 if (list == NULL)
2108 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002109 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2110 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002111 fdp->suffix, fdp->mode, fdp->type);
2112 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002113 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002114 return NULL;
2115 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002116 if (PyList_Append(list, item) < 0) {
2117 Py_DECREF(list);
2118 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002119 return NULL;
2120 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002121 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002122 }
2123 return list;
2124}
2125
Guido van Rossum79f25d91997-04-29 20:08:16 +00002126static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002127call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002128{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002129 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002130 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002131 struct filedescr *fdp;
2132 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002133 FILE *fp = NULL;
2134
2135 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002136 if (path == Py_None)
2137 path = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002138 fdp = find_module(name, path, pathname, MAXPATHLEN+1, &fp);
2139 if (fdp == NULL)
2140 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002141 if (fp != NULL) {
2142 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2143 if (fob == NULL) {
2144 fclose(fp);
2145 return NULL;
2146 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002147 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002148 else {
2149 fob = Py_None;
2150 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002151 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002152 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002153 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002154 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002155 return ret;
2156}
2157
Guido van Rossum79f25d91997-04-29 20:08:16 +00002158static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002159imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002160{
2161 char *name;
2162 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002163 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002164 return NULL;
2165 return call_find_module(name, path);
2166}
2167
2168static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002169imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002170{
2171 char *name;
2172 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002173 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002174 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002175 return NULL;
2176 ret = init_builtin(name);
2177 if (ret < 0)
2178 return NULL;
2179 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002180 Py_INCREF(Py_None);
2181 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002182 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002183 m = PyImport_AddModule(name);
2184 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002185 return m;
2186}
2187
Guido van Rossum79f25d91997-04-29 20:08:16 +00002188static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002189imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002190{
2191 char *name;
2192 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002193 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002194 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002195 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002196 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002197 if (ret < 0)
2198 return NULL;
2199 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002200 Py_INCREF(Py_None);
2201 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002202 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002203 m = PyImport_AddModule(name);
2204 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002205 return m;
2206}
2207
Guido van Rossum79f25d91997-04-29 20:08:16 +00002208static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002209imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002210{
2211 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002212
Guido van Rossum43713e52000-02-29 13:59:29 +00002213 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002214 return NULL;
2215 return get_frozen_object(name);
2216}
2217
Guido van Rossum79f25d91997-04-29 20:08:16 +00002218static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002219imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002220{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002221 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002222 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002223 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002224 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002225}
2226
Guido van Rossum79f25d91997-04-29 20:08:16 +00002227static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002228imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002229{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002230 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002231 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002232 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002233 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002234 p = find_frozen(name);
2235 return PyInt_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002236}
2237
2238static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002239get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002240{
2241 FILE *fp;
2242 if (fob == NULL) {
2243 fp = fopen(pathname, mode);
2244 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002245 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002246 }
2247 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002248 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002249 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002250 PyErr_SetString(PyExc_ValueError,
2251 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002252 }
2253 return fp;
2254}
2255
Guido van Rossum79f25d91997-04-29 20:08:16 +00002256static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002257imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002258{
2259 char *name;
2260 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002261 PyObject *fob = NULL;
2262 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002263 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002264 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002265 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002266 return NULL;
2267 fp = get_file(pathname, fob, "rb");
2268 if (fp == NULL)
2269 return NULL;
2270 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002271 if (fob == NULL)
2272 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002273 return m;
2274}
2275
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002276#ifdef HAVE_DYNAMIC_LOADING
2277
Guido van Rossum79f25d91997-04-29 20:08:16 +00002278static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002279imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002280{
2281 char *name;
2282 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002283 PyObject *fob = NULL;
2284 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002285 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002286 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002287 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002288 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002289 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002290 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002291 if (fp == NULL)
2292 return NULL;
2293 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002294 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002295 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002296}
2297
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002298#endif /* HAVE_DYNAMIC_LOADING */
2299
Guido van Rossum79f25d91997-04-29 20:08:16 +00002300static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002301imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002302{
2303 char *name;
2304 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002305 PyObject *fob = NULL;
2306 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002307 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002308 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002309 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002310 return NULL;
2311 fp = get_file(pathname, fob, "r");
2312 if (fp == NULL)
2313 return NULL;
2314 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002315 if (fob == NULL)
2316 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002317 return m;
2318}
2319
Jack Jansen9c96a921995-02-15 22:57:06 +00002320#ifdef macintosh
Guido van Rossum79f25d91997-04-29 20:08:16 +00002321static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002322imp_load_resource(PyObject *self, PyObject *args)
Jack Jansen9c96a921995-02-15 22:57:06 +00002323{
2324 char *name;
2325 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002326 PyObject *m;
Jack Jansen9c96a921995-02-15 22:57:06 +00002327
Guido van Rossum43713e52000-02-29 13:59:29 +00002328 if (!PyArg_ParseTuple(args, "ss:load_resource", &name, &pathname))
Jack Jansen9c96a921995-02-15 22:57:06 +00002329 return NULL;
2330 m = PyMac_LoadResourceModule(name, pathname);
2331 return m;
2332}
2333#endif /* macintosh */
2334
Guido van Rossum79f25d91997-04-29 20:08:16 +00002335static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002336imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002337{
2338 char *name;
2339 PyObject *fob;
2340 char *pathname;
2341 char *suffix; /* Unused */
2342 char *mode;
2343 int type;
2344 FILE *fp;
2345
Guido van Rossum43713e52000-02-29 13:59:29 +00002346 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002347 &name, &fob, &pathname,
2348 &suffix, &mode, &type))
2349 return NULL;
2350 if (*mode && (*mode != 'r' || strchr(mode, '+') != NULL)) {
2351 PyErr_Format(PyExc_ValueError,
2352 "invalid file open mode %.200s", mode);
2353 return NULL;
2354 }
2355 if (fob == Py_None)
2356 fp = NULL;
2357 else {
2358 if (!PyFile_Check(fob)) {
2359 PyErr_SetString(PyExc_ValueError,
2360 "load_module arg#2 should be a file or None");
2361 return NULL;
2362 }
2363 fp = get_file(pathname, fob, mode);
2364 if (fp == NULL)
2365 return NULL;
2366 }
2367 return load_module(name, fp, pathname, type);
2368}
2369
2370static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002371imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002372{
2373 char *name;
2374 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002375 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002376 return NULL;
2377 return load_package(name, pathname);
2378}
2379
2380static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002381imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002382{
2383 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002384 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002385 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002386 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002387}
2388
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002389/* Doc strings */
2390
2391static char doc_imp[] = "\
2392This module provides the components needed to build your own\n\
2393__import__ function. Undocumented functions are obsolete.\n\
2394";
2395
2396static char doc_find_module[] = "\
2397find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
2398Search for a module. If path is omitted or None, search for a\n\
2399built-in, frozen or special module and continue search in sys.path.\n\
2400The module name cannot contain '.'; to search for a submodule of a\n\
2401package, pass the submodule name and the package's __path__.\
2402";
2403
2404static char doc_load_module[] = "\
2405load_module(name, file, filename, (suffix, mode, type)) -> module\n\
2406Load a module, given information returned by find_module().\n\
2407The module name must include the full package name, if any.\
2408";
2409
2410static char doc_get_magic[] = "\
2411get_magic() -> string\n\
2412Return the magic number for .pyc or .pyo files.\
2413";
2414
2415static char doc_get_suffixes[] = "\
2416get_suffixes() -> [(suffix, mode, type), ...]\n\
2417Return a list of (suffix, mode, type) tuples describing the files\n\
2418that find_module() looks for.\
2419";
2420
2421static char doc_new_module[] = "\
2422new_module(name) -> module\n\
2423Create a new module. Do not enter it in sys.modules.\n\
2424The module name must include the full package name, if any.\
2425";
2426
Tim Peters69232342001-08-30 05:16:13 +00002427static char doc_lock_held[] = "\
2428lock_held() -> 0 or 1\n\
2429Return 1 if the import lock is currently held.\n\
2430On platforms without threads, return 0.\
2431";
2432
Guido van Rossum79f25d91997-04-29 20:08:16 +00002433static PyMethodDef imp_methods[] = {
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002434 {"find_module", imp_find_module, 1, doc_find_module},
2435 {"get_magic", imp_get_magic, 1, doc_get_magic},
2436 {"get_suffixes", imp_get_suffixes, 1, doc_get_suffixes},
2437 {"load_module", imp_load_module, 1, doc_load_module},
2438 {"new_module", imp_new_module, 1, doc_new_module},
Tim Peters69232342001-08-30 05:16:13 +00002439 {"lock_held", imp_lock_held, 1, doc_lock_held},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002440 /* The rest are obsolete */
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002441 {"get_frozen_object", imp_get_frozen_object, 1},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002442 {"init_builtin", imp_init_builtin, 1},
2443 {"init_frozen", imp_init_frozen, 1},
2444 {"is_builtin", imp_is_builtin, 1},
2445 {"is_frozen", imp_is_frozen, 1},
2446 {"load_compiled", imp_load_compiled, 1},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002447#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002448 {"load_dynamic", imp_load_dynamic, 1},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002449#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002450 {"load_package", imp_load_package, 1},
Jack Jansen9c96a921995-02-15 22:57:06 +00002451#ifdef macintosh
2452 {"load_resource", imp_load_resource, 1},
2453#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002454 {"load_source", imp_load_source, 1},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002455 {NULL, NULL} /* sentinel */
2456};
2457
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002458static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002459setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002460{
2461 PyObject *v;
2462 int err;
2463
2464 v = PyInt_FromLong((long)value);
2465 err = PyDict_SetItemString(d, name, v);
2466 Py_XDECREF(v);
2467 return err;
2468}
2469
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002470void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002471initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002472{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002473 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002474
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002475 m = Py_InitModule4("imp", imp_methods, doc_imp,
2476 NULL, PYTHON_API_VERSION);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002477 d = PyModule_GetDict(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002478
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002479 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
2480 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
2481 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
2482 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
2483 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
2484 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
2485 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
2486 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00002487 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002488
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002489 failure:
2490 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002491}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002492
2493
Guido van Rossumb18618d2000-05-03 23:44:39 +00002494/* API for embedding applications that want to add their own entries
2495 to the table of built-in modules. This should normally be called
2496 *before* Py_Initialize(). When the table resize fails, -1 is
2497 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002498
2499 After a similar function by Just van Rossum. */
2500
2501int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002502PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002503{
2504 static struct _inittab *our_copy = NULL;
2505 struct _inittab *p;
2506 int i, n;
2507
2508 /* Count the number of entries in both tables */
2509 for (n = 0; newtab[n].name != NULL; n++)
2510 ;
2511 if (n == 0)
2512 return 0; /* Nothing to do */
2513 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2514 ;
2515
2516 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00002517 p = our_copy;
2518 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002519 if (p == NULL)
2520 return -1;
2521
2522 /* Copy the tables into the new memory */
2523 if (our_copy != PyImport_Inittab)
2524 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2525 PyImport_Inittab = our_copy = p;
2526 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
2527
2528 return 0;
2529}
2530
2531/* Shorthand to add a single entry given a name and a function */
2532
2533int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002534PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002535{
2536 struct _inittab newtab[2];
2537
2538 memset(newtab, '\0', sizeof newtab);
2539
2540 newtab[0].name = name;
2541 newtab[0].initfunc = initfunc;
2542
2543 return PyImport_ExtendInittab(newtab);
2544}