Python/frozenmain.c

   1
   2 /* Python interpreter main program for frozen scripts */
   3
   4 #include "Python.h"
   5 #include <locale.h>
   6
   7 #ifdef MS_WINDOWS
   8 extern void PyWinFreeze_ExeInit(void);
   9 extern void PyWinFreeze_ExeTerm(void);
  10 extern int PyInitFrozenExtensions(void);
  11 #endif
  12
  13 /* Main program */
  14
  15 int
  16 Py_FrozenMain(int argc, char **argv)
  17 {
  18         char *p;
  19         int i, n, sts;
  20         int inspect = 0;
  21         int unbuffered = 0;
  22         char *oldloc;
  23         wchar_t **argv_copy = PyMem_Malloc(sizeof(wchar_t*)*argc);
  24         /* We need a second copies, as Python might modify the first one. */
  25         wchar_t **argv_copy2 = PyMem_Malloc(sizeof(wchar_t*)*argc);
  26
  27         Py_FrozenFlag = 1; /* Suppress errors from getpath.c */
  28
  29         if ((p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
  30                 inspect = 1;
  31         if ((p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0')
  32                 unbuffered = 1;
  33
  34         if (unbuffered) {
  35                 setbuf(stdin, (char *)NULL);
  36                 setbuf(stdout, (char *)NULL);
  37                 setbuf(stderr, (char *)NULL);
  38         }
  39
  40         if (!argv_copy) {
  41                 fprintf(stderr, "out of memory\n");
  42                 return 1;
  43         }
  44
  45         oldloc = setlocale(LC_ALL, NULL);
  46         setlocale(LC_ALL, "");
  47         for (i = 0; i < argc; i++) {
  48 #ifdef HAVE_BROKEN_MBSTOWCS
  49                 size_t argsize = strlen(argv[i]);
  50 #else
  51                 size_t argsize = mbstowcs(NULL, argv[i], 0);
  52 #endif
  53                 size_t count;
  54                 if (argsize == (size_t)-1) {
  55                         fprintf(stderr, "Could not convert argument %d to string\n", i);
  56                         return 1;
  57                 }
  58                 argv_copy[i] = PyMem_Malloc((argsize+1)*sizeof(wchar_t));
  59                 argv_copy2[i] = argv_copy[i];
  60                 if (!argv_copy[i]) {
  61                         fprintf(stderr, "out of memory\n");
  62                         return 1;
  63                 }
  64                 count = mbstowcs(argv_copy[i], argv[i], argsize+1);
  65                 if (count == (size_t)-1) {
  66                         fprintf(stderr, "Could not convert argument %d to string\n", i);
  67                         return 1;
  68                 }
  69         }
  70         setlocale(LC_ALL, oldloc);
  71
  72 #ifdef MS_WINDOWS
  73         PyInitFrozenExtensions();
  74 #endif /* MS_WINDOWS */
  75         Py_SetProgramName(argv_copy[0]);
  76         Py_Initialize();
  77 #ifdef MS_WINDOWS
  78         PyWinFreeze_ExeInit();
  79 #endif
  80
  81         if (Py_VerboseFlag)
  82                 fprintf(stderr, "Python %s\n%s\n",
  83                         Py_GetVersion(), Py_GetCopyright());
  84
  85         PySys_SetArgv(argc, argv_copy);
  86
  87         n = PyImport_ImportFrozenModule("__main__");
  88         if (n == 0)
  89                 Py_FatalError("__main__ not frozen");
  90         if (n < 0) {
  91                 PyErr_Print();
  92                 sts = 1;
  93         }
  94         else
  95                 sts = 0;
  96
  97         if (inspect && isatty((int)fileno(stdin)))
  98                 sts = PyRun_AnyFile(stdin, "<stdin>") != 0;
  99
 100 #ifdef MS_WINDOWS
 101         PyWinFreeze_ExeTerm();
 102 #endif
 103         Py_Finalize();
 104         for (i = 0; i < argc; i++) {
 105                 PyMem_Free(argv_copy2[i]);
 106         }
 107         PyMem_Free(argv_copy);
 108         PyMem_Free(argv_copy2);
 109         return sts;
 110 }