Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / python / cpython3 / bebfbe2d1138216a5023ce067f6d1fbc29a437c6 / . / Python / frozenmain.c
blob: 88c34651dc16bc46366cd84543124c2910b383a6 [file] [log] [blame]
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]1
2/* Python interpreter main program for frozen scripts */
3
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]4#include "Python.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]5#include <locale.h>
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]6
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000[diff] [blame]7#ifdef MS_WINDOWS
Thomas Woutersb4bd21c2000-07-22 23:38:01 +0000[diff] [blame]8extern void PyWinFreeze_ExeInit(void);
9extern void PyWinFreeze_ExeTerm(void);
10extern int PyInitFrozenExtensions(void);
Guido van Rossum6deac7a1998-04-03 21:11:15 +0000[diff] [blame]11#endif
12
Guido van Rossum47ad5e71995-08-04 04:10:43 +0000[diff] [blame]13/* Main program */
14
15int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000[diff] [blame]16Py_FrozenMain(int argc, char **argv)
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]17{
18 char *p;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]19 int i, n, sts;
Guido van Rossum094885b1993-11-05 10:16:09 +0000[diff] [blame]20 int inspect = 0;
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]21 int unbuffered = 0;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]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);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]26
Guido van Rossum919b83d1998-02-06 22:30:29 +0000[diff] [blame]27 Py_FrozenFlag = 1; /* Suppress errors from getpath.c */
28
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000[diff] [blame]29 if ((p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]30 inspect = 1;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000[diff] [blame]31 if ((p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0')
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]32 unbuffered = 1;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]33
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]34 if (unbuffered) {
Guido van Rossumc425d2f1997-12-02 20:41:39 +0000[diff] [blame]35 setbuf(stdin, (char *)NULL);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]36 setbuf(stdout, (char *)NULL);
37 setbuf(stderr, (char *)NULL);
38 }
39
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]40 if (!argv_copy) {
41 fprintf(stderr, "out of memory");
42 return 1;
43 }
44
45 oldloc = setlocale(LC_ALL, NULL);
46 setlocale(LC_ALL, "");
47 for (i = 0; i < argc; i++) {
48 size_t argsize = mbstowcs(NULL, argv[i], 0);
49 if (argsize == (size_t)-1) {
50 fprintf(stderr, "Could not convert argument %d to string", i);
51 return 1;
52 }
53 argv_copy[i] = PyMem_Malloc((argsize+1)*sizeof(wchar_t));
54 argv_copy2[i] = argv_copy[i];
55 if (!argv_copy[i]) {
56 fprintf(stderr, "out of memory");
57 return 1;
58 }
59 mbstowcs(argv_copy[i], argv[i], argsize+1);
60 }
61 setlocale(LC_ALL, oldloc);
62
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000[diff] [blame]63#ifdef MS_WINDOWS
Guido van Rossum47ae0281999-01-02 21:42:48 +0000[diff] [blame]64 PyInitFrozenExtensions();
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000[diff] [blame]65#endif /* MS_WINDOWS */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]66 Py_SetProgramName(argv_copy[0]);
Guido van Rossum3768fb11997-07-19 19:24:41 +0000[diff] [blame]67 Py_Initialize();
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000[diff] [blame]68#ifdef MS_WINDOWS
Guido van Rossum6deac7a1998-04-03 21:11:15 +0000[diff] [blame]69 PyWinFreeze_ExeInit();
70#endif
Guido van Rossum3768fb11997-07-19 19:24:41 +0000[diff] [blame]71
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]72 if (Py_VerboseFlag)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000[diff] [blame]73 fprintf(stderr, "Python %s\n%s\n",
Guido van Rossum582646a1996-05-28 22:30:17 +0000[diff] [blame]74 Py_GetVersion(), Py_GetCopyright());
Guido van Rossum3768fb11997-07-19 19:24:41 +0000[diff] [blame]75
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]76 PySys_SetArgv(argc, argv_copy);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]77
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]78 n = PyImport_ImportFrozenModule("__main__");
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]79 if (n == 0)
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]80 Py_FatalError("__main__ not frozen");
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]81 if (n < 0) {
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]82 PyErr_Print();
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]83 sts = 1;
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]84 }
85 else
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]86 sts = 0;
87
Guido van Rossumf1dc5661993-07-05 10:31:29 +0000[diff] [blame]88 if (inspect && isatty((int)fileno(stdin)))
Guido van Rossum4e2e0f91995-03-31 10:27:23 +0000[diff] [blame]89 sts = PyRun_AnyFile(stdin, "<stdin>") != 0;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000[diff] [blame]90
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000[diff] [blame]91#ifdef MS_WINDOWS
Guido van Rossum6deac7a1998-04-03 21:11:15 +0000[diff] [blame]92 PyWinFreeze_ExeTerm();
93#endif
Guido van Rossum7c141031997-08-15 02:52:08 +0000[diff] [blame]94 Py_Finalize();
Martin v. Löwis790465f2008-04-05 20:41:37 +0000[diff] [blame]95 for (i = 0; i < argc; i++) {
96 PyMem_Free(argv_copy2[i]);
97 }
98 PyMem_Free(argv_copy);
99 PyMem_Free(argv_copy2);
Guido van Rossum7c141031997-08-15 02:52:08 +0000[diff] [blame]100 return sts;
Guido van Rossumf56e3db1993-04-01 20:59:32 +0000[diff] [blame]101}