blob: f342143563239cada78b7ef91551af0b46d9b6df [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);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000957 if (cpathname)
958 write_compiled_module(co, cpathname, mtime);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000959 }
Guido van Rossumafd3dae1998-08-25 18:44:34 +0000960 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname);
Guido van Rossum79f25d91997-04-29 20:08:16 +0000961 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000962
963 return m;
964}
965
Christian Heimes3b06e532008-01-07 20:12:44 +0000966/* Get source file -> unicode or None
967 * Returns the path to the py file if available, else the given path
968 */
969static PyObject *
970get_sourcefile(const char *file)
971{
972 char py[MAXPATHLEN + 1];
973 Py_ssize_t len;
974 PyObject *u;
975 struct stat statbuf;
976
977 if (!file || !*file) {
978 Py_RETURN_NONE;
979 }
980
981 len = strlen(file);
982 if (len > MAXPATHLEN || PyOS_stricmp(&file[len-4], ".pyc") != 0) {
983 return PyUnicode_DecodeFSDefault(file);
984 }
985
986 strncpy(py, file, len-1);
987 py[len] = '\0';
988 if (stat(py, &statbuf) == 0 &&
989 S_ISREG(statbuf.st_mode)) {
990 u = PyUnicode_DecodeFSDefault(py);
991 }
992 else {
993 u = PyUnicode_DecodeFSDefault(file);
994 }
995 return u;
996}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000997
Guido van Rossumaee0bad1997-09-05 07:33:22 +0000998/* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +0000999static PyObject *load_module(char *, FILE *, char *, int, PyObject *);
1000static struct filedescr *find_module(char *, char *, PyObject *,
1001 char *, size_t, FILE **, PyObject **);
Christian Heimes3b06e532008-01-07 20:12:44 +00001002static struct _frozen * find_frozen(char *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001003
1004/* Load a package and return its module object WITH INCREMENTED
1005 REFERENCE COUNT */
1006
1007static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001008load_package(char *name, char *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001009{
Tim Peters1cd70172004-08-02 03:52:12 +00001010 PyObject *m, *d;
1011 PyObject *file = NULL;
1012 PyObject *path = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001013 int err;
1014 char buf[MAXPATHLEN+1];
1015 FILE *fp = NULL;
1016 struct filedescr *fdp;
1017
1018 m = PyImport_AddModule(name);
1019 if (m == NULL)
1020 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001021 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001022 PySys_WriteStderr("import %s # directory %s\n",
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001023 name, pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001024 d = PyModule_GetDict(m);
Christian Heimes3b06e532008-01-07 20:12:44 +00001025 file = get_sourcefile(pathname);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001026 if (file == NULL)
Tim Peters1cd70172004-08-02 03:52:12 +00001027 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001028 path = Py_BuildValue("[O]", file);
Tim Peters1cd70172004-08-02 03:52:12 +00001029 if (path == NULL)
1030 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001031 err = PyDict_SetItemString(d, "__file__", file);
1032 if (err == 0)
1033 err = PyDict_SetItemString(d, "__path__", path);
Tim Peters1cd70172004-08-02 03:52:12 +00001034 if (err != 0)
1035 goto error;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001036 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00001037 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001038 if (fdp == NULL) {
1039 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1040 PyErr_Clear();
Thomas Heller25653242004-06-07 15:04:10 +00001041 Py_INCREF(m);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001042 }
1043 else
1044 m = NULL;
1045 goto cleanup;
1046 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001047 m = load_module(name, fp, buf, fdp->type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001048 if (fp != NULL)
1049 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00001050 goto cleanup;
1051
1052 error:
1053 m = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001054 cleanup:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001055 Py_XDECREF(path);
1056 Py_XDECREF(file);
1057 return m;
1058}
1059
1060
1061/* Helper to test for built-in module */
1062
1063static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001064is_builtin(char *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001065{
1066 int i;
Guido van Rossum771c6c81997-10-31 18:37:24 +00001067 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
1068 if (strcmp(name, PyImport_Inittab[i].name) == 0) {
1069 if (PyImport_Inittab[i].initfunc == NULL)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001070 return -1;
1071 else
1072 return 1;
1073 }
1074 }
1075 return 0;
1076}
1077
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001078
Just van Rossum52e14d62002-12-30 22:08:05 +00001079/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1080 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +00001081 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +00001082 that can handle the path item. Return None if no hook could;
1083 this tells our caller it should fall back to the builtin
1084 import mechanism. Cache the result in path_importer_cache.
1085 Returns a borrowed reference. */
1086
1087static PyObject *
1088get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
1089 PyObject *p)
1090{
1091 PyObject *importer;
Martin v. Löwis725507b2006-03-07 12:08:51 +00001092 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001093
1094 /* These conditions are the caller's responsibility: */
1095 assert(PyList_Check(path_hooks));
1096 assert(PyDict_Check(path_importer_cache));
1097
1098 nhooks = PyList_Size(path_hooks);
1099 if (nhooks < 0)
1100 return NULL; /* Shouldn't happen */
1101
1102 importer = PyDict_GetItem(path_importer_cache, p);
1103 if (importer != NULL)
1104 return importer;
1105
1106 /* set path_importer_cache[p] to None to avoid recursion */
1107 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1108 return NULL;
1109
1110 for (j = 0; j < nhooks; j++) {
1111 PyObject *hook = PyList_GetItem(path_hooks, j);
1112 if (hook == NULL)
1113 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001114 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
Just van Rossum52e14d62002-12-30 22:08:05 +00001115 if (importer != NULL)
1116 break;
1117
1118 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1119 return NULL;
1120 }
1121 PyErr_Clear();
1122 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001123 if (importer == NULL) {
1124 importer = PyObject_CallFunctionObjArgs(
Christian Heimes9cd17752007-11-18 19:35:23 +00001125 (PyObject *)&PyNullImporter_Type, p, NULL
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 );
1127 if (importer == NULL) {
1128 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1129 PyErr_Clear();
1130 return Py_None;
1131 }
1132 }
1133 }
1134 if (importer != NULL) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001135 int err = PyDict_SetItem(path_importer_cache, p, importer);
1136 Py_DECREF(importer);
1137 if (err != 0)
1138 return NULL;
1139 }
1140 return importer;
1141}
1142
Christian Heimes9cd17752007-11-18 19:35:23 +00001143PyAPI_FUNC(PyObject *)
1144PyImport_GetImporter(PyObject *path) {
1145 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
1146
1147 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) {
1148 if ((path_hooks = PySys_GetObject("path_hooks"))) {
1149 importer = get_path_importer(path_importer_cache,
1150 path_hooks, path);
1151 }
1152 }
1153 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1154 return importer;
1155}
1156
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001157/* Search the path (default sys.path) for a module. Return the
1158 corresponding filedescr struct, and (via return arguments) the
1159 pathname and an open file. Return NULL if the module is not found. */
1160
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001161#ifdef MS_COREDLL
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00001162extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001163 char *, Py_ssize_t);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001164#endif
1165
Martin v. Löwis18e16552006-02-15 17:27:45 +00001166static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001167static int find_init_module(char *); /* Forward */
Just van Rossum52e14d62002-12-30 22:08:05 +00001168static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001169
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001170static struct filedescr *
Just van Rossum52e14d62002-12-30 22:08:05 +00001171find_module(char *fullname, char *subname, PyObject *path, char *buf,
1172 size_t buflen, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001173{
Martin v. Löwis725507b2006-03-07 12:08:51 +00001174 Py_ssize_t i, npath;
Fred Drake4c82b232000-06-30 16:18:57 +00001175 size_t len, namelen;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001176 struct filedescr *fdp = NULL;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001177 char *filemode;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001178 FILE *fp = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001179 PyObject *path_hooks, *path_importer_cache;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001180 struct stat statbuf;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001181 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1182 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1183 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
Guido van Rossum0506a431998-08-11 15:07:39 +00001184 char name[MAXPATHLEN+1];
Andrew MacIntyred9400542002-02-26 11:41:34 +00001185#if defined(PYOS_OS2)
1186 size_t saved_len;
1187 size_t saved_namelen;
1188 char *saved_buf = NULL;
1189#endif
Just van Rossum52e14d62002-12-30 22:08:05 +00001190 if (p_loader != NULL)
1191 *p_loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001192
Just van Rossum52e14d62002-12-30 22:08:05 +00001193 if (strlen(subname) > MAXPATHLEN) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001194 PyErr_SetString(PyExc_OverflowError,
1195 "module name is too long");
Fred Drake4c82b232000-06-30 16:18:57 +00001196 return NULL;
1197 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001198 strcpy(name, subname);
1199
1200 /* sys.meta_path import hook */
1201 if (p_loader != NULL) {
1202 PyObject *meta_path;
1203
1204 meta_path = PySys_GetObject("meta_path");
1205 if (meta_path == NULL || !PyList_Check(meta_path)) {
1206 PyErr_SetString(PyExc_ImportError,
1207 "sys.meta_path must be a list of "
1208 "import hooks");
1209 return NULL;
1210 }
1211 Py_INCREF(meta_path); /* zap guard */
1212 npath = PyList_Size(meta_path);
1213 for (i = 0; i < npath; i++) {
1214 PyObject *loader;
1215 PyObject *hook = PyList_GetItem(meta_path, i);
1216 loader = PyObject_CallMethod(hook, "find_module",
1217 "sO", fullname,
1218 path != NULL ?
1219 path : Py_None);
1220 if (loader == NULL) {
1221 Py_DECREF(meta_path);
1222 return NULL; /* true error */
1223 }
1224 if (loader != Py_None) {
1225 /* a loader was found */
1226 *p_loader = loader;
1227 Py_DECREF(meta_path);
1228 return &importhookdescr;
1229 }
1230 Py_DECREF(loader);
1231 }
1232 Py_DECREF(meta_path);
1233 }
Guido van Rossum0506a431998-08-11 15:07:39 +00001234
Guido van Rossum13d77992007-07-23 03:16:50 +00001235 if (path != NULL && PyUnicode_Check(path)) {
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001236 /* The only type of submodule allowed inside a "frozen"
1237 package are other frozen modules or packages. */
Guido van Rossum13d77992007-07-23 03:16:50 +00001238 char *p = PyUnicode_AsString(path);
1239 if (strlen(p) + 1 + strlen(name) >= (size_t)buflen) {
Guido van Rossum0506a431998-08-11 15:07:39 +00001240 PyErr_SetString(PyExc_ImportError,
1241 "full frozen module name too long");
1242 return NULL;
1243 }
Guido van Rossum13d77992007-07-23 03:16:50 +00001244 strcpy(buf, p);
Guido van Rossum0506a431998-08-11 15:07:39 +00001245 strcat(buf, ".");
1246 strcat(buf, name);
1247 strcpy(name, buf);
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001248 if (find_frozen(name) != NULL) {
1249 strcpy(buf, name);
1250 return &fd_frozen;
1251 }
1252 PyErr_Format(PyExc_ImportError,
1253 "No frozen submodule named %.200s", name);
1254 return NULL;
Guido van Rossum0506a431998-08-11 15:07:39 +00001255 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001256 if (path == NULL) {
1257 if (is_builtin(name)) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001258 strcpy(buf, name);
1259 return &fd_builtin;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001260 }
Greg Ward201baee2001-10-04 14:52:06 +00001261 if ((find_frozen(name)) != NULL) {
Guido van Rossuma5568d31998-03-05 03:45:08 +00001262 strcpy(buf, name);
1263 return &fd_frozen;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001264 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001265
Guido van Rossumac279101996-08-22 23:10:58 +00001266#ifdef MS_COREDLL
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001267 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen);
1268 if (fp != NULL) {
1269 *p_fp = fp;
1270 return fdp;
1271 }
Guido van Rossuma5a3db71996-04-09 02:39:59 +00001272#endif
Guido van Rossuma5568d31998-03-05 03:45:08 +00001273 path = PySys_GetObject("path");
1274 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001275 if (path == NULL || !PyList_Check(path)) {
1276 PyErr_SetString(PyExc_ImportError,
Guido van Rossuma5568d31998-03-05 03:45:08 +00001277 "sys.path must be a list of directory names");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001278 return NULL;
1279 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001280
1281 path_hooks = PySys_GetObject("path_hooks");
1282 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
1283 PyErr_SetString(PyExc_ImportError,
1284 "sys.path_hooks must be a list of "
1285 "import hooks");
1286 return NULL;
1287 }
1288 path_importer_cache = PySys_GetObject("path_importer_cache");
1289 if (path_importer_cache == NULL ||
1290 !PyDict_Check(path_importer_cache)) {
1291 PyErr_SetString(PyExc_ImportError,
1292 "sys.path_importer_cache must be a dict");
1293 return NULL;
1294 }
1295
Guido van Rossum79f25d91997-04-29 20:08:16 +00001296 npath = PyList_Size(path);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001297 namelen = strlen(name);
1298 for (i = 0; i < npath; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001299 PyObject *v = PyList_GetItem(path, i);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001300 PyObject *origv = v;
Guido van Rossum6262cc72007-05-09 23:29:27 +00001301 const char *base;
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001302 Py_ssize_t size;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001303 if (!v)
1304 return NULL;
Guido van Rossuma4b8d1d2007-08-27 15:02:28 +00001305 if (PyUnicode_Check(v)) {
1306 v = _PyUnicode_AsDefaultEncodedString(v, NULL);
1307 if (v == NULL)
1308 return NULL;
1309 }
1310 if (!PyString_Check(v))
1311 continue;
1312 base = PyString_AS_STRING(v);
1313 size = PyString_GET_SIZE(v);
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001314 len = size;
Walter Dörwald3430d702002-06-17 10:43:59 +00001315 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001316 continue; /* Too long */
Walter Dörwald3430d702002-06-17 10:43:59 +00001317 }
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001318 strcpy(buf, base);
Walter Dörwald3430d702002-06-17 10:43:59 +00001319 if (strlen(buf) != len) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001320 continue; /* v contains '\0' */
Walter Dörwald3430d702002-06-17 10:43:59 +00001321 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001322
1323 /* sys.path_hooks import hook */
1324 if (p_loader != NULL) {
1325 PyObject *importer;
1326
1327 importer = get_path_importer(path_importer_cache,
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001328 path_hooks, origv);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001329 if (importer == NULL) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001330 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001331 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001332 /* Note: importer is a borrowed reference */
1333 if (importer != Py_None) {
Just van Rossum52e14d62002-12-30 22:08:05 +00001334 PyObject *loader;
1335 loader = PyObject_CallMethod(importer,
1336 "find_module",
1337 "s", fullname);
1338 if (loader == NULL)
1339 return NULL; /* error */
1340 if (loader != Py_None) {
1341 /* a loader was found */
1342 *p_loader = loader;
1343 return &importhookdescr;
1344 }
1345 Py_DECREF(loader);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001346 continue;
Just van Rossum52e14d62002-12-30 22:08:05 +00001347 }
Just van Rossum52e14d62002-12-30 22:08:05 +00001348 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001349 /* no hook was found, use builtin import */
Just van Rossum52e14d62002-12-30 22:08:05 +00001350
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001351 if (len > 0 && buf[len-1] != SEP
1352#ifdef ALTSEP
1353 && buf[len-1] != ALTSEP
1354#endif
1355 )
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001356 buf[len++] = SEP;
Guido van Rossum215c3402000-11-13 17:26:32 +00001357 strcpy(buf+len, name);
1358 len += namelen;
Tim Peters50d8d372001-02-28 05:34:27 +00001359
1360 /* Check for package import (buf holds a directory name,
1361 and there's an __init__ module in that directory */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001362#ifdef HAVE_STAT
Walter Dörwald3430d702002-06-17 10:43:59 +00001363 if (stat(buf, &statbuf) == 0 && /* it exists */
1364 S_ISDIR(statbuf.st_mode) && /* it's a directory */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001365 case_ok(buf, len, namelen, name)) { /* case matches */
1366 if (find_init_module(buf)) { /* and has __init__.py */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001367 return &fd_package;
1368 }
1369 else {
1370 char warnstr[MAXPATHLEN+80];
1371 sprintf(warnstr, "Not importing directory "
1372 "'%.*s': missing __init__.py",
1373 MAXPATHLEN, buf);
Skip Montanaro46fc3372007-08-12 11:44:53 +00001374 if (PyErr_WarnEx(PyExc_ImportWarning,
1375 warnstr, 1)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001376 return NULL;
1377 }
1378 }
Walter Dörwald3430d702002-06-17 10:43:59 +00001379 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001380#endif
Andrew MacIntyred9400542002-02-26 11:41:34 +00001381#if defined(PYOS_OS2)
1382 /* take a snapshot of the module spec for restoration
1383 * after the 8 character DLL hackery
1384 */
1385 saved_buf = strdup(buf);
1386 saved_len = len;
1387 saved_namelen = namelen;
1388#endif /* PYOS_OS2 */
Guido van Rossum79f25d91997-04-29 20:08:16 +00001389 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Guido van Rossum04110fb2007-08-24 16:32:05 +00001390#if defined(PYOS_OS2) && defined(HAVE_DYNAMIC_LOADING)
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001391 /* OS/2 limits DLLs to 8 character names (w/o
1392 extension)
Andrew MacIntyred9400542002-02-26 11:41:34 +00001393 * so if the name is longer than that and its a
1394 * dynamically loaded module we're going to try,
1395 * truncate the name before trying
1396 */
Just van Rossum52e14d62002-12-30 22:08:05 +00001397 if (strlen(subname) > 8) {
Andrew MacIntyred9400542002-02-26 11:41:34 +00001398 /* is this an attempt to load a C extension? */
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001399 const struct filedescr *scan;
1400 scan = _PyImport_DynLoadFiletab;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001401 while (scan->suffix != NULL) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001402 if (!strcmp(scan->suffix, fdp->suffix))
Andrew MacIntyred9400542002-02-26 11:41:34 +00001403 break;
1404 else
1405 scan++;
1406 }
1407 if (scan->suffix != NULL) {
1408 /* yes, so truncate the name */
1409 namelen = 8;
Just van Rossum52e14d62002-12-30 22:08:05 +00001410 len -= strlen(subname) - namelen;
Andrew MacIntyred9400542002-02-26 11:41:34 +00001411 buf[len] = '\0';
1412 }
1413 }
1414#endif /* PYOS_OS2 */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001415 strcpy(buf+len, fdp->suffix);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001416 if (Py_VerboseFlag > 1)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001417 PySys_WriteStderr("# trying %s\n", buf);
Jack Jansenc88da1f2002-05-28 10:58:19 +00001418 filemode = fdp->mode;
Tim Peters86c7d2f2004-08-01 23:24:21 +00001419 if (filemode[0] == 'U')
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00001420 filemode = "r" PY_STDIOTEXTMODE;
Jack Jansenc88da1f2002-05-28 10:58:19 +00001421 fp = fopen(buf, filemode);
Tim Peters50d8d372001-02-28 05:34:27 +00001422 if (fp != NULL) {
1423 if (case_ok(buf, len, namelen, name))
1424 break;
1425 else { /* continue search */
1426 fclose(fp);
1427 fp = NULL;
Barry Warsaw914a0b12001-02-02 19:12:16 +00001428 }
Barry Warsaw914a0b12001-02-02 19:12:16 +00001429 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001430#if defined(PYOS_OS2)
1431 /* restore the saved snapshot */
1432 strcpy(buf, saved_buf);
1433 len = saved_len;
1434 namelen = saved_namelen;
1435#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001436 }
Andrew MacIntyred9400542002-02-26 11:41:34 +00001437#if defined(PYOS_OS2)
1438 /* don't need/want the module name snapshot anymore */
1439 if (saved_buf)
1440 {
1441 free(saved_buf);
1442 saved_buf = NULL;
1443 }
1444#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001445 if (fp != NULL)
1446 break;
1447 }
1448 if (fp == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001449 PyErr_Format(PyExc_ImportError,
1450 "No module named %.200s", name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001451 return NULL;
1452 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001453 *p_fp = fp;
1454 return fdp;
1455}
1456
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +00001457/* Helpers for main.c
1458 * Find the source file corresponding to a named module
1459 */
1460struct filedescr *
1461_PyImport_FindModule(const char *name, PyObject *path, char *buf,
1462 size_t buflen, FILE **p_fp, PyObject **p_loader)
1463{
1464 return find_module((char *) name, (char *) name, path,
1465 buf, buflen, p_fp, p_loader);
1466}
1467
1468PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd)
1469{
1470 return fd->type == PY_SOURCE || fd->type == PY_COMPILED;
1471}
1472
Martin v. Löwis18e16552006-02-15 17:27:45 +00001473/* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00001474 * The arguments here are tricky, best shown by example:
1475 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1476 * ^ ^ ^ ^
1477 * |--------------------- buf ---------------------|
1478 * |------------------- len ------------------|
1479 * |------ name -------|
1480 * |----- namelen -----|
1481 * buf is the full path, but len only counts up to (& exclusive of) the
1482 * extension. name is the module name, also exclusive of extension.
1483 *
1484 * We've already done a successful stat() or fopen() on buf, so know that
1485 * there's some match, possibly case-insensitive.
1486 *
Tim Peters50d8d372001-02-28 05:34:27 +00001487 * case_ok() is to return 1 if there's a case-sensitive match for
1488 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK
1489 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00001490 *
Tim Peters50d8d372001-02-28 05:34:27 +00001491 * case_ok() is used to implement case-sensitive import semantics even
1492 * on platforms with case-insensitive filesystems. It's trivial to implement
1493 * for case-sensitive filesystems. It's pretty much a cross-platform
1494 * nightmare for systems with case-insensitive filesystems.
1495 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00001496
Tim Peters50d8d372001-02-28 05:34:27 +00001497/* First we may need a pile of platform-specific header files; the sequence
1498 * of #if's here should match the sequence in the body of case_ok().
1499 */
Jason Tishler7961aa62005-05-20 00:56:54 +00001500#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001501#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00001502
Tim Peters50d8d372001-02-28 05:34:27 +00001503#elif defined(DJGPP)
1504#include <dir.h>
1505
Jason Tishler7961aa62005-05-20 00:56:54 +00001506#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001507#include <sys/types.h>
1508#include <dirent.h>
1509
Andrew MacIntyred9400542002-02-26 11:41:34 +00001510#elif defined(PYOS_OS2)
1511#define INCL_DOS
1512#define INCL_DOSERRORS
1513#define INCL_NOPMAPI
1514#include <os2.h>
Tim Peters50d8d372001-02-28 05:34:27 +00001515#endif
1516
Guido van Rossum0980bd91998-02-13 17:18:36 +00001517static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001518case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001519{
Tim Peters50d8d372001-02-28 05:34:27 +00001520/* Pick a platform-specific implementation; the sequence of #if's here should
1521 * match the sequence just above.
1522 */
1523
Jason Tishler7961aa62005-05-20 00:56:54 +00001524/* MS_WINDOWS */
1525#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001526 WIN32_FIND_DATA data;
1527 HANDLE h;
Tim Peters50d8d372001-02-28 05:34:27 +00001528
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001529 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum0980bd91998-02-13 17:18:36 +00001530 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001531
Guido van Rossum0980bd91998-02-13 17:18:36 +00001532 h = FindFirstFile(buf, &data);
1533 if (h == INVALID_HANDLE_VALUE) {
1534 PyErr_Format(PyExc_NameError,
1535 "Can't find file for module %.100s\n(filename %.300s)",
1536 name, buf);
1537 return 0;
1538 }
1539 FindClose(h);
Tim Peters50d8d372001-02-28 05:34:27 +00001540 return strncmp(data.cFileName, name, namelen) == 0;
1541
1542/* DJGPP */
1543#elif defined(DJGPP)
1544 struct ffblk ffblk;
1545 int done;
1546
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001547 if (Py_GETENV("PYTHONCASEOK") != NULL)
Guido van Rossum323bf5e1998-06-24 03:54:06 +00001548 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00001549
1550 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
1551 if (done) {
Guido van Rossum0980bd91998-02-13 17:18:36 +00001552 PyErr_Format(PyExc_NameError,
Tim Peters50d8d372001-02-28 05:34:27 +00001553 "Can't find file for module %.100s\n(filename %.300s)",
Guido van Rossum0980bd91998-02-13 17:18:36 +00001554 name, buf);
1555 return 0;
1556 }
Tim Peters50d8d372001-02-28 05:34:27 +00001557 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001558
Jason Tishler7961aa62005-05-20 00:56:54 +00001559/* new-fangled macintosh (macosx) or Cygwin */
1560#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00001561 DIR *dirp;
1562 struct dirent *dp;
Tim Petersd1e87a82001-03-01 18:12:00 +00001563 char dirname[MAXPATHLEN + 1];
1564 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00001565
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +00001566 if (Py_GETENV("PYTHONCASEOK") != NULL)
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001567 return 1;
1568
Tim Petersd1e87a82001-03-01 18:12:00 +00001569 /* Copy the dir component into dirname; substitute "." if empty */
1570 if (dirlen <= 0) {
1571 dirname[0] = '.';
1572 dirname[1] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001573 }
1574 else {
Tim Petersd1e87a82001-03-01 18:12:00 +00001575 assert(dirlen <= MAXPATHLEN);
1576 memcpy(dirname, buf, dirlen);
1577 dirname[dirlen] = '\0';
Tim Peters430f5d42001-03-01 01:30:56 +00001578 }
1579 /* Open the directory and search the entries for an exact match. */
Tim Petersd1e87a82001-03-01 18:12:00 +00001580 dirp = opendir(dirname);
Tim Peters430f5d42001-03-01 01:30:56 +00001581 if (dirp) {
Tim Peters677898a2001-03-02 03:28:03 +00001582 char *nameWithExt = buf + len - namelen;
Tim Peters430f5d42001-03-01 01:30:56 +00001583 while ((dp = readdir(dirp)) != NULL) {
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001584 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00001585#ifdef _DIRENT_HAVE_D_NAMELEN
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001586 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00001587#else
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001588 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00001589#endif
Tim Petersd1e87a82001-03-01 18:12:00 +00001590 if (thislen >= namelen &&
Tim Peters677898a2001-03-02 03:28:03 +00001591 strcmp(dp->d_name, nameWithExt) == 0) {
Tim Peters430f5d42001-03-01 01:30:56 +00001592 (void)closedir(dirp);
1593 return 1; /* Found */
1594 }
1595 }
Tim Petersdbe6ebb2001-03-01 08:47:29 +00001596 (void)closedir(dirp);
Tim Peters430f5d42001-03-01 01:30:56 +00001597 }
Tim Peters430f5d42001-03-01 01:30:56 +00001598 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00001599
Andrew MacIntyred9400542002-02-26 11:41:34 +00001600/* OS/2 */
1601#elif defined(PYOS_OS2)
1602 HDIR hdir = 1;
1603 ULONG srchcnt = 1;
1604 FILEFINDBUF3 ffbuf;
1605 APIRET rc;
1606
1607 if (getenv("PYTHONCASEOK") != NULL)
1608 return 1;
1609
1610 rc = DosFindFirst(buf,
1611 &hdir,
1612 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
1613 &ffbuf, sizeof(ffbuf),
1614 &srchcnt,
1615 FIL_STANDARD);
1616 if (rc != NO_ERROR)
1617 return 0;
1618 return strncmp(ffbuf.achName, name, namelen) == 0;
1619
Tim Peters50d8d372001-02-28 05:34:27 +00001620/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
1621#else
Guido van Rossum0980bd91998-02-13 17:18:36 +00001622 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00001623
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001624#endif
Tim Peters50d8d372001-02-28 05:34:27 +00001625}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00001626
Guido van Rossum0980bd91998-02-13 17:18:36 +00001627
Guido van Rossum197346f1997-10-31 18:38:52 +00001628#ifdef HAVE_STAT
1629/* Helper to look for __init__.py or __init__.py[co] in potential package */
1630static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001631find_init_module(char *buf)
Guido van Rossum197346f1997-10-31 18:38:52 +00001632{
Tim Peters0f9431f2001-07-05 03:47:53 +00001633 const size_t save_len = strlen(buf);
Fred Drake4c82b232000-06-30 16:18:57 +00001634 size_t i = save_len;
Tim Peters0f9431f2001-07-05 03:47:53 +00001635 char *pname; /* pointer to start of __init__ */
Guido van Rossum197346f1997-10-31 18:38:52 +00001636 struct stat statbuf;
1637
Tim Peters0f9431f2001-07-05 03:47:53 +00001638/* For calling case_ok(buf, len, namelen, name):
1639 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
1640 * ^ ^ ^ ^
1641 * |--------------------- buf ---------------------|
1642 * |------------------- len ------------------|
1643 * |------ name -------|
1644 * |----- namelen -----|
1645 */
Guido van Rossum197346f1997-10-31 18:38:52 +00001646 if (save_len + 13 >= MAXPATHLEN)
1647 return 0;
1648 buf[i++] = SEP;
Tim Peters0f9431f2001-07-05 03:47:53 +00001649 pname = buf + i;
1650 strcpy(pname, "__init__.py");
Guido van Rossum197346f1997-10-31 18:38:52 +00001651 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001652 if (case_ok(buf,
1653 save_len + 9, /* len("/__init__") */
1654 8, /* len("__init__") */
1655 pname)) {
1656 buf[save_len] = '\0';
1657 return 1;
1658 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001659 }
Tim Peters0f9431f2001-07-05 03:47:53 +00001660 i += strlen(pname);
1661 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c");
Guido van Rossum197346f1997-10-31 18:38:52 +00001662 if (stat(buf, &statbuf) == 0) {
Tim Peters0f9431f2001-07-05 03:47:53 +00001663 if (case_ok(buf,
1664 save_len + 9, /* len("/__init__") */
1665 8, /* len("__init__") */
1666 pname)) {
1667 buf[save_len] = '\0';
1668 return 1;
1669 }
Guido van Rossum197346f1997-10-31 18:38:52 +00001670 }
1671 buf[save_len] = '\0';
1672 return 0;
1673}
Guido van Rossum48a680c2001-03-02 06:34:14 +00001674
Guido van Rossum197346f1997-10-31 18:38:52 +00001675#endif /* HAVE_STAT */
1676
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001677
Tim Petersdbd9ba62000-07-09 03:09:57 +00001678static int init_builtin(char *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001679
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001680/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001681 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001682
Guido van Rossum79f25d91997-04-29 20:08:16 +00001683static PyObject *
Just van Rossum52e14d62002-12-30 22:08:05 +00001684load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001685{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001686 PyObject *modules;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001687 PyObject *m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001688 int err;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001689
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001690 /* First check that there's an open file (if we need one) */
1691 switch (type) {
1692 case PY_SOURCE:
1693 case PY_COMPILED:
1694 if (fp == NULL) {
1695 PyErr_Format(PyExc_ValueError,
1696 "file object required for import (type code %d)",
1697 type);
1698 return NULL;
1699 }
1700 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001701
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001702 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001703
1704 case PY_SOURCE:
1705 m = load_source_module(name, buf, fp);
1706 break;
1707
1708 case PY_COMPILED:
1709 m = load_compiled_module(name, buf, fp);
1710 break;
1711
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001712#ifdef HAVE_DYNAMIC_LOADING
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001713 case C_EXTENSION:
Guido van Rossum79f25d91997-04-29 20:08:16 +00001714 m = _PyImport_LoadDynamicModule(name, buf, fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001715 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00001716#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001717
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001718 case PKG_DIRECTORY:
1719 m = load_package(name, buf);
1720 break;
1721
1722 case C_BUILTIN:
1723 case PY_FROZEN:
Guido van Rossuma5568d31998-03-05 03:45:08 +00001724 if (buf != NULL && buf[0] != '\0')
1725 name = buf;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001726 if (type == C_BUILTIN)
1727 err = init_builtin(name);
1728 else
1729 err = PyImport_ImportFrozenModule(name);
1730 if (err < 0)
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001731 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001732 if (err == 0) {
1733 PyErr_Format(PyExc_ImportError,
1734 "Purported %s module %.200s not found",
1735 type == C_BUILTIN ?
1736 "builtin" : "frozen",
1737 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001738 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001739 }
1740 modules = PyImport_GetModuleDict();
1741 m = PyDict_GetItemString(modules, name);
1742 if (m == NULL) {
1743 PyErr_Format(
1744 PyExc_ImportError,
1745 "%s module %.200s not properly initialized",
1746 type == C_BUILTIN ?
1747 "builtin" : "frozen",
1748 name);
Guido van Rossuma86f77d1997-09-09 18:53:47 +00001749 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001750 }
1751 Py_INCREF(m);
1752 break;
1753
Just van Rossum52e14d62002-12-30 22:08:05 +00001754 case IMP_HOOK: {
1755 if (loader == NULL) {
1756 PyErr_SetString(PyExc_ImportError,
1757 "import hook without loader");
1758 return NULL;
1759 }
1760 m = PyObject_CallMethod(loader, "load_module", "s", name);
1761 break;
1762 }
1763
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001764 default:
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001765 PyErr_Format(PyExc_ImportError,
1766 "Don't know how to import %.200s (type code %d)",
1767 name, type);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001768 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001769
1770 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001771
1772 return m;
1773}
1774
1775
1776/* Initialize a built-in module.
Thomas Wouters89f507f2006-12-13 04:49:30 +00001777 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001778 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001779
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001780static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001781init_builtin(char *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001782{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001783 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001784
Greg Ward201baee2001-10-04 14:52:06 +00001785 if (_PyImport_FindExtension(name, name) != NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +00001786 return 1;
1787
Guido van Rossum771c6c81997-10-31 18:37:24 +00001788 for (p = PyImport_Inittab; p->name != NULL; p++) {
Guido van Rossum25ce5661997-08-02 03:10:38 +00001789 if (strcmp(name, p->name) == 0) {
1790 if (p->initfunc == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001791 PyErr_Format(PyExc_ImportError,
1792 "Cannot re-init internal module %.200s",
1793 name);
Guido van Rossum74e6a111994-08-29 12:54:38 +00001794 return -1;
1795 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00001796 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001797 PySys_WriteStderr("import %s # builtin\n", name);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001798 (*p->initfunc)();
Guido van Rossum79f25d91997-04-29 20:08:16 +00001799 if (PyErr_Occurred())
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001800 return -1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001801 if (_PyImport_FixupExtension(name, name) == NULL)
1802 return -1;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00001803 return 1;
1804 }
1805 }
1806 return 0;
1807}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001808
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001809
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001810/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001811
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001812static struct _frozen *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001813find_frozen(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001814{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001815 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001816
Guido van Rossum79f25d91997-04-29 20:08:16 +00001817 for (p = PyImport_FrozenModules; ; p++) {
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001818 if (p->name == NULL)
1819 return NULL;
1820 if (strcmp(p->name, name) == 0)
1821 break;
1822 }
1823 return p;
1824}
1825
Guido van Rossum79f25d91997-04-29 20:08:16 +00001826static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001827get_frozen_object(char *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001828{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001829 struct _frozen *p = find_frozen(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001830 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001831
1832 if (p == NULL) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001833 PyErr_Format(PyExc_ImportError,
1834 "No such frozen object named %.200s",
1835 name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001836 return NULL;
1837 }
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001838 if (p->code == NULL) {
1839 PyErr_Format(PyExc_ImportError,
1840 "Excluded frozen object named %.200s",
1841 name);
1842 return NULL;
1843 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001844 size = p->size;
1845 if (size < 0)
1846 size = -size;
1847 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001848}
1849
1850/* Initialize a frozen module.
1851 Return 1 for succes, 0 if the module is not found, and -1 with
1852 an exception set if the initialization failed.
1853 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00001854
1855int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001856PyImport_ImportFrozenModule(char *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001857{
Guido van Rossumcfd0a221996-06-17 17:06:34 +00001858 struct _frozen *p = find_frozen(name);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001859 PyObject *co;
1860 PyObject *m;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001861 int ispackage;
1862 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00001863
1864 if (p == NULL)
1865 return 0;
Guido van Rossum8f4d3312001-10-18 18:54:11 +00001866 if (p->code == NULL) {
1867 PyErr_Format(PyExc_ImportError,
1868 "Excluded frozen object named %.200s",
1869 name);
1870 return -1;
1871 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001872 size = p->size;
1873 ispackage = (size < 0);
1874 if (ispackage)
1875 size = -size;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001876 if (Py_VerboseFlag)
Guido van Rossum2f3667a1998-10-12 18:23:55 +00001877 PySys_WriteStderr("import %s # frozen%s\n",
Guido van Rossuma5568d31998-03-05 03:45:08 +00001878 name, ispackage ? " package" : "");
1879 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001880 if (co == NULL)
1881 return -1;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001882 if (!PyCode_Check(co)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001883 PyErr_Format(PyExc_TypeError,
1884 "frozen object %.200s is not a code object",
1885 name);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001886 goto err_return;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001887 }
Guido van Rossuma5568d31998-03-05 03:45:08 +00001888 if (ispackage) {
1889 /* Set __path__ to the package name */
1890 PyObject *d, *s;
1891 int err;
1892 m = PyImport_AddModule(name);
1893 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001894 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001895 d = PyModule_GetDict(m);
Martin v. Löwis5b222132007-06-10 09:51:05 +00001896 s = PyUnicode_InternFromString(name);
Guido van Rossuma5568d31998-03-05 03:45:08 +00001897 if (s == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001898 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001899 err = PyDict_SetItemString(d, "__path__", s);
1900 Py_DECREF(s);
1901 if (err != 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001902 goto err_return;
Guido van Rossuma5568d31998-03-05 03:45:08 +00001903 }
Guido van Rossume32bf6e1998-02-11 05:53:02 +00001904 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>");
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001905 if (m == NULL)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001906 goto err_return;
1907 Py_DECREF(co);
Guido van Rossum79f25d91997-04-29 20:08:16 +00001908 Py_DECREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001909 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001910err_return:
1911 Py_DECREF(co);
1912 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001913}
Guido van Rossum74e6a111994-08-29 12:54:38 +00001914
1915
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001916/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001917 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00001918
Guido van Rossum79f25d91997-04-29 20:08:16 +00001919PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001920PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00001921{
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001922 PyObject *pname;
1923 PyObject *result;
1924
Martin v. Löwis5b222132007-06-10 09:51:05 +00001925 pname = PyUnicode_FromString(name);
Neal Norwitza11e4c12003-03-23 14:31:01 +00001926 if (pname == NULL)
1927 return NULL;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00001928 result = PyImport_Import(pname);
1929 Py_DECREF(pname);
1930 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001931}
1932
Christian Heimes072c0f12008-01-03 23:01:04 +00001933/* Import a module without blocking
1934 *
1935 * At first it tries to fetch the module from sys.modules. If the module was
1936 * never loaded before it loads it with PyImport_ImportModule() unless another
1937 * thread holds the import lock. In the latter case the function raises an
1938 * ImportError instead of blocking.
1939 *
1940 * Returns the module object with incremented ref count.
1941 */
1942PyObject *
1943PyImport_ImportModuleNoBlock(const char *name)
1944{
1945 PyObject *result;
1946 PyObject *modules;
1947 long me;
1948
1949 /* Try to get the module from sys.modules[name] */
1950 modules = PyImport_GetModuleDict();
1951 if (modules == NULL)
1952 return NULL;
1953
1954 result = PyDict_GetItemString(modules, name);
1955 if (result != NULL) {
1956 Py_INCREF(result);
1957 return result;
1958 }
1959 else {
1960 PyErr_Clear();
1961 }
1962
1963 /* check the import lock
1964 * me might be -1 but I ignore the error here, the lock function
1965 * takes care of the problem */
1966 me = PyThread_get_thread_ident();
1967 if (import_lock_thread == -1 || import_lock_thread == me) {
1968 /* no thread or me is holding the lock */
1969 return PyImport_ImportModule(name);
1970 }
1971 else {
1972 PyErr_Format(PyExc_ImportError,
1973 "Failed to import %.200s because the import lock"
1974 "is held by another thread.",
1975 name);
1976 return NULL;
1977 }
1978}
1979
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001980/* Forward declarations for helper routines */
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001981static PyObject *get_parent(PyObject *globals, char *buf,
1982 Py_ssize_t *p_buflen, int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001983static PyObject *load_next(PyObject *mod, PyObject *altmod,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001984 char **p_name, char *buf, Py_ssize_t *p_buflen);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001985static int mark_miss(char *name);
1986static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Martin v. Löwis18e16552006-02-15 17:27:45 +00001987 char *buf, Py_ssize_t buflen, int recursive);
Tim Petersdbd9ba62000-07-09 03:09:57 +00001988static PyObject * import_submodule(PyObject *mod, char *name, char *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001989
1990/* The Magnum Opus of dotted-name import :-) */
1991
Guido van Rossum75acc9c1998-03-03 22:26:50 +00001992static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001993import_module_level(char *name, PyObject *globals, PyObject *locals,
1994 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001995{
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001996 char buf[MAXPATHLEN+1];
Martin v. Löwis18e16552006-02-15 17:27:45 +00001997 Py_ssize_t buflen = 0;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00001998 PyObject *parent, *head, *next, *tail;
1999
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002000 parent = get_parent(globals, buf, &buflen, level);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002001 if (parent == NULL)
2002 return NULL;
2003
2004 head = load_next(parent, Py_None, &name, buf, &buflen);
2005 if (head == NULL)
2006 return NULL;
2007
2008 tail = head;
2009 Py_INCREF(tail);
2010 while (name) {
2011 next = load_next(tail, tail, &name, buf, &buflen);
2012 Py_DECREF(tail);
2013 if (next == NULL) {
2014 Py_DECREF(head);
2015 return NULL;
2016 }
2017 tail = next;
2018 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002019 if (tail == Py_None) {
2020 /* If tail is Py_None, both get_parent and load_next found
2021 an empty module name: someone called __import__("") or
2022 doctored faulty bytecode */
2023 Py_DECREF(tail);
2024 Py_DECREF(head);
2025 PyErr_SetString(PyExc_ValueError,
2026 "Empty module name");
2027 return NULL;
2028 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002029
2030 if (fromlist != NULL) {
2031 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2032 fromlist = NULL;
2033 }
2034
2035 if (fromlist == NULL) {
2036 Py_DECREF(tail);
2037 return head;
2038 }
2039
2040 Py_DECREF(head);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002041 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002042 Py_DECREF(tail);
2043 return NULL;
2044 }
2045
2046 return tail;
2047}
2048
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002049PyObject *
2050PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals,
2051 PyObject *fromlist, int level)
2052{
2053 PyObject *result;
2054 lock_import();
2055 result = import_module_level(name, globals, locals, fromlist, level);
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00002056 if (unlock_import() < 0) {
2057 Py_XDECREF(result);
2058 PyErr_SetString(PyExc_RuntimeError,
2059 "not holding the import lock");
2060 return NULL;
2061 }
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002062 return result;
2063}
2064
Fred Drake87590902004-05-28 20:21:36 +00002065/* Return the package that an import is being performed in. If globals comes
2066 from the module foo.bar.bat (not itself a package), this returns the
2067 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002068 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002069
2070 The *name* of the returned package is returned in buf, with the length of
2071 the name in *p_buflen.
2072
2073 If globals doesn't come from a package or a module in a package, or a
2074 corresponding entry is not found in sys.modules, Py_None is returned.
2075*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002076static PyObject *
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002077get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002078{
2079 static PyObject *namestr = NULL;
2080 static PyObject *pathstr = NULL;
Nick Coghlande10c852007-12-04 12:22:52 +00002081 static PyObject *pkgstr = NULL;
2082 PyObject *pkgname, *modname, *modpath, *modules, *parent;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002083
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002084 if (globals == NULL || !PyDict_Check(globals) || !level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002085 return Py_None;
2086
2087 if (namestr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002088 namestr = PyUnicode_InternFromString("__name__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002089 if (namestr == NULL)
2090 return NULL;
2091 }
2092 if (pathstr == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002093 pathstr = PyUnicode_InternFromString("__path__");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002094 if (pathstr == NULL)
2095 return NULL;
2096 }
Nick Coghlande10c852007-12-04 12:22:52 +00002097 if (pkgstr == NULL) {
2098 pkgstr = PyUnicode_InternFromString("__package__");
2099 if (pkgstr == NULL)
2100 return NULL;
2101 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002102
2103 *buf = '\0';
2104 *p_buflen = 0;
Nick Coghlande10c852007-12-04 12:22:52 +00002105 pkgname = PyDict_GetItem(globals, pkgstr);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002106
Nick Coghlande10c852007-12-04 12:22:52 +00002107 if ((pkgname != NULL) && (pkgname != Py_None)) {
2108 /* __package__ is set, so use it */
2109 Py_ssize_t len;
2110 if (!PyUnicode_Check(pkgname)) {
2111 PyErr_SetString(PyExc_ValueError,
2112 "__package__ set to non-string");
2113 return NULL;
2114 }
2115 len = PyUnicode_GET_SIZE(pkgname);
2116 if (len == 0) {
2117 if (level > 0) {
2118 PyErr_SetString(PyExc_ValueError,
2119 "Attempted relative import in non-package");
2120 return NULL;
2121 }
2122 return Py_None;
2123 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002124 if (len > MAXPATHLEN) {
2125 PyErr_SetString(PyExc_ValueError,
Nick Coghlande10c852007-12-04 12:22:52 +00002126 "Package name too long");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002127 return NULL;
2128 }
Nick Coghlande10c852007-12-04 12:22:52 +00002129 strcpy(buf, PyUnicode_AsString(pkgname));
2130 } else {
2131 /* __package__ not set, so figure it out and set it */
2132 modname = PyDict_GetItem(globals, namestr);
2133 if (modname == NULL || !PyUnicode_Check(modname))
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002134 return Py_None;
Nick Coghlande10c852007-12-04 12:22:52 +00002135
2136 modpath = PyDict_GetItem(globals, pathstr);
2137 if (modpath != NULL) {
2138 /* __path__ is set, so modname is already the package name */
2139 Py_ssize_t len = PyUnicode_GET_SIZE(modname);
2140 int error;
2141 if (len > MAXPATHLEN) {
2142 PyErr_SetString(PyExc_ValueError,
2143 "Module name too long");
2144 return NULL;
2145 }
2146 strcpy(buf, PyUnicode_AsString(modname));
2147 error = PyDict_SetItem(globals, pkgstr, modname);
2148 if (error) {
2149 PyErr_SetString(PyExc_ValueError,
2150 "Could not set __package__");
2151 return NULL;
2152 }
2153 } else {
2154 /* Normal module, so work out the package name if any */
2155 char *start = PyUnicode_AsString(modname);
2156 char *lastdot = strrchr(start, '.');
2157 size_t len;
2158 int error;
2159 if (lastdot == NULL && level > 0) {
2160 PyErr_SetString(PyExc_ValueError,
2161 "Attempted relative import in non-package");
2162 return NULL;
2163 }
2164 if (lastdot == NULL) {
2165 error = PyDict_SetItem(globals, pkgstr, Py_None);
2166 if (error) {
2167 PyErr_SetString(PyExc_ValueError,
2168 "Could not set __package__");
2169 return NULL;
2170 }
2171 return Py_None;
2172 }
2173 len = lastdot - start;
2174 if (len >= MAXPATHLEN) {
2175 PyErr_SetString(PyExc_ValueError,
2176 "Module name too long");
2177 return NULL;
2178 }
2179 strncpy(buf, start, len);
2180 buf[len] = '\0';
2181 pkgname = PyUnicode_FromString(buf);
2182 if (pkgname == NULL) {
2183 return NULL;
2184 }
2185 error = PyDict_SetItem(globals, pkgstr, pkgname);
2186 Py_DECREF(pkgname);
2187 if (error) {
2188 PyErr_SetString(PyExc_ValueError,
2189 "Could not set __package__");
2190 return NULL;
2191 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002192 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002193 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002194 while (--level > 0) {
2195 char *dot = strrchr(buf, '.');
2196 if (dot == NULL) {
2197 PyErr_SetString(PyExc_ValueError,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002198 "Attempted relative import beyond "
2199 "toplevel package");
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002200 return NULL;
2201 }
2202 *dot = '\0';
2203 }
2204 *p_buflen = strlen(buf);
2205
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002206 modules = PyImport_GetModuleDict();
2207 parent = PyDict_GetItemString(modules, buf);
2208 if (parent == NULL)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002209 PyErr_Format(PyExc_SystemError,
2210 "Parent module '%.200s' not loaded", buf);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002211 return parent;
2212 /* We expect, but can't guarantee, if parent != None, that:
2213 - parent.__name__ == buf
2214 - parent.__dict__ is globals
2215 If this is violated... Who cares? */
2216}
2217
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002218/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002219static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002220load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf,
Martin v. Löwis18e16552006-02-15 17:27:45 +00002221 Py_ssize_t *p_buflen)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002222{
2223 char *name = *p_name;
2224 char *dot = strchr(name, '.');
Fred Drake4c82b232000-06-30 16:18:57 +00002225 size_t len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002226 char *p;
2227 PyObject *result;
2228
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002229 if (strlen(name) == 0) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002230 /* completely empty module name should only happen in
2231 'from . import' (or '__import__("")')*/
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002232 Py_INCREF(mod);
2233 *p_name = NULL;
2234 return mod;
2235 }
2236
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002237 if (dot == NULL) {
2238 *p_name = NULL;
2239 len = strlen(name);
2240 }
2241 else {
2242 *p_name = dot+1;
2243 len = dot-name;
2244 }
Guido van Rossum111c20b1998-04-11 17:38:22 +00002245 if (len == 0) {
2246 PyErr_SetString(PyExc_ValueError,
2247 "Empty module name");
2248 return NULL;
2249 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002250
2251 p = buf + *p_buflen;
2252 if (p != buf)
2253 *p++ = '.';
2254 if (p+len-buf >= MAXPATHLEN) {
2255 PyErr_SetString(PyExc_ValueError,
2256 "Module name too long");
2257 return NULL;
2258 }
2259 strncpy(p, name, len);
2260 p[len] = '\0';
2261 *p_buflen = p+len-buf;
2262
2263 result = import_submodule(mod, p, buf);
2264 if (result == Py_None && altmod != mod) {
2265 Py_DECREF(result);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002266 /* Here, altmod must be None and mod must not be None */
Guido van Rossum0c819451997-09-07 06:16:57 +00002267 result = import_submodule(altmod, p, p);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002268 if (result != NULL && result != Py_None) {
2269 if (mark_miss(buf) != 0) {
2270 Py_DECREF(result);
2271 return NULL;
2272 }
2273 strncpy(buf, name, len);
2274 buf[len] = '\0';
2275 *p_buflen = len;
2276 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002277 }
2278 if (result == NULL)
2279 return NULL;
2280
2281 if (result == Py_None) {
2282 Py_DECREF(result);
2283 PyErr_Format(PyExc_ImportError,
2284 "No module named %.200s", name);
2285 return NULL;
2286 }
2287
2288 return result;
2289}
2290
2291static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002292mark_miss(char *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00002293{
2294 PyObject *modules = PyImport_GetModuleDict();
2295 return PyDict_SetItemString(modules, name, Py_None);
2296}
2297
2298static int
Martin v. Löwis18e16552006-02-15 17:27:45 +00002299ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen,
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002300 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002301{
2302 int i;
2303
2304 if (!PyObject_HasAttrString(mod, "__path__"))
2305 return 1;
2306
2307 for (i = 0; ; i++) {
2308 PyObject *item = PySequence_GetItem(fromlist, i);
2309 int hasit;
2310 if (item == NULL) {
2311 if (PyErr_ExceptionMatches(PyExc_IndexError)) {
2312 PyErr_Clear();
2313 return 1;
2314 }
2315 return 0;
2316 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002317 if (!PyUnicode_Check(item)) {
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002318 PyErr_SetString(PyExc_TypeError,
Neal Norwitz80e7f272007-08-26 06:45:23 +00002319 "Item in ``from list'' not a string");
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002320 Py_DECREF(item);
2321 return 0;
2322 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00002323 if (PyUnicode_AS_UNICODE(item)[0] == '*') {
Guido van Rossum9905ef91997-09-08 16:07:11 +00002324 PyObject *all;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002325 Py_DECREF(item);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002326 /* See if the package defines __all__ */
2327 if (recursive)
2328 continue; /* Avoid endless recursion */
2329 all = PyObject_GetAttrString(mod, "__all__");
2330 if (all == NULL)
2331 PyErr_Clear();
2332 else {
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002333 int ret = ensure_fromlist(mod, all, buf, buflen, 1);
Guido van Rossum9905ef91997-09-08 16:07:11 +00002334 Py_DECREF(all);
Martin v. Löwis83969ee2004-03-23 16:28:13 +00002335 if (!ret)
2336 return 0;
Guido van Rossum9905ef91997-09-08 16:07:11 +00002337 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002338 continue;
2339 }
2340 hasit = PyObject_HasAttr(mod, item);
2341 if (!hasit) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002342 PyObject *item8;
2343 char *subname;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002344 PyObject *submod;
2345 char *p;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002346 if (!Py_FileSystemDefaultEncoding) {
2347 item8 = PyUnicode_EncodeASCII(PyUnicode_AsUnicode(item),
2348 PyUnicode_GetSize(item),
Martin v. Löwis427dbff2007-06-12 05:53:00 +00002349 NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002350 } else {
Guido van Rossum98297ee2007-11-06 21:34:58 +00002351 item8 = PyUnicode_AsEncodedString(item,
2352 Py_FileSystemDefaultEncoding, NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002353 }
2354 if (!item8) {
2355 PyErr_SetString(PyExc_ValueError, "Cannot encode path item");
2356 return 0;
2357 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00002358 subname = PyString_AS_STRING(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002359 if (buflen + strlen(subname) >= MAXPATHLEN) {
2360 PyErr_SetString(PyExc_ValueError,
2361 "Module name too long");
2362 Py_DECREF(item);
2363 return 0;
2364 }
2365 p = buf + buflen;
2366 *p++ = '.';
2367 strcpy(p, subname);
2368 submod = import_submodule(mod, subname, buf);
Martin v. Löwis5b222132007-06-10 09:51:05 +00002369 Py_DECREF(item8);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002370 Py_XDECREF(submod);
2371 if (submod == NULL) {
2372 Py_DECREF(item);
2373 return 0;
2374 }
2375 }
2376 Py_DECREF(item);
2377 }
2378
Guido van Rossuma7f2e811997-10-03 15:33:32 +00002379 /* NOTREACHED */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002380}
2381
Neil Schemenauer00b09662003-06-16 21:03:07 +00002382static int
2383add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname,
2384 PyObject *modules)
2385{
2386 if (mod == Py_None)
2387 return 1;
2388 /* Irrespective of the success of this load, make a
2389 reference to it in the parent package module. A copy gets
2390 saved in the modules dictionary under the full name, so get a
2391 reference from there, if need be. (The exception is when the
2392 load failed with a SyntaxError -- then there's no trace in
2393 sys.modules. In that case, of course, do nothing extra.) */
2394 if (submod == NULL) {
2395 submod = PyDict_GetItemString(modules, fullname);
2396 if (submod == NULL)
2397 return 1;
2398 }
2399 if (PyModule_Check(mod)) {
2400 /* We can't use setattr here since it can give a
2401 * spurious warning if the submodule name shadows a
2402 * builtin name */
2403 PyObject *dict = PyModule_GetDict(mod);
2404 if (!dict)
2405 return 0;
2406 if (PyDict_SetItemString(dict, subname, submod) < 0)
2407 return 0;
2408 }
2409 else {
2410 if (PyObject_SetAttrString(mod, subname, submod) < 0)
2411 return 0;
2412 }
2413 return 1;
2414}
2415
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002416static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002417import_submodule(PyObject *mod, char *subname, char *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002418{
Guido van Rossum25ce5661997-08-02 03:10:38 +00002419 PyObject *modules = PyImport_GetModuleDict();
Neil Schemenauer00b09662003-06-16 21:03:07 +00002420 PyObject *m = NULL;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002421
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002422 /* Require:
2423 if mod == None: subname == fullname
2424 else: mod.__name__ + "." + subname == fullname
2425 */
2426
Tim Peters50d8d372001-02-28 05:34:27 +00002427 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002428 Py_INCREF(m);
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002429 }
2430 else {
Just van Rossum52e14d62002-12-30 22:08:05 +00002431 PyObject *path, *loader = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002432 char buf[MAXPATHLEN+1];
2433 struct filedescr *fdp;
2434 FILE *fp = NULL;
2435
Guido van Rossum9c0afe51998-05-19 15:09:05 +00002436 if (mod == Py_None)
2437 path = NULL;
2438 else {
2439 path = PyObject_GetAttrString(mod, "__path__");
2440 if (path == NULL) {
2441 PyErr_Clear();
2442 Py_INCREF(Py_None);
2443 return Py_None;
2444 }
2445 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002446
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002447 buf[0] = '\0';
Just van Rossum52e14d62002-12-30 22:08:05 +00002448 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1,
2449 &fp, &loader);
Guido van Rossumb68cd421998-07-01 17:36:26 +00002450 Py_XDECREF(path);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002451 if (fdp == NULL) {
2452 if (!PyErr_ExceptionMatches(PyExc_ImportError))
2453 return NULL;
2454 PyErr_Clear();
2455 Py_INCREF(Py_None);
2456 return Py_None;
2457 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002458 m = load_module(fullname, fp, buf, fdp->type, loader);
2459 Py_XDECREF(loader);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002460 if (fp)
2461 fclose(fp);
Neil Schemenauer00b09662003-06-16 21:03:07 +00002462 if (!add_submodule(mod, m, fullname, subname, modules)) {
2463 Py_XDECREF(m);
2464 m = NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002465 }
Guido van Rossum74e6a111994-08-29 12:54:38 +00002466 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002467
2468 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00002469}
2470
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002471
2472/* Re-import a module of any kind and return its module object, WITH
2473 INCREMENTED REFERENCE COUNT */
2474
Guido van Rossum79f25d91997-04-29 20:08:16 +00002475PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002476PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002477{
Guido van Rossumd8faa362007-04-27 19:54:29 +00002478 PyInterpreterState *interp = PyThreadState_Get()->interp;
2479 PyObject *modules_reloading = interp->modules_reloading;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002480 PyObject *modules = PyImport_GetModuleDict();
Guido van Rossumd8faa362007-04-27 19:54:29 +00002481 PyObject *path = NULL, *loader = NULL, *existing_m = NULL;
Guido van Rossum222ef561997-09-06 19:41:09 +00002482 char *name, *subname;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002483 char buf[MAXPATHLEN+1];
2484 struct filedescr *fdp;
2485 FILE *fp = NULL;
Tim Peters1cd70172004-08-02 03:52:12 +00002486 PyObject *newm;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002487
2488 if (modules_reloading == NULL) {
2489 Py_FatalError("PyImport_ReloadModule: "
Guido van Rossume7ba4952007-06-06 23:52:48 +00002490 "no modules_reloading dictionary!");
Guido van Rossumd8faa362007-04-27 19:54:29 +00002491 return NULL;
2492 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002493
Guido van Rossum79f25d91997-04-29 20:08:16 +00002494 if (m == NULL || !PyModule_Check(m)) {
2495 PyErr_SetString(PyExc_TypeError,
2496 "reload() argument must be module");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002497 return NULL;
2498 }
Neal Norwitz5616c6d2007-08-26 05:32:41 +00002499 name = (char*)PyModule_GetName(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002500 if (name == NULL)
2501 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002502 if (m != PyDict_GetItemString(modules, name)) {
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002503 PyErr_Format(PyExc_ImportError,
2504 "reload(): module %.200s not in sys.modules",
2505 name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002506 return NULL;
2507 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002508 existing_m = PyDict_GetItemString(modules_reloading, name);
2509 if (existing_m != NULL) {
2510 /* Due to a recursive reload, this module is already
2511 being reloaded. */
2512 Py_INCREF(existing_m);
2513 return existing_m;
2514 }
2515 if (PyDict_SetItemString(modules_reloading, name, m) < 0)
2516 return NULL;
2517
Guido van Rossum222ef561997-09-06 19:41:09 +00002518 subname = strrchr(name, '.');
2519 if (subname == NULL)
2520 subname = name;
2521 else {
2522 PyObject *parentname, *parent;
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002523 parentname = PyUnicode_FromStringAndSize(name, (subname-name));
Guido van Rossumd8faa362007-04-27 19:54:29 +00002524 if (parentname == NULL) {
2525 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002526 return NULL;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002527 }
Guido van Rossum222ef561997-09-06 19:41:09 +00002528 parent = PyDict_GetItem(modules, parentname);
2529 if (parent == NULL) {
2530 PyErr_Format(PyExc_ImportError,
2531 "reload(): parent %.200s not in sys.modules",
Christian Heimesc4cb3b82007-11-04 12:10:01 +00002532 PyUnicode_AsString(parentname));
Georg Brandl0c55f292005-09-14 06:56:20 +00002533 Py_DECREF(parentname);
Guido van Rossume7ba4952007-06-06 23:52:48 +00002534 imp_modules_reloading_clear();
Guido van Rossum222ef561997-09-06 19:41:09 +00002535 return NULL;
2536 }
Georg Brandl0c55f292005-09-14 06:56:20 +00002537 Py_DECREF(parentname);
Guido van Rossum222ef561997-09-06 19:41:09 +00002538 subname++;
2539 path = PyObject_GetAttrString(parent, "__path__");
2540 if (path == NULL)
2541 PyErr_Clear();
2542 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002543 buf[0] = '\0';
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002544 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
Guido van Rossum222ef561997-09-06 19:41:09 +00002545 Py_XDECREF(path);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002546
2547 if (fdp == NULL) {
2548 Py_XDECREF(loader);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002549 imp_modules_reloading_clear();
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002550 return NULL;
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00002551 }
2552
2553 newm = load_module(name, fp, buf, fdp->type, loader);
2554 Py_XDECREF(loader);
2555
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002556 if (fp)
2557 fclose(fp);
Tim Peters1cd70172004-08-02 03:52:12 +00002558 if (newm == NULL) {
2559 /* load_module probably removed name from modules because of
2560 * the error. Put back the original module object. We're
2561 * going to return NULL in this case regardless of whether
2562 * replacing name succeeds, so the return value is ignored.
2563 */
2564 PyDict_SetItemString(modules, name, m);
2565 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002566 imp_modules_reloading_clear();
Tim Peters1cd70172004-08-02 03:52:12 +00002567 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002568}
2569
2570
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002571/* Higher-level import emulator which emulates the "import" statement
2572 more accurately -- it invokes the __import__() function from the
2573 builtins of the current globals. This means that the import is
2574 done using whatever import hooks are installed in the current
Guido van Rossum6058eb41998-12-21 19:51:00 +00002575 environment, e.g. by "rexec".
2576 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00002577 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00002578 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002579
2580PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002581PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002582{
2583 static PyObject *silly_list = NULL;
2584 static PyObject *builtins_str = NULL;
2585 static PyObject *import_str = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002586 PyObject *globals = NULL;
2587 PyObject *import = NULL;
2588 PyObject *builtins = NULL;
2589 PyObject *r = NULL;
2590
2591 /* Initialize constant string objects */
2592 if (silly_list == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002593 import_str = PyUnicode_InternFromString("__import__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002594 if (import_str == NULL)
2595 return NULL;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002596 builtins_str = PyUnicode_InternFromString("__builtins__");
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002597 if (builtins_str == NULL)
2598 return NULL;
2599 silly_list = Py_BuildValue("[s]", "__doc__");
2600 if (silly_list == NULL)
2601 return NULL;
2602 }
2603
2604 /* Get the builtins from current globals */
2605 globals = PyEval_GetGlobals();
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002606 if (globals != NULL) {
Guido van Rossum66468561998-10-22 15:46:50 +00002607 Py_INCREF(globals);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002608 builtins = PyObject_GetItem(globals, builtins_str);
2609 if (builtins == NULL)
2610 goto err;
2611 }
2612 else {
2613 /* No globals -- use standard builtins, and fake globals */
2614 PyErr_Clear();
2615
Georg Brandl1a3284e2007-12-02 09:40:06 +00002616 builtins = PyImport_ImportModuleLevel("builtins",
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002617 NULL, NULL, NULL, 0);
Guido van Rossum85cd1d62001-02-20 21:43:24 +00002618 if (builtins == NULL)
2619 return NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002620 globals = Py_BuildValue("{OO}", builtins_str, builtins);
2621 if (globals == NULL)
2622 goto err;
2623 }
2624
2625 /* Get the __import__ function from the builtins */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002626 if (PyDict_Check(builtins)) {
Fred Drakea76ba6e2001-03-06 06:31:15 +00002627 import = PyObject_GetItem(builtins, import_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002628 if (import == NULL)
2629 PyErr_SetObject(PyExc_KeyError, import_str);
2630 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002631 else
Fred Drakea76ba6e2001-03-06 06:31:15 +00002632 import = PyObject_GetAttr(builtins, import_str);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002633 if (import == NULL)
2634 goto err;
2635
Christian Heimes072c0f12008-01-03 23:01:04 +00002636 /* Call the __import__ function with the proper argument list
2637 * Always use absolute import here. */
2638 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
2639 globals, silly_list, 0, NULL);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002640
2641 err:
2642 Py_XDECREF(globals);
2643 Py_XDECREF(builtins);
2644 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00002645
Guido van Rossumd47a0a81997-08-14 20:11:26 +00002646 return r;
2647}
2648
2649
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002650/* Module 'imp' provides Python access to the primitives used for
2651 importing modules.
2652*/
2653
Guido van Rossum79f25d91997-04-29 20:08:16 +00002654static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002655imp_get_magic(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002656{
2657 char buf[4];
2658
Guido van Rossum96774c12000-05-01 20:19:08 +00002659 buf[0] = (char) ((pyc_magic >> 0) & 0xff);
2660 buf[1] = (char) ((pyc_magic >> 8) & 0xff);
2661 buf[2] = (char) ((pyc_magic >> 16) & 0xff);
2662 buf[3] = (char) ((pyc_magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002663
Guido van Rossumad8d3002007-08-03 18:40:49 +00002664 return PyBytes_FromStringAndSize(buf, 4);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002665}
2666
Guido van Rossum79f25d91997-04-29 20:08:16 +00002667static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00002668imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002669{
Guido van Rossum79f25d91997-04-29 20:08:16 +00002670 PyObject *list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002671 struct filedescr *fdp;
2672
Guido van Rossum79f25d91997-04-29 20:08:16 +00002673 list = PyList_New(0);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002674 if (list == NULL)
2675 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002676 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
2677 PyObject *item = Py_BuildValue("ssi",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002678 fdp->suffix, fdp->mode, fdp->type);
2679 if (item == NULL) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002680 Py_DECREF(list);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002681 return NULL;
2682 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002683 if (PyList_Append(list, item) < 0) {
2684 Py_DECREF(list);
2685 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002686 return NULL;
2687 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002688 Py_DECREF(item);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002689 }
2690 return list;
2691}
2692
Guido van Rossum79f25d91997-04-29 20:08:16 +00002693static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002694call_find_module(char *name, PyObject *path)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002695{
Tim Petersdbd9ba62000-07-09 03:09:57 +00002696 extern int fclose(FILE *);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002697 PyObject *fob, *ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002698 struct filedescr *fdp;
2699 char pathname[MAXPATHLEN+1];
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002700 FILE *fp = NULL;
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002701 int fd = -1;
Brett Cannon3bb42d92007-10-20 03:43:15 +00002702 char *found_encoding = NULL;
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002703 char *encoding = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002704
2705 pathname[0] = '\0';
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002706 if (path == Py_None)
2707 path = NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00002708 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002709 if (fdp == NULL)
2710 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002711 if (fp != NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002712 fd = fileno(fp);
2713 if (fd != -1)
2714 fd = dup(fd);
2715 fclose(fp);
2716 fp = NULL;
2717 }
2718 if (fd != -1) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002719 if (strchr(fdp->mode, 'b') == NULL) {
Brett Cannon3bb42d92007-10-20 03:43:15 +00002720 /* PyTokenizer_FindEncoding() returns PyMem_MALLOC'ed
2721 memory. */
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002722 found_encoding = PyTokenizer_FindEncoding(fd);
2723 lseek(fd, 0, 0); /* Reset position */
Brett Cannon3bb42d92007-10-20 03:43:15 +00002724 encoding = (found_encoding != NULL) ? found_encoding :
Guido van Rossumce3a72a2007-10-19 23:16:50 +00002725 (char*)PyUnicode_GetDefaultEncoding();
2726 }
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002727 fob = PyFile_FromFd(fd, pathname, fdp->mode, -1,
Guido van Rossume7fc50f2007-12-03 22:54:21 +00002728 (char*)encoding, NULL, NULL, 1);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002729 if (fob == NULL) {
Guido van Rossum40d20bc2007-10-22 00:09:51 +00002730 close(fd);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002731 PyMem_FREE(found_encoding);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002732 return NULL;
2733 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002734 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002735 else {
2736 fob = Py_None;
2737 Py_INCREF(fob);
Tim Peters50d8d372001-02-28 05:34:27 +00002738 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002739 ret = Py_BuildValue("Os(ssi)",
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002740 fob, pathname, fdp->suffix, fdp->mode, fdp->type);
Guido van Rossum79f25d91997-04-29 20:08:16 +00002741 Py_DECREF(fob);
Brett Cannon3bb42d92007-10-20 03:43:15 +00002742 PyMem_FREE(found_encoding);
2743
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002744 return ret;
2745}
2746
Guido van Rossum79f25d91997-04-29 20:08:16 +00002747static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002748imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002749{
2750 char *name;
2751 PyObject *path = NULL;
Guido van Rossum43713e52000-02-29 13:59:29 +00002752 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002753 return NULL;
2754 return call_find_module(name, path);
2755}
2756
2757static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002758imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002759{
2760 char *name;
2761 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002762 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002763 if (!PyArg_ParseTuple(args, "s:init_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002764 return NULL;
2765 ret = init_builtin(name);
2766 if (ret < 0)
2767 return NULL;
2768 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002769 Py_INCREF(Py_None);
2770 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002771 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002772 m = PyImport_AddModule(name);
2773 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002774 return m;
2775}
2776
Guido van Rossum79f25d91997-04-29 20:08:16 +00002777static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002778imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002779{
2780 char *name;
2781 int ret;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002782 PyObject *m;
Guido van Rossum43713e52000-02-29 13:59:29 +00002783 if (!PyArg_ParseTuple(args, "s:init_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002784 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002785 ret = PyImport_ImportFrozenModule(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002786 if (ret < 0)
2787 return NULL;
2788 if (ret == 0) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00002789 Py_INCREF(Py_None);
2790 return Py_None;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002791 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002792 m = PyImport_AddModule(name);
2793 Py_XINCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002794 return m;
2795}
2796
Guido van Rossum79f25d91997-04-29 20:08:16 +00002797static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002798imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002799{
2800 char *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00002801
Guido van Rossum43713e52000-02-29 13:59:29 +00002802 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name))
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002803 return NULL;
2804 return get_frozen_object(name);
2805}
2806
Guido van Rossum79f25d91997-04-29 20:08:16 +00002807static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002808imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002809{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002810 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002811 if (!PyArg_ParseTuple(args, "s:is_builtin", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002812 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00002813 return PyLong_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002814}
2815
Guido van Rossum79f25d91997-04-29 20:08:16 +00002816static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002817imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002818{
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002819 char *name;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002820 struct _frozen *p;
Guido van Rossum43713e52000-02-29 13:59:29 +00002821 if (!PyArg_ParseTuple(args, "s:is_frozen", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002822 return NULL;
Guido van Rossum323bf5e1998-06-24 03:54:06 +00002823 p = find_frozen(name);
Guido van Rossumb8bff3f2002-04-07 06:34:38 +00002824 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002825}
2826
2827static FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002828get_file(char *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002829{
2830 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002831 if (mode[0] == 'U')
2832 mode = "r" PY_STDIOTEXTMODE;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002833 if (fob == NULL) {
2834 fp = fopen(pathname, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002835 }
2836 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002837 int fd = PyObject_AsFileDescriptor(fob);
2838 if (fd == -1)
2839 return NULL;
2840 /* XXX This will leak a FILE struct. Fix this!!!!
2841 (But it doesn't leak a file descrioptor!) */
2842 fp = fdopen(fd, mode);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002843 }
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002844 if (fp == NULL)
2845 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002846 return fp;
2847}
2848
Guido van Rossum79f25d91997-04-29 20:08:16 +00002849static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002850imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002851{
2852 char *name;
2853 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002854 PyObject *fob = NULL;
2855 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002856 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002857 if (!PyArg_ParseTuple(args, "ss|O:load_compiled",
2858 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002859 return NULL;
2860 fp = get_file(pathname, fob, "rb");
2861 if (fp == NULL)
2862 return NULL;
2863 m = load_compiled_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002864 if (fob == NULL)
2865 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002866 return m;
2867}
2868
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002869#ifdef HAVE_DYNAMIC_LOADING
2870
Guido van Rossum79f25d91997-04-29 20:08:16 +00002871static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002872imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002873{
2874 char *name;
2875 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002876 PyObject *fob = NULL;
2877 PyObject *m;
Guido van Rossum7faeab31995-07-07 22:50:36 +00002878 FILE *fp = NULL;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002879 if (!PyArg_ParseTuple(args, "ss|O:load_dynamic",
2880 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002881 return NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002882 if (fob) {
Guido van Rossum7faeab31995-07-07 22:50:36 +00002883 fp = get_file(pathname, fob, "r");
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002884 if (fp == NULL)
2885 return NULL;
2886 }
Guido van Rossum79f25d91997-04-29 20:08:16 +00002887 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Guido van Rossum7faeab31995-07-07 22:50:36 +00002888 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002889}
2890
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002891#endif /* HAVE_DYNAMIC_LOADING */
2892
Guido van Rossum79f25d91997-04-29 20:08:16 +00002893static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002894imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002895{
2896 char *name;
2897 char *pathname;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002898 PyObject *fob = NULL;
2899 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002900 FILE *fp;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002901 if (!PyArg_ParseTuple(args, "ss|O:load_source",
2902 &name, &pathname, &fob))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002903 return NULL;
2904 fp = get_file(pathname, fob, "r");
2905 if (fp == NULL)
2906 return NULL;
2907 m = load_source_module(name, pathname, fp);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002908 if (fob == NULL)
2909 fclose(fp);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002910 return m;
2911}
2912
Guido van Rossum79f25d91997-04-29 20:08:16 +00002913static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002914imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002915{
2916 char *name;
2917 PyObject *fob;
2918 char *pathname;
2919 char *suffix; /* Unused */
2920 char *mode;
2921 int type;
2922 FILE *fp;
2923
Guido van Rossum43713e52000-02-29 13:59:29 +00002924 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module",
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002925 &name, &fob, &pathname,
2926 &suffix, &mode, &type))
2927 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002928 if (*mode) {
2929 /* Mode must start with 'r' or 'U' and must not contain '+'.
2930 Implicit in this test is the assumption that the mode
2931 may contain other modifiers like 'b' or 't'. */
2932
2933 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
Jeremy Hylton4ae6fae2002-05-30 17:15:25 +00002934 PyErr_Format(PyExc_ValueError,
2935 "invalid file open mode %.200s", mode);
2936 return NULL;
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00002937 }
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002938 }
2939 if (fob == Py_None)
2940 fp = NULL;
2941 else {
Guido van Rossumda5b8f22007-06-12 23:30:11 +00002942 fp = get_file(NULL, fob, mode);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002943 if (fp == NULL)
2944 return NULL;
2945 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002946 return load_module(name, fp, pathname, type, NULL);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002947}
2948
2949static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002950imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002951{
2952 char *name;
2953 char *pathname;
Guido van Rossum43713e52000-02-29 13:59:29 +00002954 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname))
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002955 return NULL;
2956 return load_package(name, pathname);
2957}
2958
2959static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002960imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002961{
2962 char *name;
Guido van Rossum43713e52000-02-29 13:59:29 +00002963 if (!PyArg_ParseTuple(args, "s:new_module", &name))
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002964 return NULL;
Guido van Rossum79f25d91997-04-29 20:08:16 +00002965 return PyModule_New(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002966}
2967
Christian Heimes13a7a212008-01-07 17:13:09 +00002968static PyObject *
2969imp_reload(PyObject *self, PyObject *v)
2970{
2971 return PyImport_ReloadModule(v);
2972}
2973
2974PyDoc_STRVAR(doc_reload,
2975"reload(module) -> module\n\
2976\n\
2977Reload the module. The module must have been successfully imported before.");
2978
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002979/* Doc strings */
2980
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002981PyDoc_STRVAR(doc_imp,
2982"This module provides the components needed to build your own\n\
2983__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002984
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002985PyDoc_STRVAR(doc_find_module,
2986"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002987Search for a module. If path is omitted or None, search for a\n\
2988built-in, frozen or special module and continue search in sys.path.\n\
2989The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002990package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002991
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002992PyDoc_STRVAR(doc_load_module,
2993"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002994Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002995The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00002996
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002997PyDoc_STRVAR(doc_get_magic,
2998"get_magic() -> string\n\
2999Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003000
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003001PyDoc_STRVAR(doc_get_suffixes,
3002"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003003Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003004that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003005
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003006PyDoc_STRVAR(doc_new_module,
3007"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003008Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003009The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003010
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003011PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00003012"lock_held() -> boolean\n\
3013Return True if the import lock is currently held, else False.\n\
3014On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00003015
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003016PyDoc_STRVAR(doc_acquire_lock,
3017"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00003018Acquires the interpreter's import lock for the current thread.\n\
3019This lock should be used by import hooks to ensure thread-safety\n\
3020when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00003021On platforms without threads, this function does nothing.");
3022
3023PyDoc_STRVAR(doc_release_lock,
3024"release_lock() -> None\n\
3025Release the interpreter's import lock.\n\
3026On platforms without threads, this function does nothing.");
3027
Guido van Rossum79f25d91997-04-29 20:08:16 +00003028static PyMethodDef imp_methods[] = {
Neal Norwitz08ea61a2003-02-17 18:18:00 +00003029 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
3030 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
3031 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
3032 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
3033 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
3034 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
3035 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
3036 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
Christian Heimes13a7a212008-01-07 17:13:09 +00003037 {"reload", imp_reload, METH_O, doc_reload},
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003038 /* The rest are obsolete */
Neal Norwitz031829d2002-03-31 14:37:44 +00003039 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
3040 {"init_builtin", imp_init_builtin, METH_VARARGS},
3041 {"init_frozen", imp_init_frozen, METH_VARARGS},
3042 {"is_builtin", imp_is_builtin, METH_VARARGS},
3043 {"is_frozen", imp_is_frozen, METH_VARARGS},
3044 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003045#ifdef HAVE_DYNAMIC_LOADING
Neal Norwitz031829d2002-03-31 14:37:44 +00003046 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003047#endif
Neal Norwitz031829d2002-03-31 14:37:44 +00003048 {"load_package", imp_load_package, METH_VARARGS},
Neal Norwitz031829d2002-03-31 14:37:44 +00003049 {"load_source", imp_load_source, METH_VARARGS},
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003050 {NULL, NULL} /* sentinel */
3051};
3052
Guido van Rossum1a8791e1998-08-04 22:46:29 +00003053static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003054setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003055{
3056 PyObject *v;
3057 int err;
3058
Christian Heimes217cfd12007-12-02 14:31:20 +00003059 v = PyLong_FromLong((long)value);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003060 err = PyDict_SetItemString(d, name, v);
3061 Py_XDECREF(v);
3062 return err;
3063}
3064
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003065typedef struct {
3066 PyObject_HEAD
3067} NullImporter;
3068
3069static int
3070NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
3071{
3072 char *path;
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003073 Py_ssize_t pathlen;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003074
3075 if (!_PyArg_NoKeywords("NullImporter()", kwds))
3076 return -1;
3077
3078 if (!PyArg_ParseTuple(args, "s:NullImporter",
3079 &path))
3080 return -1;
3081
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003082 pathlen = strlen(path);
3083 if (pathlen == 0) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003084 PyErr_SetString(PyExc_ImportError, "empty pathname");
3085 return -1;
3086 } else {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003087 struct stat statbuf;
3088 int rv;
3089
3090 rv = stat(path, &statbuf);
Guido van Rossumc1bdbc32007-11-07 17:46:34 +00003091#ifdef MS_WINDOWS
3092 /* MS Windows stat() chokes on paths like C:\path\. Try to
3093 * recover *one* time by stripping off a trailing slash or
3094 * backslash. http://bugs.python.org/issue1293
3095 */
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003096 if (rv != 0 && pathlen <= MAXPATHLEN &&
3097 (path[pathlen-1] == '/' || path[pathlen-1] == '\\')) {
3098 char mangled[MAXPATHLEN+1];
3099
3100 strcpy(mangled, path);
3101 mangled[pathlen-1] = '\0';
3102 rv = stat(mangled, &statbuf);
3103 }
Christian Heimes7d3bc0a2007-11-07 17:26:16 +00003104#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003105 if (rv == 0) {
3106 /* it exists */
3107 if (S_ISDIR(statbuf.st_mode)) {
3108 /* it's a directory */
3109 PyErr_SetString(PyExc_ImportError,
3110 "existing directory");
3111 return -1;
3112 }
3113 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003114 }
3115 return 0;
3116}
3117
3118static PyObject *
3119NullImporter_find_module(NullImporter *self, PyObject *args)
3120{
3121 Py_RETURN_NONE;
3122}
3123
3124static PyMethodDef NullImporter_methods[] = {
3125 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
3126 "Always return None"
3127 },
3128 {NULL} /* Sentinel */
3129};
3130
3131
Christian Heimes9cd17752007-11-18 19:35:23 +00003132PyTypeObject PyNullImporter_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00003133 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003134 "imp.NullImporter", /*tp_name*/
3135 sizeof(NullImporter), /*tp_basicsize*/
3136 0, /*tp_itemsize*/
3137 0, /*tp_dealloc*/
3138 0, /*tp_print*/
3139 0, /*tp_getattr*/
3140 0, /*tp_setattr*/
3141 0, /*tp_compare*/
3142 0, /*tp_repr*/
3143 0, /*tp_as_number*/
3144 0, /*tp_as_sequence*/
3145 0, /*tp_as_mapping*/
3146 0, /*tp_hash */
3147 0, /*tp_call*/
3148 0, /*tp_str*/
3149 0, /*tp_getattro*/
3150 0, /*tp_setattro*/
3151 0, /*tp_as_buffer*/
3152 Py_TPFLAGS_DEFAULT, /*tp_flags*/
3153 "Null importer object", /* tp_doc */
3154 0, /* tp_traverse */
3155 0, /* tp_clear */
3156 0, /* tp_richcompare */
3157 0, /* tp_weaklistoffset */
3158 0, /* tp_iter */
3159 0, /* tp_iternext */
3160 NullImporter_methods, /* tp_methods */
3161 0, /* tp_members */
3162 0, /* tp_getset */
3163 0, /* tp_base */
3164 0, /* tp_dict */
3165 0, /* tp_descr_get */
3166 0, /* tp_descr_set */
3167 0, /* tp_dictoffset */
3168 (initproc)NullImporter_init, /* tp_init */
3169 0, /* tp_alloc */
3170 PyType_GenericNew /* tp_new */
3171};
3172
3173
Jason Tishler6bc06ec2003-09-04 11:59:50 +00003174PyMODINIT_FUNC
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003175initimp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003176{
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003177 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003178
Christian Heimes9cd17752007-11-18 19:35:23 +00003179 if (PyType_Ready(&PyNullImporter_Type) < 0)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003180 goto failure;
3181
Guido van Rossum0207e6d1997-09-09 22:04:42 +00003182 m = Py_InitModule4("imp", imp_methods, doc_imp,
3183 NULL, PYTHON_API_VERSION);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00003184 if (m == NULL)
3185 goto failure;
Guido van Rossum79f25d91997-04-29 20:08:16 +00003186 d = PyModule_GetDict(m);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003187 if (d == NULL)
3188 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003189
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003190 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
3191 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
3192 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
3193 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
3194 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
3195 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
3196 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
3197 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
Guido van Rossum0f84a341998-08-06 13:36:01 +00003198 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
Just van Rossum52e14d62002-12-30 22:08:05 +00003199 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003200
Christian Heimes9cd17752007-11-18 19:35:23 +00003201 Py_INCREF(&PyNullImporter_Type);
3202 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003203 failure:
3204 ;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003205}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003206
3207
Guido van Rossumb18618d2000-05-03 23:44:39 +00003208/* API for embedding applications that want to add their own entries
3209 to the table of built-in modules. This should normally be called
3210 *before* Py_Initialize(). When the table resize fails, -1 is
3211 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003212
3213 After a similar function by Just van Rossum. */
3214
3215int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003216PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003217{
3218 static struct _inittab *our_copy = NULL;
3219 struct _inittab *p;
3220 int i, n;
3221
3222 /* Count the number of entries in both tables */
3223 for (n = 0; newtab[n].name != NULL; n++)
3224 ;
3225 if (n == 0)
3226 return 0; /* Nothing to do */
3227 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
3228 ;
3229
3230 /* Allocate new memory for the combined table */
Guido van Rossumb18618d2000-05-03 23:44:39 +00003231 p = our_copy;
3232 PyMem_RESIZE(p, struct _inittab, i+n+1);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003233 if (p == NULL)
3234 return -1;
3235
3236 /* Copy the tables into the new memory */
3237 if (our_copy != PyImport_Inittab)
3238 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
3239 PyImport_Inittab = our_copy = p;
3240 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
3241
3242 return 0;
3243}
3244
3245/* Shorthand to add a single entry given a name and a function */
3246
3247int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003248PyImport_AppendInittab(char *name, void (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00003249{
3250 struct _inittab newtab[2];
3251
3252 memset(newtab, '\0', sizeof newtab);
3253
3254 newtab[0].name = name;
3255 newtab[0].initfunc = initfunc;
3256
3257 return PyImport_ExtendInittab(newtab);
3258}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003259
3260#ifdef __cplusplus
3261}
3262#endif