|
1 |
|
2 /* Module definition and import implementation */ |
|
3 |
|
4 #include "Python.h" |
|
5 |
|
6 #include "Python-ast.h" |
|
7 #undef Yield /* undefine macro conflicting with winbase.h */ |
|
8 #include "pyarena.h" |
|
9 #include "pythonrun.h" |
|
10 #include "errcode.h" |
|
11 #include "marshal.h" |
|
12 #include "code.h" |
|
13 #include "compile.h" |
|
14 #include "eval.h" |
|
15 #include "osdefs.h" |
|
16 #include "importdl.h" |
|
17 |
|
18 #ifdef HAVE_FCNTL_H |
|
19 #include <fcntl.h> |
|
20 #endif |
|
21 #ifdef __cplusplus |
|
22 extern "C" { |
|
23 #endif |
|
24 |
|
25 #ifdef MS_WINDOWS |
|
26 /* for stat.st_mode */ |
|
27 typedef unsigned short mode_t; |
|
28 #endif |
|
29 |
|
30 extern time_t PyOS_GetLastModificationTime(char *, FILE *); |
|
31 /* In getmtime.c */ |
|
32 |
|
33 /* Magic word to reject .pyc files generated by other Python versions. |
|
34 It should change for each incompatible change to the bytecode. |
|
35 |
|
36 The value of CR and LF is incorporated so if you ever read or write |
|
37 a .pyc file in text mode the magic number will be wrong; also, the |
|
38 Apple MPW compiler swaps their values, botching string constants. |
|
39 |
|
40 The magic numbers must be spaced apart atleast 2 values, as the |
|
41 -U interpeter flag will cause MAGIC+1 being used. They have been |
|
42 odd numbers for some time now. |
|
43 |
|
44 There were a variety of old schemes for setting the magic number. |
|
45 The current working scheme is to increment the previous value by |
|
46 10. |
|
47 |
|
48 Known values: |
|
49 Python 1.5: 20121 |
|
50 Python 1.5.1: 20121 |
|
51 Python 1.5.2: 20121 |
|
52 Python 1.6: 50428 |
|
53 Python 2.0: 50823 |
|
54 Python 2.0.1: 50823 |
|
55 Python 2.1: 60202 |
|
56 Python 2.1.1: 60202 |
|
57 Python 2.1.2: 60202 |
|
58 Python 2.2: 60717 |
|
59 Python 2.3a0: 62011 |
|
60 Python 2.3a0: 62021 |
|
61 Python 2.3a0: 62011 (!) |
|
62 Python 2.4a0: 62041 |
|
63 Python 2.4a3: 62051 |
|
64 Python 2.4b1: 62061 |
|
65 Python 2.5a0: 62071 |
|
66 Python 2.5a0: 62081 (ast-branch) |
|
67 Python 2.5a0: 62091 (with) |
|
68 Python 2.5a0: 62092 (changed WITH_CLEANUP opcode) |
|
69 Python 2.5b3: 62101 (fix wrong code: for x, in ...) |
|
70 Python 2.5b3: 62111 (fix wrong code: x += yield) |
|
71 Python 2.5c1: 62121 (fix wrong lnotab with for loops and |
|
72 storing constants that should have been removed) |
|
73 Python 2.5c2: 62131 (fix wrong code: for x, in ... in listcomp/genexp) |
|
74 Python 2.6a0: 62151 (peephole optimizations and STORE_MAP opcode) |
|
75 Python 2.6a1: 62161 (WITH_CLEANUP optimization) |
|
76 . |
|
77 */ |
|
78 #define MAGIC (62161 | ((long)'\r'<<16) | ((long)'\n'<<24)) |
|
79 |
|
80 /* Magic word as global; note that _PyImport_Init() can change the |
|
81 value of this global to accommodate for alterations of how the |
|
82 compiler works which are enabled by command line switches. */ |
|
83 static long pyc_magic = MAGIC; |
|
84 |
|
85 /* See _PyImport_FixupExtension() below */ |
|
86 static PyObject *extensions = NULL; |
|
87 |
|
88 /* This table is defined in config.c: */ |
|
89 extern struct _inittab _PyImport_Inittab[]; |
|
90 |
|
91 struct _inittab *PyImport_Inittab = _PyImport_Inittab; |
|
92 |
|
93 /* these tables define the module suffixes that Python recognizes */ |
|
94 struct filedescr * _PyImport_Filetab = NULL; |
|
95 |
|
96 #ifdef RISCOS |
|
97 static const struct filedescr _PyImport_StandardFiletab[] = { |
|
98 {"/py", "U", PY_SOURCE}, |
|
99 {"/pyc", "rb", PY_COMPILED}, |
|
100 {0, 0} |
|
101 }; |
|
102 #else |
|
103 static const struct filedescr _PyImport_StandardFiletab[] = { |
|
104 {".py", "U", PY_SOURCE}, |
|
105 #ifdef MS_WINDOWS |
|
106 {".pyw", "U", PY_SOURCE}, |
|
107 #endif |
|
108 {".pyc", "rb", PY_COMPILED}, |
|
109 {0, 0} |
|
110 }; |
|
111 #endif |
|
112 |
|
113 |
|
114 /* Initialize things */ |
|
115 |
|
116 void |
|
117 _PyImport_Init(void) |
|
118 { |
|
119 const struct filedescr *scan; |
|
120 struct filedescr *filetab; |
|
121 int countD = 0; |
|
122 int countS = 0; |
|
123 |
|
124 /* prepare _PyImport_Filetab: copy entries from |
|
125 _PyImport_DynLoadFiletab and _PyImport_StandardFiletab. |
|
126 */ |
|
127 #ifdef HAVE_DYNAMIC_LOADING |
|
128 for (scan = _PyImport_DynLoadFiletab; scan->suffix != NULL; ++scan) |
|
129 ++countD; |
|
130 #endif |
|
131 for (scan = _PyImport_StandardFiletab; scan->suffix != NULL; ++scan) |
|
132 ++countS; |
|
133 filetab = PyMem_NEW(struct filedescr, countD + countS + 1); |
|
134 if (filetab == NULL) |
|
135 Py_FatalError("Can't initialize import file table."); |
|
136 #ifdef HAVE_DYNAMIC_LOADING |
|
137 memcpy(filetab, _PyImport_DynLoadFiletab, |
|
138 countD * sizeof(struct filedescr)); |
|
139 #endif |
|
140 memcpy(filetab + countD, _PyImport_StandardFiletab, |
|
141 countS * sizeof(struct filedescr)); |
|
142 filetab[countD + countS].suffix = NULL; |
|
143 |
|
144 _PyImport_Filetab = filetab; |
|
145 |
|
146 if (Py_OptimizeFlag) { |
|
147 /* Replace ".pyc" with ".pyo" in _PyImport_Filetab */ |
|
148 for (; filetab->suffix != NULL; filetab++) { |
|
149 #ifndef RISCOS |
|
150 if (strcmp(filetab->suffix, ".pyc") == 0) |
|
151 filetab->suffix = ".pyo"; |
|
152 #else |
|
153 if (strcmp(filetab->suffix, "/pyc") == 0) |
|
154 filetab->suffix = "/pyo"; |
|
155 #endif |
|
156 } |
|
157 } |
|
158 |
|
159 if (Py_UnicodeFlag) { |
|
160 /* Fix the pyc_magic so that byte compiled code created |
|
161 using the all-Unicode method doesn't interfere with |
|
162 code created in normal operation mode. */ |
|
163 pyc_magic = MAGIC + 1; |
|
164 } |
|
165 } |
|
166 |
|
167 void |
|
168 _PyImportHooks_Init(void) |
|
169 { |
|
170 PyObject *v, *path_hooks = NULL, *zimpimport; |
|
171 int err = 0; |
|
172 |
|
173 /* adding sys.path_hooks and sys.path_importer_cache, setting up |
|
174 zipimport */ |
|
175 if (PyType_Ready(&PyNullImporter_Type) < 0) |
|
176 goto error; |
|
177 |
|
178 if (Py_VerboseFlag) |
|
179 PySys_WriteStderr("# installing zipimport hook\n"); |
|
180 |
|
181 v = PyList_New(0); |
|
182 if (v == NULL) |
|
183 goto error; |
|
184 err = PySys_SetObject("meta_path", v); |
|
185 Py_DECREF(v); |
|
186 if (err) |
|
187 goto error; |
|
188 v = PyDict_New(); |
|
189 if (v == NULL) |
|
190 goto error; |
|
191 err = PySys_SetObject("path_importer_cache", v); |
|
192 Py_DECREF(v); |
|
193 if (err) |
|
194 goto error; |
|
195 path_hooks = PyList_New(0); |
|
196 if (path_hooks == NULL) |
|
197 goto error; |
|
198 err = PySys_SetObject("path_hooks", path_hooks); |
|
199 if (err) { |
|
200 error: |
|
201 PyErr_Print(); |
|
202 Py_FatalError("initializing sys.meta_path, sys.path_hooks, " |
|
203 "path_importer_cache, or NullImporter failed" |
|
204 ); |
|
205 } |
|
206 |
|
207 zimpimport = PyImport_ImportModule("zipimport"); |
|
208 if (zimpimport == NULL) { |
|
209 PyErr_Clear(); /* No zip import module -- okay */ |
|
210 if (Py_VerboseFlag) |
|
211 PySys_WriteStderr("# can't import zipimport\n"); |
|
212 } |
|
213 else { |
|
214 PyObject *zipimporter = PyObject_GetAttrString(zimpimport, |
|
215 "zipimporter"); |
|
216 Py_DECREF(zimpimport); |
|
217 if (zipimporter == NULL) { |
|
218 PyErr_Clear(); /* No zipimporter object -- okay */ |
|
219 if (Py_VerboseFlag) |
|
220 PySys_WriteStderr( |
|
221 "# can't import zipimport.zipimporter\n"); |
|
222 } |
|
223 else { |
|
224 /* sys.path_hooks.append(zipimporter) */ |
|
225 err = PyList_Append(path_hooks, zipimporter); |
|
226 Py_DECREF(zipimporter); |
|
227 if (err) |
|
228 goto error; |
|
229 if (Py_VerboseFlag) |
|
230 PySys_WriteStderr( |
|
231 "# installed zipimport hook\n"); |
|
232 } |
|
233 } |
|
234 Py_DECREF(path_hooks); |
|
235 } |
|
236 |
|
237 void |
|
238 _PyImport_Fini(void) |
|
239 { |
|
240 Py_XDECREF(extensions); |
|
241 extensions = NULL; |
|
242 PyMem_DEL(_PyImport_Filetab); |
|
243 _PyImport_Filetab = NULL; |
|
244 } |
|
245 |
|
246 |
|
247 /* Locking primitives to prevent parallel imports of the same module |
|
248 in different threads to return with a partially loaded module. |
|
249 These calls are serialized by the global interpreter lock. */ |
|
250 |
|
251 #ifdef WITH_THREAD |
|
252 |
|
253 #include "pythread.h" |
|
254 |
|
255 static PyThread_type_lock import_lock = 0; |
|
256 static long import_lock_thread = -1; |
|
257 static int import_lock_level = 0; |
|
258 |
|
259 static void |
|
260 lock_import(void) |
|
261 { |
|
262 long me = PyThread_get_thread_ident(); |
|
263 if (me == -1) |
|
264 return; /* Too bad */ |
|
265 if (import_lock == NULL) { |
|
266 import_lock = PyThread_allocate_lock(); |
|
267 if (import_lock == NULL) |
|
268 return; /* Nothing much we can do. */ |
|
269 } |
|
270 if (import_lock_thread == me) { |
|
271 import_lock_level++; |
|
272 return; |
|
273 } |
|
274 if (import_lock_thread != -1 || !PyThread_acquire_lock(import_lock, 0)) |
|
275 { |
|
276 PyThreadState *tstate = PyEval_SaveThread(); |
|
277 PyThread_acquire_lock(import_lock, 1); |
|
278 PyEval_RestoreThread(tstate); |
|
279 } |
|
280 import_lock_thread = me; |
|
281 import_lock_level = 1; |
|
282 } |
|
283 |
|
284 static int |
|
285 unlock_import(void) |
|
286 { |
|
287 long me = PyThread_get_thread_ident(); |
|
288 if (me == -1 || import_lock == NULL) |
|
289 return 0; /* Too bad */ |
|
290 if (import_lock_thread != me) |
|
291 return -1; |
|
292 import_lock_level--; |
|
293 if (import_lock_level == 0) { |
|
294 import_lock_thread = -1; |
|
295 PyThread_release_lock(import_lock); |
|
296 } |
|
297 return 1; |
|
298 } |
|
299 |
|
300 /* This function is called from PyOS_AfterFork to ensure that newly |
|
301 created child processes do not share locks with the parent. */ |
|
302 |
|
303 void |
|
304 _PyImport_ReInitLock(void) |
|
305 { |
|
306 #ifdef _AIX |
|
307 if (import_lock != NULL) |
|
308 import_lock = PyThread_allocate_lock(); |
|
309 #endif |
|
310 } |
|
311 |
|
312 #else |
|
313 |
|
314 #define lock_import() |
|
315 #define unlock_import() 0 |
|
316 |
|
317 #endif |
|
318 |
|
319 static PyObject * |
|
320 imp_lock_held(PyObject *self, PyObject *noargs) |
|
321 { |
|
322 #ifdef WITH_THREAD |
|
323 return PyBool_FromLong(import_lock_thread != -1); |
|
324 #else |
|
325 return PyBool_FromLong(0); |
|
326 #endif |
|
327 } |
|
328 |
|
329 static PyObject * |
|
330 imp_acquire_lock(PyObject *self, PyObject *noargs) |
|
331 { |
|
332 #ifdef WITH_THREAD |
|
333 lock_import(); |
|
334 #endif |
|
335 Py_INCREF(Py_None); |
|
336 return Py_None; |
|
337 } |
|
338 |
|
339 static PyObject * |
|
340 imp_release_lock(PyObject *self, PyObject *noargs) |
|
341 { |
|
342 #ifdef WITH_THREAD |
|
343 if (unlock_import() < 0) { |
|
344 PyErr_SetString(PyExc_RuntimeError, |
|
345 "not holding the import lock"); |
|
346 return NULL; |
|
347 } |
|
348 #endif |
|
349 Py_INCREF(Py_None); |
|
350 return Py_None; |
|
351 } |
|
352 |
|
353 static void |
|
354 imp_modules_reloading_clear(void) |
|
355 { |
|
356 PyInterpreterState *interp = PyThreadState_Get()->interp; |
|
357 if (interp->modules_reloading != NULL) |
|
358 PyDict_Clear(interp->modules_reloading); |
|
359 } |
|
360 |
|
361 /* Helper for sys */ |
|
362 |
|
363 PyObject * |
|
364 PyImport_GetModuleDict(void) |
|
365 { |
|
366 PyInterpreterState *interp = PyThreadState_GET()->interp; |
|
367 if (interp->modules == NULL) |
|
368 Py_FatalError("PyImport_GetModuleDict: no module dictionary!"); |
|
369 return interp->modules; |
|
370 } |
|
371 |
|
372 |
|
373 /* List of names to clear in sys */ |
|
374 static char* sys_deletes[] = { |
|
375 "path", "argv", "ps1", "ps2", "exitfunc", |
|
376 "exc_type", "exc_value", "exc_traceback", |
|
377 "last_type", "last_value", "last_traceback", |
|
378 "path_hooks", "path_importer_cache", "meta_path", |
|
379 /* misc stuff */ |
|
380 "flags", "float_info", |
|
381 NULL |
|
382 }; |
|
383 |
|
384 static char* sys_files[] = { |
|
385 "stdin", "__stdin__", |
|
386 "stdout", "__stdout__", |
|
387 "stderr", "__stderr__", |
|
388 NULL |
|
389 }; |
|
390 |
|
391 |
|
392 /* Un-initialize things, as good as we can */ |
|
393 |
|
394 void |
|
395 PyImport_Cleanup(void) |
|
396 { |
|
397 Py_ssize_t pos, ndone; |
|
398 char *name; |
|
399 PyObject *key, *value, *dict; |
|
400 PyInterpreterState *interp = PyThreadState_GET()->interp; |
|
401 PyObject *modules = interp->modules; |
|
402 |
|
403 if (modules == NULL) |
|
404 return; /* Already done */ |
|
405 |
|
406 /* Delete some special variables first. These are common |
|
407 places where user values hide and people complain when their |
|
408 destructors fail. Since the modules containing them are |
|
409 deleted *last* of all, they would come too late in the normal |
|
410 destruction order. Sigh. */ |
|
411 |
|
412 value = PyDict_GetItemString(modules, "__builtin__"); |
|
413 if (value != NULL && PyModule_Check(value)) { |
|
414 dict = PyModule_GetDict(value); |
|
415 if (Py_VerboseFlag) |
|
416 PySys_WriteStderr("# clear __builtin__._\n"); |
|
417 PyDict_SetItemString(dict, "_", Py_None); |
|
418 } |
|
419 value = PyDict_GetItemString(modules, "sys"); |
|
420 if (value != NULL && PyModule_Check(value)) { |
|
421 char **p; |
|
422 PyObject *v; |
|
423 dict = PyModule_GetDict(value); |
|
424 for (p = sys_deletes; *p != NULL; p++) { |
|
425 if (Py_VerboseFlag) |
|
426 PySys_WriteStderr("# clear sys.%s\n", *p); |
|
427 PyDict_SetItemString(dict, *p, Py_None); |
|
428 } |
|
429 for (p = sys_files; *p != NULL; p+=2) { |
|
430 if (Py_VerboseFlag) |
|
431 PySys_WriteStderr("# restore sys.%s\n", *p); |
|
432 v = PyDict_GetItemString(dict, *(p+1)); |
|
433 if (v == NULL) |
|
434 v = Py_None; |
|
435 PyDict_SetItemString(dict, *p, v); |
|
436 } |
|
437 } |
|
438 |
|
439 /* First, delete __main__ */ |
|
440 value = PyDict_GetItemString(modules, "__main__"); |
|
441 if (value != NULL && PyModule_Check(value)) { |
|
442 if (Py_VerboseFlag) |
|
443 PySys_WriteStderr("# cleanup __main__\n"); |
|
444 _PyModule_Clear(value); |
|
445 PyDict_SetItemString(modules, "__main__", Py_None); |
|
446 } |
|
447 |
|
448 /* The special treatment of __builtin__ here is because even |
|
449 when it's not referenced as a module, its dictionary is |
|
450 referenced by almost every module's __builtins__. Since |
|
451 deleting a module clears its dictionary (even if there are |
|
452 references left to it), we need to delete the __builtin__ |
|
453 module last. Likewise, we don't delete sys until the very |
|
454 end because it is implicitly referenced (e.g. by print). |
|
455 |
|
456 Also note that we 'delete' modules by replacing their entry |
|
457 in the modules dict with None, rather than really deleting |
|
458 them; this avoids a rehash of the modules dictionary and |
|
459 also marks them as "non existent" so they won't be |
|
460 re-imported. */ |
|
461 |
|
462 /* Next, repeatedly delete modules with a reference count of |
|
463 one (skipping __builtin__ and sys) and delete them */ |
|
464 do { |
|
465 ndone = 0; |
|
466 pos = 0; |
|
467 while (PyDict_Next(modules, &pos, &key, &value)) { |
|
468 if (value->ob_refcnt != 1) |
|
469 continue; |
|
470 if (PyString_Check(key) && PyModule_Check(value)) { |
|
471 name = PyString_AS_STRING(key); |
|
472 if (strcmp(name, "__builtin__") == 0) |
|
473 continue; |
|
474 if (strcmp(name, "sys") == 0) |
|
475 continue; |
|
476 if (Py_VerboseFlag) |
|
477 PySys_WriteStderr( |
|
478 "# cleanup[1] %s\n", name); |
|
479 _PyModule_Clear(value); |
|
480 PyDict_SetItem(modules, key, Py_None); |
|
481 ndone++; |
|
482 } |
|
483 } |
|
484 } while (ndone > 0); |
|
485 |
|
486 /* Next, delete all modules (still skipping __builtin__ and sys) */ |
|
487 pos = 0; |
|
488 while (PyDict_Next(modules, &pos, &key, &value)) { |
|
489 if (PyString_Check(key) && PyModule_Check(value)) { |
|
490 name = PyString_AS_STRING(key); |
|
491 if (strcmp(name, "__builtin__") == 0) |
|
492 continue; |
|
493 if (strcmp(name, "sys") == 0) |
|
494 continue; |
|
495 if (Py_VerboseFlag) |
|
496 PySys_WriteStderr("# cleanup[2] %s\n", name); |
|
497 _PyModule_Clear(value); |
|
498 PyDict_SetItem(modules, key, Py_None); |
|
499 } |
|
500 } |
|
501 |
|
502 /* Next, delete sys and __builtin__ (in that order) */ |
|
503 value = PyDict_GetItemString(modules, "sys"); |
|
504 if (value != NULL && PyModule_Check(value)) { |
|
505 if (Py_VerboseFlag) |
|
506 PySys_WriteStderr("# cleanup sys\n"); |
|
507 _PyModule_Clear(value); |
|
508 PyDict_SetItemString(modules, "sys", Py_None); |
|
509 } |
|
510 value = PyDict_GetItemString(modules, "__builtin__"); |
|
511 if (value != NULL && PyModule_Check(value)) { |
|
512 if (Py_VerboseFlag) |
|
513 PySys_WriteStderr("# cleanup __builtin__\n"); |
|
514 _PyModule_Clear(value); |
|
515 PyDict_SetItemString(modules, "__builtin__", Py_None); |
|
516 } |
|
517 |
|
518 /* Finally, clear and delete the modules directory */ |
|
519 PyDict_Clear(modules); |
|
520 interp->modules = NULL; |
|
521 Py_DECREF(modules); |
|
522 Py_CLEAR(interp->modules_reloading); |
|
523 } |
|
524 |
|
525 |
|
526 /* Helper for pythonrun.c -- return magic number */ |
|
527 |
|
528 long |
|
529 PyImport_GetMagicNumber(void) |
|
530 { |
|
531 return pyc_magic; |
|
532 } |
|
533 |
|
534 |
|
535 /* Magic for extension modules (built-in as well as dynamically |
|
536 loaded). To prevent initializing an extension module more than |
|
537 once, we keep a static dictionary 'extensions' keyed by module name |
|
538 (for built-in modules) or by filename (for dynamically loaded |
|
539 modules), containing these modules. A copy of the module's |
|
540 dictionary is stored by calling _PyImport_FixupExtension() |
|
541 immediately after the module initialization function succeeds. A |
|
542 copy can be retrieved from there by calling |
|
543 _PyImport_FindExtension(). */ |
|
544 |
|
545 PyObject * |
|
546 _PyImport_FixupExtension(char *name, char *filename) |
|
547 { |
|
548 PyObject *modules, *mod, *dict, *copy; |
|
549 if (extensions == NULL) { |
|
550 extensions = PyDict_New(); |
|
551 if (extensions == NULL) |
|
552 return NULL; |
|
553 } |
|
554 modules = PyImport_GetModuleDict(); |
|
555 mod = PyDict_GetItemString(modules, name); |
|
556 if (mod == NULL || !PyModule_Check(mod)) { |
|
557 PyErr_Format(PyExc_SystemError, |
|
558 "_PyImport_FixupExtension: module %.200s not loaded", name); |
|
559 return NULL; |
|
560 } |
|
561 dict = PyModule_GetDict(mod); |
|
562 if (dict == NULL) |
|
563 return NULL; |
|
564 copy = PyDict_Copy(dict); |
|
565 if (copy == NULL) |
|
566 return NULL; |
|
567 PyDict_SetItemString(extensions, filename, copy); |
|
568 Py_DECREF(copy); |
|
569 return copy; |
|
570 } |
|
571 |
|
572 PyObject * |
|
573 _PyImport_FindExtension(char *name, char *filename) |
|
574 { |
|
575 PyObject *dict, *mod, *mdict; |
|
576 if (extensions == NULL) |
|
577 return NULL; |
|
578 dict = PyDict_GetItemString(extensions, filename); |
|
579 if (dict == NULL) |
|
580 return NULL; |
|
581 mod = PyImport_AddModule(name); |
|
582 if (mod == NULL) |
|
583 return NULL; |
|
584 mdict = PyModule_GetDict(mod); |
|
585 if (mdict == NULL) |
|
586 return NULL; |
|
587 if (PyDict_Update(mdict, dict)) |
|
588 return NULL; |
|
589 if (Py_VerboseFlag) |
|
590 PySys_WriteStderr("import %s # previously loaded (%s)\n", |
|
591 name, filename); |
|
592 return mod; |
|
593 } |
|
594 |
|
595 |
|
596 /* Get the module object corresponding to a module name. |
|
597 First check the modules dictionary if there's one there, |
|
598 if not, create a new one and insert it in the modules dictionary. |
|
599 Because the former action is most common, THIS DOES NOT RETURN A |
|
600 'NEW' REFERENCE! */ |
|
601 |
|
602 PyObject * |
|
603 PyImport_AddModule(const char *name) |
|
604 { |
|
605 PyObject *modules = PyImport_GetModuleDict(); |
|
606 PyObject *m; |
|
607 |
|
608 if ((m = PyDict_GetItemString(modules, name)) != NULL && |
|
609 PyModule_Check(m)) |
|
610 return m; |
|
611 m = PyModule_New(name); |
|
612 if (m == NULL) |
|
613 return NULL; |
|
614 if (PyDict_SetItemString(modules, name, m) != 0) { |
|
615 Py_DECREF(m); |
|
616 return NULL; |
|
617 } |
|
618 Py_DECREF(m); /* Yes, it still exists, in modules! */ |
|
619 |
|
620 return m; |
|
621 } |
|
622 |
|
623 /* Remove name from sys.modules, if it's there. */ |
|
624 static void |
|
625 _RemoveModule(const char *name) |
|
626 { |
|
627 PyObject *modules = PyImport_GetModuleDict(); |
|
628 if (PyDict_GetItemString(modules, name) == NULL) |
|
629 return; |
|
630 if (PyDict_DelItemString(modules, name) < 0) |
|
631 Py_FatalError("import: deleting existing key in" |
|
632 "sys.modules failed"); |
|
633 } |
|
634 |
|
635 /* Execute a code object in a module and return the module object |
|
636 * WITH INCREMENTED REFERENCE COUNT. If an error occurs, name is |
|
637 * removed from sys.modules, to avoid leaving damaged module objects |
|
638 * in sys.modules. The caller may wish to restore the original |
|
639 * module object (if any) in this case; PyImport_ReloadModule is an |
|
640 * example. |
|
641 */ |
|
642 PyObject * |
|
643 PyImport_ExecCodeModule(char *name, PyObject *co) |
|
644 { |
|
645 return PyImport_ExecCodeModuleEx(name, co, (char *)NULL); |
|
646 } |
|
647 |
|
648 PyObject * |
|
649 PyImport_ExecCodeModuleEx(char *name, PyObject *co, char *pathname) |
|
650 { |
|
651 PyObject *modules = PyImport_GetModuleDict(); |
|
652 PyObject *m, *d, *v; |
|
653 |
|
654 m = PyImport_AddModule(name); |
|
655 if (m == NULL) |
|
656 return NULL; |
|
657 /* If the module is being reloaded, we get the old module back |
|
658 and re-use its dict to exec the new code. */ |
|
659 d = PyModule_GetDict(m); |
|
660 if (PyDict_GetItemString(d, "__builtins__") == NULL) { |
|
661 if (PyDict_SetItemString(d, "__builtins__", |
|
662 PyEval_GetBuiltins()) != 0) |
|
663 goto error; |
|
664 } |
|
665 /* Remember the filename as the __file__ attribute */ |
|
666 v = NULL; |
|
667 if (pathname != NULL) { |
|
668 v = PyString_FromString(pathname); |
|
669 if (v == NULL) |
|
670 PyErr_Clear(); |
|
671 } |
|
672 if (v == NULL) { |
|
673 v = ((PyCodeObject *)co)->co_filename; |
|
674 Py_INCREF(v); |
|
675 } |
|
676 if (PyDict_SetItemString(d, "__file__", v) != 0) |
|
677 PyErr_Clear(); /* Not important enough to report */ |
|
678 Py_DECREF(v); |
|
679 |
|
680 v = PyEval_EvalCode((PyCodeObject *)co, d, d); |
|
681 if (v == NULL) |
|
682 goto error; |
|
683 Py_DECREF(v); |
|
684 |
|
685 if ((m = PyDict_GetItemString(modules, name)) == NULL) { |
|
686 PyErr_Format(PyExc_ImportError, |
|
687 "Loaded module %.200s not found in sys.modules", |
|
688 name); |
|
689 return NULL; |
|
690 } |
|
691 |
|
692 Py_INCREF(m); |
|
693 |
|
694 return m; |
|
695 |
|
696 error: |
|
697 _RemoveModule(name); |
|
698 return NULL; |
|
699 } |
|
700 |
|
701 |
|
702 /* Given a pathname for a Python source file, fill a buffer with the |
|
703 pathname for the corresponding compiled file. Return the pathname |
|
704 for the compiled file, or NULL if there's no space in the buffer. |
|
705 Doesn't set an exception. */ |
|
706 |
|
707 static char * |
|
708 make_compiled_pathname(char *pathname, char *buf, size_t buflen) |
|
709 { |
|
710 size_t len = strlen(pathname); |
|
711 if (len+2 > buflen) |
|
712 return NULL; |
|
713 |
|
714 #ifdef MS_WINDOWS |
|
715 /* Treat .pyw as if it were .py. The case of ".pyw" must match |
|
716 that used in _PyImport_StandardFiletab. */ |
|
717 if (len >= 4 && strcmp(&pathname[len-4], ".pyw") == 0) |
|
718 --len; /* pretend 'w' isn't there */ |
|
719 #endif |
|
720 memcpy(buf, pathname, len); |
|
721 buf[len] = Py_OptimizeFlag ? 'o' : 'c'; |
|
722 buf[len+1] = '\0'; |
|
723 |
|
724 return buf; |
|
725 } |
|
726 |
|
727 |
|
728 /* Given a pathname for a Python source file, its time of last |
|
729 modification, and a pathname for a compiled file, check whether the |
|
730 compiled file represents the same version of the source. If so, |
|
731 return a FILE pointer for the compiled file, positioned just after |
|
732 the header; if not, return NULL. |
|
733 Doesn't set an exception. */ |
|
734 |
|
735 static FILE * |
|
736 check_compiled_module(char *pathname, time_t mtime, char *cpathname) |
|
737 { |
|
738 FILE *fp; |
|
739 long magic; |
|
740 long pyc_mtime; |
|
741 |
|
742 fp = fopen(cpathname, "rb"); |
|
743 if (fp == NULL) |
|
744 return NULL; |
|
745 magic = PyMarshal_ReadLongFromFile(fp); |
|
746 if (magic != pyc_magic) { |
|
747 if (Py_VerboseFlag) |
|
748 PySys_WriteStderr("# %s has bad magic\n", cpathname); |
|
749 fclose(fp); |
|
750 return NULL; |
|
751 } |
|
752 pyc_mtime = PyMarshal_ReadLongFromFile(fp); |
|
753 if (pyc_mtime != mtime) { |
|
754 if (Py_VerboseFlag) |
|
755 PySys_WriteStderr("# %s has bad mtime\n", cpathname); |
|
756 fclose(fp); |
|
757 return NULL; |
|
758 } |
|
759 if (Py_VerboseFlag) |
|
760 PySys_WriteStderr("# %s matches %s\n", cpathname, pathname); |
|
761 return fp; |
|
762 } |
|
763 |
|
764 |
|
765 /* Read a code object from a file and check it for validity */ |
|
766 |
|
767 static PyCodeObject * |
|
768 read_compiled_module(char *cpathname, FILE *fp) |
|
769 { |
|
770 PyObject *co; |
|
771 |
|
772 co = PyMarshal_ReadLastObjectFromFile(fp); |
|
773 if (co == NULL) |
|
774 return NULL; |
|
775 if (!PyCode_Check(co)) { |
|
776 PyErr_Format(PyExc_ImportError, |
|
777 "Non-code object in %.200s", cpathname); |
|
778 Py_DECREF(co); |
|
779 return NULL; |
|
780 } |
|
781 return (PyCodeObject *)co; |
|
782 } |
|
783 |
|
784 |
|
785 /* Load a module from a compiled file, execute it, and return its |
|
786 module object WITH INCREMENTED REFERENCE COUNT */ |
|
787 |
|
788 static PyObject * |
|
789 load_compiled_module(char *name, char *cpathname, FILE *fp) |
|
790 { |
|
791 long magic; |
|
792 PyCodeObject *co; |
|
793 PyObject *m; |
|
794 |
|
795 magic = PyMarshal_ReadLongFromFile(fp); |
|
796 if (magic != pyc_magic) { |
|
797 PyErr_Format(PyExc_ImportError, |
|
798 "Bad magic number in %.200s", cpathname); |
|
799 return NULL; |
|
800 } |
|
801 (void) PyMarshal_ReadLongFromFile(fp); |
|
802 co = read_compiled_module(cpathname, fp); |
|
803 if (co == NULL) |
|
804 return NULL; |
|
805 if (Py_VerboseFlag) |
|
806 PySys_WriteStderr("import %s # precompiled from %s\n", |
|
807 name, cpathname); |
|
808 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, cpathname); |
|
809 Py_DECREF(co); |
|
810 |
|
811 return m; |
|
812 } |
|
813 |
|
814 /* Parse a source file and return the corresponding code object */ |
|
815 |
|
816 static PyCodeObject * |
|
817 parse_source_module(const char *pathname, FILE *fp) |
|
818 { |
|
819 PyCodeObject *co = NULL; |
|
820 mod_ty mod; |
|
821 PyCompilerFlags flags; |
|
822 PyArena *arena = PyArena_New(); |
|
823 if (arena == NULL) |
|
824 return NULL; |
|
825 |
|
826 flags.cf_flags = 0; |
|
827 |
|
828 mod = PyParser_ASTFromFile(fp, pathname, Py_file_input, 0, 0, &flags, |
|
829 NULL, arena); |
|
830 if (mod) { |
|
831 co = PyAST_Compile(mod, pathname, NULL, arena); |
|
832 } |
|
833 PyArena_Free(arena); |
|
834 return co; |
|
835 } |
|
836 |
|
837 |
|
838 /* Helper to open a bytecode file for writing in exclusive mode */ |
|
839 |
|
840 static FILE * |
|
841 open_exclusive(char *filename, mode_t mode) |
|
842 { |
|
843 #if defined(O_EXCL)&&defined(O_CREAT)&&defined(O_WRONLY)&&defined(O_TRUNC) |
|
844 /* Use O_EXCL to avoid a race condition when another process tries to |
|
845 write the same file. When that happens, our open() call fails, |
|
846 which is just fine (since it's only a cache). |
|
847 XXX If the file exists and is writable but the directory is not |
|
848 writable, the file will never be written. Oh well. |
|
849 */ |
|
850 int fd; |
|
851 (void) unlink(filename); |
|
852 fd = open(filename, O_EXCL|O_CREAT|O_WRONLY|O_TRUNC |
|
853 #ifdef O_BINARY |
|
854 |O_BINARY /* necessary for Windows */ |
|
855 #endif |
|
856 #ifdef __VMS |
|
857 , mode, "ctxt=bin", "shr=nil" |
|
858 #else |
|
859 , mode |
|
860 #endif |
|
861 ); |
|
862 if (fd < 0) |
|
863 return NULL; |
|
864 return fdopen(fd, "wb"); |
|
865 #else |
|
866 /* Best we can do -- on Windows this can't happen anyway */ |
|
867 return fopen(filename, "wb"); |
|
868 #endif |
|
869 } |
|
870 |
|
871 |
|
872 /* Write a compiled module to a file, placing the time of last |
|
873 modification of its source into the header. |
|
874 Errors are ignored, if a write error occurs an attempt is made to |
|
875 remove the file. */ |
|
876 |
|
877 static void |
|
878 write_compiled_module(PyCodeObject *co, char *cpathname, struct stat *srcstat) |
|
879 { |
|
880 FILE *fp; |
|
881 time_t mtime = srcstat->st_mtime; |
|
882 mode_t mode = srcstat->st_mode; |
|
883 |
|
884 fp = open_exclusive(cpathname, mode); |
|
885 if (fp == NULL) { |
|
886 if (Py_VerboseFlag) |
|
887 PySys_WriteStderr( |
|
888 "# can't create %s\n", cpathname); |
|
889 return; |
|
890 } |
|
891 PyMarshal_WriteLongToFile(pyc_magic, fp, Py_MARSHAL_VERSION); |
|
892 /* First write a 0 for mtime */ |
|
893 PyMarshal_WriteLongToFile(0L, fp, Py_MARSHAL_VERSION); |
|
894 PyMarshal_WriteObjectToFile((PyObject *)co, fp, Py_MARSHAL_VERSION); |
|
895 if (fflush(fp) != 0 || ferror(fp)) { |
|
896 if (Py_VerboseFlag) |
|
897 PySys_WriteStderr("# can't write %s\n", cpathname); |
|
898 /* Don't keep partial file */ |
|
899 fclose(fp); |
|
900 (void) unlink(cpathname); |
|
901 return; |
|
902 } |
|
903 /* Now write the true mtime */ |
|
904 fseek(fp, 4L, 0); |
|
905 assert(mtime < LONG_MAX); |
|
906 PyMarshal_WriteLongToFile((long)mtime, fp, Py_MARSHAL_VERSION); |
|
907 fflush(fp); |
|
908 fclose(fp); |
|
909 if (Py_VerboseFlag) |
|
910 PySys_WriteStderr("# wrote %s\n", cpathname); |
|
911 } |
|
912 |
|
913 |
|
914 /* Load a source module from a given file and return its module |
|
915 object WITH INCREMENTED REFERENCE COUNT. If there's a matching |
|
916 byte-compiled file, use that instead. */ |
|
917 |
|
918 static PyObject * |
|
919 load_source_module(char *name, char *pathname, FILE *fp) |
|
920 { |
|
921 struct stat st; |
|
922 FILE *fpc; |
|
923 char buf[MAXPATHLEN+1]; |
|
924 char *cpathname; |
|
925 PyCodeObject *co; |
|
926 PyObject *m; |
|
927 |
|
928 if (fstat(fileno(fp), &st) != 0) { |
|
929 PyErr_Format(PyExc_RuntimeError, |
|
930 "unable to get file status from '%s'", |
|
931 pathname); |
|
932 return NULL; |
|
933 } |
|
934 #if SIZEOF_TIME_T > 4 |
|
935 /* Python's .pyc timestamp handling presumes that the timestamp fits |
|
936 in 4 bytes. This will be fine until sometime in the year 2038, |
|
937 when a 4-byte signed time_t will overflow. |
|
938 */ |
|
939 if (st.st_mtime >> 32) { |
|
940 PyErr_SetString(PyExc_OverflowError, |
|
941 "modification time overflows a 4 byte field"); |
|
942 return NULL; |
|
943 } |
|
944 #endif |
|
945 cpathname = make_compiled_pathname(pathname, buf, |
|
946 (size_t)MAXPATHLEN + 1); |
|
947 if (cpathname != NULL && |
|
948 (fpc = check_compiled_module(pathname, st.st_mtime, cpathname))) { |
|
949 co = read_compiled_module(cpathname, fpc); |
|
950 fclose(fpc); |
|
951 if (co == NULL) |
|
952 return NULL; |
|
953 if (Py_VerboseFlag) |
|
954 PySys_WriteStderr("import %s # precompiled from %s\n", |
|
955 name, cpathname); |
|
956 pathname = cpathname; |
|
957 } |
|
958 else { |
|
959 co = parse_source_module(pathname, fp); |
|
960 if (co == NULL) |
|
961 return NULL; |
|
962 if (Py_VerboseFlag) |
|
963 PySys_WriteStderr("import %s # from %s\n", |
|
964 name, pathname); |
|
965 if (cpathname) { |
|
966 PyObject *ro = PySys_GetObject("dont_write_bytecode"); |
|
967 if (ro == NULL || !PyObject_IsTrue(ro)) |
|
968 write_compiled_module(co, cpathname, &st); |
|
969 } |
|
970 } |
|
971 m = PyImport_ExecCodeModuleEx(name, (PyObject *)co, pathname); |
|
972 Py_DECREF(co); |
|
973 |
|
974 return m; |
|
975 } |
|
976 |
|
977 |
|
978 /* Forward */ |
|
979 static PyObject *load_module(char *, FILE *, char *, int, PyObject *); |
|
980 static struct filedescr *find_module(char *, char *, PyObject *, |
|
981 char *, size_t, FILE **, PyObject **); |
|
982 static struct _frozen *find_frozen(char *name); |
|
983 |
|
984 /* Load a package and return its module object WITH INCREMENTED |
|
985 REFERENCE COUNT */ |
|
986 |
|
987 static PyObject * |
|
988 load_package(char *name, char *pathname) |
|
989 { |
|
990 PyObject *m, *d; |
|
991 PyObject *file = NULL; |
|
992 PyObject *path = NULL; |
|
993 int err; |
|
994 char buf[MAXPATHLEN+1]; |
|
995 FILE *fp = NULL; |
|
996 struct filedescr *fdp; |
|
997 |
|
998 m = PyImport_AddModule(name); |
|
999 if (m == NULL) |
|
1000 return NULL; |
|
1001 if (Py_VerboseFlag) |
|
1002 PySys_WriteStderr("import %s # directory %s\n", |
|
1003 name, pathname); |
|
1004 d = PyModule_GetDict(m); |
|
1005 file = PyString_FromString(pathname); |
|
1006 if (file == NULL) |
|
1007 goto error; |
|
1008 path = Py_BuildValue("[O]", file); |
|
1009 if (path == NULL) |
|
1010 goto error; |
|
1011 err = PyDict_SetItemString(d, "__file__", file); |
|
1012 if (err == 0) |
|
1013 err = PyDict_SetItemString(d, "__path__", path); |
|
1014 if (err != 0) |
|
1015 goto error; |
|
1016 buf[0] = '\0'; |
|
1017 fdp = find_module(name, "__init__", path, buf, sizeof(buf), &fp, NULL); |
|
1018 if (fdp == NULL) { |
|
1019 if (PyErr_ExceptionMatches(PyExc_ImportError)) { |
|
1020 PyErr_Clear(); |
|
1021 Py_INCREF(m); |
|
1022 } |
|
1023 else |
|
1024 m = NULL; |
|
1025 goto cleanup; |
|
1026 } |
|
1027 m = load_module(name, fp, buf, fdp->type, NULL); |
|
1028 if (fp != NULL) |
|
1029 fclose(fp); |
|
1030 goto cleanup; |
|
1031 |
|
1032 error: |
|
1033 m = NULL; |
|
1034 cleanup: |
|
1035 Py_XDECREF(path); |
|
1036 Py_XDECREF(file); |
|
1037 return m; |
|
1038 } |
|
1039 |
|
1040 |
|
1041 /* Helper to test for built-in module */ |
|
1042 |
|
1043 static int |
|
1044 is_builtin(char *name) |
|
1045 { |
|
1046 int i; |
|
1047 for (i = 0; PyImport_Inittab[i].name != NULL; i++) { |
|
1048 if (strcmp(name, PyImport_Inittab[i].name) == 0) { |
|
1049 if (PyImport_Inittab[i].initfunc == NULL) |
|
1050 return -1; |
|
1051 else |
|
1052 return 1; |
|
1053 } |
|
1054 } |
|
1055 return 0; |
|
1056 } |
|
1057 |
|
1058 |
|
1059 /* Return an importer object for a sys.path/pkg.__path__ item 'p', |
|
1060 possibly by fetching it from the path_importer_cache dict. If it |
|
1061 wasn't yet cached, traverse path_hooks until a hook is found |
|
1062 that can handle the path item. Return None if no hook could; |
|
1063 this tells our caller it should fall back to the builtin |
|
1064 import mechanism. Cache the result in path_importer_cache. |
|
1065 Returns a borrowed reference. */ |
|
1066 |
|
1067 static PyObject * |
|
1068 get_path_importer(PyObject *path_importer_cache, PyObject *path_hooks, |
|
1069 PyObject *p) |
|
1070 { |
|
1071 PyObject *importer; |
|
1072 Py_ssize_t j, nhooks; |
|
1073 |
|
1074 /* These conditions are the caller's responsibility: */ |
|
1075 assert(PyList_Check(path_hooks)); |
|
1076 assert(PyDict_Check(path_importer_cache)); |
|
1077 |
|
1078 nhooks = PyList_Size(path_hooks); |
|
1079 if (nhooks < 0) |
|
1080 return NULL; /* Shouldn't happen */ |
|
1081 |
|
1082 importer = PyDict_GetItem(path_importer_cache, p); |
|
1083 if (importer != NULL) |
|
1084 return importer; |
|
1085 |
|
1086 /* set path_importer_cache[p] to None to avoid recursion */ |
|
1087 if (PyDict_SetItem(path_importer_cache, p, Py_None) != 0) |
|
1088 return NULL; |
|
1089 |
|
1090 for (j = 0; j < nhooks; j++) { |
|
1091 PyObject *hook = PyList_GetItem(path_hooks, j); |
|
1092 if (hook == NULL) |
|
1093 return NULL; |
|
1094 importer = PyObject_CallFunctionObjArgs(hook, p, NULL); |
|
1095 if (importer != NULL) |
|
1096 break; |
|
1097 |
|
1098 if (!PyErr_ExceptionMatches(PyExc_ImportError)) { |
|
1099 return NULL; |
|
1100 } |
|
1101 PyErr_Clear(); |
|
1102 } |
|
1103 if (importer == NULL) { |
|
1104 importer = PyObject_CallFunctionObjArgs( |
|
1105 (PyObject *)&PyNullImporter_Type, p, NULL |
|
1106 ); |
|
1107 if (importer == NULL) { |
|
1108 if (PyErr_ExceptionMatches(PyExc_ImportError)) { |
|
1109 PyErr_Clear(); |
|
1110 return Py_None; |
|
1111 } |
|
1112 } |
|
1113 } |
|
1114 if (importer != NULL) { |
|
1115 int err = PyDict_SetItem(path_importer_cache, p, importer); |
|
1116 Py_DECREF(importer); |
|
1117 if (err != 0) |
|
1118 return NULL; |
|
1119 } |
|
1120 return importer; |
|
1121 } |
|
1122 |
|
1123 PyAPI_FUNC(PyObject *) |
|
1124 PyImport_GetImporter(PyObject *path) { |
|
1125 PyObject *importer=NULL, *path_importer_cache=NULL, *path_hooks=NULL; |
|
1126 |
|
1127 if ((path_importer_cache = PySys_GetObject("path_importer_cache"))) { |
|
1128 if ((path_hooks = PySys_GetObject("path_hooks"))) { |
|
1129 importer = get_path_importer(path_importer_cache, |
|
1130 path_hooks, path); |
|
1131 } |
|
1132 } |
|
1133 Py_XINCREF(importer); /* get_path_importer returns a borrowed reference */ |
|
1134 return importer; |
|
1135 } |
|
1136 |
|
1137 /* Search the path (default sys.path) for a module. Return the |
|
1138 corresponding filedescr struct, and (via return arguments) the |
|
1139 pathname and an open file. Return NULL if the module is not found. */ |
|
1140 |
|
1141 #ifdef MS_COREDLL |
|
1142 extern FILE *PyWin_FindRegisteredModule(const char *, struct filedescr **, |
|
1143 char *, Py_ssize_t); |
|
1144 #endif |
|
1145 |
|
1146 static int case_ok(char *, Py_ssize_t, Py_ssize_t, char *); |
|
1147 static int find_init_module(char *); /* Forward */ |
|
1148 static struct filedescr importhookdescr = {"", "", IMP_HOOK}; |
|
1149 |
|
1150 static struct filedescr * |
|
1151 find_module(char *fullname, char *subname, PyObject *path, char *buf, |
|
1152 size_t buflen, FILE **p_fp, PyObject **p_loader) |
|
1153 { |
|
1154 Py_ssize_t i, npath; |
|
1155 size_t len, namelen; |
|
1156 struct filedescr *fdp = NULL; |
|
1157 char *filemode; |
|
1158 FILE *fp = NULL; |
|
1159 PyObject *path_hooks, *path_importer_cache; |
|
1160 #ifndef RISCOS |
|
1161 struct stat statbuf; |
|
1162 #endif |
|
1163 static struct filedescr fd_frozen = {"", "", PY_FROZEN}; |
|
1164 static struct filedescr fd_builtin = {"", "", C_BUILTIN}; |
|
1165 static struct filedescr fd_package = {"", "", PKG_DIRECTORY}; |
|
1166 char name[MAXPATHLEN+1]; |
|
1167 #if defined(PYOS_OS2) |
|
1168 size_t saved_len; |
|
1169 size_t saved_namelen; |
|
1170 char *saved_buf = NULL; |
|
1171 #endif |
|
1172 if (p_loader != NULL) |
|
1173 *p_loader = NULL; |
|
1174 |
|
1175 if (strlen(subname) > MAXPATHLEN) { |
|
1176 PyErr_SetString(PyExc_OverflowError, |
|
1177 "module name is too long"); |
|
1178 return NULL; |
|
1179 } |
|
1180 strcpy(name, subname); |
|
1181 |
|
1182 /* sys.meta_path import hook */ |
|
1183 if (p_loader != NULL) { |
|
1184 PyObject *meta_path; |
|
1185 |
|
1186 meta_path = PySys_GetObject("meta_path"); |
|
1187 if (meta_path == NULL || !PyList_Check(meta_path)) { |
|
1188 PyErr_SetString(PyExc_ImportError, |
|
1189 "sys.meta_path must be a list of " |
|
1190 "import hooks"); |
|
1191 return NULL; |
|
1192 } |
|
1193 Py_INCREF(meta_path); /* zap guard */ |
|
1194 npath = PyList_Size(meta_path); |
|
1195 for (i = 0; i < npath; i++) { |
|
1196 PyObject *loader; |
|
1197 PyObject *hook = PyList_GetItem(meta_path, i); |
|
1198 loader = PyObject_CallMethod(hook, "find_module", |
|
1199 "sO", fullname, |
|
1200 path != NULL ? |
|
1201 path : Py_None); |
|
1202 if (loader == NULL) { |
|
1203 Py_DECREF(meta_path); |
|
1204 return NULL; /* true error */ |
|
1205 } |
|
1206 if (loader != Py_None) { |
|
1207 /* a loader was found */ |
|
1208 *p_loader = loader; |
|
1209 Py_DECREF(meta_path); |
|
1210 return &importhookdescr; |
|
1211 } |
|
1212 Py_DECREF(loader); |
|
1213 } |
|
1214 Py_DECREF(meta_path); |
|
1215 } |
|
1216 |
|
1217 if (path != NULL && PyString_Check(path)) { |
|
1218 /* The only type of submodule allowed inside a "frozen" |
|
1219 package are other frozen modules or packages. */ |
|
1220 if (PyString_Size(path) + 1 + strlen(name) >= (size_t)buflen) { |
|
1221 PyErr_SetString(PyExc_ImportError, |
|
1222 "full frozen module name too long"); |
|
1223 return NULL; |
|
1224 } |
|
1225 strcpy(buf, PyString_AsString(path)); |
|
1226 strcat(buf, "."); |
|
1227 strcat(buf, name); |
|
1228 strcpy(name, buf); |
|
1229 if (find_frozen(name) != NULL) { |
|
1230 strcpy(buf, name); |
|
1231 return &fd_frozen; |
|
1232 } |
|
1233 PyErr_Format(PyExc_ImportError, |
|
1234 "No frozen submodule named %.200s", name); |
|
1235 return NULL; |
|
1236 } |
|
1237 if (path == NULL) { |
|
1238 if (is_builtin(name)) { |
|
1239 strcpy(buf, name); |
|
1240 return &fd_builtin; |
|
1241 } |
|
1242 if ((find_frozen(name)) != NULL) { |
|
1243 strcpy(buf, name); |
|
1244 return &fd_frozen; |
|
1245 } |
|
1246 |
|
1247 #ifdef MS_COREDLL |
|
1248 fp = PyWin_FindRegisteredModule(name, &fdp, buf, buflen); |
|
1249 if (fp != NULL) { |
|
1250 *p_fp = fp; |
|
1251 return fdp; |
|
1252 } |
|
1253 #endif |
|
1254 path = PySys_GetObject("path"); |
|
1255 } |
|
1256 if (path == NULL || !PyList_Check(path)) { |
|
1257 PyErr_SetString(PyExc_ImportError, |
|
1258 "sys.path must be a list of directory names"); |
|
1259 return NULL; |
|
1260 } |
|
1261 |
|
1262 path_hooks = PySys_GetObject("path_hooks"); |
|
1263 if (path_hooks == NULL || !PyList_Check(path_hooks)) { |
|
1264 PyErr_SetString(PyExc_ImportError, |
|
1265 "sys.path_hooks must be a list of " |
|
1266 "import hooks"); |
|
1267 return NULL; |
|
1268 } |
|
1269 path_importer_cache = PySys_GetObject("path_importer_cache"); |
|
1270 if (path_importer_cache == NULL || |
|
1271 !PyDict_Check(path_importer_cache)) { |
|
1272 PyErr_SetString(PyExc_ImportError, |
|
1273 "sys.path_importer_cache must be a dict"); |
|
1274 return NULL; |
|
1275 } |
|
1276 |
|
1277 npath = PyList_Size(path); |
|
1278 namelen = strlen(name); |
|
1279 for (i = 0; i < npath; i++) { |
|
1280 PyObject *copy = NULL; |
|
1281 PyObject *v = PyList_GetItem(path, i); |
|
1282 if (!v) |
|
1283 return NULL; |
|
1284 #ifdef Py_USING_UNICODE |
|
1285 if (PyUnicode_Check(v)) { |
|
1286 copy = PyUnicode_Encode(PyUnicode_AS_UNICODE(v), |
|
1287 PyUnicode_GET_SIZE(v), Py_FileSystemDefaultEncoding, NULL); |
|
1288 if (copy == NULL) |
|
1289 return NULL; |
|
1290 v = copy; |
|
1291 } |
|
1292 else |
|
1293 #endif |
|
1294 if (!PyString_Check(v)) |
|
1295 continue; |
|
1296 len = PyString_GET_SIZE(v); |
|
1297 if (len + 2 + namelen + MAXSUFFIXSIZE >= buflen) { |
|
1298 Py_XDECREF(copy); |
|
1299 continue; /* Too long */ |
|
1300 } |
|
1301 strcpy(buf, PyString_AS_STRING(v)); |
|
1302 if (strlen(buf) != len) { |
|
1303 Py_XDECREF(copy); |
|
1304 continue; /* v contains '\0' */ |
|
1305 } |
|
1306 |
|
1307 /* sys.path_hooks import hook */ |
|
1308 if (p_loader != NULL) { |
|
1309 PyObject *importer; |
|
1310 |
|
1311 importer = get_path_importer(path_importer_cache, |
|
1312 path_hooks, v); |
|
1313 if (importer == NULL) { |
|
1314 Py_XDECREF(copy); |
|
1315 return NULL; |
|
1316 } |
|
1317 /* Note: importer is a borrowed reference */ |
|
1318 if (importer != Py_None) { |
|
1319 PyObject *loader; |
|
1320 loader = PyObject_CallMethod(importer, |
|
1321 "find_module", |
|
1322 "s", fullname); |
|
1323 Py_XDECREF(copy); |
|
1324 if (loader == NULL) |
|
1325 return NULL; /* error */ |
|
1326 if (loader != Py_None) { |
|
1327 /* a loader was found */ |
|
1328 *p_loader = loader; |
|
1329 return &importhookdescr; |
|
1330 } |
|
1331 Py_DECREF(loader); |
|
1332 continue; |
|
1333 } |
|
1334 } |
|
1335 /* no hook was found, use builtin import */ |
|
1336 |
|
1337 if (len > 0 && buf[len-1] != SEP |
|
1338 #ifdef ALTSEP |
|
1339 && buf[len-1] != ALTSEP |
|
1340 #endif |
|
1341 ) |
|
1342 buf[len++] = SEP; |
|
1343 strcpy(buf+len, name); |
|
1344 len += namelen; |
|
1345 |
|
1346 /* Check for package import (buf holds a directory name, |
|
1347 and there's an __init__ module in that directory */ |
|
1348 #ifdef HAVE_STAT |
|
1349 if (stat(buf, &statbuf) == 0 && /* it exists */ |
|
1350 S_ISDIR(statbuf.st_mode) && /* it's a directory */ |
|
1351 case_ok(buf, len, namelen, name)) { /* case matches */ |
|
1352 if (find_init_module(buf)) { /* and has __init__.py */ |
|
1353 Py_XDECREF(copy); |
|
1354 return &fd_package; |
|
1355 } |
|
1356 else { |
|
1357 char warnstr[MAXPATHLEN+80]; |
|
1358 sprintf(warnstr, "Not importing directory " |
|
1359 "'%.*s': missing __init__.py", |
|
1360 MAXPATHLEN, buf); |
|
1361 if (PyErr_Warn(PyExc_ImportWarning, |
|
1362 warnstr)) { |
|
1363 Py_XDECREF(copy); |
|
1364 return NULL; |
|
1365 } |
|
1366 } |
|
1367 } |
|
1368 #else |
|
1369 /* XXX How are you going to test for directories? */ |
|
1370 #ifdef RISCOS |
|
1371 if (isdir(buf) && |
|
1372 case_ok(buf, len, namelen, name)) { |
|
1373 if (find_init_module(buf)) { |
|
1374 Py_XDECREF(copy); |
|
1375 return &fd_package; |
|
1376 } |
|
1377 else { |
|
1378 char warnstr[MAXPATHLEN+80]; |
|
1379 sprintf(warnstr, "Not importing directory " |
|
1380 "'%.*s': missing __init__.py", |
|
1381 MAXPATHLEN, buf); |
|
1382 if (PyErr_Warn(PyExc_ImportWarning, |
|
1383 warnstr)) { |
|
1384 Py_XDECREF(copy); |
|
1385 return NULL; |
|
1386 } |
|
1387 } |
|
1388 #endif |
|
1389 #endif |
|
1390 #if defined(PYOS_OS2) |
|
1391 /* take a snapshot of the module spec for restoration |
|
1392 * after the 8 character DLL hackery |
|
1393 */ |
|
1394 saved_buf = strdup(buf); |
|
1395 saved_len = len; |
|
1396 saved_namelen = namelen; |
|
1397 #endif /* PYOS_OS2 */ |
|
1398 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) { |
|
1399 #if defined(PYOS_OS2) && defined(HAVE_DYNAMIC_LOADING) |
|
1400 /* OS/2 limits DLLs to 8 character names (w/o |
|
1401 extension) |
|
1402 * so if the name is longer than that and its a |
|
1403 * dynamically loaded module we're going to try, |
|
1404 * truncate the name before trying |
|
1405 */ |
|
1406 if (strlen(subname) > 8) { |
|
1407 /* is this an attempt to load a C extension? */ |
|
1408 const struct filedescr *scan; |
|
1409 scan = _PyImport_DynLoadFiletab; |
|
1410 while (scan->suffix != NULL) { |
|
1411 if (!strcmp(scan->suffix, fdp->suffix)) |
|
1412 break; |
|
1413 else |
|
1414 scan++; |
|
1415 } |
|
1416 if (scan->suffix != NULL) { |
|
1417 /* yes, so truncate the name */ |
|
1418 namelen = 8; |
|
1419 len -= strlen(subname) - namelen; |
|
1420 buf[len] = '\0'; |
|
1421 } |
|
1422 } |
|
1423 #endif /* PYOS_OS2 */ |
|
1424 strcpy(buf+len, fdp->suffix); |
|
1425 if (Py_VerboseFlag > 1) |
|
1426 PySys_WriteStderr("# trying %s\n", buf); |
|
1427 filemode = fdp->mode; |
|
1428 if (filemode[0] == 'U') |
|
1429 filemode = "r" PY_STDIOTEXTMODE; |
|
1430 fp = fopen(buf, filemode); |
|
1431 if (fp != NULL) { |
|
1432 if (case_ok(buf, len, namelen, name)) |
|
1433 break; |
|
1434 else { /* continue search */ |
|
1435 fclose(fp); |
|
1436 fp = NULL; |
|
1437 } |
|
1438 } |
|
1439 #if defined(PYOS_OS2) |
|
1440 /* restore the saved snapshot */ |
|
1441 strcpy(buf, saved_buf); |
|
1442 len = saved_len; |
|
1443 namelen = saved_namelen; |
|
1444 #endif |
|
1445 } |
|
1446 #if defined(PYOS_OS2) |
|
1447 /* don't need/want the module name snapshot anymore */ |
|
1448 if (saved_buf) |
|
1449 { |
|
1450 free(saved_buf); |
|
1451 saved_buf = NULL; |
|
1452 } |
|
1453 #endif |
|
1454 Py_XDECREF(copy); |
|
1455 if (fp != NULL) |
|
1456 break; |
|
1457 } |
|
1458 if (fp == NULL) { |
|
1459 PyErr_Format(PyExc_ImportError, |
|
1460 "No module named %.200s", name); |
|
1461 return NULL; |
|
1462 } |
|
1463 *p_fp = fp; |
|
1464 return fdp; |
|
1465 } |
|
1466 |
|
1467 /* Helpers for main.c |
|
1468 * Find the source file corresponding to a named module |
|
1469 */ |
|
1470 struct filedescr * |
|
1471 _PyImport_FindModule(const char *name, PyObject *path, char *buf, |
|
1472 size_t buflen, FILE **p_fp, PyObject **p_loader) |
|
1473 { |
|
1474 return find_module((char *) name, (char *) name, path, |
|
1475 buf, buflen, p_fp, p_loader); |
|
1476 } |
|
1477 |
|
1478 PyAPI_FUNC(int) _PyImport_IsScript(struct filedescr * fd) |
|
1479 { |
|
1480 return fd->type == PY_SOURCE || fd->type == PY_COMPILED; |
|
1481 } |
|
1482 |
|
1483 /* case_ok(char* buf, Py_ssize_t len, Py_ssize_t namelen, char* name) |
|
1484 * The arguments here are tricky, best shown by example: |
|
1485 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0 |
|
1486 * ^ ^ ^ ^ |
|
1487 * |--------------------- buf ---------------------| |
|
1488 * |------------------- len ------------------| |
|
1489 * |------ name -------| |
|
1490 * |----- namelen -----| |
|
1491 * buf is the full path, but len only counts up to (& exclusive of) the |
|
1492 * extension. name is the module name, also exclusive of extension. |
|
1493 * |
|
1494 * We've already done a successful stat() or fopen() on buf, so know that |
|
1495 * there's some match, possibly case-insensitive. |
|
1496 * |
|
1497 * case_ok() is to return 1 if there's a case-sensitive match for |
|
1498 * name, else 0. case_ok() is also to return 1 if envar PYTHONCASEOK |
|
1499 * exists. |
|
1500 * |
|
1501 * case_ok() is used to implement case-sensitive import semantics even |
|
1502 * on platforms with case-insensitive filesystems. It's trivial to implement |
|
1503 * for case-sensitive filesystems. It's pretty much a cross-platform |
|
1504 * nightmare for systems with case-insensitive filesystems. |
|
1505 */ |
|
1506 |
|
1507 /* First we may need a pile of platform-specific header files; the sequence |
|
1508 * of #if's here should match the sequence in the body of case_ok(). |
|
1509 */ |
|
1510 #if defined(MS_WINDOWS) |
|
1511 #include <windows.h> |
|
1512 |
|
1513 #elif defined(DJGPP) |
|
1514 #include <dir.h> |
|
1515 |
|
1516 #elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H) |
|
1517 #include <sys/types.h> |
|
1518 #include <dirent.h> |
|
1519 |
|
1520 #elif defined(PYOS_OS2) |
|
1521 #define INCL_DOS |
|
1522 #define INCL_DOSERRORS |
|
1523 #define INCL_NOPMAPI |
|
1524 #include <os2.h> |
|
1525 |
|
1526 #elif defined(RISCOS) |
|
1527 #include "oslib/osfscontrol.h" |
|
1528 #endif |
|
1529 |
|
1530 static int |
|
1531 case_ok(char *buf, Py_ssize_t len, Py_ssize_t namelen, char *name) |
|
1532 { |
|
1533 /* Pick a platform-specific implementation; the sequence of #if's here should |
|
1534 * match the sequence just above. |
|
1535 */ |
|
1536 |
|
1537 /* MS_WINDOWS */ |
|
1538 #if defined(MS_WINDOWS) |
|
1539 WIN32_FIND_DATA data; |
|
1540 HANDLE h; |
|
1541 |
|
1542 if (Py_GETENV("PYTHONCASEOK") != NULL) |
|
1543 return 1; |
|
1544 |
|
1545 h = FindFirstFile(buf, &data); |
|
1546 if (h == INVALID_HANDLE_VALUE) { |
|
1547 PyErr_Format(PyExc_NameError, |
|
1548 "Can't find file for module %.100s\n(filename %.300s)", |
|
1549 name, buf); |
|
1550 return 0; |
|
1551 } |
|
1552 FindClose(h); |
|
1553 return strncmp(data.cFileName, name, namelen) == 0; |
|
1554 |
|
1555 /* DJGPP */ |
|
1556 #elif defined(DJGPP) |
|
1557 struct ffblk ffblk; |
|
1558 int done; |
|
1559 |
|
1560 if (Py_GETENV("PYTHONCASEOK") != NULL) |
|
1561 return 1; |
|
1562 |
|
1563 done = findfirst(buf, &ffblk, FA_ARCH|FA_RDONLY|FA_HIDDEN|FA_DIREC); |
|
1564 if (done) { |
|
1565 PyErr_Format(PyExc_NameError, |
|
1566 "Can't find file for module %.100s\n(filename %.300s)", |
|
1567 name, buf); |
|
1568 return 0; |
|
1569 } |
|
1570 return strncmp(ffblk.ff_name, name, namelen) == 0; |
|
1571 |
|
1572 /* new-fangled macintosh (macosx) or Cygwin */ |
|
1573 #elif (defined(__MACH__) && defined(__APPLE__) || defined(__CYGWIN__)) && defined(HAVE_DIRENT_H) |
|
1574 DIR *dirp; |
|
1575 struct dirent *dp; |
|
1576 char dirname[MAXPATHLEN + 1]; |
|
1577 const int dirlen = len - namelen - 1; /* don't want trailing SEP */ |
|
1578 |
|
1579 if (Py_GETENV("PYTHONCASEOK") != NULL) |
|
1580 return 1; |
|
1581 |
|
1582 /* Copy the dir component into dirname; substitute "." if empty */ |
|
1583 if (dirlen <= 0) { |
|
1584 dirname[0] = '.'; |
|
1585 dirname[1] = '\0'; |
|
1586 } |
|
1587 else { |
|
1588 assert(dirlen <= MAXPATHLEN); |
|
1589 memcpy(dirname, buf, dirlen); |
|
1590 dirname[dirlen] = '\0'; |
|
1591 } |
|
1592 /* Open the directory and search the entries for an exact match. */ |
|
1593 dirp = opendir(dirname); |
|
1594 if (dirp) { |
|
1595 char *nameWithExt = buf + len - namelen; |
|
1596 while ((dp = readdir(dirp)) != NULL) { |
|
1597 const int thislen = |
|
1598 #ifdef _DIRENT_HAVE_D_NAMELEN |
|
1599 dp->d_namlen; |
|
1600 #else |
|
1601 strlen(dp->d_name); |
|
1602 #endif |
|
1603 if (thislen >= namelen && |
|
1604 strcmp(dp->d_name, nameWithExt) == 0) { |
|
1605 (void)closedir(dirp); |
|
1606 return 1; /* Found */ |
|
1607 } |
|
1608 } |
|
1609 (void)closedir(dirp); |
|
1610 } |
|
1611 return 0 ; /* Not found */ |
|
1612 |
|
1613 /* RISC OS */ |
|
1614 #elif defined(RISCOS) |
|
1615 char canon[MAXPATHLEN+1]; /* buffer for the canonical form of the path */ |
|
1616 char buf2[MAXPATHLEN+2]; |
|
1617 char *nameWithExt = buf+len-namelen; |
|
1618 int canonlen; |
|
1619 os_error *e; |
|
1620 |
|
1621 if (Py_GETENV("PYTHONCASEOK") != NULL) |
|
1622 return 1; |
|
1623 |
|
1624 /* workaround: |
|
1625 append wildcard, otherwise case of filename wouldn't be touched */ |
|
1626 strcpy(buf2, buf); |
|
1627 strcat(buf2, "*"); |
|
1628 |
|
1629 e = xosfscontrol_canonicalise_path(buf2,canon,0,0,MAXPATHLEN+1,&canonlen); |
|
1630 canonlen = MAXPATHLEN+1-canonlen; |
|
1631 if (e || canonlen<=0 || canonlen>(MAXPATHLEN+1) ) |
|
1632 return 0; |
|
1633 if (strcmp(nameWithExt, canon+canonlen-strlen(nameWithExt))==0) |
|
1634 return 1; /* match */ |
|
1635 |
|
1636 return 0; |
|
1637 |
|
1638 /* OS/2 */ |
|
1639 #elif defined(PYOS_OS2) |
|
1640 HDIR hdir = 1; |
|
1641 ULONG srchcnt = 1; |
|
1642 FILEFINDBUF3 ffbuf; |
|
1643 APIRET rc; |
|
1644 |
|
1645 if (Py_GETENV("PYTHONCASEOK") != NULL) |
|
1646 return 1; |
|
1647 |
|
1648 rc = DosFindFirst(buf, |
|
1649 &hdir, |
|
1650 FILE_READONLY | FILE_HIDDEN | FILE_SYSTEM | FILE_DIRECTORY, |
|
1651 &ffbuf, sizeof(ffbuf), |
|
1652 &srchcnt, |
|
1653 FIL_STANDARD); |
|
1654 if (rc != NO_ERROR) |
|
1655 return 0; |
|
1656 return strncmp(ffbuf.achName, name, namelen) == 0; |
|
1657 |
|
1658 /* assuming it's a case-sensitive filesystem, so there's nothing to do! */ |
|
1659 #else |
|
1660 return 1; |
|
1661 |
|
1662 #endif |
|
1663 } |
|
1664 |
|
1665 |
|
1666 #ifdef HAVE_STAT |
|
1667 /* Helper to look for __init__.py or __init__.py[co] in potential package */ |
|
1668 static int |
|
1669 find_init_module(char *buf) |
|
1670 { |
|
1671 const size_t save_len = strlen(buf); |
|
1672 size_t i = save_len; |
|
1673 char *pname; /* pointer to start of __init__ */ |
|
1674 struct stat statbuf; |
|
1675 |
|
1676 /* For calling case_ok(buf, len, namelen, name): |
|
1677 * /a/b/c/d/e/f/g/h/i/j/k/some_long_module_name.py\0 |
|
1678 * ^ ^ ^ ^ |
|
1679 * |--------------------- buf ---------------------| |
|
1680 * |------------------- len ------------------| |
|
1681 * |------ name -------| |
|
1682 * |----- namelen -----| |
|
1683 */ |
|
1684 if (save_len + 13 >= MAXPATHLEN) |
|
1685 return 0; |
|
1686 buf[i++] = SEP; |
|
1687 pname = buf + i; |
|
1688 strcpy(pname, "__init__.py"); |
|
1689 if (stat(buf, &statbuf) == 0) { |
|
1690 if (case_ok(buf, |
|
1691 save_len + 9, /* len("/__init__") */ |
|
1692 8, /* len("__init__") */ |
|
1693 pname)) { |
|
1694 buf[save_len] = '\0'; |
|
1695 return 1; |
|
1696 } |
|
1697 } |
|
1698 i += strlen(pname); |
|
1699 strcpy(buf+i, Py_OptimizeFlag ? "o" : "c"); |
|
1700 if (stat(buf, &statbuf) == 0) { |
|
1701 if (case_ok(buf, |
|
1702 save_len + 9, /* len("/__init__") */ |
|
1703 8, /* len("__init__") */ |
|
1704 pname)) { |
|
1705 buf[save_len] = '\0'; |
|
1706 return 1; |
|
1707 } |
|
1708 } |
|
1709 buf[save_len] = '\0'; |
|
1710 return 0; |
|
1711 } |
|
1712 |
|
1713 #else |
|
1714 |
|
1715 #ifdef RISCOS |
|
1716 static int |
|
1717 find_init_module(buf) |
|
1718 char *buf; |
|
1719 { |
|
1720 int save_len = strlen(buf); |
|
1721 int i = save_len; |
|
1722 |
|
1723 if (save_len + 13 >= MAXPATHLEN) |
|
1724 return 0; |
|
1725 buf[i++] = SEP; |
|
1726 strcpy(buf+i, "__init__/py"); |
|
1727 if (isfile(buf)) { |
|
1728 buf[save_len] = '\0'; |
|
1729 return 1; |
|
1730 } |
|
1731 |
|
1732 if (Py_OptimizeFlag) |
|
1733 strcpy(buf+i, "o"); |
|
1734 else |
|
1735 strcpy(buf+i, "c"); |
|
1736 if (isfile(buf)) { |
|
1737 buf[save_len] = '\0'; |
|
1738 return 1; |
|
1739 } |
|
1740 buf[save_len] = '\0'; |
|
1741 return 0; |
|
1742 } |
|
1743 #endif /*RISCOS*/ |
|
1744 |
|
1745 #endif /* HAVE_STAT */ |
|
1746 |
|
1747 |
|
1748 static int init_builtin(char *); /* Forward */ |
|
1749 |
|
1750 /* Load an external module using the default search path and return |
|
1751 its module object WITH INCREMENTED REFERENCE COUNT */ |
|
1752 |
|
1753 static PyObject * |
|
1754 load_module(char *name, FILE *fp, char *buf, int type, PyObject *loader) |
|
1755 { |
|
1756 PyObject *modules; |
|
1757 PyObject *m; |
|
1758 int err; |
|
1759 |
|
1760 /* First check that there's an open file (if we need one) */ |
|
1761 switch (type) { |
|
1762 case PY_SOURCE: |
|
1763 case PY_COMPILED: |
|
1764 if (fp == NULL) { |
|
1765 PyErr_Format(PyExc_ValueError, |
|
1766 "file object required for import (type code %d)", |
|
1767 type); |
|
1768 return NULL; |
|
1769 } |
|
1770 } |
|
1771 |
|
1772 switch (type) { |
|
1773 |
|
1774 case PY_SOURCE: |
|
1775 m = load_source_module(name, buf, fp); |
|
1776 break; |
|
1777 |
|
1778 case PY_COMPILED: |
|
1779 m = load_compiled_module(name, buf, fp); |
|
1780 break; |
|
1781 |
|
1782 #ifdef HAVE_DYNAMIC_LOADING |
|
1783 case C_EXTENSION: |
|
1784 m = _PyImport_LoadDynamicModule(name, buf, fp); |
|
1785 break; |
|
1786 #endif |
|
1787 |
|
1788 case PKG_DIRECTORY: |
|
1789 m = load_package(name, buf); |
|
1790 break; |
|
1791 |
|
1792 case C_BUILTIN: |
|
1793 case PY_FROZEN: |
|
1794 if (buf != NULL && buf[0] != '\0') |
|
1795 name = buf; |
|
1796 if (type == C_BUILTIN) |
|
1797 err = init_builtin(name); |
|
1798 else |
|
1799 err = PyImport_ImportFrozenModule(name); |
|
1800 if (err < 0) |
|
1801 return NULL; |
|
1802 if (err == 0) { |
|
1803 PyErr_Format(PyExc_ImportError, |
|
1804 "Purported %s module %.200s not found", |
|
1805 type == C_BUILTIN ? |
|
1806 "builtin" : "frozen", |
|
1807 name); |
|
1808 return NULL; |
|
1809 } |
|
1810 modules = PyImport_GetModuleDict(); |
|
1811 m = PyDict_GetItemString(modules, name); |
|
1812 if (m == NULL) { |
|
1813 PyErr_Format( |
|
1814 PyExc_ImportError, |
|
1815 "%s module %.200s not properly initialized", |
|
1816 type == C_BUILTIN ? |
|
1817 "builtin" : "frozen", |
|
1818 name); |
|
1819 return NULL; |
|
1820 } |
|
1821 Py_INCREF(m); |
|
1822 break; |
|
1823 |
|
1824 case IMP_HOOK: { |
|
1825 if (loader == NULL) { |
|
1826 PyErr_SetString(PyExc_ImportError, |
|
1827 "import hook without loader"); |
|
1828 return NULL; |
|
1829 } |
|
1830 m = PyObject_CallMethod(loader, "load_module", "s", name); |
|
1831 break; |
|
1832 } |
|
1833 |
|
1834 default: |
|
1835 PyErr_Format(PyExc_ImportError, |
|
1836 "Don't know how to import %.200s (type code %d)", |
|
1837 name, type); |
|
1838 m = NULL; |
|
1839 |
|
1840 } |
|
1841 |
|
1842 return m; |
|
1843 } |
|
1844 |
|
1845 |
|
1846 /* Initialize a built-in module. |
|
1847 Return 1 for success, 0 if the module is not found, and -1 with |
|
1848 an exception set if the initialization failed. */ |
|
1849 |
|
1850 static int |
|
1851 init_builtin(char *name) |
|
1852 { |
|
1853 struct _inittab *p; |
|
1854 |
|
1855 if (_PyImport_FindExtension(name, name) != NULL) |
|
1856 return 1; |
|
1857 |
|
1858 for (p = PyImport_Inittab; p->name != NULL; p++) { |
|
1859 if (strcmp(name, p->name) == 0) { |
|
1860 if (p->initfunc == NULL) { |
|
1861 PyErr_Format(PyExc_ImportError, |
|
1862 "Cannot re-init internal module %.200s", |
|
1863 name); |
|
1864 return -1; |
|
1865 } |
|
1866 if (Py_VerboseFlag) |
|
1867 PySys_WriteStderr("import %s # builtin\n", name); |
|
1868 (*p->initfunc)(); |
|
1869 if (PyErr_Occurred()) |
|
1870 return -1; |
|
1871 if (_PyImport_FixupExtension(name, name) == NULL) |
|
1872 return -1; |
|
1873 return 1; |
|
1874 } |
|
1875 } |
|
1876 return 0; |
|
1877 } |
|
1878 |
|
1879 |
|
1880 /* Frozen modules */ |
|
1881 |
|
1882 static struct _frozen * |
|
1883 find_frozen(char *name) |
|
1884 { |
|
1885 struct _frozen *p; |
|
1886 |
|
1887 for (p = PyImport_FrozenModules; ; p++) { |
|
1888 if (p->name == NULL) |
|
1889 return NULL; |
|
1890 if (strcmp(p->name, name) == 0) |
|
1891 break; |
|
1892 } |
|
1893 return p; |
|
1894 } |
|
1895 |
|
1896 static PyObject * |
|
1897 get_frozen_object(char *name) |
|
1898 { |
|
1899 struct _frozen *p = find_frozen(name); |
|
1900 int size; |
|
1901 |
|
1902 if (p == NULL) { |
|
1903 PyErr_Format(PyExc_ImportError, |
|
1904 "No such frozen object named %.200s", |
|
1905 name); |
|
1906 return NULL; |
|
1907 } |
|
1908 if (p->code == NULL) { |
|
1909 PyErr_Format(PyExc_ImportError, |
|
1910 "Excluded frozen object named %.200s", |
|
1911 name); |
|
1912 return NULL; |
|
1913 } |
|
1914 size = p->size; |
|
1915 if (size < 0) |
|
1916 size = -size; |
|
1917 return PyMarshal_ReadObjectFromString((char *)p->code, size); |
|
1918 } |
|
1919 |
|
1920 /* Initialize a frozen module. |
|
1921 Return 1 for succes, 0 if the module is not found, and -1 with |
|
1922 an exception set if the initialization failed. |
|
1923 This function is also used from frozenmain.c */ |
|
1924 |
|
1925 int |
|
1926 PyImport_ImportFrozenModule(char *name) |
|
1927 { |
|
1928 struct _frozen *p = find_frozen(name); |
|
1929 PyObject *co; |
|
1930 PyObject *m; |
|
1931 int ispackage; |
|
1932 int size; |
|
1933 |
|
1934 if (p == NULL) |
|
1935 return 0; |
|
1936 if (p->code == NULL) { |
|
1937 PyErr_Format(PyExc_ImportError, |
|
1938 "Excluded frozen object named %.200s", |
|
1939 name); |
|
1940 return -1; |
|
1941 } |
|
1942 size = p->size; |
|
1943 ispackage = (size < 0); |
|
1944 if (ispackage) |
|
1945 size = -size; |
|
1946 if (Py_VerboseFlag) |
|
1947 PySys_WriteStderr("import %s # frozen%s\n", |
|
1948 name, ispackage ? " package" : ""); |
|
1949 co = PyMarshal_ReadObjectFromString((char *)p->code, size); |
|
1950 if (co == NULL) |
|
1951 return -1; |
|
1952 if (!PyCode_Check(co)) { |
|
1953 PyErr_Format(PyExc_TypeError, |
|
1954 "frozen object %.200s is not a code object", |
|
1955 name); |
|
1956 goto err_return; |
|
1957 } |
|
1958 if (ispackage) { |
|
1959 /* Set __path__ to the package name */ |
|
1960 PyObject *d, *s; |
|
1961 int err; |
|
1962 m = PyImport_AddModule(name); |
|
1963 if (m == NULL) |
|
1964 goto err_return; |
|
1965 d = PyModule_GetDict(m); |
|
1966 s = PyString_InternFromString(name); |
|
1967 if (s == NULL) |
|
1968 goto err_return; |
|
1969 err = PyDict_SetItemString(d, "__path__", s); |
|
1970 Py_DECREF(s); |
|
1971 if (err != 0) |
|
1972 goto err_return; |
|
1973 } |
|
1974 m = PyImport_ExecCodeModuleEx(name, co, "<frozen>"); |
|
1975 if (m == NULL) |
|
1976 goto err_return; |
|
1977 Py_DECREF(co); |
|
1978 Py_DECREF(m); |
|
1979 return 1; |
|
1980 err_return: |
|
1981 Py_DECREF(co); |
|
1982 return -1; |
|
1983 } |
|
1984 |
|
1985 |
|
1986 /* Import a module, either built-in, frozen, or external, and return |
|
1987 its module object WITH INCREMENTED REFERENCE COUNT */ |
|
1988 |
|
1989 PyObject * |
|
1990 PyImport_ImportModule(const char *name) |
|
1991 { |
|
1992 PyObject *pname; |
|
1993 PyObject *result; |
|
1994 |
|
1995 pname = PyString_FromString(name); |
|
1996 if (pname == NULL) |
|
1997 return NULL; |
|
1998 result = PyImport_Import(pname); |
|
1999 Py_DECREF(pname); |
|
2000 return result; |
|
2001 } |
|
2002 |
|
2003 /* Import a module without blocking |
|
2004 * |
|
2005 * At first it tries to fetch the module from sys.modules. If the module was |
|
2006 * never loaded before it loads it with PyImport_ImportModule() unless another |
|
2007 * thread holds the import lock. In the latter case the function raises an |
|
2008 * ImportError instead of blocking. |
|
2009 * |
|
2010 * Returns the module object with incremented ref count. |
|
2011 */ |
|
2012 PyObject * |
|
2013 PyImport_ImportModuleNoBlock(const char *name) |
|
2014 { |
|
2015 PyObject *result; |
|
2016 PyObject *modules; |
|
2017 long me; |
|
2018 |
|
2019 /* Try to get the module from sys.modules[name] */ |
|
2020 modules = PyImport_GetModuleDict(); |
|
2021 if (modules == NULL) |
|
2022 return NULL; |
|
2023 |
|
2024 result = PyDict_GetItemString(modules, name); |
|
2025 if (result != NULL) { |
|
2026 Py_INCREF(result); |
|
2027 return result; |
|
2028 } |
|
2029 else { |
|
2030 PyErr_Clear(); |
|
2031 } |
|
2032 #ifdef WITH_THREAD |
|
2033 /* check the import lock |
|
2034 * me might be -1 but I ignore the error here, the lock function |
|
2035 * takes care of the problem */ |
|
2036 me = PyThread_get_thread_ident(); |
|
2037 if (import_lock_thread == -1 || import_lock_thread == me) { |
|
2038 /* no thread or me is holding the lock */ |
|
2039 return PyImport_ImportModule(name); |
|
2040 } |
|
2041 else { |
|
2042 PyErr_Format(PyExc_ImportError, |
|
2043 "Failed to import %.200s because the import lock" |
|
2044 "is held by another thread.", |
|
2045 name); |
|
2046 return NULL; |
|
2047 } |
|
2048 #else |
|
2049 return PyImport_ImportModule(name); |
|
2050 #endif |
|
2051 } |
|
2052 |
|
2053 /* Forward declarations for helper routines */ |
|
2054 static PyObject *get_parent(PyObject *globals, char *buf, |
|
2055 Py_ssize_t *p_buflen, int level); |
|
2056 static PyObject *load_next(PyObject *mod, PyObject *altmod, |
|
2057 char **p_name, char *buf, Py_ssize_t *p_buflen); |
|
2058 static int mark_miss(char *name); |
|
2059 static int ensure_fromlist(PyObject *mod, PyObject *fromlist, |
|
2060 char *buf, Py_ssize_t buflen, int recursive); |
|
2061 static PyObject * import_submodule(PyObject *mod, char *name, char *fullname); |
|
2062 |
|
2063 /* The Magnum Opus of dotted-name import :-) */ |
|
2064 |
|
2065 static PyObject * |
|
2066 import_module_level(char *name, PyObject *globals, PyObject *locals, |
|
2067 PyObject *fromlist, int level) |
|
2068 { |
|
2069 char buf[MAXPATHLEN+1]; |
|
2070 Py_ssize_t buflen = 0; |
|
2071 PyObject *parent, *head, *next, *tail; |
|
2072 |
|
2073 if (strchr(name, '/') != NULL |
|
2074 #ifdef MS_WINDOWS |
|
2075 || strchr(name, '\\') != NULL |
|
2076 #endif |
|
2077 ) { |
|
2078 PyErr_SetString(PyExc_ImportError, |
|
2079 "Import by filename is not supported."); |
|
2080 return NULL; |
|
2081 } |
|
2082 |
|
2083 parent = get_parent(globals, buf, &buflen, level); |
|
2084 if (parent == NULL) |
|
2085 return NULL; |
|
2086 |
|
2087 head = load_next(parent, Py_None, &name, buf, &buflen); |
|
2088 if (head == NULL) |
|
2089 return NULL; |
|
2090 |
|
2091 tail = head; |
|
2092 Py_INCREF(tail); |
|
2093 while (name) { |
|
2094 next = load_next(tail, tail, &name, buf, &buflen); |
|
2095 Py_DECREF(tail); |
|
2096 if (next == NULL) { |
|
2097 Py_DECREF(head); |
|
2098 return NULL; |
|
2099 } |
|
2100 tail = next; |
|
2101 } |
|
2102 if (tail == Py_None) { |
|
2103 /* If tail is Py_None, both get_parent and load_next found |
|
2104 an empty module name: someone called __import__("") or |
|
2105 doctored faulty bytecode */ |
|
2106 Py_DECREF(tail); |
|
2107 Py_DECREF(head); |
|
2108 PyErr_SetString(PyExc_ValueError, |
|
2109 "Empty module name"); |
|
2110 return NULL; |
|
2111 } |
|
2112 |
|
2113 if (fromlist != NULL) { |
|
2114 if (fromlist == Py_None || !PyObject_IsTrue(fromlist)) |
|
2115 fromlist = NULL; |
|
2116 } |
|
2117 |
|
2118 if (fromlist == NULL) { |
|
2119 Py_DECREF(tail); |
|
2120 return head; |
|
2121 } |
|
2122 |
|
2123 Py_DECREF(head); |
|
2124 if (!ensure_fromlist(tail, fromlist, buf, buflen, 0)) { |
|
2125 Py_DECREF(tail); |
|
2126 return NULL; |
|
2127 } |
|
2128 |
|
2129 return tail; |
|
2130 } |
|
2131 |
|
2132 PyObject * |
|
2133 PyImport_ImportModuleLevel(char *name, PyObject *globals, PyObject *locals, |
|
2134 PyObject *fromlist, int level) |
|
2135 { |
|
2136 PyObject *result; |
|
2137 lock_import(); |
|
2138 result = import_module_level(name, globals, locals, fromlist, level); |
|
2139 if (unlock_import() < 0) { |
|
2140 Py_XDECREF(result); |
|
2141 PyErr_SetString(PyExc_RuntimeError, |
|
2142 "not holding the import lock"); |
|
2143 return NULL; |
|
2144 } |
|
2145 return result; |
|
2146 } |
|
2147 |
|
2148 /* Return the package that an import is being performed in. If globals comes |
|
2149 from the module foo.bar.bat (not itself a package), this returns the |
|
2150 sys.modules entry for foo.bar. If globals is from a package's __init__.py, |
|
2151 the package's entry in sys.modules is returned, as a borrowed reference. |
|
2152 |
|
2153 The *name* of the returned package is returned in buf, with the length of |
|
2154 the name in *p_buflen. |
|
2155 |
|
2156 If globals doesn't come from a package or a module in a package, or a |
|
2157 corresponding entry is not found in sys.modules, Py_None is returned. |
|
2158 */ |
|
2159 static PyObject * |
|
2160 get_parent(PyObject *globals, char *buf, Py_ssize_t *p_buflen, int level) |
|
2161 { |
|
2162 static PyObject *namestr = NULL; |
|
2163 static PyObject *pathstr = NULL; |
|
2164 static PyObject *pkgstr = NULL; |
|
2165 PyObject *pkgname, *modname, *modpath, *modules, *parent; |
|
2166 int orig_level = level; |
|
2167 |
|
2168 if (globals == NULL || !PyDict_Check(globals) || !level) |
|
2169 return Py_None; |
|
2170 |
|
2171 if (namestr == NULL) { |
|
2172 namestr = PyString_InternFromString("__name__"); |
|
2173 if (namestr == NULL) |
|
2174 return NULL; |
|
2175 } |
|
2176 if (pathstr == NULL) { |
|
2177 pathstr = PyString_InternFromString("__path__"); |
|
2178 if (pathstr == NULL) |
|
2179 return NULL; |
|
2180 } |
|
2181 if (pkgstr == NULL) { |
|
2182 pkgstr = PyString_InternFromString("__package__"); |
|
2183 if (pkgstr == NULL) |
|
2184 return NULL; |
|
2185 } |
|
2186 |
|
2187 *buf = '\0'; |
|
2188 *p_buflen = 0; |
|
2189 pkgname = PyDict_GetItem(globals, pkgstr); |
|
2190 |
|
2191 if ((pkgname != NULL) && (pkgname != Py_None)) { |
|
2192 /* __package__ is set, so use it */ |
|
2193 Py_ssize_t len; |
|
2194 if (!PyString_Check(pkgname)) { |
|
2195 PyErr_SetString(PyExc_ValueError, |
|
2196 "__package__ set to non-string"); |
|
2197 return NULL; |
|
2198 } |
|
2199 len = PyString_GET_SIZE(pkgname); |
|
2200 if (len == 0) { |
|
2201 if (level > 0) { |
|
2202 PyErr_SetString(PyExc_ValueError, |
|
2203 "Attempted relative import in non-package"); |
|
2204 return NULL; |
|
2205 } |
|
2206 return Py_None; |
|
2207 } |
|
2208 if (len > MAXPATHLEN) { |
|
2209 PyErr_SetString(PyExc_ValueError, |
|
2210 "Package name too long"); |
|
2211 return NULL; |
|
2212 } |
|
2213 strcpy(buf, PyString_AS_STRING(pkgname)); |
|
2214 } else { |
|
2215 /* __package__ not set, so figure it out and set it */ |
|
2216 modname = PyDict_GetItem(globals, namestr); |
|
2217 if (modname == NULL || !PyString_Check(modname)) |
|
2218 return Py_None; |
|
2219 |
|
2220 modpath = PyDict_GetItem(globals, pathstr); |
|
2221 if (modpath != NULL) { |
|
2222 /* __path__ is set, so modname is already the package name */ |
|
2223 Py_ssize_t len = PyString_GET_SIZE(modname); |
|
2224 int error; |
|
2225 if (len > MAXPATHLEN) { |
|
2226 PyErr_SetString(PyExc_ValueError, |
|
2227 "Module name too long"); |
|
2228 return NULL; |
|
2229 } |
|
2230 strcpy(buf, PyString_AS_STRING(modname)); |
|
2231 error = PyDict_SetItem(globals, pkgstr, modname); |
|
2232 if (error) { |
|
2233 PyErr_SetString(PyExc_ValueError, |
|
2234 "Could not set __package__"); |
|
2235 return NULL; |
|
2236 } |
|
2237 } else { |
|
2238 /* Normal module, so work out the package name if any */ |
|
2239 char *start = PyString_AS_STRING(modname); |
|
2240 char *lastdot = strrchr(start, '.'); |
|
2241 size_t len; |
|
2242 int error; |
|
2243 if (lastdot == NULL && level > 0) { |
|
2244 PyErr_SetString(PyExc_ValueError, |
|
2245 "Attempted relative import in non-package"); |
|
2246 return NULL; |
|
2247 } |
|
2248 if (lastdot == NULL) { |
|
2249 error = PyDict_SetItem(globals, pkgstr, Py_None); |
|
2250 if (error) { |
|
2251 PyErr_SetString(PyExc_ValueError, |
|
2252 "Could not set __package__"); |
|
2253 return NULL; |
|
2254 } |
|
2255 return Py_None; |
|
2256 } |
|
2257 len = lastdot - start; |
|
2258 if (len >= MAXPATHLEN) { |
|
2259 PyErr_SetString(PyExc_ValueError, |
|
2260 "Module name too long"); |
|
2261 return NULL; |
|
2262 } |
|
2263 strncpy(buf, start, len); |
|
2264 buf[len] = '\0'; |
|
2265 pkgname = PyString_FromString(buf); |
|
2266 if (pkgname == NULL) { |
|
2267 return NULL; |
|
2268 } |
|
2269 error = PyDict_SetItem(globals, pkgstr, pkgname); |
|
2270 Py_DECREF(pkgname); |
|
2271 if (error) { |
|
2272 PyErr_SetString(PyExc_ValueError, |
|
2273 "Could not set __package__"); |
|
2274 return NULL; |
|
2275 } |
|
2276 } |
|
2277 } |
|
2278 while (--level > 0) { |
|
2279 char *dot = strrchr(buf, '.'); |
|
2280 if (dot == NULL) { |
|
2281 PyErr_SetString(PyExc_ValueError, |
|
2282 "Attempted relative import beyond " |
|
2283 "toplevel package"); |
|
2284 return NULL; |
|
2285 } |
|
2286 *dot = '\0'; |
|
2287 } |
|
2288 *p_buflen = strlen(buf); |
|
2289 |
|
2290 modules = PyImport_GetModuleDict(); |
|
2291 parent = PyDict_GetItemString(modules, buf); |
|
2292 if (parent == NULL) { |
|
2293 if (orig_level < 1) { |
|
2294 PyObject *err_msg = PyString_FromFormat( |
|
2295 "Parent module '%.200s' not found " |
|
2296 "while handling absolute import", buf); |
|
2297 if (err_msg == NULL) { |
|
2298 return NULL; |
|
2299 } |
|
2300 if (!PyErr_WarnEx(PyExc_RuntimeWarning, |
|
2301 PyString_AsString(err_msg), 1)) { |
|
2302 *buf = '\0'; |
|
2303 *p_buflen = 0; |
|
2304 parent = Py_None; |
|
2305 } |
|
2306 Py_DECREF(err_msg); |
|
2307 } else { |
|
2308 PyErr_Format(PyExc_SystemError, |
|
2309 "Parent module '%.200s' not loaded, " |
|
2310 "cannot perform relative import", buf); |
|
2311 } |
|
2312 } |
|
2313 return parent; |
|
2314 /* We expect, but can't guarantee, if parent != None, that: |
|
2315 - parent.__name__ == buf |
|
2316 - parent.__dict__ is globals |
|
2317 If this is violated... Who cares? */ |
|
2318 } |
|
2319 |
|
2320 /* altmod is either None or same as mod */ |
|
2321 static PyObject * |
|
2322 load_next(PyObject *mod, PyObject *altmod, char **p_name, char *buf, |
|
2323 Py_ssize_t *p_buflen) |
|
2324 { |
|
2325 char *name = *p_name; |
|
2326 char *dot = strchr(name, '.'); |
|
2327 size_t len; |
|
2328 char *p; |
|
2329 PyObject *result; |
|
2330 |
|
2331 if (strlen(name) == 0) { |
|
2332 /* completely empty module name should only happen in |
|
2333 'from . import' (or '__import__("")')*/ |
|
2334 Py_INCREF(mod); |
|
2335 *p_name = NULL; |
|
2336 return mod; |
|
2337 } |
|
2338 |
|
2339 if (dot == NULL) { |
|
2340 *p_name = NULL; |
|
2341 len = strlen(name); |
|
2342 } |
|
2343 else { |
|
2344 *p_name = dot+1; |
|
2345 len = dot-name; |
|
2346 } |
|
2347 if (len == 0) { |
|
2348 PyErr_SetString(PyExc_ValueError, |
|
2349 "Empty module name"); |
|
2350 return NULL; |
|
2351 } |
|
2352 |
|
2353 p = buf + *p_buflen; |
|
2354 if (p != buf) |
|
2355 *p++ = '.'; |
|
2356 if (p+len-buf >= MAXPATHLEN) { |
|
2357 PyErr_SetString(PyExc_ValueError, |
|
2358 "Module name too long"); |
|
2359 return NULL; |
|
2360 } |
|
2361 strncpy(p, name, len); |
|
2362 p[len] = '\0'; |
|
2363 *p_buflen = p+len-buf; |
|
2364 |
|
2365 result = import_submodule(mod, p, buf); |
|
2366 if (result == Py_None && altmod != mod) { |
|
2367 Py_DECREF(result); |
|
2368 /* Here, altmod must be None and mod must not be None */ |
|
2369 result = import_submodule(altmod, p, p); |
|
2370 if (result != NULL && result != Py_None) { |
|
2371 if (mark_miss(buf) != 0) { |
|
2372 Py_DECREF(result); |
|
2373 return NULL; |
|
2374 } |
|
2375 strncpy(buf, name, len); |
|
2376 buf[len] = '\0'; |
|
2377 *p_buflen = len; |
|
2378 } |
|
2379 } |
|
2380 if (result == NULL) |
|
2381 return NULL; |
|
2382 |
|
2383 if (result == Py_None) { |
|
2384 Py_DECREF(result); |
|
2385 PyErr_Format(PyExc_ImportError, |
|
2386 "No module named %.200s", name); |
|
2387 return NULL; |
|
2388 } |
|
2389 |
|
2390 return result; |
|
2391 } |
|
2392 |
|
2393 static int |
|
2394 mark_miss(char *name) |
|
2395 { |
|
2396 PyObject *modules = PyImport_GetModuleDict(); |
|
2397 return PyDict_SetItemString(modules, name, Py_None); |
|
2398 } |
|
2399 |
|
2400 static int |
|
2401 ensure_fromlist(PyObject *mod, PyObject *fromlist, char *buf, Py_ssize_t buflen, |
|
2402 int recursive) |
|
2403 { |
|
2404 int i; |
|
2405 |
|
2406 if (!PyObject_HasAttrString(mod, "__path__")) |
|
2407 return 1; |
|
2408 |
|
2409 for (i = 0; ; i++) { |
|
2410 PyObject *item = PySequence_GetItem(fromlist, i); |
|
2411 int hasit; |
|
2412 if (item == NULL) { |
|
2413 if (PyErr_ExceptionMatches(PyExc_IndexError)) { |
|
2414 PyErr_Clear(); |
|
2415 return 1; |
|
2416 } |
|
2417 return 0; |
|
2418 } |
|
2419 if (!PyString_Check(item)) { |
|
2420 PyErr_SetString(PyExc_TypeError, |
|
2421 "Item in ``from list'' not a string"); |
|
2422 Py_DECREF(item); |
|
2423 return 0; |
|
2424 } |
|
2425 if (PyString_AS_STRING(item)[0] == '*') { |
|
2426 PyObject *all; |
|
2427 Py_DECREF(item); |
|
2428 /* See if the package defines __all__ */ |
|
2429 if (recursive) |
|
2430 continue; /* Avoid endless recursion */ |
|
2431 all = PyObject_GetAttrString(mod, "__all__"); |
|
2432 if (all == NULL) |
|
2433 PyErr_Clear(); |
|
2434 else { |
|
2435 int ret = ensure_fromlist(mod, all, buf, buflen, 1); |
|
2436 Py_DECREF(all); |
|
2437 if (!ret) |
|
2438 return 0; |
|
2439 } |
|
2440 continue; |
|
2441 } |
|
2442 hasit = PyObject_HasAttr(mod, item); |
|
2443 if (!hasit) { |
|
2444 char *subname = PyString_AS_STRING(item); |
|
2445 PyObject *submod; |
|
2446 char *p; |
|
2447 if (buflen + strlen(subname) >= MAXPATHLEN) { |
|
2448 PyErr_SetString(PyExc_ValueError, |
|
2449 "Module name too long"); |
|
2450 Py_DECREF(item); |
|
2451 return 0; |
|
2452 } |
|
2453 p = buf + buflen; |
|
2454 *p++ = '.'; |
|
2455 strcpy(p, subname); |
|
2456 submod = import_submodule(mod, subname, buf); |
|
2457 Py_XDECREF(submod); |
|
2458 if (submod == NULL) { |
|
2459 Py_DECREF(item); |
|
2460 return 0; |
|
2461 } |
|
2462 } |
|
2463 Py_DECREF(item); |
|
2464 } |
|
2465 |
|
2466 /* NOTREACHED */ |
|
2467 } |
|
2468 |
|
2469 static int |
|
2470 add_submodule(PyObject *mod, PyObject *submod, char *fullname, char *subname, |
|
2471 PyObject *modules) |
|
2472 { |
|
2473 if (mod == Py_None) |
|
2474 return 1; |
|
2475 /* Irrespective of the success of this load, make a |
|
2476 reference to it in the parent package module. A copy gets |
|
2477 saved in the modules dictionary under the full name, so get a |
|
2478 reference from there, if need be. (The exception is when the |
|
2479 load failed with a SyntaxError -- then there's no trace in |
|
2480 sys.modules. In that case, of course, do nothing extra.) */ |
|
2481 if (submod == NULL) { |
|
2482 submod = PyDict_GetItemString(modules, fullname); |
|
2483 if (submod == NULL) |
|
2484 return 1; |
|
2485 } |
|
2486 if (PyModule_Check(mod)) { |
|
2487 /* We can't use setattr here since it can give a |
|
2488 * spurious warning if the submodule name shadows a |
|
2489 * builtin name */ |
|
2490 PyObject *dict = PyModule_GetDict(mod); |
|
2491 if (!dict) |
|
2492 return 0; |
|
2493 if (PyDict_SetItemString(dict, subname, submod) < 0) |
|
2494 return 0; |
|
2495 } |
|
2496 else { |
|
2497 if (PyObject_SetAttrString(mod, subname, submod) < 0) |
|
2498 return 0; |
|
2499 } |
|
2500 return 1; |
|
2501 } |
|
2502 |
|
2503 static PyObject * |
|
2504 import_submodule(PyObject *mod, char *subname, char *fullname) |
|
2505 { |
|
2506 PyObject *modules = PyImport_GetModuleDict(); |
|
2507 PyObject *m = NULL; |
|
2508 |
|
2509 /* Require: |
|
2510 if mod == None: subname == fullname |
|
2511 else: mod.__name__ + "." + subname == fullname |
|
2512 */ |
|
2513 |
|
2514 if ((m = PyDict_GetItemString(modules, fullname)) != NULL) { |
|
2515 Py_INCREF(m); |
|
2516 } |
|
2517 else { |
|
2518 PyObject *path, *loader = NULL; |
|
2519 char buf[MAXPATHLEN+1]; |
|
2520 struct filedescr *fdp; |
|
2521 FILE *fp = NULL; |
|
2522 |
|
2523 if (mod == Py_None) |
|
2524 path = NULL; |
|
2525 else { |
|
2526 path = PyObject_GetAttrString(mod, "__path__"); |
|
2527 if (path == NULL) { |
|
2528 PyErr_Clear(); |
|
2529 Py_INCREF(Py_None); |
|
2530 return Py_None; |
|
2531 } |
|
2532 } |
|
2533 |
|
2534 buf[0] = '\0'; |
|
2535 fdp = find_module(fullname, subname, path, buf, MAXPATHLEN+1, |
|
2536 &fp, &loader); |
|
2537 Py_XDECREF(path); |
|
2538 if (fdp == NULL) { |
|
2539 if (!PyErr_ExceptionMatches(PyExc_ImportError)) |
|
2540 return NULL; |
|
2541 PyErr_Clear(); |
|
2542 Py_INCREF(Py_None); |
|
2543 return Py_None; |
|
2544 } |
|
2545 m = load_module(fullname, fp, buf, fdp->type, loader); |
|
2546 Py_XDECREF(loader); |
|
2547 if (fp) |
|
2548 fclose(fp); |
|
2549 if (!add_submodule(mod, m, fullname, subname, modules)) { |
|
2550 Py_XDECREF(m); |
|
2551 m = NULL; |
|
2552 } |
|
2553 } |
|
2554 |
|
2555 return m; |
|
2556 } |
|
2557 |
|
2558 |
|
2559 /* Re-import a module of any kind and return its module object, WITH |
|
2560 INCREMENTED REFERENCE COUNT */ |
|
2561 |
|
2562 PyObject * |
|
2563 PyImport_ReloadModule(PyObject *m) |
|
2564 { |
|
2565 PyInterpreterState *interp = PyThreadState_Get()->interp; |
|
2566 PyObject *modules_reloading = interp->modules_reloading; |
|
2567 PyObject *modules = PyImport_GetModuleDict(); |
|
2568 PyObject *path = NULL, *loader = NULL, *existing_m = NULL; |
|
2569 char *name, *subname; |
|
2570 char buf[MAXPATHLEN+1]; |
|
2571 struct filedescr *fdp; |
|
2572 FILE *fp = NULL; |
|
2573 PyObject *newm; |
|
2574 |
|
2575 if (modules_reloading == NULL) { |
|
2576 Py_FatalError("PyImport_ReloadModule: " |
|
2577 "no modules_reloading dictionary!"); |
|
2578 return NULL; |
|
2579 } |
|
2580 |
|
2581 if (m == NULL || !PyModule_Check(m)) { |
|
2582 PyErr_SetString(PyExc_TypeError, |
|
2583 "reload() argument must be module"); |
|
2584 return NULL; |
|
2585 } |
|
2586 name = PyModule_GetName(m); |
|
2587 if (name == NULL) |
|
2588 return NULL; |
|
2589 if (m != PyDict_GetItemString(modules, name)) { |
|
2590 PyErr_Format(PyExc_ImportError, |
|
2591 "reload(): module %.200s not in sys.modules", |
|
2592 name); |
|
2593 return NULL; |
|
2594 } |
|
2595 existing_m = PyDict_GetItemString(modules_reloading, name); |
|
2596 if (existing_m != NULL) { |
|
2597 /* Due to a recursive reload, this module is already |
|
2598 being reloaded. */ |
|
2599 Py_INCREF(existing_m); |
|
2600 return existing_m; |
|
2601 } |
|
2602 if (PyDict_SetItemString(modules_reloading, name, m) < 0) |
|
2603 return NULL; |
|
2604 |
|
2605 subname = strrchr(name, '.'); |
|
2606 if (subname == NULL) |
|
2607 subname = name; |
|
2608 else { |
|
2609 PyObject *parentname, *parent; |
|
2610 parentname = PyString_FromStringAndSize(name, (subname-name)); |
|
2611 if (parentname == NULL) { |
|
2612 imp_modules_reloading_clear(); |
|
2613 return NULL; |
|
2614 } |
|
2615 parent = PyDict_GetItem(modules, parentname); |
|
2616 if (parent == NULL) { |
|
2617 PyErr_Format(PyExc_ImportError, |
|
2618 "reload(): parent %.200s not in sys.modules", |
|
2619 PyString_AS_STRING(parentname)); |
|
2620 Py_DECREF(parentname); |
|
2621 imp_modules_reloading_clear(); |
|
2622 return NULL; |
|
2623 } |
|
2624 Py_DECREF(parentname); |
|
2625 subname++; |
|
2626 path = PyObject_GetAttrString(parent, "__path__"); |
|
2627 if (path == NULL) |
|
2628 PyErr_Clear(); |
|
2629 } |
|
2630 buf[0] = '\0'; |
|
2631 fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader); |
|
2632 Py_XDECREF(path); |
|
2633 |
|
2634 if (fdp == NULL) { |
|
2635 Py_XDECREF(loader); |
|
2636 imp_modules_reloading_clear(); |
|
2637 return NULL; |
|
2638 } |
|
2639 |
|
2640 newm = load_module(name, fp, buf, fdp->type, loader); |
|
2641 Py_XDECREF(loader); |
|
2642 |
|
2643 if (fp) |
|
2644 fclose(fp); |
|
2645 if (newm == NULL) { |
|
2646 /* load_module probably removed name from modules because of |
|
2647 * the error. Put back the original module object. We're |
|
2648 * going to return NULL in this case regardless of whether |
|
2649 * replacing name succeeds, so the return value is ignored. |
|
2650 */ |
|
2651 PyDict_SetItemString(modules, name, m); |
|
2652 } |
|
2653 imp_modules_reloading_clear(); |
|
2654 return newm; |
|
2655 } |
|
2656 |
|
2657 |
|
2658 /* Higher-level import emulator which emulates the "import" statement |
|
2659 more accurately -- it invokes the __import__() function from the |
|
2660 builtins of the current globals. This means that the import is |
|
2661 done using whatever import hooks are installed in the current |
|
2662 environment, e.g. by "rexec". |
|
2663 A dummy list ["__doc__"] is passed as the 4th argument so that |
|
2664 e.g. PyImport_Import(PyString_FromString("win32com.client.gencache")) |
|
2665 will return <module "gencache"> instead of <module "win32com">. */ |
|
2666 |
|
2667 PyObject * |
|
2668 PyImport_Import(PyObject *module_name) |
|
2669 { |
|
2670 static PyObject *silly_list = NULL; |
|
2671 static PyObject *builtins_str = NULL; |
|
2672 static PyObject *import_str = NULL; |
|
2673 PyObject *globals = NULL; |
|
2674 PyObject *import = NULL; |
|
2675 PyObject *builtins = NULL; |
|
2676 PyObject *r = NULL; |
|
2677 |
|
2678 /* Initialize constant string objects */ |
|
2679 if (silly_list == NULL) { |
|
2680 import_str = PyString_InternFromString("__import__"); |
|
2681 if (import_str == NULL) |
|
2682 return NULL; |
|
2683 builtins_str = PyString_InternFromString("__builtins__"); |
|
2684 if (builtins_str == NULL) |
|
2685 return NULL; |
|
2686 silly_list = Py_BuildValue("[s]", "__doc__"); |
|
2687 if (silly_list == NULL) |
|
2688 return NULL; |
|
2689 } |
|
2690 |
|
2691 /* Get the builtins from current globals */ |
|
2692 globals = PyEval_GetGlobals(); |
|
2693 if (globals != NULL) { |
|
2694 Py_INCREF(globals); |
|
2695 builtins = PyObject_GetItem(globals, builtins_str); |
|
2696 if (builtins == NULL) |
|
2697 goto err; |
|
2698 } |
|
2699 else { |
|
2700 /* No globals -- use standard builtins, and fake globals */ |
|
2701 PyErr_Clear(); |
|
2702 |
|
2703 builtins = PyImport_ImportModuleLevel("__builtin__", |
|
2704 NULL, NULL, NULL, 0); |
|
2705 if (builtins == NULL) |
|
2706 return NULL; |
|
2707 globals = Py_BuildValue("{OO}", builtins_str, builtins); |
|
2708 if (globals == NULL) |
|
2709 goto err; |
|
2710 } |
|
2711 |
|
2712 /* Get the __import__ function from the builtins */ |
|
2713 if (PyDict_Check(builtins)) { |
|
2714 import = PyObject_GetItem(builtins, import_str); |
|
2715 if (import == NULL) |
|
2716 PyErr_SetObject(PyExc_KeyError, import_str); |
|
2717 } |
|
2718 else |
|
2719 import = PyObject_GetAttr(builtins, import_str); |
|
2720 if (import == NULL) |
|
2721 goto err; |
|
2722 |
|
2723 /* Call the __import__ function with the proper argument list |
|
2724 * Always use absolute import here. */ |
|
2725 r = PyObject_CallFunction(import, "OOOOi", module_name, globals, |
|
2726 globals, silly_list, 0, NULL); |
|
2727 |
|
2728 err: |
|
2729 Py_XDECREF(globals); |
|
2730 Py_XDECREF(builtins); |
|
2731 Py_XDECREF(import); |
|
2732 |
|
2733 return r; |
|
2734 } |
|
2735 |
|
2736 |
|
2737 /* Module 'imp' provides Python access to the primitives used for |
|
2738 importing modules. |
|
2739 */ |
|
2740 |
|
2741 static PyObject * |
|
2742 imp_get_magic(PyObject *self, PyObject *noargs) |
|
2743 { |
|
2744 char buf[4]; |
|
2745 |
|
2746 buf[0] = (char) ((pyc_magic >> 0) & 0xff); |
|
2747 buf[1] = (char) ((pyc_magic >> 8) & 0xff); |
|
2748 buf[2] = (char) ((pyc_magic >> 16) & 0xff); |
|
2749 buf[3] = (char) ((pyc_magic >> 24) & 0xff); |
|
2750 |
|
2751 return PyString_FromStringAndSize(buf, 4); |
|
2752 } |
|
2753 |
|
2754 static PyObject * |
|
2755 imp_get_suffixes(PyObject *self, PyObject *noargs) |
|
2756 { |
|
2757 PyObject *list; |
|
2758 struct filedescr *fdp; |
|
2759 |
|
2760 list = PyList_New(0); |
|
2761 if (list == NULL) |
|
2762 return NULL; |
|
2763 for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) { |
|
2764 PyObject *item = Py_BuildValue("ssi", |
|
2765 fdp->suffix, fdp->mode, fdp->type); |
|
2766 if (item == NULL) { |
|
2767 Py_DECREF(list); |
|
2768 return NULL; |
|
2769 } |
|
2770 if (PyList_Append(list, item) < 0) { |
|
2771 Py_DECREF(list); |
|
2772 Py_DECREF(item); |
|
2773 return NULL; |
|
2774 } |
|
2775 Py_DECREF(item); |
|
2776 } |
|
2777 return list; |
|
2778 } |
|
2779 |
|
2780 static PyObject * |
|
2781 call_find_module(char *name, PyObject *path) |
|
2782 { |
|
2783 extern int fclose(FILE *); |
|
2784 PyObject *fob, *ret; |
|
2785 struct filedescr *fdp; |
|
2786 char pathname[MAXPATHLEN+1]; |
|
2787 FILE *fp = NULL; |
|
2788 |
|
2789 pathname[0] = '\0'; |
|
2790 if (path == Py_None) |
|
2791 path = NULL; |
|
2792 fdp = find_module(NULL, name, path, pathname, MAXPATHLEN+1, &fp, NULL); |
|
2793 if (fdp == NULL) |
|
2794 return NULL; |
|
2795 if (fp != NULL) { |
|
2796 fob = PyFile_FromFile(fp, pathname, fdp->mode, fclose); |
|
2797 if (fob == NULL) { |
|
2798 fclose(fp); |
|
2799 return NULL; |
|
2800 } |
|
2801 } |
|
2802 else { |
|
2803 fob = Py_None; |
|
2804 Py_INCREF(fob); |
|
2805 } |
|
2806 ret = Py_BuildValue("Os(ssi)", |
|
2807 fob, pathname, fdp->suffix, fdp->mode, fdp->type); |
|
2808 Py_DECREF(fob); |
|
2809 return ret; |
|
2810 } |
|
2811 |
|
2812 static PyObject * |
|
2813 imp_find_module(PyObject *self, PyObject *args) |
|
2814 { |
|
2815 char *name; |
|
2816 PyObject *path = NULL; |
|
2817 if (!PyArg_ParseTuple(args, "s|O:find_module", &name, &path)) |
|
2818 return NULL; |
|
2819 return call_find_module(name, path); |
|
2820 } |
|
2821 |
|
2822 static PyObject * |
|
2823 imp_init_builtin(PyObject *self, PyObject *args) |
|
2824 { |
|
2825 char *name; |
|
2826 int ret; |
|
2827 PyObject *m; |
|
2828 if (!PyArg_ParseTuple(args, "s:init_builtin", &name)) |
|
2829 return NULL; |
|
2830 ret = init_builtin(name); |
|
2831 if (ret < 0) |
|
2832 return NULL; |
|
2833 if (ret == 0) { |
|
2834 Py_INCREF(Py_None); |
|
2835 return Py_None; |
|
2836 } |
|
2837 m = PyImport_AddModule(name); |
|
2838 Py_XINCREF(m); |
|
2839 return m; |
|
2840 } |
|
2841 |
|
2842 static PyObject * |
|
2843 imp_init_frozen(PyObject *self, PyObject *args) |
|
2844 { |
|
2845 char *name; |
|
2846 int ret; |
|
2847 PyObject *m; |
|
2848 if (!PyArg_ParseTuple(args, "s:init_frozen", &name)) |
|
2849 return NULL; |
|
2850 ret = PyImport_ImportFrozenModule(name); |
|
2851 if (ret < 0) |
|
2852 return NULL; |
|
2853 if (ret == 0) { |
|
2854 Py_INCREF(Py_None); |
|
2855 return Py_None; |
|
2856 } |
|
2857 m = PyImport_AddModule(name); |
|
2858 Py_XINCREF(m); |
|
2859 return m; |
|
2860 } |
|
2861 |
|
2862 static PyObject * |
|
2863 imp_get_frozen_object(PyObject *self, PyObject *args) |
|
2864 { |
|
2865 char *name; |
|
2866 |
|
2867 if (!PyArg_ParseTuple(args, "s:get_frozen_object", &name)) |
|
2868 return NULL; |
|
2869 return get_frozen_object(name); |
|
2870 } |
|
2871 |
|
2872 static PyObject * |
|
2873 imp_is_builtin(PyObject *self, PyObject *args) |
|
2874 { |
|
2875 char *name; |
|
2876 if (!PyArg_ParseTuple(args, "s:is_builtin", &name)) |
|
2877 return NULL; |
|
2878 return PyInt_FromLong(is_builtin(name)); |
|
2879 } |
|
2880 |
|
2881 static PyObject * |
|
2882 imp_is_frozen(PyObject *self, PyObject *args) |
|
2883 { |
|
2884 char *name; |
|
2885 struct _frozen *p; |
|
2886 if (!PyArg_ParseTuple(args, "s:is_frozen", &name)) |
|
2887 return NULL; |
|
2888 p = find_frozen(name); |
|
2889 return PyBool_FromLong((long) (p == NULL ? 0 : p->size)); |
|
2890 } |
|
2891 |
|
2892 static FILE * |
|
2893 get_file(char *pathname, PyObject *fob, char *mode) |
|
2894 { |
|
2895 FILE *fp; |
|
2896 if (fob == NULL) { |
|
2897 if (mode[0] == 'U') |
|
2898 mode = "r" PY_STDIOTEXTMODE; |
|
2899 fp = fopen(pathname, mode); |
|
2900 if (fp == NULL) |
|
2901 PyErr_SetFromErrno(PyExc_IOError); |
|
2902 } |
|
2903 else { |
|
2904 fp = PyFile_AsFile(fob); |
|
2905 if (fp == NULL) |
|
2906 PyErr_SetString(PyExc_ValueError, |
|
2907 "bad/closed file object"); |
|
2908 } |
|
2909 return fp; |
|
2910 } |
|
2911 |
|
2912 static PyObject * |
|
2913 imp_load_compiled(PyObject *self, PyObject *args) |
|
2914 { |
|
2915 char *name; |
|
2916 char *pathname; |
|
2917 PyObject *fob = NULL; |
|
2918 PyObject *m; |
|
2919 FILE *fp; |
|
2920 if (!PyArg_ParseTuple(args, "ss|O!:load_compiled", &name, &pathname, |
|
2921 &PyFile_Type, &fob)) |
|
2922 return NULL; |
|
2923 fp = get_file(pathname, fob, "rb"); |
|
2924 if (fp == NULL) |
|
2925 return NULL; |
|
2926 m = load_compiled_module(name, pathname, fp); |
|
2927 if (fob == NULL) |
|
2928 fclose(fp); |
|
2929 return m; |
|
2930 } |
|
2931 |
|
2932 #ifdef HAVE_DYNAMIC_LOADING |
|
2933 |
|
2934 static PyObject * |
|
2935 imp_load_dynamic(PyObject *self, PyObject *args) |
|
2936 { |
|
2937 char *name; |
|
2938 char *pathname; |
|
2939 PyObject *fob = NULL; |
|
2940 PyObject *m; |
|
2941 FILE *fp = NULL; |
|
2942 if (!PyArg_ParseTuple(args, "ss|O!:load_dynamic", &name, &pathname, |
|
2943 &PyFile_Type, &fob)) |
|
2944 return NULL; |
|
2945 if (fob) { |
|
2946 fp = get_file(pathname, fob, "r"); |
|
2947 if (fp == NULL) |
|
2948 return NULL; |
|
2949 } |
|
2950 m = _PyImport_LoadDynamicModule(name, pathname, fp); |
|
2951 return m; |
|
2952 } |
|
2953 |
|
2954 #endif /* HAVE_DYNAMIC_LOADING */ |
|
2955 |
|
2956 static PyObject * |
|
2957 imp_load_source(PyObject *self, PyObject *args) |
|
2958 { |
|
2959 char *name; |
|
2960 char *pathname; |
|
2961 PyObject *fob = NULL; |
|
2962 PyObject *m; |
|
2963 FILE *fp; |
|
2964 if (!PyArg_ParseTuple(args, "ss|O!:load_source", &name, &pathname, |
|
2965 &PyFile_Type, &fob)) |
|
2966 return NULL; |
|
2967 fp = get_file(pathname, fob, "r"); |
|
2968 if (fp == NULL) |
|
2969 return NULL; |
|
2970 m = load_source_module(name, pathname, fp); |
|
2971 if (fob == NULL) |
|
2972 fclose(fp); |
|
2973 return m; |
|
2974 } |
|
2975 |
|
2976 static PyObject * |
|
2977 imp_load_module(PyObject *self, PyObject *args) |
|
2978 { |
|
2979 char *name; |
|
2980 PyObject *fob; |
|
2981 char *pathname; |
|
2982 char *suffix; /* Unused */ |
|
2983 char *mode; |
|
2984 int type; |
|
2985 FILE *fp; |
|
2986 |
|
2987 if (!PyArg_ParseTuple(args, "sOs(ssi):load_module", |
|
2988 &name, &fob, &pathname, |
|
2989 &suffix, &mode, &type)) |
|
2990 return NULL; |
|
2991 if (*mode) { |
|
2992 /* Mode must start with 'r' or 'U' and must not contain '+'. |
|
2993 Implicit in this test is the assumption that the mode |
|
2994 may contain other modifiers like 'b' or 't'. */ |
|
2995 |
|
2996 if (!(*mode == 'r' || *mode == 'U') || strchr(mode, '+')) { |
|
2997 PyErr_Format(PyExc_ValueError, |
|
2998 "invalid file open mode %.200s", mode); |
|
2999 return NULL; |
|
3000 } |
|
3001 } |
|
3002 if (fob == Py_None) |
|
3003 fp = NULL; |
|
3004 else { |
|
3005 if (!PyFile_Check(fob)) { |
|
3006 PyErr_SetString(PyExc_ValueError, |
|
3007 "load_module arg#2 should be a file or None"); |
|
3008 return NULL; |
|
3009 } |
|
3010 fp = get_file(pathname, fob, mode); |
|
3011 if (fp == NULL) |
|
3012 return NULL; |
|
3013 } |
|
3014 return load_module(name, fp, pathname, type, NULL); |
|
3015 } |
|
3016 |
|
3017 static PyObject * |
|
3018 imp_load_package(PyObject *self, PyObject *args) |
|
3019 { |
|
3020 char *name; |
|
3021 char *pathname; |
|
3022 if (!PyArg_ParseTuple(args, "ss:load_package", &name, &pathname)) |
|
3023 return NULL; |
|
3024 return load_package(name, pathname); |
|
3025 } |
|
3026 |
|
3027 static PyObject * |
|
3028 imp_new_module(PyObject *self, PyObject *args) |
|
3029 { |
|
3030 char *name; |
|
3031 if (!PyArg_ParseTuple(args, "s:new_module", &name)) |
|
3032 return NULL; |
|
3033 return PyModule_New(name); |
|
3034 } |
|
3035 |
|
3036 static PyObject * |
|
3037 imp_reload(PyObject *self, PyObject *v) |
|
3038 { |
|
3039 return PyImport_ReloadModule(v); |
|
3040 } |
|
3041 |
|
3042 |
|
3043 /* Doc strings */ |
|
3044 |
|
3045 PyDoc_STRVAR(doc_imp, |
|
3046 "This module provides the components needed to build your own\n\ |
|
3047 __import__ function. Undocumented functions are obsolete."); |
|
3048 |
|
3049 PyDoc_STRVAR(doc_reload, |
|
3050 "reload(module) -> module\n\ |
|
3051 \n\ |
|
3052 Reload the module. The module must have been successfully imported before."); |
|
3053 |
|
3054 PyDoc_STRVAR(doc_find_module, |
|
3055 "find_module(name, [path]) -> (file, filename, (suffix, mode, type))\n\ |
|
3056 Search for a module. If path is omitted or None, search for a\n\ |
|
3057 built-in, frozen or special module and continue search in sys.path.\n\ |
|
3058 The module name cannot contain '.'; to search for a submodule of a\n\ |
|
3059 package, pass the submodule name and the package's __path__."); |
|
3060 |
|
3061 PyDoc_STRVAR(doc_load_module, |
|
3062 "load_module(name, file, filename, (suffix, mode, type)) -> module\n\ |
|
3063 Load a module, given information returned by find_module().\n\ |
|
3064 The module name must include the full package name, if any."); |
|
3065 |
|
3066 PyDoc_STRVAR(doc_get_magic, |
|
3067 "get_magic() -> string\n\ |
|
3068 Return the magic number for .pyc or .pyo files."); |
|
3069 |
|
3070 PyDoc_STRVAR(doc_get_suffixes, |
|
3071 "get_suffixes() -> [(suffix, mode, type), ...]\n\ |
|
3072 Return a list of (suffix, mode, type) tuples describing the files\n\ |
|
3073 that find_module() looks for."); |
|
3074 |
|
3075 PyDoc_STRVAR(doc_new_module, |
|
3076 "new_module(name) -> module\n\ |
|
3077 Create a new module. Do not enter it in sys.modules.\n\ |
|
3078 The module name must include the full package name, if any."); |
|
3079 |
|
3080 PyDoc_STRVAR(doc_lock_held, |
|
3081 "lock_held() -> boolean\n\ |
|
3082 Return True if the import lock is currently held, else False.\n\ |
|
3083 On platforms without threads, return False."); |
|
3084 |
|
3085 PyDoc_STRVAR(doc_acquire_lock, |
|
3086 "acquire_lock() -> None\n\ |
|
3087 Acquires the interpreter's import lock for the current thread.\n\ |
|
3088 This lock should be used by import hooks to ensure thread-safety\n\ |
|
3089 when importing modules.\n\ |
|
3090 On platforms without threads, this function does nothing."); |
|
3091 |
|
3092 PyDoc_STRVAR(doc_release_lock, |
|
3093 "release_lock() -> None\n\ |
|
3094 Release the interpreter's import lock.\n\ |
|
3095 On platforms without threads, this function does nothing."); |
|
3096 |
|
3097 static PyMethodDef imp_methods[] = { |
|
3098 {"reload", imp_reload, METH_O, doc_reload}, |
|
3099 {"find_module", imp_find_module, METH_VARARGS, doc_find_module}, |
|
3100 {"get_magic", imp_get_magic, METH_NOARGS, doc_get_magic}, |
|
3101 {"get_suffixes", imp_get_suffixes, METH_NOARGS, doc_get_suffixes}, |
|
3102 {"load_module", imp_load_module, METH_VARARGS, doc_load_module}, |
|
3103 {"new_module", imp_new_module, METH_VARARGS, doc_new_module}, |
|
3104 {"lock_held", imp_lock_held, METH_NOARGS, doc_lock_held}, |
|
3105 {"acquire_lock", imp_acquire_lock, METH_NOARGS, doc_acquire_lock}, |
|
3106 {"release_lock", imp_release_lock, METH_NOARGS, doc_release_lock}, |
|
3107 /* The rest are obsolete */ |
|
3108 {"get_frozen_object", imp_get_frozen_object, METH_VARARGS}, |
|
3109 {"init_builtin", imp_init_builtin, METH_VARARGS}, |
|
3110 {"init_frozen", imp_init_frozen, METH_VARARGS}, |
|
3111 {"is_builtin", imp_is_builtin, METH_VARARGS}, |
|
3112 {"is_frozen", imp_is_frozen, METH_VARARGS}, |
|
3113 {"load_compiled", imp_load_compiled, METH_VARARGS}, |
|
3114 #ifdef HAVE_DYNAMIC_LOADING |
|
3115 {"load_dynamic", imp_load_dynamic, METH_VARARGS}, |
|
3116 #endif |
|
3117 {"load_package", imp_load_package, METH_VARARGS}, |
|
3118 {"load_source", imp_load_source, METH_VARARGS}, |
|
3119 {NULL, NULL} /* sentinel */ |
|
3120 }; |
|
3121 |
|
3122 static int |
|
3123 setint(PyObject *d, char *name, int value) |
|
3124 { |
|
3125 PyObject *v; |
|
3126 int err; |
|
3127 |
|
3128 v = PyInt_FromLong((long)value); |
|
3129 err = PyDict_SetItemString(d, name, v); |
|
3130 Py_XDECREF(v); |
|
3131 return err; |
|
3132 } |
|
3133 |
|
3134 typedef struct { |
|
3135 PyObject_HEAD |
|
3136 } NullImporter; |
|
3137 |
|
3138 static int |
|
3139 NullImporter_init(NullImporter *self, PyObject *args, PyObject *kwds) |
|
3140 { |
|
3141 char *path; |
|
3142 Py_ssize_t pathlen; |
|
3143 |
|
3144 if (!_PyArg_NoKeywords("NullImporter()", kwds)) |
|
3145 return -1; |
|
3146 |
|
3147 if (!PyArg_ParseTuple(args, "s:NullImporter", |
|
3148 &path)) |
|
3149 return -1; |
|
3150 |
|
3151 pathlen = strlen(path); |
|
3152 if (pathlen == 0) { |
|
3153 PyErr_SetString(PyExc_ImportError, "empty pathname"); |
|
3154 return -1; |
|
3155 } else { |
|
3156 #ifndef RISCOS |
|
3157 struct stat statbuf; |
|
3158 int rv; |
|
3159 |
|
3160 rv = stat(path, &statbuf); |
|
3161 #ifdef MS_WINDOWS |
|
3162 /* MS Windows stat() chokes on paths like C:\path\. Try to |
|
3163 * recover *one* time by stripping off a trailing slash or |
|
3164 * backslash. http://bugs.python.org/issue1293 |
|
3165 */ |
|
3166 if (rv != 0 && pathlen <= MAXPATHLEN && |
|
3167 (path[pathlen-1] == '/' || path[pathlen-1] == '\\')) { |
|
3168 char mangled[MAXPATHLEN+1]; |
|
3169 |
|
3170 strcpy(mangled, path); |
|
3171 mangled[pathlen-1] = '\0'; |
|
3172 rv = stat(mangled, &statbuf); |
|
3173 } |
|
3174 #endif |
|
3175 if (rv == 0) { |
|
3176 /* it exists */ |
|
3177 if (S_ISDIR(statbuf.st_mode)) { |
|
3178 /* it's a directory */ |
|
3179 PyErr_SetString(PyExc_ImportError, |
|
3180 "existing directory"); |
|
3181 return -1; |
|
3182 } |
|
3183 } |
|
3184 #else |
|
3185 if (object_exists(path)) { |
|
3186 /* it exists */ |
|
3187 if (isdir(path)) { |
|
3188 /* it's a directory */ |
|
3189 PyErr_SetString(PyExc_ImportError, |
|
3190 "existing directory"); |
|
3191 return -1; |
|
3192 } |
|
3193 } |
|
3194 #endif |
|
3195 } |
|
3196 return 0; |
|
3197 } |
|
3198 |
|
3199 static PyObject * |
|
3200 NullImporter_find_module(NullImporter *self, PyObject *args) |
|
3201 { |
|
3202 Py_RETURN_NONE; |
|
3203 } |
|
3204 |
|
3205 static PyMethodDef NullImporter_methods[] = { |
|
3206 {"find_module", (PyCFunction)NullImporter_find_module, METH_VARARGS, |
|
3207 "Always return None" |
|
3208 }, |
|
3209 {NULL} /* Sentinel */ |
|
3210 }; |
|
3211 |
|
3212 |
|
3213 PyTypeObject PyNullImporter_Type = { |
|
3214 PyVarObject_HEAD_INIT(NULL, 0) |
|
3215 "imp.NullImporter", /*tp_name*/ |
|
3216 sizeof(NullImporter), /*tp_basicsize*/ |
|
3217 0, /*tp_itemsize*/ |
|
3218 0, /*tp_dealloc*/ |
|
3219 0, /*tp_print*/ |
|
3220 0, /*tp_getattr*/ |
|
3221 0, /*tp_setattr*/ |
|
3222 0, /*tp_compare*/ |
|
3223 0, /*tp_repr*/ |
|
3224 0, /*tp_as_number*/ |
|
3225 0, /*tp_as_sequence*/ |
|
3226 0, /*tp_as_mapping*/ |
|
3227 0, /*tp_hash */ |
|
3228 0, /*tp_call*/ |
|
3229 0, /*tp_str*/ |
|
3230 0, /*tp_getattro*/ |
|
3231 0, /*tp_setattro*/ |
|
3232 0, /*tp_as_buffer*/ |
|
3233 Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
|
3234 "Null importer object", /* tp_doc */ |
|
3235 0, /* tp_traverse */ |
|
3236 0, /* tp_clear */ |
|
3237 0, /* tp_richcompare */ |
|
3238 0, /* tp_weaklistoffset */ |
|
3239 0, /* tp_iter */ |
|
3240 0, /* tp_iternext */ |
|
3241 NullImporter_methods, /* tp_methods */ |
|
3242 0, /* tp_members */ |
|
3243 0, /* tp_getset */ |
|
3244 0, /* tp_base */ |
|
3245 0, /* tp_dict */ |
|
3246 0, /* tp_descr_get */ |
|
3247 0, /* tp_descr_set */ |
|
3248 0, /* tp_dictoffset */ |
|
3249 (initproc)NullImporter_init, /* tp_init */ |
|
3250 0, /* tp_alloc */ |
|
3251 PyType_GenericNew /* tp_new */ |
|
3252 }; |
|
3253 |
|
3254 |
|
3255 PyMODINIT_FUNC |
|
3256 initimp(void) |
|
3257 { |
|
3258 PyObject *m, *d; |
|
3259 |
|
3260 if (PyType_Ready(&PyNullImporter_Type) < 0) |
|
3261 goto failure; |
|
3262 |
|
3263 m = Py_InitModule4("imp", imp_methods, doc_imp, |
|
3264 NULL, PYTHON_API_VERSION); |
|
3265 if (m == NULL) |
|
3266 goto failure; |
|
3267 d = PyModule_GetDict(m); |
|
3268 if (d == NULL) |
|
3269 goto failure; |
|
3270 |
|
3271 if (setint(d, "SEARCH_ERROR", SEARCH_ERROR) < 0) goto failure; |
|
3272 if (setint(d, "PY_SOURCE", PY_SOURCE) < 0) goto failure; |
|
3273 if (setint(d, "PY_COMPILED", PY_COMPILED) < 0) goto failure; |
|
3274 if (setint(d, "C_EXTENSION", C_EXTENSION) < 0) goto failure; |
|
3275 if (setint(d, "PY_RESOURCE", PY_RESOURCE) < 0) goto failure; |
|
3276 if (setint(d, "PKG_DIRECTORY", PKG_DIRECTORY) < 0) goto failure; |
|
3277 if (setint(d, "C_BUILTIN", C_BUILTIN) < 0) goto failure; |
|
3278 if (setint(d, "PY_FROZEN", PY_FROZEN) < 0) goto failure; |
|
3279 if (setint(d, "PY_CODERESOURCE", PY_CODERESOURCE) < 0) goto failure; |
|
3280 if (setint(d, "IMP_HOOK", IMP_HOOK) < 0) goto failure; |
|
3281 |
|
3282 Py_INCREF(&PyNullImporter_Type); |
|
3283 PyModule_AddObject(m, "NullImporter", (PyObject *)&PyNullImporter_Type); |
|
3284 failure: |
|
3285 ; |
|
3286 } |
|
3287 |
|
3288 |
|
3289 /* API for embedding applications that want to add their own entries |
|
3290 to the table of built-in modules. This should normally be called |
|
3291 *before* Py_Initialize(). When the table resize fails, -1 is |
|
3292 returned and the existing table is unchanged. |
|
3293 |
|
3294 After a similar function by Just van Rossum. */ |
|
3295 |
|
3296 int |
|
3297 PyImport_ExtendInittab(struct _inittab *newtab) |
|
3298 { |
|
3299 static struct _inittab *our_copy = NULL; |
|
3300 struct _inittab *p; |
|
3301 int i, n; |
|
3302 |
|
3303 /* Count the number of entries in both tables */ |
|
3304 for (n = 0; newtab[n].name != NULL; n++) |
|
3305 ; |
|
3306 if (n == 0) |
|
3307 return 0; /* Nothing to do */ |
|
3308 for (i = 0; PyImport_Inittab[i].name != NULL; i++) |
|
3309 ; |
|
3310 |
|
3311 /* Allocate new memory for the combined table */ |
|
3312 p = our_copy; |
|
3313 PyMem_RESIZE(p, struct _inittab, i+n+1); |
|
3314 if (p == NULL) |
|
3315 return -1; |
|
3316 |
|
3317 /* Copy the tables into the new memory */ |
|
3318 if (our_copy != PyImport_Inittab) |
|
3319 memcpy(p, PyImport_Inittab, (i+1) * sizeof(struct _inittab)); |
|
3320 PyImport_Inittab = our_copy = p; |
|
3321 memcpy(p+i, newtab, (n+1) * sizeof(struct _inittab)); |
|
3322 |
|
3323 return 0; |
|
3324 } |
|
3325 |
|
3326 /* Shorthand to add a single entry given a name and a function */ |
|
3327 |
|
3328 int |
|
3329 PyImport_AppendInittab(char *name, void (*initfunc)(void)) |
|
3330 { |
|
3331 struct _inittab newtab[2]; |
|
3332 |
|
3333 memset(newtab, '\0', sizeof newtab); |
|
3334 |
|
3335 newtab[0].name = name; |
|
3336 newtab[0].initfunc = initfunc; |
|
3337 |
|
3338 return PyImport_ExtendInittab(newtab); |
|
3339 } |
|
3340 |
|
3341 #ifdef __cplusplus |
|
3342 } |
|
3343 #endif |