blob: 8c9cd2f0f520ce19d8e10ab168259bd91bd64991 [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[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000084 {"/py", "U", PY_SOURCE},
Guido van Rossum48a680c2001-03-02 06:34:14 +000085 {"/pyc", "rb", PY_COMPILED},
86 {0, 0}
87};
88#else
Guido van Rossumed1170e1999-12-20 21:23:41 +000089static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +000090 {".py", "U", PY_SOURCE},
Tim Petersc1731372001-08-04 08:12:36 +000091#ifdef MS_WIN32
Jack Jansenc88da1f2002-05-28 10:58:19 +000092 {".pyw", "U", 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
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000217 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000218#else
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000219 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000220#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;
Jack Jansenc88da1f2002-05-28 10:58:19 +0000895 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +0000896 FILE *fp = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000897#ifndef RISCOS
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000898 struct stat statbuf;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000899#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +0000900 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
901 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
902 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +0000903 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +0000904#if defined(PYOS_OS2)
905 size_t saved_len;
906 size_t saved_namelen;
907 char *saved_buf = NULL;
908#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000909
Fred Drake4c82b232000-06-30 16:18:57 +0000910 if (strlen(realname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000911 PyErr_SetString(PyExc_OverflowError,
912 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +0000913 return NULL;
914 }
Guido van Rossum0506a431998-08-11 15:07:39 +0000915 strcpy(name, realname);
916
917 if (path != NULL && PyString_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000918 /* The only type of submodule allowed inside a "frozen"
919 package are other frozen modules or packages. */
Guido van Rossum0506a431998-08-11 15:07:39 +0000920 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) {
921 PyErr_SetString(PyExc_ImportError,
922 "full frozen module name too long");
923 return NULL;
924 }
925 strcpy(buf, PyString_AsString(path));
926 strcat(buf, ".");
927 strcat(buf, name);
928 strcpy(name, buf);
Jack Jansen550fdae2001-10-30 13:08:39 +0000929#ifdef macintosh
930 /* Freezing on the mac works different, and the modules are
931 ** actually on sys.path. So we don't take the quick exit but
932 ** continue with the normal flow.
933 */
934 path = NULL;
935#else
Guido van Rossum8f4d3312001-10-18 18:54:11 +0000936 if (find_frozen(name) != NULL) {
937 strcpy(buf, name);
938 return &fd_frozen;
939 }
940 PyErr_Format(PyExc_ImportError,
941 "No frozen submodule named %.200s", name);
942 return NULL;
Jack Jansen550fdae2001-10-30 13:08:39 +0000943#endif
Guido van Rossum0506a431998-08-11 15:07:39 +0000944 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000945 if (path == NULL) {
946 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000947 strcpy(buf, name);
948 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000949 }
Greg Ward201baee2001-10-04 14:52:06 +0000950 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +0000951 strcpy(buf, name);
952 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000953 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000954
Guido van Rossumac279101996-08-22 23:10:58 +0000955#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000956 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
957 if (fp != NULL) {
958 *p_fp = fp;
959 return fdp;
960 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +0000961#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +0000962 path = PySys_GetObject("path");
963 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000964 if (path == NULL || !PyList_Check(path)) {
965 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +0000966 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000967 return NULL;
968 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000969 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000970 namelen = strlen(name);
971 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000972 PyObject *v = PyList_GetItem(path, i);
973 if (!PyString_Check(v))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000974 continue;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000975 len = PyString_Size(v);
Guido van Rossumef3d02e1997-07-21 14:54:36 +0000976 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000977 continue; /* Too long */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000978 strcpy(buf, PyString_AsString(v));
Fred Drake4c82b232000-06-30 16:18:57 +0000979 if (strlen(buf) != len)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000980 continue; /* v contains '\0' */
Jack Jansen9c96a921995-02-15 22:57:06 +0000981#ifdef macintosh
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);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000990 if (PyMac_FindResourceModule((PyStringObject *)v, name, buf)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000991 static struct filedescr resfiledescr =
992 {"", "", PY_RESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +0000993
Jack Jansen9c96a921995-02-15 22:57:06 +0000994 return &resfiledescr;
995 }
Guido van Rossum0f84a341998-08-06 13:36:01 +0000996 if (PyMac_FindCodeResourceModule((PyStringObject *)v, name, buf)) {
997 static struct filedescr resfiledescr =
998 {"", "", PY_CODERESOURCE};
Tim Peters50d8d372001-02-28 05:34:27 +0000999
Guido van Rossum0f84a341998-08-06 13:36:01 +00001000 return &resfiledescr;
1001 }
Jack Jansen9c96a921995-02-15 22:57:06 +00001002#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001003 if (len > 0 && buf[len-1] != SEP
1004#ifdef ALTSEP
1005 && buf[len-1] != ALTSEP
1006#endif
1007 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001008 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001009 strcpy(buf+len, name);
1010 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001011
1012 /* Check for package import (buf holds a directory name,
1013 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001014#ifdef HAVE_STAT
Tim Peterscab3f682001-04-29 22:21:25 +00001015 if (stat(buf, &statbuf) == 0 && /* it exists */
1016 S_ISDIR(statbuf.st_mode) && /* it's a directory */
1017 find_init_module(buf) && /* it has __init__.py */
1018 case_ok(buf, len, namelen, name)) /* and case matches */
1019 return &fd_package;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001020#else
1021 /* XXX How are you going to test for directories? */
Guido van Rossum48a680c2001-03-02 06:34:14 +00001022#ifdef RISCOS
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001023 if (isdir(buf) &&
1024 find_init_module(buf) &&
1025 case_ok(buf, len, namelen, name))
1026 return &fd_package;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001027#endif
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001028#endif
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001029#ifdef macintosh
1030 fdp = PyMac_FindModuleExtension(buf, &len, name);
Jack Jansen4df3c522001-03-20 23:09:54 +00001031 if (fdp) {
Guido van Rossum9a61dc91997-10-08 15:25:08 +00001032#else
Andrew MacIntyred9400542002-02-26 11:41:34 +00001033#if defined(PYOS_OS2)
1034 /* take a snapshot of the module spec for restoration
1035 * after the 8 character DLL hackery
1036 */
1037 saved_buf = strdup(buf);
1038 saved_len = len;
1039 saved_namelen = namelen;
1040#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001041 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001042#if defined(PYOS_OS2)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001043 /* OS/2 limits DLLs to 8 character names (w/o
1044 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001045 * 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? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001051 const struct filedescr *scan;
1052 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001053 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001054 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001055 break;
1056 else
1057 scan++;
1058 }
1059 if (scan->suffix != NULL) {
1060 /* yes, so truncate the name */
1061 namelen = 8;
1062 len -= strlen(realname) - namelen;
1063 buf[len] = '\0';
1064 }
1065 }
1066#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001067 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001068 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001069 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansen4df3c522001-03-20 23:09:54 +00001070#endif /* !macintosh */
Jack Jansenc88da1f2002-05-28 10:58:19 +00001071 filemode = fdp->mode;
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001072 if (filemode[0] == 'U')
1073 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001074 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001075 if (fp != NULL) {
1076 if (case_ok(buf, len, namelen, name))
1077 break;
1078 else { /* continue search */
1079 fclose(fp);
1080 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001081 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001082 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001083#if defined(PYOS_OS2)
1084 /* restore the saved snapshot */
1085 strcpy(buf, saved_buf);
1086 len = saved_len;
1087 namelen = saved_namelen;
1088#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001089 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001090#if defined(PYOS_OS2)
1091 /* don't need/want the module name snapshot anymore */
1092 if (saved_buf)
1093 {
1094 free(saved_buf);
1095 saved_buf = NULL;
1096 }
1097#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001098 if (fp != NULL)
1099 break;
1100 }
1101 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001102 PyErr_Format(PyExc_ImportError,
1103 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001104 return NULL;
1105 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001106 *p_fp = fp;
1107 return fdp;
1108}
1109
Tim Petersd1e87a82001-03-01 18:12:00 +00001110/* case_ok(char* buf, int len, int namelen, char* name)
1111 * The arguments here are tricky, best shown by example:
1112 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1113 * ^ ^ ^ ^
1114 * |--------------------- buf ---------------------|
1115 * |------------------- len ------------------|
1116 * |------ name -------|
1117 * |----- namelen -----|
1118 * buf is the full path, but len only counts up to (& exclusive of) the
1119 * extension. name is the module name, also exclusive of extension.
1120 *
1121 * We've already done a successful stat() or fopen() on buf, so know that
1122 * there's some match, possibly case-insensitive.
1123 *
Tim Peters50d8d372001-02-28 05:34:27 +00001124 * case_ok() is to return 1 if there's a case-sensitive match for
1125 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1126 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001127 *
Tim Peters50d8d372001-02-28 05:34:27 +00001128 * case_ok() is used to implement case-sensitive import semantics even
1129 * on platforms with case-insensitive filesystems. It's trivial to implement
1130 * for case-sensitive filesystems. It's pretty much a cross-platform
1131 * nightmare for systems with case-insensitive filesystems.
1132 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001133
Tim Peters50d8d372001-02-28 05:34:27 +00001134/* First we may need a pile of platform-specific header files; the sequence
1135 * of #if's here should match the sequence in the body of case_ok().
1136 */
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001137#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001138#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001139#ifdef __CYGWIN__
1140#include <sys/cygwin.h>
1141#endif
1142
Tim Peters50d8d372001-02-28 05:34:27 +00001143#elif defined(DJGPP)
1144#include <dir.h>
1145
1146#elif defined(macintosh)
1147#include <TextUtils.h>
Tim Peters50d8d372001-02-28 05:34:27 +00001148
Tim Petersd1e87a82001-03-01 18:12:00 +00001149#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001150#include <sys/types.h>
1151#include <dirent.h>
1152
Andrew MacIntyred9400542002-02-26 11:41:34 +00001153#elif defined(PYOS_OS2)
1154#define INCL_DOS
1155#define INCL_DOSERRORS
1156#define INCL_NOPMAPI
1157#include <os2.h>
1158
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001159#elif defined(RISCOS)
1160#include "oslib/osfscontrol.h"
Tim Peters50d8d372001-02-28 05:34:27 +00001161#endif
1162
Guido van Rossum0980bd91998-02-13 17:18:36 +00001163static int
Tim Peters50d8d372001-02-28 05:34:27 +00001164case_ok(char *buf, int len, int namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001165{
Tim Peters50d8d372001-02-28 05:34:27 +00001166/* Pick a platform-specific implementation; the sequence of #if's here should
1167 * match the sequence just above.
1168 */
1169
1170/* MS_WIN32 || __CYGWIN__ */
1171#if defined(MS_WIN32) || defined(__CYGWIN__)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001172 WIN32_FIND_DATA data;
1173 HANDLE h;
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001174#ifdef __CYGWIN__
1175 char tempbuf[MAX_PATH];
1176#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001177
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001178 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001179 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001180
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001181#ifdef __CYGWIN__
1182 cygwin32_conv_to_win32_path(buf, tempbuf);
1183 h = FindFirstFile(tempbuf, &data);
1184#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001185 h = FindFirstFile(buf, &data);
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001186#endif
Guido van Rossum0980bd91998-02-13 17:18:36 +00001187 if (h == INVALID_HANDLE_VALUE) {
1188 PyErr_Format(PyExc_NameError,
1189 "Can't find file for module %.100s\n(filename %.300s)",
1190 name, buf);
1191 return 0;
1192 }
1193 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001194 return strncmp(data.cFileName, name, namelen) == 0;
1195
1196/* DJGPP */
1197#elif defined(DJGPP)
1198 struct ffblk ffblk;
1199 int done;
1200
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001201 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001202 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001203
1204 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1205 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001206 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001207 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001208 name, buf);
1209 return 0;
1210 }
Tim Peters50d8d372001-02-28 05:34:27 +00001211 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001212
Tim Peters50d8d372001-02-28 05:34:27 +00001213/* macintosh */
1214#elif defined(macintosh)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001215 FSSpec fss;
1216 OSErr err;
Tim Peters50d8d372001-02-28 05:34:27 +00001217
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001218 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Peters50d8d372001-02-28 05:34:27 +00001219 return 1;
1220
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001221 err = FSMakeFSSpec(0, 0, Pstring(buf), &fss);
Guido van Rossum0980bd91998-02-13 17:18:36 +00001222 if (err) {
1223 PyErr_Format(PyExc_NameError,
Guido van Rossuma0f0a331998-09-14 13:40:53 +00001224 "Can't find file for module %.100s\n(filename %.300s)",
1225 name, buf);
Guido van Rossum0980bd91998-02-13 17:18:36 +00001226 return 0;
1227 }
Tim Peters50d8d372001-02-28 05:34:27 +00001228 return fss.name[0] >= namelen &&
1229 strncmp(name, (char *)fss.name+1, namelen) == 0;
1230
Tim Peters677898a2001-03-02 03:28:03 +00001231/* new-fangled macintosh (macosx) */
Tim Petersd1e87a82001-03-01 18:12:00 +00001232#elif defined(__MACH__) && defined(__APPLE__) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001233 DIR *dirp;
1234 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001235 char dirname[MAXPATHLEN + 1];
1236 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001237
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001238 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001239 return 1;
1240
Tim Petersd1e87a82001-03-01 18:12:00 +00001241 /* Copy the dir component into dirname; substitute "." if empty */
1242 if (dirlen <= 0) {
1243 dirname[0] = '.';
1244 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001245 }
1246 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001247 assert(dirlen <= MAXPATHLEN);
1248 memcpy(dirname, buf, dirlen);
1249 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001250 }
1251 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001252 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001253 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001254 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001255 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001256 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001257#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001258 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001259#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001260 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001261#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001262 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001263 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001264 (void)closedir(dirp);
1265 return 1; /* Found */
1266 }
1267 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001268 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001269 }
Tim Peters430f5d42001-03-01 01:30:56 +00001270 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001271
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001272/* RISC OS */
1273#elif defined(RISCOS)
1274 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */
1275 char buf2[MAXPATHLEN+2];
1276 char *nameWithExt = buf+len-namelen;
1277 int canonlen;
1278 os_error *e;
1279
1280 if (Py_GETENV("PYTHONCASEOK") != NULL)
1281 return 1;
1282
1283 /* workaround:
1284 append wildcard, otherwise case of filename wouldn't be touched */
1285 strcpy(buf2, buf);
1286 strcat(buf2, "*");
1287
1288 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen);
1289 canonlen = MAXPATHLEN+1-canonlen;
1290 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) )
1291 return 0;
1292 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0)
1293 return 1; /* match */
1294
1295 return 0;
1296
Andrew MacIntyred9400542002-02-26 11:41:34 +00001297/* OS/2 */
1298#elif defined(PYOS_OS2)
1299 HDIR hdir = 1;
1300 ULONG srchcnt = 1;
1301 FILEFINDBUF3 ffbuf;
1302 APIRET rc;
1303
1304 if (getenv("PYTHONCASEOK") != NULL)
1305 return 1;
1306
1307 rc = DosFindFirst(buf,
1308 &hdir,
1309 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1310 &ffbuf, sizeof(ffbuf),
1311 &srchcnt,
1312 FIL_STANDARD);
1313 if (rc != NO_ERROR)
1314 return 0;
1315 return strncmp(ffbuf.achName, name, namelen) == 0;
1316
Tim Peters50d8d372001-02-28 05:34:27 +00001317/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1318#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001319 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001320
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001321#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001322}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001323
Guido van Rossum0980bd91998-02-13 17:18:36 +00001324
Guido van Rossum197346f1997-10-31 18:38:52 +00001325#ifdef HAVE_STAT
1326/* Helper to look for __init__.py or __init__.py[co] in potential package */
1327static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001328find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001329{
Tim Peters0f9431f2001-07-05 03:47:53 +00001330 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001331 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001332 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001333 struct stat statbuf;
1334
Tim Peters0f9431f2001-07-05 03:47:53 +00001335/* For calling case_ok(buf, len, namelen, name):
1336 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1337 * ^ ^ ^ ^
1338 * |--------------------- buf ---------------------|
1339 * |------------------- len ------------------|
1340 * |------ name -------|
1341 * |----- namelen -----|
1342 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001343 if (save_len + 13 >= MAXPATHLEN)
1344 return 0;
1345 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001346 pname = buf + i;
1347 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001348 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001349 if (case_ok(buf,
1350 save_len + 9, /* len("/__init__") */
1351 8, /* len("__init__") */
1352 pname)) {
1353 buf[save_len] = '\0';
1354 return 1;
1355 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001356 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001357 i += strlen(pname);
1358 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001359 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001360 if (case_ok(buf,
1361 save_len + 9, /* len("/__init__") */
1362 8, /* len("__init__") */
1363 pname)) {
1364 buf[save_len] = '\0';
1365 return 1;
1366 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001367 }
1368 buf[save_len] = '\0';
1369 return 0;
1370}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001371
1372#else
1373
1374#ifdef RISCOS
1375static int
1376find_init_module(buf)
1377 char *buf;
1378{
1379 int save_len = strlen(buf);
1380 int i = save_len;
1381
1382 if (save_len + 13 >= MAXPATHLEN)
1383 return 0;
1384 buf[i++] = SEP;
1385 strcpy(buf+i, "__init__/py");
1386 if (isfile(buf)) {
1387 buf[save_len] = '\0';
1388 return 1;
1389 }
1390
1391 if (Py_OptimizeFlag)
1392 strcpy(buf+i, "o");
1393 else
1394 strcpy(buf+i, "c");
1395 if (isfile(buf)) {
1396 buf[save_len] = '\0';
1397 return 1;
1398 }
1399 buf[save_len] = '\0';
1400 return 0;
1401}
1402#endif /*RISCOS*/
1403
Guido van Rossum197346f1997-10-31 18:38:52 +00001404#endif /* HAVE_STAT */
1405
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001406
Tim Petersdbd9ba62000-07-09 03:09:57 +00001407static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001408
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001409/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001410 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001411
Guido van Rossum79f25d91997-04-29 20:08:16 +00001412static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001413load_module(char *name, FILE *fp, char *buf, int type)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001414{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001415 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001416 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001417 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001418
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001419 /* First check that there's an open file (if we need one) */
1420 switch (type) {
1421 case PY_SOURCE:
1422 case PY_COMPILED:
1423 if (fp == NULL) {
1424 PyErr_Format(PyExc_ValueError,
1425 "file object required for import (type code %d)",
1426 type);
1427 return NULL;
1428 }
1429 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001430
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001431 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001432
1433 case PY_SOURCE:
1434 m = load_source_module(name, buf, fp);
1435 break;
1436
1437 case PY_COMPILED:
1438 m = load_compiled_module(name, buf, fp);
1439 break;
1440
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001441#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001442 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001443 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001444 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001445#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001446
Jack Jansen9c96a921995-02-15 22:57:06 +00001447#ifdef macintosh
1448 case PY_RESOURCE:
1449 m = PyMac_LoadResourceModule(name, buf);
1450 break;
Guido van Rossum0f84a341998-08-06 13:36:01 +00001451 case PY_CODERESOURCE:
1452 m = PyMac_LoadCodeResourceModule(name, buf);
1453 break;
Jack Jansen9c96a921995-02-15 22:57:06 +00001454#endif
1455
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001456 case PKG_DIRECTORY:
1457 m = load_package(name, buf);
1458 break;
1459
1460 case C_BUILTIN:
1461 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001462 if (buf != NULL && buf[0] != '\0')
1463 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001464 if (type == C_BUILTIN)
1465 err = init_builtin(name);
1466 else
1467 err = PyImport_ImportFrozenModule(name);
1468 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001469 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001470 if (err == 0) {
1471 PyErr_Format(PyExc_ImportError,
1472 "Purported %s module %.200s not found",
1473 type == C_BUILTIN ?
1474 "builtin" : "frozen",
1475 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001476 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001477 }
1478 modules = PyImport_GetModuleDict();
1479 m = PyDict_GetItemString(modules, name);
1480 if (m == NULL) {
1481 PyErr_Format(
1482 PyExc_ImportError,
1483 "%s module %.200s not properly initialized",
1484 type == C_BUILTIN ?
1485 "builtin" : "frozen",
1486 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001487 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001488 }
1489 Py_INCREF(m);
1490 break;
1491
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001492 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001493 PyErr_Format(PyExc_ImportError,
1494 "Don't know how to import %.200s (type code %d)",
1495 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001496 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001497
1498 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001499
1500 return m;
1501}
1502
1503
1504/* Initialize a built-in module.
1505 Return 1 for succes, 0 if the module is not found, and -1 with
1506 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001507
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001508static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001509init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001510{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001511 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001512
Greg Ward201baee2001-10-04 14:52:06 +00001513 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001514 return 1;
1515
Guido van Rossum771c6c81997-10-31 18:37:24 +00001516 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001517 if (strcmp(name, p->name) == 0) {
1518 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001519 PyErr_Format(PyExc_ImportError,
1520 "Cannot re-init internal module %.200s",
1521 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001522 return -1;
1523 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001524 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001525 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001526 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001527 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001528 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001529 if (_PyImport_FixupExtension(name, name) == NULL)
1530 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001531 return 1;
1532 }
1533 }
1534 return 0;
1535}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001536
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001537
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001538/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001539
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001540static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001541find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001542{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001543 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001544
Guido van Rossum79f25d91997-04-29 20:08:16 +00001545 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001546 if (p->name == NULL)
1547 return NULL;
1548 if (strcmp(p->name, name) == 0)
1549 break;
1550 }
1551 return p;
1552}
1553
Guido van Rossum79f25d91997-04-29 20:08:16 +00001554static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001555get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001556{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001557 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001558 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001559
1560 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001561 PyErr_Format(PyExc_ImportError,
1562 "No such frozen object named %.200s",
1563 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001564 return NULL;
1565 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001566 if (p->code == NULL) {
1567 PyErr_Format(PyExc_ImportError,
1568 "Excluded frozen object named %.200s",
1569 name);
1570 return NULL;
1571 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001572 size = p->size;
1573 if (size < 0)
1574 size = -size;
1575 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001576}
1577
1578/* Initialize a frozen module.
1579 Return 1 for succes, 0 if the module is not found, and -1 with
1580 an exception set if the initialization failed.
1581 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001582
1583int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001584PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001585{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001586 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001587 PyObject *co;
1588 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001589 int ispackage;
1590 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001591
1592 if (p == NULL)
1593 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001594 if (p->code == NULL) {
1595 PyErr_Format(PyExc_ImportError,
1596 "Excluded frozen object named %.200s",
1597 name);
1598 return -1;
1599 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001600 size = p->size;
1601 ispackage = (size < 0);
1602 if (ispackage)
1603 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001604 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001605 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001606 name, ispackage ? " package" : "");
1607 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001608 if (co == NULL)
1609 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001610 if (!PyCode_Check(co)) {
1611 Py_DECREF(co);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001612 PyErr_Format(PyExc_TypeError,
1613 "frozen object %.200s is not a code object",
1614 name);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001615 return -1;
1616 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001617 if (ispackage) {
1618 /* Set __path__ to the package name */
1619 PyObject *d, *s;
1620 int err;
1621 m = PyImport_AddModule(name);
1622 if (m == NULL)
1623 return -1;
1624 d = PyModule_GetDict(m);
1625 s = PyString_InternFromString(name);
1626 if (s == NULL)
1627 return -1;
1628 err = PyDict_SetItemString(d, "__path__", s);
1629 Py_DECREF(s);
1630 if (err != 0)
1631 return err;
1632 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001633 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum79f25d91997-04-29 20:08:16 +00001634 Py_DECREF(co);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001635 if (m == NULL)
1636 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001637 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001638 return 1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001639}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001640
1641
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001642/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001643 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001644
Guido van Rossum79f25d91997-04-29 20:08:16 +00001645PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001646PyImport_ImportModule(char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001647{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001648 PyObject *pname;
1649 PyObject *result;
1650
1651 pname = PyString_FromString(name);
1652 result = PyImport_Import(pname);
1653 Py_DECREF(pname);
1654 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001655}
1656
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001657/* Forward declarations for helper routines */
Tim Petersdbd9ba62000-07-09 03:09:57 +00001658static PyObject *get_parent(PyObject *globals, char *buf, int *p_buflen);
1659static PyObject *load_next(PyObject *mod, PyObject *altmod,
1660 char **p_name, char *buf, int *p_buflen);
1661static int mark_miss(char *name);
1662static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
1663 char *buf, int buflen, int recursive);
1664static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001665
1666/* The Magnum Opus of dotted-name import :-) */
1667
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001668static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001669import_module_ex(char *name, PyObject *globals, PyObject *locals,
1670 PyObject *fromlist)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001671{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001672 char buf[MAXPATHLEN+1];
1673 int buflen = 0;
1674 PyObject *parent, *head, *next, *tail;
1675
1676 parent = get_parent(globals, buf, &buflen);
1677 if (parent == NULL)
1678 return NULL;
1679
1680 head = load_next(parent, Py_None, &name, buf, &buflen);
1681 if (head == NULL)
1682 return NULL;
1683
1684 tail = head;
1685 Py_INCREF(tail);
1686 while (name) {
1687 next = load_next(tail, tail, &name, buf, &buflen);
1688 Py_DECREF(tail);
1689 if (next == NULL) {
1690 Py_DECREF(head);
1691 return NULL;
1692 }
1693 tail = next;
1694 }
1695
1696 if (fromlist != NULL) {
1697 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
1698 fromlist = NULL;
1699 }
1700
1701 if (fromlist == NULL) {
1702 Py_DECREF(tail);
1703 return head;
1704 }
1705
1706 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001707 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001708 Py_DECREF(tail);
1709 return NULL;
1710 }
1711
1712 return tail;
1713}
1714
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001715PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001716PyImport_ImportModuleEx(char *name, PyObject *globals, PyObject *locals,
1717 PyObject *fromlist)
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001718{
1719 PyObject *result;
1720 lock_import();
Guido van Rossumd65911b1998-03-03 22:33:27 +00001721 result = import_module_ex(name, globals, locals, fromlist);
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001722 unlock_import();
1723 return result;
1724}
1725
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001726static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001727get_parent(PyObject *globals, char *buf, int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001728{
1729 static PyObject *namestr = NULL;
1730 static PyObject *pathstr = NULL;
1731 PyObject *modname, *modpath, *modules, *parent;
1732
1733 if (globals == NULL || !PyDict_Check(globals))
1734 return Py_None;
1735
1736 if (namestr == NULL) {
1737 namestr = PyString_InternFromString("__name__");
1738 if (namestr == NULL)
1739 return NULL;
1740 }
1741 if (pathstr == NULL) {
1742 pathstr = PyString_InternFromString("__path__");
1743 if (pathstr == NULL)
1744 return NULL;
1745 }
1746
1747 *buf = '\0';
1748 *p_buflen = 0;
1749 modname = PyDict_GetItem(globals, namestr);
1750 if (modname == NULL || !PyString_Check(modname))
1751 return Py_None;
1752
1753 modpath = PyDict_GetItem(globals, pathstr);
1754 if (modpath != NULL) {
1755 int len = PyString_GET_SIZE(modname);
1756 if (len > MAXPATHLEN) {
1757 PyErr_SetString(PyExc_ValueError,
1758 "Module name too long");
1759 return NULL;
1760 }
1761 strcpy(buf, PyString_AS_STRING(modname));
1762 *p_buflen = len;
1763 }
1764 else {
1765 char *start = PyString_AS_STRING(modname);
1766 char *lastdot = strrchr(start, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001767 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001768 if (lastdot == NULL)
1769 return Py_None;
1770 len = lastdot - start;
1771 if (len >= MAXPATHLEN) {
1772 PyErr_SetString(PyExc_ValueError,
1773 "Module name too long");
1774 return NULL;
1775 }
1776 strncpy(buf, start, len);
1777 buf[len] = '\0';
1778 *p_buflen = len;
1779 }
1780
1781 modules = PyImport_GetModuleDict();
1782 parent = PyDict_GetItemString(modules, buf);
1783 if (parent == NULL)
1784 parent = Py_None;
1785 return parent;
1786 /* We expect, but can't guarantee, if parent != None, that:
1787 - parent.__name__ == buf
1788 - parent.__dict__ is globals
1789 If this is violated... Who cares? */
1790}
1791
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001792/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001793static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001794load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
1795 int *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001796{
1797 char *name = *p_name;
1798 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00001799 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001800 char *p;
1801 PyObject *result;
1802
1803 if (dot == NULL) {
1804 *p_name = NULL;
1805 len = strlen(name);
1806 }
1807 else {
1808 *p_name = dot+1;
1809 len = dot-name;
1810 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00001811 if (len == 0) {
1812 PyErr_SetString(PyExc_ValueError,
1813 "Empty module name");
1814 return NULL;
1815 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001816
1817 p = buf + *p_buflen;
1818 if (p != buf)
1819 *p++ = '.';
1820 if (p+len-buf >= MAXPATHLEN) {
1821 PyErr_SetString(PyExc_ValueError,
1822 "Module name too long");
1823 return NULL;
1824 }
1825 strncpy(p, name, len);
1826 p[len] = '\0';
1827 *p_buflen = p+len-buf;
1828
1829 result = import_submodule(mod, p, buf);
1830 if (result == Py_None && altmod != mod) {
1831 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001832 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00001833 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001834 if (result != NULL && result != Py_None) {
1835 if (mark_miss(buf) != 0) {
1836 Py_DECREF(result);
1837 return NULL;
1838 }
1839 strncpy(buf, name, len);
1840 buf[len] = '\0';
1841 *p_buflen = len;
1842 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001843 }
1844 if (result == NULL)
1845 return NULL;
1846
1847 if (result == Py_None) {
1848 Py_DECREF(result);
1849 PyErr_Format(PyExc_ImportError,
1850 "No module named %.200s", name);
1851 return NULL;
1852 }
1853
1854 return result;
1855}
1856
1857static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001858mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00001859{
1860 PyObject *modules = PyImport_GetModuleDict();
1861 return PyDict_SetItemString(modules, name, Py_None);
1862}
1863
1864static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001865ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, int buflen,
1866 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001867{
1868 int i;
1869
1870 if (!PyObject_HasAttrString(mod, "__path__"))
1871 return 1;
1872
1873 for (i = 0; ; i++) {
1874 PyObject *item = PySequence_GetItem(fromlist, i);
1875 int hasit;
1876 if (item == NULL) {
1877 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
1878 PyErr_Clear();
1879 return 1;
1880 }
1881 return 0;
1882 }
1883 if (!PyString_Check(item)) {
1884 PyErr_SetString(PyExc_TypeError,
1885 "Item in ``from list'' not a string");
1886 Py_DECREF(item);
1887 return 0;
1888 }
1889 if (PyString_AS_STRING(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00001890 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001891 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00001892 /* See if the package defines __all__ */
1893 if (recursive)
1894 continue; /* Avoid endless recursion */
1895 all = PyObject_GetAttrString(mod, "__all__");
1896 if (all == NULL)
1897 PyErr_Clear();
1898 else {
1899 if (!ensure_fromlist(mod, all, buf, buflen, 1))
1900 return 0;
1901 Py_DECREF(all);
1902 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001903 continue;
1904 }
1905 hasit = PyObject_HasAttr(mod, item);
1906 if (!hasit) {
1907 char *subname = PyString_AS_STRING(item);
1908 PyObject *submod;
1909 char *p;
1910 if (buflen + strlen(subname) >= MAXPATHLEN) {
1911 PyErr_SetString(PyExc_ValueError,
1912 "Module name too long");
1913 Py_DECREF(item);
1914 return 0;
1915 }
1916 p = buf + buflen;
1917 *p++ = '.';
1918 strcpy(p, subname);
1919 submod = import_submodule(mod, subname, buf);
1920 Py_XDECREF(submod);
1921 if (submod == NULL) {
1922 Py_DECREF(item);
1923 return 0;
1924 }
1925 }
1926 Py_DECREF(item);
1927 }
1928
Guido van Rossuma7f2e811997-10-03 15:33:32 +00001929 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001930}
1931
1932static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001933import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001934{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001935 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001936 PyObject *m, *res = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00001937
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001938 /* Require:
1939 if mod == None: subname == fullname
1940 else: mod.__name__ + "." + subname == fullname
1941 */
1942
Tim Peters50d8d372001-02-28 05:34:27 +00001943 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001944 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001945 }
1946 else {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001947 PyObject *path;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001948 char buf[MAXPATHLEN+1];
1949 struct filedescr *fdp;
1950 FILE *fp = NULL;
1951
Guido van Rossum9c0afe51998-05-19 15:09:05 +00001952 if (mod == Py_None)
1953 path = NULL;
1954 else {
1955 path = PyObject_GetAttrString(mod, "__path__");
1956 if (path == NULL) {
1957 PyErr_Clear();
1958 Py_INCREF(Py_None);
1959 return Py_None;
1960 }
1961 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001962
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001963 buf[0] = '\0';
Guido van Rossumb68cd421998-07-01 17:36:26 +00001964 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
1965 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001966 if (fdp == NULL) {
1967 if (!PyErr_ExceptionMatches(PyExc_ImportError))
1968 return NULL;
1969 PyErr_Clear();
1970 Py_INCREF(Py_None);
1971 return Py_None;
1972 }
1973 m = load_module(fullname, fp, buf, fdp->type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001974 if (fp)
1975 fclose(fp);
Guido van Rossumf48f11c2001-07-23 13:27:49 +00001976 if (mod != Py_None) {
1977 /* Irrespective of the success of this load, make a
1978 reference to it in the parent package module.
1979 A copy gets saved in the modules dictionary
1980 under the full name, so get a reference from
1981 there, if need be. (The exception is when
1982 the load failed with a SyntaxError -- then
1983 there's no trace in sys.modules. In that case,
1984 of course, do nothing extra.) */
1985 res = m;
1986 if (res == NULL)
1987 res = PyDict_GetItemString(modules, fullname);
1988 if (res != NULL &&
1989 PyObject_SetAttrString(mod, subname, res) < 0) {
1990 Py_XDECREF(m);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001991 m = NULL;
1992 }
1993 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00001994 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001995
1996 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00001997}
1998
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001999
2000/* Re-import a module of any kind and return its module object, WITH
2001 INCREMENTED REFERENCE COUNT */
2002
Guido van Rossum79f25d91997-04-29 20:08:16 +00002003PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002004PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002005{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002006 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum222ef561997-09-06 19:41:09 +00002007 PyObject *path = NULL;
2008 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002009 char buf[MAXPATHLEN+1];
2010 struct filedescr *fdp;
2011 FILE *fp = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002012
Guido van Rossum79f25d91997-04-29 20:08:16 +00002013 if (m == NULL || !PyModule_Check(m)) {
2014 PyErr_SetString(PyExc_TypeError,
2015 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002016 return NULL;
2017 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002018 name = PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002019 if (name == NULL)
2020 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002021 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002022 PyErr_Format(PyExc_ImportError,
2023 "reload(): module %.200s not in sys.modules",
2024 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002025 return NULL;
2026 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002027 subname = strrchr(name, '.');
2028 if (subname == NULL)
2029 subname = name;
2030 else {
2031 PyObject *parentname, *parent;
2032 parentname = PyString_FromStringAndSize(name, (subname-name));
2033 if (parentname == NULL)
2034 return NULL;
2035 parent = PyDict_GetItem(modules, parentname);
Barry Warsaw38793331999-01-27 17:54:20 +00002036 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002037 if (parent == NULL) {
2038 PyErr_Format(PyExc_ImportError,
2039 "reload(): parent %.200s not in sys.modules",
2040 name);
2041 return NULL;
2042 }
2043 subname++;
2044 path = PyObject_GetAttrString(parent, "__path__");
2045 if (path == NULL)
2046 PyErr_Clear();
2047 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002048 buf[0] = '\0';
Guido van Rossum222ef561997-09-06 19:41:09 +00002049 fdp = find_module(subname, path, buf, MAXPATHLEN+1, &fp);
2050 Py_XDECREF(path);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002051 if (fdp == NULL)
2052 return NULL;
2053 m = load_module(name, fp, buf, fdp->type);
2054 if (fp)
2055 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002056 return m;
2057}
2058
2059
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002060/* Higher-level import emulator which emulates the "import" statement
2061 more accurately -- it invokes the __import__() function from the
2062 builtins of the current globals. This means that the import is
2063 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002064 environment, e.g. by "rexec".
2065 A dummy list ["__doc__"] is passed as the 4th argument so that
2066 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache"))
2067 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002068
2069PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002070PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002071{
2072 static PyObject *silly_list = NULL;
2073 static PyObject *builtins_str = NULL;
2074 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002075 PyObject *globals = NULL;
2076 PyObject *import = NULL;
2077 PyObject *builtins = NULL;
2078 PyObject *r = NULL;
2079
2080 /* Initialize constant string objects */
2081 if (silly_list == NULL) {
2082 import_str = PyString_InternFromString("__import__");
2083 if (import_str == NULL)
2084 return NULL;
2085 builtins_str = PyString_InternFromString("__builtins__");
2086 if (builtins_str == NULL)
2087 return NULL;
2088 silly_list = Py_BuildValue("[s]", "__doc__");
2089 if (silly_list == NULL)
2090 return NULL;
2091 }
2092
2093 /* Get the builtins from current globals */
2094 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002095 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002096 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002097 builtins = PyObject_GetItem(globals, builtins_str);
2098 if (builtins == NULL)
2099 goto err;
2100 }
2101 else {
2102 /* No globals -- use standard builtins, and fake globals */
2103 PyErr_Clear();
2104
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002105 builtins = PyImport_ImportModuleEx("__builtin__",
2106 NULL, NULL, NULL);
2107 if (builtins == NULL)
2108 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002109 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2110 if (globals == NULL)
2111 goto err;
2112 }
2113
2114 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002115 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002116 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002117 if (import == NULL)
2118 PyErr_SetObject(PyExc_KeyError, import_str);
2119 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002120 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002121 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002122 if (import == NULL)
2123 goto err;
2124
2125 /* Call the _import__ function with the proper argument list */
2126 r = PyObject_CallFunction(import, "OOOO",
2127 module_name, globals, globals, silly_list);
2128
2129 err:
2130 Py_XDECREF(globals);
2131 Py_XDECREF(builtins);
2132 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002133
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002134 return r;
2135}
2136
2137
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002138/* Module 'imp' provides Python access to the primitives used for
2139 importing modules.
2140*/
2141
Guido van Rossum79f25d91997-04-29 20:08:16 +00002142static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002143imp_get_magic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002144{
2145 char buf[4];
2146
Guido van Rossum43713e52000-02-29 13:59:29 +00002147 if (!PyArg_ParseTuple(args, ":get_magic"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002148 return NULL;
Guido van Rossum96774c12000-05-01 20:19:08 +00002149 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2150 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2151 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2152 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002153
Guido van Rossum79f25d91997-04-29 20:08:16 +00002154 return PyString_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002155}
2156
Guido van Rossum79f25d91997-04-29 20:08:16 +00002157static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002158imp_get_suffixes(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002159{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002160 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002161 struct filedescr *fdp;
2162
Guido van Rossum43713e52000-02-29 13:59:29 +00002163 if (!PyArg_ParseTuple(args, ":get_suffixes"))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002164 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002165 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002166 if (list == NULL)
2167 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002168 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2169 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002170 fdp->suffix, fdp->mode, fdp->type);
2171 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002172 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002173 return NULL;
2174 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002175 if (PyList_Append(list, item) < 0) {
2176 Py_DECREF(list);
2177 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002178 return NULL;
2179 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002180 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002181 }
2182 return list;
2183}
2184
Guido van Rossum79f25d91997-04-29 20:08:16 +00002185static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002186call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002187{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002188 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002189 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002190 struct filedescr *fdp;
2191 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002192 FILE *fp = NULL;
2193
2194 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002195 if (path == Py_None)
2196 path = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002197 fdp = find_module(name, path, pathname, MAXPATHLEN+1, &fp);
2198 if (fdp == NULL)
2199 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002200 if (fp != NULL) {
2201 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose);
2202 if (fob == NULL) {
2203 fclose(fp);
2204 return NULL;
2205 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002206 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002207 else {
2208 fob = Py_None;
2209 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002210 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002211 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002212 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002213 Py_DECREF(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002214 return ret;
2215}
2216
Guido van Rossum79f25d91997-04-29 20:08:16 +00002217static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002218imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002219{
2220 char *name;
2221 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002222 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002223 return NULL;
2224 return call_find_module(name, path);
2225}
2226
2227static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002228imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002229{
2230 char *name;
2231 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002232 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002233 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002234 return NULL;
2235 ret = init_builtin(name);
2236 if (ret < 0)
2237 return NULL;
2238 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002239 Py_INCREF(Py_None);
2240 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002241 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002242 m = PyImport_AddModule(name);
2243 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002244 return m;
2245}
2246
Guido van Rossum79f25d91997-04-29 20:08:16 +00002247static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002248imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002249{
2250 char *name;
2251 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002252 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002253 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002254 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002255 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002256 if (ret < 0)
2257 return NULL;
2258 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002259 Py_INCREF(Py_None);
2260 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002261 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002262 m = PyImport_AddModule(name);
2263 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002264 return m;
2265}
2266
Guido van Rossum79f25d91997-04-29 20:08:16 +00002267static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002268imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002269{
2270 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002271
Guido van Rossum43713e52000-02-29 13:59:29 +00002272 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002273 return NULL;
2274 return get_frozen_object(name);
2275}
2276
Guido van Rossum79f25d91997-04-29 20:08:16 +00002277static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002278imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002279{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002280 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002281 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002282 return NULL;
Guido van Rossum50ee94f2002-04-09 18:00:58 +00002283 return PyInt_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002284}
2285
Guido van Rossum79f25d91997-04-29 20:08:16 +00002286static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002287imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002288{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002289 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002290 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002291 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002292 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002293 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002294 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002295}
2296
2297static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002298get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002299{
2300 FILE *fp;
2301 if (fob == NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002302 if (mode[0] == 'U')
2303 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002304 fp = fopen(pathname, mode);
2305 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002306 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002307 }
2308 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002309 fp = PyFile_AsFile(fob);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002310 if (fp == NULL)
Guido van Rossum79f25d91997-04-29 20:08:16 +00002311 PyErr_SetString(PyExc_ValueError,
2312 "bad/closed file object");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002313 }
2314 return fp;
2315}
2316
Guido van Rossum79f25d91997-04-29 20:08:16 +00002317static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002318imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002319{
2320 char *name;
2321 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002322 PyObject *fob = NULL;
2323 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002324 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002325 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002326 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002327 return NULL;
2328 fp = get_file(pathname, fob, "rb");
2329 if (fp == NULL)
2330 return NULL;
2331 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002332 if (fob == NULL)
2333 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002334 return m;
2335}
2336
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002337#ifdef HAVE_DYNAMIC_LOADING
2338
Guido van Rossum79f25d91997-04-29 20:08:16 +00002339static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002340imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002341{
2342 char *name;
2343 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002344 PyObject *fob = NULL;
2345 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002346 FILE *fp = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002347 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002348 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002349 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002350 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002351 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002352 if (fp == NULL)
2353 return NULL;
2354 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002355 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002356 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002357}
2358
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002359#endif /* HAVE_DYNAMIC_LOADING */
2360
Guido van Rossum79f25d91997-04-29 20:08:16 +00002361static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002362imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002363{
2364 char *name;
2365 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002366 PyObject *fob = NULL;
2367 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002368 FILE *fp;
Guido van Rossum43713e52000-02-29 13:59:29 +00002369 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname,
Guido van Rossum79f25d91997-04-29 20:08:16 +00002370 &PyFile_Type, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002371 return NULL;
2372 fp = get_file(pathname, fob, "r");
2373 if (fp == NULL)
2374 return NULL;
2375 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002376 if (fob == NULL)
2377 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002378 return m;
2379}
2380
Jack Jansen9c96a921995-02-15 22:57:06 +00002381#ifdef macintosh
Guido van Rossum79f25d91997-04-29 20:08:16 +00002382static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002383imp_load_resource(PyObject *self, PyObject *args)
Jack Jansen9c96a921995-02-15 22:57:06 +00002384{
2385 char *name;
2386 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002387 PyObject *m;
Jack Jansen9c96a921995-02-15 22:57:06 +00002388
Guido van Rossum43713e52000-02-29 13:59:29 +00002389 if (!PyArg_ParseTuple(args, "ss:load_resource", &name, &pathname))
Jack Jansen9c96a921995-02-15 22:57:06 +00002390 return NULL;
2391 m = PyMac_LoadResourceModule(name, pathname);
2392 return m;
2393}
2394#endif /* macintosh */
2395
Guido van Rossum79f25d91997-04-29 20:08:16 +00002396static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002397imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002398{
2399 char *name;
2400 PyObject *fob;
2401 char *pathname;
2402 char *suffix; /* Unused */
2403 char *mode;
2404 int type;
2405 FILE *fp;
2406
Guido van Rossum43713e52000-02-29 13:59:29 +00002407 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002408 &name, &fob, &pathname,
2409 &suffix, &mode, &type))
2410 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002411 if (*mode) {
2412 /* Mode must start with 'r' or 'U' and must not contain '+'.
2413 Implicit in this test is the assumption that the mode
2414 may contain other modifiers like 'b' or 't'. */
2415
2416 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002417 PyErr_Format(PyExc_ValueError,
2418 "invalid file open mode %.200s", mode);
2419 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002420 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002421 }
2422 if (fob == Py_None)
2423 fp = NULL;
2424 else {
2425 if (!PyFile_Check(fob)) {
2426 PyErr_SetString(PyExc_ValueError,
2427 "load_module arg#2 should be a file or None");
2428 return NULL;
2429 }
2430 fp = get_file(pathname, fob, mode);
2431 if (fp == NULL)
2432 return NULL;
2433 }
2434 return load_module(name, fp, pathname, type);
2435}
2436
2437static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002438imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002439{
2440 char *name;
2441 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002442 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002443 return NULL;
2444 return load_package(name, pathname);
2445}
2446
2447static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002448imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002449{
2450 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002451 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002452 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002453 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002454}
2455
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002456/* Doc strings */
2457
2458static char doc_imp[] = "\
2459This module provides the components needed to build your own\n\
2460__import__ function. Undocumented functions are obsolete.\n\
2461";
2462
2463static char doc_find_module[] = "\
2464find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
2465Search for a module. If path is omitted or None, search for a\n\
2466built-in, frozen or special module and continue search in sys.path.\n\
2467The module name cannot contain '.'; to search for a submodule of a\n\
2468package, pass the submodule name and the package's __path__.\
2469";
2470
2471static char doc_load_module[] = "\
2472load_module(name, file, filename, (suffix, mode, type)) -> module\n\
2473Load a module, given information returned by find_module().\n\
2474The module name must include the full package name, if any.\
2475";
2476
2477static char doc_get_magic[] = "\
2478get_magic() -> string\n\
2479Return the magic number for .pyc or .pyo files.\
2480";
2481
2482static char doc_get_suffixes[] = "\
2483get_suffixes() -> [(suffix, mode, type), ...]\n\
2484Return a list of (suffix, mode, type) tuples describing the files\n\
2485that find_module() looks for.\
2486";
2487
2488static char doc_new_module[] = "\
2489new_module(name) -> module\n\
2490Create a new module. Do not enter it in sys.modules.\n\
2491The module name must include the full package name, if any.\
2492";
2493
Tim Peters69232342001-08-30 05:16:13 +00002494static char doc_lock_held[] = "\
2495lock_held() -> 0 or 1\n\
2496Return 1 if the import lock is currently held.\n\
2497On platforms without threads, return 0.\
2498";
2499
Guido van Rossum79f25d91997-04-29 20:08:16 +00002500static PyMethodDef imp_methods[] = {
Neal Norwitz031829d2002-03-31 14:37:44 +00002501 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
2502 {"get_magic", imp_get_magic, METH_VARARGS, doc_get_magic},
2503 {"get_suffixes", imp_get_suffixes, METH_VARARGS, doc_get_suffixes},
2504 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
2505 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
2506 {"lock_held", imp_lock_held, METH_VARARGS, doc_lock_held},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002507 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00002508 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
2509 {"init_builtin", imp_init_builtin, METH_VARARGS},
2510 {"init_frozen", imp_init_frozen, METH_VARARGS},
2511 {"is_builtin", imp_is_builtin, METH_VARARGS},
2512 {"is_frozen", imp_is_frozen, METH_VARARGS},
2513 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002514#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00002515 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002516#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002517 {"load_package", imp_load_package, METH_VARARGS},
Jack Jansen9c96a921995-02-15 22:57:06 +00002518#ifdef macintosh
Neal Norwitz031829d2002-03-31 14:37:44 +00002519 {"load_resource", imp_load_resource, METH_VARARGS},
Jack Jansen9c96a921995-02-15 22:57:06 +00002520#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00002521 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002522 {NULL, NULL} /* sentinel */
2523};
2524
Guido van Rossum1a8791e1998-08-04 22:46:29 +00002525static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002526setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002527{
2528 PyObject *v;
2529 int err;
2530
2531 v = PyInt_FromLong((long)value);
2532 err = PyDict_SetItemString(d, name, v);
2533 Py_XDECREF(v);
2534 return err;
2535}
2536
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002537void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002538initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002539{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002540 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002541
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002542 m = Py_InitModule4("imp", imp_methods, doc_imp,
2543 NULL, PYTHON_API_VERSION);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002544 d = PyModule_GetDict(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002545
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002546 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
2547 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
2548 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
2549 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
2550 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
2551 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
2552 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
2553 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00002554 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002555
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002556 failure:
2557 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002558}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002559
2560
Guido van Rossumb18618d2000-05-03 23:44:39 +00002561/* API for embedding applications that want to add their own entries
2562 to the table of built-in modules. This should normally be called
2563 *before* Py_Initialize(). When the table resize fails, -1 is
2564 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002565
2566 After a similar function by Just van Rossum. */
2567
2568int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002569PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002570{
2571 static struct _inittab *our_copy = NULL;
2572 struct _inittab *p;
2573 int i, n;
2574
2575 /* Count the number of entries in both tables */
2576 for (n = 0; newtab[n].name != NULL; n++)
2577 ;
2578 if (n == 0)
2579 return 0; /* Nothing to do */
2580 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
2581 ;
2582
2583 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00002584 p = our_copy;
2585 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002586 if (p == NULL)
2587 return -1;
2588
2589 /* Copy the tables into the new memory */
2590 if (our_copy != PyImport_Inittab)
2591 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
2592 PyImport_Inittab = our_copy = p;
2593 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
2594
2595 return 0;
2596}
2597
2598/* Shorthand to add a single entry given a name and a function */
2599
2600int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002601PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00002602{
2603 struct _inittab newtab[2];
2604
2605 memset(newtab, '\0', sizeof newtab);
2606
2607 newtab[0].name = name;
2608 newtab[0].initfunc = initfunc;
2609
2610 return PyImport_ExtendInittab(newtab);
2611}