blob: 98f46ebdb2dd5f8299304c69a4901e0ca504115d [file] [log] [blame]
Guido van Rossumf56e3db1993-04-01 20:59:32 +00001
2/* Python interpreter main program for frozen scripts */
3
Guido van Rossum4e2e0f91995-03-31 10:27:23 +00004#include "Python.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +00005#include <locale.h>
Guido van Rossumf56e3db1993-04-01 20:59:32 +00006
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00007#ifdef MS_WINDOWS
Thomas Woutersb4bd21c2000-07-22 23:38:01 +00008extern void PyWinFreeze_ExeInit(void);
9extern void PyWinFreeze_ExeTerm(void);
10extern int PyInitFrozenExtensions(void);
Guido van Rossum6deac7a1998-04-03 21:11:15 +000011#endif
12
Guido van Rossum47ad5e71995-08-04 04:10:43 +000013/* Main program */
14
15int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000016Py_FrozenMain(int argc, char **argv)
Guido van Rossumf56e3db1993-04-01 20:59:32 +000017{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000018 char *p;
19 int i, n, sts;
20 int inspect = 0;
21 int unbuffered = 0;
22 char *oldloc;
Victor Stinner36577e42013-07-27 01:04:56 +020023 wchar_t **argv_copy;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000024 /* We need a second copies, as Python might modify the first one. */
Victor Stinner36577e42013-07-27 01:04:56 +020025 wchar_t **argv_copy2;
26
27 argv_copy = PyMem_Malloc(sizeof(wchar_t*)*argc);
28 argv_copy2 = PyMem_Malloc(sizeof(wchar_t*)*argc);
29 if (!argv_copy || !argv_copy2) {
30 fprintf(stderr, "out of memory\n");
31 return 1;
32 }
Guido van Rossum1d5735e1994-08-30 08:27:36 +000033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000034 Py_FrozenFlag = 1; /* Suppress errors from getpath.c */
Guido van Rossum919b83d1998-02-06 22:30:29 +000035
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000036 if ((p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
37 inspect = 1;
38 if ((p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0')
39 unbuffered = 1;
Guido van Rossum9e90a671993-06-24 11:10:19 +000040
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000041 if (unbuffered) {
42 setbuf(stdin, (char *)NULL);
43 setbuf(stdout, (char *)NULL);
44 setbuf(stderr, (char *)NULL);
45 }
Guido van Rossum1d5735e1994-08-30 08:27:36 +000046
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000047 if (!argv_copy) {
48 fprintf(stderr, "out of memory\n");
49 return 1;
50 }
Martin v. Löwis790465f2008-04-05 20:41:37 +000051
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000052 oldloc = setlocale(LC_ALL, NULL);
53 setlocale(LC_ALL, "");
54 for (i = 0; i < argc; i++) {
Antoine Pitroufff95302008-09-03 18:58:51 +000055#ifdef HAVE_BROKEN_MBSTOWCS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000056 size_t argsize = strlen(argv[i]);
Antoine Pitroufff95302008-09-03 18:58:51 +000057#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000058 size_t argsize = mbstowcs(NULL, argv[i], 0);
Antoine Pitroufff95302008-09-03 18:58:51 +000059#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000060 size_t count;
61 if (argsize == (size_t)-1) {
62 fprintf(stderr, "Could not convert argument %d to string\n", i);
63 return 1;
64 }
65 argv_copy[i] = PyMem_Malloc((argsize+1)*sizeof(wchar_t));
66 argv_copy2[i] = argv_copy[i];
67 if (!argv_copy[i]) {
68 fprintf(stderr, "out of memory\n");
69 return 1;
70 }
71 count = mbstowcs(argv_copy[i], argv[i], argsize+1);
72 if (count == (size_t)-1) {
73 fprintf(stderr, "Could not convert argument %d to string\n", i);
74 return 1;
75 }
76 }
77 setlocale(LC_ALL, oldloc);
Martin v. Löwis790465f2008-04-05 20:41:37 +000078
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000079#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000080 PyInitFrozenExtensions();
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000081#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082 Py_SetProgramName(argv_copy[0]);
83 Py_Initialize();
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000084#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000085 PyWinFreeze_ExeInit();
Guido van Rossum6deac7a1998-04-03 21:11:15 +000086#endif
Guido van Rossum3768fb11997-07-19 19:24:41 +000087
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000088 if (Py_VerboseFlag)
89 fprintf(stderr, "Python %s\n%s\n",
90 Py_GetVersion(), Py_GetCopyright());
Guido van Rossum3768fb11997-07-19 19:24:41 +000091
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000092 PySys_SetArgv(argc, argv_copy);
Guido van Rossum9e90a671993-06-24 11:10:19 +000093
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000094 n = PyImport_ImportFrozenModule("__main__");
95 if (n == 0)
96 Py_FatalError("__main__ not frozen");
97 if (n < 0) {
98 PyErr_Print();
99 sts = 1;
100 }
101 else
102 sts = 0;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000103
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000104 if (inspect && isatty((int)fileno(stdin)))
105 sts = PyRun_AnyFile(stdin, "<stdin>") != 0;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000106
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000107#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000108 PyWinFreeze_ExeTerm();
Guido van Rossum6deac7a1998-04-03 21:11:15 +0000109#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000110 Py_Finalize();
111 for (i = 0; i < argc; i++) {
112 PyMem_Free(argv_copy2[i]);
113 }
114 PyMem_Free(argv_copy);
115 PyMem_Free(argv_copy2);
116 return sts;
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000117}