LCOV - code coverage report
Current view: top level - Python - import.c (source / functions) Hit Total Coverage
Test: CPython lcov report Lines: 747 1427 52.3 %
Date: 2017-04-19 Functions: 45 77 58.4 %

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

Generated by: LCOV version 1.10