blob: 6500f23bc89f267637b3e0f2fae4ec9fca2ce1ce [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Module definition and import implementation */
3
Guido van Rossum79f25d91997-04-29 20:08:16 +00004#include "Python.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Guido van Rossumd8faa362007-04-27 19:54:29 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Neal Norwitzadb69fc2005-12-17 20:54:49 +00008#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00009#include "pythonrun.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000011#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000012#include "code.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000013#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000014#include "eval.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000015#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000016#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000017
Guido van Rossum55a83382000-09-20 20:31:38 +000018#ifdef HAVE_FCNTL_H
19#include <fcntl.h>
20#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000021#ifdef __cplusplus
22extern "C" {
23#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000024
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000025extern time_t PyOS_GetLastModificationTime(char *, FILE *);
26 /* In getmtime.c */
Guido van Rossum21d335e1993-10-15 13:01:11 +000027
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000028/* Magic word to reject .pyc files generated by other Python versions.
29 It should change for each incompatible change to the bytecode.
30
31 The value of CR and LF is incorporated so if you ever read or write
Guido van Rossum7faeab31995-07-07 22:50:36 +000032 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000033 Apple MPW compiler swaps their values, botching string constants.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000034
Guido van Rossum45aecf42006-03-15 04:58:47 +000035 The magic numbers must be spaced apart at least 2 values, as the
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000036 -U interpeter flag will cause MAGIC+1 being used. They have been
37 odd numbers for some time now.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000038
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000039 There were a variety of old schemes for setting the magic number.
40 The current working scheme is to increment the previous value by
41 10.
Guido van Rossumf6894922002-08-31 15:16:14 +000042
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000043 Known values:
44 Python 1.5: 20121
45 Python 1.5.1: 20121
46 Python 1.5.2: 20121
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000047 Python 1.6: 50428
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000048 Python 2.0: 50823
49 Python 2.0.1: 50823
50 Python 2.1: 60202
51 Python 2.1.1: 60202
52 Python 2.1.2: 60202
53 Python 2.2: 60717
Neal Norwitz7fdcb412002-06-14 01:07:39 +000054 Python 2.3a0: 62011
Michael W. Hudsondd32a912002-08-15 14:59:02 +000055 Python 2.3a0: 62021
Guido van Rossumf6894922002-08-31 15:16:14 +000056 Python 2.3a0: 62011 (!)
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000057 Python 2.4a0: 62041
Raymond Hettingerfd2d1f72004-08-23 23:37:48 +000058 Python 2.4a3: 62051
Raymond Hettinger2c31a052004-09-22 18:44:21 +000059 Python 2.4b1: 62061
Michael W. Hudsondf888462005-06-03 14:41:55 +000060 Python 2.5a0: 62071
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000061 Python 2.5a0: 62081 (ast-branch)
Guido van Rossumc2e20742006-02-27 22:32:47 +000062 Python 2.5a0: 62091 (with)
Guido van Rossumf6694362006-03-10 02:28:35 +000063 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000064 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
65 Python 2.5b3: 62111 (fix wrong code: x += yield)
66 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
67 storing constants that should have been removed)
Thomas Wouters89f507f2006-12-13 04:49:30 +000068 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp)
Christian Heimes99170a52007-12-19 02:07:34 +000069 Python 2.6a0: 62151 (peephole optimizations and STORE_MAP opcode)
Guido van Rossum45aecf42006-03-15 04:58:47 +000070 Python 3000: 3000
Brett Cannone2e23ef2006-08-25 05:05:30 +000071 3010 (removed UNARY_CONVERT)
Guido van Rossum86e58e22006-08-28 15:27:34 +000072 3020 (added BUILD_SET)
Guido van Rossum4f72a782006-10-27 23:31:49 +000073 3030 (added keyword-only parameters)
Guido van Rossum3203bdc2006-12-28 21:03:31 +000074 3040 (added signature annotations)
Guido van Rossum452bf512007-02-09 05:32:43 +000075 3050 (print becomes a function)
Guido van Rossum52cc1d82007-03-18 15:41:51 +000076 3060 (PEP 3115 metaclass syntax)
Guido van Rossum00bc0e02007-10-15 02:52:41 +000077 3070 (PEP 3109 raise changes)
78 3080 (PEP 3137 make __file__ and __name__ unicode)
Guido van Rossum98297ee2007-11-06 21:34:58 +000079 3090 (kill str8 interning)
Christian Heimes99170a52007-12-19 02:07:34 +000080 3100 (merge from 2.6a0, see 62151)
Christian Heimes3b06e532008-01-07 20:12:44 +000081 3102 (__file__ points to source file)
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000082.
Tim Peters36515e22001-11-18 04:06:29 +000083*/
Christian Heimes3b06e532008-01-07 20:12:44 +000084#define MAGIC (3102 | ((long)'\r'<<16) | ((long)'\n'<<24))
Guido van Rossum3ddee711991-12-16 13:06:34 +000085
Guido van Rossum96774c12000-05-01 20:19:08 +000086/* Magic word as global; note that _PyImport_Init() can change the
Jeremy Hylton9262b8a2000-06-30 04:59:17 +000087 value of this global to accommodate for alterations of how the
Guido van Rossum96774c12000-05-01 20:19:08 +000088 compiler works which are enabled by command line switches. */
89static long pyc_magic = MAGIC;
90
Guido van Rossum25ce5661997-08-02 03:10:38 +000091/* See _PyImport_FixupExtension() below */
92static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000093
Guido van Rossum771c6c81997-10-31 18:37:24 +000094/* This table is defined in config.c: */
95extern struct _inittab _PyImport_Inittab[];
96
Guido van Rossumce3a72a2007-10-19 23:16:50 +000097/* Method from Parser/tokenizer.c */
Guido van Rossum40d20bc2007-10-22 00:09:51 +000098extern char * PyTokenizer_FindEncoding(int);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000099
Guido van Rossum771c6c81997-10-31 18:37:24 +0000100struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +0000101
Guido van Rossumed1170e1999-12-20 21:23:41 +0000102/* these tables define the module suffixes that Python recognizes */
103struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000104
Guido van Rossumed1170e1999-12-20 21:23:41 +0000105static const struct filedescr _PyImport_StandardFiletab[] = {
Jack Jansenc88da1f2002-05-28 10:58:19 +0000106 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000107#ifdef MS_WINDOWS
Jack Jansenc88da1f2002-05-28 10:58:19 +0000108 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +0000109#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000110 {".pyc", "rb", PY_COMPILED},
111 {0, 0}
112};
113
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000114
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000115/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000116
117void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000118_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000119{
Guido van Rossumed1170e1999-12-20 21:23:41 +0000120 const struct filedescr *scan;
121 struct filedescr *filetab;
122 int countD = 0;
123 int countS = 0;
124
125 /* prepare _PyImport_Filetab: copy entries from
126 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
127 */
Guido van Rossum04110fb2007-08-24 16:32:05 +0000128#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000129 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
130 ++countD;
Guido van Rossum04110fb2007-08-24 16:32:05 +0000131#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000132 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
133 ++countS;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000134 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000135 if (filetab == NULL)
136 Py_FatalError("Can't initialize import file table.");
Guido van Rossum04110fb2007-08-24 16:32:05 +0000137#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossumed1170e1999-12-20 21:23:41 +0000138 memcpy(filetab, _PyImport_DynLoadFiletab,
139 countD * sizeof(struct filedescr));
Guido van Rossum04110fb2007-08-24 16:32:05 +0000140#endif
Guido van Rossumed1170e1999-12-20 21:23:41 +0000141 memcpy(filetab + countD, _PyImport_StandardFiletab,
142 countS * sizeof(struct filedescr));
143 filetab[countD + countS].suffix = NULL;
144
145 _PyImport_Filetab = filetab;
146
Guido van Rossum0824f631997-03-11 18:37:35 +0000147 if (Py_OptimizeFlag) {
Guido van Rossumed1170e1999-12-20 21:23:41 +0000148 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
149 for (; filetab->suffix != NULL; filetab++) {
150 if (strcmp(filetab->suffix, ".pyc") == 0)
151 filetab->suffix = ".pyo";
Guido van Rossum0824f631997-03-11 18:37:35 +0000152 }
153 }
Guido van Rossum96774c12000-05-01 20:19:08 +0000154
Guido van Rossum572dbf82007-04-27 23:53:51 +0000155 {
Guido van Rossum96774c12000-05-01 20:19:08 +0000156 /* Fix the pyc_magic so that byte compiled code created
157 using the all-Unicode method doesn't interfere with
158 code created in normal operation mode. */
159 pyc_magic = MAGIC + 1;
160 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000161}
162
Guido van Rossum25ce5661997-08-02 03:10:38 +0000163void
Just van Rossum52e14d62002-12-30 22:08:05 +0000164_PyImportHooks_Init(void)
165{
166 PyObject *v, *path_hooks = NULL, *zimpimport;
167 int err = 0;
168
169 /* adding sys.path_hooks and sys.path_importer_cache, setting up
170 zipimport */
Christian Heimes9cd17752007-11-18 19:35:23 +0000171 if (PyType_Ready(&PyNullImporter_Type) < 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000172 goto error;
Just van Rossum52e14d62002-12-30 22:08:05 +0000173
174 if (Py_VerboseFlag)
175 PySys_WriteStderr("# installing zipimport hook\n");
176
177 v = PyList_New(0);
178 if (v == NULL)
179 goto error;
180 err = PySys_SetObject("meta_path", v);
181 Py_DECREF(v);
182 if (err)
183 goto error;
184 v = PyDict_New();
185 if (v == NULL)
186 goto error;
187 err = PySys_SetObject("path_importer_cache", v);
188 Py_DECREF(v);
189 if (err)
190 goto error;
191 path_hooks = PyList_New(0);
192 if (path_hooks == NULL)
193 goto error;
194 err = PySys_SetObject("path_hooks", path_hooks);
195 if (err) {
196 error:
197 PyErr_Print();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000198 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
199 "path_importer_cache, or NullImporter failed"
200 );
Just van Rossum52e14d62002-12-30 22:08:05 +0000201 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000202
Just van Rossum52e14d62002-12-30 22:08:05 +0000203 zimpimport = PyImport_ImportModule("zipimport");
204 if (zimpimport == NULL) {
205 PyErr_Clear(); /* No zip import module -- okay */
206 if (Py_VerboseFlag)
207 PySys_WriteStderr("# can't import zipimport\n");
208 }
209 else {
210 PyObject *zipimporter = PyObject_GetAttrString(zimpimport,
211 "zipimporter");
212 Py_DECREF(zimpimport);
213 if (zipimporter == NULL) {
214 PyErr_Clear(); /* No zipimporter object -- okay */
215 if (Py_VerboseFlag)
216 PySys_WriteStderr(
Fred Drake1e5fc552003-07-11 15:01:02 +0000217 "# can't import zipimport.zipimporter\n");
Just van Rossum52e14d62002-12-30 22:08:05 +0000218 }
219 else {
220 /* sys.path_hooks.append(zipimporter) */
221 err = PyList_Append(path_hooks, zipimporter);
222 Py_DECREF(zipimporter);
223 if (err)
224 goto error;
225 if (Py_VerboseFlag)
226 PySys_WriteStderr(
227 "# installed zipimport hook\n");
228 }
229 }
230 Py_DECREF(path_hooks);
231}
232
233void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000234_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000235{
236 Py_XDECREF(extensions);
237 extensions = NULL;
Barry Warsaw84294482000-10-03 16:02:05 +0000238 PyMem_DEL(_PyImport_Filetab);
239 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000240}
241
242
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000243/* Locking primitives to prevent parallel imports of the same module
244 in different threads to return with a partially loaded module.
245 These calls are serialized by the global interpreter lock. */
246
247#ifdef WITH_THREAD
248
Guido van Rossum49b56061998-10-01 20:42:43 +0000249#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000250
Guido van Rossum65d5b571998-12-21 19:32:43 +0000251static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000252static long import_lock_thread = -1;
253static int import_lock_level = 0;
254
255static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000256lock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000257{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000258 long me = PyThread_get_thread_ident();
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000259 if (me == -1)
260 return; /* Too bad */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000261 if (import_lock == NULL) {
Guido van Rossum65d5b571998-12-21 19:32:43 +0000262 import_lock = PyThread_allocate_lock();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000263 if (import_lock == NULL)
264 return; /* Nothing much we can do. */
265 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000266 if (import_lock_thread == me) {
267 import_lock_level++;
268 return;
269 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000270 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
271 {
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000272 PyThreadState *tstate = PyEval_SaveThread();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000273 PyThread_acquire_lock(import_lock, 1);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000274 PyEval_RestoreThread(tstate);
275 }
276 import_lock_thread = me;
277 import_lock_level = 1;
278}
279
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000280static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000281unlock_import(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000282{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000283 long me = PyThread_get_thread_ident();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000284 if (me == -1 || import_lock == NULL)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000285 return 0; /* Too bad */
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000286 if (import_lock_thread != me)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000287 return -1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000288 import_lock_level--;
289 if (import_lock_level == 0) {
290 import_lock_thread = -1;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000291 PyThread_release_lock(import_lock);
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000292 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000293 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000294}
295
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000296/* This function is called from PyOS_AfterFork to ensure that newly
297 created child processes do not share locks with the parent. */
298
299void
300_PyImport_ReInitLock(void)
301{
302#ifdef _AIX
303 if (import_lock != NULL)
304 import_lock = PyThread_allocate_lock();
305#endif
306}
307
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000308#else
309
310#define lock_import()
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000311#define unlock_import() 0
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000312
313#endif
314
Tim Peters69232342001-08-30 05:16:13 +0000315static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000316imp_lock_held(PyObject *self, PyObject *noargs)
Tim Peters69232342001-08-30 05:16:13 +0000317{
Tim Peters69232342001-08-30 05:16:13 +0000318#ifdef WITH_THREAD
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000319 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000320#else
Guido van Rossumb8bff3f2002-04-07 06:34:38 +0000321 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000322#endif
323}
324
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000325static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000326imp_acquire_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000327{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000328#ifdef WITH_THREAD
329 lock_import();
330#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000331 Py_INCREF(Py_None);
332 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000333}
334
335static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000336imp_release_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000337{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000338#ifdef WITH_THREAD
339 if (unlock_import() < 0) {
340 PyErr_SetString(PyExc_RuntimeError,
341 "not holding the import lock");
342 return NULL;
343 }
344#endif
Neal Norwitz2294c0d2003-02-12 23:02:21 +0000345 Py_INCREF(Py_None);
346 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000347}
348
Guido van Rossumd8faa362007-04-27 19:54:29 +0000349static void
350imp_modules_reloading_clear(void)
351{
352 PyInterpreterState *interp = PyThreadState_Get()->interp;
353 if (interp->modules_reloading != NULL)
354 PyDict_Clear(interp->modules_reloading);
355}
356
Guido van Rossum25ce5661997-08-02 03:10:38 +0000357/* Helper for sys */
358
359PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000360PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000361{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000362 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000363 if (interp->modules == NULL)
364 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
365 return interp->modules;
366}
367
Guido van Rossum3f5da241990-12-20 15:06:42 +0000368
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000369/* List of names to clear in sys */
370static char* sys_deletes[] = {
Collin Winter670e6922007-03-21 02:57:17 +0000371 "path", "argv", "ps1", "ps2",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000372 "last_type", "last_value", "last_traceback",
Just van Rossum52e14d62002-12-30 22:08:05 +0000373 "path_hooks", "path_importer_cache", "meta_path",
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000374 NULL
375};
376
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000377static char* sys_files[] = {
378 "stdin", "__stdin__",
379 "stdout", "__stdout__",
380 "stderr", "__stderr__",
381 NULL
382};
383
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000384
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000385/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000386
Guido van Rossum3f5da241990-12-20 15:06:42 +0000387void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000388PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000389{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000390 Py_ssize_t pos, ndone;
Guido van Rossum758eec01998-01-19 21:58:26 +0000391 char *name;
392 PyObject *key, *value, *dict;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000393 PyInterpreterState *interp = PyThreadState_GET()->interp;
Guido van Rossum758eec01998-01-19 21:58:26 +0000394 PyObject *modules = interp->modules;
395
396 if (modules == NULL)
397 return; /* Already done */
398
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000399 /* Delete some special variables first. These are common
400 places where user values hide and people complain when their
401 destructors fail. Since the modules containing them are
402 deleted *last* of all, they would come too late in the normal
403 destruction order. Sigh. */
404
Georg Brandl1a3284e2007-12-02 09:40:06 +0000405 value = PyDict_GetItemString(modules, "builtins");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000406 if (value != NULL && PyModule_Check(value)) {
407 dict = PyModule_GetDict(value);
408 if (Py_VerboseFlag)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000409 PySys_WriteStderr("# clear builtins._\n");
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000410 PyDict_SetItemString(dict, "_", Py_None);
411 }
412 value = PyDict_GetItemString(modules, "sys");
413 if (value != NULL && PyModule_Check(value)) {
414 char **p;
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000415 PyObject *v;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000416 dict = PyModule_GetDict(value);
417 for (p = sys_deletes; *p != NULL; p++) {
418 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000419 PySys_WriteStderr("# clear sys.%s\n", *p);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000420 PyDict_SetItemString(dict, *p, Py_None);
421 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000422 for (p = sys_files; *p != NULL; p+=2) {
423 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000424 PySys_WriteStderr("# restore sys.%s\n", *p);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000425 v = PyDict_GetItemString(dict, *(p+1));
426 if (v == NULL)
427 v = Py_None;
428 PyDict_SetItemString(dict, *p, v);
429 }
430 }
431
432 /* First, delete __main__ */
433 value = PyDict_GetItemString(modules, "__main__");
434 if (value != NULL && PyModule_Check(value)) {
435 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000436 PySys_WriteStderr("# cleanup __main__\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000437 _PyModule_Clear(value);
438 PyDict_SetItemString(modules, "__main__", Py_None);
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000439 }
440
Georg Brandl1a3284e2007-12-02 09:40:06 +0000441 /* The special treatment of "builtins" here is because even
Guido van Rossum758eec01998-01-19 21:58:26 +0000442 when it's not referenced as a module, its dictionary is
443 referenced by almost every module's __builtins__. Since
444 deleting a module clears its dictionary (even if there are
Georg Brandl1a3284e2007-12-02 09:40:06 +0000445 references left to it), we need to delete the "builtins"
Guido van Rossum758eec01998-01-19 21:58:26 +0000446 module last. Likewise, we don't delete sys until the very
447 end because it is implicitly referenced (e.g. by print).
448
449 Also note that we 'delete' modules by replacing their entry
450 in the modules dict with None, rather than really deleting
451 them; this avoids a rehash of the modules dictionary and
452 also marks them as "non existent" so they won't be
453 re-imported. */
454
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000455 /* Next, repeatedly delete modules with a reference count of
Georg Brandl1a3284e2007-12-02 09:40:06 +0000456 one (skipping builtins and sys) and delete them */
Guido van Rossum758eec01998-01-19 21:58:26 +0000457 do {
458 ndone = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000459 pos = 0;
Guido van Rossum758eec01998-01-19 21:58:26 +0000460 while (PyDict_Next(modules, &pos, &key, &value)) {
461 if (value->ob_refcnt != 1)
462 continue;
Neal Norwitz80e7f272007-08-26 06:45:23 +0000463 if (PyUnicode_Check(key) && PyModule_Check(value)) {
464 name = PyUnicode_AsString(key);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000465 if (strcmp(name, "builtins") == 0)
Guido van Rossum758eec01998-01-19 21:58:26 +0000466 continue;
467 if (strcmp(name, "sys") == 0)
468 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000469 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000470 PySys_WriteStderr(
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000471 "# cleanup[1] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000472 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000473 PyDict_SetItem(modules, key, Py_None);
474 ndone++;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000475 }
476 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000477 } while (ndone > 0);
478
Georg Brandl1a3284e2007-12-02 09:40:06 +0000479 /* Next, delete all modules (still skipping builtins and sys) */
Guido van Rossum758eec01998-01-19 21:58:26 +0000480 pos = 0;
481 while (PyDict_Next(modules, &pos, &key, &value)) {
Neal Norwitz80e7f272007-08-26 06:45:23 +0000482 if (PyUnicode_Check(key) && PyModule_Check(value)) {
483 name = PyUnicode_AsString(key);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000484 if (strcmp(name, "builtins") == 0)
Guido van Rossum758eec01998-01-19 21:58:26 +0000485 continue;
486 if (strcmp(name, "sys") == 0)
487 continue;
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000488 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000489 PySys_WriteStderr("# cleanup[2] %s\n", name);
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000490 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000491 PyDict_SetItem(modules, key, Py_None);
492 }
493 }
494
Georg Brandl1a3284e2007-12-02 09:40:06 +0000495 /* Next, delete sys and builtins (in that order) */
Guido van Rossum758eec01998-01-19 21:58:26 +0000496 value = PyDict_GetItemString(modules, "sys");
497 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000498 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000499 PySys_WriteStderr("# cleanup sys\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000500 _PyModule_Clear(value);
Guido van Rossum758eec01998-01-19 21:58:26 +0000501 PyDict_SetItemString(modules, "sys", Py_None);
502 }
Georg Brandl1a3284e2007-12-02 09:40:06 +0000503 value = PyDict_GetItemString(modules, "builtins");
Guido van Rossum758eec01998-01-19 21:58:26 +0000504 if (value != NULL && PyModule_Check(value)) {
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000505 if (Py_VerboseFlag)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000506 PySys_WriteStderr("# cleanup builtins\n");
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000507 _PyModule_Clear(value);
Georg Brandl1a3284e2007-12-02 09:40:06 +0000508 PyDict_SetItemString(modules, "builtins", Py_None);
Guido van Rossum758eec01998-01-19 21:58:26 +0000509 }
510
511 /* Finally, clear and delete the modules directory */
512 PyDict_Clear(modules);
513 interp->modules = NULL;
514 Py_DECREF(modules);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000515 Py_CLEAR(interp->modules_reloading);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000516}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000517
518
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000519/* Helper for pythonrun.c -- return magic number */
520
521long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000522PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000523{
Guido van Rossum96774c12000-05-01 20:19:08 +0000524 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000525}
526
527
Guido van Rossum25ce5661997-08-02 03:10:38 +0000528/* Magic for extension modules (built-in as well as dynamically
529 loaded). To prevent initializing an extension module more than
530 once, we keep a static dictionary 'extensions' keyed by module name
531 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000532 modules), containing these modules. A copy of the module's
Guido van Rossum25ce5661997-08-02 03:10:38 +0000533 dictionary is stored by calling _PyImport_FixupExtension()
534 immediately after the module initialization function succeeds. A
535 copy can be retrieved from there by calling
536 _PyImport_FindExtension(). */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000537
Guido van Rossum79f25d91997-04-29 20:08:16 +0000538PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000539_PyImport_FixupExtension(char *name, char *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000540{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000541 PyObject *modules, *mod, *dict, *copy;
542 if (extensions == NULL) {
543 extensions = PyDict_New();
544 if (extensions == NULL)
545 return NULL;
546 }
547 modules = PyImport_GetModuleDict();
548 mod = PyDict_GetItemString(modules, name);
549 if (mod == NULL || !PyModule_Check(mod)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000550 PyErr_Format(PyExc_SystemError,
551 "_PyImport_FixupExtension: module %.200s not loaded", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000552 return NULL;
553 }
554 dict = PyModule_GetDict(mod);
555 if (dict == NULL)
556 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000557 copy = PyDict_Copy(dict);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000558 if (copy == NULL)
559 return NULL;
560 PyDict_SetItemString(extensions, filename, copy);
561 Py_DECREF(copy);
562 return copy;
563}
564
565PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000566_PyImport_FindExtension(char *name, char *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567{
Fred Drake9cd0efc2001-10-25 21:38:59 +0000568 PyObject *dict, *mod, *mdict;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000569 if (extensions == NULL)
570 return NULL;
571 dict = PyDict_GetItemString(extensions, filename);
572 if (dict == NULL)
573 return NULL;
574 mod = PyImport_AddModule(name);
575 if (mod == NULL)
576 return NULL;
577 mdict = PyModule_GetDict(mod);
578 if (mdict == NULL)
579 return NULL;
Fred Drake9cd0efc2001-10-25 21:38:59 +0000580 if (PyDict_Update(mdict, dict))
Guido van Rossum25ce5661997-08-02 03:10:38 +0000581 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000582 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000583 PySys_WriteStderr("import %s # previously loaded (%s)\n",
Guido van Rossum25ce5661997-08-02 03:10:38 +0000584 name, filename);
585 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000586}
587
588
589/* Get the module object corresponding to a module name.
590 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000591 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000592 Because the former action is most common, THIS DOES NOT RETURN A
593 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000594
Guido van Rossum79f25d91997-04-29 20:08:16 +0000595PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +0000596PyImport_AddModule(const char *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000597{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000598 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000599 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000600
Guido van Rossum25ce5661997-08-02 03:10:38 +0000601 if ((m = PyDict_GetItemString(modules, name)) != NULL &&
Guido van Rossum79f25d91997-04-29 20:08:16 +0000602 PyModule_Check(m))
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000603 return m;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000604 m = PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000605 if (m == NULL)
606 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000607 if (PyDict_SetItemString(modules, name, m) != 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000608 Py_DECREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000609 return NULL;
610 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000611 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000612
613 return m;
614}
615
Tim Peters1cd70172004-08-02 03:52:12 +0000616/* Remove name from sys.modules, if it's there. */
617static void
618_RemoveModule(const char *name)
619{
620 PyObject *modules = PyImport_GetModuleDict();
621 if (PyDict_GetItemString(modules, name) == NULL)
622 return;
623 if (PyDict_DelItemString(modules, name) < 0)
624 Py_FatalError("import: deleting existing key in"
625 "sys.modules failed");
626}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000627
Christian Heimes3b06e532008-01-07 20:12:44 +0000628static PyObject * get_sourcefile(const char *file);
629
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000630/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000631 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
632 * removed from sys.modules, to avoid leaving damaged module objects
633 * in sys.modules. The caller may wish to restore the original
634 * module object (if any) in this case; PyImport_ReloadModule is an
635 * example.
636 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000637PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000638PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000639{
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000640 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL);
641}
642
643PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000644PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000645{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000646 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossum79f25d91997-04-29 20:08:16 +0000647 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000648
Guido van Rossum79f25d91997-04-29 20:08:16 +0000649 m = PyImport_AddModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000650 if (m == NULL)
651 return NULL;
Jeremy Hyltonecd91292004-01-02 23:25:32 +0000652 /* If the module is being reloaded, we get the old module back
653 and re-use its dict to exec the new code. */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000654 d = PyModule_GetDict(m);
655 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
656 if (PyDict_SetItemString(d, "__builtins__",
657 PyEval_GetBuiltins()) != 0)
Tim Peters1cd70172004-08-02 03:52:12 +0000658 goto error;
Guido van Rossum6135a871995-01-09 17:53:26 +0000659 }
Guido van Rossum9c9a07c1996-05-16 20:43:40 +0000660 /* Remember the filename as the __file__ attribute */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000661 v = NULL;
662 if (pathname != NULL) {
Christian Heimes3b06e532008-01-07 20:12:44 +0000663 v = get_sourcefile(pathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000664 if (v == NULL)
665 PyErr_Clear();
666 }
667 if (v == NULL) {
668 v = ((PyCodeObject *)co)->co_filename;
669 Py_INCREF(v);
670 }
671 if (PyDict_SetItemString(d, "__file__", v) != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000672 PyErr_Clear(); /* Not important enough to report */
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000673 Py_DECREF(v);
674
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000675 v = PyEval_EvalCode((PyCodeObject *)co, d, d);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000676 if (v == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +0000677 goto error;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000678 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000679
Guido van Rossum25ce5661997-08-02 03:10:38 +0000680 if ((m = PyDict_GetItemString(modules, name)) == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000681 PyErr_Format(PyExc_ImportError,
682 "Loaded module %.200s not found in sys.modules",
683 name);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000684 return NULL;
685 }
686
Guido van Rossum79f25d91997-04-29 20:08:16 +0000687 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000688
689 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000690
691 error:
692 _RemoveModule(name);
693 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000694}
695
696
697/* Given a pathname for a Python source file, fill a buffer with the
698 pathname for the corresponding compiled file. Return the pathname
699 for the compiled file, or NULL if there's no space in the buffer.
700 Doesn't set an exception. */
701
702static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000703make_compiled_pathname(char *pathname, char *buf, size_t buflen)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000704{
Tim Petersc1731372001-08-04 08:12:36 +0000705 size_t len = strlen(pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000706 if (len+2 > buflen)
707 return NULL;
Tim Petersc1731372001-08-04 08:12:36 +0000708
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000709#ifdef MS_WINDOWS
Tim Petersc1731372001-08-04 08:12:36 +0000710 /* Treat .pyw as if it were .py. The case of ".pyw" must match
711 that used in _PyImport_StandardFiletab. */
712 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0)
713 --len; /* pretend 'w' isn't there */
714#endif
715 memcpy(buf, pathname, len);
716 buf[len] = Py_OptimizeFlag ? 'o' : 'c';
717 buf[len+1] = '\0';
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000718
719 return buf;
720}
721
722
723/* Given a pathname for a Python source file, its time of last
724 modification, and a pathname for a compiled file, check whether the
725 compiled file represents the same version of the source. If so,
726 return a FILE pointer for the compiled file, positioned just after
727 the header; if not, return NULL.
728 Doesn't set an exception. */
729
730static FILE *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000731check_compiled_module(char *pathname, time_t mtime, char *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000732{
733 FILE *fp;
734 long magic;
735 long pyc_mtime;
736
737 fp = fopen(cpathname, "rb");
738 if (fp == NULL)
739 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000740 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000741 if (magic != pyc_magic) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000742 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000743 PySys_WriteStderr("# %s has bad magic\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000744 fclose(fp);
745 return NULL;
746 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000747 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000748 if (pyc_mtime != mtime) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000749 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000750 PySys_WriteStderr("# %s has bad mtime\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000751 fclose(fp);
752 return NULL;
753 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000754 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000755 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000756 return fp;
757}
758
759
760/* Read a code object from a file and check it for validity */
761
Guido van Rossum79f25d91997-04-29 20:08:16 +0000762static PyCodeObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000763read_compiled_module(char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000764{
Guido van Rossum79f25d91997-04-29 20:08:16 +0000765 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000766
Tim Petersd9b9ac82001-01-28 00:27:39 +0000767 co = PyMarshal_ReadLastObjectFromFile(fp);
Armin Rigo01ab2792004-03-26 15:09:27 +0000768 if (co == NULL)
769 return NULL;
770 if (!PyCode_Check(co)) {
771 PyErr_Format(PyExc_ImportError,
772 "Non-code object in %.200s", cpathname);
773 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000774 return NULL;
775 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000776 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000777}
778
779
780/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000781 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000782
Guido van Rossum79f25d91997-04-29 20:08:16 +0000783static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000784load_compiled_module(char *name, char *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000785{
786 long magic;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000787 PyCodeObject *co;
788 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000789
Guido van Rossum79f25d91997-04-29 20:08:16 +0000790 magic = PyMarshal_ReadLongFromFile(fp);
Guido van Rossum96774c12000-05-01 20:19:08 +0000791 if (magic != pyc_magic) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000792 PyErr_Format(PyExc_ImportError,
793 "Bad magic number in %.200s", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000794 return NULL;
795 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000796 (void) PyMarshal_ReadLongFromFile(fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000797 co = read_compiled_module(cpathname, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000798 if (co == NULL)
799 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000800 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000801 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000802 name, cpathname);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000803 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000804 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000805
806 return m;
807}
808
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000809/* Parse a source file and return the corresponding code object */
810
Guido van Rossum79f25d91997-04-29 20:08:16 +0000811static PyCodeObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000812parse_source_module(const char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000813{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000814 PyCodeObject *co = NULL;
815 mod_ty mod;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000816 PyArena *arena = PyArena_New();
817 if (arena == NULL)
818 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000819
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000820 mod = PyParser_ASTFromFile(fp, pathname, NULL,
821 Py_file_input, 0, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000822 NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000823 if (mod) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000824 co = PyAST_Compile(mod, pathname, NULL, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000825 }
Thomas Wouters89f507f2006-12-13 04:49:30 +0000826 PyArena_Free(arena);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000827 return co;
828}
829
830
Guido van Rossum55a83382000-09-20 20:31:38 +0000831/* Helper to open a bytecode file for writing in exclusive mode */
832
833static FILE *
834open_exclusive(char *filename)
835{
836#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
837 /* Use O_EXCL to avoid a race condition when another process tries to
838 write the same file. When that happens, our open() call fails,
839 which is just fine (since it's only a cache).
840 XXX If the file exists and is writable but the directory is not
841 writable, the file will never be written. Oh well.
842 */
843 int fd;
844 (void) unlink(filename);
Tim Peters42c83af2000-09-29 04:03:10 +0000845 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
846#ifdef O_BINARY
847 |O_BINARY /* necessary for Windows */
848#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000849#ifdef __VMS
Martin v. Löwis18e16552006-02-15 17:27:45 +0000850 , 0666, "ctxt=bin", "shr=nil"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000851#else
Martin v. Löwis18e16552006-02-15 17:27:45 +0000852 , 0666
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000853#endif
Martin v. Löwis18e16552006-02-15 17:27:45 +0000854 );
Guido van Rossum55a83382000-09-20 20:31:38 +0000855 if (fd < 0)
856 return NULL;
857 return fdopen(fd, "wb");
858#else
859 /* Best we can do -- on Windows this can't happen anyway */
860 return fopen(filename, "wb");
861#endif
862}
863
864
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000865/* Write a compiled module to a file, placing the time of last
866 modification of its source into the header.
867 Errors are ignored, if a write error occurs an attempt is made to
868 remove the file. */
869
870static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000871write_compiled_module(PyCodeObject *co, char *cpathname, time_t mtime)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000872{
873 FILE *fp;
874
Guido van Rossum55a83382000-09-20 20:31:38 +0000875 fp = open_exclusive(cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000876 if (fp == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000877 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000878 PySys_WriteStderr(
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000879 "# can't create %s\n", cpathname);
880 return;
881 }
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000882 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000883 /* First write a 0 for mtime */
Martin v. Löwisef82d2f2004-06-27 16:51:46 +0000884 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION);
885 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION);
Armin Rigo01ab2792004-03-26 15:09:27 +0000886 if (fflush(fp) != 0 || ferror(fp)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000887 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000888 PySys_WriteStderr("# can't write %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000889 /* Don't keep partial file */
890 fclose(fp);
891 (void) unlink(cpathname);
892 return;
893 }
894 /* Now write the true mtime */
895 fseek(fp, 4L, 0);
Martin v. Löwis18e16552006-02-15 17:27:45 +0000896 assert(mtime < LONG_MAX);
Martin v. Löwis725507b2006-03-07 12:08:51 +0000897 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000898 fflush(fp);
899 fclose(fp);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000900 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000901 PySys_WriteStderr("# wrote %s\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000902}
903
904
905/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000906 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
907 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000908
Guido van Rossum79f25d91997-04-29 20:08:16 +0000909static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000910load_source_module(char *name, char *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000911{
Fred Drake4c82b232000-06-30 16:18:57 +0000912 time_t mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000913 FILE *fpc;
914 char buf[MAXPATHLEN+1];
915 char *cpathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000916 PyCodeObject *co;
917 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000918
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000919 mtime = PyOS_GetLastModificationTime(pathname, fp);
Neal Norwitz708e51a2005-10-03 04:48:15 +0000920 if (mtime == (time_t)(-1)) {
921 PyErr_Format(PyExc_RuntimeError,
922 "unable to get modification time from '%s'",
923 pathname);
Fred Drake4c82b232000-06-30 16:18:57 +0000924 return NULL;
Neal Norwitz708e51a2005-10-03 04:48:15 +0000925 }
Fred Drake4c82b232000-06-30 16:18:57 +0000926#if SIZEOF_TIME_T > 4
927 /* Python's .pyc timestamp handling presumes that the timestamp fits
928 in 4 bytes. This will be fine until sometime in the year 2038,
929 when a 4-byte signed time_t will overflow.
930 */
931 if (mtime >> 32) {
932 PyErr_SetString(PyExc_OverflowError,
Fred Drakec63d3e92001-03-01 06:33:32 +0000933 "modification time overflows a 4 byte field");
Fred Drake4c82b232000-06-30 16:18:57 +0000934 return NULL;
935 }
936#endif
Tim Peters36515e22001-11-18 04:06:29 +0000937 cpathname = make_compiled_pathname(pathname, buf,
Jeremy Hylton37832f02001-04-13 17:50:20 +0000938 (size_t)MAXPATHLEN + 1);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000939 if (cpathname != NULL &&
940 (fpc = check_compiled_module(pathname, mtime, cpathname))) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000941 co = read_compiled_module(cpathname, fpc);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000942 fclose(fpc);
943 if (co == NULL)
944 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000945 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000946 PySys_WriteStderr("import %s # precompiled from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000947 name, cpathname);
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000948 pathname = cpathname;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000949 }
950 else {
951 co = parse_source_module(pathname, fp);
952 if (co == NULL)
953 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000954 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +0000955 PySys_WriteStderr("import %s # from %s\n",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000956 name, pathname);
Christian Heimes790c8232008-01-07 21:14:23 +0000957 if (cpathname) {
958 PyObject *ro = PySys_GetObject("dont_write_bytecode");
959 if (ro == NULL || !PyObject_IsTrue(ro))
960 write_compiled_module(co, cpathname, mtime);
961 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000962 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000963 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000964 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000965
966 return m;
967}
968
Christian Heimes3b06e532008-01-07 20:12:44 +0000969/* Get source file -> unicode or None
970 * Returns the path to the py file if available, else the given path
971 */
972static PyObject *
973get_sourcefile(const char *file)
974{
975 char py[MAXPATHLEN + 1];
976 Py_ssize_t len;
977 PyObject *u;
978 struct stat statbuf;
979
980 if (!file || !*file) {
981 Py_RETURN_NONE;
982 }
983
984 len = strlen(file);
985 if (len > MAXPATHLEN || PyOS_stricmp(&file[len-4], ".pyc") != 0) {
986 return PyUnicode_DecodeFSDefault(file);
987 }
988
989 strncpy(py, file, len-1);
990 py[len] = '\0';
991 if (stat(py, &statbuf) == 0 &&
992 S_ISREG(statbuf.st_mode)) {
993 u = PyUnicode_DecodeFSDefault(py);
994 }
995 else {
996 u = PyUnicode_DecodeFSDefault(file);
997 }
998 return u;
999}
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001000
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001001/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001002static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
1003static struct filedescr *find_module(char *, char *, PyObject *,
1004 char *, size_t, FILE **, PyObject **);
Christian Heimes3b06e532008-01-07 20:12:44 +00001005static struct _frozen * find_frozen(char *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001006
1007/* Load a package and return its module object WITH INCREMENTED
1008 REFERENCE COUNT */
1009
1010static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001011load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001012{
Tim Peters1cd70172004-08-02 03:52:12 +00001013 PyObject *m, *d;
1014 PyObject *file = NULL;
1015 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001016 int err;
1017 char buf[MAXPATHLEN+1];
1018 FILE *fp = NULL;
1019 struct filedescr *fdp;
1020
1021 m = PyImport_AddModule(name);
1022 if (m == NULL)
1023 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001024 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001025 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001026 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001027 d = PyModule_GetDict(m);
Christian Heimes3b06e532008-01-07 20:12:44 +00001028 file = get_sourcefile(pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001029 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +00001030 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001031 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +00001032 if (path == NULL)
1033 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001034 err = PyDict_SetItemString(d, "__file__", file);
1035 if (err == 0)
1036 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +00001037 if (err != 0)
1038 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001039 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00001040 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001041 if (fdp == NULL) {
1042 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1043 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +00001044 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001045 }
1046 else
1047 m = NULL;
1048 goto cleanup;
1049 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001050 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001051 if (fp != NULL)
1052 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00001053 goto cleanup;
1054
1055 error:
1056 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001057 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001058 Py_XDECREF(path);
1059 Py_XDECREF(file);
1060 return m;
1061}
1062
1063
1064/* Helper to test for built-in module */
1065
1066static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001067is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001068{
1069 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001070 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1071 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1072 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001073 return -1;
1074 else
1075 return 1;
1076 }
1077 }
1078 return 0;
1079}
1080
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001081
Just van Rossum52e14d62002-12-30 22:08:05 +00001082/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1083 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +00001084 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +00001085 that can handle the path item. Return None if no hook could;
1086 this tells our caller it should fall back to the builtin
1087 import mechanism. Cache the result in path_importer_cache.
1088 Returns a borrowed reference. */
1089
1090static PyObject *
1091get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1092 PyObject *p)
1093{
1094 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001095 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001096
1097 /* These conditions are the caller's responsibility: */
1098 assert(PyList_Check(path_hooks));
1099 assert(PyDict_Check(path_importer_cache));
1100
1101 nhooks = PyList_Size(path_hooks);
1102 if (nhooks < 0)
1103 return NULL; /* Shouldn't happen */
1104
1105 importer = PyDict_GetItem(path_importer_cache, p);
1106 if (importer != NULL)
1107 return importer;
1108
1109 /* set path_importer_cache[p] to None to avoid recursion */
1110 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1111 return NULL;
1112
1113 for (j = 0; j < nhooks; j++) {
1114 PyObject *hook = PyList_GetItem(path_hooks, j);
1115 if (hook == NULL)
1116 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001117 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001118 if (importer != NULL)
1119 break;
1120
1121 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1122 return NULL;
1123 }
1124 PyErr_Clear();
1125 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 if (importer == NULL) {
1127 importer = PyObject_CallFunctionObjArgs(
Christian Heimes9cd17752007-11-18 19:35:23 +00001128 (PyObject *)&PyNullImporter_Type, p, NULL
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001129 );
1130 if (importer == NULL) {
1131 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1132 PyErr_Clear();
1133 return Py_None;
1134 }
1135 }
1136 }
1137 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001138 int err = PyDict_SetItem(path_importer_cache, p, importer);
1139 Py_DECREF(importer);
1140 if (err != 0)
1141 return NULL;
1142 }
1143 return importer;
1144}
1145
Christian Heimes9cd17752007-11-18 19:35:23 +00001146PyAPI_FUNC(PyObject *)
1147PyImport_GetImporter(PyObject *path) {
1148 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
1149
1150 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) {
1151 if ((path_hooks = PySys_GetObject("path_hooks"))) {
1152 importer = get_path_importer(path_importer_cache,
1153 path_hooks, path);
1154 }
1155 }
1156 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1157 return importer;
1158}
1159
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001160/* Search the path (default sys.path) for a module. Return the
1161 corresponding filedescr struct, and (via return arguments) the
1162 pathname and an open file. Return NULL if the module is not found. */
1163
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001164#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001165extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001166 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001167#endif
1168
Martin v. Löwis18e16552006-02-15 17:27:45 +00001169static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001170static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001171static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001172
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001173static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001174find_module(char *fullname, char *subname, PyObject *path, char *buf,
1175 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001176{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001177 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001178 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001179 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001180 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001181 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001182 PyObject *path_hooks, *path_importer_cache;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001183 struct stat statbuf;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001184 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1185 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1186 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001187 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001188#if defined(PYOS_OS2)
1189 size_t saved_len;
1190 size_t saved_namelen;
1191 char *saved_buf = NULL;
1192#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001193 if (p_loader != NULL)
1194 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001195
Just van Rossum52e14d62002-12-30 22:08:05 +00001196 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001197 PyErr_SetString(PyExc_OverflowError,
1198 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001199 return NULL;
1200 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001201 strcpy(name, subname);
1202
1203 /* sys.meta_path import hook */
1204 if (p_loader != NULL) {
1205 PyObject *meta_path;
1206
1207 meta_path = PySys_GetObject("meta_path");
1208 if (meta_path == NULL || !PyList_Check(meta_path)) {
1209 PyErr_SetString(PyExc_ImportError,
1210 "sys.meta_path must be a list of "
1211 "import hooks");
1212 return NULL;
1213 }
1214 Py_INCREF(meta_path); /* zap guard */
1215 npath = PyList_Size(meta_path);
1216 for (i = 0; i < npath; i++) {
1217 PyObject *loader;
1218 PyObject *hook = PyList_GetItem(meta_path, i);
1219 loader = PyObject_CallMethod(hook, "find_module",
1220 "sO", fullname,
1221 path != NULL ?
1222 path : Py_None);
1223 if (loader == NULL) {
1224 Py_DECREF(meta_path);
1225 return NULL; /* true error */
1226 }
1227 if (loader != Py_None) {
1228 /* a loader was found */
1229 *p_loader = loader;
1230 Py_DECREF(meta_path);
1231 return &importhookdescr;
1232 }
1233 Py_DECREF(loader);
1234 }
1235 Py_DECREF(meta_path);
1236 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001237
Guido van Rossum13d77992007-07-23 03:16:50 +00001238 if (path != NULL && PyUnicode_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001239 /* The only type of submodule allowed inside a "frozen"
1240 package are other frozen modules or packages. */
Guido van Rossum13d77992007-07-23 03:16:50 +00001241 char *p = PyUnicode_AsString(path);
1242 if (strlen(p) + 1 + strlen(name) >= (size_t)buflen) {
Guido van Rossum0506a431998-08-11 15:07:39 +00001243 PyErr_SetString(PyExc_ImportError,
1244 "full frozen module name too long");
1245 return NULL;
1246 }
Guido van Rossum13d77992007-07-23 03:16:50 +00001247 strcpy(buf, p);
Guido van Rossum0506a431998-08-11 15:07:39 +00001248 strcat(buf, ".");
1249 strcat(buf, name);
1250 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001251 if (find_frozen(name) != NULL) {
1252 strcpy(buf, name);
1253 return &fd_frozen;
1254 }
1255 PyErr_Format(PyExc_ImportError,
1256 "No frozen submodule named %.200s", name);
1257 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001258 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001259 if (path == NULL) {
1260 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001261 strcpy(buf, name);
1262 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001263 }
Greg Ward201baee2001-10-04 14:52:06 +00001264 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001265 strcpy(buf, name);
1266 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001267 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001268
Guido van Rossumac279101996-08-22 23:10:58 +00001269#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001270 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1271 if (fp != NULL) {
1272 *p_fp = fp;
1273 return fdp;
1274 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001275#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001276 path = PySys_GetObject("path");
1277 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001278 if (path == NULL || !PyList_Check(path)) {
1279 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001280 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001281 return NULL;
1282 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001283
1284 path_hooks = PySys_GetObject("path_hooks");
1285 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1286 PyErr_SetString(PyExc_ImportError,
1287 "sys.path_hooks must be a list of "
1288 "import hooks");
1289 return NULL;
1290 }
1291 path_importer_cache = PySys_GetObject("path_importer_cache");
1292 if (path_importer_cache == NULL ||
1293 !PyDict_Check(path_importer_cache)) {
1294 PyErr_SetString(PyExc_ImportError,
1295 "sys.path_importer_cache must be a dict");
1296 return NULL;
1297 }
1298
Guido van Rossum79f25d91997-04-29 20:08:16 +00001299 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001300 namelen = strlen(name);
1301 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001302 PyObject *v = PyList_GetItem(path, i);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001303 PyObject *origv = v;
Guido van Rossum6262cc72007-05-09 23:29:27 +00001304 const char *base;
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001305 Py_ssize_t size;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001306 if (!v)
1307 return NULL;
Guido van Rossuma4b8d1d2007-08-27 15:02:28 +00001308 if (PyUnicode_Check(v)) {
1309 v = _PyUnicode_AsDefaultEncodedString(v, NULL);
1310 if (v == NULL)
1311 return NULL;
1312 }
1313 if (!PyString_Check(v))
1314 continue;
1315 base = PyString_AS_STRING(v);
1316 size = PyString_GET_SIZE(v);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001317 len = size;
Walter Dörwald3430d702002-06-17 10:43:59 +00001318 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001319 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001320 }
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001321 strcpy(buf, base);
Walter Dörwald3430d702002-06-17 10:43:59 +00001322 if (strlen(buf) != len) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001323 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001324 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001325
1326 /* sys.path_hooks import hook */
1327 if (p_loader != NULL) {
1328 PyObject *importer;
1329
1330 importer = get_path_importer(path_importer_cache,
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001331 path_hooks, origv);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001332 if (importer == NULL) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001333 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001334 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001335 /* Note: importer is a borrowed reference */
1336 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001337 PyObject *loader;
1338 loader = PyObject_CallMethod(importer,
1339 "find_module",
1340 "s", fullname);
1341 if (loader == NULL)
1342 return NULL; /* error */
1343 if (loader != Py_None) {
1344 /* a loader was found */
1345 *p_loader = loader;
1346 return &importhookdescr;
1347 }
1348 Py_DECREF(loader);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001349 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001350 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001351 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001352 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001353
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001354 if (len > 0 && buf[len-1] != SEP
1355#ifdef ALTSEP
1356 && buf[len-1] != ALTSEP
1357#endif
1358 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001359 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001360 strcpy(buf+len, name);
1361 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001362
1363 /* Check for package import (buf holds a directory name,
1364 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001365#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001366 if (stat(buf, &statbuf) == 0 && /* it exists */
1367 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001368 case_ok(buf, len, namelen, name)) { /* case matches */
1369 if (find_init_module(buf)) { /* and has __init__.py */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001370 return &fd_package;
1371 }
1372 else {
1373 char warnstr[MAXPATHLEN+80];
1374 sprintf(warnstr, "Not importing directory "
1375 "'%.*s': missing __init__.py",
1376 MAXPATHLEN, buf);
Skip Montanaro46fc3372007-08-12 11:44:53 +00001377 if (PyErr_WarnEx(PyExc_ImportWarning,
1378 warnstr, 1)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001379 return NULL;
1380 }
1381 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001382 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001383#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001384#if defined(PYOS_OS2)
1385 /* take a snapshot of the module spec for restoration
1386 * after the 8 character DLL hackery
1387 */
1388 saved_buf = strdup(buf);
1389 saved_len = len;
1390 saved_namelen = namelen;
1391#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001392 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Guido van Rossum04110fb2007-08-24 16:32:05 +00001393#if defined(PYOS_OS2) && defined(HAVE_DYNAMIC_LOADING)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001394 /* OS/2 limits DLLs to 8 character names (w/o
1395 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001396 * so if the name is longer than that and its a
1397 * dynamically loaded module we're going to try,
1398 * truncate the name before trying
1399 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001400 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001401 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001402 const struct filedescr *scan;
1403 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001404 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001405 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001406 break;
1407 else
1408 scan++;
1409 }
1410 if (scan->suffix != NULL) {
1411 /* yes, so truncate the name */
1412 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001413 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001414 buf[len] = '\0';
1415 }
1416 }
1417#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001418 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001419 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001420 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001421 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001422 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001423 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001424 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001425 if (fp != NULL) {
1426 if (case_ok(buf, len, namelen, name))
1427 break;
1428 else { /* continue search */
1429 fclose(fp);
1430 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001431 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001432 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001433#if defined(PYOS_OS2)
1434 /* restore the saved snapshot */
1435 strcpy(buf, saved_buf);
1436 len = saved_len;
1437 namelen = saved_namelen;
1438#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001439 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001440#if defined(PYOS_OS2)
1441 /* don't need/want the module name snapshot anymore */
1442 if (saved_buf)
1443 {
1444 free(saved_buf);
1445 saved_buf = NULL;
1446 }
1447#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001448 if (fp != NULL)
1449 break;
1450 }
1451 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001452 PyErr_Format(PyExc_ImportError,
1453 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001454 return NULL;
1455 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001456 *p_fp = fp;
1457 return fdp;
1458}
1459
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001460/* Helpers for main.c
1461 * Find the source file corresponding to a named module
1462 */
1463struct filedescr *
1464_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1465 size_t buflen, FILE **p_fp, PyObject **p_loader)
1466{
1467 return find_module((char *) name, (char *) name, path,
1468 buf, buflen, p_fp, p_loader);
1469}
1470
1471PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1472{
1473 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1474}
1475
Martin v. Löwis18e16552006-02-15 17:27:45 +00001476/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001477 * The arguments here are tricky, best shown by example:
1478 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1479 * ^ ^ ^ ^
1480 * |--------------------- buf ---------------------|
1481 * |------------------- len ------------------|
1482 * |------ name -------|
1483 * |----- namelen -----|
1484 * buf is the full path, but len only counts up to (& exclusive of) the
1485 * extension. name is the module name, also exclusive of extension.
1486 *
1487 * We've already done a successful stat() or fopen() on buf, so know that
1488 * there's some match, possibly case-insensitive.
1489 *
Tim Peters50d8d372001-02-28 05:34:27 +00001490 * case_ok() is to return 1 if there's a case-sensitive match for
1491 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1492 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001493 *
Tim Peters50d8d372001-02-28 05:34:27 +00001494 * case_ok() is used to implement case-sensitive import semantics even
1495 * on platforms with case-insensitive filesystems. It's trivial to implement
1496 * for case-sensitive filesystems. It's pretty much a cross-platform
1497 * nightmare for systems with case-insensitive filesystems.
1498 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001499
Tim Peters50d8d372001-02-28 05:34:27 +00001500/* First we may need a pile of platform-specific header files; the sequence
1501 * of #if's here should match the sequence in the body of case_ok().
1502 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001503#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001504#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001505
Tim Peters50d8d372001-02-28 05:34:27 +00001506#elif defined(DJGPP)
1507#include <dir.h>
1508
Jason Tishler7961aa62005-05-20 00:56:54 +00001509#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001510#include <sys/types.h>
1511#include <dirent.h>
1512
Andrew MacIntyred9400542002-02-26 11:41:34 +00001513#elif defined(PYOS_OS2)
1514#define INCL_DOS
1515#define INCL_DOSERRORS
1516#define INCL_NOPMAPI
1517#include <os2.h>
Tim Peters50d8d372001-02-28 05:34:27 +00001518#endif
1519
Guido van Rossum0980bd91998-02-13 17:18:36 +00001520static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001521case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001522{
Tim Peters50d8d372001-02-28 05:34:27 +00001523/* Pick a platform-specific implementation; the sequence of #if's here should
1524 * match the sequence just above.
1525 */
1526
Jason Tishler7961aa62005-05-20 00:56:54 +00001527/* MS_WINDOWS */
1528#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001529 WIN32_FIND_DATA data;
1530 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001531
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001532 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001533 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001534
Guido van Rossum0980bd91998-02-13 17:18:36 +00001535 h = FindFirstFile(buf, &data);
1536 if (h == INVALID_HANDLE_VALUE) {
1537 PyErr_Format(PyExc_NameError,
1538 "Can't find file for module %.100s\n(filename %.300s)",
1539 name, buf);
1540 return 0;
1541 }
1542 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001543 return strncmp(data.cFileName, name, namelen) == 0;
1544
1545/* DJGPP */
1546#elif defined(DJGPP)
1547 struct ffblk ffblk;
1548 int done;
1549
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001550 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001551 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001552
1553 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1554 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001555 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001556 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001557 name, buf);
1558 return 0;
1559 }
Tim Peters50d8d372001-02-28 05:34:27 +00001560 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001561
Jason Tishler7961aa62005-05-20 00:56:54 +00001562/* new-fangled macintosh (macosx) or Cygwin */
1563#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001564 DIR *dirp;
1565 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001566 char dirname[MAXPATHLEN + 1];
1567 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001568
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001569 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001570 return 1;
1571
Tim Petersd1e87a82001-03-01 18:12:00 +00001572 /* Copy the dir component into dirname; substitute "." if empty */
1573 if (dirlen <= 0) {
1574 dirname[0] = '.';
1575 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001576 }
1577 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001578 assert(dirlen <= MAXPATHLEN);
1579 memcpy(dirname, buf, dirlen);
1580 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001581 }
1582 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001583 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001584 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001585 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001586 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001587 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001588#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001589 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001590#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001591 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001592#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001593 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001594 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001595 (void)closedir(dirp);
1596 return 1; /* Found */
1597 }
1598 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001599 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001600 }
Tim Peters430f5d42001-03-01 01:30:56 +00001601 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001602
Andrew MacIntyred9400542002-02-26 11:41:34 +00001603/* OS/2 */
1604#elif defined(PYOS_OS2)
1605 HDIR hdir = 1;
1606 ULONG srchcnt = 1;
1607 FILEFINDBUF3 ffbuf;
1608 APIRET rc;
1609
Christian Heimes790c8232008-01-07 21:14:23 +00001610 if (Py_GETENV("PYTHONCASEOK") != NULL)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001611 return 1;
1612
1613 rc = DosFindFirst(buf,
1614 &hdir,
1615 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1616 &ffbuf, sizeof(ffbuf),
1617 &srchcnt,
1618 FIL_STANDARD);
1619 if (rc != NO_ERROR)
1620 return 0;
1621 return strncmp(ffbuf.achName, name, namelen) == 0;
1622
Tim Peters50d8d372001-02-28 05:34:27 +00001623/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1624#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001625 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001626
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001627#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001628}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001629
Guido van Rossum0980bd91998-02-13 17:18:36 +00001630
Guido van Rossum197346f1997-10-31 18:38:52 +00001631#ifdef HAVE_STAT
1632/* Helper to look for __init__.py or __init__.py[co] in potential package */
1633static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001634find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001635{
Tim Peters0f9431f2001-07-05 03:47:53 +00001636 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001637 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001638 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001639 struct stat statbuf;
1640
Tim Peters0f9431f2001-07-05 03:47:53 +00001641/* For calling case_ok(buf, len, namelen, name):
1642 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1643 * ^ ^ ^ ^
1644 * |--------------------- buf ---------------------|
1645 * |------------------- len ------------------|
1646 * |------ name -------|
1647 * |----- namelen -----|
1648 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001649 if (save_len + 13 >= MAXPATHLEN)
1650 return 0;
1651 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001652 pname = buf + i;
1653 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001654 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001655 if (case_ok(buf,
1656 save_len + 9, /* len("/__init__") */
1657 8, /* len("__init__") */
1658 pname)) {
1659 buf[save_len] = '\0';
1660 return 1;
1661 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001662 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001663 i += strlen(pname);
1664 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001665 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001666 if (case_ok(buf,
1667 save_len + 9, /* len("/__init__") */
1668 8, /* len("__init__") */
1669 pname)) {
1670 buf[save_len] = '\0';
1671 return 1;
1672 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001673 }
1674 buf[save_len] = '\0';
1675 return 0;
1676}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001677
Guido van Rossum197346f1997-10-31 18:38:52 +00001678#endif /* HAVE_STAT */
1679
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001680
Tim Petersdbd9ba62000-07-09 03:09:57 +00001681static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001682
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001683/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001684 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001685
Guido van Rossum79f25d91997-04-29 20:08:16 +00001686static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001687load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001688{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001689 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001690 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001691 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001692
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001693 /* First check that there's an open file (if we need one) */
1694 switch (type) {
1695 case PY_SOURCE:
1696 case PY_COMPILED:
1697 if (fp == NULL) {
1698 PyErr_Format(PyExc_ValueError,
1699 "file object required for import (type code %d)",
1700 type);
1701 return NULL;
1702 }
1703 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001704
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001705 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001706
1707 case PY_SOURCE:
1708 m = load_source_module(name, buf, fp);
1709 break;
1710
1711 case PY_COMPILED:
1712 m = load_compiled_module(name, buf, fp);
1713 break;
1714
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001715#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001716 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001717 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001718 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001719#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001720
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001721 case PKG_DIRECTORY:
1722 m = load_package(name, buf);
1723 break;
1724
1725 case C_BUILTIN:
1726 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001727 if (buf != NULL && buf[0] != '\0')
1728 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001729 if (type == C_BUILTIN)
1730 err = init_builtin(name);
1731 else
1732 err = PyImport_ImportFrozenModule(name);
1733 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001734 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001735 if (err == 0) {
1736 PyErr_Format(PyExc_ImportError,
1737 "Purported %s module %.200s not found",
1738 type == C_BUILTIN ?
1739 "builtin" : "frozen",
1740 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001741 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001742 }
1743 modules = PyImport_GetModuleDict();
1744 m = PyDict_GetItemString(modules, name);
1745 if (m == NULL) {
1746 PyErr_Format(
1747 PyExc_ImportError,
1748 "%s module %.200s not properly initialized",
1749 type == C_BUILTIN ?
1750 "builtin" : "frozen",
1751 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001752 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001753 }
1754 Py_INCREF(m);
1755 break;
1756
Just van Rossum52e14d62002-12-30 22:08:05 +00001757 case IMP_HOOK: {
1758 if (loader == NULL) {
1759 PyErr_SetString(PyExc_ImportError,
1760 "import hook without loader");
1761 return NULL;
1762 }
1763 m = PyObject_CallMethod(loader, "load_module", "s", name);
1764 break;
1765 }
1766
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001767 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001768 PyErr_Format(PyExc_ImportError,
1769 "Don't know how to import %.200s (type code %d)",
1770 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001771 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001772
1773 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001774
1775 return m;
1776}
1777
1778
1779/* Initialize a built-in module.
Thomas Wouters89f507f2006-12-13 04:49:30 +00001780 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001781 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001782
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001783static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001784init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001785{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001786 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001787
Greg Ward201baee2001-10-04 14:52:06 +00001788 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001789 return 1;
1790
Guido van Rossum771c6c81997-10-31 18:37:24 +00001791 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001792 if (strcmp(name, p->name) == 0) {
1793 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001794 PyErr_Format(PyExc_ImportError,
1795 "Cannot re-init internal module %.200s",
1796 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001797 return -1;
1798 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001799 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001800 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001801 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001802 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001803 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001804 if (_PyImport_FixupExtension(name, name) == NULL)
1805 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001806 return 1;
1807 }
1808 }
1809 return 0;
1810}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001811
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001812
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001813/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001814
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001815static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001816find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001817{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001818 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001819
Guido van Rossum79f25d91997-04-29 20:08:16 +00001820 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001821 if (p->name == NULL)
1822 return NULL;
1823 if (strcmp(p->name, name) == 0)
1824 break;
1825 }
1826 return p;
1827}
1828
Guido van Rossum79f25d91997-04-29 20:08:16 +00001829static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001830get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001831{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001832 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001833 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001834
1835 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001836 PyErr_Format(PyExc_ImportError,
1837 "No such frozen object named %.200s",
1838 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001839 return NULL;
1840 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001841 if (p->code == NULL) {
1842 PyErr_Format(PyExc_ImportError,
1843 "Excluded frozen object named %.200s",
1844 name);
1845 return NULL;
1846 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001847 size = p->size;
1848 if (size < 0)
1849 size = -size;
1850 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001851}
1852
1853/* Initialize a frozen module.
1854 Return 1 for succes, 0 if the module is not found, and -1 with
1855 an exception set if the initialization failed.
1856 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001857
1858int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001859PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001860{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001861 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001862 PyObject *co;
1863 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001864 int ispackage;
1865 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001866
1867 if (p == NULL)
1868 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001869 if (p->code == NULL) {
1870 PyErr_Format(PyExc_ImportError,
1871 "Excluded frozen object named %.200s",
1872 name);
1873 return -1;
1874 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001875 size = p->size;
1876 ispackage = (size < 0);
1877 if (ispackage)
1878 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001879 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001880 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001881 name, ispackage ? " package" : "");
1882 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001883 if (co == NULL)
1884 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001885 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001886 PyErr_Format(PyExc_TypeError,
1887 "frozen object %.200s is not a code object",
1888 name);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001889 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001890 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001891 if (ispackage) {
1892 /* Set __path__ to the package name */
1893 PyObject *d, *s;
1894 int err;
1895 m = PyImport_AddModule(name);
1896 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001897 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001898 d = PyModule_GetDict(m);
Martin v. Löwis5b222132007-06-10 09:51:05 +00001899 s = PyUnicode_InternFromString(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001900 if (s == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001901 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001902 err = PyDict_SetItemString(d, "__path__", s);
1903 Py_DECREF(s);
1904 if (err != 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001905 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001906 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001907 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001908 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001909 goto err_return;
1910 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001911 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001912 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001913err_return:
1914 Py_DECREF(co);
1915 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001916}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001917
1918
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001919/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001920 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001921
Guido van Rossum79f25d91997-04-29 20:08:16 +00001922PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001923PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001924{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001925 PyObject *pname;
1926 PyObject *result;
1927
Martin v. Löwis5b222132007-06-10 09:51:05 +00001928 pname = PyUnicode_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001929 if (pname == NULL)
1930 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001931 result = PyImport_Import(pname);
1932 Py_DECREF(pname);
1933 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001934}
1935
Christian Heimes072c0f12008-01-03 23:01:04 +00001936/* Import a module without blocking
1937 *
1938 * At first it tries to fetch the module from sys.modules. If the module was
1939 * never loaded before it loads it with PyImport_ImportModule() unless another
1940 * thread holds the import lock. In the latter case the function raises an
1941 * ImportError instead of blocking.
1942 *
1943 * Returns the module object with incremented ref count.
1944 */
1945PyObject *
1946PyImport_ImportModuleNoBlock(const char *name)
1947{
1948 PyObject *result;
1949 PyObject *modules;
1950 long me;
1951
1952 /* Try to get the module from sys.modules[name] */
1953 modules = PyImport_GetModuleDict();
1954 if (modules == NULL)
1955 return NULL;
1956
1957 result = PyDict_GetItemString(modules, name);
1958 if (result != NULL) {
1959 Py_INCREF(result);
1960 return result;
1961 }
1962 else {
1963 PyErr_Clear();
1964 }
1965
1966 /* check the import lock
1967 * me might be -1 but I ignore the error here, the lock function
1968 * takes care of the problem */
1969 me = PyThread_get_thread_ident();
1970 if (import_lock_thread == -1 || import_lock_thread == me) {
1971 /* no thread or me is holding the lock */
1972 return PyImport_ImportModule(name);
1973 }
1974 else {
1975 PyErr_Format(PyExc_ImportError,
1976 "Failed to import %.200s because the import lock"
1977 "is held by another thread.",
1978 name);
1979 return NULL;
1980 }
1981}
1982
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001983/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001984static PyObject *get_parent(PyObject *globals, char *buf,
1985 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001986static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001987 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001988static int mark_miss(char *name);
1989static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001990 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001991static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001992
1993/* The Magnum Opus of dotted-name import :-) */
1994
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001995static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001996import_module_level(char *name, PyObject *globals, PyObject *locals,
1997 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001998{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001999 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00002000 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002001 PyObject *parent, *head, *next, *tail;
2002
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002003 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002004 if (parent == NULL)
2005 return NULL;
2006
2007 head = load_next(parent, Py_None, &name, buf, &buflen);
2008 if (head == NULL)
2009 return NULL;
2010
2011 tail = head;
2012 Py_INCREF(tail);
2013 while (name) {
2014 next = load_next(tail, tail, &name, buf, &buflen);
2015 Py_DECREF(tail);
2016 if (next == NULL) {
2017 Py_DECREF(head);
2018 return NULL;
2019 }
2020 tail = next;
2021 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002022 if (tail == Py_None) {
2023 /* If tail is Py_None, both get_parent and load_next found
2024 an empty module name: someone called __import__("") or
2025 doctored faulty bytecode */
2026 Py_DECREF(tail);
2027 Py_DECREF(head);
2028 PyErr_SetString(PyExc_ValueError,
2029 "Empty module name");
2030 return NULL;
2031 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002032
2033 if (fromlist != NULL) {
2034 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2035 fromlist = NULL;
2036 }
2037
2038 if (fromlist == NULL) {
2039 Py_DECREF(tail);
2040 return head;
2041 }
2042
2043 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002044 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002045 Py_DECREF(tail);
2046 return NULL;
2047 }
2048
2049 return tail;
2050}
2051
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002052PyObject *
2053PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
2054 PyObject *fromlist, int level)
2055{
2056 PyObject *result;
2057 lock_import();
2058 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002059 if (unlock_import() < 0) {
2060 Py_XDECREF(result);
2061 PyErr_SetString(PyExc_RuntimeError,
2062 "not holding the import lock");
2063 return NULL;
2064 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002065 return result;
2066}
2067
Fred Drake87590902004-05-28 20:21:36 +00002068/* Return the package that an import is being performed in. If globals comes
2069 from the module foo.bar.bat (not itself a package), this returns the
2070 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002071 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002072
2073 The *name* of the returned package is returned in buf, with the length of
2074 the name in *p_buflen.
2075
2076 If globals doesn't come from a package or a module in a package, or a
2077 corresponding entry is not found in sys.modules, Py_None is returned.
2078*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002079static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002080get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002081{
2082 static PyObject *namestr = NULL;
2083 static PyObject *pathstr = NULL;
Nick Coghlande10c852007-12-04 12:22:52 +00002084 static PyObject *pkgstr = NULL;
2085 PyObject *pkgname, *modname, *modpath, *modules, *parent;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002086
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002087 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002088 return Py_None;
2089
2090 if (namestr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002091 namestr = PyUnicode_InternFromString("__name__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002092 if (namestr == NULL)
2093 return NULL;
2094 }
2095 if (pathstr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002096 pathstr = PyUnicode_InternFromString("__path__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002097 if (pathstr == NULL)
2098 return NULL;
2099 }
Nick Coghlande10c852007-12-04 12:22:52 +00002100 if (pkgstr == NULL) {
2101 pkgstr = PyUnicode_InternFromString("__package__");
2102 if (pkgstr == NULL)
2103 return NULL;
2104 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002105
2106 *buf = '\0';
2107 *p_buflen = 0;
Nick Coghlande10c852007-12-04 12:22:52 +00002108 pkgname = PyDict_GetItem(globals, pkgstr);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002109
Nick Coghlande10c852007-12-04 12:22:52 +00002110 if ((pkgname != NULL) && (pkgname != Py_None)) {
2111 /* __package__ is set, so use it */
2112 Py_ssize_t len;
2113 if (!PyUnicode_Check(pkgname)) {
2114 PyErr_SetString(PyExc_ValueError,
2115 "__package__ set to non-string");
2116 return NULL;
2117 }
2118 len = PyUnicode_GET_SIZE(pkgname);
2119 if (len == 0) {
2120 if (level > 0) {
2121 PyErr_SetString(PyExc_ValueError,
2122 "Attempted relative import in non-package");
2123 return NULL;
2124 }
2125 return Py_None;
2126 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002127 if (len > MAXPATHLEN) {
2128 PyErr_SetString(PyExc_ValueError,
Nick Coghlande10c852007-12-04 12:22:52 +00002129 "Package name too long");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002130 return NULL;
2131 }
Nick Coghlande10c852007-12-04 12:22:52 +00002132 strcpy(buf, PyUnicode_AsString(pkgname));
2133 } else {
2134 /* __package__ not set, so figure it out and set it */
2135 modname = PyDict_GetItem(globals, namestr);
2136 if (modname == NULL || !PyUnicode_Check(modname))
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002137 return Py_None;
Nick Coghlande10c852007-12-04 12:22:52 +00002138
2139 modpath = PyDict_GetItem(globals, pathstr);
2140 if (modpath != NULL) {
2141 /* __path__ is set, so modname is already the package name */
2142 Py_ssize_t len = PyUnicode_GET_SIZE(modname);
2143 int error;
2144 if (len > MAXPATHLEN) {
2145 PyErr_SetString(PyExc_ValueError,
2146 "Module name too long");
2147 return NULL;
2148 }
2149 strcpy(buf, PyUnicode_AsString(modname));
2150 error = PyDict_SetItem(globals, pkgstr, modname);
2151 if (error) {
2152 PyErr_SetString(PyExc_ValueError,
2153 "Could not set __package__");
2154 return NULL;
2155 }
2156 } else {
2157 /* Normal module, so work out the package name if any */
2158 char *start = PyUnicode_AsString(modname);
2159 char *lastdot = strrchr(start, '.');
2160 size_t len;
2161 int error;
2162 if (lastdot == NULL && level > 0) {
2163 PyErr_SetString(PyExc_ValueError,
2164 "Attempted relative import in non-package");
2165 return NULL;
2166 }
2167 if (lastdot == NULL) {
2168 error = PyDict_SetItem(globals, pkgstr, Py_None);
2169 if (error) {
2170 PyErr_SetString(PyExc_ValueError,
2171 "Could not set __package__");
2172 return NULL;
2173 }
2174 return Py_None;
2175 }
2176 len = lastdot - start;
2177 if (len >= MAXPATHLEN) {
2178 PyErr_SetString(PyExc_ValueError,
2179 "Module name too long");
2180 return NULL;
2181 }
2182 strncpy(buf, start, len);
2183 buf[len] = '\0';
2184 pkgname = PyUnicode_FromString(buf);
2185 if (pkgname == NULL) {
2186 return NULL;
2187 }
2188 error = PyDict_SetItem(globals, pkgstr, pkgname);
2189 Py_DECREF(pkgname);
2190 if (error) {
2191 PyErr_SetString(PyExc_ValueError,
2192 "Could not set __package__");
2193 return NULL;
2194 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002195 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002196 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002197 while (--level > 0) {
2198 char *dot = strrchr(buf, '.');
2199 if (dot == NULL) {
2200 PyErr_SetString(PyExc_ValueError,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002201 "Attempted relative import beyond "
2202 "toplevel package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002203 return NULL;
2204 }
2205 *dot = '\0';
2206 }
2207 *p_buflen = strlen(buf);
2208
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002209 modules = PyImport_GetModuleDict();
2210 parent = PyDict_GetItemString(modules, buf);
2211 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002212 PyErr_Format(PyExc_SystemError,
2213 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002214 return parent;
2215 /* We expect, but can't guarantee, if parent != None, that:
2216 - parent.__name__ == buf
2217 - parent.__dict__ is globals
2218 If this is violated... Who cares? */
2219}
2220
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002221/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002222static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002223load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002224 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002225{
2226 char *name = *p_name;
2227 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002228 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002229 char *p;
2230 PyObject *result;
2231
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002232 if (strlen(name) == 0) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002233 /* completely empty module name should only happen in
2234 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002235 Py_INCREF(mod);
2236 *p_name = NULL;
2237 return mod;
2238 }
2239
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002240 if (dot == NULL) {
2241 *p_name = NULL;
2242 len = strlen(name);
2243 }
2244 else {
2245 *p_name = dot+1;
2246 len = dot-name;
2247 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002248 if (len == 0) {
2249 PyErr_SetString(PyExc_ValueError,
2250 "Empty module name");
2251 return NULL;
2252 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002253
2254 p = buf + *p_buflen;
2255 if (p != buf)
2256 *p++ = '.';
2257 if (p+len-buf >= MAXPATHLEN) {
2258 PyErr_SetString(PyExc_ValueError,
2259 "Module name too long");
2260 return NULL;
2261 }
2262 strncpy(p, name, len);
2263 p[len] = '\0';
2264 *p_buflen = p+len-buf;
2265
2266 result = import_submodule(mod, p, buf);
2267 if (result == Py_None && altmod != mod) {
2268 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002269 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002270 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002271 if (result != NULL && result != Py_None) {
2272 if (mark_miss(buf) != 0) {
2273 Py_DECREF(result);
2274 return NULL;
2275 }
2276 strncpy(buf, name, len);
2277 buf[len] = '\0';
2278 *p_buflen = len;
2279 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002280 }
2281 if (result == NULL)
2282 return NULL;
2283
2284 if (result == Py_None) {
2285 Py_DECREF(result);
2286 PyErr_Format(PyExc_ImportError,
2287 "No module named %.200s", name);
2288 return NULL;
2289 }
2290
2291 return result;
2292}
2293
2294static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002295mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002296{
2297 PyObject *modules = PyImport_GetModuleDict();
2298 return PyDict_SetItemString(modules, name, Py_None);
2299}
2300
2301static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002302ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002303 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002304{
2305 int i;
2306
2307 if (!PyObject_HasAttrString(mod, "__path__"))
2308 return 1;
2309
2310 for (i = 0; ; i++) {
2311 PyObject *item = PySequence_GetItem(fromlist, i);
2312 int hasit;
2313 if (item == NULL) {
2314 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2315 PyErr_Clear();
2316 return 1;
2317 }
2318 return 0;
2319 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002320 if (!PyUnicode_Check(item)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002321 PyErr_SetString(PyExc_TypeError,
Neal Norwitz80e7f272007-08-26 06:45:23 +00002322 "Item in ``from list'' not a string");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002323 Py_DECREF(item);
2324 return 0;
2325 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002326 if (PyUnicode_AS_UNICODE(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002327 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002328 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002329 /* See if the package defines __all__ */
2330 if (recursive)
2331 continue; /* Avoid endless recursion */
2332 all = PyObject_GetAttrString(mod, "__all__");
2333 if (all == NULL)
2334 PyErr_Clear();
2335 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002336 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002337 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002338 if (!ret)
2339 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002340 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002341 continue;
2342 }
2343 hasit = PyObject_HasAttr(mod, item);
2344 if (!hasit) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002345 PyObject *item8;
2346 char *subname;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002347 PyObject *submod;
2348 char *p;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002349 if (!Py_FileSystemDefaultEncoding) {
2350 item8 = PyUnicode_EncodeASCII(PyUnicode_AsUnicode(item),
2351 PyUnicode_GetSize(item),
Martin v. Löwis427dbff2007-06-12 05:53:00 +00002352 NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002353 } else {
Guido van Rossum98297ee2007-11-06 21:34:58 +00002354 item8 = PyUnicode_AsEncodedString(item,
2355 Py_FileSystemDefaultEncoding, NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002356 }
2357 if (!item8) {
2358 PyErr_SetString(PyExc_ValueError, "Cannot encode path item");
2359 return 0;
2360 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00002361 subname = PyString_AS_STRING(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002362 if (buflen + strlen(subname) >= MAXPATHLEN) {
2363 PyErr_SetString(PyExc_ValueError,
2364 "Module name too long");
2365 Py_DECREF(item);
2366 return 0;
2367 }
2368 p = buf + buflen;
2369 *p++ = '.';
2370 strcpy(p, subname);
2371 submod = import_submodule(mod, subname, buf);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002372 Py_DECREF(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002373 Py_XDECREF(submod);
2374 if (submod == NULL) {
2375 Py_DECREF(item);
2376 return 0;
2377 }
2378 }
2379 Py_DECREF(item);
2380 }
2381
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002382 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002383}
2384
Neil Schemenauer00b09662003-06-16 21:03:07 +00002385static int
2386add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2387 PyObject *modules)
2388{
2389 if (mod == Py_None)
2390 return 1;
2391 /* Irrespective of the success of this load, make a
2392 reference to it in the parent package module. A copy gets
2393 saved in the modules dictionary under the full name, so get a
2394 reference from there, if need be. (The exception is when the
2395 load failed with a SyntaxError -- then there's no trace in
2396 sys.modules. In that case, of course, do nothing extra.) */
2397 if (submod == NULL) {
2398 submod = PyDict_GetItemString(modules, fullname);
2399 if (submod == NULL)
2400 return 1;
2401 }
2402 if (PyModule_Check(mod)) {
2403 /* We can't use setattr here since it can give a
2404 * spurious warning if the submodule name shadows a
2405 * builtin name */
2406 PyObject *dict = PyModule_GetDict(mod);
2407 if (!dict)
2408 return 0;
2409 if (PyDict_SetItemString(dict, subname, submod) < 0)
2410 return 0;
2411 }
2412 else {
2413 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2414 return 0;
2415 }
2416 return 1;
2417}
2418
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002419static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002420import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002421{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002422 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002423 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002424
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002425 /* Require:
2426 if mod == None: subname == fullname
2427 else: mod.__name__ + "." + subname == fullname
2428 */
2429
Tim Peters50d8d372001-02-28 05:34:27 +00002430 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002431 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002432 }
2433 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002434 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002435 char buf[MAXPATHLEN+1];
2436 struct filedescr *fdp;
2437 FILE *fp = NULL;
2438
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002439 if (mod == Py_None)
2440 path = NULL;
2441 else {
2442 path = PyObject_GetAttrString(mod, "__path__");
2443 if (path == NULL) {
2444 PyErr_Clear();
2445 Py_INCREF(Py_None);
2446 return Py_None;
2447 }
2448 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002449
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002450 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002451 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2452 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002453 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002454 if (fdp == NULL) {
2455 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2456 return NULL;
2457 PyErr_Clear();
2458 Py_INCREF(Py_None);
2459 return Py_None;
2460 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002461 m = load_module(fullname, fp, buf, fdp->type, loader);
2462 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002463 if (fp)
2464 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002465 if (!add_submodule(mod, m, fullname, subname, modules)) {
2466 Py_XDECREF(m);
2467 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002468 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002469 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002470
2471 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002472}
2473
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002474
2475/* Re-import a module of any kind and return its module object, WITH
2476 INCREMENTED REFERENCE COUNT */
2477
Guido van Rossum79f25d91997-04-29 20:08:16 +00002478PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002479PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002480{
Guido van Rossumd8faa362007-04-27 19:54:29 +00002481 PyInterpreterState *interp = PyThreadState_Get()->interp;
2482 PyObject *modules_reloading = interp->modules_reloading;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002483 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumd8faa362007-04-27 19:54:29 +00002484 PyObject *path = NULL, *loader = NULL, *existing_m = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002485 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002486 char buf[MAXPATHLEN+1];
2487 struct filedescr *fdp;
2488 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002489 PyObject *newm;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002490
2491 if (modules_reloading == NULL) {
2492 Py_FatalError("PyImport_ReloadModule: "
Guido van Rossume7ba4952007-06-06 23:52:48 +00002493 "no modules_reloading dictionary!");
Guido van Rossumd8faa362007-04-27 19:54:29 +00002494 return NULL;
2495 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002496
Guido van Rossum79f25d91997-04-29 20:08:16 +00002497 if (m == NULL || !PyModule_Check(m)) {
2498 PyErr_SetString(PyExc_TypeError,
2499 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002500 return NULL;
2501 }
Neal Norwitz5616c6d2007-08-26 05:32:41 +00002502 name = (char*)PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002503 if (name == NULL)
2504 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002505 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002506 PyErr_Format(PyExc_ImportError,
2507 "reload(): module %.200s not in sys.modules",
2508 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002509 return NULL;
2510 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002511 existing_m = PyDict_GetItemString(modules_reloading, name);
2512 if (existing_m != NULL) {
2513 /* Due to a recursive reload, this module is already
2514 being reloaded. */
2515 Py_INCREF(existing_m);
2516 return existing_m;
2517 }
2518 if (PyDict_SetItemString(modules_reloading, name, m) < 0)
2519 return NULL;
2520
Guido van Rossum222ef561997-09-06 19:41:09 +00002521 subname = strrchr(name, '.');
2522 if (subname == NULL)
2523 subname = name;
2524 else {
2525 PyObject *parentname, *parent;
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002526 parentname = PyUnicode_FromStringAndSize(name, (subname-name));
Guido van Rossumd8faa362007-04-27 19:54:29 +00002527 if (parentname == NULL) {
2528 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002529 return NULL;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002530 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002531 parent = PyDict_GetItem(modules, parentname);
2532 if (parent == NULL) {
2533 PyErr_Format(PyExc_ImportError,
2534 "reload(): parent %.200s not in sys.modules",
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002535 PyUnicode_AsString(parentname));
Georg Brandl0c55f292005-09-14 06:56:20 +00002536 Py_DECREF(parentname);
Guido van Rossume7ba4952007-06-06 23:52:48 +00002537 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002538 return NULL;
2539 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002540 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002541 subname++;
2542 path = PyObject_GetAttrString(parent, "__path__");
2543 if (path == NULL)
2544 PyErr_Clear();
2545 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002546 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002547 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002548 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002549
2550 if (fdp == NULL) {
2551 Py_XDECREF(loader);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002552 imp_modules_reloading_clear();
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002553 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002554 }
2555
2556 newm = load_module(name, fp, buf, fdp->type, loader);
2557 Py_XDECREF(loader);
2558
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002559 if (fp)
2560 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002561 if (newm == NULL) {
2562 /* load_module probably removed name from modules because of
2563 * the error. Put back the original module object. We're
2564 * going to return NULL in this case regardless of whether
2565 * replacing name succeeds, so the return value is ignored.
2566 */
2567 PyDict_SetItemString(modules, name, m);
2568 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002569 imp_modules_reloading_clear();
Tim Peters1cd70172004-08-02 03:52:12 +00002570 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002571}
2572
2573
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002574/* Higher-level import emulator which emulates the "import" statement
2575 more accurately -- it invokes the __import__() function from the
2576 builtins of the current globals. This means that the import is
2577 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002578 environment, e.g. by "rexec".
2579 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00002580 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00002581 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002582
2583PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002584PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002585{
2586 static PyObject *silly_list = NULL;
2587 static PyObject *builtins_str = NULL;
2588 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002589 PyObject *globals = NULL;
2590 PyObject *import = NULL;
2591 PyObject *builtins = NULL;
2592 PyObject *r = NULL;
2593
2594 /* Initialize constant string objects */
2595 if (silly_list == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002596 import_str = PyUnicode_InternFromString("__import__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002597 if (import_str == NULL)
2598 return NULL;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002599 builtins_str = PyUnicode_InternFromString("__builtins__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002600 if (builtins_str == NULL)
2601 return NULL;
2602 silly_list = Py_BuildValue("[s]", "__doc__");
2603 if (silly_list == NULL)
2604 return NULL;
2605 }
2606
2607 /* Get the builtins from current globals */
2608 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002609 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002610 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002611 builtins = PyObject_GetItem(globals, builtins_str);
2612 if (builtins == NULL)
2613 goto err;
2614 }
2615 else {
2616 /* No globals -- use standard builtins, and fake globals */
2617 PyErr_Clear();
2618
Georg Brandl1a3284e2007-12-02 09:40:06 +00002619 builtins = PyImport_ImportModuleLevel("builtins",
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002620 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002621 if (builtins == NULL)
2622 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002623 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2624 if (globals == NULL)
2625 goto err;
2626 }
2627
2628 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002629 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002630 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002631 if (import == NULL)
2632 PyErr_SetObject(PyExc_KeyError, import_str);
2633 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002634 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002635 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002636 if (import == NULL)
2637 goto err;
2638
Christian Heimes072c0f12008-01-03 23:01:04 +00002639 /* Call the __import__ function with the proper argument list
2640 * Always use absolute import here. */
2641 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
2642 globals, silly_list, 0, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002643
2644 err:
2645 Py_XDECREF(globals);
2646 Py_XDECREF(builtins);
2647 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002648
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002649 return r;
2650}
2651
2652
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002653/* Module 'imp' provides Python access to the primitives used for
2654 importing modules.
2655*/
2656
Guido van Rossum79f25d91997-04-29 20:08:16 +00002657static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002658imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002659{
2660 char buf[4];
2661
Guido van Rossum96774c12000-05-01 20:19:08 +00002662 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2663 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2664 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2665 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002666
Guido van Rossumad8d3002007-08-03 18:40:49 +00002667 return PyBytes_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002668}
2669
Guido van Rossum79f25d91997-04-29 20:08:16 +00002670static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002671imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002672{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002673 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002674 struct filedescr *fdp;
2675
Guido van Rossum79f25d91997-04-29 20:08:16 +00002676 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002677 if (list == NULL)
2678 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002679 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2680 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002681 fdp->suffix, fdp->mode, fdp->type);
2682 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002683 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002684 return NULL;
2685 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002686 if (PyList_Append(list, item) < 0) {
2687 Py_DECREF(list);
2688 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002689 return NULL;
2690 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002691 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002692 }
2693 return list;
2694}
2695
Guido van Rossum79f25d91997-04-29 20:08:16 +00002696static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002697call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002698{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002699 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002700 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002701 struct filedescr *fdp;
2702 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002703 FILE *fp = NULL;
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002704 int fd = -1;
Brett Cannon3bb42d92007-10-20 03:43:15 +00002705 char *found_encoding = NULL;
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002706 char *encoding = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002707
2708 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002709 if (path == Py_None)
2710 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002711 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002712 if (fdp == NULL)
2713 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002714 if (fp != NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002715 fd = fileno(fp);
2716 if (fd != -1)
2717 fd = dup(fd);
2718 fclose(fp);
2719 fp = NULL;
2720 }
2721 if (fd != -1) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002722 if (strchr(fdp->mode, 'b') == NULL) {
Brett Cannon3bb42d92007-10-20 03:43:15 +00002723 /* PyTokenizer_FindEncoding() returns PyMem_MALLOC'ed
2724 memory. */
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002725 found_encoding = PyTokenizer_FindEncoding(fd);
2726 lseek(fd, 0, 0); /* Reset position */
Brett Cannon3bb42d92007-10-20 03:43:15 +00002727 encoding = (found_encoding != NULL) ? found_encoding :
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002728 (char*)PyUnicode_GetDefaultEncoding();
2729 }
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002730 fob = PyFile_FromFd(fd, pathname, fdp->mode, -1,
Guido van Rossume7fc50f2007-12-03 22:54:21 +00002731 (char*)encoding, NULL, NULL, 1);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002732 if (fob == NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002733 close(fd);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002734 PyMem_FREE(found_encoding);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002735 return NULL;
2736 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002737 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002738 else {
2739 fob = Py_None;
2740 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002741 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002742 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002743 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002744 Py_DECREF(fob);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002745 PyMem_FREE(found_encoding);
2746
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002747 return ret;
2748}
2749
Guido van Rossum79f25d91997-04-29 20:08:16 +00002750static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002751imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002752{
2753 char *name;
2754 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002755 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002756 return NULL;
2757 return call_find_module(name, path);
2758}
2759
2760static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002761imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002762{
2763 char *name;
2764 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002765 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002766 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002767 return NULL;
2768 ret = init_builtin(name);
2769 if (ret < 0)
2770 return NULL;
2771 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002772 Py_INCREF(Py_None);
2773 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002774 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002775 m = PyImport_AddModule(name);
2776 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002777 return m;
2778}
2779
Guido van Rossum79f25d91997-04-29 20:08:16 +00002780static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002781imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002782{
2783 char *name;
2784 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002785 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002786 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002787 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002788 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002789 if (ret < 0)
2790 return NULL;
2791 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002792 Py_INCREF(Py_None);
2793 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002794 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002795 m = PyImport_AddModule(name);
2796 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002797 return m;
2798}
2799
Guido van Rossum79f25d91997-04-29 20:08:16 +00002800static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002801imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002802{
2803 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002804
Guido van Rossum43713e52000-02-29 13:59:29 +00002805 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002806 return NULL;
2807 return get_frozen_object(name);
2808}
2809
Guido van Rossum79f25d91997-04-29 20:08:16 +00002810static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002811imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002812{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002813 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002814 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002815 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00002816 return PyLong_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002817}
2818
Guido van Rossum79f25d91997-04-29 20:08:16 +00002819static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002820imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002821{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002822 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002823 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002824 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002825 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002826 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002827 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002828}
2829
2830static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002831get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002832{
2833 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002834 if (mode[0] == 'U')
2835 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002836 if (fob == NULL) {
2837 fp = fopen(pathname, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002838 }
2839 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002840 int fd = PyObject_AsFileDescriptor(fob);
2841 if (fd == -1)
2842 return NULL;
2843 /* XXX This will leak a FILE struct. Fix this!!!!
2844 (But it doesn't leak a file descrioptor!) */
2845 fp = fdopen(fd, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002846 }
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002847 if (fp == NULL)
2848 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002849 return fp;
2850}
2851
Guido van Rossum79f25d91997-04-29 20:08:16 +00002852static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002853imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002854{
2855 char *name;
2856 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002857 PyObject *fob = NULL;
2858 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002859 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002860 if (!PyArg_ParseTuple(args, "ss|O:load_compiled",
2861 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002862 return NULL;
2863 fp = get_file(pathname, fob, "rb");
2864 if (fp == NULL)
2865 return NULL;
2866 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002867 if (fob == NULL)
2868 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002869 return m;
2870}
2871
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002872#ifdef HAVE_DYNAMIC_LOADING
2873
Guido van Rossum79f25d91997-04-29 20:08:16 +00002874static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002875imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002876{
2877 char *name;
2878 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002879 PyObject *fob = NULL;
2880 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002881 FILE *fp = NULL;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002882 if (!PyArg_ParseTuple(args, "ss|O:load_dynamic",
2883 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002884 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002885 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002886 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002887 if (fp == NULL)
2888 return NULL;
2889 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002890 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002891 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002892}
2893
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002894#endif /* HAVE_DYNAMIC_LOADING */
2895
Guido van Rossum79f25d91997-04-29 20:08:16 +00002896static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002897imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002898{
2899 char *name;
2900 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002901 PyObject *fob = NULL;
2902 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002903 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002904 if (!PyArg_ParseTuple(args, "ss|O:load_source",
2905 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002906 return NULL;
2907 fp = get_file(pathname, fob, "r");
2908 if (fp == NULL)
2909 return NULL;
2910 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002911 if (fob == NULL)
2912 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002913 return m;
2914}
2915
Guido van Rossum79f25d91997-04-29 20:08:16 +00002916static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002917imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002918{
2919 char *name;
2920 PyObject *fob;
2921 char *pathname;
2922 char *suffix; /* Unused */
2923 char *mode;
2924 int type;
2925 FILE *fp;
2926
Guido van Rossum43713e52000-02-29 13:59:29 +00002927 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002928 &name, &fob, &pathname,
2929 &suffix, &mode, &type))
2930 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002931 if (*mode) {
2932 /* Mode must start with 'r' or 'U' and must not contain '+'.
2933 Implicit in this test is the assumption that the mode
2934 may contain other modifiers like 'b' or 't'. */
2935
2936 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002937 PyErr_Format(PyExc_ValueError,
2938 "invalid file open mode %.200s", mode);
2939 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002940 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002941 }
2942 if (fob == Py_None)
2943 fp = NULL;
2944 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002945 fp = get_file(NULL, fob, mode);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002946 if (fp == NULL)
2947 return NULL;
2948 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002949 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002950}
2951
2952static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002953imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002954{
2955 char *name;
2956 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002957 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002958 return NULL;
2959 return load_package(name, pathname);
2960}
2961
2962static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002963imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002964{
2965 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002966 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002967 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002968 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002969}
2970
Christian Heimes13a7a212008-01-07 17:13:09 +00002971static PyObject *
2972imp_reload(PyObject *self, PyObject *v)
2973{
2974 return PyImport_ReloadModule(v);
2975}
2976
2977PyDoc_STRVAR(doc_reload,
2978"reload(module) -> module\n\
2979\n\
2980Reload the module. The module must have been successfully imported before.");
2981
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002982/* Doc strings */
2983
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002984PyDoc_STRVAR(doc_imp,
2985"This module provides the components needed to build your own\n\
2986__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002987
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002988PyDoc_STRVAR(doc_find_module,
2989"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002990Search for a module. If path is omitted or None, search for a\n\
2991built-in, frozen or special module and continue search in sys.path.\n\
2992The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002993package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002994
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002995PyDoc_STRVAR(doc_load_module,
2996"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002997Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002998The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002999
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003000PyDoc_STRVAR(doc_get_magic,
3001"get_magic() -> string\n\
3002Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003003
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003004PyDoc_STRVAR(doc_get_suffixes,
3005"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003006Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003007that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003008
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003009PyDoc_STRVAR(doc_new_module,
3010"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003011Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003012The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003013
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003014PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00003015"lock_held() -> boolean\n\
3016Return True if the import lock is currently held, else False.\n\
3017On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00003018
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003019PyDoc_STRVAR(doc_acquire_lock,
3020"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00003021Acquires the interpreter's import lock for the current thread.\n\
3022This lock should be used by import hooks to ensure thread-safety\n\
3023when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003024On platforms without threads, this function does nothing.");
3025
3026PyDoc_STRVAR(doc_release_lock,
3027"release_lock() -> None\n\
3028Release the interpreter's import lock.\n\
3029On platforms without threads, this function does nothing.");
3030
Guido van Rossum79f25d91997-04-29 20:08:16 +00003031static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00003032 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
3033 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
3034 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
3035 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
3036 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
3037 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
3038 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
3039 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Christian Heimes13a7a212008-01-07 17:13:09 +00003040 {"reload", imp_reload, METH_O, doc_reload},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003041 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00003042 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
3043 {"init_builtin", imp_init_builtin, METH_VARARGS},
3044 {"init_frozen", imp_init_frozen, METH_VARARGS},
3045 {"is_builtin", imp_is_builtin, METH_VARARGS},
3046 {"is_frozen", imp_is_frozen, METH_VARARGS},
3047 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003048#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00003049 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003050#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00003051 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00003052 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003053 {NULL, NULL} /* sentinel */
3054};
3055
Guido van Rossum1a8791e1998-08-04 22:46:29 +00003056static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003057setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003058{
3059 PyObject *v;
3060 int err;
3061
Christian Heimes217cfd12007-12-02 14:31:20 +00003062 v = PyLong_FromLong((long)value);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003063 err = PyDict_SetItemString(d, name, v);
3064 Py_XDECREF(v);
3065 return err;
3066}
3067
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003068typedef struct {
3069 PyObject_HEAD
3070} NullImporter;
3071
3072static int
3073NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
3074{
3075 char *path;
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003076 Py_ssize_t pathlen;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003077
3078 if (!_PyArg_NoKeywords("NullImporter()", kwds))
3079 return -1;
3080
3081 if (!PyArg_ParseTuple(args, "s:NullImporter",
3082 &path))
3083 return -1;
3084
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003085 pathlen = strlen(path);
3086 if (pathlen == 0) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003087 PyErr_SetString(PyExc_ImportError, "empty pathname");
3088 return -1;
3089 } else {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003090 struct stat statbuf;
3091 int rv;
3092
3093 rv = stat(path, &statbuf);
Guido van Rossumc1bdbc32007-11-07 17:46:34 +00003094#ifdef MS_WINDOWS
3095 /* MS Windows stat() chokes on paths like C:\path\. Try to
3096 * recover *one* time by stripping off a trailing slash or
3097 * backslash. http://bugs.python.org/issue1293
3098 */
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003099 if (rv != 0 && pathlen <= MAXPATHLEN &&
3100 (path[pathlen-1] == '/' || path[pathlen-1] == '\\')) {
3101 char mangled[MAXPATHLEN+1];
3102
3103 strcpy(mangled, path);
3104 mangled[pathlen-1] = '\0';
3105 rv = stat(mangled, &statbuf);
3106 }
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003107#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003108 if (rv == 0) {
3109 /* it exists */
3110 if (S_ISDIR(statbuf.st_mode)) {
3111 /* it's a directory */
3112 PyErr_SetString(PyExc_ImportError,
3113 "existing directory");
3114 return -1;
3115 }
3116 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003117 }
3118 return 0;
3119}
3120
3121static PyObject *
3122NullImporter_find_module(NullImporter *self, PyObject *args)
3123{
3124 Py_RETURN_NONE;
3125}
3126
3127static PyMethodDef NullImporter_methods[] = {
3128 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
3129 "Always return None"
3130 },
3131 {NULL} /* Sentinel */
3132};
3133
3134
Christian Heimes9cd17752007-11-18 19:35:23 +00003135PyTypeObject PyNullImporter_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00003136 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003137 "imp.NullImporter", /*tp_name*/
3138 sizeof(NullImporter), /*tp_basicsize*/
3139 0, /*tp_itemsize*/
3140 0, /*tp_dealloc*/
3141 0, /*tp_print*/
3142 0, /*tp_getattr*/
3143 0, /*tp_setattr*/
3144 0, /*tp_compare*/
3145 0, /*tp_repr*/
3146 0, /*tp_as_number*/
3147 0, /*tp_as_sequence*/
3148 0, /*tp_as_mapping*/
3149 0, /*tp_hash */
3150 0, /*tp_call*/
3151 0, /*tp_str*/
3152 0, /*tp_getattro*/
3153 0, /*tp_setattro*/
3154 0, /*tp_as_buffer*/
3155 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3156 "Null importer object", /* tp_doc */
3157 0, /* tp_traverse */
3158 0, /* tp_clear */
3159 0, /* tp_richcompare */
3160 0, /* tp_weaklistoffset */
3161 0, /* tp_iter */
3162 0, /* tp_iternext */
3163 NullImporter_methods, /* tp_methods */
3164 0, /* tp_members */
3165 0, /* tp_getset */
3166 0, /* tp_base */
3167 0, /* tp_dict */
3168 0, /* tp_descr_get */
3169 0, /* tp_descr_set */
3170 0, /* tp_dictoffset */
3171 (initproc)NullImporter_init, /* tp_init */
3172 0, /* tp_alloc */
3173 PyType_GenericNew /* tp_new */
3174};
3175
3176
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003177PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003178initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003179{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003180 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003181
Christian Heimes9cd17752007-11-18 19:35:23 +00003182 if (PyType_Ready(&PyNullImporter_Type) < 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003183 goto failure;
3184
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003185 m = Py_InitModule4("imp", imp_methods, doc_imp,
3186 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003187 if (m == NULL)
3188 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003189 d = PyModule_GetDict(m);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003190 if (d == NULL)
3191 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003192
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003193 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3194 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3195 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3196 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3197 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3198 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3199 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3200 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003201 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003202 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003203
Christian Heimes9cd17752007-11-18 19:35:23 +00003204 Py_INCREF(&PyNullImporter_Type);
3205 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003206 failure:
3207 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003208}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003209
3210
Guido van Rossumb18618d2000-05-03 23:44:39 +00003211/* API for embedding applications that want to add their own entries
3212 to the table of built-in modules. This should normally be called
3213 *before* Py_Initialize(). When the table resize fails, -1 is
3214 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003215
3216 After a similar function by Just van Rossum. */
3217
3218int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003219PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003220{
3221 static struct _inittab *our_copy = NULL;
3222 struct _inittab *p;
3223 int i, n;
3224
3225 /* Count the number of entries in both tables */
3226 for (n = 0; newtab[n].name != NULL; n++)
3227 ;
3228 if (n == 0)
3229 return 0; /* Nothing to do */
3230 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3231 ;
3232
3233 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003234 p = our_copy;
3235 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003236 if (p == NULL)
3237 return -1;
3238
3239 /* Copy the tables into the new memory */
3240 if (our_copy != PyImport_Inittab)
3241 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3242 PyImport_Inittab = our_copy = p;
3243 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3244
3245 return 0;
3246}
3247
3248/* Shorthand to add a single entry given a name and a function */
3249
3250int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003251PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003252{
3253 struct _inittab newtab[2];
3254
3255 memset(newtab, '\0', sizeof newtab);
3256
3257 newtab[0].name = name;
3258 newtab[0].initfunc = initfunc;
3259
3260 return PyImport_ExtendInittab(newtab);
3261}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003262
3263#ifdef __cplusplus
3264}
3265#endif