blob: e106aad6806ff9ac48097b53a41ec80caebf22af [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];
Andrew MacIntyred9400542002-02-26 11:41:34 +0000903#if defined(PYOS_OS2)
904 size_t saved_len;
905 size_t saved_namelen;
906 char *saved_buf = NULL;
907#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000908
Fred Drake4c82b232000-06-30 16:18:57 +0000909 if (strlen(realname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000910 PyErr_SetString(PyExc_OverflowError,
911 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +0000912 return NULL;
913 }
Guido van Rossum0506a431998-08-11 15:07:39 +0000914 strcpy(name, realname);
915
916 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000917 /* The only type of submodule allowed inside a "frozen"
918 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +0000919 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
920 PyErr_SetString(PyExc_ImportError,
921 "full frozen module name too long");
922 return NULL;
923 }
924 strcpy(buf, PyString_AsString(path));
925 strcat(buf, ".");
926 strcat(buf, name);
927 strcpy(name, buf);
Jack Jansen550fdae2001-10-30 13:08:39 +0000928#ifdef macintosh
929 /* Freezing on the mac works different, and the modules are
930 ** actually on sys.path. So we don't take the quick exit but
931 ** continue with the normal flow.
932 */
933 path = NULL;
934#else
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000935 if (find_frozen(name) != NULL) {
936 strcpy(buf, name);
937 return &fd_frozen;
938 }
939 PyErr_Format(PyExc_ImportError,
940 "No frozen submodule named %.200s", name);
941 return NULL;
Jack Jansen550fdae2001-10-30 13:08:39 +0000942#endif
Guido van Rossum0506a431998-08-11 15:07:39 +0000943 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000944 if (path == NULL) {
945 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000946 strcpy(buf, name);
947 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000948 }
Greg Ward201baee2001-10-04 14:52:06 +0000949 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000950 strcpy(buf, name);
951 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000952 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000953
Guido van Rossumac279101996-08-22 23:10:58 +0000954#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000955 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
956 if (fp != NULL) {
957 *p_fp = fp;
958 return fdp;
959 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +0000960#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +0000961 path = PySys_GetObject("path");
962 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000963 if (path == NULL || !PyList_Check(path)) {
964 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +0000965 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000966 return NULL;
967 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000968 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000969 namelen = strlen(name);
970 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000971 PyObject *v = PyList_GetItem(path, i);
972 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000973 continue;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000974 len = PyString_Size(v);
Guido van Rossumef3d02e1997-07-21 14:54:36 +0000975 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000976 continue; /* Too long */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000977 strcpy(buf, PyString_AsString(v));
Fred Drake4c82b232000-06-30 16:18:57 +0000978 if (strlen(buf) != len)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000979 continue; /* v contains '\0' */
Jack Jansen9c96a921995-02-15 22:57:06 +0000980#ifdef macintosh
Guido van Rossum741689d1997-08-12 14:53:39 +0000981#ifdef INTERN_STRINGS
Tim Peters50d8d372001-02-28 05:34:27 +0000982 /*
Guido van Rossum741689d1997-08-12 14:53:39 +0000983 ** Speedup: each sys.path item is interned, and
984 ** FindResourceModule remembers which items refer to
985 ** folders (so we don't have to bother trying to look
Tim Peters50d8d372001-02-28 05:34:27 +0000986 ** into them for resources).
Guido van Rossum741689d1997-08-12 14:53:39 +0000987 */
988 PyString_InternInPlace(&PyList_GET_ITEM(path, i));
989 v = PyList_GET_ITEM(path, i);
990#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000991 if (PyMac_FindResourceModule((PyStringObject *)v, name, buf)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000992 static struct filedescr resfiledescr =
993 {"", "", PY_RESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +0000994
Jack Jansen9c96a921995-02-15 22:57:06 +0000995 return &resfiledescr;
996 }
Guido van Rossum0f84a341998-08-06 13:36:01 +0000997 if (PyMac_FindCodeResourceModule((PyStringObject *)v, name, buf)) {
998 static struct filedescr resfiledescr =
999 {"", "", PY_CODERESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +00001000
Guido van Rossum0f84a341998-08-06 13:36:01 +00001001 return &resfiledescr;
1002 }
Jack Jansen9c96a921995-02-15 22:57:06 +00001003#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001004 if (len > 0 && buf[len-1] != SEP
1005#ifdef ALTSEP
1006 && buf[len-1] != ALTSEP
1007#endif
1008 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001009 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001010 strcpy(buf+len, name);
1011 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001012
1013 /* Check for package import (buf holds a directory name,
1014 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001015#ifdef HAVE_STAT
Tim Peterscab3f682001-04-29 22:21:25 +00001016 if (stat(buf, &statbuf) == 0 && /* it exists */
1017 S_ISDIR(statbuf.st_mode) && /* it's a directory */
1018 find_init_module(buf) && /* it has __init__.py */
1019 case_ok(buf, len, namelen, name)) /* and case matches */
1020 return &fd_package;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001021#else
1022 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001023#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001024 if (isdir(buf) &&
1025 find_init_module(buf) &&
1026 case_ok(buf, len, namelen, name))
1027 return &fd_package;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001028#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001029#endif
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001030#ifdef macintosh
1031 fdp = PyMac_FindModuleExtension(buf, &len, name);
Jack Jansen4df3c522001-03-20 23:09:54 +00001032 if (fdp) {
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001033#else
Andrew MacIntyred9400542002-02-26 11:41:34 +00001034#if defined(PYOS_OS2)
1035 /* take a snapshot of the module spec for restoration
1036 * after the 8 character DLL hackery
1037 */
1038 saved_buf = strdup(buf);
1039 saved_len = len;
1040 saved_namelen = namelen;
1041#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001042 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001043#if defined(PYOS_OS2)
1044 /* OS/2 limits DLLs to 8 character names (w/o extension)
1045 * so if the name is longer than that and its a
1046 * dynamically loaded module we're going to try,
1047 * truncate the name before trying
1048 */
1049 if (strlen(realname) > 8) {
1050 /* is this an attempt to load a C extension? */
1051 const struct filedescr *scan = _PyImport_DynLoadFiletab;
1052 while (scan->suffix != NULL) {
1053 if (strcmp(scan->suffix, fdp->suffix) == 0)
1054 break;
1055 else
1056 scan++;
1057 }
1058 if (scan->suffix != NULL) {
1059 /* yes, so truncate the name */
1060 namelen = 8;
1061 len -= strlen(realname) - namelen;
1062 buf[len] = '\0';
1063 }
1064 }
1065#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001066 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001067 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001068 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansen4df3c522001-03-20 23:09:54 +00001069#endif /* !macintosh */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001070 fp = fopen(buf, fdp->mode);
Tim Peters50d8d372001-02-28 05:34:27 +00001071 if (fp != NULL) {
1072 if (case_ok(buf, len, namelen, name))
1073 break;
1074 else { /* continue search */
1075 fclose(fp);
1076 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001077 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001078 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001079#if defined(PYOS_OS2)
1080 /* restore the saved snapshot */
1081 strcpy(buf, saved_buf);
1082 len = saved_len;
1083 namelen = saved_namelen;
1084#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001085 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001086#if defined(PYOS_OS2)
1087 /* don't need/want the module name snapshot anymore */
1088 if (saved_buf)
1089 {
1090 free(saved_buf);
1091 saved_buf = NULL;
1092 }
1093#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001094 if (fp != NULL)
1095 break;
1096 }
1097 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001098 PyErr_Format(PyExc_ImportError,
1099 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001100 return NULL;
1101 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001102 *p_fp = fp;
1103 return fdp;
1104}
1105
Tim Petersd1e87a82001-03-01 18:12:00 +00001106/* case_ok(char* buf, int len, int namelen, char* name)
1107 * The arguments here are tricky, best shown by example:
1108 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1109 * ^ ^ ^ ^
1110 * |--------------------- buf ---------------------|
1111 * |------------------- len ------------------|
1112 * |------ name -------|
1113 * |----- namelen -----|
1114 * buf is the full path, but len only counts up to (& exclusive of) the
1115 * extension. name is the module name, also exclusive of extension.
1116 *
1117 * We've already done a successful stat() or fopen() on buf, so know that
1118 * there's some match, possibly case-insensitive.
1119 *
Tim Peters50d8d372001-02-28 05:34:27 +00001120 * case_ok() is to return 1 if there's a case-sensitive match for
1121 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1122 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001123 *
Tim Peters50d8d372001-02-28 05:34:27 +00001124 * case_ok() is used to implement case-sensitive import semantics even
1125 * on platforms with case-insensitive filesystems. It's trivial to implement
1126 * for case-sensitive filesystems. It's pretty much a cross-platform
1127 * nightmare for systems with case-insensitive filesystems.
1128 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001129
Tim Peters50d8d372001-02-28 05:34:27 +00001130/* First we may need a pile of platform-specific header files; the sequence
1131 * of #if's here should match the sequence in the body of case_ok().
1132 */
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001133#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001134#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001135#ifdef __CYGWIN__
1136#include <sys/cygwin.h>
1137#endif
1138
Tim Peters50d8d372001-02-28 05:34:27 +00001139#elif defined(DJGPP)
1140#include <dir.h>
1141
1142#elif defined(macintosh)
1143#include <TextUtils.h>
1144#ifdef USE_GUSI1
1145#include "TFileSpec.h" /* for Path2FSSpec() */
1146#endif
1147
Tim Petersd1e87a82001-03-01 18:12:00 +00001148#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001149#include <sys/types.h>
1150#include <dirent.h>
1151
Andrew MacIntyred9400542002-02-26 11:41:34 +00001152#elif defined(PYOS_OS2)
1153#define INCL_DOS
1154#define INCL_DOSERRORS
1155#define INCL_NOPMAPI
1156#include <os2.h>
1157
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001158#elif defined(RISCOS)
1159#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001160#endif
1161
Guido van Rossum0980bd91998-02-13 17:18:36 +00001162static int
Tim Peters50d8d372001-02-28 05:34:27 +00001163case_ok(char *buf, int len, int namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001164{
Tim Peters50d8d372001-02-28 05:34:27 +00001165/* Pick a platform-specific implementation; the sequence of #if's here should
1166 * match the sequence just above.
1167 */
1168
1169/* MS_WIN32 || __CYGWIN__ */
1170#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001171 WIN32_FIND_DATA data;
1172 HANDLE h;
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001173#ifdef __CYGWIN__
1174 char tempbuf[MAX_PATH];
1175#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001176
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001177 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001178 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001179
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001180#ifdef __CYGWIN__
1181 cygwin32_conv_to_win32_path(buf, tempbuf);
1182 h = FindFirstFile(tempbuf, &data);
1183#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001184 h = FindFirstFile(buf, &data);
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001185#endif
Guido van Rossum0980bd91998-02-13 17:18:36 +00001186 if (h == INVALID_HANDLE_VALUE) {
1187 PyErr_Format(PyExc_NameError,
1188 "Can't find file for module %.100s\n(filename %.300s)",
1189 name, buf);
1190 return 0;
1191 }
1192 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001193 return strncmp(data.cFileName, name, namelen) == 0;
1194
1195/* DJGPP */
1196#elif defined(DJGPP)
1197 struct ffblk ffblk;
1198 int done;
1199
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001200 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001201 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001202
1203 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1204 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001205 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001206 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001207 name, buf);
1208 return 0;
1209 }
Tim Peters50d8d372001-02-28 05:34:27 +00001210 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001211
Tim Peters50d8d372001-02-28 05:34:27 +00001212/* macintosh */
1213#elif defined(macintosh)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001214 FSSpec fss;
1215 OSErr err;
Tim Peters50d8d372001-02-28 05:34:27 +00001216
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001217 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Peters50d8d372001-02-28 05:34:27 +00001218 return 1;
1219
Guido van Rossumb33aa1a2000-04-24 15:08:18 +00001220#ifndef USE_GUSI1
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001221 err = FSMakeFSSpec(0, 0, Pstring(buf), &fss);
1222#else
1223 /* GUSI's Path2FSSpec() resolves all possible aliases nicely on
1224 the way, which is fine for all directories, but here we need
1225 the original name of the alias file (say, Dlg.ppc.slb, not
1226 toolboxmodules.ppc.slb). */
1227 char *colon;
1228 err = Path2FSSpec(buf, &fss);
1229 if (err == noErr) {
1230 colon = strrchr(buf, ':'); /* find filename */
1231 if (colon != NULL)
1232 err = FSMakeFSSpec(fss.vRefNum, fss.parID,
1233 Pstring(colon+1), &fss);
1234 else
1235 err = FSMakeFSSpec(fss.vRefNum, fss.parID,
1236 fss.name, &fss);
1237 }
1238#endif
Guido van Rossum0980bd91998-02-13 17:18:36 +00001239 if (err) {
1240 PyErr_Format(PyExc_NameError,
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001241 "Can't find file for module %.100s\n(filename %.300s)",
1242 name, buf);
Guido van Rossum0980bd91998-02-13 17:18:36 +00001243 return 0;
1244 }
Tim Peters50d8d372001-02-28 05:34:27 +00001245 return fss.name[0] >= namelen &&
1246 strncmp(name, (char *)fss.name+1, namelen) == 0;
1247
Tim Peters677898a2001-03-02 03:28:03 +00001248/* new-fangled macintosh (macosx) */
Tim Petersd1e87a82001-03-01 18:12:00 +00001249#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001250 DIR *dirp;
1251 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001252 char dirname[MAXPATHLEN + 1];
1253 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001254
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001255 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001256 return 1;
1257
Tim Petersd1e87a82001-03-01 18:12:00 +00001258 /* Copy the dir component into dirname; substitute "." if empty */
1259 if (dirlen <= 0) {
1260 dirname[0] = '.';
1261 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001262 }
1263 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001264 assert(dirlen <= MAXPATHLEN);
1265 memcpy(dirname, buf, dirlen);
1266 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001267 }
1268 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001269 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001270 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001271 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001272 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001273 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001274#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001275 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001276#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001277 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001278#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001279 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001280 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001281 (void)closedir(dirp);
1282 return 1; /* Found */
1283 }
1284 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001285 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001286 }
Tim Peters430f5d42001-03-01 01:30:56 +00001287 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001288
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001289/* RISC OS */
1290#elif defined(RISCOS)
1291 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1292 char buf2[MAXPATHLEN+2];
1293 char *nameWithExt = buf+len-namelen;
1294 int canonlen;
1295 os_error *e;
1296
1297 if (Py_GETENV("PYTHONCASEOK") != NULL)
1298 return 1;
1299
1300 /* workaround:
1301 append wildcard, otherwise case of filename wouldn't be touched */
1302 strcpy(buf2, buf);
1303 strcat(buf2, "*");
1304
1305 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1306 canonlen = MAXPATHLEN+1-canonlen;
1307 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1308 return 0;
1309 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1310 return 1; /* match */
1311
1312 return 0;
1313
Andrew MacIntyred9400542002-02-26 11:41:34 +00001314/* OS/2 */
1315#elif defined(PYOS_OS2)
1316 HDIR hdir = 1;
1317 ULONG srchcnt = 1;
1318 FILEFINDBUF3 ffbuf;
1319 APIRET rc;
1320
1321 if (getenv("PYTHONCASEOK") != NULL)
1322 return 1;
1323
1324 rc = DosFindFirst(buf,
1325 &hdir,
1326 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1327 &ffbuf, sizeof(ffbuf),
1328 &srchcnt,
1329 FIL_STANDARD);
1330 if (rc != NO_ERROR)
1331 return 0;
1332 return strncmp(ffbuf.achName, name, namelen) == 0;
1333
Tim Peters50d8d372001-02-28 05:34:27 +00001334/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1335#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001336 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001337
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001338#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001339}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001340
Guido van Rossum0980bd91998-02-13 17:18:36 +00001341
Guido van Rossum197346f1997-10-31 18:38:52 +00001342#ifdef HAVE_STAT
1343/* Helper to look for __init__.py or __init__.py[co] in potential package */
1344static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001345find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001346{
Tim Peters0f9431f2001-07-05 03:47:53 +00001347 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001348 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001349 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001350 struct stat statbuf;
1351
Tim Peters0f9431f2001-07-05 03:47:53 +00001352/* For calling case_ok(buf, len, namelen, name):
1353 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1354 * ^ ^ ^ ^
1355 * |--------------------- buf ---------------------|
1356 * |------------------- len ------------------|
1357 * |------ name -------|
1358 * |----- namelen -----|
1359 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001360 if (save_len + 13 >= MAXPATHLEN)
1361 return 0;
1362 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001363 pname = buf + i;
1364 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001365 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001366 if (case_ok(buf,
1367 save_len + 9, /* len("/__init__") */
1368 8, /* len("__init__") */
1369 pname)) {
1370 buf[save_len] = '\0';
1371 return 1;
1372 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001373 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001374 i += strlen(pname);
1375 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001376 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001377 if (case_ok(buf,
1378 save_len + 9, /* len("/__init__") */
1379 8, /* len("__init__") */
1380 pname)) {
1381 buf[save_len] = '\0';
1382 return 1;
1383 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001384 }
1385 buf[save_len] = '\0';
1386 return 0;
1387}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001388
1389#else
1390
1391#ifdef RISCOS
1392static int
1393find_init_module(buf)
1394 char *buf;
1395{
1396 int save_len = strlen(buf);
1397 int i = save_len;
1398
1399 if (save_len + 13 >= MAXPATHLEN)
1400 return 0;
1401 buf[i++] = SEP;
1402 strcpy(buf+i, "__init__/py");
1403 if (isfile(buf)) {
1404 buf[save_len] = '\0';
1405 return 1;
1406 }
1407
1408 if (Py_OptimizeFlag)
1409 strcpy(buf+i, "o");
1410 else
1411 strcpy(buf+i, "c");
1412 if (isfile(buf)) {
1413 buf[save_len] = '\0';
1414 return 1;
1415 }
1416 buf[save_len] = '\0';
1417 return 0;
1418}
1419#endif /*RISCOS*/
1420
Guido van Rossum197346f1997-10-31 18:38:52 +00001421#endif /* HAVE_STAT */
1422
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001423
Tim Petersdbd9ba62000-07-09 03:09:57 +00001424static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001425
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001426/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001427 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001428
Guido van Rossum79f25d91997-04-29 20:08:16 +00001429static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001430load_module(char *name, FILE *fp, char *buf, int type)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001431{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001432 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001433 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001434 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001435
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001436 /* First check that there's an open file (if we need one) */
1437 switch (type) {
1438 case PY_SOURCE:
1439 case PY_COMPILED:
1440 if (fp == NULL) {
1441 PyErr_Format(PyExc_ValueError,
1442 "file object required for import (type code %d)",
1443 type);
1444 return NULL;
1445 }
1446 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001447
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001448 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001449
1450 case PY_SOURCE:
1451 m = load_source_module(name, buf, fp);
1452 break;
1453
1454 case PY_COMPILED:
1455 m = load_compiled_module(name, buf, fp);
1456 break;
1457
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001458#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001459 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001460 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001461 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001462#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001463
Jack Jansen9c96a921995-02-15 22:57:06 +00001464#ifdef macintosh
1465 case PY_RESOURCE:
1466 m = PyMac_LoadResourceModule(name, buf);
1467 break;
Guido van Rossum0f84a341998-08-06 13:36:01 +00001468 case PY_CODERESOURCE:
1469 m = PyMac_LoadCodeResourceModule(name, buf);
1470 break;
Jack Jansen9c96a921995-02-15 22:57:06 +00001471#endif
1472
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001473 case PKG_DIRECTORY:
1474 m = load_package(name, buf);
1475 break;
1476
1477 case C_BUILTIN:
1478 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001479 if (buf != NULL && buf[0] != '\0')
1480 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001481 if (type == C_BUILTIN)
1482 err = init_builtin(name);
1483 else
1484 err = PyImport_ImportFrozenModule(name);
1485 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001486 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001487 if (err == 0) {
1488 PyErr_Format(PyExc_ImportError,
1489 "Purported %s module %.200s not found",
1490 type == C_BUILTIN ?
1491 "builtin" : "frozen",
1492 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001493 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001494 }
1495 modules = PyImport_GetModuleDict();
1496 m = PyDict_GetItemString(modules, name);
1497 if (m == NULL) {
1498 PyErr_Format(
1499 PyExc_ImportError,
1500 "%s module %.200s not properly initialized",
1501 type == C_BUILTIN ?
1502 "builtin" : "frozen",
1503 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001504 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001505 }
1506 Py_INCREF(m);
1507 break;
1508
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001509 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001510 PyErr_Format(PyExc_ImportError,
1511 "Don't know how to import %.200s (type code %d)",
1512 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001513 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001514
1515 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001516
1517 return m;
1518}
1519
1520
1521/* Initialize a built-in module.
1522 Return 1 for succes, 0 if the module is not found, and -1 with
1523 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001524
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001525static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001526init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001527{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001528 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001529
Greg Ward201baee2001-10-04 14:52:06 +00001530 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001531 return 1;
1532
Guido van Rossum771c6c81997-10-31 18:37:24 +00001533 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001534 if (strcmp(name, p->name) == 0) {
1535 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001536 PyErr_Format(PyExc_ImportError,
1537 "Cannot re-init internal module %.200s",
1538 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001539 return -1;
1540 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001541 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001542 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001543 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001544 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001545 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001546 if (_PyImport_FixupExtension(name, name) == NULL)
1547 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001548 return 1;
1549 }
1550 }
1551 return 0;
1552}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001553
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001554
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001555/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001556
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001557static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001558find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001559{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001560 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001561
Guido van Rossum79f25d91997-04-29 20:08:16 +00001562 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001563 if (p->name == NULL)
1564 return NULL;
1565 if (strcmp(p->name, name) == 0)
1566 break;
1567 }
1568 return p;
1569}
1570
Guido van Rossum79f25d91997-04-29 20:08:16 +00001571static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001572get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001573{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001574 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001575 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001576
1577 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001578 PyErr_Format(PyExc_ImportError,
1579 "No such frozen object named %.200s",
1580 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001581 return NULL;
1582 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001583 if (p->code == NULL) {
1584 PyErr_Format(PyExc_ImportError,
1585 "Excluded frozen object named %.200s",
1586 name);
1587 return NULL;
1588 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001589 size = p->size;
1590 if (size < 0)
1591 size = -size;
1592 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001593}
1594
1595/* Initialize a frozen module.
1596 Return 1 for succes, 0 if the module is not found, and -1 with
1597 an exception set if the initialization failed.
1598 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001599
1600int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001601PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001602{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001603 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001604 PyObject *co;
1605 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001606 int ispackage;
1607 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001608
1609 if (p == NULL)
1610 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001611 if (p->code == NULL) {
1612 PyErr_Format(PyExc_ImportError,
1613 "Excluded frozen object named %.200s",
1614 name);
1615 return -1;
1616 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001617 size = p->size;
1618 ispackage = (size < 0);
1619 if (ispackage)
1620 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001621 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001622 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001623 name, ispackage ? " package" : "");
1624 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001625 if (co == NULL)
1626 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001627 if (!PyCode_Check(co)) {
1628 Py_DECREF(co);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001629 PyErr_Format(PyExc_TypeError,
1630 "frozen object %.200s is not a code object",
1631 name);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001632 return -1;
1633 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001634 if (ispackage) {
1635 /* Set __path__ to the package name */
1636 PyObject *d, *s;
1637 int err;
1638 m = PyImport_AddModule(name);
1639 if (m == NULL)
1640 return -1;
1641 d = PyModule_GetDict(m);
1642 s = PyString_InternFromString(name);
1643 if (s == NULL)
1644 return -1;
1645 err = PyDict_SetItemString(d, "__path__", s);
1646 Py_DECREF(s);
1647 if (err != 0)
1648 return err;
1649 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001650 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum79f25d91997-04-29 20:08:16 +00001651 Py_DECREF(co);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001652 if (m == NULL)
1653 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001654 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001655 return 1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001656}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001657
1658
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001659/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001660 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001661
Guido van Rossum79f25d91997-04-29 20:08:16 +00001662PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001663PyImport_ImportModule(char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001664{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001665 PyObject *pname;
1666 PyObject *result;
1667
1668 pname = PyString_FromString(name);
1669 result = PyImport_Import(pname);
1670 Py_DECREF(pname);
1671 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001672}
1673
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001674/* Forward declarations for helper routines */
Tim Petersdbd9ba62000-07-09 03:09:57 +00001675static PyObject *get_parent(PyObject *globals, char *buf, int *p_buflen);
1676static PyObject *load_next(PyObject *mod, PyObject *altmod,
1677 char **p_name, char *buf, int *p_buflen);
1678static int mark_miss(char *name);
1679static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
1680 char *buf, int buflen, int recursive);
1681static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001682
1683/* The Magnum Opus of dotted-name import :-) */
1684
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001685static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001686import_module_ex(char *name, PyObject *globals, PyObject *locals,
1687 PyObject *fromlist)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001688{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001689 char buf[MAXPATHLEN+1];
1690 int buflen = 0;
1691 PyObject *parent, *head, *next, *tail;
1692
1693 parent = get_parent(globals, buf, &buflen);
1694 if (parent == NULL)
1695 return NULL;
1696
1697 head = load_next(parent, Py_None, &name, buf, &buflen);
1698 if (head == NULL)
1699 return NULL;
1700
1701 tail = head;
1702 Py_INCREF(tail);
1703 while (name) {
1704 next = load_next(tail, tail, &name, buf, &buflen);
1705 Py_DECREF(tail);
1706 if (next == NULL) {
1707 Py_DECREF(head);
1708 return NULL;
1709 }
1710 tail = next;
1711 }
1712
1713 if (fromlist != NULL) {
1714 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
1715 fromlist = NULL;
1716 }
1717
1718 if (fromlist == NULL) {
1719 Py_DECREF(tail);
1720 return head;
1721 }
1722
1723 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001724 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001725 Py_DECREF(tail);
1726 return NULL;
1727 }
1728
1729 return tail;
1730}
1731
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001732PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001733PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
1734 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001735{
1736 PyObject *result;
1737 lock_import();
Guido van Rossumd65911b1998-03-03 22:33:27 +00001738 result = import_module_ex(name, globals, locals, fromlist);
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001739 unlock_import();
1740 return result;
1741}
1742
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001743static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001744get_parent(PyObject *globals, char *buf, int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001745{
1746 static PyObject *namestr = NULL;
1747 static PyObject *pathstr = NULL;
1748 PyObject *modname, *modpath, *modules, *parent;
1749
1750 if (globals == NULL || !PyDict_Check(globals))
1751 return Py_None;
1752
1753 if (namestr == NULL) {
1754 namestr = PyString_InternFromString("__name__");
1755 if (namestr == NULL)
1756 return NULL;
1757 }
1758 if (pathstr == NULL) {
1759 pathstr = PyString_InternFromString("__path__");
1760 if (pathstr == NULL)
1761 return NULL;
1762 }
1763
1764 *buf = '\0';
1765 *p_buflen = 0;
1766 modname = PyDict_GetItem(globals, namestr);
1767 if (modname == NULL || !PyString_Check(modname))
1768 return Py_None;
1769
1770 modpath = PyDict_GetItem(globals, pathstr);
1771 if (modpath != NULL) {
1772 int len = PyString_GET_SIZE(modname);
1773 if (len > MAXPATHLEN) {
1774 PyErr_SetString(PyExc_ValueError,
1775 "Module name too long");
1776 return NULL;
1777 }
1778 strcpy(buf, PyString_AS_STRING(modname));
1779 *p_buflen = len;
1780 }
1781 else {
1782 char *start = PyString_AS_STRING(modname);
1783 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001784 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001785 if (lastdot == NULL)
1786 return Py_None;
1787 len = lastdot - start;
1788 if (len >= MAXPATHLEN) {
1789 PyErr_SetString(PyExc_ValueError,
1790 "Module name too long");
1791 return NULL;
1792 }
1793 strncpy(buf, start, len);
1794 buf[len] = '\0';
1795 *p_buflen = len;
1796 }
1797
1798 modules = PyImport_GetModuleDict();
1799 parent = PyDict_GetItemString(modules, buf);
1800 if (parent == NULL)
1801 parent = Py_None;
1802 return parent;
1803 /* We expect, but can't guarantee, if parent != None, that:
1804 - parent.__name__ == buf
1805 - parent.__dict__ is globals
1806 If this is violated... Who cares? */
1807}
1808
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001809/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001810static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001811load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
1812 int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001813{
1814 char *name = *p_name;
1815 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001816 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001817 char *p;
1818 PyObject *result;
1819
1820 if (dot == NULL) {
1821 *p_name = NULL;
1822 len = strlen(name);
1823 }
1824 else {
1825 *p_name = dot+1;
1826 len = dot-name;
1827 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00001828 if (len == 0) {
1829 PyErr_SetString(PyExc_ValueError,
1830 "Empty module name");
1831 return NULL;
1832 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001833
1834 p = buf + *p_buflen;
1835 if (p != buf)
1836 *p++ = '.';
1837 if (p+len-buf >= MAXPATHLEN) {
1838 PyErr_SetString(PyExc_ValueError,
1839 "Module name too long");
1840 return NULL;
1841 }
1842 strncpy(p, name, len);
1843 p[len] = '\0';
1844 *p_buflen = p+len-buf;
1845
1846 result = import_submodule(mod, p, buf);
1847 if (result == Py_None && altmod != mod) {
1848 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001849 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00001850 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001851 if (result != NULL && result != Py_None) {
1852 if (mark_miss(buf) != 0) {
1853 Py_DECREF(result);
1854 return NULL;
1855 }
1856 strncpy(buf, name, len);
1857 buf[len] = '\0';
1858 *p_buflen = len;
1859 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001860 }
1861 if (result == NULL)
1862 return NULL;
1863
1864 if (result == Py_None) {
1865 Py_DECREF(result);
1866 PyErr_Format(PyExc_ImportError,
1867 "No module named %.200s", name);
1868 return NULL;
1869 }
1870
1871 return result;
1872}
1873
1874static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001875mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001876{
1877 PyObject *modules = PyImport_GetModuleDict();
1878 return PyDict_SetItemString(modules, name, Py_None);
1879}
1880
1881static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001882ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, int buflen,
1883 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001884{
1885 int i;
1886
1887 if (!PyObject_HasAttrString(mod, "__path__"))
1888 return 1;
1889
1890 for (i = 0; ; i++) {
1891 PyObject *item = PySequence_GetItem(fromlist, i);
1892 int hasit;
1893 if (item == NULL) {
1894 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
1895 PyErr_Clear();
1896 return 1;
1897 }
1898 return 0;
1899 }
1900 if (!PyString_Check(item)) {
1901 PyErr_SetString(PyExc_TypeError,
1902 "Item in ``from list'' not a string");
1903 Py_DECREF(item);
1904 return 0;
1905 }
1906 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00001907 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001908 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001909 /* See if the package defines __all__ */
1910 if (recursive)
1911 continue; /* Avoid endless recursion */
1912 all = PyObject_GetAttrString(mod, "__all__");
1913 if (all == NULL)
1914 PyErr_Clear();
1915 else {
1916 if (!ensure_fromlist(mod, all, buf, buflen, 1))
1917 return 0;
1918 Py_DECREF(all);
1919 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001920 continue;
1921 }
1922 hasit = PyObject_HasAttr(mod, item);
1923 if (!hasit) {
1924 char *subname = PyString_AS_STRING(item);
1925 PyObject *submod;
1926 char *p;
1927 if (buflen + strlen(subname) >= MAXPATHLEN) {
1928 PyErr_SetString(PyExc_ValueError,
1929 "Module name too long");
1930 Py_DECREF(item);
1931 return 0;
1932 }
1933 p = buf + buflen;
1934 *p++ = '.';
1935 strcpy(p, subname);
1936 submod = import_submodule(mod, subname, buf);
1937 Py_XDECREF(submod);
1938 if (submod == NULL) {
1939 Py_DECREF(item);
1940 return 0;
1941 }
1942 }
1943 Py_DECREF(item);
1944 }
1945
Guido van Rossuma7f2e811997-10-03 15:33:32 +00001946 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001947}
1948
1949static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001950import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001951{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001952 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001953 PyObject *m, *res = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00001954
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001955 /* Require:
1956 if mod == None: subname == fullname
1957 else: mod.__name__ + "." + subname == fullname
1958 */
1959
Tim Peters50d8d372001-02-28 05:34:27 +00001960 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001961 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001962 }
1963 else {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001964 PyObject *path;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001965 char buf[MAXPATHLEN+1];
1966 struct filedescr *fdp;
1967 FILE *fp = NULL;
1968
Guido van Rossum9c0afe51998-05-19 15:09:05 +00001969 if (mod == Py_None)
1970 path = NULL;
1971 else {
1972 path = PyObject_GetAttrString(mod, "__path__");
1973 if (path == NULL) {
1974 PyErr_Clear();
1975 Py_INCREF(Py_None);
1976 return Py_None;
1977 }
1978 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001979
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001980 buf[0] = '\0';
Guido van Rossumb68cd421998-07-01 17:36:26 +00001981 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
1982 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001983 if (fdp == NULL) {
1984 if (!PyErr_ExceptionMatches(PyExc_ImportError))
1985 return NULL;
1986 PyErr_Clear();
1987 Py_INCREF(Py_None);
1988 return Py_None;
1989 }
1990 m = load_module(fullname, fp, buf, fdp->type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001991 if (fp)
1992 fclose(fp);
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001993 if (mod != Py_None) {
1994 /* Irrespective of the success of this load, make a
1995 reference to it in the parent package module.
1996 A copy gets saved in the modules dictionary
1997 under the full name, so get a reference from
1998 there, if need be. (The exception is when
1999 the load failed with a SyntaxError -- then
2000 there's no trace in sys.modules. In that case,
2001 of course, do nothing extra.) */
2002 res = m;
2003 if (res == NULL)
2004 res = PyDict_GetItemString(modules, fullname);
2005 if (res != NULL &&
2006 PyObject_SetAttrString(mod, subname, res) < 0) {
2007 Py_XDECREF(m);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002008 m = NULL;
2009 }
2010 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002011 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002012
2013 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002014}
2015
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002016
2017/* Re-import a module of any kind and return its module object, WITH
2018 INCREMENTED REFERENCE COUNT */
2019
Guido van Rossum79f25d91997-04-29 20:08:16 +00002020PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002021PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002022{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002023 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum222ef561997-09-06 19:41:09 +00002024 PyObject *path = NULL;
2025 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002026 char buf[MAXPATHLEN+1];
2027 struct filedescr *fdp;
2028 FILE *fp = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002029
Guido van Rossum79f25d91997-04-29 20:08:16 +00002030 if (m == NULL || !PyModule_Check(m)) {
2031 PyErr_SetString(PyExc_TypeError,
2032 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002033 return NULL;
2034 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002035 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002036 if (name == NULL)
2037 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002038 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002039 PyErr_Format(PyExc_ImportError,
2040 "reload(): module %.200s not in sys.modules",
2041 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002042 return NULL;
2043 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002044 subname = strrchr(name, '.');
2045 if (subname == NULL)
2046 subname = name;
2047 else {
2048 PyObject *parentname, *parent;
2049 parentname = PyString_FromStringAndSize(name, (subname-name));
2050 if (parentname == NULL)
2051 return NULL;
2052 parent = PyDict_GetItem(modules, parentname);
Barry Warsaw38793331999-01-27 17:54:20 +00002053 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002054 if (parent == NULL) {
2055 PyErr_Format(PyExc_ImportError,
2056 "reload(): parent %.200s not in sys.modules",
2057 name);
2058 return NULL;
2059 }
2060 subname++;
2061 path = PyObject_GetAttrString(parent, "__path__");
2062 if (path == NULL)
2063 PyErr_Clear();
2064 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002065 buf[0] = '\0';
Guido van Rossum222ef561997-09-06 19:41:09 +00002066 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
2067 Py_XDECREF(path);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002068 if (fdp == NULL)
2069 return NULL;
2070 m = load_module(name, fp, buf, fdp->type);
2071 if (fp)
2072 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002073 return m;
2074}
2075
2076
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002077/* Higher-level import emulator which emulates the "import" statement
2078 more accurately -- it invokes the __import__() function from the
2079 builtins of the current globals. This means that the import is
2080 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002081 environment, e.g. by "rexec".
2082 A dummy list ["__doc__"] is passed as the 4th argument so that
2083 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2084 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002085
2086PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002087PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002088{
2089 static PyObject *silly_list = NULL;
2090 static PyObject *builtins_str = NULL;
2091 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002092 PyObject *globals = NULL;
2093 PyObject *import = NULL;
2094 PyObject *builtins = NULL;
2095 PyObject *r = NULL;
2096
2097 /* Initialize constant string objects */
2098 if (silly_list == NULL) {
2099 import_str = PyString_InternFromString("__import__");
2100 if (import_str == NULL)
2101 return NULL;
2102 builtins_str = PyString_InternFromString("__builtins__");
2103 if (builtins_str == NULL)
2104 return NULL;
2105 silly_list = Py_BuildValue("[s]", "__doc__");
2106 if (silly_list == NULL)
2107 return NULL;
2108 }
2109
2110 /* Get the builtins from current globals */
2111 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002112 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002113 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002114 builtins = PyObject_GetItem(globals, builtins_str);
2115 if (builtins == NULL)
2116 goto err;
2117 }
2118 else {
2119 /* No globals -- use standard builtins, and fake globals */
2120 PyErr_Clear();
2121
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002122 builtins = PyImport_ImportModuleEx("__builtin__",
2123 NULL, NULL, NULL);
2124 if (builtins == NULL)
2125 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002126 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2127 if (globals == NULL)
2128 goto err;
2129 }
2130
2131 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002132 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002133 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002134 if (import == NULL)
2135 PyErr_SetObject(PyExc_KeyError, import_str);
2136 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002137 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002138 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002139 if (import == NULL)
2140 goto err;
2141
2142 /* Call the _import__ function with the proper argument list */
2143 r = PyObject_CallFunction(import, "OOOO",
2144 module_name, globals, globals, silly_list);
2145
2146 err:
2147 Py_XDECREF(globals);
2148 Py_XDECREF(builtins);
2149 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002150
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002151 return r;
2152}
2153
2154
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002155/* Module 'imp' provides Python access to the primitives used for
2156 importing modules.
2157*/
2158
Guido van Rossum79f25d91997-04-29 20:08:16 +00002159static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002160imp_get_magic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002161{
2162 char buf[4];
2163
Guido van Rossum43713e52000-02-29 13:59:29 +00002164 if (!PyArg_ParseTuple(args, ":get_magic"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002165 return NULL;
Guido van Rossum96774c12000-05-01 20:19:08 +00002166 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2167 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2168 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2169 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002170
Guido van Rossum79f25d91997-04-29 20:08:16 +00002171 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002172}
2173
Guido van Rossum79f25d91997-04-29 20:08:16 +00002174static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002175imp_get_suffixes(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002176{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002177 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002178 struct filedescr *fdp;
2179
Guido van Rossum43713e52000-02-29 13:59:29 +00002180 if (!PyArg_ParseTuple(args, ":get_suffixes"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002181 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002182 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002183 if (list == NULL)
2184 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002185 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2186 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002187 fdp->suffix, fdp->mode, fdp->type);
2188 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002189 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002190 return NULL;
2191 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002192 if (PyList_Append(list, item) < 0) {
2193 Py_DECREF(list);
2194 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002195 return NULL;
2196 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002197 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002198 }
2199 return list;
2200}
2201
Guido van Rossum79f25d91997-04-29 20:08:16 +00002202static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002203call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002204{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002205 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002206 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002207 struct filedescr *fdp;
2208 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002209 FILE *fp = NULL;
2210
2211 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002212 if (path == Py_None)
2213 path = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002214 fdp = find_module(name, path, pathname, MAXPATHLEN+1, &fp);
2215 if (fdp == NULL)
2216 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002217 if (fp != NULL) {
2218 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2219 if (fob == NULL) {
2220 fclose(fp);
2221 return NULL;
2222 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002223 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002224 else {
2225 fob = Py_None;
2226 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002227 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002228 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002229 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002230 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002231 return ret;
2232}
2233
Guido van Rossum79f25d91997-04-29 20:08:16 +00002234static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002235imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002236{
2237 char *name;
2238 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002239 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002240 return NULL;
2241 return call_find_module(name, path);
2242}
2243
2244static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002245imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002246{
2247 char *name;
2248 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002249 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002250 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002251 return NULL;
2252 ret = init_builtin(name);
2253 if (ret < 0)
2254 return NULL;
2255 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002256 Py_INCREF(Py_None);
2257 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002258 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002259 m = PyImport_AddModule(name);
2260 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002261 return m;
2262}
2263
Guido van Rossum79f25d91997-04-29 20:08:16 +00002264static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002265imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002266{
2267 char *name;
2268 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002269 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002270 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002271 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002272 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002273 if (ret < 0)
2274 return NULL;
2275 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002276 Py_INCREF(Py_None);
2277 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002278 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002279 m = PyImport_AddModule(name);
2280 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002281 return m;
2282}
2283
Guido van Rossum79f25d91997-04-29 20:08:16 +00002284static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002285imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002286{
2287 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002288
Guido van Rossum43713e52000-02-29 13:59:29 +00002289 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002290 return NULL;
2291 return get_frozen_object(name);
2292}
2293
Guido van Rossum79f25d91997-04-29 20:08:16 +00002294static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002295imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002296{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002297 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002298 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002299 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002300 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002301}
2302
Guido van Rossum79f25d91997-04-29 20:08:16 +00002303static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002304imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002305{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002306 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002307 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002308 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002309 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002310 p = find_frozen(name);
2311 return PyInt_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002312}
2313
2314static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002315get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002316{
2317 FILE *fp;
2318 if (fob == NULL) {
2319 fp = fopen(pathname, mode);
2320 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002321 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002322 }
2323 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002324 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002325 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002326 PyErr_SetString(PyExc_ValueError,
2327 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002328 }
2329 return fp;
2330}
2331
Guido van Rossum79f25d91997-04-29 20:08:16 +00002332static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002333imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002334{
2335 char *name;
2336 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002337 PyObject *fob = NULL;
2338 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002339 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002340 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002341 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002342 return NULL;
2343 fp = get_file(pathname, fob, "rb");
2344 if (fp == NULL)
2345 return NULL;
2346 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002347 if (fob == NULL)
2348 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002349 return m;
2350}
2351
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002352#ifdef HAVE_DYNAMIC_LOADING
2353
Guido van Rossum79f25d91997-04-29 20:08:16 +00002354static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002355imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002356{
2357 char *name;
2358 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002359 PyObject *fob = NULL;
2360 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002361 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002362 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002363 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002364 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002365 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002366 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002367 if (fp == NULL)
2368 return NULL;
2369 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002370 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002371 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002372}
2373
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002374#endif /* HAVE_DYNAMIC_LOADING */
2375
Guido van Rossum79f25d91997-04-29 20:08:16 +00002376static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002377imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002378{
2379 char *name;
2380 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002381 PyObject *fob = NULL;
2382 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002383 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002384 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002385 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002386 return NULL;
2387 fp = get_file(pathname, fob, "r");
2388 if (fp == NULL)
2389 return NULL;
2390 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002391 if (fob == NULL)
2392 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002393 return m;
2394}
2395
Jack Jansen9c96a921995-02-15 22:57:06 +00002396#ifdef macintosh
Guido van Rossum79f25d91997-04-29 20:08:16 +00002397static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002398imp_load_resource(PyObject *self, PyObject *args)
Jack Jansen9c96a921995-02-15 22:57:06 +00002399{
2400 char *name;
2401 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002402 PyObject *m;
Jack Jansen9c96a921995-02-15 22:57:06 +00002403
Guido van Rossum43713e52000-02-29 13:59:29 +00002404 if (!PyArg_ParseTuple(args, "ss:load_resource", &name, &pathname))
Jack Jansen9c96a921995-02-15 22:57:06 +00002405 return NULL;
2406 m = PyMac_LoadResourceModule(name, pathname);
2407 return m;
2408}
2409#endif /* macintosh */
2410
Guido van Rossum79f25d91997-04-29 20:08:16 +00002411static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002412imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002413{
2414 char *name;
2415 PyObject *fob;
2416 char *pathname;
2417 char *suffix; /* Unused */
2418 char *mode;
2419 int type;
2420 FILE *fp;
2421
Guido van Rossum43713e52000-02-29 13:59:29 +00002422 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002423 &name, &fob, &pathname,
2424 &suffix, &mode, &type))
2425 return NULL;
2426 if (*mode && (*mode != 'r' || strchr(mode, '+') != NULL)) {
2427 PyErr_Format(PyExc_ValueError,
2428 "invalid file open mode %.200s", mode);
2429 return NULL;
2430 }
2431 if (fob == Py_None)
2432 fp = NULL;
2433 else {
2434 if (!PyFile_Check(fob)) {
2435 PyErr_SetString(PyExc_ValueError,
2436 "load_module arg#2 should be a file or None");
2437 return NULL;
2438 }
2439 fp = get_file(pathname, fob, mode);
2440 if (fp == NULL)
2441 return NULL;
2442 }
2443 return load_module(name, fp, pathname, type);
2444}
2445
2446static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002447imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002448{
2449 char *name;
2450 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002451 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002452 return NULL;
2453 return load_package(name, pathname);
2454}
2455
2456static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002457imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002458{
2459 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002460 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002461 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002462 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002463}
2464
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002465/* Doc strings */
2466
2467static char doc_imp[] = "\
2468This module provides the components needed to build your own\n\
2469__import__ function. Undocumented functions are obsolete.\n\
2470";
2471
2472static char doc_find_module[] = "\
2473find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
2474Search for a module. If path is omitted or None, search for a\n\
2475built-in, frozen or special module and continue search in sys.path.\n\
2476The module name cannot contain '.'; to search for a submodule of a\n\
2477package, pass the submodule name and the package's __path__.\
2478";
2479
2480static char doc_load_module[] = "\
2481load_module(name, file, filename, (suffix, mode, type)) -> module\n\
2482Load a module, given information returned by find_module().\n\
2483The module name must include the full package name, if any.\
2484";
2485
2486static char doc_get_magic[] = "\
2487get_magic() -> string\n\
2488Return the magic number for .pyc or .pyo files.\
2489";
2490
2491static char doc_get_suffixes[] = "\
2492get_suffixes() -> [(suffix, mode, type), ...]\n\
2493Return a list of (suffix, mode, type) tuples describing the files\n\
2494that find_module() looks for.\
2495";
2496
2497static char doc_new_module[] = "\
2498new_module(name) -> module\n\
2499Create a new module. Do not enter it in sys.modules.\n\
2500The module name must include the full package name, if any.\
2501";
2502
Tim Peters69232342001-08-30 05:16:13 +00002503static char doc_lock_held[] = "\
2504lock_held() -> 0 or 1\n\
2505Return 1 if the import lock is currently held.\n\
2506On platforms without threads, return 0.\
2507";
2508
Guido van Rossum79f25d91997-04-29 20:08:16 +00002509static PyMethodDef imp_methods[] = {
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002510 {"find_module", imp_find_module, 1, doc_find_module},
2511 {"get_magic", imp_get_magic, 1, doc_get_magic},
2512 {"get_suffixes", imp_get_suffixes, 1, doc_get_suffixes},
2513 {"load_module", imp_load_module, 1, doc_load_module},
2514 {"new_module", imp_new_module, 1, doc_new_module},
Tim Peters69232342001-08-30 05:16:13 +00002515 {"lock_held", imp_lock_held, 1, doc_lock_held},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002516 /* The rest are obsolete */
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002517 {"get_frozen_object", imp_get_frozen_object, 1},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002518 {"init_builtin", imp_init_builtin, 1},
2519 {"init_frozen", imp_init_frozen, 1},
2520 {"is_builtin", imp_is_builtin, 1},
2521 {"is_frozen", imp_is_frozen, 1},
2522 {"load_compiled", imp_load_compiled, 1},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002523#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002524 {"load_dynamic", imp_load_dynamic, 1},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002525#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002526 {"load_package", imp_load_package, 1},
Jack Jansen9c96a921995-02-15 22:57:06 +00002527#ifdef macintosh
2528 {"load_resource", imp_load_resource, 1},
2529#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002530 {"load_source", imp_load_source, 1},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002531 {NULL, NULL} /* sentinel */
2532};
2533
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002534static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002535setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002536{
2537 PyObject *v;
2538 int err;
2539
2540 v = PyInt_FromLong((long)value);
2541 err = PyDict_SetItemString(d, name, v);
2542 Py_XDECREF(v);
2543 return err;
2544}
2545
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002546void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002547initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002548{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002549 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002550
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002551 m = Py_InitModule4("imp", imp_methods, doc_imp,
2552 NULL, PYTHON_API_VERSION);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002553 d = PyModule_GetDict(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002554
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002555 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
2556 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
2557 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
2558 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
2559 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
2560 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
2561 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
2562 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00002563 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002564
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002565 failure:
2566 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002567}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002568
2569
Guido van Rossumb18618d2000-05-03 23:44:39 +00002570/* API for embedding applications that want to add their own entries
2571 to the table of built-in modules. This should normally be called
2572 *before* Py_Initialize(). When the table resize fails, -1 is
2573 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002574
2575 After a similar function by Just van Rossum. */
2576
2577int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002578PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002579{
2580 static struct _inittab *our_copy = NULL;
2581 struct _inittab *p;
2582 int i, n;
2583
2584 /* Count the number of entries in both tables */
2585 for (n = 0; newtab[n].name != NULL; n++)
2586 ;
2587 if (n == 0)
2588 return 0; /* Nothing to do */
2589 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2590 ;
2591
2592 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00002593 p = our_copy;
2594 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002595 if (p == NULL)
2596 return -1;
2597
2598 /* Copy the tables into the new memory */
2599 if (our_copy != PyImport_Inittab)
2600 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2601 PyImport_Inittab = our_copy = p;
2602 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
2603
2604 return 0;
2605}
2606
2607/* Shorthand to add a single entry given a name and a function */
2608
2609int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002610PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002611{
2612 struct _inittab newtab[2];
2613
2614 memset(newtab, '\0', sizeof newtab);
2615
2616 newtab[0].name = name;
2617 newtab[0].initfunc = initfunc;
2618
2619 return PyImport_ExtendInittab(newtab);
2620}