Branch data     Line data    Source code

       1                 :            : 
       2                 :            : /* Frozen modules bootstrap
       3                 :            :  *
       4                 :            :  * Limited and restricted Python interpreter to run
       5                 :            :  * "Tools/scripts/deepfreeze.py" on systems with no or older Python
       6                 :            :  * interpreter.
       7                 :            :  */
       8                 :            : 
       9                 :            : #include "Python.h"
      10                 :            : #include "pycore_import.h"
      11                 :            : 
      12                 :            : /* Includes for frozen modules: */
      13                 :            : #include "Python/frozen_modules/importlib._bootstrap.h"
      14                 :            : #include "Python/frozen_modules/importlib._bootstrap_external.h"
      15                 :            : /* End includes */
      16                 :            : 
      17                 :            : /* Empty initializer for deepfrozen modules */
      18                 :         21 : int _Py_Deepfreeze_Init(void)
      19                 :            : {
      20                 :         21 :     return 0;
      21                 :            : }
      22                 :            : /* Empty finalizer for deepfrozen modules */
      23                 :            : void
      24                 :         21 : _Py_Deepfreeze_Fini(void)
      25                 :            : {
      26                 :         21 : }
      27                 :            : 
      28                 :            : /* Note that a negative size indicates a package. */
      29                 :            : 
      30                 :            : static const struct _frozen bootstrap_modules[] = {
      31                 :            :     {"_frozen_importlib", _Py_M__importlib__bootstrap, (int)sizeof(_Py_M__importlib__bootstrap)},
      32                 :            :     {"_frozen_importlib_external", _Py_M__importlib__bootstrap_external, (int)sizeof(_Py_M__importlib__bootstrap_external)},
      33                 :            :     {0, 0, 0} /* bootstrap sentinel */
      34                 :            : };
      35                 :            : static const struct _frozen stdlib_modules[] = {
      36                 :            :     {0, 0, 0} /* stdlib sentinel */
      37                 :            : };
      38                 :            : static const struct _frozen test_modules[] = {
      39                 :            :     {0, 0, 0} /* test sentinel */
      40                 :            : };
      41                 :            : const struct _frozen *_PyImport_FrozenBootstrap = bootstrap_modules;
      42                 :            : const struct _frozen *_PyImport_FrozenStdlib = stdlib_modules;
      43                 :            : const struct _frozen *_PyImport_FrozenTest = test_modules;
      44                 :            : 
      45                 :            : static const struct _module_alias aliases[] = {
      46                 :            :     {"_frozen_importlib", "importlib._bootstrap"},
      47                 :            :     {"_frozen_importlib_external", "importlib._bootstrap_external"},
      48                 :            :     {0, 0} /* aliases sentinel */
      49                 :            : };
      50                 :            : const struct _module_alias *_PyImport_FrozenAliases = aliases;
      51                 :            : 
      52                 :            : /* Embedding apps may change this pointer to point to their favorite
      53                 :            :    collection of frozen modules: */
      54                 :            : 
      55                 :            : const struct _frozen *PyImport_FrozenModules = NULL;
      56                 :            : 
      57                 :            : int
      58                 :            : #ifdef MS_WINDOWS
      59                 :            : wmain(int argc, wchar_t **argv)
      60                 :            : #else
      61                 :         21 : main(int argc, char **argv)
      62                 :            : #endif
      63                 :            : {
      64                 :            :     PyStatus status;
      65                 :            : 
      66                 :            :     PyConfig config;
      67                 :         21 :     PyConfig_InitIsolatedConfig(&config);
      68                 :            :     // don't warn, pybuilddir.txt does not exist yet
      69                 :         21 :     config.pathconfig_warnings = 0;
      70                 :            :     // parse arguments
      71                 :         21 :     config.parse_argv = 1;
      72                 :            :     // add current script dir to sys.path
      73                 :         21 :     config.isolated = 0;
      74                 :         21 :     config.safe_path = 0;
      75                 :            : 
      76                 :            : #ifdef MS_WINDOWS
      77                 :            :     status = PyConfig_SetArgv(&config, argc, argv);
      78                 :            : #else
      79                 :         21 :     status = PyConfig_SetBytesArgv(&config, argc, argv);
      80                 :            : #endif
      81         [ -  + ]:         21 :     if (PyStatus_Exception(status)) {
      82                 :          0 :         goto error;
      83                 :            :     }
      84                 :            : 
      85                 :         21 :     status = PyConfig_Read(&config);
      86         [ -  + ]:         21 :     if (config.run_filename == NULL) {
      87                 :          0 :         status = PyStatus_Error("Run filename expected");
      88                 :          0 :         goto error;
      89                 :            :     }
      90                 :            : 
      91                 :            : #define CLEAR(ATTR) \
      92                 :            :     do { \
      93                 :            :         PyMem_RawFree(ATTR); \
      94                 :            :         ATTR = NULL; \
      95                 :            :     } while (0)
      96                 :            : 
      97                 :            :     // isolate from system Python
      98                 :         21 :     CLEAR(config.base_prefix);
      99                 :         21 :     CLEAR(config.prefix);
     100                 :         21 :     CLEAR(config.base_exec_prefix);
     101                 :         21 :     CLEAR(config.exec_prefix);
     102                 :            : 
     103                 :         21 :     status = Py_InitializeFromConfig(&config);
     104         [ -  + ]:         21 :     if (PyStatus_Exception(status)) {
     105                 :          0 :         goto error;
     106                 :            :     }
     107                 :         21 :     PyConfig_Clear(&config);
     108                 :            : 
     109                 :         21 :     return Py_RunMain();
     110                 :            : 
     111                 :          0 : error:
     112                 :          0 :     PyConfig_Clear(&config);
     113         [ #  # ]:          0 :     if (PyStatus_IsExit(status)) {
     114                 :          0 :         return status.exitcode;
     115                 :            :     }
     116                 :          0 :     Py_ExitStatusException(status);
     117                 :            : }
     118                 :            :