blob: ebaaeacebd8c0fddd68768ec0a37f9c677854d5b [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 */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008#include "errcode.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +00009#include "marshal.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000010#include "code.h"
Guido van Rossumd8bac6d1992-02-26 15:19:13 +000011#include "osdefs.h"
Guido van Rossum1ae940a1995-01-02 19:04:15 +000012#include "importdl.h"
Guido van Rossumc405b7b1991-06-04 19:39:42 +000013
Guido van Rossum55a83382000-09-20 20:31:38 +000014#ifdef HAVE_FCNTL_H
15#include <fcntl.h>
16#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000017#ifdef __cplusplus
Brett Cannon9a5b25a2010-03-01 02:09:17 +000018extern "C" {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000019#endif
Guido van Rossum55a83382000-09-20 20:31:38 +000020
Christian Heimesd3eb5a152008-02-24 00:38:49 +000021#ifdef MS_WINDOWS
22/* for stat.st_mode */
23typedef unsigned short mode_t;
Daniel Stutzbachc7937792010-09-09 21:18:04 +000024/* for _mkdir */
25#include <direct.h>
Christian Heimesd3eb5a152008-02-24 00:38:49 +000026#endif
27
Guido van Rossum21d335e1993-10-15 13:01:11 +000028
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000029/* Magic word to reject .pyc files generated by other Python versions.
30 It should change for each incompatible change to the bytecode.
31
32 The value of CR and LF is incorporated so if you ever read or write
Guido van Rossum7faeab31995-07-07 22:50:36 +000033 a .pyc file in text mode the magic number will be wrong; also, the
Tim Peters36515e22001-11-18 04:06:29 +000034 Apple MPW compiler swaps their values, botching string constants.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000035
Guido van Rossum45aecf42006-03-15 04:58:47 +000036 The magic numbers must be spaced apart at least 2 values, as the
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000037 -U interpeter flag will cause MAGIC+1 being used. They have been
38 odd numbers for some time now.
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000039
Jeremy Hyltond4ceb312004-04-01 02:45:22 +000040 There were a variety of old schemes for setting the magic number.
41 The current working scheme is to increment the previous value by
42 10.
Guido van Rossumf6894922002-08-31 15:16:14 +000043
Barry Warsaw28a691b2010-04-17 00:19:56 +000044 Starting with the adoption of PEP 3147 in Python 3.2, every bump in magic
45 number also includes a new "magic tag", i.e. a human readable string used
46 to represent the magic number in __pycache__ directories. When you change
47 the magic number, you must also set a new unique magic tag. Generally this
48 can be named after the Python major version of the magic number bump, but
49 it can really be anything, as long as it's different than anything else
50 that's come before. The tags are included in the following table, starting
51 with Python 3.2a0.
52
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000053 Known values:
54 Python 1.5: 20121
55 Python 1.5.1: 20121
56 Python 1.5.2: 20121
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000057 Python 1.6: 50428
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +000058 Python 2.0: 50823
59 Python 2.0.1: 50823
60 Python 2.1: 60202
61 Python 2.1.1: 60202
62 Python 2.1.2: 60202
63 Python 2.2: 60717
Neal Norwitz7fdcb412002-06-14 01:07:39 +000064 Python 2.3a0: 62011
Michael W. Hudsondd32a912002-08-15 14:59:02 +000065 Python 2.3a0: 62021
Guido van Rossumf6894922002-08-31 15:16:14 +000066 Python 2.3a0: 62011 (!)
Martin v. Löwisef82d2f2004-06-27 16:51:46 +000067 Python 2.4a0: 62041
Raymond Hettingerfd2d1f72004-08-23 23:37:48 +000068 Python 2.4a3: 62051
Raymond Hettinger2c31a052004-09-22 18:44:21 +000069 Python 2.4b1: 62061
Michael W. Hudsondf888462005-06-03 14:41:55 +000070 Python 2.5a0: 62071
Michael W. Hudsonaee2e282005-10-21 11:32:20 +000071 Python 2.5a0: 62081 (ast-branch)
Guido van Rossumc2e20742006-02-27 22:32:47 +000072 Python 2.5a0: 62091 (with)
Guido van Rossumf6694362006-03-10 02:28:35 +000073 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000074 Python 2.5b3: 62101 (fix wrong code: for x, in ...)
75 Python 2.5b3: 62111 (fix wrong code: x += yield)
76 Python 2.5c1: 62121 (fix wrong lnotab with for loops and
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000077 storing constants that should have been removed)
Thomas Wouters89f507f2006-12-13 04:49:30 +000078 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp)
Christian Heimes99170a52007-12-19 02:07:34 +000079 Python 2.6a0: 62151 (peephole optimizations and STORE_MAP opcode)
Christian Heimesdd15f6c2008-03-16 00:07:10 +000080 Python 2.6a1: 62161 (WITH_CLEANUP optimization)
Guido van Rossum45aecf42006-03-15 04:58:47 +000081 Python 3000: 3000
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082 3010 (removed UNARY_CONVERT)
83 3020 (added BUILD_SET)
84 3030 (added keyword-only parameters)
85 3040 (added signature annotations)
86 3050 (print becomes a function)
87 3060 (PEP 3115 metaclass syntax)
88 3061 (string literals become unicode)
89 3071 (PEP 3109 raise changes)
90 3081 (PEP 3137 make __file__ and __name__ unicode)
91 3091 (kill str8 interning)
92 3101 (merge from 2.6a0, see 62151)
93 3103 (__file__ points to source file)
Benjamin Peterson0f6cae02009-12-10 02:09:08 +000094 Python 3.0a4: 3111 (WITH_CLEANUP optimization).
95 Python 3.0a5: 3131 (lexical exception stacking, including POP_EXCEPT)
96 Python 3.1a0: 3141 (optimize list, set and dict comprehensions:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000097 change LIST_APPEND and SET_ADD, add MAP_ADD)
Benjamin Peterson0f6cae02009-12-10 02:09:08 +000098 Python 3.1a0: 3151 (optimize conditional branches:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000099 introduce POP_JUMP_IF_FALSE and POP_JUMP_IF_TRUE)
Benjamin Peterson876b2f22009-06-28 03:18:59 +0000100 Python 3.2a0: 3160 (add SETUP_WITH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000101 tag: cpython-32
Antoine Pitrou74a69fa2010-09-04 18:43:52 +0000102 Python 3.2a1: 3170 (add DUP_TOP_TWO, remove DUP_TOPX and ROT_FOUR)
103 tag: cpython-32
Benjamin Peterson6246d6d2010-09-10 21:51:44 +0000104 Python 3.2a2 3180 (add DELETE_DEREF)
Benjamin Petersone7c15fa2011-06-19 19:54:45 -0500105 Python 3.3a0 3190 __class__ super closure changed
Tim Peters36515e22001-11-18 04:06:29 +0000106*/
Guido van Rossum3ddee711991-12-16 13:06:34 +0000107
Nick Coghlancd419ab2010-09-11 00:39:25 +0000108/* MAGIC must change whenever the bytecode emitted by the compiler may no
109 longer be understood by older implementations of the eval loop (usually
110 due to the addition of new opcodes)
Martin v. Löwisfadcd312011-10-23 18:08:20 +0200111 TAG must change for each major Python release. The magic number will take
112 care of any bytecode changes that occur during development.
Barry Warsaw28a691b2010-04-17 00:19:56 +0000113*/
Benjamin Peterson48deae12011-06-03 17:50:16 -0500114#define QUOTE(arg) #arg
115#define STRIFY(name) QUOTE(name)
116#define MAJOR STRIFY(PY_MAJOR_VERSION)
117#define MINOR STRIFY(PY_MINOR_VERSION)
Benjamin Petersone7c15fa2011-06-19 19:54:45 -0500118#define MAGIC (3190 | ((long)'\r'<<16) | ((long)'\n'<<24))
Benjamin Peterson48deae12011-06-03 17:50:16 -0500119#define TAG "cpython-" MAJOR MINOR;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000120#define CACHEDIR "__pycache__"
121/* Current magic word and string tag as globals. */
Guido van Rossum96774c12000-05-01 20:19:08 +0000122static long pyc_magic = MAGIC;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000123static const char *pyc_tag = TAG;
Benjamin Peterson48deae12011-06-03 17:50:16 -0500124#undef QUOTE
125#undef STRIFY
126#undef MAJOR
127#undef MINOR
Guido van Rossum96774c12000-05-01 20:19:08 +0000128
Victor Stinner95872862011-03-07 18:20:56 +0100129/* See _PyImport_FixupExtensionObject() below */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000130static PyObject *extensions = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000131
Victor Stinnerfe7c5b52011-04-05 01:48:03 +0200132/* Function from Parser/tokenizer.c */
133extern char * PyTokenizer_FindEncodingFilename(int, PyObject *);
134
Guido van Rossum771c6c81997-10-31 18:37:24 +0000135/* This table is defined in config.c: */
136extern struct _inittab _PyImport_Inittab[];
137
138struct _inittab *PyImport_Inittab = _PyImport_Inittab;
Guido van Rossum66f1fa81991-04-03 19:03:52 +0000139
Guido van Rossumed1170e1999-12-20 21:23:41 +0000140/* these tables define the module suffixes that Python recognizes */
141struct filedescr * _PyImport_Filetab = NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000142
Guido van Rossumed1170e1999-12-20 21:23:41 +0000143static const struct filedescr _PyImport_StandardFiletab[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000144 {".py", "U", PY_SOURCE},
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000145#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000146 {".pyw", "U", PY_SOURCE},
Tim Peters36515e22001-11-18 04:06:29 +0000147#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000148 {".pyc", "rb", PY_COMPILED},
149 {0, 0}
Guido van Rossumed1170e1999-12-20 21:23:41 +0000150};
151
Victor Stinnerd0296212011-03-14 14:04:10 -0400152static PyObject *initstr = NULL;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200153_Py_IDENTIFIER(__path__);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000154
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000155/* Initialize things */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000156
157void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000158_PyImport_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000159{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000160 const struct filedescr *scan;
161 struct filedescr *filetab;
162 int countD = 0;
163 int countS = 0;
Guido van Rossumed1170e1999-12-20 21:23:41 +0000164
Victor Stinnerd0296212011-03-14 14:04:10 -0400165 initstr = PyUnicode_InternFromString("__init__");
166 if (initstr == NULL)
167 Py_FatalError("Can't initialize import variables");
168
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000169 /* prepare _PyImport_Filetab: copy entries from
170 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab.
171 */
Guido van Rossum04110fb2007-08-24 16:32:05 +0000172#ifdef HAVE_DYNAMIC_LOADING
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000173 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan)
174 ++countD;
Guido van Rossum04110fb2007-08-24 16:32:05 +0000175#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000176 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan)
177 ++countS;
178 filetab = PyMem_NEW(struct filedescr, countD + countS + 1);
179 if (filetab == NULL)
180 Py_FatalError("Can't initialize import file table.");
Guido van Rossum04110fb2007-08-24 16:32:05 +0000181#ifdef HAVE_DYNAMIC_LOADING
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000182 memcpy(filetab, _PyImport_DynLoadFiletab,
183 countD * sizeof(struct filedescr));
Guido van Rossum04110fb2007-08-24 16:32:05 +0000184#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000185 memcpy(filetab + countD, _PyImport_StandardFiletab,
186 countS * sizeof(struct filedescr));
187 filetab[countD + countS].suffix = NULL;
Guido van Rossumed1170e1999-12-20 21:23:41 +0000188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000189 _PyImport_Filetab = filetab;
Guido van Rossumed1170e1999-12-20 21:23:41 +0000190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000191 if (Py_OptimizeFlag) {
192 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */
193 for (; filetab->suffix != NULL; filetab++) {
194 if (strcmp(filetab->suffix, ".pyc") == 0)
195 filetab->suffix = ".pyo";
196 }
197 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000198}
199
Guido van Rossum25ce5661997-08-02 03:10:38 +0000200void
Just van Rossum52e14d62002-12-30 22:08:05 +0000201_PyImportHooks_Init(void)
202{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000203 PyObject *v, *path_hooks = NULL, *zimpimport;
204 int err = 0;
Just van Rossum52e14d62002-12-30 22:08:05 +0000205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000206 /* adding sys.path_hooks and sys.path_importer_cache, setting up
207 zipimport */
208 if (PyType_Ready(&PyNullImporter_Type) < 0)
209 goto error;
Just van Rossum52e14d62002-12-30 22:08:05 +0000210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000211 if (Py_VerboseFlag)
212 PySys_WriteStderr("# installing zipimport hook\n");
Just van Rossum52e14d62002-12-30 22:08:05 +0000213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000214 v = PyList_New(0);
215 if (v == NULL)
216 goto error;
217 err = PySys_SetObject("meta_path", v);
218 Py_DECREF(v);
219 if (err)
220 goto error;
221 v = PyDict_New();
222 if (v == NULL)
223 goto error;
224 err = PySys_SetObject("path_importer_cache", v);
225 Py_DECREF(v);
226 if (err)
227 goto error;
228 path_hooks = PyList_New(0);
229 if (path_hooks == NULL)
230 goto error;
231 err = PySys_SetObject("path_hooks", path_hooks);
232 if (err) {
Just van Rossum52e14d62002-12-30 22:08:05 +0000233 error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000234 PyErr_Print();
235 Py_FatalError("initializing sys.meta_path, sys.path_hooks, "
236 "path_importer_cache, or NullImporter failed"
237 );
238 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000239
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000240 zimpimport = PyImport_ImportModule("zipimport");
241 if (zimpimport == NULL) {
242 PyErr_Clear(); /* No zip import module -- okay */
243 if (Py_VerboseFlag)
244 PySys_WriteStderr("# can't import zipimport\n");
245 }
246 else {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200247 _Py_IDENTIFIER(zipimporter);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200248 PyObject *zipimporter = _PyObject_GetAttrId(zimpimport,
249 &PyId_zipimporter);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000250 Py_DECREF(zimpimport);
251 if (zipimporter == NULL) {
252 PyErr_Clear(); /* No zipimporter object -- okay */
253 if (Py_VerboseFlag)
254 PySys_WriteStderr(
255 "# can't import zipimport.zipimporter\n");
256 }
257 else {
258 /* sys.path_hooks.append(zipimporter) */
259 err = PyList_Append(path_hooks, zipimporter);
260 Py_DECREF(zipimporter);
261 if (err)
262 goto error;
263 if (Py_VerboseFlag)
264 PySys_WriteStderr(
265 "# installed zipimport hook\n");
266 }
267 }
268 Py_DECREF(path_hooks);
Just van Rossum52e14d62002-12-30 22:08:05 +0000269}
270
271void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000272_PyImport_Fini(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000273{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000274 Py_XDECREF(extensions);
275 extensions = NULL;
276 PyMem_DEL(_PyImport_Filetab);
277 _PyImport_Filetab = NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000278}
279
280
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000281/* Locking primitives to prevent parallel imports of the same module
282 in different threads to return with a partially loaded module.
283 These calls are serialized by the global interpreter lock. */
284
285#ifdef WITH_THREAD
286
Guido van Rossum49b56061998-10-01 20:42:43 +0000287#include "pythread.h"
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000288
Guido van Rossum65d5b571998-12-21 19:32:43 +0000289static PyThread_type_lock import_lock = 0;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000290static long import_lock_thread = -1;
291static int import_lock_level = 0;
292
Benjamin Peterson0df35a92009-10-04 20:32:25 +0000293void
294_PyImport_AcquireLock(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000295{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000296 long me = PyThread_get_thread_ident();
297 if (me == -1)
298 return; /* Too bad */
299 if (import_lock == NULL) {
300 import_lock = PyThread_allocate_lock();
301 if (import_lock == NULL)
302 return; /* Nothing much we can do. */
303 }
304 if (import_lock_thread == me) {
305 import_lock_level++;
306 return;
307 }
308 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0))
309 {
310 PyThreadState *tstate = PyEval_SaveThread();
311 PyThread_acquire_lock(import_lock, 1);
312 PyEval_RestoreThread(tstate);
313 }
314 import_lock_thread = me;
315 import_lock_level = 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000316}
317
Benjamin Peterson0df35a92009-10-04 20:32:25 +0000318int
319_PyImport_ReleaseLock(void)
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000320{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000321 long me = PyThread_get_thread_ident();
322 if (me == -1 || import_lock == NULL)
323 return 0; /* Too bad */
324 if (import_lock_thread != me)
325 return -1;
326 import_lock_level--;
327 if (import_lock_level == 0) {
328 import_lock_thread = -1;
329 PyThread_release_lock(import_lock);
330 }
331 return 1;
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000332}
333
Gregory P. Smith24cec9f2010-03-01 06:18:41 +0000334/* This function is called from PyOS_AfterFork to ensure that newly
335 created child processes do not share locks with the parent.
336 We now acquire the import lock around fork() calls but on some platforms
337 (Solaris 9 and earlier? see isue7242) that still left us with problems. */
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000338
339void
340_PyImport_ReInitLock(void)
341{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000342 if (import_lock != NULL)
343 import_lock = PyThread_allocate_lock();
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000344 if (import_lock_level > 1) {
345 /* Forked as a side effect of import */
346 long me = PyThread_get_thread_ident();
347 PyThread_acquire_lock(import_lock, 0);
Victor Stinner942003c2011-03-07 16:57:48 +0100348 /* XXX: can the previous line fail? */
Nick Coghlanb2ddf792010-12-02 04:11:46 +0000349 import_lock_thread = me;
350 import_lock_level--;
351 } else {
352 import_lock_thread = -1;
353 import_lock_level = 0;
354 }
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000355}
356
Guido van Rossum75acc9c1998-03-03 22:26:50 +0000357#endif
358
Tim Peters69232342001-08-30 05:16:13 +0000359static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000360imp_lock_held(PyObject *self, PyObject *noargs)
Tim Peters69232342001-08-30 05:16:13 +0000361{
Tim Peters69232342001-08-30 05:16:13 +0000362#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000363 return PyBool_FromLong(import_lock_thread != -1);
Tim Peters69232342001-08-30 05:16:13 +0000364#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000365 return PyBool_FromLong(0);
Tim Peters69232342001-08-30 05:16:13 +0000366#endif
367}
368
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000369static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000370imp_acquire_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000371{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000372#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000373 _PyImport_AcquireLock();
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000374#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000375 Py_INCREF(Py_None);
376 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000377}
378
379static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +0000380imp_release_lock(PyObject *self, PyObject *noargs)
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000381{
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000382#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000383 if (_PyImport_ReleaseLock() < 0) {
384 PyErr_SetString(PyExc_RuntimeError,
385 "not holding the import lock");
386 return NULL;
387 }
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000388#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000389 Py_INCREF(Py_None);
390 return Py_None;
Guido van Rossumc4f4ca92003-02-12 21:46:11 +0000391}
392
Guido van Rossumd8faa362007-04-27 19:54:29 +0000393static void
394imp_modules_reloading_clear(void)
395{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000396 PyInterpreterState *interp = PyThreadState_Get()->interp;
397 if (interp->modules_reloading != NULL)
398 PyDict_Clear(interp->modules_reloading);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000399}
400
Guido van Rossum25ce5661997-08-02 03:10:38 +0000401/* Helper for sys */
402
403PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000404PyImport_GetModuleDict(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000405{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 PyInterpreterState *interp = PyThreadState_GET()->interp;
407 if (interp->modules == NULL)
408 Py_FatalError("PyImport_GetModuleDict: no module dictionary!");
409 return interp->modules;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000410}
411
Guido van Rossum3f5da241990-12-20 15:06:42 +0000412
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000413/* List of names to clear in sys */
414static char* sys_deletes[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 "path", "argv", "ps1", "ps2",
416 "last_type", "last_value", "last_traceback",
417 "path_hooks", "path_importer_cache", "meta_path",
418 /* misc stuff */
419 "flags", "float_info",
420 NULL
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000421};
422
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000423static char* sys_files[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000424 "stdin", "__stdin__",
425 "stdout", "__stdout__",
426 "stderr", "__stderr__",
427 NULL
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000428};
429
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000430
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000431/* Un-initialize things, as good as we can */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000432
Guido van Rossum3f5da241990-12-20 15:06:42 +0000433void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000434PyImport_Cleanup(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000435{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 Py_ssize_t pos, ndone;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000437 PyObject *key, *value, *dict;
438 PyInterpreterState *interp = PyThreadState_GET()->interp;
439 PyObject *modules = interp->modules;
Guido van Rossum758eec01998-01-19 21:58:26 +0000440
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000441 if (modules == NULL)
442 return; /* Already done */
Guido van Rossum758eec01998-01-19 21:58:26 +0000443
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000444 /* Delete some special variables first. These are common
445 places where user values hide and people complain when their
446 destructors fail. Since the modules containing them are
447 deleted *last* of all, they would come too late in the normal
448 destruction order. Sigh. */
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000449
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000450 value = PyDict_GetItemString(modules, "builtins");
451 if (value != NULL && PyModule_Check(value)) {
452 dict = PyModule_GetDict(value);
453 if (Py_VerboseFlag)
454 PySys_WriteStderr("# clear builtins._\n");
455 PyDict_SetItemString(dict, "_", Py_None);
456 }
457 value = PyDict_GetItemString(modules, "sys");
458 if (value != NULL && PyModule_Check(value)) {
459 char **p;
460 PyObject *v;
461 dict = PyModule_GetDict(value);
462 for (p = sys_deletes; *p != NULL; p++) {
463 if (Py_VerboseFlag)
464 PySys_WriteStderr("# clear sys.%s\n", *p);
465 PyDict_SetItemString(dict, *p, Py_None);
466 }
467 for (p = sys_files; *p != NULL; p+=2) {
468 if (Py_VerboseFlag)
469 PySys_WriteStderr("# restore sys.%s\n", *p);
470 v = PyDict_GetItemString(dict, *(p+1));
471 if (v == NULL)
472 v = Py_None;
473 PyDict_SetItemString(dict, *p, v);
474 }
475 }
Guido van Rossum05f9dce1998-02-19 20:58:44 +0000476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000477 /* First, delete __main__ */
478 value = PyDict_GetItemString(modules, "__main__");
479 if (value != NULL && PyModule_Check(value)) {
480 if (Py_VerboseFlag)
481 PySys_WriteStderr("# cleanup __main__\n");
482 _PyModule_Clear(value);
483 PyDict_SetItemString(modules, "__main__", Py_None);
484 }
Guido van Rossuma0fec2b1998-02-06 17:16:02 +0000485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000486 /* The special treatment of "builtins" here is because even
487 when it's not referenced as a module, its dictionary is
488 referenced by almost every module's __builtins__. Since
489 deleting a module clears its dictionary (even if there are
490 references left to it), we need to delete the "builtins"
491 module last. Likewise, we don't delete sys until the very
492 end because it is implicitly referenced (e.g. by print).
Guido van Rossum758eec01998-01-19 21:58:26 +0000493
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000494 Also note that we 'delete' modules by replacing their entry
495 in the modules dict with None, rather than really deleting
496 them; this avoids a rehash of the modules dictionary and
497 also marks them as "non existent" so they won't be
498 re-imported. */
Guido van Rossum758eec01998-01-19 21:58:26 +0000499
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000500 /* Next, repeatedly delete modules with a reference count of
501 one (skipping builtins and sys) and delete them */
502 do {
503 ndone = 0;
504 pos = 0;
505 while (PyDict_Next(modules, &pos, &key, &value)) {
506 if (value->ob_refcnt != 1)
507 continue;
508 if (PyUnicode_Check(key) && PyModule_Check(value)) {
Victor Stinner9464d612011-03-07 17:08:21 +0100509 if (PyUnicode_CompareWithASCIIString(key, "builtins") == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000510 continue;
Victor Stinner9464d612011-03-07 17:08:21 +0100511 if (PyUnicode_CompareWithASCIIString(key, "sys") == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000512 continue;
513 if (Py_VerboseFlag)
Victor Stinner9464d612011-03-07 17:08:21 +0100514 PySys_FormatStderr(
515 "# cleanup[1] %U\n", key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000516 _PyModule_Clear(value);
517 PyDict_SetItem(modules, key, Py_None);
518 ndone++;
519 }
520 }
521 } while (ndone > 0);
Guido van Rossum758eec01998-01-19 21:58:26 +0000522
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000523 /* Next, delete all modules (still skipping builtins and sys) */
524 pos = 0;
525 while (PyDict_Next(modules, &pos, &key, &value)) {
526 if (PyUnicode_Check(key) && PyModule_Check(value)) {
Victor Stinner9464d612011-03-07 17:08:21 +0100527 if (PyUnicode_CompareWithASCIIString(key, "builtins") == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000528 continue;
Victor Stinner9464d612011-03-07 17:08:21 +0100529 if (PyUnicode_CompareWithASCIIString(key, "sys") == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 continue;
531 if (Py_VerboseFlag)
Victor Stinner9464d612011-03-07 17:08:21 +0100532 PySys_FormatStderr("# cleanup[2] %U\n", key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000533 _PyModule_Clear(value);
534 PyDict_SetItem(modules, key, Py_None);
535 }
536 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 /* Next, delete sys and builtins (in that order) */
539 value = PyDict_GetItemString(modules, "sys");
540 if (value != NULL && PyModule_Check(value)) {
541 if (Py_VerboseFlag)
542 PySys_WriteStderr("# cleanup sys\n");
543 _PyModule_Clear(value);
544 PyDict_SetItemString(modules, "sys", Py_None);
545 }
546 value = PyDict_GetItemString(modules, "builtins");
547 if (value != NULL && PyModule_Check(value)) {
548 if (Py_VerboseFlag)
549 PySys_WriteStderr("# cleanup builtins\n");
550 _PyModule_Clear(value);
551 PyDict_SetItemString(modules, "builtins", Py_None);
552 }
Guido van Rossum758eec01998-01-19 21:58:26 +0000553
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000554 /* Finally, clear and delete the modules directory */
555 PyDict_Clear(modules);
556 interp->modules = NULL;
557 Py_DECREF(modules);
558 Py_CLEAR(interp->modules_reloading);
Guido van Rossum8d15b5d1990-10-26 14:58:58 +0000559}
Guido van Rossum7f133ed1991-02-19 12:23:57 +0000560
561
Barry Warsaw28a691b2010-04-17 00:19:56 +0000562/* Helper for pythonrun.c -- return magic number and tag. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000563
564long
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000565PyImport_GetMagicNumber(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000566{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000567 return pyc_magic;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000568}
569
570
Barry Warsaw28a691b2010-04-17 00:19:56 +0000571const char *
572PyImport_GetMagicTag(void)
573{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 return pyc_tag;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000575}
576
Guido van Rossum25ce5661997-08-02 03:10:38 +0000577/* Magic for extension modules (built-in as well as dynamically
578 loaded). To prevent initializing an extension module more than
579 once, we keep a static dictionary 'extensions' keyed by module name
580 (for built-in modules) or by filename (for dynamically loaded
Barry Warsaw92883382001-08-13 23:05:44 +0000581 modules), containing these modules. A copy of the module's
Victor Stinner95872862011-03-07 18:20:56 +0100582 dictionary is stored by calling _PyImport_FixupExtensionObject()
Guido van Rossum25ce5661997-08-02 03:10:38 +0000583 immediately after the module initialization function succeeds. A
584 copy can be retrieved from there by calling
Victor Stinner95872862011-03-07 18:20:56 +0100585 _PyImport_FindExtensionObject().
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000586
Barry Warsaw7c9627b2010-06-17 18:38:20 +0000587 Modules which do support multiple initialization set their m_size
588 field to a non-negative number (indicating the size of the
589 module-specific state). They are still recorded in the extensions
590 dictionary, to avoid loading shared libraries twice.
Martin v. Löwis1a214512008-06-11 05:26:20 +0000591*/
592
593int
Victor Stinner95872862011-03-07 18:20:56 +0100594_PyImport_FixupExtensionObject(PyObject *mod, PyObject *name,
595 PyObject *filename)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000596{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000597 PyObject *modules, *dict;
598 struct PyModuleDef *def;
599 if (extensions == NULL) {
600 extensions = PyDict_New();
601 if (extensions == NULL)
602 return -1;
603 }
604 if (mod == NULL || !PyModule_Check(mod)) {
605 PyErr_BadInternalCall();
606 return -1;
607 }
608 def = PyModule_GetDef(mod);
609 if (!def) {
610 PyErr_BadInternalCall();
611 return -1;
612 }
613 modules = PyImport_GetModuleDict();
Victor Stinner95872862011-03-07 18:20:56 +0100614 if (PyDict_SetItem(modules, name, mod) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000615 return -1;
616 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100617 PyDict_DelItem(modules, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 return -1;
619 }
620 if (def->m_size == -1) {
621 if (def->m_base.m_copy) {
622 /* Somebody already imported the module,
623 likely under a different name.
624 XXX this should really not happen. */
625 Py_DECREF(def->m_base.m_copy);
626 def->m_base.m_copy = NULL;
627 }
628 dict = PyModule_GetDict(mod);
629 if (dict == NULL)
630 return -1;
631 def->m_base.m_copy = PyDict_Copy(dict);
632 if (def->m_base.m_copy == NULL)
633 return -1;
634 }
Victor Stinner49d3f252010-10-17 01:24:53 +0000635 PyDict_SetItem(extensions, filename, (PyObject*)def);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 return 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000637}
638
Victor Stinner49d3f252010-10-17 01:24:53 +0000639int
640_PyImport_FixupBuiltin(PyObject *mod, char *name)
641{
642 int res;
Victor Stinner95872862011-03-07 18:20:56 +0100643 PyObject *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100644 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100645 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000646 return -1;
Victor Stinner95872862011-03-07 18:20:56 +0100647 res = _PyImport_FixupExtensionObject(mod, nameobj, nameobj);
648 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000649 return res;
650}
651
Guido van Rossum25ce5661997-08-02 03:10:38 +0000652PyObject *
Victor Stinner95872862011-03-07 18:20:56 +0100653_PyImport_FindExtensionObject(PyObject *name, PyObject *filename)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000654{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 PyObject *mod, *mdict;
656 PyModuleDef* def;
657 if (extensions == NULL)
658 return NULL;
Victor Stinner49d3f252010-10-17 01:24:53 +0000659 def = (PyModuleDef*)PyDict_GetItem(extensions, filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000660 if (def == NULL)
661 return NULL;
662 if (def->m_size == -1) {
663 /* Module does not support repeated initialization */
664 if (def->m_base.m_copy == NULL)
665 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100666 mod = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 if (mod == NULL)
668 return NULL;
669 mdict = PyModule_GetDict(mod);
670 if (mdict == NULL)
671 return NULL;
672 if (PyDict_Update(mdict, def->m_base.m_copy))
673 return NULL;
674 }
675 else {
676 if (def->m_base.m_init == NULL)
677 return NULL;
678 mod = def->m_base.m_init();
679 if (mod == NULL)
680 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100681 PyDict_SetItem(PyImport_GetModuleDict(), name, mod);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000682 Py_DECREF(mod);
683 }
684 if (_PyState_AddModule(mod, def) < 0) {
Victor Stinner95872862011-03-07 18:20:56 +0100685 PyDict_DelItem(PyImport_GetModuleDict(), name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 Py_DECREF(mod);
687 return NULL;
688 }
689 if (Py_VerboseFlag)
Victor Stinner95872862011-03-07 18:20:56 +0100690 PySys_FormatStderr("import %U # previously loaded (%R)\n",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000691 name, filename);
692 return mod;
Brett Cannon9a5b25a2010-03-01 02:09:17 +0000693
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000694}
695
Victor Stinner49d3f252010-10-17 01:24:53 +0000696PyObject *
Victor Stinner501c09a2011-02-23 00:02:00 +0000697_PyImport_FindBuiltin(const char *name)
Victor Stinner49d3f252010-10-17 01:24:53 +0000698{
Victor Stinner95872862011-03-07 18:20:56 +0100699 PyObject *res, *nameobj;
Victor Stinner21fcd0c2011-03-07 18:28:15 +0100700 nameobj = PyUnicode_InternFromString(name);
Victor Stinner95872862011-03-07 18:20:56 +0100701 if (nameobj == NULL)
Victor Stinner49d3f252010-10-17 01:24:53 +0000702 return NULL;
Victor Stinner95872862011-03-07 18:20:56 +0100703 res = _PyImport_FindExtensionObject(nameobj, nameobj);
704 Py_DECREF(nameobj);
Victor Stinner49d3f252010-10-17 01:24:53 +0000705 return res;
706}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000707
708/* Get the module object corresponding to a module name.
709 First check the modules dictionary if there's one there,
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000710 if not, create a new one and insert it in the modules dictionary.
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000711 Because the former action is most common, THIS DOES NOT RETURN A
712 'NEW' REFERENCE! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000713
Guido van Rossum79f25d91997-04-29 20:08:16 +0000714PyObject *
Victor Stinner27ee0892011-03-04 12:57:09 +0000715PyImport_AddModuleObject(PyObject *name)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000716{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000717 PyObject *modules = PyImport_GetModuleDict();
718 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000719
Victor Stinner27ee0892011-03-04 12:57:09 +0000720 if ((m = PyDict_GetItem(modules, name)) != NULL &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000721 PyModule_Check(m))
722 return m;
Victor Stinner27ee0892011-03-04 12:57:09 +0000723 m = PyModule_NewObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000724 if (m == NULL)
725 return NULL;
Victor Stinner27ee0892011-03-04 12:57:09 +0000726 if (PyDict_SetItem(modules, name, m) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000727 Py_DECREF(m);
728 return NULL;
729 }
730 Py_DECREF(m); /* Yes, it still exists, in modules! */
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000731
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000732 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000733}
734
Victor Stinner27ee0892011-03-04 12:57:09 +0000735PyObject *
736PyImport_AddModule(const char *name)
737{
738 PyObject *nameobj, *module;
739 nameobj = PyUnicode_FromString(name);
740 if (nameobj == NULL)
741 return NULL;
742 module = PyImport_AddModuleObject(nameobj);
743 Py_DECREF(nameobj);
744 return module;
745}
746
747
Tim Peters1cd70172004-08-02 03:52:12 +0000748/* Remove name from sys.modules, if it's there. */
749static void
Victor Stinner27ee0892011-03-04 12:57:09 +0000750remove_module(PyObject *name)
Tim Peters1cd70172004-08-02 03:52:12 +0000751{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000752 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner27ee0892011-03-04 12:57:09 +0000753 if (PyDict_GetItem(modules, name) == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000754 return;
Victor Stinner27ee0892011-03-04 12:57:09 +0000755 if (PyDict_DelItem(modules, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000756 Py_FatalError("import: deleting existing key in"
757 "sys.modules failed");
Tim Peters1cd70172004-08-02 03:52:12 +0000758}
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000759
Victor Stinnerc9abda02011-03-14 13:33:46 -0400760static PyObject * get_sourcefile(PyObject *filename);
761static PyObject *make_source_pathname(PyObject *pathname);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200762static PyObject* make_compiled_pathname(PyObject *pathname, int debug);
Christian Heimes3b06e532008-01-07 20:12:44 +0000763
Guido van Rossum7f9fa971995-01-20 16:53:12 +0000764/* Execute a code object in a module and return the module object
Tim Peters1cd70172004-08-02 03:52:12 +0000765 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is
766 * removed from sys.modules, to avoid leaving damaged module objects
767 * in sys.modules. The caller may wish to restore the original
768 * module object (if any) in this case; PyImport_ReloadModule is an
769 * example.
Barry Warsaw28a691b2010-04-17 00:19:56 +0000770 *
771 * Note that PyImport_ExecCodeModuleWithPathnames() is the preferred, richer
772 * interface. The other two exist primarily for backward compatibility.
Tim Peters1cd70172004-08-02 03:52:12 +0000773 */
Guido van Rossum79f25d91997-04-29 20:08:16 +0000774PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000775PyImport_ExecCodeModule(char *name, PyObject *co)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000776{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000777 return PyImport_ExecCodeModuleWithPathnames(
778 name, co, (char *)NULL, (char *)NULL);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000779}
780
781PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000782PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname)
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000783{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000784 return PyImport_ExecCodeModuleWithPathnames(
785 name, co, pathname, (char *)NULL);
Barry Warsaw28a691b2010-04-17 00:19:56 +0000786}
787
788PyObject *
789PyImport_ExecCodeModuleWithPathnames(char *name, PyObject *co, char *pathname,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000790 char *cpathname)
Barry Warsaw28a691b2010-04-17 00:19:56 +0000791{
Victor Stinner27ee0892011-03-04 12:57:09 +0000792 PyObject *m = NULL;
793 PyObject *nameobj, *pathobj = NULL, *cpathobj = NULL;
794
795 nameobj = PyUnicode_FromString(name);
796 if (nameobj == NULL)
797 return NULL;
798
799 if (pathname != NULL) {
800 pathobj = PyUnicode_DecodeFSDefault(pathname);
801 if (pathobj == NULL)
802 goto error;
803 } else
804 pathobj = NULL;
805 if (cpathname != NULL) {
806 cpathobj = PyUnicode_DecodeFSDefault(cpathname);
807 if (cpathobj == NULL)
808 goto error;
809 } else
810 cpathobj = NULL;
811 m = PyImport_ExecCodeModuleObject(nameobj, co, pathobj, cpathobj);
812error:
813 Py_DECREF(nameobj);
814 Py_XDECREF(pathobj);
815 Py_XDECREF(cpathobj);
816 return m;
817}
818
819PyObject*
820PyImport_ExecCodeModuleObject(PyObject *name, PyObject *co, PyObject *pathname,
821 PyObject *cpathname)
822{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000823 PyObject *modules = PyImport_GetModuleDict();
824 PyObject *m, *d, *v;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000825
Victor Stinner27ee0892011-03-04 12:57:09 +0000826 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 if (m == NULL)
828 return NULL;
829 /* If the module is being reloaded, we get the old module back
830 and re-use its dict to exec the new code. */
831 d = PyModule_GetDict(m);
832 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
833 if (PyDict_SetItemString(d, "__builtins__",
834 PyEval_GetBuiltins()) != 0)
835 goto error;
836 }
837 /* Remember the filename as the __file__ attribute */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000838 if (pathname != NULL) {
Victor Stinnerc9abda02011-03-14 13:33:46 -0400839 v = get_sourcefile(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000840 if (v == NULL)
841 PyErr_Clear();
842 }
Victor Stinner27ee0892011-03-04 12:57:09 +0000843 else
844 v = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000845 if (v == NULL) {
846 v = ((PyCodeObject *)co)->co_filename;
847 Py_INCREF(v);
848 }
849 if (PyDict_SetItemString(d, "__file__", v) != 0)
850 PyErr_Clear(); /* Not important enough to report */
851 Py_DECREF(v);
Guido van Rossume32bf6e1998-02-11 05:53:02 +0000852
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000853 /* Remember the pyc path name as the __cached__ attribute. */
Victor Stinner27ee0892011-03-04 12:57:09 +0000854 if (cpathname != NULL)
855 v = cpathname;
856 else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 v = Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 if (PyDict_SetItemString(d, "__cached__", v) != 0)
859 PyErr_Clear(); /* Not important enough to report */
Barry Warsaw28a691b2010-04-17 00:19:56 +0000860
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000861 v = PyEval_EvalCode(co, d, d);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000862 if (v == NULL)
863 goto error;
864 Py_DECREF(v);
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000865
Victor Stinner27ee0892011-03-04 12:57:09 +0000866 if ((m = PyDict_GetItem(modules, name)) == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000867 PyErr_Format(PyExc_ImportError,
Victor Stinner27ee0892011-03-04 12:57:09 +0000868 "Loaded module %R not found in sys.modules",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 name);
870 return NULL;
871 }
Guido van Rossumb65e85c1997-07-10 18:00:45 +0000872
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000873 Py_INCREF(m);
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000874
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000875 return m;
Tim Peters1cd70172004-08-02 03:52:12 +0000876
877 error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000878 remove_module(name);
879 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000880}
881
882
Barry Warsaw28a691b2010-04-17 00:19:56 +0000883/* Like strrchr(string, '/') but searches for the rightmost of either SEP
884 or ALTSEP, if the latter is defined.
885*/
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200886static Py_UCS4*
887rightmost_sep(Py_UCS4 *s)
Victor Stinnerc9abda02011-03-14 13:33:46 -0400888{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200889 Py_UCS4 *found, c;
Victor Stinnerc9abda02011-03-14 13:33:46 -0400890 for (found = NULL; (c = *s); s++) {
891 if (c == SEP
892#ifdef ALTSEP
893 || c == ALTSEP
894#endif
895 )
896 {
897 found = s;
898 }
899 }
900 return found;
901}
902
Martin v. Löwis2db72862011-10-23 17:29:08 +0200903/* Like rightmost_sep, but operate on unicode objects. */
904static Py_ssize_t
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200905rightmost_sep_obj(PyObject* o, Py_ssize_t start, Py_ssize_t end)
Martin v. Löwis2db72862011-10-23 17:29:08 +0200906{
907 Py_ssize_t found, i;
908 Py_UCS4 c;
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200909 for (found = -1, i = start; i < end; i++) {
Martin v. Löwis2db72862011-10-23 17:29:08 +0200910 c = PyUnicode_READ_CHAR(o, i);
911 if (c == SEP
912#ifdef ALTSEP
913 || c == ALTSEP
914#endif
915 )
916 {
917 found = i;
918 }
919 }
920 return found;
921}
Victor Stinnerc9abda02011-03-14 13:33:46 -0400922
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000923/* Given a pathname for a Python source file, fill a buffer with the
924 pathname for the corresponding compiled file. Return the pathname
925 for the compiled file, or NULL if there's no space in the buffer.
Victor Stinner2f42ae52011-03-20 00:41:24 +0100926 Doesn't set an exception.
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000927
Martin v. Löwis30260a72011-10-23 17:35:46 +0200928 foo.py -> __pycache__/foo.<tag>.pyc
929
930 pathstr is assumed to be "ready".
931*/
Victor Stinner2f42ae52011-03-20 00:41:24 +0100932
933static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200934make_compiled_pathname(PyObject *pathstr, int debug)
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000935{
Martin v. Löwis2db72862011-10-23 17:29:08 +0200936 PyObject *result;
937 Py_ssize_t fname, ext, len, i, pos, taglen;
Martin v. Löwisfadcd312011-10-23 18:08:20 +0200938 Py_ssize_t pycache_len = sizeof(CACHEDIR) - 1;
Martin v. Löwis2db72862011-10-23 17:29:08 +0200939 int kind;
940 void *data;
Barry Warsaw28a691b2010-04-17 00:19:56 +0000941
Martin v. Löwis2db72862011-10-23 17:29:08 +0200942 /* Compute the output string size. */
943 len = PyUnicode_GET_LENGTH(pathstr);
944 /* If there is no separator, this returns -1, so
945 lastsep will be 0. */
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200946 fname = rightmost_sep_obj(pathstr, 0, len) + 1;
Martin v. Löwis2db72862011-10-23 17:29:08 +0200947 ext = fname - 1;
948 for(i = fname; i < len; i++)
949 if (PyUnicode_READ_CHAR(pathstr, i) == '.')
950 ext = i + 1;
951 if (ext < fname)
952 /* No dot in filename; use entire filename */
953 ext = len;
954
955 /* result = pathstr[:fname] + "__pycache__" + SEP +
956 pathstr[fname:ext] + tag + ".py[co]" */
957 taglen = strlen(pyc_tag);
958 result = PyUnicode_New(ext + pycache_len + taglen + 4,
959 PyUnicode_MAX_CHAR_VALUE(pathstr));
960 if (!result)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200961 return NULL;
Martin v. Löwis2db72862011-10-23 17:29:08 +0200962 kind = PyUnicode_KIND(result);
963 data = PyUnicode_DATA(result);
964 PyUnicode_CopyCharacters(result, 0, pathstr, 0, fname);
965 pos = fname;
966 for (i = 0; i < pycache_len - 1; i++)
Martin v. Löwisfadcd312011-10-23 18:08:20 +0200967 PyUnicode_WRITE(kind, data, pos++, CACHEDIR[i]);
Martin v. Löwis2db72862011-10-23 17:29:08 +0200968 PyUnicode_WRITE(kind, data, pos++, SEP);
969 PyUnicode_CopyCharacters(result, pos, pathstr,
970 fname, ext - fname);
971 pos += ext - fname;
972 for (i = 0; pyc_tag[i]; i++)
973 PyUnicode_WRITE(kind, data, pos++, pyc_tag[i]);
974 PyUnicode_WRITE(kind, data, pos++, '.');
975 PyUnicode_WRITE(kind, data, pos++, 'p');
976 PyUnicode_WRITE(kind, data, pos++, 'y');
977 PyUnicode_WRITE(kind, data, pos++, debug ? 'c' : 'o');
978 return result;
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000979}
980
981
Barry Warsaw28a691b2010-04-17 00:19:56 +0000982/* Given a pathname to a Python byte compiled file, return the path to the
983 source file, if the path matches the PEP 3147 format. This does not check
984 for any file existence, however, if the pyc file name does not match PEP
985 3147 style, NULL is returned. buf must be at least as big as pathname;
Victor Stinnerc9abda02011-03-14 13:33:46 -0400986 the resulting path will always be shorter.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000987
Victor Stinnerc9abda02011-03-14 13:33:46 -0400988 (...)/__pycache__/foo.<tag>.pyc -> (...)/foo.py */
989
990static PyObject*
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200991make_source_pathname(PyObject *path)
Barry Warsaw28a691b2010-04-17 00:19:56 +0000992{
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200993 Py_ssize_t left, right, dot0, dot1, len;
994 Py_ssize_t i, j;
995 PyObject *result;
996 int kind;
997 void *data;
Victor Stinnerc9abda02011-03-14 13:33:46 -0400998
Martin v. Löwis8a0ef782011-10-23 18:05:06 +0200999 len = PyUnicode_GET_LENGTH(path);
1000 if (len > MAXPATHLEN)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001001 return NULL;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001002
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 /* Look back two slashes from the end. In between these two slashes
1004 must be the string __pycache__ or this is not a PEP 3147 style
1005 path. It's possible for there to be only one slash.
1006 */
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001007 right = rightmost_sep_obj(path, 0, len);
1008 if (right == -1)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001009 return NULL;
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001010 left = rightmost_sep_obj(path, 0, right);
1011 if (left == -1)
1012 left = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001013 else
1014 left++;
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001015 if (right-left != sizeof(CACHEDIR)-1)
1016 return NULL;
1017 for (i = 0; i < sizeof(CACHEDIR)-1; i++)
1018 if (PyUnicode_READ_CHAR(path, left+i) != CACHEDIR[i])
1019 return NULL;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 /* Now verify that the path component to the right of the last slash
1022 has two dots in it.
1023 */
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001024 dot0 = PyUnicode_FindChar(path, '.', right+1, len, 1);
1025 if (dot0 < 0)
1026 return NULL;
1027 dot1 = PyUnicode_FindChar(path, '.', dot0+1, len, 1);
1028 if (dot1 < 0)
1029 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001030 /* Too many dots? */
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001031 if (PyUnicode_FindChar(path, '.', dot1+1, len, 1) != -1)
1032 return NULL;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001034 /* This is a PEP 3147 path. Start by copying everything from the
1035 start of pathname up to and including the leftmost slash. Then
1036 copy the file's basename, removing the magic tag and adding a .py
1037 suffix.
1038 */
Martin v. Löwis8a0ef782011-10-23 18:05:06 +02001039 result = PyUnicode_New(left + (dot0-right) + 2,
1040 PyUnicode_MAX_CHAR_VALUE(path));
1041 if (!result)
1042 return NULL;
1043 kind = PyUnicode_KIND(result);
1044 data = PyUnicode_DATA(result);
1045 PyUnicode_CopyCharacters(result, 0, path, 0, (i = left));
1046 PyUnicode_CopyCharacters(result, left, path, right+1,
1047 (j = dot0-right));
1048 PyUnicode_WRITE(kind, data, i+j, 'p');
1049 PyUnicode_WRITE(kind, data, i+j+1, 'y');
1050 return result;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001051}
1052
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001053/* Given a pathname for a Python source file, its time of last
1054 modification, and a pathname for a compiled file, check whether the
1055 compiled file represents the same version of the source. If so,
1056 return a FILE pointer for the compiled file, positioned just after
1057 the header; if not, return NULL.
1058 Doesn't set an exception. */
1059
1060static FILE *
Victor Stinner2f42ae52011-03-20 00:41:24 +01001061check_compiled_module(PyObject *pathname, time_t mtime, PyObject *cpathname)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001062{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 FILE *fp;
1064 long magic;
1065 long pyc_mtime;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001066
Victor Stinner2f42ae52011-03-20 00:41:24 +01001067 fp = _Py_fopen(cpathname, "rb");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001068 if (fp == NULL)
1069 return NULL;
1070 magic = PyMarshal_ReadLongFromFile(fp);
1071 if (magic != pyc_magic) {
1072 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001073 PySys_FormatStderr("# %R has bad magic\n", cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001074 fclose(fp);
1075 return NULL;
1076 }
1077 pyc_mtime = PyMarshal_ReadLongFromFile(fp);
1078 if (pyc_mtime != mtime) {
1079 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001080 PySys_FormatStderr("# %R has bad mtime\n", cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001081 fclose(fp);
1082 return NULL;
1083 }
1084 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001085 PySys_FormatStderr("# %R matches %R\n", cpathname, pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001086 return fp;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001087}
1088
1089
1090/* Read a code object from a file and check it for validity */
1091
Guido van Rossum79f25d91997-04-29 20:08:16 +00001092static PyCodeObject *
Victor Stinner2f42ae52011-03-20 00:41:24 +01001093read_compiled_module(PyObject *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001094{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001095 PyObject *co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001096
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001097 co = PyMarshal_ReadLastObjectFromFile(fp);
1098 if (co == NULL)
1099 return NULL;
1100 if (!PyCode_Check(co)) {
1101 PyErr_Format(PyExc_ImportError,
Victor Stinner2f42ae52011-03-20 00:41:24 +01001102 "Non-code object in %R", cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001103 Py_DECREF(co);
1104 return NULL;
1105 }
1106 return (PyCodeObject *)co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001107}
1108
1109
1110/* Load a module from a compiled file, execute it, and return its
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001111 module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001112
Guido van Rossum79f25d91997-04-29 20:08:16 +00001113static PyObject *
Victor Stinner2f42ae52011-03-20 00:41:24 +01001114load_compiled_module(PyObject *name, PyObject *cpathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001115{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001116 long magic;
1117 PyCodeObject *co;
1118 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001120 magic = PyMarshal_ReadLongFromFile(fp);
1121 if (magic != pyc_magic) {
1122 PyErr_Format(PyExc_ImportError,
Victor Stinner2f42ae52011-03-20 00:41:24 +01001123 "Bad magic number in %R", cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001124 return NULL;
1125 }
1126 (void) PyMarshal_ReadLongFromFile(fp);
1127 co = read_compiled_module(cpathname, fp);
1128 if (co == NULL)
1129 return NULL;
1130 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001131 PySys_FormatStderr("import %U # precompiled from %R\n",
1132 name, cpathname);
1133 m = PyImport_ExecCodeModuleObject(name, (PyObject *)co,
1134 cpathname, cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001135 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001136
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001137 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001138}
1139
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001140/* Parse a source file and return the corresponding code object */
1141
Guido van Rossum79f25d91997-04-29 20:08:16 +00001142static PyCodeObject *
Victor Stinner2f42ae52011-03-20 00:41:24 +01001143parse_source_module(PyObject *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001144{
Victor Stinner2f42ae52011-03-20 00:41:24 +01001145 PyCodeObject *co;
1146 PyObject *pathbytes;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001147 mod_ty mod;
1148 PyCompilerFlags flags;
Victor Stinner2f42ae52011-03-20 00:41:24 +01001149 PyArena *arena;
1150
1151 pathbytes = PyUnicode_EncodeFSDefault(pathname);
1152 if (pathbytes == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001153 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001154
Victor Stinner2f42ae52011-03-20 00:41:24 +01001155 arena = PyArena_New();
1156 if (arena == NULL) {
1157 Py_DECREF(pathbytes);
1158 return NULL;
1159 }
1160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001161 flags.cf_flags = 0;
Victor Stinner2f42ae52011-03-20 00:41:24 +01001162 mod = PyParser_ASTFromFile(fp, PyBytes_AS_STRING(pathbytes), NULL,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001163 Py_file_input, 0, 0, &flags,
1164 NULL, arena);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001165 if (mod != NULL)
1166 co = PyAST_Compile(mod, PyBytes_AS_STRING(pathbytes), NULL, arena);
1167 else
1168 co = NULL;
1169 Py_DECREF(pathbytes);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001170 PyArena_Free(arena);
1171 return co;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001172}
1173
Guido van Rossum55a83382000-09-20 20:31:38 +00001174/* Helper to open a bytecode file for writing in exclusive mode */
1175
Victor Stinner783c82c2011-04-20 03:27:51 +02001176#ifndef MS_WINDOWS
Guido van Rossum55a83382000-09-20 20:31:38 +00001177static FILE *
Christian Heimes05e8be12008-02-23 18:30:17 +00001178open_exclusive(char *filename, mode_t mode)
Guido van Rossum55a83382000-09-20 20:31:38 +00001179{
1180#if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001181 /* Use O_EXCL to avoid a race condition when another process tries to
1182 write the same file. When that happens, our open() call fails,
1183 which is just fine (since it's only a cache).
1184 XXX If the file exists and is writable but the directory is not
1185 writable, the file will never be written. Oh well.
1186 */
1187 int fd;
1188 (void) unlink(filename);
1189 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC
Tim Peters42c83af2000-09-29 04:03:10 +00001190#ifdef O_BINARY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 |O_BINARY /* necessary for Windows */
Tim Peters42c83af2000-09-29 04:03:10 +00001192#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001193#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001194 , mode, "ctxt=bin", "shr=nil"
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001195#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 , mode
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001197#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001198 );
1199 if (fd < 0)
1200 return NULL;
1201 return fdopen(fd, "wb");
Guido van Rossum55a83382000-09-20 20:31:38 +00001202#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001203 /* Best we can do -- on Windows this can't happen anyway */
1204 return fopen(filename, "wb");
Guido van Rossum55a83382000-09-20 20:31:38 +00001205#endif
1206}
Victor Stinner783c82c2011-04-20 03:27:51 +02001207#endif
Guido van Rossum55a83382000-09-20 20:31:38 +00001208
1209
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001210/* Write a compiled module to a file, placing the time of last
1211 modification of its source into the header.
1212 Errors are ignored, if a write error occurs an attempt is made to
1213 remove the file. */
1214
1215static void
Victor Stinner2f42ae52011-03-20 00:41:24 +01001216write_compiled_module(PyCodeObject *co, PyObject *cpathname,
1217 struct stat *srcstat)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001218{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001219 Py_UCS4 *cpathname_ucs4;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 FILE *fp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001221 time_t mtime = srcstat->st_mtime;
Alexandre Vassalotti9d58e3e2009-07-17 10:55:50 +00001222#ifdef MS_WINDOWS /* since Windows uses different permissions */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001223 mode_t mode = srcstat->st_mode & ~S_IEXEC;
Alexandre Vassalotti9d58e3e2009-07-17 10:55:50 +00001224#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001225 mode_t mode = srcstat->st_mode & ~S_IXUSR & ~S_IXGRP & ~S_IXOTH;
1226 mode_t dirmode = (srcstat->st_mode |
1227 S_IXUSR | S_IXGRP | S_IXOTH |
1228 S_IWUSR | S_IWGRP | S_IWOTH);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001229 PyObject *dirbytes;
Brett Cannon9a5b25a2010-03-01 02:09:17 +00001230#endif
Victor Stinner783c82c2011-04-20 03:27:51 +02001231#ifdef MS_WINDOWS
1232 int fd;
1233#else
Antoine Pitrou707033a2011-10-17 19:28:44 +02001234 PyObject *cpathbytes, *cpathbytes_tmp;
1235 Py_ssize_t cpathbytes_len;
Victor Stinner783c82c2011-04-20 03:27:51 +02001236#endif
1237 PyObject *dirname;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001238 Py_UCS4 *dirsep;
Victor Stinner2f42ae52011-03-20 00:41:24 +01001239 int res, ok;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001241 /* Ensure that the __pycache__ directory exists. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001242 cpathname_ucs4 = PyUnicode_AsUCS4Copy(cpathname);
1243 if (!cpathname_ucs4)
1244 return;
1245 dirsep = rightmost_sep(cpathname_ucs4);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001246 if (dirsep == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001247 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001248 PySys_FormatStderr("# no %s path found %R\n", CACHEDIR, cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 return;
1250 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001251 dirname = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
1252 cpathname_ucs4,
1253 dirsep - cpathname_ucs4);
1254 PyMem_Free(cpathname_ucs4);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001255 if (dirname == NULL) {
1256 PyErr_Clear();
1257 return;
1258 }
Daniel Stutzbachc7937792010-09-09 21:18:04 +00001259
1260#ifdef MS_WINDOWS
Victor Stinner2f42ae52011-03-20 00:41:24 +01001261 res = CreateDirectoryW(PyUnicode_AS_UNICODE(dirname), NULL);
1262 ok = (res != 0);
1263 if (!ok && GetLastError() == ERROR_ALREADY_EXISTS)
1264 ok = 1;
Daniel Stutzbachc7937792010-09-09 21:18:04 +00001265#else
Victor Stinner2f42ae52011-03-20 00:41:24 +01001266 dirbytes = PyUnicode_EncodeFSDefault(dirname);
1267 if (dirbytes == NULL) {
1268 PyErr_Clear();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001269 return;
1270 }
Victor Stinner2f42ae52011-03-20 00:41:24 +01001271 res = mkdir(PyBytes_AS_STRING(dirbytes), dirmode);
1272 Py_DECREF(dirbytes);
1273 if (0 <= res)
1274 ok = 1;
1275 else
1276 ok = (errno == EEXIST);
1277#endif
1278 if (!ok) {
1279 if (Py_VerboseFlag)
1280 PySys_FormatStderr("# cannot create cache dir %R\n", dirname);
1281 Py_DECREF(dirname);
1282 return;
1283 }
1284 Py_DECREF(dirname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001285
Victor Stinner783c82c2011-04-20 03:27:51 +02001286#ifdef MS_WINDOWS
1287 (void)DeleteFileW(PyUnicode_AS_UNICODE(cpathname));
1288 fd = _wopen(PyUnicode_AS_UNICODE(cpathname),
1289 O_EXCL | O_CREAT | O_WRONLY | O_TRUNC | O_BINARY,
1290 mode);
1291 if (0 <= fd)
1292 fp = fdopen(fd, "wb");
1293 else
1294 fp = NULL;
1295#else
Antoine Pitrou707033a2011-10-17 19:28:44 +02001296 /* Under POSIX, we first write to a tmp file and then take advantage
1297 of atomic renaming. */
Victor Stinner2f42ae52011-03-20 00:41:24 +01001298 cpathbytes = PyUnicode_EncodeFSDefault(cpathname);
1299 if (cpathbytes == NULL) {
1300 PyErr_Clear();
1301 return;
1302 }
Antoine Pitrou707033a2011-10-17 19:28:44 +02001303 cpathbytes_len = PyBytes_GET_SIZE(cpathbytes);
1304 cpathbytes_tmp = PyBytes_FromStringAndSize(NULL, cpathbytes_len + 4);
1305 if (cpathbytes_tmp == NULL) {
1306 Py_DECREF(cpathbytes);
1307 PyErr_Clear();
1308 return;
1309 }
1310 memcpy(PyBytes_AS_STRING(cpathbytes_tmp), PyBytes_AS_STRING(cpathbytes),
1311 cpathbytes_len);
1312 memcpy(PyBytes_AS_STRING(cpathbytes_tmp) + cpathbytes_len, ".tmp", 4);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001313
Antoine Pitrou707033a2011-10-17 19:28:44 +02001314 fp = open_exclusive(PyBytes_AS_STRING(cpathbytes_tmp), mode);
Victor Stinner783c82c2011-04-20 03:27:51 +02001315#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001316 if (fp == NULL) {
1317 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001318 PySys_FormatStderr(
1319 "# can't create %R\n", cpathname);
Victor Stinner783c82c2011-04-20 03:27:51 +02001320#ifndef MS_WINDOWS
Victor Stinner2f42ae52011-03-20 00:41:24 +01001321 Py_DECREF(cpathbytes);
Antoine Pitrou707033a2011-10-17 19:28:44 +02001322 Py_DECREF(cpathbytes_tmp);
Victor Stinner783c82c2011-04-20 03:27:51 +02001323#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001324 return;
1325 }
1326 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION);
1327 /* First write a 0 for mtime */
1328 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION);
1329 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION);
Antoine Pitrou707033a2011-10-17 19:28:44 +02001330 fflush(fp);
1331 /* Now write the true mtime */
1332 fseek(fp, 4L, 0);
1333 assert(mtime < LONG_MAX);
1334 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001335 if (fflush(fp) != 0 || ferror(fp)) {
1336 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001337 PySys_FormatStderr("# can't write %R\n", cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001338 /* Don't keep partial file */
1339 fclose(fp);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001340#ifdef MS_WINDOWS
1341 (void)DeleteFileW(PyUnicode_AS_UNICODE(cpathname));
1342#else
Antoine Pitrou707033a2011-10-17 19:28:44 +02001343 (void) unlink(PyBytes_AS_STRING(cpathbytes_tmp));
Victor Stinner2f42ae52011-03-20 00:41:24 +01001344 Py_DECREF(cpathbytes);
Antoine Pitrou707033a2011-10-17 19:28:44 +02001345 Py_DECREF(cpathbytes_tmp);
Victor Stinner783c82c2011-04-20 03:27:51 +02001346#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001347 return;
1348 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001349 fclose(fp);
Antoine Pitrou707033a2011-10-17 19:28:44 +02001350 /* Under POSIX, do an atomic rename */
1351#ifndef MS_WINDOWS
1352 if (rename(PyBytes_AS_STRING(cpathbytes_tmp),
1353 PyBytes_AS_STRING(cpathbytes))) {
1354 if (Py_VerboseFlag)
1355 PySys_FormatStderr("# can't write %R\n", cpathname);
1356 /* Don't keep tmp file */
1357 unlink(PyBytes_AS_STRING(cpathbytes_tmp));
1358 Py_DECREF(cpathbytes);
1359 Py_DECREF(cpathbytes_tmp);
1360 return;
1361 }
1362 Py_DECREF(cpathbytes);
1363 Py_DECREF(cpathbytes_tmp);
1364#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001365 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001366 PySys_FormatStderr("# wrote %R\n", cpathname);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001367}
1368
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001369static void
1370update_code_filenames(PyCodeObject *co, PyObject *oldname, PyObject *newname)
1371{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 PyObject *constants, *tmp;
1373 Py_ssize_t i, n;
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001374
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001375 if (PyUnicode_Compare(co->co_filename, oldname))
1376 return;
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 tmp = co->co_filename;
1379 co->co_filename = newname;
1380 Py_INCREF(co->co_filename);
1381 Py_DECREF(tmp);
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001382
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 constants = co->co_consts;
1384 n = PyTuple_GET_SIZE(constants);
1385 for (i = 0; i < n; i++) {
1386 tmp = PyTuple_GET_ITEM(constants, i);
1387 if (PyCode_Check(tmp))
1388 update_code_filenames((PyCodeObject *)tmp,
1389 oldname, newname);
1390 }
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001391}
1392
Victor Stinner2f42ae52011-03-20 00:41:24 +01001393static void
1394update_compiled_module(PyCodeObject *co, PyObject *newname)
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001395{
Victor Stinner2f42ae52011-03-20 00:41:24 +01001396 PyObject *oldname;
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001397
Victor Stinner2f42ae52011-03-20 00:41:24 +01001398 if (PyUnicode_Compare(co->co_filename, newname) == 0)
1399 return;
Hirokazu Yamamoto4f447fb2009-03-04 01:52:10 +00001400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001401 oldname = co->co_filename;
1402 Py_INCREF(oldname);
1403 update_code_filenames(co, oldname, newname);
1404 Py_DECREF(oldname);
Antoine Pitroud35cbf62009-01-06 19:02:24 +00001405}
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001406
Brett Cannon442c9b92011-03-23 16:14:42 -07001407static PyObject *
1408imp_fix_co_filename(PyObject *self, PyObject *args)
1409{
1410 PyObject *co;
1411 PyObject *file_path;
1412
1413 if (!PyArg_ParseTuple(args, "OO:_fix_co_filename", &co, &file_path))
1414 return NULL;
1415
1416 if (!PyCode_Check(co)) {
1417 PyErr_SetString(PyExc_TypeError,
1418 "first argument must be a code object");
1419 return NULL;
1420 }
1421
1422 if (!PyUnicode_Check(file_path)) {
1423 PyErr_SetString(PyExc_TypeError,
1424 "second argument must be a string");
1425 return NULL;
1426 }
1427
1428 update_compiled_module((PyCodeObject*)co, file_path);
1429
1430 Py_RETURN_NONE;
1431}
1432
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001433/* Load a source module from a given file and return its module
Guido van Rossum7f9fa971995-01-20 16:53:12 +00001434 object WITH INCREMENTED REFERENCE COUNT. If there's a matching
1435 byte-compiled file, use that instead. */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001436
Guido van Rossum79f25d91997-04-29 20:08:16 +00001437static PyObject *
Victor Stinner2f42ae52011-03-20 00:41:24 +01001438load_source_module(PyObject *name, PyObject *pathname, FILE *fp)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001439{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001440 struct stat st;
1441 FILE *fpc;
Victor Stinner2f42ae52011-03-20 00:41:24 +01001442 PyObject *cpathname = NULL, *cpathbytes = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001443 PyCodeObject *co;
Victor Stinner2f42ae52011-03-20 00:41:24 +01001444 PyObject *m = NULL;
Brett Cannon9a5b25a2010-03-01 02:09:17 +00001445
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 if (fstat(fileno(fp), &st) != 0) {
1447 PyErr_Format(PyExc_RuntimeError,
Victor Stinner2f42ae52011-03-20 00:41:24 +01001448 "unable to get file status from %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001449 pathname);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001450 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001451 }
Fred Drake4c82b232000-06-30 16:18:57 +00001452#if SIZEOF_TIME_T > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001453 /* Python's .pyc timestamp handling presumes that the timestamp fits
1454 in 4 bytes. This will be fine until sometime in the year 2038,
1455 when a 4-byte signed time_t will overflow.
1456 */
1457 if (st.st_mtime >> 32) {
1458 PyErr_SetString(PyExc_OverflowError,
1459 "modification time overflows a 4 byte field");
Victor Stinner2f42ae52011-03-20 00:41:24 +01001460 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001461 }
Fred Drake4c82b232000-06-30 16:18:57 +00001462#endif
Martin v. Löwis30260a72011-10-23 17:35:46 +02001463 if (PyUnicode_READY(pathname) < 0)
1464 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001465 cpathname = make_compiled_pathname(pathname, !Py_OptimizeFlag);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001466
1467 if (cpathname != NULL)
1468 fpc = check_compiled_module(pathname, st.st_mtime, cpathname);
1469 else
1470 fpc = NULL;
1471
1472 if (fpc) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001473 co = read_compiled_module(cpathname, fpc);
1474 fclose(fpc);
1475 if (co == NULL)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001476 goto error;
1477 update_compiled_module(co, pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001478 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001479 PySys_FormatStderr("import %U # precompiled from %R\n",
1480 name, cpathname);
1481 m = PyImport_ExecCodeModuleObject(name, (PyObject *)co,
1482 cpathname, cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001483 }
1484 else {
1485 co = parse_source_module(pathname, fp);
1486 if (co == NULL)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001487 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001488 if (Py_VerboseFlag)
Victor Stinner2f42ae52011-03-20 00:41:24 +01001489 PySys_FormatStderr("import %U # from %R\n",
Victor Stinner98dbba52011-03-14 15:15:47 -04001490 name, pathname);
Victor Stinner2f42ae52011-03-20 00:41:24 +01001491 if (cpathname != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001492 PyObject *ro = PySys_GetObject("dont_write_bytecode");
1493 if (ro == NULL || !PyObject_IsTrue(ro))
1494 write_compiled_module(co, cpathname, &st);
1495 }
Victor Stinner2f42ae52011-03-20 00:41:24 +01001496 m = PyImport_ExecCodeModuleObject(name, (PyObject *)co,
1497 pathname, cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001498 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001499 Py_DECREF(co);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001500
Victor Stinner2f42ae52011-03-20 00:41:24 +01001501error:
1502 Py_XDECREF(cpathbytes);
1503 Py_XDECREF(cpathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001504 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001505}
1506
Christian Heimes3b06e532008-01-07 20:12:44 +00001507/* Get source file -> unicode or None
1508 * Returns the path to the py file if available, else the given path
1509 */
1510static PyObject *
Victor Stinnerc9abda02011-03-14 13:33:46 -04001511get_sourcefile(PyObject *filename)
Christian Heimes3b06e532008-01-07 20:12:44 +00001512{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001513 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001514 Py_UCS4 *fileuni;
Victor Stinnerc9abda02011-03-14 13:33:46 -04001515 PyObject *py;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001516 struct stat statbuf;
Christian Heimes3b06e532008-01-07 20:12:44 +00001517
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001518 len = PyUnicode_GET_LENGTH(filename);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001519 if (len == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001520 Py_RETURN_NONE;
Christian Heimes3b06e532008-01-07 20:12:44 +00001521
Victor Stinnerc9abda02011-03-14 13:33:46 -04001522 /* don't match *.pyc or *.pyo? */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001523 fileuni = PyUnicode_AsUCS4Copy(filename);
1524 if (!fileuni)
1525 return NULL;
Victor Stinnerc9abda02011-03-14 13:33:46 -04001526 if (len < 5
1527 || fileuni[len-4] != '.'
1528 || (fileuni[len-3] != 'p' && fileuni[len-3] != 'P')
1529 || (fileuni[len-2] != 'y' && fileuni[len-2] != 'Y'))
1530 goto unchanged;
Christian Heimes3b06e532008-01-07 20:12:44 +00001531
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001532 /* Start by trying to turn PEP 3147 path into source path. If that
1533 * fails, just chop off the trailing character, i.e. legacy pyc path
1534 * to py.
1535 */
Victor Stinnerc9abda02011-03-14 13:33:46 -04001536 py = make_source_pathname(filename);
1537 if (py == NULL) {
1538 PyErr_Clear();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001539 py = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND, fileuni, len - 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001540 }
Victor Stinnerc9abda02011-03-14 13:33:46 -04001541 if (py == NULL)
1542 goto error;
Barry Warsaw28a691b2010-04-17 00:19:56 +00001543
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02001544 if (_Py_stat(py, &statbuf) == 0 && S_ISREG(statbuf.st_mode)) {
1545 PyMem_Free(fileuni);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001546 return py;
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02001547 }
Victor Stinnerc9abda02011-03-14 13:33:46 -04001548 Py_DECREF(py);
1549 goto unchanged;
1550
1551error:
1552 PyErr_Clear();
1553unchanged:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001554 PyMem_Free(fileuni);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001555 Py_INCREF(filename);
1556 return filename;
Christian Heimes3b06e532008-01-07 20:12:44 +00001557}
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001558
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001559/* Forward */
Victor Stinner41c5fec2011-03-13 21:46:30 -04001560static PyObject *load_module(PyObject *, FILE *, PyObject *, int, PyObject *);
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001561static struct filedescr *find_module(PyObject *, PyObject *, PyObject *,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001562 PyObject **, FILE **, PyObject **);
Victor Stinner53dc7352011-03-20 01:50:21 +01001563static struct _frozen * find_frozen(PyObject *);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001564
1565/* Load a package and return its module object WITH INCREMENTED
1566 REFERENCE COUNT */
1567
1568static PyObject *
Victor Stinnerc9abda02011-03-14 13:33:46 -04001569load_package(PyObject *name, PyObject *pathname)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001570{
Victor Stinner41c5fec2011-03-13 21:46:30 -04001571 PyObject *m, *d, *bufobj;
Victor Stinnerc9abda02011-03-14 13:33:46 -04001572 PyObject *file = NULL, *path_list = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001573 int err;
Victor Stinnerc9abda02011-03-14 13:33:46 -04001574 FILE *fp = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001575 struct filedescr *fdp;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001576
Victor Stinnerc9abda02011-03-14 13:33:46 -04001577 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001578 if (m == NULL)
1579 return NULL;
1580 if (Py_VerboseFlag)
Victor Stinner98dbba52011-03-14 15:15:47 -04001581 PySys_FormatStderr("import %U # directory %R\n",
1582 name, pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001583 file = get_sourcefile(pathname);
1584 if (file == NULL)
Victor Stinnerc9abda02011-03-14 13:33:46 -04001585 return NULL;
1586 path_list = Py_BuildValue("[O]", file);
1587 if (path_list == NULL) {
1588 Py_DECREF(file);
1589 return NULL;
1590 }
1591 d = PyModule_GetDict(m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001592 err = PyDict_SetItemString(d, "__file__", file);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001593 Py_DECREF(file);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001594 if (err == 0)
Victor Stinnerc9abda02011-03-14 13:33:46 -04001595 err = PyDict_SetItemString(d, "__path__", path_list);
1596 if (err != 0) {
1597 Py_DECREF(path_list);
1598 return NULL;
1599 }
Victor Stinner533d7832011-03-14 13:22:54 -04001600 fdp = find_module(name, initstr, path_list,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001601 &bufobj, &fp, NULL);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001602 Py_DECREF(path_list);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001603 if (fdp == NULL) {
1604 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1605 PyErr_Clear();
1606 Py_INCREF(m);
Victor Stinnerc9abda02011-03-14 13:33:46 -04001607 return m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001608 }
1609 else
Victor Stinnerc9abda02011-03-14 13:33:46 -04001610 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001611 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04001612 m = load_module(name, fp, bufobj, fdp->type, NULL);
1613 Py_XDECREF(bufobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001614 if (fp != NULL)
1615 fclose(fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001616 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001617}
1618
1619
1620/* Helper to test for built-in module */
1621
1622static int
Victor Stinner95872862011-03-07 18:20:56 +01001623is_builtin(PyObject *name)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001624{
Victor Stinner95872862011-03-07 18:20:56 +01001625 int i, cmp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001626 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Victor Stinner95872862011-03-07 18:20:56 +01001627 cmp = PyUnicode_CompareWithASCIIString(name, PyImport_Inittab[i].name);
1628 if (cmp == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001629 if (PyImport_Inittab[i].initfunc == NULL)
1630 return -1;
1631 else
1632 return 1;
1633 }
1634 }
1635 return 0;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001636}
1637
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001638
Just van Rossum52e14d62002-12-30 22:08:05 +00001639/* Return an importer object for a sys.path/pkg.__path__ item 'p',
1640 possibly by fetching it from the path_importer_cache dict. If it
Thomas Wouters89f507f2006-12-13 04:49:30 +00001641 wasn't yet cached, traverse path_hooks until a hook is found
Just van Rossum52e14d62002-12-30 22:08:05 +00001642 that can handle the path item. Return None if no hook could;
1643 this tells our caller it should fall back to the builtin
1644 import mechanism. Cache the result in path_importer_cache.
1645 Returns a borrowed reference. */
1646
1647static PyObject *
1648get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001649 PyObject *p)
Just van Rossum52e14d62002-12-30 22:08:05 +00001650{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 PyObject *importer;
1652 Py_ssize_t j, nhooks;
Just van Rossum52e14d62002-12-30 22:08:05 +00001653
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001654 /* These conditions are the caller's responsibility: */
1655 assert(PyList_Check(path_hooks));
1656 assert(PyDict_Check(path_importer_cache));
Just van Rossum52e14d62002-12-30 22:08:05 +00001657
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001658 nhooks = PyList_Size(path_hooks);
1659 if (nhooks < 0)
1660 return NULL; /* Shouldn't happen */
Just van Rossum52e14d62002-12-30 22:08:05 +00001661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001662 importer = PyDict_GetItem(path_importer_cache, p);
1663 if (importer != NULL)
1664 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +00001665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001666 /* set path_importer_cache[p] to None to avoid recursion */
1667 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0)
1668 return NULL;
Just van Rossum52e14d62002-12-30 22:08:05 +00001669
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001670 for (j = 0; j < nhooks; j++) {
1671 PyObject *hook = PyList_GetItem(path_hooks, j);
1672 if (hook == NULL)
1673 return NULL;
1674 importer = PyObject_CallFunctionObjArgs(hook, p, NULL);
1675 if (importer != NULL)
1676 break;
Just van Rossum52e14d62002-12-30 22:08:05 +00001677
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001678 if (!PyErr_ExceptionMatches(PyExc_ImportError)) {
1679 return NULL;
1680 }
1681 PyErr_Clear();
1682 }
1683 if (importer == NULL) {
1684 importer = PyObject_CallFunctionObjArgs(
1685 (PyObject *)&PyNullImporter_Type, p, NULL
1686 );
1687 if (importer == NULL) {
1688 if (PyErr_ExceptionMatches(PyExc_ImportError)) {
1689 PyErr_Clear();
1690 return Py_None;
1691 }
1692 }
1693 }
1694 if (importer != NULL) {
1695 int err = PyDict_SetItem(path_importer_cache, p, importer);
1696 Py_DECREF(importer);
1697 if (err != 0)
1698 return NULL;
1699 }
1700 return importer;
Just van Rossum52e14d62002-12-30 22:08:05 +00001701}
1702
Christian Heimes9cd17752007-11-18 19:35:23 +00001703PyAPI_FUNC(PyObject *)
1704PyImport_GetImporter(PyObject *path) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001705 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL;
Christian Heimes9cd17752007-11-18 19:35:23 +00001706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001707 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) {
1708 if ((path_hooks = PySys_GetObject("path_hooks"))) {
1709 importer = get_path_importer(path_importer_cache,
1710 path_hooks, path);
1711 }
1712 }
1713 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */
1714 return importer;
Christian Heimes9cd17752007-11-18 19:35:23 +00001715}
1716
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001717/* Search the path (default sys.path) for a module. Return the
1718 corresponding filedescr struct, and (via return arguments) the
1719 pathname and an open file. Return NULL if the module is not found. */
1720
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001721#ifdef MS_COREDLL
Victor Stinner4d6c1c42011-03-08 23:49:04 +01001722extern FILE *_PyWin_FindRegisteredModule(PyObject *, struct filedescr **,
1723 PyObject **p_path);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00001724#endif
1725
Victor Stinner547a2a62011-03-20 03:07:28 +01001726/* Forward */
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001727static int case_ok(PyObject *, Py_ssize_t, PyObject *);
Victor Stinner547a2a62011-03-20 03:07:28 +01001728static int find_init_module(PyObject *);
Just van Rossum52e14d62002-12-30 22:08:05 +00001729static struct filedescr importhookdescr = {"", "", IMP_HOOK};
Guido van Rossum197346f1997-10-31 18:38:52 +00001730
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001731/* Get the path of a module: get its importer and call importer.find_module()
1732 hook, or check if the module if a package (if path/__init__.py exists).
1733
1734 -1: error: a Python error occurred
1735 0: ignore: an error occurred because of invalid data, but the error is not
1736 important enough to be reported.
1737 1: get path: module not found, but *buf contains its path
1738 2: found: *p_fd is the file descriptor (IMP_HOOK or PKG_DIRECTORY)
1739 and *buf is the path */
1740
1741static int
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001742find_module_path(PyObject *fullname, PyObject *name, PyObject *path,
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001743 PyObject *path_hooks, PyObject *path_importer_cache,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001744 PyObject **p_path, PyObject **p_loader, struct filedescr **p_fd)
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001745{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001746 Py_UCS4 buf[MAXPATHLEN+1];
Victor Stinner2fd76e42011-03-14 15:19:39 -04001747 PyObject *path_unicode, *filename;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001748 Py_ssize_t len;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001749 struct stat statbuf;
1750 static struct filedescr fd_package = {"", "", PKG_DIRECTORY};
1751
1752 if (PyUnicode_Check(path)) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04001753 Py_INCREF(path);
1754 path_unicode = path;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001755 }
1756 else if (PyBytes_Check(path)) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04001757 path_unicode = PyUnicode_DecodeFSDefaultAndSize(
1758 PyBytes_AS_STRING(path), PyBytes_GET_SIZE(path));
1759 if (path_unicode == NULL)
1760 return -1;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001761 }
1762 else
1763 return 0;
1764
Victor Stinner46084ba2011-10-06 02:39:42 +02001765 if (PyUnicode_READY(path_unicode))
1766 return -1;
1767
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001768 len = PyUnicode_GET_LENGTH(path_unicode);
Georg Brandl4cb0de22011-09-28 21:49:49 +02001769 if (!PyUnicode_AsUCS4(path_unicode, buf, Py_ARRAY_LENGTH(buf), 1)) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04001770 Py_DECREF(path_unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001771 PyErr_Clear();
1772 return 0;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001773 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04001774 Py_DECREF(path_unicode);
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001775
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001776 if (Py_UCS4_strlen(buf) != len)
Victor Stinner2fd76e42011-03-14 15:19:39 -04001777 return 0; /* path contains '\0' */
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001778
1779 /* sys.path_hooks import hook */
1780 if (p_loader != NULL) {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02001781 _Py_IDENTIFIER(find_module);
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001782 PyObject *importer;
1783
1784 importer = get_path_importer(path_importer_cache,
1785 path_hooks, path);
1786 if (importer == NULL) {
1787 return -1;
1788 }
1789 /* Note: importer is a borrowed reference */
1790 if (importer != Py_None) {
1791 PyObject *loader;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001792 loader = _PyObject_CallMethodId(importer,
1793 &PyId_find_module, "O", fullname);
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001794 if (loader == NULL)
1795 return -1; /* error */
1796 if (loader != Py_None) {
1797 /* a loader was found */
1798 *p_loader = loader;
1799 *p_fd = &importhookdescr;
1800 return 2;
1801 }
1802 Py_DECREF(loader);
1803 return 0;
1804 }
1805 }
1806 /* no hook was found, use builtin import */
1807
1808 if (len > 0 && buf[len-1] != SEP
1809#ifdef ALTSEP
1810 && buf[len-1] != ALTSEP
1811#endif
1812 )
1813 buf[len++] = SEP;
Georg Brandl4cb0de22011-09-28 21:49:49 +02001814 if (!PyUnicode_AsUCS4(name, buf+len, Py_ARRAY_LENGTH(buf)-len, 1)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001815 PyErr_Clear();
1816 return 0;
1817 }
1818 len += PyUnicode_GET_LENGTH(name);
Victor Stinner2fd76e42011-03-14 15:19:39 -04001819
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001820 filename = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
1821 buf, len);
Victor Stinner2fd76e42011-03-14 15:19:39 -04001822 if (filename == NULL)
1823 return -1;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001824
1825 /* Check for package import (buf holds a directory name,
1826 and there's an __init__ module in that directory */
1827#ifdef HAVE_STAT
Victor Stinner2fd76e42011-03-14 15:19:39 -04001828 if (_Py_stat(filename, &statbuf) == 0 && /* it exists */
Victor Stinnerd0296212011-03-14 14:04:10 -04001829 S_ISDIR(statbuf.st_mode)) /* it's a directory */
1830 {
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001831 int match;
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001832
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001833 match = case_ok(filename, 0, name);
1834 if (match < 0) {
1835 Py_DECREF(filename);
1836 return -1;
1837 }
1838 if (match) { /* case matches */
1839 if (find_init_module(filename)) { /* and has __init__.py */
Victor Stinner2fd76e42011-03-14 15:19:39 -04001840 *p_path = filename;
Victor Stinnerd0296212011-03-14 14:04:10 -04001841 *p_fd = &fd_package;
1842 return 2;
1843 }
1844 else {
1845 int err;
Victor Stinnerd0296212011-03-14 14:04:10 -04001846 err = PyErr_WarnFormat(PyExc_ImportWarning, 1,
Victor Stinner547a2a62011-03-20 03:07:28 +01001847 "Not importing directory %R: missing __init__.py",
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001848 filename);
Victor Stinner547a2a62011-03-20 03:07:28 +01001849 if (err) {
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001850 Py_DECREF(filename);
Victor Stinnerd0296212011-03-14 14:04:10 -04001851 return -1;
Victor Stinner547a2a62011-03-20 03:07:28 +01001852 }
Victor Stinnerd0296212011-03-14 14:04:10 -04001853 }
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001854 }
1855 }
1856#endif
Victor Stinner2fd76e42011-03-14 15:19:39 -04001857 *p_path = filename;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001858 return 1;
1859}
1860
1861/* Find a module in search_path_list. For each path, try
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001862 find_module_path() or try each _PyImport_Filetab suffix.
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001863
1864 If the module is found, return a file descriptor, write the path in
1865 *p_filename, write the pointer to the file object into *p_fp, and (if
1866 p_loader is not NULL) the loader into *p_loader.
1867
1868 Otherwise, raise an exception and return NULL. */
1869
Victor Stinner37580282011-03-20 01:34:43 +01001870static struct filedescr*
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001871find_module_path_list(PyObject *fullname, PyObject *name,
Victor Stinner37580282011-03-20 01:34:43 +01001872 PyObject *search_path_list, PyObject *path_hooks,
1873 PyObject *path_importer_cache,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001874 PyObject **p_path, FILE **p_fp, PyObject **p_loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001875{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001876 Py_ssize_t i, npath;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 struct filedescr *fdp = NULL;
1878 char *filemode;
1879 FILE *fp = NULL;
Victor Stinner2fd76e42011-03-14 15:19:39 -04001880 PyObject *prefix, *filename;
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001881 int match;
Victor Stinner53dc7352011-03-20 01:50:21 +01001882
Victor Stinner37580282011-03-20 01:34:43 +01001883 npath = PyList_Size(search_path_list);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001884 for (i = 0; i < npath; i++) {
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001885 PyObject *path;
1886 int ok;
1887
1888 path = PyList_GetItem(search_path_list, i);
1889 if (path == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 return NULL;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001891
Victor Stinner2fd76e42011-03-14 15:19:39 -04001892 prefix = NULL;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001893 ok = find_module_path(fullname, name, path,
1894 path_hooks, path_importer_cache,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001895 &prefix, p_loader, &fdp);
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001896 if (ok < 0)
1897 return NULL;
1898 if (ok == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001899 continue;
Victor Stinner2fd76e42011-03-14 15:19:39 -04001900 if (ok == 2) {
1901 *p_path = prefix;
Victor Stinnerd68c2cf2011-03-12 16:02:28 -05001902 return fdp;
Victor Stinner2fd76e42011-03-14 15:19:39 -04001903 }
Amaury Forgeot d'Arcf1ca0b12008-06-11 17:40:47 +00001904
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001905 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
Victor Stinnera1fe1f82011-09-23 18:59:08 +02001906 struct stat statbuf;
1907
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001908 filemode = fdp->mode;
1909 if (filemode[0] == 'U')
1910 filemode = "r" PY_STDIOTEXTMODE;
Victor Stinnerd0296212011-03-14 14:04:10 -04001911
Victor Stinner2fd76e42011-03-14 15:19:39 -04001912 filename = PyUnicode_FromFormat("%U%s", prefix, fdp->suffix);
1913 if (filename == NULL) {
1914 Py_DECREF(prefix);
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001915 return NULL;
Victor Stinner2fd76e42011-03-14 15:19:39 -04001916 }
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001917
1918 if (Py_VerboseFlag > 1)
1919 PySys_FormatStderr("# trying %R\n", filename);
1920
Victor Stinnera1fe1f82011-09-23 18:59:08 +02001921 if (_Py_stat(filename, &statbuf) == 0 && /* it exists */
1922 S_ISDIR(statbuf.st_mode)) /* it's a directory */
1923 {
1924 Py_DECREF(filename);
1925 continue;
1926 }
1927
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001928 fp = _Py_fopen(filename, filemode);
1929 if (fp == NULL) {
1930 Py_DECREF(filename);
Victor Stinner925ef392011-06-20 15:01:10 +02001931 if (PyErr_Occurred()) {
1932 Py_DECREF(prefix);
1933 return NULL;
1934 }
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001935 continue;
1936 }
1937 match = case_ok(filename, -(Py_ssize_t)strlen(fdp->suffix), name);
1938 if (match < 0) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04001939 Py_DECREF(prefix);
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001940 Py_DECREF(filename);
1941 return NULL;
1942 }
1943 if (match) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04001944 Py_DECREF(prefix);
1945 *p_path = filename;
Victor Stinnerd0296212011-03-14 14:04:10 -04001946 *p_fp = fp;
1947 return fdp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001948 }
Victor Stinnerc9a271c2011-03-14 14:34:13 -04001949 Py_DECREF(filename);
Victor Stinnerd0296212011-03-14 14:04:10 -04001950
1951 fclose(fp);
1952 fp = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001953 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04001954 Py_DECREF(prefix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001955 }
Victor Stinnerd0296212011-03-14 14:04:10 -04001956 PyErr_Format(PyExc_ImportError,
Victor Stinner98dbba52011-03-14 15:15:47 -04001957 "No module named %R", name);
Victor Stinnerd0296212011-03-14 14:04:10 -04001958 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001959}
1960
Victor Stinner37580282011-03-20 01:34:43 +01001961/* Find a module:
1962
1963 - try find_module() of each sys.meta_path hook
1964 - try find_frozen()
1965 - try is_builtin()
1966 - try _PyWin_FindRegisteredModule() (Windows only)
1967 - otherwise, call find_module_path_list() with search_path_list (if not
1968 NULL) or sys.path
1969
Victor Stinner2fd76e42011-03-14 15:19:39 -04001970 fullname can be NULL, but only if p_loader is NULL.
1971
Victor Stinner37580282011-03-20 01:34:43 +01001972 Return:
1973
1974 - &fd_builtin (C_BUILTIN) if it is a builtin
1975 - &fd_frozen (PY_FROZEN) if it is frozen
Victor Stinner2fd76e42011-03-14 15:19:39 -04001976 - &fd_package (PKG_DIRECTORY) and write the filename into *p_path
Victor Stinner37580282011-03-20 01:34:43 +01001977 if it is a package
1978 - &importhookdescr (IMP_HOOK) and write the loader into *p_loader if a
1979 importer loader was found
1980 - a file descriptor (PY_SOURCE, PY_COMPILED, C_EXTENSION, PY_RESOURCE or
1981 PY_CODERESOURCE: see _PyImport_Filetab), write the filename into
Victor Stinner2fd76e42011-03-14 15:19:39 -04001982 *p_path and the pointer to the open file into *p_fp
Victor Stinner37580282011-03-20 01:34:43 +01001983 - NULL on error
1984
Victor Stinner2fd76e42011-03-14 15:19:39 -04001985 By default, *p_path, *p_fp and *p_loader (if set) are set to NULL.
1986 Eg. *p_path is set to NULL for a builtin package.
1987*/
Victor Stinner37580282011-03-20 01:34:43 +01001988
1989static struct filedescr *
Victor Stinnerad3c03b2011-03-14 09:21:33 -04001990find_module(PyObject *fullname, PyObject *name, PyObject *search_path_list,
Victor Stinner2fd76e42011-03-14 15:19:39 -04001991 PyObject **p_path, FILE **p_fp, PyObject **p_loader)
Victor Stinner37580282011-03-20 01:34:43 +01001992{
1993 Py_ssize_t i, npath;
1994 static struct filedescr fd_frozen = {"", "", PY_FROZEN};
1995 static struct filedescr fd_builtin = {"", "", C_BUILTIN};
1996 PyObject *path_hooks, *path_importer_cache;
Victor Stinner37580282011-03-20 01:34:43 +01001997
Victor Stinner2fd76e42011-03-14 15:19:39 -04001998 *p_path = NULL;
Victor Stinner37580282011-03-20 01:34:43 +01001999 *p_fp = NULL;
2000 if (p_loader != NULL)
2001 *p_loader = NULL;
2002
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002003 if (PyUnicode_GET_LENGTH(name) > MAXPATHLEN) {
2004 PyErr_SetString(PyExc_OverflowError,
2005 "module name is too long");
2006 return NULL;
2007 }
2008
Victor Stinner37580282011-03-20 01:34:43 +01002009 /* sys.meta_path import hook */
2010 if (p_loader != NULL) {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002011 _Py_IDENTIFIER(find_module);
Victor Stinner37580282011-03-20 01:34:43 +01002012 PyObject *meta_path;
2013
2014 meta_path = PySys_GetObject("meta_path");
2015 if (meta_path == NULL || !PyList_Check(meta_path)) {
Victor Stinner16191322011-09-15 19:28:05 +02002016 PyErr_SetString(PyExc_RuntimeError,
Victor Stinner37580282011-03-20 01:34:43 +01002017 "sys.meta_path must be a list of "
2018 "import hooks");
2019 return NULL;
2020 }
2021 Py_INCREF(meta_path); /* zap guard */
2022 npath = PyList_Size(meta_path);
2023 for (i = 0; i < npath; i++) {
2024 PyObject *loader;
2025 PyObject *hook = PyList_GetItem(meta_path, i);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002026 loader = _PyObject_CallMethodId(hook, &PyId_find_module,
Victor Stinnerad3c03b2011-03-14 09:21:33 -04002027 "OO", fullname,
Victor Stinner37580282011-03-20 01:34:43 +01002028 search_path_list != NULL ?
2029 search_path_list : Py_None);
2030 if (loader == NULL) {
2031 Py_DECREF(meta_path);
2032 return NULL; /* true error */
2033 }
2034 if (loader != Py_None) {
2035 /* a loader was found */
2036 *p_loader = loader;
2037 Py_DECREF(meta_path);
2038 return &importhookdescr;
2039 }
2040 Py_DECREF(loader);
2041 }
2042 Py_DECREF(meta_path);
2043 }
2044
Victor Stinnerdf75a022011-03-14 13:40:04 -04002045 if (find_frozen(fullname) != NULL)
Victor Stinnerad3c03b2011-03-14 09:21:33 -04002046 return &fd_frozen;
Victor Stinner37580282011-03-20 01:34:43 +01002047
2048 if (search_path_list == NULL) {
2049#ifdef MS_COREDLL
2050 FILE *fp;
2051 struct filedescr *fdp;
Victor Stinner37580282011-03-20 01:34:43 +01002052#endif
Victor Stinnerdf75a022011-03-14 13:40:04 -04002053 if (is_builtin(name))
Victor Stinner37580282011-03-20 01:34:43 +01002054 return &fd_builtin;
Victor Stinner37580282011-03-20 01:34:43 +01002055#ifdef MS_COREDLL
Victor Stinner2fd76e42011-03-14 15:19:39 -04002056 fp = _PyWin_FindRegisteredModule(name, &fdp, p_path);
Victor Stinner37580282011-03-20 01:34:43 +01002057 if (fp != NULL) {
Victor Stinner37580282011-03-20 01:34:43 +01002058 *p_fp = fp;
2059 return fdp;
2060 }
2061 else if (PyErr_Occurred())
2062 return NULL;
2063#endif
Victor Stinner37580282011-03-20 01:34:43 +01002064 search_path_list = PySys_GetObject("path");
2065 }
2066
2067 if (search_path_list == NULL || !PyList_Check(search_path_list)) {
Victor Stinner16191322011-09-15 19:28:05 +02002068 PyErr_SetString(PyExc_RuntimeError,
Victor Stinner37580282011-03-20 01:34:43 +01002069 "sys.path must be a list of directory names");
2070 return NULL;
2071 }
2072
2073 path_hooks = PySys_GetObject("path_hooks");
2074 if (path_hooks == NULL || !PyList_Check(path_hooks)) {
Victor Stinner16191322011-09-15 19:28:05 +02002075 PyErr_SetString(PyExc_RuntimeError,
Victor Stinner37580282011-03-20 01:34:43 +01002076 "sys.path_hooks must be a list of "
2077 "import hooks");
2078 return NULL;
2079 }
2080 path_importer_cache = PySys_GetObject("path_importer_cache");
2081 if (path_importer_cache == NULL ||
2082 !PyDict_Check(path_importer_cache)) {
Victor Stinner16191322011-09-15 19:28:05 +02002083 PyErr_SetString(PyExc_RuntimeError,
Victor Stinner37580282011-03-20 01:34:43 +01002084 "sys.path_importer_cache must be a dict");
2085 return NULL;
2086 }
2087
2088 return find_module_path_list(fullname, name,
2089 search_path_list, path_hooks,
2090 path_importer_cache,
Victor Stinner2fd76e42011-03-14 15:19:39 -04002091 p_path, p_fp, p_loader);
Victor Stinner37580282011-03-20 01:34:43 +01002092}
2093
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002094/* case_bytes(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name)
Tim Petersd1e87a82001-03-01 18:12:00 +00002095 * The arguments here are tricky, best shown by example:
2096 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0
2097 * ^ ^ ^ ^
2098 * |--------------------- buf ---------------------|
2099 * |------------------- len ------------------|
2100 * |------ name -------|
2101 * |----- namelen -----|
2102 * buf is the full path, but len only counts up to (& exclusive of) the
2103 * extension. name is the module name, also exclusive of extension.
2104 *
2105 * We've already done a successful stat() or fopen() on buf, so know that
2106 * there's some match, possibly case-insensitive.
2107 *
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002108 * case_bytes() is to return 1 if there's a case-sensitive match for
2109 * name, else 0. case_bytes() is also to return 1 if envar PYTHONCASEOK
Tim Peters50d8d372001-02-28 05:34:27 +00002110 * exists.
Tim Petersd1e87a82001-03-01 18:12:00 +00002111 *
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002112 * case_bytes() is used to implement case-sensitive import semantics even
Tim Peters50d8d372001-02-28 05:34:27 +00002113 * on platforms with case-insensitive filesystems. It's trivial to implement
2114 * for case-sensitive filesystems. It's pretty much a cross-platform
2115 * nightmare for systems with case-insensitive filesystems.
2116 */
Guido van Rossum0980bd91998-02-13 17:18:36 +00002117
Tim Peters50d8d372001-02-28 05:34:27 +00002118/* First we may need a pile of platform-specific header files; the sequence
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002119 * of #if's here should match the sequence in the body of case_bytes().
Tim Peters50d8d372001-02-28 05:34:27 +00002120 */
Jason Tishler7961aa62005-05-20 00:56:54 +00002121#if defined(MS_WINDOWS)
Guido van Rossum0980bd91998-02-13 17:18:36 +00002122#include <windows.h>
Guido van Rossum4c3f57c2001-01-10 20:40:46 +00002123
Tim Peters50d8d372001-02-28 05:34:27 +00002124#elif defined(DJGPP)
2125#include <dir.h>
2126
Jason Tishler7961aa62005-05-20 00:56:54 +00002127#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Tim Peters430f5d42001-03-01 01:30:56 +00002128#include <sys/types.h>
2129#include <dirent.h>
2130
Andrew MacIntyred9400542002-02-26 11:41:34 +00002131#elif defined(PYOS_OS2)
2132#define INCL_DOS
2133#define INCL_DOSERRORS
2134#define INCL_NOPMAPI
2135#include <os2.h>
Tim Peters50d8d372001-02-28 05:34:27 +00002136#endif
2137
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002138#if defined(DJGPP) \
2139 || ((defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) \
2140 && defined(HAVE_DIRENT_H)) \
2141 || defined(PYOS_OS2)
2142# define USE_CASE_OK_BYTES
2143#endif
2144
2145
2146#ifdef USE_CASE_OK_BYTES
Guido van Rossum0980bd91998-02-13 17:18:36 +00002147static int
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002148case_bytes(char *buf, Py_ssize_t len, Py_ssize_t namelen, const char *name)
Guido van Rossum0980bd91998-02-13 17:18:36 +00002149{
Tim Peters50d8d372001-02-28 05:34:27 +00002150/* Pick a platform-specific implementation; the sequence of #if's here should
2151 * match the sequence just above.
2152 */
2153
Tim Peters50d8d372001-02-28 05:34:27 +00002154/* DJGPP */
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002155#if defined(DJGPP)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002156 struct ffblk ffblk;
2157 int done;
Tim Peters50d8d372001-02-28 05:34:27 +00002158
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002159 if (Py_GETENV("PYTHONCASEOK") != NULL)
2160 return 1;
Tim Peters50d8d372001-02-28 05:34:27 +00002161
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002162 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC);
2163 if (done) {
2164 PyErr_Format(PyExc_NameError,
2165 "Can't find file for module %.100s\n(filename %.300s)",
2166 name, buf);
Victor Stinner9c61e242011-03-22 01:22:27 +01002167 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002168 }
2169 return strncmp(ffblk.ff_name, name, namelen) == 0;
Guido van Rossum0980bd91998-02-13 17:18:36 +00002170
Jason Tishler7961aa62005-05-20 00:56:54 +00002171/* new-fangled macintosh (macosx) or Cygwin */
2172#elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002173 DIR *dirp;
2174 struct dirent *dp;
2175 char dirname[MAXPATHLEN + 1];
2176 const int dirlen = len - namelen - 1; /* don't want trailing SEP */
Tim Peters430f5d42001-03-01 01:30:56 +00002177
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002178 if (Py_GETENV("PYTHONCASEOK") != NULL)
2179 return 1;
Tim Petersdbe6ebb2001-03-01 08:47:29 +00002180
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002181 /* Copy the dir component into dirname; substitute "." if empty */
2182 if (dirlen <= 0) {
2183 dirname[0] = '.';
2184 dirname[1] = '\0';
2185 }
2186 else {
2187 assert(dirlen <= MAXPATHLEN);
2188 memcpy(dirname, buf, dirlen);
2189 dirname[dirlen] = '\0';
2190 }
2191 /* Open the directory and search the entries for an exact match. */
2192 dirp = opendir(dirname);
2193 if (dirp) {
2194 char *nameWithExt = buf + len - namelen;
2195 while ((dp = readdir(dirp)) != NULL) {
2196 const int thislen =
Tim Peters430f5d42001-03-01 01:30:56 +00002197#ifdef _DIRENT_HAVE_D_NAMELEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002198 dp->d_namlen;
Tim Peters430f5d42001-03-01 01:30:56 +00002199#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002200 strlen(dp->d_name);
Tim Peters430f5d42001-03-01 01:30:56 +00002201#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002202 if (thislen >= namelen &&
2203 strcmp(dp->d_name, nameWithExt) == 0) {
2204 (void)closedir(dirp);
2205 return 1; /* Found */
2206 }
2207 }
2208 (void)closedir(dirp);
2209 }
2210 return 0 ; /* Not found */
Tim Peters430f5d42001-03-01 01:30:56 +00002211
Andrew MacIntyred9400542002-02-26 11:41:34 +00002212/* OS/2 */
2213#elif defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002214 HDIR hdir = 1;
2215 ULONG srchcnt = 1;
2216 FILEFINDBUF3 ffbuf;
2217 APIRET rc;
Andrew MacIntyred9400542002-02-26 11:41:34 +00002218
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002219 if (Py_GETENV("PYTHONCASEOK") != NULL)
2220 return 1;
Andrew MacIntyred9400542002-02-26 11:41:34 +00002221
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002222 rc = DosFindFirst(buf,
2223 &hdir,
2224 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY,
2225 &ffbuf, sizeof(ffbuf),
2226 &srchcnt,
2227 FIL_STANDARD);
2228 if (rc != NO_ERROR)
2229 return 0;
2230 return strncmp(ffbuf.achName, name, namelen) == 0;
Andrew MacIntyred9400542002-02-26 11:41:34 +00002231
Tim Peters50d8d372001-02-28 05:34:27 +00002232/* assuming it's a case-sensitive filesystem, so there's nothing to do! */
2233#else
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002234# error "USE_CASE_OK_BYTES is not correctly defined"
2235#endif
2236}
2237#endif
2238
2239/*
2240 * Check if a filename case matchs the name case. We've already done a
2241 * successful stat() or fopen() on buf, so know that there's some match,
2242 * possibly case-insensitive.
2243 *
2244 * case_ok() is to return 1 if there's a case-sensitive match for name, 0 if it
2245 * the filename doesn't match, or -1 on error. case_ok() is also to return 1
2246 * if envar PYTHONCASEOK exists.
2247 *
2248 * case_ok() is used to implement case-sensitive import semantics even
2249 * on platforms with case-insensitive filesystems. It's trivial to implement
2250 * for case-sensitive filesystems. It's pretty much a cross-platform
2251 * nightmare for systems with case-insensitive filesystems.
2252 */
2253
2254static int
2255case_ok(PyObject *filename, Py_ssize_t prefix_delta, PyObject *name)
2256{
2257#ifdef MS_WINDOWS
2258 WIN32_FIND_DATAW data;
2259 HANDLE h;
2260 int cmp;
Victor Stinnerbeac78b2011-10-11 21:55:01 +02002261 wchar_t *wname;
Victor Stinner8c981892011-10-11 22:27:13 +02002262 Py_ssize_t wname_len;
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002263
2264 if (Py_GETENV("PYTHONCASEOK") != NULL)
2265 return 1;
2266
2267 h = FindFirstFileW(PyUnicode_AS_UNICODE(filename), &data);
2268 if (h == INVALID_HANDLE_VALUE) {
2269 PyErr_Format(PyExc_NameError,
2270 "Can't find file for module %R\n(filename %R)",
2271 name, filename);
2272 return 0;
2273 }
2274 FindClose(h);
Victor Stinnerbeac78b2011-10-11 21:55:01 +02002275
2276 wname = PyUnicode_AsUnicodeAndSize(name, &wname_len);
2277 if (wname == NULL)
2278 return -1;
2279
2280 cmp = wcsncmp(data.cFileName, wname, wname_len);
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002281 return cmp == 0;
2282#elif defined(USE_CASE_OK_BYTES)
2283 int match;
2284 PyObject *filebytes, *namebytes;
2285 filebytes = PyUnicode_EncodeFSDefault(filename);
2286 if (filebytes == NULL)
2287 return -1;
2288 namebytes = PyUnicode_EncodeFSDefault(name);
2289 if (namebytes == NULL) {
2290 Py_DECREF(filebytes);
2291 return -1;
2292 }
2293 match = case_bytes(
2294 PyBytes_AS_STRING(filebytes),
2295 PyBytes_GET_SIZE(filebytes) + prefix_delta,
Victor Stinner1304f2d2011-03-20 04:28:55 +01002296 PyBytes_GET_SIZE(namebytes),
2297 PyBytes_AS_STRING(namebytes));
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002298 Py_DECREF(filebytes);
2299 Py_DECREF(namebytes);
2300 return match;
2301#else
2302 /* assuming it's a case-sensitive filesystem, so there's nothing to do! */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002303 return 1;
Guido van Rossum0980bd91998-02-13 17:18:36 +00002304
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00002305#endif
Tim Peters50d8d372001-02-28 05:34:27 +00002306}
Guido van Rossum4d1b3b91998-02-13 23:27:59 +00002307
Guido van Rossum197346f1997-10-31 18:38:52 +00002308#ifdef HAVE_STAT
Victor Stinner4f4402c2010-08-14 14:50:26 +00002309
Victor Stinner547a2a62011-03-20 03:07:28 +01002310/* Helper to look for __init__.py or __init__.py[co] in potential package.
2311 Return 1 if __init__ was found, 0 if not, or -1 on error. */
Guido van Rossum197346f1997-10-31 18:38:52 +00002312static int
Victor Stinner547a2a62011-03-20 03:07:28 +01002313find_init_module(PyObject *directory)
Guido van Rossum197346f1997-10-31 18:38:52 +00002314{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002315 struct stat statbuf;
Victor Stinner547a2a62011-03-20 03:07:28 +01002316 PyObject *filename;
2317 int match;
Guido van Rossum197346f1997-10-31 18:38:52 +00002318
Victor Stinner547a2a62011-03-20 03:07:28 +01002319 filename = PyUnicode_FromFormat("%U%c__init__.py", directory, SEP);
2320 if (filename == NULL)
2321 return -1;
2322 if (_Py_stat(filename, &statbuf) == 0) {
Victor Stinnercc9564e2011-03-20 04:58:29 +01002323 /* 3=len(".py") */
2324 match = case_ok(filename, -3, initstr);
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002325 if (match < 0) {
2326 Py_DECREF(filename);
2327 return -1;
2328 }
2329 if (match) {
Victor Stinner547a2a62011-03-20 03:07:28 +01002330 Py_DECREF(filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002331 return 1;
2332 }
2333 }
Victor Stinner547a2a62011-03-20 03:07:28 +01002334 Py_DECREF(filename);
2335
2336 filename = PyUnicode_FromFormat("%U%c__init__.py%c",
2337 directory, SEP, Py_OptimizeFlag ? 'o' : 'c');
2338 if (filename == NULL)
2339 return -1;
2340 if (_Py_stat(filename, &statbuf) == 0) {
Victor Stinnercc9564e2011-03-20 04:58:29 +01002341 /* 4=len(".pyc") */
2342 match = case_ok(filename, -4, initstr);
Victor Stinnerc9a271c2011-03-14 14:34:13 -04002343 if (match < 0) {
2344 Py_DECREF(filename);
2345 return -1;
2346 }
2347 if (match) {
Victor Stinner547a2a62011-03-20 03:07:28 +01002348 Py_DECREF(filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002349 return 1;
2350 }
2351 }
Victor Stinner547a2a62011-03-20 03:07:28 +01002352 Py_DECREF(filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002353 return 0;
Guido van Rossum197346f1997-10-31 18:38:52 +00002354}
Guido van Rossum48a680c2001-03-02 06:34:14 +00002355
Guido van Rossum197346f1997-10-31 18:38:52 +00002356#endif /* HAVE_STAT */
2357
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002358
Victor Stinner95872862011-03-07 18:20:56 +01002359static int init_builtin(PyObject *); /* Forward */
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002360
Victor Stinner44c6c152010-08-09 00:59:10 +00002361static PyObject*
Victor Stinner95872862011-03-07 18:20:56 +01002362load_builtin(PyObject *name, int type)
Victor Stinner44c6c152010-08-09 00:59:10 +00002363{
2364 PyObject *m, *modules;
2365 int err;
2366
Victor Stinner44c6c152010-08-09 00:59:10 +00002367 if (type == C_BUILTIN)
2368 err = init_builtin(name);
2369 else
Victor Stinner95872862011-03-07 18:20:56 +01002370 err = PyImport_ImportFrozenModuleObject(name);
Victor Stinner44c6c152010-08-09 00:59:10 +00002371 if (err < 0)
2372 return NULL;
2373 if (err == 0) {
2374 PyErr_Format(PyExc_ImportError,
Victor Stinner95872862011-03-07 18:20:56 +01002375 "Purported %s module %R not found",
2376 type == C_BUILTIN ? "builtin" : "frozen",
Victor Stinner44c6c152010-08-09 00:59:10 +00002377 name);
2378 return NULL;
2379 }
2380
2381 modules = PyImport_GetModuleDict();
Victor Stinner95872862011-03-07 18:20:56 +01002382 m = PyDict_GetItem(modules, name);
Victor Stinner44c6c152010-08-09 00:59:10 +00002383 if (m == NULL) {
2384 PyErr_Format(
2385 PyExc_ImportError,
Victor Stinner95872862011-03-07 18:20:56 +01002386 "%s module %R not properly initialized",
2387 type == C_BUILTIN ? "builtin" : "frozen",
Victor Stinner44c6c152010-08-09 00:59:10 +00002388 name);
2389 return NULL;
2390 }
2391 Py_INCREF(m);
2392 return m;
2393}
2394
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002395/* Load an external module using the default search path and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002396 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002397
Guido van Rossum79f25d91997-04-29 20:08:16 +00002398static PyObject *
Victor Stinner41c5fec2011-03-13 21:46:30 -04002399load_module(PyObject *name, FILE *fp, PyObject *pathname, int type, PyObject *loader)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002400{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002401 PyObject *m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002403 /* First check that there's an open file (if we need one) */
2404 switch (type) {
2405 case PY_SOURCE:
2406 case PY_COMPILED:
2407 if (fp == NULL) {
2408 PyErr_Format(PyExc_ValueError,
Victor Stinner41c5fec2011-03-13 21:46:30 -04002409 "file object required for import (type code %d)",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002410 type);
2411 return NULL;
2412 }
2413 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002415 switch (type) {
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002416
Victor Stinner41c5fec2011-03-13 21:46:30 -04002417 case PY_SOURCE:
2418 m = load_source_module(name, pathname, fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002419 break;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002420
Victor Stinner41c5fec2011-03-13 21:46:30 -04002421 case PY_COMPILED:
2422 m = load_compiled_module(name, pathname, fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002423 break;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002424
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002425#ifdef HAVE_DYNAMIC_LOADING
Victor Stinner41c5fec2011-03-13 21:46:30 -04002426 case C_EXTENSION:
2427 m = _PyImport_LoadDynamicModule(name, pathname, fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002428 break;
Guido van Rossum96a8fb71999-12-22 14:09:35 +00002429#endif
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002430
Victor Stinner41c5fec2011-03-13 21:46:30 -04002431 case PKG_DIRECTORY:
2432 m = load_package(name, pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002433 break;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002434
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002435 case C_BUILTIN:
Victor Stinner41c5fec2011-03-13 21:46:30 -04002436 case PY_FROZEN:
2437 m = load_builtin(name, type);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002438 break;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002439
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002440 case IMP_HOOK: {
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002441 _Py_IDENTIFIER(load_module);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002442 if (loader == NULL) {
2443 PyErr_SetString(PyExc_ImportError,
2444 "import hook without loader");
2445 return NULL;
2446 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002447 m = _PyObject_CallMethodId(loader, &PyId_load_module, "O", name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002448 break;
2449 }
Just van Rossum52e14d62002-12-30 22:08:05 +00002450
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002451 default:
2452 PyErr_Format(PyExc_ImportError,
Victor Stinner98dbba52011-03-14 15:15:47 -04002453 "Don't know how to import %R (type code %d)",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002454 name, type);
2455 m = NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002457 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002458
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002459 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002460}
2461
2462
2463/* Initialize a built-in module.
Thomas Wouters89f507f2006-12-13 04:49:30 +00002464 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002465 an exception set if the initialization failed. */
Guido van Rossum7f133ed1991-02-19 12:23:57 +00002466
Guido van Rossum7f133ed1991-02-19 12:23:57 +00002467static int
Victor Stinner95872862011-03-07 18:20:56 +01002468init_builtin(PyObject *name)
Guido van Rossum7f133ed1991-02-19 12:23:57 +00002469{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002470 struct _inittab *p;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002471
Victor Stinner95872862011-03-07 18:20:56 +01002472 if (_PyImport_FindExtensionObject(name, name) != NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002473 return 1;
Guido van Rossum25ce5661997-08-02 03:10:38 +00002474
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002475 for (p = PyImport_Inittab; p->name != NULL; p++) {
2476 PyObject *mod;
Victor Stinner95872862011-03-07 18:20:56 +01002477 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002478 if (p->initfunc == NULL) {
2479 PyErr_Format(PyExc_ImportError,
Victor Stinner95872862011-03-07 18:20:56 +01002480 "Cannot re-init internal module %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002481 name);
2482 return -1;
2483 }
2484 if (Py_VerboseFlag)
Victor Stinner95872862011-03-07 18:20:56 +01002485 PySys_FormatStderr("import %U # builtin\n", name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002486 mod = (*p->initfunc)();
2487 if (mod == 0)
2488 return -1;
Victor Stinner95872862011-03-07 18:20:56 +01002489 if (_PyImport_FixupExtensionObject(mod, name, name) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002490 return -1;
2491 /* FixupExtension has put the module into sys.modules,
2492 so we can release our own reference. */
2493 Py_DECREF(mod);
2494 return 1;
2495 }
2496 }
2497 return 0;
Guido van Rossum7f133ed1991-02-19 12:23:57 +00002498}
Guido van Rossumf56e3db1993-04-01 20:59:32 +00002499
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002500
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002501/* Frozen modules */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002502
Guido van Rossumcfd0a221996-06-17 17:06:34 +00002503static struct _frozen *
Victor Stinner53dc7352011-03-20 01:50:21 +01002504find_frozen(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002505{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002506 struct _frozen *p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002507
Victor Stinner53dc7352011-03-20 01:50:21 +01002508 if (name == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002509 return NULL;
Benjamin Petersond968e272008-11-05 22:48:33 +00002510
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002511 for (p = PyImport_FrozenModules; ; p++) {
2512 if (p->name == NULL)
2513 return NULL;
Victor Stinner53dc7352011-03-20 01:50:21 +01002514 if (PyUnicode_CompareWithASCIIString(name, p->name) == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002515 break;
2516 }
2517 return p;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002518}
2519
Guido van Rossum79f25d91997-04-29 20:08:16 +00002520static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01002521get_frozen_object(PyObject *name)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002522{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 struct _frozen *p = find_frozen(name);
2524 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 if (p == NULL) {
2527 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01002528 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002529 name);
2530 return NULL;
2531 }
2532 if (p->code == NULL) {
2533 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01002534 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002535 name);
2536 return NULL;
2537 }
2538 size = p->size;
2539 if (size < 0)
2540 size = -size;
2541 return PyMarshal_ReadObjectFromString((char *)p->code, size);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002542}
2543
Brett Cannon8d110132009-03-15 02:20:16 +00002544static PyObject *
Victor Stinner53dc7352011-03-20 01:50:21 +01002545is_frozen_package(PyObject *name)
Brett Cannon8d110132009-03-15 02:20:16 +00002546{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002547 struct _frozen *p = find_frozen(name);
2548 int size;
Brett Cannon8d110132009-03-15 02:20:16 +00002549
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002550 if (p == NULL) {
2551 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01002552 "No such frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002553 name);
2554 return NULL;
2555 }
Brett Cannon8d110132009-03-15 02:20:16 +00002556
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002557 size = p->size;
Brett Cannon8d110132009-03-15 02:20:16 +00002558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002559 if (size < 0)
2560 Py_RETURN_TRUE;
2561 else
2562 Py_RETURN_FALSE;
Brett Cannon8d110132009-03-15 02:20:16 +00002563}
2564
2565
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002566/* Initialize a frozen module.
Brett Cannon3c2ac442009-03-08 20:49:47 +00002567 Return 1 for success, 0 if the module is not found, and -1 with
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002568 an exception set if the initialization failed.
2569 This function is also used from frozenmain.c */
Guido van Rossum0b344901995-02-07 15:35:27 +00002570
2571int
Victor Stinner53dc7352011-03-20 01:50:21 +01002572PyImport_ImportFrozenModuleObject(PyObject *name)
Guido van Rossumf56e3db1993-04-01 20:59:32 +00002573{
Victor Stinner53dc7352011-03-20 01:50:21 +01002574 struct _frozen *p;
2575 PyObject *co, *m, *path;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002576 int ispackage;
2577 int size;
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00002578
Victor Stinner53dc7352011-03-20 01:50:21 +01002579 p = find_frozen(name);
2580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002581 if (p == NULL)
2582 return 0;
2583 if (p->code == NULL) {
2584 PyErr_Format(PyExc_ImportError,
Victor Stinner53dc7352011-03-20 01:50:21 +01002585 "Excluded frozen object named %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002586 name);
2587 return -1;
2588 }
2589 size = p->size;
2590 ispackage = (size < 0);
2591 if (ispackage)
2592 size = -size;
2593 if (Py_VerboseFlag)
Victor Stinner53dc7352011-03-20 01:50:21 +01002594 PySys_FormatStderr("import %U # frozen%s\n",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002595 name, ispackage ? " package" : "");
2596 co = PyMarshal_ReadObjectFromString((char *)p->code, size);
2597 if (co == NULL)
2598 return -1;
2599 if (!PyCode_Check(co)) {
2600 PyErr_Format(PyExc_TypeError,
Victor Stinner53dc7352011-03-20 01:50:21 +01002601 "frozen object %R is not a code object",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002602 name);
2603 goto err_return;
2604 }
2605 if (ispackage) {
2606 /* Set __path__ to the package name */
Victor Stinner53dc7352011-03-20 01:50:21 +01002607 PyObject *d, *l;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002608 int err;
Victor Stinner53dc7352011-03-20 01:50:21 +01002609 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002610 if (m == NULL)
2611 goto err_return;
2612 d = PyModule_GetDict(m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002613 l = PyList_New(1);
2614 if (l == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002615 goto err_return;
2616 }
Victor Stinner53dc7352011-03-20 01:50:21 +01002617 Py_INCREF(name);
2618 PyList_SET_ITEM(l, 0, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002619 err = PyDict_SetItemString(d, "__path__", l);
2620 Py_DECREF(l);
2621 if (err != 0)
2622 goto err_return;
2623 }
Victor Stinner53dc7352011-03-20 01:50:21 +01002624 path = PyUnicode_FromString("<frozen>");
2625 if (path == NULL)
2626 goto err_return;
2627 m = PyImport_ExecCodeModuleObject(name, co, path, NULL);
2628 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002629 if (m == NULL)
2630 goto err_return;
2631 Py_DECREF(co);
2632 Py_DECREF(m);
2633 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002634err_return:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002635 Py_DECREF(co);
2636 return -1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +00002637}
Guido van Rossum74e6a111994-08-29 12:54:38 +00002638
Victor Stinner53dc7352011-03-20 01:50:21 +01002639int
2640PyImport_ImportFrozenModule(char *name)
2641{
2642 PyObject *nameobj;
2643 int ret;
2644 nameobj = PyUnicode_InternFromString(name);
2645 if (nameobj == NULL)
2646 return -1;
2647 ret = PyImport_ImportFrozenModuleObject(nameobj);
2648 Py_DECREF(nameobj);
2649 return ret;
2650}
2651
Guido van Rossum74e6a111994-08-29 12:54:38 +00002652
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002653/* Import a module, either built-in, frozen, or external, and return
Guido van Rossum7f9fa971995-01-20 16:53:12 +00002654 its module object WITH INCREMENTED REFERENCE COUNT */
Guido van Rossum74e6a111994-08-29 12:54:38 +00002655
Guido van Rossum79f25d91997-04-29 20:08:16 +00002656PyObject *
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00002657PyImport_ImportModule(const char *name)
Guido van Rossum74e6a111994-08-29 12:54:38 +00002658{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002659 PyObject *pname;
2660 PyObject *result;
Marc-André Lemburg3c61c352001-02-09 19:40:15 +00002661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002662 pname = PyUnicode_FromString(name);
2663 if (pname == NULL)
2664 return NULL;
2665 result = PyImport_Import(pname);
2666 Py_DECREF(pname);
2667 return result;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002668}
2669
Christian Heimes072c0f12008-01-03 23:01:04 +00002670/* Import a module without blocking
2671 *
2672 * At first it tries to fetch the module from sys.modules. If the module was
2673 * never loaded before it loads it with PyImport_ImportModule() unless another
2674 * thread holds the import lock. In the latter case the function raises an
2675 * ImportError instead of blocking.
2676 *
2677 * Returns the module object with incremented ref count.
2678 */
2679PyObject *
2680PyImport_ImportModuleNoBlock(const char *name)
2681{
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002682 PyObject *nameobj, *modules, *result;
Victor Stinner0af03062011-09-02 00:11:43 +02002683#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002684 long me;
Victor Stinner0af03062011-09-02 00:11:43 +02002685#endif
Christian Heimes072c0f12008-01-03 23:01:04 +00002686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002687 /* Try to get the module from sys.modules[name] */
2688 modules = PyImport_GetModuleDict();
2689 if (modules == NULL)
2690 return NULL;
Christian Heimes072c0f12008-01-03 23:01:04 +00002691
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002692 nameobj = PyUnicode_FromString(name);
2693 if (nameobj == NULL)
2694 return NULL;
2695 result = PyDict_GetItem(modules, nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002696 if (result != NULL) {
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002697 Py_DECREF(nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002698 Py_INCREF(result);
2699 return result;
2700 }
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002701 PyErr_Clear();
Benjamin Peterson3e4f0552008-09-02 00:31:15 +00002702#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002703 /* check the import lock
2704 * me might be -1 but I ignore the error here, the lock function
2705 * takes care of the problem */
2706 me = PyThread_get_thread_ident();
2707 if (import_lock_thread == -1 || import_lock_thread == me) {
2708 /* no thread or me is holding the lock */
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002709 result = PyImport_Import(nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002710 }
2711 else {
2712 PyErr_Format(PyExc_ImportError,
Victor Stinnerc24c8102011-03-13 22:38:06 -04002713 "Failed to import %R because the import lock"
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002714 "is held by another thread.",
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002715 nameobj);
2716 result = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002717 }
Benjamin Peterson3e4f0552008-09-02 00:31:15 +00002718#else
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002719 result = PyImport_Import(nameobj);
Benjamin Peterson3e4f0552008-09-02 00:31:15 +00002720#endif
Victor Stinner2e5f11a2011-03-13 21:57:27 -04002721 Py_DECREF(nameobj);
2722 return result;
Christian Heimes072c0f12008-01-03 23:01:04 +00002723}
2724
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002725/* Forward declarations for helper routines */
Victor Stinner974389d2011-03-15 09:33:57 +01002726static PyObject *get_parent(PyObject *globals,
2727 PyObject **p_name,
2728 int level);
Tim Petersdbd9ba62000-07-09 03:09:57 +00002729static PyObject *load_next(PyObject *mod, PyObject *altmod,
Victor Stinner974389d2011-03-15 09:33:57 +01002730 PyObject *inputname, PyObject **p_outputname,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002731 Py_UCS4 *buf, Py_ssize_t *p_buflen,
Victor Stinner974389d2011-03-15 09:33:57 +01002732 Py_ssize_t bufsize);
2733static int mark_miss(PyObject *name);
Tim Petersdbd9ba62000-07-09 03:09:57 +00002734static int ensure_fromlist(PyObject *mod, PyObject *fromlist,
Victor Stinner974389d2011-03-15 09:33:57 +01002735 PyObject *buf, int recursive);
2736static PyObject * import_submodule(PyObject *mod, PyObject *name,
2737 PyObject *fullname);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002738
2739/* The Magnum Opus of dotted-name import :-) */
2740
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002741static PyObject *
Victor Stinner974389d2011-03-15 09:33:57 +01002742import_module_level(PyObject *name, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002743 PyObject *fromlist, int level)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00002744{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002745 Py_UCS4 buf[MAXPATHLEN+1];
Victor Stinner974389d2011-03-15 09:33:57 +01002746 Py_ssize_t buflen;
2747 Py_ssize_t bufsize = MAXPATHLEN+1;
2748 PyObject *parent, *head, *next, *tail, *inputname, *outputname;
2749 PyObject *parent_name, *ensure_name;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002750 Py_ssize_t sep, altsep;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002751
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002752 if (PyUnicode_READY(name))
2753 return NULL;
Victor Stinner974389d2011-03-15 09:33:57 +01002754
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002755 sep = PyUnicode_FindChar(name, SEP, 0, PyUnicode_GET_LENGTH(name), 1);
2756 if (sep == -2)
2757 return NULL;
Victor Stinner974389d2011-03-15 09:33:57 +01002758#ifdef ALTSEP
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002759 altsep = PyUnicode_FindChar(name, ALTSEP, 0, PyUnicode_GET_LENGTH(name), 1);
2760 if (altsep == -2)
2761 return NULL;
2762#else
2763 altsep = -1;
Christian Heimes454f37b2008-01-10 00:10:02 +00002764#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002765 if (sep != -1 || altsep != -1)
2766 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002767 PyErr_SetString(PyExc_ImportError,
2768 "Import by filename is not supported.");
2769 return NULL;
2770 }
Christian Heimes454f37b2008-01-10 00:10:02 +00002771
Victor Stinner974389d2011-03-15 09:33:57 +01002772 parent = get_parent(globals, &parent_name, level);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002773 if (parent == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002774 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002775 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002776
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002777 if (PyUnicode_READY(parent_name))
2778 return NULL;
2779 buflen = PyUnicode_GET_LENGTH(parent_name);
Georg Brandl4cb0de22011-09-28 21:49:49 +02002780 if (!PyUnicode_AsUCS4(parent_name, buf, Py_ARRAY_LENGTH(buf), 1)) {
Victor Stinner974389d2011-03-15 09:33:57 +01002781 Py_DECREF(parent_name);
2782 PyErr_SetString(PyExc_ValueError,
2783 "Module name too long");
2784 return NULL;
2785 }
Victor Stinner974389d2011-03-15 09:33:57 +01002786 Py_DECREF(parent_name);
2787
2788 head = load_next(parent, level < 0 ? Py_None : parent, name, &outputname,
2789 buf, &buflen, bufsize);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002790 if (head == NULL)
2791 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002792
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002793 tail = head;
2794 Py_INCREF(tail);
Victor Stinner974389d2011-03-15 09:33:57 +01002795
2796 if (outputname != NULL) {
2797 while (1) {
2798 inputname = outputname;
2799 next = load_next(tail, tail, inputname, &outputname,
2800 buf, &buflen, bufsize);
2801 Py_DECREF(tail);
2802 Py_DECREF(inputname);
2803 if (next == NULL) {
2804 Py_DECREF(head);
2805 return NULL;
2806 }
2807 tail = next;
2808
2809 if (outputname == NULL) {
2810 break;
2811 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002812 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002813 }
2814 if (tail == Py_None) {
2815 /* If tail is Py_None, both get_parent and load_next found
2816 an empty module name: someone called __import__("") or
2817 doctored faulty bytecode */
2818 Py_DECREF(tail);
2819 Py_DECREF(head);
Victor Stinner974389d2011-03-15 09:33:57 +01002820 PyErr_SetString(PyExc_ValueError, "Empty module name");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002821 return NULL;
2822 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002823
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002824 if (fromlist != NULL) {
2825 if (fromlist == Py_None || !PyObject_IsTrue(fromlist))
2826 fromlist = NULL;
2827 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002828
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002829 if (fromlist == NULL) {
2830 Py_DECREF(tail);
2831 return head;
2832 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002833
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002834 Py_DECREF(head);
Victor Stinner974389d2011-03-15 09:33:57 +01002835
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002836 ensure_name = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
2837 buf, Py_UCS4_strlen(buf));
Victor Stinner974389d2011-03-15 09:33:57 +01002838 if (ensure_name == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002839 Py_DECREF(tail);
2840 return NULL;
2841 }
Victor Stinner974389d2011-03-15 09:33:57 +01002842 if (!ensure_fromlist(tail, fromlist, ensure_name, 0)) {
2843 Py_DECREF(tail);
2844 Py_DECREF(ensure_name);
2845 return NULL;
2846 }
2847 Py_DECREF(ensure_name);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002848
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002849 return tail;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002850}
2851
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002852PyObject *
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002853PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
2854 PyObject *locals, PyObject *fromlist,
2855 int level)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002856{
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002857 PyObject *mod;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002858 _PyImport_AcquireLock();
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002859 mod = import_module_level(name, globals, locals, fromlist, level);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002860 if (_PyImport_ReleaseLock() < 0) {
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002861 Py_XDECREF(mod);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002862 PyErr_SetString(PyExc_RuntimeError,
2863 "not holding the import lock");
2864 return NULL;
2865 }
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002866 return mod;
Guido van Rossum75acc9c1998-03-03 22:26:50 +00002867}
2868
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002869PyObject *
Benjamin Peterson04778a82011-05-25 09:29:00 -05002870PyImport_ImportModuleLevel(const char *name, PyObject *globals, PyObject *locals,
Victor Stinnerfe93faf2011-03-14 15:54:52 -04002871 PyObject *fromlist, int level)
2872{
2873 PyObject *nameobj, *mod;
2874 nameobj = PyUnicode_FromString(name);
2875 if (nameobj == NULL)
2876 return NULL;
2877 mod = PyImport_ImportModuleLevelObject(nameobj, globals, locals,
2878 fromlist, level);
2879 Py_DECREF(nameobj);
2880 return mod;
2881}
2882
2883
Fred Drake87590902004-05-28 20:21:36 +00002884/* Return the package that an import is being performed in. If globals comes
2885 from the module foo.bar.bat (not itself a package), this returns the
2886 sys.modules entry for foo.bar. If globals is from a package's __init__.py,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002887 the package's entry in sys.modules is returned, as a borrowed reference.
Fred Drake87590902004-05-28 20:21:36 +00002888
Victor Stinner974389d2011-03-15 09:33:57 +01002889 The name of the returned package is returned in *p_name.
Fred Drake87590902004-05-28 20:21:36 +00002890
2891 If globals doesn't come from a package or a module in a package, or a
2892 corresponding entry is not found in sys.modules, Py_None is returned.
2893*/
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002894static PyObject *
Victor Stinner9599de52011-03-13 22:38:38 -04002895get_parent(PyObject *globals, PyObject **p_name, int level)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002896{
Victor Stinner974389d2011-03-15 09:33:57 +01002897 PyObject *nameobj;
2898
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002899 static PyObject *namestr = NULL;
2900 static PyObject *pathstr = NULL;
2901 static PyObject *pkgstr = NULL;
2902 PyObject *pkgname, *modname, *modpath, *modules, *parent;
2903 int orig_level = level;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002904
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002905 if (globals == NULL || !PyDict_Check(globals) || !level)
Victor Stinner974389d2011-03-15 09:33:57 +01002906 goto return_none;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002907
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002908 if (namestr == NULL) {
2909 namestr = PyUnicode_InternFromString("__name__");
2910 if (namestr == NULL)
2911 return NULL;
2912 }
2913 if (pathstr == NULL) {
2914 pathstr = PyUnicode_InternFromString("__path__");
2915 if (pathstr == NULL)
2916 return NULL;
2917 }
2918 if (pkgstr == NULL) {
2919 pkgstr = PyUnicode_InternFromString("__package__");
2920 if (pkgstr == NULL)
2921 return NULL;
2922 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002923
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002924 pkgname = PyDict_GetItem(globals, pkgstr);
Guido van Rossum17fc85f1997-09-06 18:52:03 +00002925
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002926 if ((pkgname != NULL) && (pkgname != Py_None)) {
2927 /* __package__ is set, so use it */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002928 if (!PyUnicode_Check(pkgname)) {
2929 PyErr_SetString(PyExc_ValueError,
2930 "__package__ set to non-string");
2931 return NULL;
2932 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002933 if (PyUnicode_GET_LENGTH(pkgname) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002934 if (level > 0) {
2935 PyErr_SetString(PyExc_ValueError,
2936 "Attempted relative import in non-package");
2937 return NULL;
2938 }
Victor Stinner974389d2011-03-15 09:33:57 +01002939 goto return_none;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002940 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002941 Py_INCREF(pkgname);
2942 nameobj = pkgname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002943 } else {
2944 /* __package__ not set, so figure it out and set it */
2945 modname = PyDict_GetItem(globals, namestr);
2946 if (modname == NULL || !PyUnicode_Check(modname))
Victor Stinner974389d2011-03-15 09:33:57 +01002947 goto return_none;
Brett Cannon9a5b25a2010-03-01 02:09:17 +00002948
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002949 modpath = PyDict_GetItem(globals, pathstr);
2950 if (modpath != NULL) {
2951 /* __path__ is set, so modname is already the package name */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002952 int error;
Alexandre Vassalottia85998a2008-05-03 18:24:43 +00002953
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002954 error = PyDict_SetItem(globals, pkgstr, modname);
2955 if (error) {
2956 PyErr_SetString(PyExc_ValueError,
2957 "Could not set __package__");
2958 return NULL;
2959 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002960 Py_INCREF(modname);
2961 nameobj = modname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002962 } else {
2963 /* Normal module, so work out the package name if any */
Victor Stinner974389d2011-03-15 09:33:57 +01002964 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002965 len = PyUnicode_FindChar(modname, '.',
2966 0, PyUnicode_GET_LENGTH(modname), -1);
2967 if (len == -2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002968 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002969 if (len < 0) {
2970 if (level > 0) {
2971 PyErr_SetString(PyExc_ValueError,
2972 "Attempted relative import in non-package");
2973 return NULL;
2974 }
2975 if (PyDict_SetItem(globals, pkgstr, Py_None)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002976 PyErr_SetString(PyExc_ValueError,
2977 "Could not set __package__");
2978 return NULL;
2979 }
Victor Stinner974389d2011-03-15 09:33:57 +01002980 goto return_none;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002981 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002982 pkgname = PyUnicode_Substring(modname, 0, len);
2983 if (pkgname == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002984 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002985 if (PyDict_SetItem(globals, pkgstr, pkgname)) {
2986 Py_DECREF(pkgname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002987 PyErr_SetString(PyExc_ValueError,
2988 "Could not set __package__");
2989 return NULL;
2990 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002991 nameobj = pkgname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002992 }
2993 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002994 if (level > 1) {
2995 Py_ssize_t dot, end = PyUnicode_GET_LENGTH(nameobj);
2996 PyObject *newname;
2997 while (--level > 0) {
2998 dot = PyUnicode_FindChar(nameobj, '.', 0, end, -1);
2999 if (dot == -2) {
3000 Py_DECREF(nameobj);
3001 return NULL;
3002 }
3003 if (dot < 0) {
3004 Py_DECREF(nameobj);
3005 PyErr_SetString(PyExc_ValueError,
3006 "Attempted relative import beyond "
3007 "toplevel package");
3008 return NULL;
3009 }
3010 end = dot;
3011 }
3012 newname = PyUnicode_Substring(nameobj, 0, end);
3013 Py_DECREF(nameobj);
3014 if (newname == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003015 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003016 nameobj = newname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003017 }
Victor Stinner974389d2011-03-15 09:33:57 +01003018
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003019 modules = PyImport_GetModuleDict();
Victor Stinner974389d2011-03-15 09:33:57 +01003020 parent = PyDict_GetItem(modules, nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003021 if (parent == NULL) {
Victor Stinner974389d2011-03-15 09:33:57 +01003022 int err;
3023
3024 if (orig_level >= 1) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003025 PyErr_Format(PyExc_SystemError,
Victor Stinner974389d2011-03-15 09:33:57 +01003026 "Parent module %R not loaded, "
3027 "cannot perform relative import", nameobj);
3028 Py_DECREF(nameobj);
3029 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003030 }
Victor Stinner974389d2011-03-15 09:33:57 +01003031
3032 err = PyErr_WarnFormat(
3033 PyExc_RuntimeWarning, 1,
3034 "Parent module %R not found while handling absolute import",
3035 nameobj);
3036 Py_DECREF(nameobj);
3037 if (err)
3038 return NULL;
3039
3040 goto return_none;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003041 }
Victor Stinner974389d2011-03-15 09:33:57 +01003042 *p_name = nameobj;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003043 return parent;
3044 /* We expect, but can't guarantee, if parent != None, that:
Victor Stinner974389d2011-03-15 09:33:57 +01003045 - parent.__name__ == name
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003046 - parent.__dict__ is globals
3047 If this is violated... Who cares? */
Victor Stinner974389d2011-03-15 09:33:57 +01003048
3049return_none:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003050 nameobj = PyUnicode_New(0, 0);
Victor Stinner974389d2011-03-15 09:33:57 +01003051 if (nameobj == NULL)
3052 return NULL;
3053 *p_name = nameobj;
3054 return Py_None;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003055}
3056
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003057/* altmod is either None or same as mod */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003058static PyObject *
Victor Stinner974389d2011-03-15 09:33:57 +01003059load_next(PyObject *mod, PyObject *altmod,
3060 PyObject *inputname, PyObject **p_outputname,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003061 Py_UCS4 *buf, Py_ssize_t *p_buflen, Py_ssize_t bufsize)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003062{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003063 Py_UCS4 *dot;
Victor Stinner974389d2011-03-15 09:33:57 +01003064 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003065 Py_UCS4 *p;
Victor Stinner974389d2011-03-15 09:33:57 +01003066 PyObject *fullname, *name, *result, *mark_name;
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003067 Py_UCS4 *nameuni;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003068
Victor Stinner974389d2011-03-15 09:33:57 +01003069 *p_outputname = NULL;
3070
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003071 if (PyUnicode_GET_LENGTH(inputname) == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003072 /* completely empty module name should only happen in
3073 'from . import' (or '__import__("")')*/
3074 Py_INCREF(mod);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003075 return mod;
3076 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00003077
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003078 nameuni = PyUnicode_AsUCS4Copy(inputname);
Victor Stinner974389d2011-03-15 09:33:57 +01003079 if (nameuni == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003080 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003081
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003082 dot = Py_UCS4_strchr(nameuni, '.');
Victor Stinner974389d2011-03-15 09:33:57 +01003083 if (dot != NULL) {
3084 len = dot - nameuni;
3085 if (len == 0) {
3086 PyErr_SetString(PyExc_ValueError,
3087 "Empty module name");
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003088 goto error;
Victor Stinner974389d2011-03-15 09:33:57 +01003089 }
3090 }
3091 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003092 len = PyUnicode_GET_LENGTH(inputname);
Victor Stinner974389d2011-03-15 09:33:57 +01003093
3094 if (*p_buflen+len+1 >= bufsize) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003095 PyErr_SetString(PyExc_ValueError,
3096 "Module name too long");
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003097 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003098 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003099
Victor Stinner974389d2011-03-15 09:33:57 +01003100 p = buf + *p_buflen;
3101 if (p != buf) {
3102 *p++ = '.';
3103 *p_buflen += 1;
3104 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003105 Py_UCS4_strncpy(p, nameuni, len);
Victor Stinner974389d2011-03-15 09:33:57 +01003106 p[len] = '\0';
3107 *p_buflen += len;
3108
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003109 fullname = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3110 buf, *p_buflen);
Victor Stinner974389d2011-03-15 09:33:57 +01003111 if (fullname == NULL)
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003112 goto error;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003113 name = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3114 p, len);
Victor Stinner974389d2011-03-15 09:33:57 +01003115 if (name == NULL) {
3116 Py_DECREF(fullname);
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003117 goto error;
Victor Stinner974389d2011-03-15 09:33:57 +01003118 }
3119 result = import_submodule(mod, name, fullname);
3120 Py_DECREF(fullname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003121 if (result == Py_None && altmod != mod) {
3122 Py_DECREF(result);
3123 /* Here, altmod must be None and mod must not be None */
Victor Stinner974389d2011-03-15 09:33:57 +01003124 result = import_submodule(altmod, name, name);
3125 Py_DECREF(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003126 if (result != NULL && result != Py_None) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003127 mark_name = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3128 buf, *p_buflen);
Victor Stinner974389d2011-03-15 09:33:57 +01003129 if (mark_name == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003130 Py_DECREF(result);
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003131 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003132 }
Victor Stinner974389d2011-03-15 09:33:57 +01003133 if (mark_miss(mark_name) != 0) {
3134 Py_DECREF(result);
3135 Py_DECREF(mark_name);
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003136 goto error;
Victor Stinner974389d2011-03-15 09:33:57 +01003137 }
3138 Py_DECREF(mark_name);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003139 Py_UCS4_strncpy(buf, nameuni, len);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003140 buf[len] = '\0';
3141 *p_buflen = len;
3142 }
3143 }
Victor Stinner974389d2011-03-15 09:33:57 +01003144 else
3145 Py_DECREF(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003146 if (result == NULL)
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003147 goto error;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003148
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003149 if (result == Py_None) {
3150 Py_DECREF(result);
3151 PyErr_Format(PyExc_ImportError,
Victor Stinner974389d2011-03-15 09:33:57 +01003152 "No module named %R", inputname);
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003153 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003154 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003155
Victor Stinner974389d2011-03-15 09:33:57 +01003156 if (dot != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003157 *p_outputname = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3158 dot+1, Py_UCS4_strlen(dot+1));
Victor Stinner974389d2011-03-15 09:33:57 +01003159 if (*p_outputname == NULL) {
3160 Py_DECREF(result);
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003161 goto error;
Victor Stinner974389d2011-03-15 09:33:57 +01003162 }
3163 }
3164
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003165 PyMem_Free(nameuni);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003166 return result;
Martin v. Löwis0b1d3482011-10-01 16:35:40 +02003167
3168error:
3169 PyMem_Free(nameuni);
3170 return NULL;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003171}
3172
3173static int
Victor Stinner974389d2011-03-15 09:33:57 +01003174mark_miss(PyObject *name)
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00003175{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003176 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner974389d2011-03-15 09:33:57 +01003177 return PyDict_SetItem(modules, name, Py_None);
Guido van Rossumf5f5fdb1997-09-06 20:29:52 +00003178}
3179
3180static int
Victor Stinner974389d2011-03-15 09:33:57 +01003181ensure_fromlist(PyObject *mod, PyObject *fromlist, PyObject *name,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003182 int recursive)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003183{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003184 int i;
Victor Stinner974389d2011-03-15 09:33:57 +01003185 PyObject *fullname;
3186 Py_ssize_t fromlist_len;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003187
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003188 if (!_PyObject_HasAttrId(mod, &PyId___path__))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003189 return 1;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003190
Victor Stinner974389d2011-03-15 09:33:57 +01003191 fromlist_len = PySequence_Size(fromlist);
3192
3193 for (i = 0; i < fromlist_len; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003194 PyObject *item = PySequence_GetItem(fromlist, i);
3195 int hasit;
Victor Stinner974389d2011-03-15 09:33:57 +01003196 if (item == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003197 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003198 if (!PyUnicode_Check(item)) {
3199 PyErr_SetString(PyExc_TypeError,
3200 "Item in ``from list'' not a string");
3201 Py_DECREF(item);
3202 return 0;
3203 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003204 if (PyUnicode_READ_CHAR(item, 0) == '*') {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003205 PyObject *all;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02003206 _Py_IDENTIFIER(__all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003207 Py_DECREF(item);
3208 /* See if the package defines __all__ */
3209 if (recursive)
3210 continue; /* Avoid endless recursion */
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003211 all = _PyObject_GetAttrId(mod, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003212 if (all == NULL)
3213 PyErr_Clear();
3214 else {
Victor Stinner974389d2011-03-15 09:33:57 +01003215 int ret = ensure_fromlist(mod, all, name, 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003216 Py_DECREF(all);
3217 if (!ret)
3218 return 0;
3219 }
3220 continue;
3221 }
3222 hasit = PyObject_HasAttr(mod, item);
3223 if (!hasit) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003224 PyObject *submod;
Victor Stinner974389d2011-03-15 09:33:57 +01003225 fullname = PyUnicode_FromFormat("%U.%U", name, item);
3226 if (fullname != NULL) {
3227 submod = import_submodule(mod, item, fullname);
3228 Py_DECREF(fullname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003229 }
Victor Stinner974389d2011-03-15 09:33:57 +01003230 else
3231 submod = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003232 Py_XDECREF(submod);
3233 if (submod == NULL) {
3234 Py_DECREF(item);
3235 return 0;
3236 }
3237 }
3238 Py_DECREF(item);
3239 }
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003240
Victor Stinner974389d2011-03-15 09:33:57 +01003241 return 1;
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003242}
3243
Neil Schemenauer00b09662003-06-16 21:03:07 +00003244static int
Victor Stinner974389d2011-03-15 09:33:57 +01003245add_submodule(PyObject *mod, PyObject *submod, PyObject *fullname,
3246 PyObject *subname, PyObject *modules)
Neil Schemenauer00b09662003-06-16 21:03:07 +00003247{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003248 if (mod == Py_None)
3249 return 1;
3250 /* Irrespective of the success of this load, make a
3251 reference to it in the parent package module. A copy gets
3252 saved in the modules dictionary under the full name, so get a
3253 reference from there, if need be. (The exception is when the
3254 load failed with a SyntaxError -- then there's no trace in
3255 sys.modules. In that case, of course, do nothing extra.) */
3256 if (submod == NULL) {
Victor Stinner974389d2011-03-15 09:33:57 +01003257 submod = PyDict_GetItem(modules, fullname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003258 if (submod == NULL)
3259 return 1;
3260 }
3261 if (PyModule_Check(mod)) {
3262 /* We can't use setattr here since it can give a
3263 * spurious warning if the submodule name shadows a
3264 * builtin name */
3265 PyObject *dict = PyModule_GetDict(mod);
3266 if (!dict)
3267 return 0;
Victor Stinner974389d2011-03-15 09:33:57 +01003268 if (PyDict_SetItem(dict, subname, submod) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003269 return 0;
3270 }
3271 else {
Victor Stinner974389d2011-03-15 09:33:57 +01003272 if (PyObject_SetAttr(mod, subname, submod) < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003273 return 0;
3274 }
3275 return 1;
Neil Schemenauer00b09662003-06-16 21:03:07 +00003276}
3277
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003278static PyObject *
Victor Stinner974389d2011-03-15 09:33:57 +01003279import_submodule(PyObject *mod, PyObject *subname, PyObject *fullname)
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003280{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003281 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner533d7832011-03-14 13:22:54 -04003282 PyObject *m = NULL, *bufobj, *path_list, *loader;
Victor Stinner9599de52011-03-13 22:38:38 -04003283 struct filedescr *fdp;
3284 FILE *fp;
Guido van Rossum74e6a111994-08-29 12:54:38 +00003285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003286 /* Require:
3287 if mod == None: subname == fullname
3288 else: mod.__name__ + "." + subname == fullname
3289 */
Guido van Rossum17fc85f1997-09-06 18:52:03 +00003290
Victor Stinner974389d2011-03-15 09:33:57 +01003291 if ((m = PyDict_GetItem(modules, fullname)) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003292 Py_INCREF(m);
Victor Stinner9599de52011-03-13 22:38:38 -04003293 return m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003294 }
Victor Stinner9599de52011-03-13 22:38:38 -04003295
3296 if (mod == Py_None)
Victor Stinner533d7832011-03-14 13:22:54 -04003297 path_list = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003298 else {
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003299 path_list = _PyObject_GetAttrId(mod, &PyId___path__);
Victor Stinner533d7832011-03-14 13:22:54 -04003300 if (path_list == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003301 PyErr_Clear();
3302 Py_INCREF(Py_None);
3303 return Py_None;
3304 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003305 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003306
Victor Stinner533d7832011-03-14 13:22:54 -04003307 fdp = find_module(fullname, subname, path_list,
Victor Stinner2fd76e42011-03-14 15:19:39 -04003308 &bufobj, &fp, &loader);
Victor Stinner533d7832011-03-14 13:22:54 -04003309 Py_XDECREF(path_list);
Victor Stinner9599de52011-03-13 22:38:38 -04003310 if (fdp == NULL) {
3311 if (!PyErr_ExceptionMatches(PyExc_ImportError))
3312 return NULL;
3313 PyErr_Clear();
3314 Py_INCREF(Py_None);
3315 return Py_None;
3316 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04003317 m = load_module(fullname, fp, bufobj, fdp->type, loader);
3318 Py_XDECREF(bufobj);
Victor Stinner9599de52011-03-13 22:38:38 -04003319 Py_XDECREF(loader);
3320 if (fp)
3321 fclose(fp);
3322 if (m == NULL)
3323 return NULL;
3324 if (!add_submodule(mod, m, fullname, subname, modules)) {
3325 Py_XDECREF(m);
3326 return NULL;
3327 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 return m;
Guido van Rossum74e6a111994-08-29 12:54:38 +00003329}
3330
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003331
3332/* Re-import a module of any kind and return its module object, WITH
3333 INCREMENTED REFERENCE COUNT */
3334
Guido van Rossum79f25d91997-04-29 20:08:16 +00003335PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003336PyImport_ReloadModule(PyObject *m)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003337{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003338 PyInterpreterState *interp = PyThreadState_Get()->interp;
3339 PyObject *modules_reloading = interp->modules_reloading;
3340 PyObject *modules = PyImport_GetModuleDict();
Victor Stinner533d7832011-03-14 13:22:54 -04003341 PyObject *path_list = NULL, *loader = NULL, *existing_m = NULL;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003342 PyObject *nameobj, *bufobj, *subnameobj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003343 Py_UCS4 *name = NULL, *subname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003344 struct filedescr *fdp;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003345 FILE *fp = NULL;
3346 PyObject *newm = NULL;
Brett Cannon9a5b25a2010-03-01 02:09:17 +00003347
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003348 if (modules_reloading == NULL) {
3349 Py_FatalError("PyImport_ReloadModule: "
3350 "no modules_reloading dictionary!");
3351 return NULL;
3352 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003354 if (m == NULL || !PyModule_Check(m)) {
3355 PyErr_SetString(PyExc_TypeError,
3356 "reload() argument must be module");
3357 return NULL;
3358 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003359 nameobj = PyModule_GetNameObject(m);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003360 if (nameobj == NULL || PyUnicode_READY(nameobj) == -1)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003361 return NULL;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003362 if (m != PyDict_GetItem(modules, nameobj)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003363 PyErr_Format(PyExc_ImportError,
Victor Stinner98dbba52011-03-14 15:15:47 -04003364 "reload(): module %R not in sys.modules",
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003365 nameobj);
3366 Py_DECREF(nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003367 return NULL;
3368 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003369 existing_m = PyDict_GetItem(modules_reloading, nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003370 if (existing_m != NULL) {
3371 /* Due to a recursive reload, this module is already
3372 being reloaded. */
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003373 Py_DECREF(nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003374 Py_INCREF(existing_m);
3375 return existing_m;
3376 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003377 if (PyDict_SetItem(modules_reloading, nameobj, m) < 0) {
3378 Py_DECREF(nameobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003379 return NULL;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003380 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00003381
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003382 name = PyUnicode_AsUCS4Copy(nameobj);
3383 if (!name) {
3384 Py_DECREF(nameobj);
3385 return NULL;
3386 }
3387 subname = Py_UCS4_strrchr(name, '.');
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003388 if (subname == NULL) {
3389 Py_INCREF(nameobj);
3390 subnameobj = nameobj;
3391 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003392 else {
3393 PyObject *parentname, *parent;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003394 Py_ssize_t len;
3395 len = subname - name;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003396 parentname = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3397 name, len);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003398 if (parentname == NULL) {
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003399 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003400 }
3401 parent = PyDict_GetItem(modules, parentname);
3402 if (parent == NULL) {
3403 PyErr_Format(PyExc_ImportError,
Victor Stinner98dbba52011-03-14 15:15:47 -04003404 "reload(): parent %R not in sys.modules",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003405 parentname);
3406 Py_DECREF(parentname);
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003407 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003408 }
3409 Py_DECREF(parentname);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003410 path_list = _PyObject_GetAttrId(parent, &PyId___path__);
Victor Stinner533d7832011-03-14 13:22:54 -04003411 if (path_list == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003412 PyErr_Clear();
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003413 subname++;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003414 len = PyUnicode_GET_LENGTH(nameobj) - (len + 1);
3415 subnameobj = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND,
3416 subname, len);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003417 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003418 if (subnameobj == NULL)
3419 goto error;
Victor Stinner533d7832011-03-14 13:22:54 -04003420 fdp = find_module(nameobj, subnameobj, path_list,
Victor Stinner2fd76e42011-03-14 15:19:39 -04003421 &bufobj, &fp, &loader);
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003422 Py_DECREF(subnameobj);
Victor Stinner533d7832011-03-14 13:22:54 -04003423 Py_XDECREF(path_list);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00003424
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003425 if (fdp == NULL) {
3426 Py_XDECREF(loader);
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003427 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003428 }
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00003429
Victor Stinner2fd76e42011-03-14 15:19:39 -04003430 newm = load_module(nameobj, fp, bufobj, fdp->type, loader);
3431 Py_XDECREF(bufobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003432 Py_XDECREF(loader);
Phillip J. Eby7ec642a2004-09-23 04:37:36 +00003433
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003434 if (fp)
3435 fclose(fp);
3436 if (newm == NULL) {
3437 /* load_module probably removed name from modules because of
3438 * the error. Put back the original module object. We're
3439 * going to return NULL in this case regardless of whether
3440 * replacing name succeeds, so the return value is ignored.
3441 */
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003442 PyDict_SetItem(modules, nameobj, m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003443 }
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003444
3445error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003446 imp_modules_reloading_clear();
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003447 Py_DECREF(nameobj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003448 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003449 return newm;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003450}
3451
3452
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003453/* Higher-level import emulator which emulates the "import" statement
3454 more accurately -- it invokes the __import__() function from the
3455 builtins of the current globals. This means that the import is
3456 done using whatever import hooks are installed in the current
Brett Cannonbc2eff32010-09-19 21:39:02 +00003457 environment.
Guido van Rossum6058eb41998-12-21 19:51:00 +00003458 A dummy list ["__doc__"] is passed as the 4th argument so that
Neal Norwitz80e7f272007-08-26 06:45:23 +00003459 e.g. PyImport_Import(PyUnicode_FromString("win32com.client.gencache"))
Guido van Rossum6058eb41998-12-21 19:51:00 +00003460 will return <module "gencache"> instead of <module "win32com">. */
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003461
3462PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003463PyImport_Import(PyObject *module_name)
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003464{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003465 static PyObject *silly_list = NULL;
3466 static PyObject *builtins_str = NULL;
3467 static PyObject *import_str = NULL;
3468 PyObject *globals = NULL;
3469 PyObject *import = NULL;
3470 PyObject *builtins = NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00003471 PyObject *modules = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003472 PyObject *r = NULL;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003474 /* Initialize constant string objects */
3475 if (silly_list == NULL) {
3476 import_str = PyUnicode_InternFromString("__import__");
3477 if (import_str == NULL)
3478 return NULL;
3479 builtins_str = PyUnicode_InternFromString("__builtins__");
3480 if (builtins_str == NULL)
3481 return NULL;
Brett Cannonbc2eff32010-09-19 21:39:02 +00003482 silly_list = PyList_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003483 if (silly_list == NULL)
3484 return NULL;
3485 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003486
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003487 /* Get the builtins from current globals */
3488 globals = PyEval_GetGlobals();
3489 if (globals != NULL) {
3490 Py_INCREF(globals);
3491 builtins = PyObject_GetItem(globals, builtins_str);
3492 if (builtins == NULL)
3493 goto err;
3494 }
3495 else {
3496 /* No globals -- use standard builtins, and fake globals */
3497 builtins = PyImport_ImportModuleLevel("builtins",
3498 NULL, NULL, NULL, 0);
3499 if (builtins == NULL)
3500 return NULL;
3501 globals = Py_BuildValue("{OO}", builtins_str, builtins);
3502 if (globals == NULL)
3503 goto err;
3504 }
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003505
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003506 /* Get the __import__ function from the builtins */
3507 if (PyDict_Check(builtins)) {
3508 import = PyObject_GetItem(builtins, import_str);
3509 if (import == NULL)
3510 PyErr_SetObject(PyExc_KeyError, import_str);
3511 }
3512 else
3513 import = PyObject_GetAttr(builtins, import_str);
3514 if (import == NULL)
3515 goto err;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003516
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003517 /* Call the __import__ function with the proper argument list
Brett Cannonbc2eff32010-09-19 21:39:02 +00003518 Always use absolute import here.
3519 Calling for side-effect of import. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003520 r = PyObject_CallFunction(import, "OOOOi", module_name, globals,
3521 globals, silly_list, 0, NULL);
Brett Cannonbc2eff32010-09-19 21:39:02 +00003522 if (r == NULL)
3523 goto err;
3524 Py_DECREF(r);
3525
3526 modules = PyImport_GetModuleDict();
3527 r = PyDict_GetItem(modules, module_name);
3528 if (r != NULL)
3529 Py_INCREF(r);
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003530
3531 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003532 Py_XDECREF(globals);
3533 Py_XDECREF(builtins);
3534 Py_XDECREF(import);
Tim Peters50d8d372001-02-28 05:34:27 +00003535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003536 return r;
Guido van Rossumd47a0a81997-08-14 20:11:26 +00003537}
3538
3539
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003540/* Module 'imp' provides Python access to the primitives used for
3541 importing modules.
3542*/
3543
Guido van Rossum79f25d91997-04-29 20:08:16 +00003544static PyObject *
Barry Warsaw28a691b2010-04-17 00:19:56 +00003545imp_make_magic(long magic)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003546{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003547 char buf[4];
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003548
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003549 buf[0] = (char) ((magic >> 0) & 0xff);
3550 buf[1] = (char) ((magic >> 8) & 0xff);
3551 buf[2] = (char) ((magic >> 16) & 0xff);
3552 buf[3] = (char) ((magic >> 24) & 0xff);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003553
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003554 return PyBytes_FromStringAndSize(buf, 4);
Victor Stinner3e2b7172010-11-09 09:32:19 +00003555}
Barry Warsaw28a691b2010-04-17 00:19:56 +00003556
3557static PyObject *
3558imp_get_magic(PyObject *self, PyObject *noargs)
3559{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003560 return imp_make_magic(pyc_magic);
Barry Warsaw28a691b2010-04-17 00:19:56 +00003561}
3562
3563static PyObject *
3564imp_get_tag(PyObject *self, PyObject *noargs)
3565{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003566 return PyUnicode_FromString(pyc_tag);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003567}
3568
Guido van Rossum79f25d91997-04-29 20:08:16 +00003569static PyObject *
Neal Norwitz08ea61a2003-02-17 18:18:00 +00003570imp_get_suffixes(PyObject *self, PyObject *noargs)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003571{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003572 PyObject *list;
3573 struct filedescr *fdp;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003574
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003575 list = PyList_New(0);
3576 if (list == NULL)
3577 return NULL;
3578 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
3579 PyObject *item = Py_BuildValue("ssi",
3580 fdp->suffix, fdp->mode, fdp->type);
3581 if (item == NULL) {
3582 Py_DECREF(list);
3583 return NULL;
3584 }
3585 if (PyList_Append(list, item) < 0) {
3586 Py_DECREF(list);
3587 Py_DECREF(item);
3588 return NULL;
3589 }
3590 Py_DECREF(item);
3591 }
3592 return list;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003593}
3594
Guido van Rossum79f25d91997-04-29 20:08:16 +00003595static PyObject *
Victor Stinner533d7832011-03-14 13:22:54 -04003596call_find_module(PyObject *name, PyObject *path_list)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003597{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003598 extern int fclose(FILE *);
3599 PyObject *fob, *ret;
3600 PyObject *pathobj;
3601 struct filedescr *fdp;
Victor Stinner7d8b77c2011-03-12 08:45:02 -05003602 FILE *fp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003603 int fd = -1;
3604 char *found_encoding = NULL;
3605 char *encoding = NULL;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003606
Victor Stinner533d7832011-03-14 13:22:54 -04003607 if (path_list == Py_None)
3608 path_list = NULL;
3609 fdp = find_module(NULL, name, path_list,
Victor Stinner2fd76e42011-03-14 15:19:39 -04003610 &pathobj, &fp, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003611 if (fdp == NULL)
3612 return NULL;
3613 if (fp != NULL) {
3614 fd = fileno(fp);
3615 if (fd != -1)
3616 fd = dup(fd);
3617 fclose(fp);
Victor Stinnerd417d012011-06-20 15:16:55 +02003618 if (fd == -1) {
3619 PyErr_SetFromErrno(PyExc_OSError);
3620 return NULL;
3621 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003622 fp = NULL;
3623 }
3624 if (fd != -1) {
3625 if (strchr(fdp->mode, 'b') == NULL) {
Victor Stinnerfe7c5b52011-04-05 01:48:03 +02003626 /* PyTokenizer_FindEncodingFilename() returns PyMem_MALLOC'ed
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003627 memory. */
Victor Stinnerfe7c5b52011-04-05 01:48:03 +02003628 found_encoding = PyTokenizer_FindEncodingFilename(fd, pathobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003629 lseek(fd, 0, 0); /* Reset position */
Victor Stinner2fd76e42011-03-14 15:19:39 -04003630 if (found_encoding == NULL && PyErr_Occurred()) {
3631 Py_XDECREF(pathobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003632 return NULL;
Victor Stinner2fd76e42011-03-14 15:19:39 -04003633 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003634 encoding = (found_encoding != NULL) ? found_encoding :
3635 (char*)PyUnicode_GetDefaultEncoding();
3636 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04003637 fob = PyFile_FromFd(fd, NULL, fdp->mode, -1,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003638 (char*)encoding, NULL, NULL, 1);
3639 if (fob == NULL) {
Victor Stinner2fd76e42011-03-14 15:19:39 -04003640 Py_XDECREF(pathobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003641 close(fd);
3642 PyMem_FREE(found_encoding);
3643 return NULL;
3644 }
3645 }
3646 else {
3647 fob = Py_None;
3648 Py_INCREF(fob);
3649 }
Victor Stinner2fd76e42011-03-14 15:19:39 -04003650 if (pathobj == NULL) {
3651 Py_INCREF(Py_None);
3652 pathobj = Py_None;
3653 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003654 ret = Py_BuildValue("NN(ssi)",
3655 fob, pathobj, fdp->suffix, fdp->mode, fdp->type);
3656 PyMem_FREE(found_encoding);
Brett Cannon3bb42d92007-10-20 03:43:15 +00003657
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003658 return ret;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003659}
3660
Guido van Rossum79f25d91997-04-29 20:08:16 +00003661static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003662imp_find_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003663{
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003664 PyObject *name, *path_list = NULL;
3665 if (!PyArg_ParseTuple(args, "U|O:find_module",
3666 &name, &path_list))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003667 return NULL;
Victor Stinnerad3c03b2011-03-14 09:21:33 -04003668 return call_find_module(name, path_list);
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003669}
3670
3671static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003672imp_init_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003673{
Victor Stinner95872862011-03-07 18:20:56 +01003674 PyObject *name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003675 int ret;
3676 PyObject *m;
Victor Stinner95872862011-03-07 18:20:56 +01003677 if (!PyArg_ParseTuple(args, "U:init_builtin", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003678 return NULL;
3679 ret = init_builtin(name);
3680 if (ret < 0)
3681 return NULL;
3682 if (ret == 0) {
3683 Py_INCREF(Py_None);
3684 return Py_None;
3685 }
Victor Stinner95872862011-03-07 18:20:56 +01003686 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003687 Py_XINCREF(m);
3688 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003689}
3690
Guido van Rossum79f25d91997-04-29 20:08:16 +00003691static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003692imp_init_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003693{
Victor Stinner53dc7352011-03-20 01:50:21 +01003694 PyObject *name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003695 int ret;
3696 PyObject *m;
Victor Stinner53dc7352011-03-20 01:50:21 +01003697 if (!PyArg_ParseTuple(args, "U:init_frozen", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003698 return NULL;
Victor Stinner53dc7352011-03-20 01:50:21 +01003699 ret = PyImport_ImportFrozenModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003700 if (ret < 0)
3701 return NULL;
3702 if (ret == 0) {
3703 Py_INCREF(Py_None);
3704 return Py_None;
3705 }
Victor Stinner53dc7352011-03-20 01:50:21 +01003706 m = PyImport_AddModuleObject(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003707 Py_XINCREF(m);
3708 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003709}
3710
Guido van Rossum79f25d91997-04-29 20:08:16 +00003711static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003712imp_get_frozen_object(PyObject *self, PyObject *args)
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00003713{
Victor Stinner53dc7352011-03-20 01:50:21 +01003714 PyObject *name;
Jack Jansen95ffa231995-10-03 14:38:41 +00003715
Victor Stinner53dc7352011-03-20 01:50:21 +01003716 if (!PyArg_ParseTuple(args, "U:get_frozen_object", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003717 return NULL;
3718 return get_frozen_object(name);
Guido van Rossum6ec1efb1995-08-04 04:08:57 +00003719}
3720
Guido van Rossum79f25d91997-04-29 20:08:16 +00003721static PyObject *
Brett Cannon8d110132009-03-15 02:20:16 +00003722imp_is_frozen_package(PyObject *self, PyObject *args)
3723{
Victor Stinner53dc7352011-03-20 01:50:21 +01003724 PyObject *name;
Brett Cannon8d110132009-03-15 02:20:16 +00003725
Victor Stinner53dc7352011-03-20 01:50:21 +01003726 if (!PyArg_ParseTuple(args, "U:is_frozen_package", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003727 return NULL;
3728 return is_frozen_package(name);
Brett Cannon8d110132009-03-15 02:20:16 +00003729}
3730
3731static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003732imp_is_builtin(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003733{
Victor Stinner95872862011-03-07 18:20:56 +01003734 PyObject *name;
3735 if (!PyArg_ParseTuple(args, "U:is_builtin", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003736 return NULL;
3737 return PyLong_FromLong(is_builtin(name));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003738}
3739
Guido van Rossum79f25d91997-04-29 20:08:16 +00003740static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003741imp_is_frozen(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003742{
Victor Stinner53dc7352011-03-20 01:50:21 +01003743 PyObject *name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003744 struct _frozen *p;
Victor Stinner53dc7352011-03-20 01:50:21 +01003745 if (!PyArg_ParseTuple(args, "U:is_frozen", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003746 return NULL;
3747 p = find_frozen(name);
3748 return PyBool_FromLong((long) (p == NULL ? 0 : p->size));
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003749}
3750
3751static FILE *
Victor Stinner2f42ae52011-03-20 00:41:24 +01003752get_file(PyObject *pathname, PyObject *fob, char *mode)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003753{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003754 FILE *fp;
3755 if (mode[0] == 'U')
3756 mode = "r" PY_STDIOTEXTMODE;
3757 if (fob == NULL) {
Victor Stinner2f42ae52011-03-20 00:41:24 +01003758 fp = _Py_fopen(pathname, mode);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003759 }
3760 else {
3761 int fd = PyObject_AsFileDescriptor(fob);
3762 if (fd == -1)
3763 return NULL;
3764 if (!_PyVerify_fd(fd))
3765 goto error;
3766 /* the FILE struct gets a new fd, so that it can be closed
3767 * independently of the file descriptor given
3768 */
3769 fd = dup(fd);
3770 if (fd == -1)
3771 goto error;
3772 fp = fdopen(fd, mode);
3773 }
3774 if (fp)
3775 return fp;
Kristján Valur Jónssone1b04452009-03-31 17:43:39 +00003776error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003777 PyErr_SetFromErrno(PyExc_IOError);
3778 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003779}
3780
Guido van Rossum79f25d91997-04-29 20:08:16 +00003781static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003782imp_load_compiled(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003783{
Victor Stinner2f42ae52011-03-20 00:41:24 +01003784 PyObject *name, *pathname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003785 PyObject *fob = NULL;
3786 PyObject *m;
3787 FILE *fp;
Victor Stinner2f42ae52011-03-20 00:41:24 +01003788 if (!PyArg_ParseTuple(args, "UO&|O:load_compiled",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003789 &name,
Victor Stinner2f42ae52011-03-20 00:41:24 +01003790 PyUnicode_FSDecoder, &pathname,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003791 &fob))
3792 return NULL;
Victor Stinner2f42ae52011-03-20 00:41:24 +01003793 fp = get_file(pathname, fob, "rb");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003794 if (fp == NULL) {
Victor Stinnerebc00522010-12-03 17:06:43 +00003795 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003796 return NULL;
3797 }
Victor Stinner2f42ae52011-03-20 00:41:24 +01003798 m = load_compiled_module(name, pathname, fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003799 fclose(fp);
Victor Stinnerebc00522010-12-03 17:06:43 +00003800 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003801 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003802}
3803
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003804#ifdef HAVE_DYNAMIC_LOADING
3805
Guido van Rossum79f25d91997-04-29 20:08:16 +00003806static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003807imp_load_dynamic(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003808{
Victor Stinnerfefd70c2011-03-14 15:54:07 -04003809 PyObject *name, *pathname, *fob = NULL, *mod;
3810 FILE *fp;
3811
3812 if (!PyArg_ParseTuple(args, "UO&|O:load_dynamic",
3813 &name, PyUnicode_FSDecoder, &pathname, &fob))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003814 return NULL;
Victor Stinnerfefd70c2011-03-14 15:54:07 -04003815 if (fob != NULL) {
3816 fp = get_file(NULL, fob, "r");
Victor Stinnere9ddbf62011-03-22 10:46:35 +01003817 if (fp == NULL) {
3818 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003819 return NULL;
Victor Stinnere9ddbf62011-03-22 10:46:35 +01003820 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003821 }
Victor Stinnerfefd70c2011-03-14 15:54:07 -04003822 else
3823 fp = NULL;
3824 mod = _PyImport_LoadDynamicModule(name, pathname, fp);
Victor Stinnere9ddbf62011-03-22 10:46:35 +01003825 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003826 if (fp)
3827 fclose(fp);
Victor Stinnerfefd70c2011-03-14 15:54:07 -04003828 return mod;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003829}
3830
Guido van Rossum96a8fb71999-12-22 14:09:35 +00003831#endif /* HAVE_DYNAMIC_LOADING */
3832
Guido van Rossum79f25d91997-04-29 20:08:16 +00003833static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003834imp_load_source(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003835{
Victor Stinner2f42ae52011-03-20 00:41:24 +01003836 PyObject *name, *pathname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003837 PyObject *fob = NULL;
3838 PyObject *m;
3839 FILE *fp;
Victor Stinner2f42ae52011-03-20 00:41:24 +01003840 if (!PyArg_ParseTuple(args, "UO&|O:load_source",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003841 &name,
Victor Stinner2f42ae52011-03-20 00:41:24 +01003842 PyUnicode_FSDecoder, &pathname,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003843 &fob))
3844 return NULL;
Victor Stinner2f42ae52011-03-20 00:41:24 +01003845 fp = get_file(pathname, fob, "r");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003846 if (fp == NULL) {
Victor Stinnerebc00522010-12-03 17:06:43 +00003847 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003848 return NULL;
3849 }
Victor Stinner2f42ae52011-03-20 00:41:24 +01003850 m = load_source_module(name, pathname, fp);
Victor Stinnerebc00522010-12-03 17:06:43 +00003851 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003852 fclose(fp);
3853 return m;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003854}
3855
Guido van Rossum79f25d91997-04-29 20:08:16 +00003856static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003857imp_load_module(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003858{
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003859 PyObject *name, *fob, *pathname, *pathname_obj, *ret;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003860 char *suffix; /* Unused */
3861 char *mode;
3862 int type;
3863 FILE *fp;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003864
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003865 if (!PyArg_ParseTuple(args, "UOO(ssi):load_module",
3866 &name, &fob, &pathname_obj, &suffix, &mode, &type))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003867 return NULL;
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003868 if (pathname_obj != Py_None) {
3869 if (!PyUnicode_FSDecoder(pathname_obj, &pathname))
3870 return NULL;
3871 }
3872 else
3873 pathname = NULL;
3874
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003875 if (*mode) {
3876 /* Mode must start with 'r' or 'U' and must not contain '+'.
3877 Implicit in this test is the assumption that the mode
3878 may contain other modifiers like 'b' or 't'. */
Guido van Rossum5e2c5fa2002-05-30 17:33:07 +00003879
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003880 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) {
3881 PyErr_Format(PyExc_ValueError,
3882 "invalid file open mode %.200s", mode);
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003883 Py_XDECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003884 return NULL;
3885 }
3886 }
3887 if (fob == Py_None)
3888 fp = NULL;
3889 else {
3890 fp = get_file(NULL, fob, mode);
3891 if (fp == NULL) {
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003892 Py_XDECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003893 return NULL;
3894 }
3895 }
Victor Stinner41c5fec2011-03-13 21:46:30 -04003896 ret = load_module(name, fp, pathname, type, NULL);
Victor Stinner6ae1e7f2011-03-20 22:37:17 +01003897 Py_XDECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003898 if (fp)
3899 fclose(fp);
3900 return ret;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003901}
3902
3903static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003904imp_load_package(PyObject *self, PyObject *args)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003905{
Victor Stinnerc9abda02011-03-14 13:33:46 -04003906 PyObject *name, *pathname;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003907 PyObject * ret;
Victor Stinnerc9abda02011-03-14 13:33:46 -04003908 if (!PyArg_ParseTuple(args, "UO&:load_package",
3909 &name, PyUnicode_FSDecoder, &pathname))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003910 return NULL;
Victor Stinnerc9abda02011-03-14 13:33:46 -04003911 ret = load_package(name, pathname);
Victor Stinnerebc00522010-12-03 17:06:43 +00003912 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003913 return ret;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00003914}
3915
3916static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003917imp_new_module(PyObject *self, PyObject *args)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003918{
Victor Stinnerfe19d212011-03-14 14:53:28 -04003919 PyObject *name;
3920 if (!PyArg_ParseTuple(args, "U:new_module", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003921 return NULL;
Victor Stinnerfe19d212011-03-14 14:53:28 -04003922 return PyModule_NewObject(name);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003923}
3924
Christian Heimes13a7a212008-01-07 17:13:09 +00003925static PyObject *
3926imp_reload(PyObject *self, PyObject *v)
3927{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003928 return PyImport_ReloadModule(v);
Christian Heimes13a7a212008-01-07 17:13:09 +00003929}
3930
3931PyDoc_STRVAR(doc_reload,
3932"reload(module) -> module\n\
3933\n\
3934Reload the module. The module must have been successfully imported before.");
3935
Barry Warsaw28a691b2010-04-17 00:19:56 +00003936static PyObject *
3937imp_cache_from_source(PyObject *self, PyObject *args, PyObject *kws)
3938{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003939 static char *kwlist[] = {"path", "debug_override", NULL};
Barry Warsaw28a691b2010-04-17 00:19:56 +00003940
Victor Stinner2f42ae52011-03-20 00:41:24 +01003941 PyObject *pathname, *cpathname;
Benjamin Peterson294a9fc2010-10-16 03:12:39 +00003942 PyObject *debug_override = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003943 int debug = !Py_OptimizeFlag;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003944
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003945 if (!PyArg_ParseTupleAndKeywords(
Victor Stinner3ea23dd2010-10-15 20:34:32 +00003946 args, kws, "O&|O", kwlist,
Victor Stinner2f42ae52011-03-20 00:41:24 +01003947 PyUnicode_FSDecoder, &pathname, &debug_override))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003948 return NULL;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003949
Benjamin Peterson294a9fc2010-10-16 03:12:39 +00003950 if (debug_override != NULL &&
3951 (debug = PyObject_IsTrue(debug_override)) < 0) {
Victor Stinner2f42ae52011-03-20 00:41:24 +01003952 Py_DECREF(pathname);
Benjamin Peterson294a9fc2010-10-16 03:12:39 +00003953 return NULL;
3954 }
Barry Warsaw28a691b2010-04-17 00:19:56 +00003955
Martin v. Löwis30260a72011-10-23 17:35:46 +02003956 if (PyUnicode_READY(pathname) < 0)
3957 return NULL;
3958
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003959 cpathname = make_compiled_pathname(pathname, debug);
Victor Stinner2f42ae52011-03-20 00:41:24 +01003960 Py_DECREF(pathname);
Barry Warsaw28a691b2010-04-17 00:19:56 +00003961
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003962 if (cpathname == NULL) {
3963 PyErr_Format(PyExc_SystemError, "path buffer too short");
3964 return NULL;
3965 }
Victor Stinner2f42ae52011-03-20 00:41:24 +01003966 return cpathname;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003967}
3968
3969PyDoc_STRVAR(doc_cache_from_source,
3970"Given the path to a .py file, return the path to its .pyc/.pyo file.\n\
3971\n\
3972The .py file does not need to exist; this simply returns the path to the\n\
3973.pyc/.pyo file calculated as if the .py file were imported. The extension\n\
3974will be .pyc unless __debug__ is not defined, then it will be .pyo.\n\
3975\n\
3976If debug_override is not None, then it must be a boolean and is taken as\n\
3977the value of __debug__ instead.");
3978
3979static PyObject *
3980imp_source_from_cache(PyObject *self, PyObject *args, PyObject *kws)
3981{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003982 static char *kwlist[] = {"path", NULL};
Victor Stinnerc9abda02011-03-14 13:33:46 -04003983 PyObject *pathname, *source;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003985 if (!PyArg_ParseTupleAndKeywords(
Victor Stinnerebc00522010-12-03 17:06:43 +00003986 args, kws, "O&", kwlist,
Victor Stinnerc9abda02011-03-14 13:33:46 -04003987 PyUnicode_FSDecoder, &pathname))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003988 return NULL;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003989
Victor Stinnerc9abda02011-03-14 13:33:46 -04003990 source = make_source_pathname(pathname);
3991 if (source == NULL) {
3992 PyErr_Format(PyExc_ValueError, "Not a PEP 3147 pyc path: %R",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003993 pathname);
Victor Stinnerc9abda02011-03-14 13:33:46 -04003994 Py_DECREF(pathname);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003995 return NULL;
3996 }
Victor Stinnerc9abda02011-03-14 13:33:46 -04003997 Py_DECREF(pathname);
3998 return source;
Barry Warsaw28a691b2010-04-17 00:19:56 +00003999}
4000
4001PyDoc_STRVAR(doc_source_from_cache,
4002"Given the path to a .pyc./.pyo file, return the path to its .py file.\n\
4003\n\
4004The .pyc/.pyo file does not need to exist; this simply returns the path to\n\
4005the .py file calculated to correspond to the .pyc/.pyo file. If path\n\
4006does not conform to PEP 3147 format, ValueError will be raised.");
4007
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004008/* Doc strings */
4009
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004010PyDoc_STRVAR(doc_imp,
4011"This module provides the components needed to build your own\n\
4012__import__ function. Undocumented functions are obsolete.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004013
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004014PyDoc_STRVAR(doc_find_module,
4015"find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004016Search for a module. If path is omitted or None, search for a\n\
4017built-in, frozen or special module and continue search in sys.path.\n\
4018The module name cannot contain '.'; to search for a submodule of a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004019package, pass the submodule name and the package's __path__.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004020
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004021PyDoc_STRVAR(doc_load_module,
4022"load_module(name, file, filename, (suffix, mode, type)) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004023Load a module, given information returned by find_module().\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004024The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004025
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004026PyDoc_STRVAR(doc_get_magic,
4027"get_magic() -> string\n\
4028Return the magic number for .pyc or .pyo files.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004029
Barry Warsaw28a691b2010-04-17 00:19:56 +00004030PyDoc_STRVAR(doc_get_tag,
4031"get_tag() -> string\n\
4032Return the magic tag for .pyc or .pyo files.");
4033
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004034PyDoc_STRVAR(doc_get_suffixes,
4035"get_suffixes() -> [(suffix, mode, type), ...]\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004036Return a list of (suffix, mode, type) tuples describing the files\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004037that find_module() looks for.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004038
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004039PyDoc_STRVAR(doc_new_module,
4040"new_module(name) -> module\n\
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004041Create a new module. Do not enter it in sys.modules.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004042The module name must include the full package name, if any.");
Guido van Rossum0207e6d1997-09-09 22:04:42 +00004043
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004044PyDoc_STRVAR(doc_lock_held,
Tim Petersa7c65092004-08-01 23:26:05 +00004045"lock_held() -> boolean\n\
4046Return True if the import lock is currently held, else False.\n\
4047On platforms without threads, return False.");
Tim Peters69232342001-08-30 05:16:13 +00004048
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00004049PyDoc_STRVAR(doc_acquire_lock,
4050"acquire_lock() -> None\n\
Neal Norwitz2294c0d2003-02-12 23:02:21 +00004051Acquires the interpreter's import lock for the current thread.\n\
4052This lock should be used by import hooks to ensure thread-safety\n\
4053when importing modules.\n\
Guido van Rossumc4f4ca92003-02-12 21:46:11 +00004054On platforms without threads, this function does nothing.");
4055
4056PyDoc_STRVAR(doc_release_lock,
4057"release_lock() -> None\n\
4058Release the interpreter's import lock.\n\
4059On platforms without threads, this function does nothing.");
4060
Guido van Rossum79f25d91997-04-29 20:08:16 +00004061static PyMethodDef imp_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004062 {"find_module", imp_find_module, METH_VARARGS, doc_find_module},
4063 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic},
4064 {"get_tag", imp_get_tag, METH_NOARGS, doc_get_tag},
4065 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes},
4066 {"load_module", imp_load_module, METH_VARARGS, doc_load_module},
4067 {"new_module", imp_new_module, METH_VARARGS, doc_new_module},
4068 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held},
4069 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock},
4070 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock},
4071 {"reload", imp_reload, METH_O, doc_reload},
4072 {"cache_from_source", (PyCFunction)imp_cache_from_source,
4073 METH_VARARGS | METH_KEYWORDS, doc_cache_from_source},
4074 {"source_from_cache", (PyCFunction)imp_source_from_cache,
4075 METH_VARARGS | METH_KEYWORDS, doc_source_from_cache},
4076 /* The rest are obsolete */
4077 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS},
4078 {"is_frozen_package", imp_is_frozen_package, METH_VARARGS},
4079 {"init_builtin", imp_init_builtin, METH_VARARGS},
4080 {"init_frozen", imp_init_frozen, METH_VARARGS},
4081 {"is_builtin", imp_is_builtin, METH_VARARGS},
4082 {"is_frozen", imp_is_frozen, METH_VARARGS},
4083 {"load_compiled", imp_load_compiled, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00004084#ifdef HAVE_DYNAMIC_LOADING
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004085 {"load_dynamic", imp_load_dynamic, METH_VARARGS},
Guido van Rossum96a8fb71999-12-22 14:09:35 +00004086#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004087 {"load_package", imp_load_package, METH_VARARGS},
4088 {"load_source", imp_load_source, METH_VARARGS},
Brett Cannon442c9b92011-03-23 16:14:42 -07004089 {"_fix_co_filename", imp_fix_co_filename, METH_VARARGS},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004090 {NULL, NULL} /* sentinel */
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004091};
4092
Guido van Rossum1a8791e1998-08-04 22:46:29 +00004093static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004094setint(PyObject *d, char *name, int value)
Guido van Rossumaee0bad1997-09-05 07:33:22 +00004095{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004096 PyObject *v;
4097 int err;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00004098
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004099 v = PyLong_FromLong((long)value);
4100 err = PyDict_SetItemString(d, name, v);
4101 Py_XDECREF(v);
4102 return err;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00004103}
4104
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004105typedef struct {
4106 PyObject_HEAD
4107} NullImporter;
4108
4109static int
4110NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds)
4111{
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004112#ifndef MS_WINDOWS
4113 PyObject *path;
4114 struct stat statbuf;
4115 int rv;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004117 if (!_PyArg_NoKeywords("NullImporter()", kwds))
4118 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004119
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004120 if (!PyArg_ParseTuple(args, "O&:NullImporter",
4121 PyUnicode_FSConverter, &path))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004122 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004123
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004124 if (PyBytes_GET_SIZE(path) == 0) {
4125 Py_DECREF(path);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004126 PyErr_SetString(PyExc_ImportError, "empty pathname");
4127 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004128 }
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004129
4130 rv = stat(PyBytes_AS_STRING(path), &statbuf);
4131 Py_DECREF(path);
4132 if (rv == 0) {
4133 /* it exists */
4134 if (S_ISDIR(statbuf.st_mode)) {
4135 /* it's a directory */
4136 PyErr_SetString(PyExc_ImportError, "existing directory");
4137 return -1;
4138 }
4139 }
4140#else /* MS_WINDOWS */
4141 PyObject *pathobj;
4142 DWORD rv;
Victor Stinner255dfdb2010-09-29 10:28:51 +00004143 wchar_t *path;
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004144
4145 if (!_PyArg_NoKeywords("NullImporter()", kwds))
4146 return -1;
4147
4148 if (!PyArg_ParseTuple(args, "U:NullImporter",
4149 &pathobj))
4150 return -1;
4151
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004152 if (PyUnicode_GET_LENGTH(pathobj) == 0) {
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004153 PyErr_SetString(PyExc_ImportError, "empty pathname");
4154 return -1;
4155 }
4156
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00004157 path = PyUnicode_AsWideCharString(pathobj, NULL);
Victor Stinner255dfdb2010-09-29 10:28:51 +00004158 if (path == NULL)
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004159 return -1;
4160 /* see issue1293 and issue3677:
4161 * stat() on Windows doesn't recognise paths like
4162 * "e:\\shared\\" and "\\\\whiterab-c2znlh\\shared" as dirs.
4163 */
4164 rv = GetFileAttributesW(path);
Victor Stinner255dfdb2010-09-29 10:28:51 +00004165 PyMem_Free(path);
Victor Stinner1a4d12d2010-08-13 13:07:29 +00004166 if (rv != INVALID_FILE_ATTRIBUTES) {
4167 /* it exists */
4168 if (rv & FILE_ATTRIBUTE_DIRECTORY) {
4169 /* it's a directory */
4170 PyErr_SetString(PyExc_ImportError, "existing directory");
4171 return -1;
4172 }
4173 }
4174#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004175 return 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004176}
4177
4178static PyObject *
4179NullImporter_find_module(NullImporter *self, PyObject *args)
4180{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004181 Py_RETURN_NONE;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004182}
4183
4184static PyMethodDef NullImporter_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004185 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS,
4186 "Always return None"
4187 },
4188 {NULL} /* Sentinel */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004189};
4190
4191
Christian Heimes9cd17752007-11-18 19:35:23 +00004192PyTypeObject PyNullImporter_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004193 PyVarObject_HEAD_INIT(NULL, 0)
4194 "imp.NullImporter", /*tp_name*/
4195 sizeof(NullImporter), /*tp_basicsize*/
4196 0, /*tp_itemsize*/
4197 0, /*tp_dealloc*/
4198 0, /*tp_print*/
4199 0, /*tp_getattr*/
4200 0, /*tp_setattr*/
4201 0, /*tp_reserved*/
4202 0, /*tp_repr*/
4203 0, /*tp_as_number*/
4204 0, /*tp_as_sequence*/
4205 0, /*tp_as_mapping*/
4206 0, /*tp_hash */
4207 0, /*tp_call*/
4208 0, /*tp_str*/
4209 0, /*tp_getattro*/
4210 0, /*tp_setattro*/
4211 0, /*tp_as_buffer*/
4212 Py_TPFLAGS_DEFAULT, /*tp_flags*/
4213 "Null importer object", /* tp_doc */
4214 0, /* tp_traverse */
4215 0, /* tp_clear */
4216 0, /* tp_richcompare */
4217 0, /* tp_weaklistoffset */
4218 0, /* tp_iter */
4219 0, /* tp_iternext */
4220 NullImporter_methods, /* tp_methods */
4221 0, /* tp_members */
4222 0, /* tp_getset */
4223 0, /* tp_base */
4224 0, /* tp_dict */
4225 0, /* tp_descr_get */
4226 0, /* tp_descr_set */
4227 0, /* tp_dictoffset */
4228 (initproc)NullImporter_init, /* tp_init */
4229 0, /* tp_alloc */
4230 PyType_GenericNew /* tp_new */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004231};
4232
Martin v. Löwis1a214512008-06-11 05:26:20 +00004233static struct PyModuleDef impmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004234 PyModuleDef_HEAD_INIT,
4235 "imp",
4236 doc_imp,
4237 0,
4238 imp_methods,
4239 NULL,
4240 NULL,
4241 NULL,
4242 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004243};
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004244
Jason Tishler6bc06ec2003-09-04 11:59:50 +00004245PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004246PyInit_imp(void)
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004247{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004248 PyObject *m, *d;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004249
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004250 if (PyType_Ready(&PyNullImporter_Type) < 0)
4251 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004252
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004253 m = PyModule_Create(&impmodule);
4254 if (m == NULL)
4255 goto failure;
4256 d = PyModule_GetDict(m);
4257 if (d == NULL)
4258 goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004260 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure;
4261 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure;
4262 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure;
4263 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure;
4264 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure;
4265 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure;
4266 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure;
4267 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure;
4268 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure;
4269 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004270
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004271 Py_INCREF(&PyNullImporter_Type);
4272 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type);
4273 return m;
Guido van Rossumaee0bad1997-09-05 07:33:22 +00004274 failure:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004275 Py_XDECREF(m);
4276 return NULL;
Guido van Rossum1ae940a1995-01-02 19:04:15 +00004277}
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004278
4279
Guido van Rossumb18618d2000-05-03 23:44:39 +00004280/* API for embedding applications that want to add their own entries
4281 to the table of built-in modules. This should normally be called
4282 *before* Py_Initialize(). When the table resize fails, -1 is
4283 returned and the existing table is unchanged.
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004284
4285 After a similar function by Just van Rossum. */
4286
4287int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004288PyImport_ExtendInittab(struct _inittab *newtab)
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004289{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004290 static struct _inittab *our_copy = NULL;
4291 struct _inittab *p;
4292 int i, n;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004294 /* Count the number of entries in both tables */
4295 for (n = 0; newtab[n].name != NULL; n++)
4296 ;
4297 if (n == 0)
4298 return 0; /* Nothing to do */
4299 for (i = 0; PyImport_Inittab[i].name != NULL; i++)
4300 ;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004302 /* Allocate new memory for the combined table */
4303 p = our_copy;
4304 PyMem_RESIZE(p, struct _inittab, i+n+1);
4305 if (p == NULL)
4306 return -1;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004307
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004308 /* Copy the tables into the new memory */
4309 if (our_copy != PyImport_Inittab)
4310 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab));
4311 PyImport_Inittab = our_copy = p;
4312 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab));
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004313
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004314 return 0;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004315}
4316
4317/* Shorthand to add a single entry given a name and a function */
4318
4319int
Brett Cannona826f322009-04-02 03:41:46 +00004320PyImport_AppendInittab(const char *name, PyObject* (*initfunc)(void))
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004321{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004322 struct _inittab newtab[2];
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004323
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004324 memset(newtab, '\0', sizeof newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004325
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004326 newtab[0].name = (char *)name;
4327 newtab[0].initfunc = initfunc;
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004329 return PyImport_ExtendInittab(newtab);
Guido van Rossum09cae1f1998-05-14 02:32:54 +00004330}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004331
4332#ifdef __cplusplus
4333}
4334#endif