| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1 |  | 
 | 2 | /* Python interpreter top-level routines, including init/exit */ | 
 | 3 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 4 | #include "Python.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 5 |  | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 6 | #include "Python-ast.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 7 | #include "grammar.h" | 
 | 8 | #include "node.h" | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 9 | #include "token.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 10 | #include "parsetok.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 11 | #include "errcode.h" | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 12 | #include "code.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 13 | #include "compile.h" | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 14 | #include "symtable.h" | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 15 | #include "pyarena.h" | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 16 | #include "ast.h" | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 17 | #include "eval.h" | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 18 | #include "marshal.h" | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 19 |  | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 20 | #include <signal.h> | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 21 |  | 
| Martin v. Löwis | 73d538b | 2003-03-05 15:13:47 +0000 | [diff] [blame] | 22 | #ifdef HAVE_LANGINFO_H | 
 | 23 | #include <locale.h> | 
 | 24 | #include <langinfo.h> | 
 | 25 | #endif | 
 | 26 |  | 
| Martin v. Löwis | 6238d2b | 2002-06-30 15:26:10 +0000 | [diff] [blame] | 27 | #ifdef MS_WINDOWS | 
| Guido van Rossum | a44823b | 1995-03-14 15:01:17 +0000 | [diff] [blame] | 28 | #undef BYTE | 
 | 29 | #include "windows.h" | 
 | 30 | #endif | 
 | 31 |  | 
| Neal Norwitz | 4281cef | 2006-03-04 19:58:13 +0000 | [diff] [blame] | 32 | #ifndef Py_REF_DEBUG | 
| Tim Peters | 62e97f0 | 2006-03-28 21:44:32 +0000 | [diff] [blame] | 33 | #define PRINT_TOTAL_REFS() | 
| Neal Norwitz | 4281cef | 2006-03-04 19:58:13 +0000 | [diff] [blame] | 34 | #else /* Py_REF_DEBUG */ | 
| Tim Peters | 62e97f0 | 2006-03-28 21:44:32 +0000 | [diff] [blame] | 35 | #define PRINT_TOTAL_REFS() fprintf(stderr,				\ | 
 | 36 | 				   "[%" PY_FORMAT_SIZE_T "d refs]\n",	\ | 
| Armin Rigo | e170937 | 2006-04-12 17:06:05 +0000 | [diff] [blame] | 37 | 				   _Py_GetRefTotal()) | 
| Neal Norwitz | 4281cef | 2006-03-04 19:58:13 +0000 | [diff] [blame] | 38 | #endif | 
 | 39 |  | 
| Anthony Baxter | ac6bd46 | 2006-04-13 02:06:09 +0000 | [diff] [blame] | 40 | #ifdef __cplusplus | 
 | 41 | extern "C" { | 
 | 42 | #endif | 
 | 43 |  | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 44 | extern char *Py_GetPath(void); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 45 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 46 | extern grammar _PyParser_Grammar; /* From graminit.c */ | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 47 |  | 
| Guido van Rossum | b73cc04 | 1993-11-01 16:28:59 +0000 | [diff] [blame] | 48 | /* Forward */ | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 49 | static void initmain(void); | 
 | 50 | static void initsite(void); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 51 | static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *, | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 52 | 			  PyCompilerFlags *, PyArena *); | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 53 | static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *, | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 54 | 			      PyCompilerFlags *); | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 55 | static void err_input(perrdetail *); | 
 | 56 | static void initsigs(void); | 
 | 57 | static void call_sys_exitfunc(void); | 
 | 58 | static void call_ll_exitfuncs(void); | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 59 | extern void _PyUnicode_Init(void); | 
 | 60 | extern void _PyUnicode_Fini(void); | 
| Guido van Rossum | c94044c | 2000-03-10 23:03:54 +0000 | [diff] [blame] | 61 |  | 
| Mark Hammond | 8d98d2c | 2003-04-19 15:41:53 +0000 | [diff] [blame] | 62 | #ifdef WITH_THREAD | 
 | 63 | extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *); | 
 | 64 | extern void _PyGILState_Fini(void); | 
 | 65 | #endif /* WITH_THREAD */ | 
 | 66 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 67 | int Py_DebugFlag; /* Needed by parser.c */ | 
 | 68 | int Py_VerboseFlag; /* Needed by import.c */ | 
| Guido van Rossum | 7433b12 | 1997-02-14 19:45:36 +0000 | [diff] [blame] | 69 | int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */ | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 70 | int Py_NoSiteFlag; /* Suppress 'import site' */ | 
| Barry Warsaw | 3ce0964 | 2000-05-02 19:18:59 +0000 | [diff] [blame] | 71 | int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */ | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 72 | int Py_FrozenFlag; /* Needed by getpath.c */ | 
| Guido van Rossum | b16d197 | 2000-05-01 17:55:15 +0000 | [diff] [blame] | 73 | int Py_UnicodeFlag = 0; /* Needed by compile.c */ | 
| Neil Schemenauer | 7d4bb9f | 2001-07-23 16:30:27 +0000 | [diff] [blame] | 74 | int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */ | 
| Tim Peters | 3caca23 | 2001-12-06 06:23:26 +0000 | [diff] [blame] | 75 | /* _XXX Py_QnewFlag should go away in 2.3.  It's true iff -Qnew is passed, | 
 | 76 |   on the command line, and is used in 2.2 by ceval.c to make all "/" divisions | 
 | 77 |   true divisions (which they will be in 2.3). */ | 
 | 78 | int _Py_QnewFlag = 0; | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 79 |  | 
| Mark Hammond | a43fd0c | 2003-02-19 00:33:33 +0000 | [diff] [blame] | 80 | /* Reference to 'warnings' module, to avoid importing it | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 81 |    on the fly when the import lock may be held.  See 683658/771097 | 
| Mark Hammond | a43fd0c | 2003-02-19 00:33:33 +0000 | [diff] [blame] | 82 | */ | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 83 | static PyObject *warnings_module = NULL; | 
 | 84 |  | 
 | 85 | /* Returns a borrowed reference to the 'warnings' module, or NULL. | 
 | 86 |    If the module is returned, it is guaranteed to have been obtained | 
 | 87 |    without acquiring the import lock | 
 | 88 | */ | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 89 | PyObject *PyModule_GetWarningsModule(void) | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 90 | { | 
 | 91 | 	PyObject *typ, *val, *tb; | 
 | 92 | 	PyObject *all_modules; | 
 | 93 | 	/* If we managed to get the module at init time, just use it */ | 
 | 94 | 	if (warnings_module) | 
 | 95 | 		return warnings_module; | 
 | 96 | 	/* If it wasn't available at init time, it may be available | 
 | 97 | 	   now in sys.modules (common scenario is frozen apps: import | 
 | 98 | 	   at init time fails, but the frozen init code sets up sys.path | 
 | 99 | 	   correctly, then does an implicit import of warnings for us | 
 | 100 | 	*/ | 
 | 101 | 	/* Save and restore any exceptions */ | 
 | 102 | 	PyErr_Fetch(&typ, &val, &tb); | 
 | 103 |  | 
| Mark Hammond | 5f4e8ca | 2003-07-16 01:54:38 +0000 | [diff] [blame] | 104 | 	all_modules = PySys_GetObject("modules"); | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 105 | 	if (all_modules) { | 
 | 106 | 		warnings_module = PyDict_GetItemString(all_modules, "warnings"); | 
 | 107 | 		/* We keep a ref in the global */ | 
 | 108 | 		Py_XINCREF(warnings_module); | 
 | 109 | 	} | 
 | 110 | 	PyErr_Restore(typ, val, tb); | 
 | 111 | 	return warnings_module; | 
 | 112 | } | 
| Mark Hammond | a43fd0c | 2003-02-19 00:33:33 +0000 | [diff] [blame] | 113 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 114 | static int initialized = 0; | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 115 |  | 
| Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 116 | /* API to access the initialized flag -- useful for esoteric use */ | 
| Guido van Rossum | e3c0d5e | 1997-08-22 04:20:13 +0000 | [diff] [blame] | 117 |  | 
 | 118 | int | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 119 | Py_IsInitialized(void) | 
| Guido van Rossum | e3c0d5e | 1997-08-22 04:20:13 +0000 | [diff] [blame] | 120 | { | 
 | 121 | 	return initialized; | 
 | 122 | } | 
 | 123 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 124 | /* Global initializations.  Can be undone by Py_Finalize().  Don't | 
 | 125 |    call this twice without an intervening Py_Finalize() call.  When | 
 | 126 |    initializations fail, a fatal error is issued and the function does | 
 | 127 |    not return.  On return, the first thread and interpreter state have | 
 | 128 |    been created. | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 129 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 130 |    Locking: you must hold the interpreter lock while calling this. | 
 | 131 |    (If the lock has not yet been initialized, that's equivalent to | 
 | 132 |    having the lock, but you cannot use multiple threads.) | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 133 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 134 | */ | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 135 |  | 
| Guido van Rossum | 9abaf4d | 2001-10-12 22:17:56 +0000 | [diff] [blame] | 136 | static int | 
 | 137 | add_flag(int flag, const char *envs) | 
 | 138 | { | 
 | 139 | 	int env = atoi(envs); | 
 | 140 | 	if (flag < env) | 
 | 141 | 		flag = env; | 
 | 142 | 	if (flag < 1) | 
 | 143 | 		flag = 1; | 
 | 144 | 	return flag; | 
 | 145 | } | 
 | 146 |  | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 147 | void | 
| Martin v. Löwis | 336e85f | 2004-08-19 11:31:58 +0000 | [diff] [blame] | 148 | Py_InitializeEx(int install_sigs) | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 149 | { | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 150 | 	PyInterpreterState *interp; | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 151 | 	PyThreadState *tstate; | 
 | 152 | 	PyObject *bimod, *sysmod; | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 153 | 	char *p; | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 154 | #if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET) | 
 | 155 | 	char *codeset; | 
 | 156 | 	char *saved_locale; | 
 | 157 | 	PyObject *sys_stream, *sys_isatty; | 
 | 158 | #endif | 
| Guido van Rossum | 70d893a | 2001-08-16 08:21:42 +0000 | [diff] [blame] | 159 | 	extern void _Py_ReadyTypes(void); | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 160 |  | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 161 | 	if (initialized) | 
| Guido van Rossum | aa61505 | 1997-08-20 22:40:18 +0000 | [diff] [blame] | 162 | 		return; | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 163 | 	initialized = 1; | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 164 |  | 
| Neil Schemenauer | 7d4bb9f | 2001-07-23 16:30:27 +0000 | [diff] [blame] | 165 | 	if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0') | 
| Guido van Rossum | 9abaf4d | 2001-10-12 22:17:56 +0000 | [diff] [blame] | 166 | 		Py_DebugFlag = add_flag(Py_DebugFlag, p); | 
| Neil Schemenauer | 7d4bb9f | 2001-07-23 16:30:27 +0000 | [diff] [blame] | 167 | 	if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0') | 
| Guido van Rossum | 9abaf4d | 2001-10-12 22:17:56 +0000 | [diff] [blame] | 168 | 		Py_VerboseFlag = add_flag(Py_VerboseFlag, p); | 
| Neil Schemenauer | 7d4bb9f | 2001-07-23 16:30:27 +0000 | [diff] [blame] | 169 | 	if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0') | 
| Guido van Rossum | 9abaf4d | 2001-10-12 22:17:56 +0000 | [diff] [blame] | 170 | 		Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p); | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 171 |  | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 172 | 	interp = PyInterpreterState_New(); | 
 | 173 | 	if (interp == NULL) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 174 | 		Py_FatalError("Py_Initialize: can't make first interpreter"); | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 175 |  | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 176 | 	tstate = PyThreadState_New(interp); | 
 | 177 | 	if (tstate == NULL) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 178 | 		Py_FatalError("Py_Initialize: can't make first thread"); | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 179 | 	(void) PyThreadState_Swap(tstate); | 
 | 180 |  | 
| Guido van Rossum | 70d893a | 2001-08-16 08:21:42 +0000 | [diff] [blame] | 181 | 	_Py_ReadyTypes(); | 
| Guido van Rossum | 528b7eb | 2001-08-07 17:24:28 +0000 | [diff] [blame] | 182 |  | 
| Neal Norwitz | b2501f4 | 2002-12-31 03:42:13 +0000 | [diff] [blame] | 183 | 	if (!_PyFrame_Init()) | 
| Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 184 | 		Py_FatalError("Py_Initialize: can't init frames"); | 
 | 185 |  | 
| Neal Norwitz | b2501f4 | 2002-12-31 03:42:13 +0000 | [diff] [blame] | 186 | 	if (!_PyInt_Init()) | 
| Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 187 | 		Py_FatalError("Py_Initialize: can't init ints"); | 
 | 188 |  | 
| Michael W. Hudson | ba283e2 | 2005-05-27 15:23:20 +0000 | [diff] [blame] | 189 | 	_PyFloat_Init(); | 
 | 190 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 191 | 	interp->modules = PyDict_New(); | 
 | 192 | 	if (interp->modules == NULL) | 
 | 193 | 		Py_FatalError("Py_Initialize: can't make modules dictionary"); | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 194 |  | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 195 | #ifdef Py_USING_UNICODE | 
| Guido van Rossum | c94044c | 2000-03-10 23:03:54 +0000 | [diff] [blame] | 196 | 	/* Init Unicode implementation; relies on the codec registry */ | 
 | 197 | 	_PyUnicode_Init(); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 198 | #endif | 
| Guido van Rossum | c94044c | 2000-03-10 23:03:54 +0000 | [diff] [blame] | 199 |  | 
| Barry Warsaw | f242aa0 | 2000-05-25 23:09:49 +0000 | [diff] [blame] | 200 | 	bimod = _PyBuiltin_Init(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 201 | 	if (bimod == NULL) | 
 | 202 | 		Py_FatalError("Py_Initialize: can't initialize __builtin__"); | 
| Guido van Rossum | 4a1f39a | 1997-11-04 19:36:18 +0000 | [diff] [blame] | 203 | 	interp->builtins = PyModule_GetDict(bimod); | 
 | 204 | 	Py_INCREF(interp->builtins); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 205 |  | 
 | 206 | 	sysmod = _PySys_Init(); | 
 | 207 | 	if (sysmod == NULL) | 
 | 208 | 		Py_FatalError("Py_Initialize: can't initialize sys"); | 
 | 209 | 	interp->sysdict = PyModule_GetDict(sysmod); | 
 | 210 | 	Py_INCREF(interp->sysdict); | 
 | 211 | 	_PyImport_FixupExtension("sys", "sys"); | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 212 | 	PySys_SetPath(Py_GetPath()); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 213 | 	PyDict_SetItemString(interp->sysdict, "modules", | 
 | 214 | 			     interp->modules); | 
 | 215 |  | 
| Guido van Rossum | 7c85ab8 | 1999-07-08 17:26:56 +0000 | [diff] [blame] | 216 | 	_PyImport_Init(); | 
 | 217 |  | 
| Barry Warsaw | f242aa0 | 2000-05-25 23:09:49 +0000 | [diff] [blame] | 218 | 	/* initialize builtin exceptions */ | 
| Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 219 | 	_PyExc_Init(); | 
| Barry Warsaw | 5821bc5 | 2001-08-13 23:04:56 +0000 | [diff] [blame] | 220 | 	_PyImport_FixupExtension("exceptions", "exceptions"); | 
| Barry Warsaw | f242aa0 | 2000-05-25 23:09:49 +0000 | [diff] [blame] | 221 |  | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 222 | 	/* phase 2 of builtins */ | 
| Barry Warsaw | 963b871 | 1997-09-18 16:42:02 +0000 | [diff] [blame] | 223 | 	_PyImport_FixupExtension("__builtin__", "__builtin__"); | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 224 |  | 
| Just van Rossum | 52e14d6 | 2002-12-30 22:08:05 +0000 | [diff] [blame] | 225 | 	_PyImportHooks_Init(); | 
 | 226 |  | 
| Martin v. Löwis | 336e85f | 2004-08-19 11:31:58 +0000 | [diff] [blame] | 227 | 	if (install_sigs) | 
 | 228 | 		initsigs(); /* Signal handling stuff, including initintr() */ | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 229 |  | 
 | 230 | 	initmain(); /* Module __main__ */ | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 231 | 	if (!Py_NoSiteFlag) | 
 | 232 | 		initsite(); /* Module site */ | 
| Just van Rossum | 5bfba3a | 2003-02-25 20:25:12 +0000 | [diff] [blame] | 233 |  | 
| Mark Hammond | 8d98d2c | 2003-04-19 15:41:53 +0000 | [diff] [blame] | 234 | 	/* auto-thread-state API, if available */ | 
 | 235 | #ifdef WITH_THREAD | 
 | 236 | 	_PyGILState_Init(interp, tstate); | 
 | 237 | #endif /* WITH_THREAD */ | 
 | 238 |  | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 239 | 	warnings_module = PyImport_ImportModule("warnings"); | 
 | 240 | 	if (!warnings_module) | 
 | 241 | 		PyErr_Clear(); | 
| Martin v. Löwis | 73d538b | 2003-03-05 15:13:47 +0000 | [diff] [blame] | 242 |  | 
 | 243 | #if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET) | 
 | 244 | 	/* On Unix, set the file system encoding according to the | 
 | 245 | 	   user's preference, if the CODESET names a well-known | 
 | 246 | 	   Python codec, and Py_FileSystemDefaultEncoding isn't | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 247 | 	   initialized by other means. Also set the encoding of | 
 | 248 | 	   stdin and stdout if these are terminals.  */ | 
 | 249 |  | 
| Martin v. Löwis | f56d015 | 2003-11-13 07:43:21 +0000 | [diff] [blame] | 250 | 	saved_locale = strdup(setlocale(LC_CTYPE, NULL)); | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 251 | 	setlocale(LC_CTYPE, ""); | 
 | 252 | 	codeset = nl_langinfo(CODESET); | 
 | 253 | 	if (codeset && *codeset) { | 
 | 254 | 		PyObject *enc = PyCodec_Encoder(codeset); | 
 | 255 | 		if (enc) { | 
 | 256 | 			codeset = strdup(codeset); | 
 | 257 | 			Py_DECREF(enc); | 
 | 258 | 		} else { | 
 | 259 | 			codeset = NULL; | 
 | 260 | 			PyErr_Clear(); | 
| Martin v. Löwis | 73d538b | 2003-03-05 15:13:47 +0000 | [diff] [blame] | 261 | 		} | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 262 | 	} else | 
 | 263 | 		codeset = NULL; | 
 | 264 | 	setlocale(LC_CTYPE, saved_locale); | 
| Martin v. Löwis | f56d015 | 2003-11-13 07:43:21 +0000 | [diff] [blame] | 265 | 	free(saved_locale); | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 266 |  | 
 | 267 | 	if (codeset) { | 
| Michael W. Hudson | 68debc9 | 2003-08-11 12:20:24 +0000 | [diff] [blame] | 268 | 		sys_stream = PySys_GetObject("stdin"); | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 269 | 		sys_isatty = PyObject_CallMethod(sys_stream, "isatty", ""); | 
 | 270 | 		if (!sys_isatty) | 
 | 271 | 			PyErr_Clear(); | 
 | 272 | 		if(sys_isatty && PyObject_IsTrue(sys_isatty)) { | 
 | 273 | 			if (!PyFile_SetEncoding(sys_stream, codeset)) | 
 | 274 | 				Py_FatalError("Cannot set codeset of stdin"); | 
 | 275 | 		} | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 276 | 		Py_XDECREF(sys_isatty); | 
 | 277 |  | 
 | 278 | 		sys_stream = PySys_GetObject("stdout"); | 
 | 279 | 		sys_isatty = PyObject_CallMethod(sys_stream, "isatty", ""); | 
 | 280 | 		if (!sys_isatty) | 
 | 281 | 			PyErr_Clear(); | 
 | 282 | 		if(sys_isatty && PyObject_IsTrue(sys_isatty)) { | 
 | 283 | 			if (!PyFile_SetEncoding(sys_stream, codeset)) | 
 | 284 | 				Py_FatalError("Cannot set codeset of stdout"); | 
 | 285 | 		} | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 286 | 		Py_XDECREF(sys_isatty); | 
 | 287 |  | 
| Martin v. Löwis | ea62d25 | 2006-04-03 10:56:49 +0000 | [diff] [blame] | 288 | 		sys_stream = PySys_GetObject("stderr"); | 
 | 289 | 		sys_isatty = PyObject_CallMethod(sys_stream, "isatty", ""); | 
 | 290 | 		if (!sys_isatty) | 
 | 291 | 			PyErr_Clear(); | 
 | 292 | 		if(sys_isatty && PyObject_IsTrue(sys_isatty)) { | 
 | 293 | 			if (!PyFile_SetEncoding(sys_stream, codeset)) | 
 | 294 | 				Py_FatalError("Cannot set codeset of stderr"); | 
 | 295 | 		} | 
 | 296 | 		Py_XDECREF(sys_isatty); | 
 | 297 |  | 
| Martin v. Löwis | a2c17c5 | 2003-08-09 09:47:11 +0000 | [diff] [blame] | 298 | 		if (!Py_FileSystemDefaultEncoding) | 
 | 299 | 			Py_FileSystemDefaultEncoding = codeset; | 
 | 300 | 		else | 
 | 301 | 			free(codeset); | 
| Martin v. Löwis | 73d538b | 2003-03-05 15:13:47 +0000 | [diff] [blame] | 302 | 	} | 
 | 303 | #endif | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 304 | } | 
 | 305 |  | 
| Martin v. Löwis | 336e85f | 2004-08-19 11:31:58 +0000 | [diff] [blame] | 306 | void | 
 | 307 | Py_Initialize(void) | 
 | 308 | { | 
 | 309 | 	Py_InitializeEx(1); | 
 | 310 | } | 
 | 311 |  | 
 | 312 |  | 
| Guido van Rossum | 2edcf0d | 1998-12-15 16:12:00 +0000 | [diff] [blame] | 313 | #ifdef COUNT_ALLOCS | 
| Martin v. Löwis | 45294a9 | 2006-04-18 06:24:08 +0000 | [diff] [blame] | 314 | extern void dump_counts(FILE*); | 
| Guido van Rossum | 2edcf0d | 1998-12-15 16:12:00 +0000 | [diff] [blame] | 315 | #endif | 
 | 316 |  | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 317 | /* Undo the effect of Py_Initialize(). | 
 | 318 |  | 
 | 319 |    Beware: if multiple interpreter and/or thread states exist, these | 
 | 320 |    are not wiped out; only the current thread and interpreter state | 
 | 321 |    are deleted.  But since everything else is deleted, those other | 
 | 322 |    interpreter and thread states should no longer be used. | 
 | 323 |  | 
 | 324 |    (XXX We should do better, e.g. wipe out all interpreters and | 
 | 325 |    threads.) | 
 | 326 |  | 
 | 327 |    Locking: as above. | 
 | 328 |  | 
 | 329 | */ | 
 | 330 |  | 
 | 331 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 332 | Py_Finalize(void) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 333 | { | 
 | 334 | 	PyInterpreterState *interp; | 
 | 335 | 	PyThreadState *tstate; | 
 | 336 |  | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 337 | 	if (!initialized) | 
| Guido van Rossum | aa61505 | 1997-08-20 22:40:18 +0000 | [diff] [blame] | 338 | 		return; | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 339 |  | 
| Tim Peters | 384fd10 | 2001-01-21 03:40:37 +0000 | [diff] [blame] | 340 | 	/* The interpreter is still entirely intact at this point, and the | 
 | 341 | 	 * exit funcs may be relying on that.  In particular, if some thread | 
 | 342 | 	 * or exit func is still waiting to do an import, the import machinery | 
 | 343 | 	 * expects Py_IsInitialized() to return true.  So don't say the | 
 | 344 | 	 * interpreter is uninitialized until after the exit funcs have run. | 
 | 345 | 	 * Note that Threading.py uses an exit func to do a join on all the | 
 | 346 | 	 * threads created thru it, so this also protects pending imports in | 
 | 347 | 	 * the threads created via Threading. | 
 | 348 | 	 */ | 
| Guido van Rossum | 4cc462e | 1998-01-19 22:00:38 +0000 | [diff] [blame] | 349 | 	call_sys_exitfunc(); | 
| Tim Peters | 384fd10 | 2001-01-21 03:40:37 +0000 | [diff] [blame] | 350 | 	initialized = 0; | 
| Guido van Rossum | 4cc462e | 1998-01-19 22:00:38 +0000 | [diff] [blame] | 351 |  | 
| Guido van Rossum | 3a44e1b | 1997-11-03 21:58:47 +0000 | [diff] [blame] | 352 | 	/* Get current thread state and interpreter pointer */ | 
| Nicholas Bastin | e5662ae | 2004-03-24 22:22:12 +0000 | [diff] [blame] | 353 | 	tstate = PyThreadState_GET(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 354 | 	interp = tstate->interp; | 
 | 355 |  | 
| Guido van Rossum | 3a44e1b | 1997-11-03 21:58:47 +0000 | [diff] [blame] | 356 | 	/* Disable signal handling */ | 
 | 357 | 	PyOS_FiniInterrupts(); | 
 | 358 |  | 
| Mark Hammond | a43fd0c | 2003-02-19 00:33:33 +0000 | [diff] [blame] | 359 | 	/* drop module references we saved */ | 
| Mark Hammond | edd0773 | 2003-07-15 23:03:55 +0000 | [diff] [blame] | 360 | 	Py_XDECREF(warnings_module); | 
 | 361 | 	warnings_module = NULL; | 
| Mark Hammond | a43fd0c | 2003-02-19 00:33:33 +0000 | [diff] [blame] | 362 |  | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 363 | 	/* Collect garbage.  This may call finalizers; it's nice to call these | 
| Tim Peters | 1d7323e | 2003-12-01 21:35:27 +0000 | [diff] [blame] | 364 | 	 * before all modules are destroyed. | 
 | 365 | 	 * XXX If a __del__ or weakref callback is triggered here, and tries to | 
 | 366 | 	 * XXX import a module, bad things can happen, because Python no | 
 | 367 | 	 * XXX longer believes it's initialized. | 
 | 368 | 	 * XXX     Fatal Python error: Interpreter not initialized (version mismatch?) | 
 | 369 | 	 * XXX is easy to provoke that way.  I've also seen, e.g., | 
 | 370 | 	 * XXX     Exception exceptions.ImportError: 'No module named sha' | 
 | 371 | 	 * XXX         in <function callback at 0x008F5718> ignored | 
 | 372 | 	 * XXX but I'm unclear on exactly how that one happens.  In any case, | 
 | 373 | 	 * XXX I haven't seen a real-life report of either of these. | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 374 | 	 */ | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 375 | 	PyGC_Collect(); | 
| Martin v. Löwis | 45294a9 | 2006-04-18 06:24:08 +0000 | [diff] [blame] | 376 | #ifdef COUNT_ALLOCS | 
 | 377 | 	/* With COUNT_ALLOCS, it helps to run GC multiple times: | 
 | 378 | 	   each collection might release some types from the type | 
 | 379 | 	   list, so they become garbage. */ | 
 | 380 | 	while (PyGC_Collect() > 0) | 
 | 381 | 		/* nothing */; | 
 | 382 | #endif | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 383 |  | 
| Guido van Rossum | 3a44e1b | 1997-11-03 21:58:47 +0000 | [diff] [blame] | 384 | 	/* Destroy all modules */ | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 385 | 	PyImport_Cleanup(); | 
| Guido van Rossum | 3a44e1b | 1997-11-03 21:58:47 +0000 | [diff] [blame] | 386 |  | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 387 | 	/* Collect final garbage.  This disposes of cycles created by | 
| Tim Peters | 1d7323e | 2003-12-01 21:35:27 +0000 | [diff] [blame] | 388 | 	 * new-style class definitions, for example. | 
 | 389 | 	 * XXX This is disabled because it caused too many problems.  If | 
 | 390 | 	 * XXX a __del__ or weakref callback triggers here, Python code has | 
 | 391 | 	 * XXX a hard time running, because even the sys module has been | 
 | 392 | 	 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc). | 
 | 393 | 	 * XXX One symptom is a sequence of information-free messages | 
 | 394 | 	 * XXX coming from threads (if a __del__ or callback is invoked, | 
 | 395 | 	 * XXX other threads can execute too, and any exception they encounter | 
 | 396 | 	 * XXX triggers a comedy of errors as subsystem after subsystem | 
 | 397 | 	 * XXX fails to find what it *expects* to find in sys to help report | 
 | 398 | 	 * XXX the exception and consequent unexpected failures).  I've also | 
 | 399 | 	 * XXX seen segfaults then, after adding print statements to the | 
 | 400 | 	 * XXX Python code getting called. | 
 | 401 | 	 */ | 
 | 402 | #if 0 | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 403 | 	PyGC_Collect(); | 
| Tim Peters | 1d7323e | 2003-12-01 21:35:27 +0000 | [diff] [blame] | 404 | #endif | 
| Guido van Rossum | e13ddc9 | 2003-04-17 17:29:22 +0000 | [diff] [blame] | 405 |  | 
| Guido van Rossum | 1707aad | 1997-12-08 23:43:45 +0000 | [diff] [blame] | 406 | 	/* Destroy the database used by _PyImport_{Fixup,Find}Extension */ | 
 | 407 | 	_PyImport_Fini(); | 
 | 408 |  | 
 | 409 | 	/* Debugging stuff */ | 
 | 410 | #ifdef COUNT_ALLOCS | 
| Martin v. Löwis | 45294a9 | 2006-04-18 06:24:08 +0000 | [diff] [blame] | 411 | 	dump_counts(stdout); | 
| Guido van Rossum | 1707aad | 1997-12-08 23:43:45 +0000 | [diff] [blame] | 412 | #endif | 
 | 413 |  | 
| Tim Peters | 62e97f0 | 2006-03-28 21:44:32 +0000 | [diff] [blame] | 414 | 	PRINT_TOTAL_REFS(); | 
| Guido van Rossum | 1707aad | 1997-12-08 23:43:45 +0000 | [diff] [blame] | 415 |  | 
| Tim Peters | 9cf25ce | 2003-04-17 15:21:01 +0000 | [diff] [blame] | 416 | #ifdef Py_TRACE_REFS | 
 | 417 | 	/* Display all objects still alive -- this can invoke arbitrary | 
 | 418 | 	 * __repr__ overrides, so requires a mostly-intact interpreter. | 
 | 419 | 	 * Alas, a lot of stuff may still be alive now that will be cleaned | 
 | 420 | 	 * up later. | 
 | 421 | 	 */ | 
| Tim Peters | 269b2a6 | 2003-04-17 19:52:29 +0000 | [diff] [blame] | 422 | 	if (Py_GETENV("PYTHONDUMPREFS")) | 
| Tim Peters | 9cf25ce | 2003-04-17 15:21:01 +0000 | [diff] [blame] | 423 | 		_Py_PrintReferences(stderr); | 
| Tim Peters | 9cf25ce | 2003-04-17 15:21:01 +0000 | [diff] [blame] | 424 | #endif /* Py_TRACE_REFS */ | 
 | 425 |  | 
| Mark Hammond | 6cb9029 | 2003-04-22 11:18:00 +0000 | [diff] [blame] | 426 | 	/* Cleanup auto-thread-state */ | 
 | 427 | #ifdef WITH_THREAD | 
 | 428 | 	_PyGILState_Fini(); | 
 | 429 | #endif /* WITH_THREAD */ | 
 | 430 |  | 
| Guido van Rossum | d922fa4 | 2003-04-15 14:10:09 +0000 | [diff] [blame] | 431 | 	/* Clear interpreter state */ | 
| Barry Warsaw | f242aa0 | 2000-05-25 23:09:49 +0000 | [diff] [blame] | 432 | 	PyInterpreterState_Clear(interp); | 
| Guido van Rossum | d922fa4 | 2003-04-15 14:10:09 +0000 | [diff] [blame] | 433 |  | 
| Anthony Baxter | 12b6f6c | 2005-03-29 13:36:16 +0000 | [diff] [blame] | 434 | 	/* Now we decref the exception classes.  After this point nothing | 
 | 435 | 	   can raise an exception.  That's okay, because each Fini() method | 
 | 436 | 	   below has been checked to make sure no exceptions are ever | 
 | 437 | 	   raised. | 
 | 438 | 	*/ | 
 | 439 |  | 
 | 440 | 	_PyExc_Fini(); | 
 | 441 |  | 
| Guido van Rossum | d922fa4 | 2003-04-15 14:10:09 +0000 | [diff] [blame] | 442 | 	/* Delete current thread */ | 
| Barry Warsaw | f242aa0 | 2000-05-25 23:09:49 +0000 | [diff] [blame] | 443 | 	PyThreadState_Swap(NULL); | 
 | 444 | 	PyInterpreterState_Delete(interp); | 
 | 445 |  | 
| Guido van Rossum | d922fa4 | 2003-04-15 14:10:09 +0000 | [diff] [blame] | 446 | 	/* Sundry finalizers */ | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 447 | 	PyMethod_Fini(); | 
 | 448 | 	PyFrame_Fini(); | 
 | 449 | 	PyCFunction_Fini(); | 
 | 450 | 	PyTuple_Fini(); | 
| Raymond Hettinger | fb09f0e | 2004-10-07 03:58:07 +0000 | [diff] [blame] | 451 | 	PyList_Fini(); | 
| Raymond Hettinger | d794666 | 2005-08-01 21:39:29 +0000 | [diff] [blame] | 452 | 	PySet_Fini(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 453 | 	PyString_Fini(); | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 454 | 	PyInt_Fini(); | 
 | 455 | 	PyFloat_Fini(); | 
 | 456 |  | 
| Marc-André Lemburg | 95de5c1 | 2002-04-08 08:19:36 +0000 | [diff] [blame] | 457 | #ifdef Py_USING_UNICODE | 
 | 458 | 	/* Cleanup Unicode implementation */ | 
 | 459 | 	_PyUnicode_Fini(); | 
 | 460 | #endif | 
 | 461 |  | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 462 | 	/* XXX Still allocated: | 
 | 463 | 	   - various static ad-hoc pointers to interned strings | 
 | 464 | 	   - int and float free list blocks | 
 | 465 | 	   - whatever various modules and libraries allocate | 
 | 466 | 	*/ | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 467 |  | 
 | 468 | 	PyGrammar_RemoveAccelerators(&_PyParser_Grammar); | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 469 |  | 
| Tim Peters | 269b2a6 | 2003-04-17 19:52:29 +0000 | [diff] [blame] | 470 | #ifdef Py_TRACE_REFS | 
 | 471 | 	/* Display addresses (& refcnts) of all objects still alive. | 
 | 472 | 	 * An address can be used to find the repr of the object, printed | 
 | 473 | 	 * above by _Py_PrintReferences. | 
 | 474 | 	 */ | 
 | 475 | 	if (Py_GETENV("PYTHONDUMPREFS")) | 
 | 476 | 		_Py_PrintReferenceAddresses(stderr); | 
 | 477 | #endif /* Py_TRACE_REFS */ | 
| Tim Peters | 0e87118 | 2002-04-13 08:29:14 +0000 | [diff] [blame] | 478 | #ifdef PYMALLOC_DEBUG | 
 | 479 | 	if (Py_GETENV("PYTHONMALLOCSTATS")) | 
 | 480 | 		_PyObject_DebugMallocStats(); | 
 | 481 | #endif | 
 | 482 |  | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 483 | 	call_ll_exitfuncs(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 484 | } | 
 | 485 |  | 
 | 486 | /* Create and initialize a new interpreter and thread, and return the | 
 | 487 |    new thread.  This requires that Py_Initialize() has been called | 
 | 488 |    first. | 
 | 489 |  | 
 | 490 |    Unsuccessful initialization yields a NULL pointer.  Note that *no* | 
 | 491 |    exception information is available even in this case -- the | 
 | 492 |    exception information is held in the thread, and there is no | 
 | 493 |    thread. | 
 | 494 |  | 
 | 495 |    Locking: as above. | 
 | 496 |  | 
 | 497 | */ | 
 | 498 |  | 
 | 499 | PyThreadState * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 500 | Py_NewInterpreter(void) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 501 | { | 
 | 502 | 	PyInterpreterState *interp; | 
 | 503 | 	PyThreadState *tstate, *save_tstate; | 
 | 504 | 	PyObject *bimod, *sysmod; | 
 | 505 |  | 
 | 506 | 	if (!initialized) | 
 | 507 | 		Py_FatalError("Py_NewInterpreter: call Py_Initialize first"); | 
 | 508 |  | 
 | 509 | 	interp = PyInterpreterState_New(); | 
 | 510 | 	if (interp == NULL) | 
 | 511 | 		return NULL; | 
 | 512 |  | 
 | 513 | 	tstate = PyThreadState_New(interp); | 
 | 514 | 	if (tstate == NULL) { | 
 | 515 | 		PyInterpreterState_Delete(interp); | 
 | 516 | 		return NULL; | 
 | 517 | 	} | 
 | 518 |  | 
 | 519 | 	save_tstate = PyThreadState_Swap(tstate); | 
 | 520 |  | 
 | 521 | 	/* XXX The following is lax in error checking */ | 
 | 522 |  | 
 | 523 | 	interp->modules = PyDict_New(); | 
 | 524 |  | 
 | 525 | 	bimod = _PyImport_FindExtension("__builtin__", "__builtin__"); | 
 | 526 | 	if (bimod != NULL) { | 
| Guido van Rossum | 4a1f39a | 1997-11-04 19:36:18 +0000 | [diff] [blame] | 527 | 		interp->builtins = PyModule_GetDict(bimod); | 
 | 528 | 		Py_INCREF(interp->builtins); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 529 | 	} | 
 | 530 | 	sysmod = _PyImport_FindExtension("sys", "sys"); | 
 | 531 | 	if (bimod != NULL && sysmod != NULL) { | 
 | 532 | 		interp->sysdict = PyModule_GetDict(sysmod); | 
 | 533 | 		Py_INCREF(interp->sysdict); | 
 | 534 | 		PySys_SetPath(Py_GetPath()); | 
 | 535 | 		PyDict_SetItemString(interp->sysdict, "modules", | 
 | 536 | 				     interp->modules); | 
| Martin v. Löwis | d7ceb22 | 2003-01-22 09:00:38 +0000 | [diff] [blame] | 537 | 		_PyImportHooks_Init(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 538 | 		initmain(); | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 539 | 		if (!Py_NoSiteFlag) | 
 | 540 | 			initsite(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 541 | 	} | 
 | 542 |  | 
 | 543 | 	if (!PyErr_Occurred()) | 
 | 544 | 		return tstate; | 
 | 545 |  | 
 | 546 | 	/* Oops, it didn't work.  Undo it all. */ | 
 | 547 |  | 
 | 548 | 	PyErr_Print(); | 
 | 549 | 	PyThreadState_Clear(tstate); | 
 | 550 | 	PyThreadState_Swap(save_tstate); | 
 | 551 | 	PyThreadState_Delete(tstate); | 
 | 552 | 	PyInterpreterState_Delete(interp); | 
 | 553 |  | 
 | 554 | 	return NULL; | 
 | 555 | } | 
 | 556 |  | 
 | 557 | /* Delete an interpreter and its last thread.  This requires that the | 
 | 558 |    given thread state is current, that the thread has no remaining | 
 | 559 |    frames, and that it is its interpreter's only remaining thread. | 
 | 560 |    It is a fatal error to violate these constraints. | 
 | 561 |  | 
 | 562 |    (Py_Finalize() doesn't have these constraints -- it zaps | 
 | 563 |    everything, regardless.) | 
 | 564 |  | 
 | 565 |    Locking: as above. | 
 | 566 |  | 
 | 567 | */ | 
 | 568 |  | 
 | 569 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 570 | Py_EndInterpreter(PyThreadState *tstate) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 571 | { | 
 | 572 | 	PyInterpreterState *interp = tstate->interp; | 
 | 573 |  | 
| Nicholas Bastin | e5662ae | 2004-03-24 22:22:12 +0000 | [diff] [blame] | 574 | 	if (tstate != PyThreadState_GET()) | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 575 | 		Py_FatalError("Py_EndInterpreter: thread is not current"); | 
 | 576 | 	if (tstate->frame != NULL) | 
 | 577 | 		Py_FatalError("Py_EndInterpreter: thread still has a frame"); | 
 | 578 | 	if (tstate != interp->tstate_head || tstate->next != NULL) | 
 | 579 | 		Py_FatalError("Py_EndInterpreter: not the last thread"); | 
 | 580 |  | 
 | 581 | 	PyImport_Cleanup(); | 
 | 582 | 	PyInterpreterState_Clear(interp); | 
 | 583 | 	PyThreadState_Swap(NULL); | 
 | 584 | 	PyInterpreterState_Delete(interp); | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 585 | } | 
 | 586 |  | 
 | 587 | static char *progname = "python"; | 
 | 588 |  | 
 | 589 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 590 | Py_SetProgramName(char *pn) | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 591 | { | 
 | 592 | 	if (pn && *pn) | 
 | 593 | 		progname = pn; | 
 | 594 | } | 
 | 595 |  | 
 | 596 | char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 597 | Py_GetProgramName(void) | 
| Guido van Rossum | ad6dfda | 1997-07-19 19:17:22 +0000 | [diff] [blame] | 598 | { | 
 | 599 | 	return progname; | 
| Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 600 | } | 
 | 601 |  | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 602 | static char *default_home = NULL; | 
 | 603 |  | 
 | 604 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 605 | Py_SetPythonHome(char *home) | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 606 | { | 
 | 607 | 	default_home = home; | 
 | 608 | } | 
 | 609 |  | 
 | 610 | char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 611 | Py_GetPythonHome(void) | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 612 | { | 
 | 613 | 	char *home = default_home; | 
| Neil Schemenauer | 7d4bb9f | 2001-07-23 16:30:27 +0000 | [diff] [blame] | 614 | 	if (home == NULL && !Py_IgnoreEnvironmentFlag) | 
 | 615 | 		home = Py_GETENV("PYTHONHOME"); | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 616 | 	return home; | 
 | 617 | } | 
 | 618 |  | 
| Guido van Rossum | 6135a87 | 1995-01-09 17:53:26 +0000 | [diff] [blame] | 619 | /* Create __main__ module */ | 
 | 620 |  | 
 | 621 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 622 | initmain(void) | 
| Guido van Rossum | 6135a87 | 1995-01-09 17:53:26 +0000 | [diff] [blame] | 623 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 624 | 	PyObject *m, *d; | 
 | 625 | 	m = PyImport_AddModule("__main__"); | 
| Guido van Rossum | 6135a87 | 1995-01-09 17:53:26 +0000 | [diff] [blame] | 626 | 	if (m == NULL) | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 627 | 		Py_FatalError("can't create __main__ module"); | 
 | 628 | 	d = PyModule_GetDict(m); | 
 | 629 | 	if (PyDict_GetItemString(d, "__builtins__") == NULL) { | 
| Guido van Rossum | 858cb73 | 1997-11-19 16:15:37 +0000 | [diff] [blame] | 630 | 		PyObject *bimod = PyImport_ImportModule("__builtin__"); | 
 | 631 | 		if (bimod == NULL || | 
 | 632 | 		    PyDict_SetItemString(d, "__builtins__", bimod) != 0) | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 633 | 			Py_FatalError("can't add __builtins__ to __main__"); | 
| Barry Warsaw | 3d05b1a | 1999-01-29 21:30:22 +0000 | [diff] [blame] | 634 | 		Py_DECREF(bimod); | 
| Guido van Rossum | 6135a87 | 1995-01-09 17:53:26 +0000 | [diff] [blame] | 635 | 	} | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 636 | } | 
 | 637 |  | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 638 | /* Import the site module (not into __main__ though) */ | 
 | 639 |  | 
 | 640 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 641 | initsite(void) | 
| Guido van Rossum | dcc0c13 | 1997-08-29 22:32:42 +0000 | [diff] [blame] | 642 | { | 
 | 643 | 	PyObject *m, *f; | 
 | 644 | 	m = PyImport_ImportModule("site"); | 
 | 645 | 	if (m == NULL) { | 
 | 646 | 		f = PySys_GetObject("stderr"); | 
 | 647 | 		if (Py_VerboseFlag) { | 
 | 648 | 			PyFile_WriteString( | 
 | 649 | 				"'import site' failed; traceback:\n", f); | 
 | 650 | 			PyErr_Print(); | 
 | 651 | 		} | 
 | 652 | 		else { | 
 | 653 | 			PyFile_WriteString( | 
 | 654 | 			  "'import site' failed; use -v for traceback\n", f); | 
 | 655 | 			PyErr_Clear(); | 
 | 656 | 		} | 
 | 657 | 	} | 
 | 658 | 	else { | 
 | 659 | 		Py_DECREF(m); | 
 | 660 | 	} | 
 | 661 | } | 
 | 662 |  | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 663 | /* Parse input from a file and execute it */ | 
 | 664 |  | 
 | 665 | int | 
| Martin v. Löwis | 056a69c | 2006-03-01 16:55:42 +0000 | [diff] [blame] | 666 | PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,  | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 667 | 		     PyCompilerFlags *flags) | 
 | 668 | { | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 669 | 	if (filename == NULL) | 
 | 670 | 		filename = "???"; | 
| Guido van Rossum | 0df002c | 2000-08-27 19:21:52 +0000 | [diff] [blame] | 671 | 	if (Py_FdIsInteractive(fp, filename)) { | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 672 | 		int err = PyRun_InteractiveLoopFlags(fp, filename, flags); | 
| Guido van Rossum | 0df002c | 2000-08-27 19:21:52 +0000 | [diff] [blame] | 673 | 		if (closeit) | 
 | 674 | 			fclose(fp); | 
 | 675 | 		return err; | 
 | 676 | 	} | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 677 | 	else | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 678 | 		return PyRun_SimpleFileExFlags(fp, filename, closeit, flags); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 679 | } | 
 | 680 |  | 
 | 681 | int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 682 | PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags) | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 683 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 684 | 	PyObject *v; | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 685 | 	int ret; | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 686 | 	PyCompilerFlags local_flags; | 
| Jeremy Hylton | 9f324e9 | 2001-03-01 22:59:14 +0000 | [diff] [blame] | 687 |  | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 688 | 	if (flags == NULL) { | 
 | 689 | 		flags = &local_flags; | 
| Tim Peters | 5ba5866 | 2001-07-16 02:29:45 +0000 | [diff] [blame] | 690 | 		local_flags.cf_flags = 0; | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 691 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 692 | 	v = PySys_GetObject("ps1"); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 693 | 	if (v == NULL) { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 694 | 		PySys_SetObject("ps1", v = PyString_FromString(">>> ")); | 
 | 695 | 		Py_XDECREF(v); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 696 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 697 | 	v = PySys_GetObject("ps2"); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 698 | 	if (v == NULL) { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 699 | 		PySys_SetObject("ps2", v = PyString_FromString("... ")); | 
 | 700 | 		Py_XDECREF(v); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 701 | 	} | 
 | 702 | 	for (;;) { | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 703 | 		ret = PyRun_InteractiveOneFlags(fp, filename, flags); | 
| Tim Peters | 62e97f0 | 2006-03-28 21:44:32 +0000 | [diff] [blame] | 704 | 		PRINT_TOTAL_REFS(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 705 | 		if (ret == E_EOF) | 
 | 706 | 			return 0; | 
 | 707 | 		/* | 
 | 708 | 		if (ret == E_NOMEM) | 
 | 709 | 			return -1; | 
 | 710 | 		*/ | 
 | 711 | 	} | 
 | 712 | } | 
 | 713 |  | 
| Neil Schemenauer | c24ea08 | 2002-03-22 23:53:36 +0000 | [diff] [blame] | 714 | /* compute parser flags based on compiler flags */ | 
| Neil Schemenauer | c24ea08 | 2002-03-22 23:53:36 +0000 | [diff] [blame] | 715 | #define PARSER_FLAGS(flags) \ | 
| Thomas Wouters | 34aa7ba | 2006-02-28 19:02:24 +0000 | [diff] [blame] | 716 | 	((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \ | 
 | 717 | 		      PyPARSE_DONT_IMPLY_DEDENT : 0) \ | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 718 | 		    | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \ | 
 | 719 | 		       PyPARSE_WITH_IS_KEYWORD : 0)) : 0) | 
| Neil Schemenauer | c24ea08 | 2002-03-22 23:53:36 +0000 | [diff] [blame] | 720 |  | 
| Jeremy Hylton | 9f324e9 | 2001-03-01 22:59:14 +0000 | [diff] [blame] | 721 | int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 722 | PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags) | 
| Jeremy Hylton | 9f324e9 | 2001-03-01 22:59:14 +0000 | [diff] [blame] | 723 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 724 | 	PyObject *m, *d, *v, *w; | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 725 | 	mod_ty mod; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 726 | 	PyArena *arena; | 
| Guido van Rossum | ddc3fb5 | 1997-11-25 20:58:13 +0000 | [diff] [blame] | 727 | 	char *ps1 = "", *ps2 = ""; | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 728 | 	int errcode = 0; | 
| Tim Peters | fe2127d | 2001-07-16 05:37:24 +0000 | [diff] [blame] | 729 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 730 | 	v = PySys_GetObject("ps1"); | 
| Guido van Rossum | ddc3fb5 | 1997-11-25 20:58:13 +0000 | [diff] [blame] | 731 | 	if (v != NULL) { | 
 | 732 | 		v = PyObject_Str(v); | 
 | 733 | 		if (v == NULL) | 
 | 734 | 			PyErr_Clear(); | 
 | 735 | 		else if (PyString_Check(v)) | 
 | 736 | 			ps1 = PyString_AsString(v); | 
 | 737 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 738 | 	w = PySys_GetObject("ps2"); | 
| Guido van Rossum | ddc3fb5 | 1997-11-25 20:58:13 +0000 | [diff] [blame] | 739 | 	if (w != NULL) { | 
 | 740 | 		w = PyObject_Str(w); | 
 | 741 | 		if (w == NULL) | 
 | 742 | 			PyErr_Clear(); | 
 | 743 | 		else if (PyString_Check(w)) | 
 | 744 | 			ps2 = PyString_AsString(w); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 745 | 	} | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 746 | 	arena = PyArena_New(); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 747 | 	mod = PyParser_ASTFromFile(fp, filename,  | 
 | 748 | 				   Py_single_input, ps1, ps2, | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 749 | 				   flags, &errcode, arena); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 750 | 	Py_XDECREF(v); | 
 | 751 | 	Py_XDECREF(w); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 752 | 	if (mod == NULL) { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 753 | 		PyArena_Free(arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 754 | 		if (errcode == E_EOF) { | 
 | 755 | 			PyErr_Clear(); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 756 | 			return E_EOF; | 
 | 757 | 		} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 758 | 		PyErr_Print(); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 759 | 		return -1; | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 760 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 761 | 	m = PyImport_AddModule("__main__"); | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 762 | 	if (m == NULL) { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 763 | 		PyArena_Free(arena); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 764 | 		return -1; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 765 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 766 | 	d = PyModule_GetDict(m); | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 767 | 	v = run_mod(mod, filename, d, d, flags, arena); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 768 | 	PyArena_Free(arena); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 769 | 	if (v == NULL) { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 770 | 		PyErr_Print(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 771 | 		return -1; | 
 | 772 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 773 | 	Py_DECREF(v); | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 774 | 	if (Py_FlushLine()) | 
 | 775 | 		PyErr_Clear(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 776 | 	return 0; | 
 | 777 | } | 
 | 778 |  | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 779 | /* Check whether a file maybe a pyc file: Look at the extension, | 
 | 780 |    the file type, and, if we may close it, at the first few bytes. */ | 
 | 781 |  | 
 | 782 | static int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 783 | maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit) | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 784 | { | 
 | 785 | 	if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0) | 
 | 786 | 		return 1; | 
 | 787 |  | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 788 | 	/* Only look into the file if we are allowed to close it, since | 
 | 789 | 	   it then should also be seekable. */ | 
 | 790 | 	if (closeit) { | 
 | 791 | 		/* Read only two bytes of the magic. If the file was opened in | 
 | 792 | 		   text mode, the bytes 3 and 4 of the magic (\r\n) might not | 
 | 793 | 		   be read as they are on disk. */ | 
 | 794 | 		unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF; | 
 | 795 | 		unsigned char buf[2]; | 
| Tim Peters | 3e87656 | 2001-02-11 04:35:39 +0000 | [diff] [blame] | 796 | 		/* Mess:  In case of -x, the stream is NOT at its start now, | 
 | 797 | 		   and ungetc() was used to push back the first newline, | 
| Tim Peters | 6f5a4ef | 2001-02-17 22:02:34 +0000 | [diff] [blame] | 798 | 		   which makes the current stream position formally undefined, | 
 | 799 | 		   and a x-platform nightmare. | 
 | 800 | 		   Unfortunately, we have no direct way to know whether -x | 
 | 801 | 		   was specified.  So we use a terrible hack:  if the current | 
 | 802 | 		   stream position is not 0, we assume -x was specified, and | 
 | 803 | 		   give up.  Bug 132850 on SourceForge spells out the | 
 | 804 | 		   hopelessness of trying anything else (fseek and ftell | 
 | 805 | 		   don't work predictably x-platform for text-mode files). | 
| Tim Peters | 3e87656 | 2001-02-11 04:35:39 +0000 | [diff] [blame] | 806 | 		*/ | 
| Tim Peters | 3e87656 | 2001-02-11 04:35:39 +0000 | [diff] [blame] | 807 | 		int ispyc = 0; | 
| Tim Peters | 6f5a4ef | 2001-02-17 22:02:34 +0000 | [diff] [blame] | 808 | 		if (ftell(fp) == 0) { | 
 | 809 | 			if (fread(buf, 1, 2, fp) == 2 && | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 810 | 			    ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic) | 
| Tim Peters | 6f5a4ef | 2001-02-17 22:02:34 +0000 | [diff] [blame] | 811 | 				ispyc = 1; | 
 | 812 | 			rewind(fp); | 
 | 813 | 		} | 
| Tim Peters | 3e87656 | 2001-02-11 04:35:39 +0000 | [diff] [blame] | 814 | 		return ispyc; | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 815 | 	} | 
 | 816 | 	return 0; | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 817 | } | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 818 |  | 
| Guido van Rossum | 0df002c | 2000-08-27 19:21:52 +0000 | [diff] [blame] | 819 | int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 820 | PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit, | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 821 | 			PyCompilerFlags *flags) | 
 | 822 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 823 | 	PyObject *m, *d, *v; | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 824 | 	const char *ext; | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 825 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 826 | 	m = PyImport_AddModule("__main__"); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 827 | 	if (m == NULL) | 
 | 828 | 		return -1; | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 829 | 	d = PyModule_GetDict(m); | 
| Fred Drake | 8ed0204 | 2002-10-17 21:24:58 +0000 | [diff] [blame] | 830 | 	if (PyDict_GetItemString(d, "__file__") == NULL) { | 
 | 831 | 		PyObject *f = PyString_FromString(filename); | 
 | 832 | 		if (f == NULL) | 
 | 833 | 			return -1; | 
 | 834 | 		if (PyDict_SetItemString(d, "__file__", f) < 0) { | 
 | 835 | 			Py_DECREF(f); | 
 | 836 | 			return -1; | 
 | 837 | 		} | 
 | 838 | 		Py_DECREF(f); | 
 | 839 | 	} | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 840 | 	ext = filename + strlen(filename) - 4; | 
| Martin v. Löwis | be4c0f5 | 2001-01-04 20:30:56 +0000 | [diff] [blame] | 841 | 	if (maybe_pyc_file(fp, filename, ext, closeit)) { | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 842 | 		/* Try to run a pyc file. First, re-open in binary */ | 
| Guido van Rossum | 0df002c | 2000-08-27 19:21:52 +0000 | [diff] [blame] | 843 | 		if (closeit) | 
 | 844 | 			fclose(fp); | 
| Fred Drake | 8ed0204 | 2002-10-17 21:24:58 +0000 | [diff] [blame] | 845 | 		if ((fp = fopen(filename, "rb")) == NULL) { | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 846 | 			fprintf(stderr, "python: Can't reopen .pyc file\n"); | 
 | 847 | 			return -1; | 
 | 848 | 		} | 
| Guido van Rossum | 2a7f58d | 1997-04-02 05:28:38 +0000 | [diff] [blame] | 849 | 		/* Turn on optimization if a .pyo file is given */ | 
 | 850 | 		if (strcmp(ext, ".pyo") == 0) | 
 | 851 | 			Py_OptimizeFlag = 1; | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 852 | 		v = run_pyc_file(fp, filename, d, d, flags); | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 853 | 	} else { | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 854 | 		v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d, | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 855 | 				      closeit, flags); | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 856 | 	} | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 857 | 	if (v == NULL) { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 858 | 		PyErr_Print(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 859 | 		return -1; | 
 | 860 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 861 | 	Py_DECREF(v); | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 862 | 	if (Py_FlushLine()) | 
 | 863 | 		PyErr_Clear(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 864 | 	return 0; | 
 | 865 | } | 
 | 866 |  | 
 | 867 | int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 868 | PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags) | 
| Guido van Rossum | 393661d | 2001-08-31 17:40:15 +0000 | [diff] [blame] | 869 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 870 | 	PyObject *m, *d, *v; | 
 | 871 | 	m = PyImport_AddModule("__main__"); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 872 | 	if (m == NULL) | 
 | 873 | 		return -1; | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 874 | 	d = PyModule_GetDict(m); | 
| Guido van Rossum | 393661d | 2001-08-31 17:40:15 +0000 | [diff] [blame] | 875 | 	v = PyRun_StringFlags(command, Py_file_input, d, d, flags); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 876 | 	if (v == NULL) { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 877 | 		PyErr_Print(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 878 | 		return -1; | 
 | 879 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 880 | 	Py_DECREF(v); | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 881 | 	if (Py_FlushLine()) | 
 | 882 | 		PyErr_Clear(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 883 | 	return 0; | 
 | 884 | } | 
 | 885 |  | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 886 | static int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 887 | parse_syntax_error(PyObject *err, PyObject **message, const char **filename, | 
 | 888 | 		   int *lineno, int *offset, const char **text) | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 889 | { | 
 | 890 | 	long hold; | 
 | 891 | 	PyObject *v; | 
 | 892 |  | 
 | 893 | 	/* old style errors */ | 
 | 894 | 	if (PyTuple_Check(err)) | 
| Neal Norwitz | 7829335 | 2002-04-01 01:41:20 +0000 | [diff] [blame] | 895 | 		return PyArg_ParseTuple(err, "O(ziiz)", message, filename, | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 896 | 					lineno, offset, text); | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 897 |  | 
 | 898 | 	/* new style errors.  `err' is an instance */ | 
 | 899 |  | 
 | 900 | 	if (! (v = PyObject_GetAttrString(err, "msg"))) | 
 | 901 | 		goto finally; | 
 | 902 | 	*message = v; | 
 | 903 |  | 
 | 904 | 	if (!(v = PyObject_GetAttrString(err, "filename"))) | 
 | 905 | 		goto finally; | 
 | 906 | 	if (v == Py_None) | 
 | 907 | 		*filename = NULL; | 
 | 908 | 	else if (! (*filename = PyString_AsString(v))) | 
 | 909 | 		goto finally; | 
 | 910 |  | 
 | 911 | 	Py_DECREF(v); | 
 | 912 | 	if (!(v = PyObject_GetAttrString(err, "lineno"))) | 
 | 913 | 		goto finally; | 
 | 914 | 	hold = PyInt_AsLong(v); | 
 | 915 | 	Py_DECREF(v); | 
 | 916 | 	v = NULL; | 
 | 917 | 	if (hold < 0 && PyErr_Occurred()) | 
 | 918 | 		goto finally; | 
 | 919 | 	*lineno = (int)hold; | 
 | 920 |  | 
 | 921 | 	if (!(v = PyObject_GetAttrString(err, "offset"))) | 
 | 922 | 		goto finally; | 
| Jeremy Hylton | 9f1b993 | 2001-02-28 07:07:43 +0000 | [diff] [blame] | 923 | 	if (v == Py_None) { | 
 | 924 | 		*offset = -1; | 
 | 925 | 		Py_DECREF(v); | 
 | 926 | 		v = NULL; | 
 | 927 | 	} else { | 
 | 928 | 		hold = PyInt_AsLong(v); | 
 | 929 | 		Py_DECREF(v); | 
 | 930 | 		v = NULL; | 
 | 931 | 		if (hold < 0 && PyErr_Occurred()) | 
 | 932 | 			goto finally; | 
 | 933 | 		*offset = (int)hold; | 
 | 934 | 	} | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 935 |  | 
 | 936 | 	if (!(v = PyObject_GetAttrString(err, "text"))) | 
 | 937 | 		goto finally; | 
 | 938 | 	if (v == Py_None) | 
 | 939 | 		*text = NULL; | 
 | 940 | 	else if (! (*text = PyString_AsString(v))) | 
 | 941 | 		goto finally; | 
 | 942 | 	Py_DECREF(v); | 
 | 943 | 	return 1; | 
 | 944 |  | 
 | 945 | finally: | 
 | 946 | 	Py_XDECREF(v); | 
 | 947 | 	return 0; | 
 | 948 | } | 
 | 949 |  | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 950 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 951 | PyErr_Print(void) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 952 | { | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 953 | 	PyErr_PrintEx(1); | 
 | 954 | } | 
 | 955 |  | 
| Jeremy Hylton | 9f1b993 | 2001-02-28 07:07:43 +0000 | [diff] [blame] | 956 | static void | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 957 | print_error_text(PyObject *f, int offset, const char *text) | 
| Jeremy Hylton | 9f1b993 | 2001-02-28 07:07:43 +0000 | [diff] [blame] | 958 | { | 
 | 959 | 	char *nl; | 
 | 960 | 	if (offset >= 0) { | 
 | 961 | 		if (offset > 0 && offset == (int)strlen(text)) | 
 | 962 | 			offset--; | 
 | 963 | 		for (;;) { | 
 | 964 | 			nl = strchr(text, '\n'); | 
 | 965 | 			if (nl == NULL || nl-text >= offset) | 
 | 966 | 				break; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 967 | 			offset -= (int)(nl+1-text); | 
| Jeremy Hylton | 9f1b993 | 2001-02-28 07:07:43 +0000 | [diff] [blame] | 968 | 			text = nl+1; | 
 | 969 | 		} | 
 | 970 | 		while (*text == ' ' || *text == '\t') { | 
 | 971 | 			text++; | 
 | 972 | 			offset--; | 
 | 973 | 		} | 
 | 974 | 	} | 
 | 975 | 	PyFile_WriteString("    ", f); | 
 | 976 | 	PyFile_WriteString(text, f); | 
 | 977 | 	if (*text == '\0' || text[strlen(text)-1] != '\n') | 
 | 978 | 		PyFile_WriteString("\n", f); | 
 | 979 | 	if (offset == -1) | 
 | 980 | 		return; | 
 | 981 | 	PyFile_WriteString("    ", f); | 
 | 982 | 	offset--; | 
 | 983 | 	while (offset > 0) { | 
 | 984 | 		PyFile_WriteString(" ", f); | 
 | 985 | 		offset--; | 
 | 986 | 	} | 
 | 987 | 	PyFile_WriteString("^\n", f); | 
 | 988 | } | 
 | 989 |  | 
| Guido van Rossum | 66e8e86 | 2001-03-23 17:54:43 +0000 | [diff] [blame] | 990 | static void | 
 | 991 | handle_system_exit(void) | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 992 | { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 993 | 	PyObject *exception, *value, *tb; | 
 | 994 | 	int exitcode = 0; | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 995 |  | 
| Guido van Rossum | 66e8e86 | 2001-03-23 17:54:43 +0000 | [diff] [blame] | 996 | 	PyErr_Fetch(&exception, &value, &tb); | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 997 | 	if (Py_FlushLine()) | 
 | 998 | 		PyErr_Clear(); | 
 | 999 | 	fflush(stdout); | 
 | 1000 | 	if (value == NULL || value == Py_None) | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1001 | 		goto done; | 
| Brett Cannon | bf36409 | 2006-03-01 04:25:17 +0000 | [diff] [blame] | 1002 | 	if (PyExceptionInstance_Check(value)) { | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1003 | 		/* The error code should be in the `code' attribute. */ | 
 | 1004 | 		PyObject *code = PyObject_GetAttrString(value, "code"); | 
 | 1005 | 		if (code) { | 
 | 1006 | 			Py_DECREF(value); | 
 | 1007 | 			value = code; | 
 | 1008 | 			if (value == Py_None) | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1009 | 				goto done; | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1010 | 		} | 
 | 1011 | 		/* If we failed to dig out the 'code' attribute, | 
 | 1012 | 		   just let the else clause below print the error. */ | 
 | 1013 | 	} | 
 | 1014 | 	if (PyInt_Check(value)) | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1015 | 		exitcode = (int)PyInt_AsLong(value); | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1016 | 	else { | 
 | 1017 | 		PyObject_Print(value, stderr, Py_PRINT_RAW); | 
 | 1018 | 		PySys_WriteStderr("\n"); | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1019 | 		exitcode = 1; | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1020 | 	} | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1021 |  done: | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1022 | 	/* Restore and clear the exception info, in order to properly decref | 
 | 1023 | 	 * the exception, value, and traceback.	 If we just exit instead, | 
 | 1024 | 	 * these leak, which confuses PYTHONDUMPREFS output, and may prevent | 
 | 1025 | 	 * some finalizers from running. | 
 | 1026 | 	 */ | 
| Tim Peters | cf615b5 | 2003-04-19 18:47:02 +0000 | [diff] [blame] | 1027 | 	PyErr_Restore(exception, value, tb); | 
 | 1028 | 	PyErr_Clear(); | 
 | 1029 | 	Py_Exit(exitcode); | 
 | 1030 | 	/* NOTREACHED */ | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1031 | } | 
 | 1032 |  | 
 | 1033 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1034 | PyErr_PrintEx(int set_sys_last_vars) | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 1035 | { | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1036 | 	PyObject *exception, *v, *tb, *hook; | 
| Guido van Rossum | 66e8e86 | 2001-03-23 17:54:43 +0000 | [diff] [blame] | 1037 |  | 
 | 1038 | 	if (PyErr_ExceptionMatches(PyExc_SystemExit)) { | 
 | 1039 | 		handle_system_exit(); | 
 | 1040 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1041 | 	PyErr_Fetch(&exception, &v, &tb); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1042 | 	if (exception == NULL) | 
 | 1043 | 		return; | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 1044 | 	PyErr_NormalizeException(&exception, &v, &tb); | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 1045 | 	if (exception == NULL) | 
| Guido van Rossum | 296b475 | 1997-05-23 00:19:20 +0000 | [diff] [blame] | 1046 | 		return; | 
| Guido van Rossum | 9d78550 | 2006-03-07 18:31:44 +0000 | [diff] [blame] | 1047 |         /* Now we know v != NULL too */ | 
| Guido van Rossum | a61691e | 1998-02-06 22:27:24 +0000 | [diff] [blame] | 1048 | 	if (set_sys_last_vars) { | 
 | 1049 | 		PySys_SetObject("last_type", exception); | 
 | 1050 | 		PySys_SetObject("last_value", v); | 
 | 1051 | 		PySys_SetObject("last_traceback", tb); | 
 | 1052 | 	} | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1053 | 	hook = PySys_GetObject("excepthook"); | 
 | 1054 | 	if (hook) { | 
| Raymond Hettinger | 8ae4689 | 2003-10-12 19:09:37 +0000 | [diff] [blame] | 1055 | 		PyObject *args = PyTuple_Pack(3, | 
| Guido van Rossum | 9d78550 | 2006-03-07 18:31:44 +0000 | [diff] [blame] | 1056 | 		    exception, v, tb ? tb : Py_None); | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1057 | 		PyObject *result = PyEval_CallObject(hook, args); | 
 | 1058 | 		if (result == NULL) { | 
 | 1059 | 			PyObject *exception2, *v2, *tb2; | 
| Guido van Rossum | 66e8e86 | 2001-03-23 17:54:43 +0000 | [diff] [blame] | 1060 | 			if (PyErr_ExceptionMatches(PyExc_SystemExit)) { | 
 | 1061 | 				handle_system_exit(); | 
 | 1062 | 			} | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1063 | 			PyErr_Fetch(&exception2, &v2, &tb2); | 
 | 1064 | 			PyErr_NormalizeException(&exception2, &v2, &tb2); | 
 | 1065 | 			if (Py_FlushLine()) | 
 | 1066 | 				PyErr_Clear(); | 
 | 1067 | 			fflush(stdout); | 
 | 1068 | 			PySys_WriteStderr("Error in sys.excepthook:\n"); | 
 | 1069 | 			PyErr_Display(exception2, v2, tb2); | 
 | 1070 | 			PySys_WriteStderr("\nOriginal exception was:\n"); | 
 | 1071 | 			PyErr_Display(exception, v, tb); | 
| Jeremy Hylton | 0702858 | 2001-12-07 15:35:35 +0000 | [diff] [blame] | 1072 | 			Py_XDECREF(exception2); | 
 | 1073 | 			Py_XDECREF(v2); | 
 | 1074 | 			Py_XDECREF(tb2); | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1075 | 		} | 
 | 1076 | 		Py_XDECREF(result); | 
 | 1077 | 		Py_XDECREF(args); | 
 | 1078 | 	} else { | 
 | 1079 | 		PySys_WriteStderr("sys.excepthook is missing\n"); | 
 | 1080 | 		PyErr_Display(exception, v, tb); | 
 | 1081 | 	} | 
 | 1082 | 	Py_XDECREF(exception); | 
 | 1083 | 	Py_XDECREF(v); | 
 | 1084 | 	Py_XDECREF(tb); | 
 | 1085 | } | 
 | 1086 |  | 
| Richard Jones | 7b9558d | 2006-05-27 12:29:24 +0000 | [diff] [blame] | 1087 | void | 
 | 1088 | PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb) | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1089 | { | 
 | 1090 | 	int err = 0; | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1091 | 	PyObject *f = PySys_GetObject("stderr"); | 
| Armin Rigo | 5d2c683 | 2004-03-22 20:16:58 +0000 | [diff] [blame] | 1092 | 	Py_INCREF(value); | 
| Guido van Rossum | 3165fe6 | 1992-09-25 21:59:05 +0000 | [diff] [blame] | 1093 | 	if (f == NULL) | 
 | 1094 | 		fprintf(stderr, "lost sys.stderr\n"); | 
 | 1095 | 	else { | 
| Guido van Rossum | 0829c75 | 1998-02-28 04:31:39 +0000 | [diff] [blame] | 1096 | 		if (Py_FlushLine()) | 
 | 1097 | 			PyErr_Clear(); | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1098 | 		fflush(stdout); | 
| Ka-Ping Yee | b5c5132 | 2001-03-23 02:46:52 +0000 | [diff] [blame] | 1099 | 		if (tb && tb != Py_None) | 
 | 1100 | 			err = PyTraceBack_Print(tb, f); | 
| Barry Warsaw | 36b8f94 | 1997-08-26 18:09:48 +0000 | [diff] [blame] | 1101 | 		if (err == 0 && | 
| Armin Rigo | 5d2c683 | 2004-03-22 20:16:58 +0000 | [diff] [blame] | 1102 | 		    PyObject_HasAttrString(value, "print_file_and_line")) | 
| Barry Warsaw | 36b8f94 | 1997-08-26 18:09:48 +0000 | [diff] [blame] | 1103 | 		{ | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1104 | 			PyObject *message; | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1105 | 			const char *filename, *text; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1106 | 			int lineno, offset; | 
| Armin Rigo | 5d2c683 | 2004-03-22 20:16:58 +0000 | [diff] [blame] | 1107 | 			if (!parse_syntax_error(value, &message, &filename, | 
| Barry Warsaw | 035574d | 1997-08-29 22:07:17 +0000 | [diff] [blame] | 1108 | 						&lineno, &offset, &text)) | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1109 | 				PyErr_Clear(); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1110 | 			else { | 
 | 1111 | 				char buf[10]; | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1112 | 				PyFile_WriteString("  File \"", f); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1113 | 				if (filename == NULL) | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1114 | 					PyFile_WriteString("<string>", f); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1115 | 				else | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1116 | 					PyFile_WriteString(filename, f); | 
 | 1117 | 				PyFile_WriteString("\", line ", f); | 
| Jeremy Hylton | 518ab1c | 2001-11-28 20:42:20 +0000 | [diff] [blame] | 1118 | 				PyOS_snprintf(buf, sizeof(buf), "%d", lineno); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1119 | 				PyFile_WriteString(buf, f); | 
 | 1120 | 				PyFile_WriteString("\n", f); | 
| Jeremy Hylton | 9f1b993 | 2001-02-28 07:07:43 +0000 | [diff] [blame] | 1121 | 				if (text != NULL) | 
 | 1122 | 					print_error_text(f, offset, text); | 
| Armin Rigo | 5d2c683 | 2004-03-22 20:16:58 +0000 | [diff] [blame] | 1123 | 				Py_DECREF(value); | 
 | 1124 | 				value = message; | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1125 | 				/* Can't be bothered to check all those | 
 | 1126 | 				   PyFile_WriteString() calls */ | 
 | 1127 | 				if (PyErr_Occurred()) | 
 | 1128 | 					err = -1; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1129 | 			} | 
 | 1130 | 		} | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1131 | 		if (err) { | 
 | 1132 | 			/* Don't do anything else */ | 
 | 1133 | 		} | 
| Brett Cannon | bf36409 | 2006-03-01 04:25:17 +0000 | [diff] [blame] | 1134 | 		else if (PyExceptionClass_Check(exception)) { | 
 | 1135 | 			char* className = PyExceptionClass_Name(exception); | 
| Richard Jones | 7b9558d | 2006-05-27 12:29:24 +0000 | [diff] [blame] | 1136 | 			char *dot = strrchr(className, '.'); | 
 | 1137 | 			PyObject* moduleName; | 
 | 1138 | 			if (dot != NULL) | 
 | 1139 | 				className = dot+1; | 
 | 1140 | 			moduleName = PyObject_GetAttrString(exception, "__module__"); | 
| Barry Warsaw | 2f5f6a2 | 1997-09-16 21:42:03 +0000 | [diff] [blame] | 1141 |  | 
 | 1142 | 			if (moduleName == NULL) | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1143 | 				err = PyFile_WriteString("<unknown>", f); | 
| Barry Warsaw | 2f5f6a2 | 1997-09-16 21:42:03 +0000 | [diff] [blame] | 1144 | 			else { | 
 | 1145 | 				char* modstr = PyString_AsString(moduleName); | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 1146 | 				if (modstr && strcmp(modstr, "exceptions")) | 
| Barry Warsaw | 2f5f6a2 | 1997-09-16 21:42:03 +0000 | [diff] [blame] | 1147 | 				{ | 
 | 1148 | 					err = PyFile_WriteString(modstr, f); | 
 | 1149 | 					err += PyFile_WriteString(".", f); | 
 | 1150 | 				} | 
| Richard Jones | 7b9558d | 2006-05-27 12:29:24 +0000 | [diff] [blame] | 1151 | 				Py_DECREF(moduleName); | 
| Barry Warsaw | 2f5f6a2 | 1997-09-16 21:42:03 +0000 | [diff] [blame] | 1152 | 			} | 
 | 1153 | 			if (err == 0) { | 
 | 1154 | 				if (className == NULL) | 
 | 1155 | 				      err = PyFile_WriteString("<unknown>", f); | 
 | 1156 | 				else | 
| Brett Cannon | bf36409 | 2006-03-01 04:25:17 +0000 | [diff] [blame] | 1157 | 				      err = PyFile_WriteString(className, f); | 
| Barry Warsaw | 2f5f6a2 | 1997-09-16 21:42:03 +0000 | [diff] [blame] | 1158 | 			} | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1159 | 		} | 
 | 1160 | 		else | 
 | 1161 | 			err = PyFile_WriteObject(exception, f, Py_PRINT_RAW); | 
| Brett Cannon | 2e63b73 | 2006-03-02 18:34:57 +0000 | [diff] [blame] | 1162 | 		if (err == 0 && (value != Py_None)) { | 
 | 1163 | 			PyObject *s = PyObject_Str(value); | 
 | 1164 | 			/* only print colon if the str() of the | 
 | 1165 | 			   object is not the empty string | 
 | 1166 | 			*/ | 
 | 1167 | 			if (s == NULL) | 
 | 1168 | 				err = -1; | 
 | 1169 | 			else if (!PyString_Check(s) || | 
 | 1170 | 				 PyString_GET_SIZE(s) != 0) | 
 | 1171 | 				err = PyFile_WriteString(": ", f); | 
 | 1172 | 			if (err == 0) | 
 | 1173 | 			  err = PyFile_WriteObject(s, f, Py_PRINT_RAW); | 
 | 1174 | 			Py_XDECREF(s); | 
| Guido van Rossum | 262e124 | 1995-02-07 15:30:45 +0000 | [diff] [blame] | 1175 | 		} | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1176 | 		if (err == 0) | 
 | 1177 | 			err = PyFile_WriteString("\n", f); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1178 | 	} | 
| Armin Rigo | 5d2c683 | 2004-03-22 20:16:58 +0000 | [diff] [blame] | 1179 | 	Py_DECREF(value); | 
| Guido van Rossum | 78a1ed3 | 1997-05-22 22:35:04 +0000 | [diff] [blame] | 1180 | 	/* If an error happened here, don't show it. | 
 | 1181 | 	   XXX This is wrong, but too many callers rely on this behavior. */ | 
 | 1182 | 	if (err != 0) | 
 | 1183 | 		PyErr_Clear(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1184 | } | 
 | 1185 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1186 | PyObject * | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1187 | PyRun_StringFlags(const char *str, int start, PyObject *globals,  | 
 | 1188 | 		  PyObject *locals, PyCompilerFlags *flags) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1189 | { | 
| Neal Norwitz | e92fba0 | 2006-03-04 18:52:26 +0000 | [diff] [blame] | 1190 | 	PyObject *ret = NULL; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1191 | 	PyArena *arena = PyArena_New(); | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1192 | 	mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags, | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1193 | 					    arena); | 
| Neal Norwitz | e92fba0 | 2006-03-04 18:52:26 +0000 | [diff] [blame] | 1194 | 	if (mod != NULL) | 
 | 1195 | 		ret = run_mod(mod, "<string>", globals, locals, flags, arena); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1196 | 	PyArena_Free(arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1197 | 	return ret; | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 1198 | } | 
 | 1199 |  | 
 | 1200 | PyObject * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1201 | PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals, | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 1202 | 		  PyObject *locals, int closeit, PyCompilerFlags *flags) | 
 | 1203 | { | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1204 | 	PyObject *ret; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1205 | 	PyArena *arena = PyArena_New(); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1206 | 	mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0, | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1207 | 					  flags, NULL, arena); | 
 | 1208 | 	if (mod == NULL) { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1209 | 		PyArena_Free(arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1210 | 		return NULL; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1211 | 	} | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 1212 | 	if (closeit) | 
 | 1213 | 		fclose(fp); | 
| Neal Norwitz | e92fba0 | 2006-03-04 18:52:26 +0000 | [diff] [blame] | 1214 | 	ret = run_mod(mod, filename, globals, locals, flags, arena); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1215 | 	PyArena_Free(arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1216 | 	return ret; | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 1217 | } | 
 | 1218 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1219 | static PyObject * | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1220 | run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals, | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1221 | 	 PyCompilerFlags *flags, PyArena *arena) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1222 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1223 | 	PyCodeObject *co; | 
 | 1224 | 	PyObject *v; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1225 | 	co = PyAST_Compile(mod, filename, flags, arena); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1226 | 	if (co == NULL) | 
 | 1227 | 		return NULL; | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1228 | 	v = PyEval_EvalCode(co, globals, locals); | 
 | 1229 | 	Py_DECREF(co); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1230 | 	return v; | 
 | 1231 | } | 
 | 1232 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1233 | static PyObject * | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1234 | run_pyc_file(FILE *fp, const char *filename, PyObject *globals,  | 
 | 1235 | 	     PyObject *locals, PyCompilerFlags *flags) | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1236 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1237 | 	PyCodeObject *co; | 
 | 1238 | 	PyObject *v; | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1239 | 	long magic; | 
| Fred Drake | e8de31c | 2000-08-31 05:38:39 +0000 | [diff] [blame] | 1240 | 	long PyImport_GetMagicNumber(void); | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1241 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1242 | 	magic = PyMarshal_ReadLongFromFile(fp); | 
 | 1243 | 	if (magic != PyImport_GetMagicNumber()) { | 
 | 1244 | 		PyErr_SetString(PyExc_RuntimeError, | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1245 | 			   "Bad magic number in .pyc file"); | 
 | 1246 | 		return NULL; | 
 | 1247 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1248 | 	(void) PyMarshal_ReadLongFromFile(fp); | 
| Tim Peters | d9b9ac8 | 2001-01-28 00:27:39 +0000 | [diff] [blame] | 1249 | 	v = PyMarshal_ReadLastObjectFromFile(fp); | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1250 | 	fclose(fp); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1251 | 	if (v == NULL || !PyCode_Check(v)) { | 
 | 1252 | 		Py_XDECREF(v); | 
 | 1253 | 		PyErr_SetString(PyExc_RuntimeError, | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1254 | 			   "Bad code object in .pyc file"); | 
 | 1255 | 		return NULL; | 
 | 1256 | 	} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1257 | 	co = (PyCodeObject *)v; | 
 | 1258 | 	v = PyEval_EvalCode(co, globals, locals); | 
| Jeremy Hylton | b857ba2 | 2001-08-10 21:41:33 +0000 | [diff] [blame] | 1259 | 	if (v && flags) | 
 | 1260 | 		flags->cf_flags |= (co->co_flags & PyCF_MASK); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1261 | 	Py_DECREF(co); | 
| Guido van Rossum | fdef271 | 1994-09-14 13:31:04 +0000 | [diff] [blame] | 1262 | 	return v; | 
 | 1263 | } | 
 | 1264 |  | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1265 | PyObject * | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 1266 | Py_CompileStringFlags(const char *str, const char *filename, int start, | 
| Jeremy Hylton | bc32024 | 2001-03-22 02:47:58 +0000 | [diff] [blame] | 1267 | 		      PyCompilerFlags *flags) | 
 | 1268 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1269 | 	PyCodeObject *co; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1270 | 	PyArena *arena = PyArena_New(); | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1271 | 	mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena); | 
 | 1272 | 	if (mod == NULL) { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1273 | 		PyArena_Free(arena); | 
| Guido van Rossum | 5b72218 | 1993-03-30 17:46:03 +0000 | [diff] [blame] | 1274 | 		return NULL; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1275 | 	} | 
| Martin v. Löwis | 2b366e4 | 2006-02-26 22:12:35 +0000 | [diff] [blame] | 1276 | 	if (flags && (flags->cf_flags & PyCF_ONLY_AST)) { | 
| Martin v. Löwis | bd260da | 2006-02-26 19:42:26 +0000 | [diff] [blame] | 1277 | 		PyObject *result = PyAST_mod2obj(mod); | 
 | 1278 | 		PyArena_Free(arena); | 
 | 1279 | 		return result; | 
 | 1280 | 	} | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1281 | 	co = PyAST_Compile(mod, filename, flags, arena); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1282 | 	PyArena_Free(arena); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1283 | 	return (PyObject *)co; | 
| Guido van Rossum | 5b72218 | 1993-03-30 17:46:03 +0000 | [diff] [blame] | 1284 | } | 
 | 1285 |  | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 1286 | struct symtable * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1287 | Py_SymtableString(const char *str, const char *filename, int start) | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 1288 | { | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 1289 | 	struct symtable *st; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1290 | 	PyArena *arena = PyArena_New(); | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1291 | 	mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena); | 
 | 1292 | 	if (mod == NULL) { | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1293 | 		PyArena_Free(arena); | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 1294 | 		return NULL; | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1295 | 	} | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1296 | 	st = PySymtable_Build(mod, filename, 0); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1297 | 	PyArena_Free(arena); | 
| Jeremy Hylton | 4b38da6 | 2001-02-02 18:19:15 +0000 | [diff] [blame] | 1298 | 	return st; | 
 | 1299 | } | 
 | 1300 |  | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1301 | /* Preferred access to parser is through AST. */ | 
 | 1302 | mod_ty | 
 | 1303 | PyParser_ASTFromString(const char *s, const char *filename, int start,  | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1304 | 		       PyCompilerFlags *flags, PyArena *arena) | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1305 | { | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1306 | 	mod_ty mod; | 
 | 1307 | 	perrdetail err; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1308 | 	node *n = PyParser_ParseStringFlagsFilename(s, filename, | 
 | 1309 | 					&_PyParser_Grammar, start, &err,  | 
 | 1310 | 					PARSER_FLAGS(flags)); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1311 | 	if (n) { | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1312 | 		mod = PyAST_FromNode(n, flags, filename, arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1313 | 		PyNode_Free(n); | 
 | 1314 | 		return mod; | 
 | 1315 | 	} | 
 | 1316 | 	else { | 
 | 1317 | 		err_input(&err); | 
 | 1318 | 		return NULL; | 
 | 1319 | 	} | 
 | 1320 | } | 
 | 1321 |  | 
 | 1322 | mod_ty | 
 | 1323 | PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,  | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1324 | 		     char *ps2, PyCompilerFlags *flags, int *errcode, | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1325 | 		     PyArena *arena) | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1326 | { | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1327 | 	mod_ty mod; | 
 | 1328 | 	perrdetail err; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1329 | 	node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar, | 
 | 1330 | 				start, ps1, ps2, &err, PARSER_FLAGS(flags)); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1331 | 	if (n) { | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1332 | 		mod = PyAST_FromNode(n, flags, filename, arena); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1333 | 		PyNode_Free(n); | 
 | 1334 | 		return mod; | 
 | 1335 | 	} | 
 | 1336 | 	else { | 
 | 1337 | 		err_input(&err); | 
 | 1338 | 		if (errcode) | 
 | 1339 | 			*errcode = err.error; | 
 | 1340 | 		return NULL; | 
 | 1341 | 	} | 
 | 1342 | } | 
 | 1343 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1344 | /* Simplified interface to parsefile -- return node or set exception */ | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1345 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1346 | node * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1347 | PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1348 | { | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1349 | 	perrdetail err; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1350 | 	node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar, | 
 | 1351 | 					  start, NULL, NULL, &err, flags); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1352 | 	if (n == NULL) | 
 | 1353 | 		err_input(&err); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1354 | 		 | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1355 | 	return n; | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1356 | } | 
 | 1357 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1358 | /* Simplified interface to parsestring -- return node or set exception */ | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1359 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1360 | node * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1361 | PyParser_SimpleParseStringFlags(const char *str, int start, int flags) | 
| Tim Peters | fe2127d | 2001-07-16 05:37:24 +0000 | [diff] [blame] | 1362 | { | 
| Tim Peters | fe2127d | 2001-07-16 05:37:24 +0000 | [diff] [blame] | 1363 | 	perrdetail err; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1364 | 	node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar, | 
 | 1365 | 					    start, &err, flags); | 
| Tim Peters | fe2127d | 2001-07-16 05:37:24 +0000 | [diff] [blame] | 1366 | 	if (n == NULL) | 
 | 1367 | 		err_input(&err); | 
 | 1368 | 	return n; | 
 | 1369 | } | 
 | 1370 |  | 
 | 1371 | node * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1372 | PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename, | 
| Thomas Heller | 6b17abf | 2002-07-09 09:23:27 +0000 | [diff] [blame] | 1373 | 					int start, int flags) | 
 | 1374 | { | 
| Thomas Heller | 6b17abf | 2002-07-09 09:23:27 +0000 | [diff] [blame] | 1375 | 	perrdetail err; | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1376 | 	node *n = PyParser_ParseStringFlagsFilename(str, filename, | 
 | 1377 | 				&_PyParser_Grammar, start, &err, flags); | 
| Thomas Heller | 6b17abf | 2002-07-09 09:23:27 +0000 | [diff] [blame] | 1378 | 	if (n == NULL) | 
 | 1379 | 		err_input(&err); | 
 | 1380 | 	return n; | 
 | 1381 | } | 
 | 1382 |  | 
 | 1383 | node * | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1384 | PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start) | 
| Thomas Heller | 6b17abf | 2002-07-09 09:23:27 +0000 | [diff] [blame] | 1385 | { | 
| Neal Norwitz | adb69fc | 2005-12-17 20:54:49 +0000 | [diff] [blame] | 1386 | 	return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0); | 
| Thomas Heller | 6b17abf | 2002-07-09 09:23:27 +0000 | [diff] [blame] | 1387 | } | 
 | 1388 |  | 
| Guido van Rossum | 66ebd91 | 2003-04-17 16:02:26 +0000 | [diff] [blame] | 1389 | /* May want to move a more generalized form of this to parsetok.c or | 
 | 1390 |    even parser modules. */ | 
 | 1391 |  | 
 | 1392 | void | 
 | 1393 | PyParser_SetError(perrdetail *err) | 
 | 1394 | { | 
 | 1395 | 	err_input(err); | 
 | 1396 | } | 
 | 1397 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1398 | /* Set the error appropriate to the given input error code (see errcode.h) */ | 
 | 1399 |  | 
 | 1400 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1401 | err_input(perrdetail *err) | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1402 | { | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1403 | 	PyObject *v, *w, *errtype; | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1404 | 	PyObject* u = NULL; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1405 | 	char *msg = NULL; | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1406 | 	errtype = PyExc_SyntaxError; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1407 | 	switch (err->error) { | 
 | 1408 | 	case E_SYNTAX: | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1409 | 		errtype = PyExc_IndentationError; | 
 | 1410 | 		if (err->expected == INDENT) | 
 | 1411 | 			msg = "expected an indented block"; | 
 | 1412 | 		else if (err->token == INDENT) | 
 | 1413 | 			msg = "unexpected indent"; | 
 | 1414 | 		else if (err->token == DEDENT) | 
 | 1415 | 			msg = "unexpected unindent"; | 
 | 1416 | 		else { | 
 | 1417 | 			errtype = PyExc_SyntaxError; | 
 | 1418 | 			msg = "invalid syntax"; | 
 | 1419 | 		} | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1420 | 		break; | 
 | 1421 | 	case E_TOKEN: | 
 | 1422 | 		msg = "invalid token"; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1423 | 		break; | 
| Skip Montanaro | 118ec70 | 2002-08-15 01:20:16 +0000 | [diff] [blame] | 1424 | 	case E_EOFS: | 
 | 1425 | 		msg = "EOF while scanning triple-quoted string"; | 
 | 1426 | 		break; | 
 | 1427 | 	case E_EOLS: | 
 | 1428 | 		msg = "EOL while scanning single-quoted string"; | 
 | 1429 | 		break; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1430 | 	case E_INTR: | 
| Michael W. Hudson | 30ea2f2 | 2004-07-07 17:44:12 +0000 | [diff] [blame] | 1431 | 		if (!PyErr_Occurred()) | 
 | 1432 | 			PyErr_SetNone(PyExc_KeyboardInterrupt); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1433 | 		return; | 
 | 1434 | 	case E_NOMEM: | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1435 | 		PyErr_NoMemory(); | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1436 | 		return; | 
 | 1437 | 	case E_EOF: | 
 | 1438 | 		msg = "unexpected EOF while parsing"; | 
 | 1439 | 		break; | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1440 | 	case E_TABSPACE: | 
 | 1441 | 		errtype = PyExc_TabError; | 
| Guido van Rossum | 560e8ad | 1998-04-10 19:43:42 +0000 | [diff] [blame] | 1442 | 		msg = "inconsistent use of tabs and spaces in indentation"; | 
 | 1443 | 		break; | 
| Jeremy Hylton | 9498806 | 2000-06-20 19:10:44 +0000 | [diff] [blame] | 1444 | 	case E_OVERFLOW: | 
 | 1445 | 		msg = "expression too long"; | 
 | 1446 | 		break; | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1447 | 	case E_DEDENT: | 
 | 1448 | 		errtype = PyExc_IndentationError; | 
 | 1449 | 		msg = "unindent does not match any outer indentation level"; | 
 | 1450 | 		break; | 
 | 1451 | 	case E_TOODEEP: | 
 | 1452 | 		errtype = PyExc_IndentationError; | 
 | 1453 | 		msg = "too many levels of indentation"; | 
 | 1454 | 		break; | 
| Martin v. Löwis | d35edda | 2005-08-24 08:39:24 +0000 | [diff] [blame] | 1455 | 	case E_DECODE: { | 
 | 1456 | 		PyObject *type, *value, *tb; | 
 | 1457 | 		PyErr_Fetch(&type, &value, &tb); | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1458 | 		if (value != NULL) { | 
| Martin v. Löwis | d35edda | 2005-08-24 08:39:24 +0000 | [diff] [blame] | 1459 | 			u = PyObject_Str(value); | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1460 | 			if (u != NULL) { | 
 | 1461 | 				msg = PyString_AsString(u); | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1462 | 			} | 
 | 1463 | 		} | 
| Martin v. Löwis | c2632a5 | 2004-07-21 05:35:02 +0000 | [diff] [blame] | 1464 | 		if (msg == NULL) | 
 | 1465 | 			msg = "unknown decode error"; | 
| Neal Norwitz | db83eb3 | 2005-12-18 05:29:30 +0000 | [diff] [blame] | 1466 | 		Py_XDECREF(type); | 
 | 1467 | 		Py_XDECREF(value); | 
| Neal Norwitz | 40d3781 | 2005-10-02 01:48:49 +0000 | [diff] [blame] | 1468 | 		Py_XDECREF(tb); | 
| Martin v. Löwis | c2632a5 | 2004-07-21 05:35:02 +0000 | [diff] [blame] | 1469 | 		break; | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1470 | 	} | 
| Martin v. Löwis | 4bf108d | 2005-03-03 11:45:45 +0000 | [diff] [blame] | 1471 | 	case E_LINECONT: | 
 | 1472 | 		msg = "unexpected character after line continuation character"; | 
 | 1473 | 		break; | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1474 | 	default: | 
 | 1475 | 		fprintf(stderr, "error=%d\n", err->error); | 
 | 1476 | 		msg = "unknown parsing error"; | 
 | 1477 | 		break; | 
 | 1478 | 	} | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1479 | 	v = Py_BuildValue("(ziiz)", err->filename, | 
 | 1480 | 			  err->lineno, err->offset, err->text); | 
| Neal Norwitz | 9589ee2 | 2006-03-04 19:01:22 +0000 | [diff] [blame] | 1481 | 	if (err->text != NULL) { | 
| Tim Peters | c9d78aa | 2006-03-26 23:27:58 +0000 | [diff] [blame] | 1482 | 		PyObject_FREE(err->text); | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1483 | 		err->text = NULL; | 
 | 1484 | 	} | 
 | 1485 | 	w = NULL; | 
 | 1486 | 	if (v != NULL) | 
 | 1487 | 		w = Py_BuildValue("(sO)", msg, v); | 
| Martin v. Löwis | 00f1e3f | 2002-08-04 17:29:52 +0000 | [diff] [blame] | 1488 | 	Py_XDECREF(u); | 
| Guido van Rossum | 4131830 | 2001-03-23 04:01:07 +0000 | [diff] [blame] | 1489 | 	Py_XDECREF(v); | 
| Fred Drake | 85f3639 | 2000-07-11 17:53:00 +0000 | [diff] [blame] | 1490 | 	PyErr_SetObject(errtype, w); | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1491 | 	Py_XDECREF(w); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1492 | } | 
 | 1493 |  | 
 | 1494 | /* Print fatal error message and abort */ | 
 | 1495 |  | 
 | 1496 | void | 
| Tim Peters | 7c321a8 | 2002-07-09 02:57:01 +0000 | [diff] [blame] | 1497 | Py_FatalError(const char *msg) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1498 | { | 
| Guido van Rossum | 83dd6c3 | 1994-09-29 09:38:33 +0000 | [diff] [blame] | 1499 | 	fprintf(stderr, "Fatal Python error: %s\n", msg); | 
| Martin v. Löwis | 6238d2b | 2002-06-30 15:26:10 +0000 | [diff] [blame] | 1500 | #ifdef MS_WINDOWS | 
| Guido van Rossum | 23c9446 | 1997-05-22 20:21:30 +0000 | [diff] [blame] | 1501 | 	OutputDebugString("Fatal Python error: "); | 
| Guido van Rossum | a44823b | 1995-03-14 15:01:17 +0000 | [diff] [blame] | 1502 | 	OutputDebugString(msg); | 
 | 1503 | 	OutputDebugString("\n"); | 
| Guido van Rossum | 0ba3536 | 1998-08-13 13:33:16 +0000 | [diff] [blame] | 1504 | #ifdef _DEBUG | 
 | 1505 | 	DebugBreak(); | 
| Guido van Rossum | a44823b | 1995-03-14 15:01:17 +0000 | [diff] [blame] | 1506 | #endif | 
| Martin v. Löwis | 6238d2b | 2002-06-30 15:26:10 +0000 | [diff] [blame] | 1507 | #endif /* MS_WINDOWS */ | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1508 | 	abort(); | 
 | 1509 | } | 
 | 1510 |  | 
 | 1511 | /* Clean up and exit */ | 
 | 1512 |  | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1513 | #ifdef WITH_THREAD | 
| Guido van Rossum | 49b5606 | 1998-10-01 20:42:43 +0000 | [diff] [blame] | 1514 | #include "pythread.h" | 
| Guido van Rossum | f9f2e82 | 1992-08-17 08:59:08 +0000 | [diff] [blame] | 1515 | #endif | 
 | 1516 |  | 
| Guido van Rossum | 2dcfc96 | 1998-10-01 16:01:57 +0000 | [diff] [blame] | 1517 | #define NEXITFUNCS 32 | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1518 | static void (*exitfuncs[NEXITFUNCS])(void); | 
| Guido van Rossum | 1662dd5 | 1994-09-07 14:38:28 +0000 | [diff] [blame] | 1519 | static int nexitfuncs = 0; | 
 | 1520 |  | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1521 | int Py_AtExit(void (*func)(void)) | 
| Guido van Rossum | 1662dd5 | 1994-09-07 14:38:28 +0000 | [diff] [blame] | 1522 | { | 
 | 1523 | 	if (nexitfuncs >= NEXITFUNCS) | 
 | 1524 | 		return -1; | 
 | 1525 | 	exitfuncs[nexitfuncs++] = func; | 
 | 1526 | 	return 0; | 
 | 1527 | } | 
 | 1528 |  | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 1529 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1530 | call_sys_exitfunc(void) | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1531 | { | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1532 | 	PyObject *exitfunc = PySys_GetObject("exitfunc"); | 
| Guido van Rossum | 59bff39 | 1992-09-03 20:28:00 +0000 | [diff] [blame] | 1533 |  | 
 | 1534 | 	if (exitfunc) { | 
| Fred Drake | 6a12d8d | 2001-03-23 17:34:02 +0000 | [diff] [blame] | 1535 | 		PyObject *res; | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1536 | 		Py_INCREF(exitfunc); | 
 | 1537 | 		PySys_SetObject("exitfunc", (PyObject *)NULL); | 
 | 1538 | 		res = PyEval_CallObject(exitfunc, (PyObject *)NULL); | 
| Guido van Rossum | 59bff39 | 1992-09-03 20:28:00 +0000 | [diff] [blame] | 1539 | 		if (res == NULL) { | 
| Ka-Ping Yee | 26fabb0 | 2001-03-23 15:36:41 +0000 | [diff] [blame] | 1540 | 			if (!PyErr_ExceptionMatches(PyExc_SystemExit)) { | 
 | 1541 | 				PySys_WriteStderr("Error in sys.exitfunc:\n"); | 
 | 1542 | 			} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1543 | 			PyErr_Print(); | 
| Guido van Rossum | 59bff39 | 1992-09-03 20:28:00 +0000 | [diff] [blame] | 1544 | 		} | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1545 | 		Py_DECREF(exitfunc); | 
| Guido van Rossum | 59bff39 | 1992-09-03 20:28:00 +0000 | [diff] [blame] | 1546 | 	} | 
 | 1547 |  | 
| Guido van Rossum | 0829c75 | 1998-02-28 04:31:39 +0000 | [diff] [blame] | 1548 | 	if (Py_FlushLine()) | 
 | 1549 | 		PyErr_Clear(); | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 1550 | } | 
| Guido van Rossum | 1662dd5 | 1994-09-07 14:38:28 +0000 | [diff] [blame] | 1551 |  | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 1552 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1553 | call_ll_exitfuncs(void) | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 1554 | { | 
| Guido van Rossum | 1662dd5 | 1994-09-07 14:38:28 +0000 | [diff] [blame] | 1555 | 	while (nexitfuncs > 0) | 
 | 1556 | 		(*exitfuncs[--nexitfuncs])(); | 
| Guido van Rossum | 25ce566 | 1997-08-02 03:10:38 +0000 | [diff] [blame] | 1557 |  | 
 | 1558 | 	fflush(stdout); | 
 | 1559 | 	fflush(stderr); | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 1560 | } | 
 | 1561 |  | 
 | 1562 | void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1563 | Py_Exit(int sts) | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 1564 | { | 
| Guido van Rossum | cc283f5 | 1997-08-05 02:22:03 +0000 | [diff] [blame] | 1565 | 	Py_Finalize(); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1566 |  | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1567 | 	exit(sts); | 
| Guido van Rossum | 1984f1e | 1992-08-04 12:41:02 +0000 | [diff] [blame] | 1568 | } | 
 | 1569 |  | 
| Guido van Rossum | f1dc566 | 1993-07-05 10:31:29 +0000 | [diff] [blame] | 1570 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1571 | initsigs(void) | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 1572 | { | 
| Guido van Rossum | a110aa6 | 1994-08-29 12:50:44 +0000 | [diff] [blame] | 1573 | #ifdef SIGPIPE | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1574 | 	PyOS_setsig(SIGPIPE, SIG_IGN); | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 1575 | #endif | 
| Guido van Rossum | 70d893a | 2001-08-16 08:21:42 +0000 | [diff] [blame] | 1576 | #ifdef SIGXFZ | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1577 | 	PyOS_setsig(SIGXFZ, SIG_IGN); | 
| Guido van Rossum | 70d893a | 2001-08-16 08:21:42 +0000 | [diff] [blame] | 1578 | #endif | 
| Jeremy Hylton | 1b0bf9b | 2002-04-23 20:31:01 +0000 | [diff] [blame] | 1579 | #ifdef SIGXFSZ | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1580 | 	PyOS_setsig(SIGXFSZ, SIG_IGN); | 
| Jeremy Hylton | 1b0bf9b | 2002-04-23 20:31:01 +0000 | [diff] [blame] | 1581 | #endif | 
| Guido van Rossum | 8259805 | 1997-03-05 00:20:32 +0000 | [diff] [blame] | 1582 | 	PyOS_InitInterrupts(); /* May imply initsignal() */ | 
| Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 1583 | } | 
 | 1584 |  | 
| Guido van Rossum | 7433b12 | 1997-02-14 19:45:36 +0000 | [diff] [blame] | 1585 |  | 
 | 1586 | /* | 
 | 1587 |  * The file descriptor fd is considered ``interactive'' if either | 
 | 1588 |  *   a) isatty(fd) is TRUE, or | 
 | 1589 |  *   b) the -i flag was given, and the filename associated with | 
 | 1590 |  *      the descriptor is NULL or "<stdin>" or "???". | 
 | 1591 |  */ | 
 | 1592 | int | 
| Martin v. Löwis | 95292d6 | 2002-12-11 14:04:59 +0000 | [diff] [blame] | 1593 | Py_FdIsInteractive(FILE *fp, const char *filename) | 
| Guido van Rossum | 7433b12 | 1997-02-14 19:45:36 +0000 | [diff] [blame] | 1594 | { | 
 | 1595 | 	if (isatty((int)fileno(fp))) | 
 | 1596 | 		return 1; | 
 | 1597 | 	if (!Py_InteractiveFlag) | 
 | 1598 | 		return 0; | 
 | 1599 | 	return (filename == NULL) || | 
 | 1600 | 	       (strcmp(filename, "<stdin>") == 0) || | 
 | 1601 | 	       (strcmp(filename, "???") == 0); | 
 | 1602 | } | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1603 |  | 
 | 1604 |  | 
| Tim Peters | d08e382 | 2003-04-17 15:24:21 +0000 | [diff] [blame] | 1605 | #if defined(USE_STACKCHECK) | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1606 | #if defined(WIN32) && defined(_MSC_VER) | 
 | 1607 |  | 
 | 1608 | /* Stack checking for Microsoft C */ | 
 | 1609 |  | 
 | 1610 | #include <malloc.h> | 
 | 1611 | #include <excpt.h> | 
 | 1612 |  | 
| Fred Drake | e8de31c | 2000-08-31 05:38:39 +0000 | [diff] [blame] | 1613 | /* | 
 | 1614 |  * Return non-zero when we run out of memory on the stack; zero otherwise. | 
 | 1615 |  */ | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1616 | int | 
| Fred Drake | 399739f | 2000-08-31 05:52:44 +0000 | [diff] [blame] | 1617 | PyOS_CheckStack(void) | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1618 | { | 
 | 1619 | 	__try { | 
| Tim Peters | 92e4dd8 | 2002-10-05 01:47:34 +0000 | [diff] [blame] | 1620 | 		/* alloca throws a stack overflow exception if there's | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1621 | 		   not enough space left on the stack */ | 
| Tim Peters | 92e4dd8 | 2002-10-05 01:47:34 +0000 | [diff] [blame] | 1622 | 		alloca(PYOS_STACK_MARGIN * sizeof(void*)); | 
| Fredrik Lundh | 2f15b25 | 2000-08-27 19:15:31 +0000 | [diff] [blame] | 1623 | 		return 0; | 
 | 1624 | 	} __except (EXCEPTION_EXECUTE_HANDLER) { | 
 | 1625 | 		/* just ignore all errors */ | 
 | 1626 | 	} | 
 | 1627 | 	return 1; | 
 | 1628 | } | 
 | 1629 |  | 
 | 1630 | #endif /* WIN32 && _MSC_VER */ | 
 | 1631 |  | 
 | 1632 | /* Alternate implementations can be added here... */ | 
 | 1633 |  | 
 | 1634 | #endif /* USE_STACKCHECK */ | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1635 |  | 
 | 1636 |  | 
 | 1637 | /* Wrappers around sigaction() or signal(). */ | 
 | 1638 |  | 
 | 1639 | PyOS_sighandler_t | 
 | 1640 | PyOS_getsig(int sig) | 
 | 1641 | { | 
 | 1642 | #ifdef HAVE_SIGACTION | 
 | 1643 | 	struct sigaction context; | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1644 | 	if (sigaction(sig, NULL, &context) == -1) | 
 | 1645 | 		return SIG_ERR; | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1646 | 	return context.sa_handler; | 
 | 1647 | #else | 
 | 1648 | 	PyOS_sighandler_t handler; | 
| Martin v. Löwis | b45b315 | 2005-11-28 17:34:23 +0000 | [diff] [blame] | 1649 | /* Special signal handling for the secure CRT in Visual Studio 2005 */ | 
 | 1650 | #if defined(_MSC_VER) && _MSC_VER >= 1400 | 
 | 1651 | 	switch (sig) { | 
 | 1652 | 	/* Only these signals are valid */ | 
 | 1653 | 	case SIGINT: | 
 | 1654 | 	case SIGILL: | 
 | 1655 | 	case SIGFPE: | 
 | 1656 | 	case SIGSEGV: | 
 | 1657 | 	case SIGTERM: | 
 | 1658 | 	case SIGBREAK: | 
 | 1659 | 	case SIGABRT: | 
 | 1660 | 		break; | 
 | 1661 | 	/* Don't call signal() with other values or it will assert */ | 
 | 1662 | 	default: | 
 | 1663 | 		return SIG_ERR; | 
 | 1664 | 	} | 
 | 1665 | #endif /* _MSC_VER && _MSC_VER >= 1400 */ | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1666 | 	handler = signal(sig, SIG_IGN); | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1667 | 	if (handler != SIG_ERR) | 
 | 1668 | 		signal(sig, handler); | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1669 | 	return handler; | 
 | 1670 | #endif | 
 | 1671 | } | 
 | 1672 |  | 
 | 1673 | PyOS_sighandler_t | 
 | 1674 | PyOS_setsig(int sig, PyOS_sighandler_t handler) | 
 | 1675 | { | 
 | 1676 | #ifdef HAVE_SIGACTION | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1677 | 	struct sigaction context, ocontext; | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1678 | 	context.sa_handler = handler; | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1679 | 	sigemptyset(&context.sa_mask); | 
 | 1680 | 	context.sa_flags = 0; | 
 | 1681 | 	if (sigaction(sig, &context, &ocontext) == -1) | 
 | 1682 | 		return SIG_ERR; | 
 | 1683 | 	return ocontext.sa_handler; | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1684 | #else | 
| Anthony Baxter | 9ceaa72 | 2004-10-13 14:48:50 +0000 | [diff] [blame] | 1685 | 	PyOS_sighandler_t oldhandler; | 
 | 1686 | 	oldhandler = signal(sig, handler); | 
 | 1687 | #ifdef HAVE_SIGINTERRUPT | 
 | 1688 | 	siginterrupt(sig, 1); | 
 | 1689 | #endif | 
 | 1690 | 	return oldhandler; | 
| Guido van Rossum | 6f25618 | 2000-09-16 16:32:19 +0000 | [diff] [blame] | 1691 | #endif | 
 | 1692 | } | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1693 |  | 
 | 1694 | /* Deprecated C API functions still provided for binary compatiblity */ | 
 | 1695 |  | 
 | 1696 | #undef PyParser_SimpleParseFile | 
| Thomas Heller | 1b04664 | 2006-04-18 18:51:06 +0000 | [diff] [blame] | 1697 | PyAPI_FUNC(node *) | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1698 | PyParser_SimpleParseFile(FILE *fp, const char *filename, int start) | 
 | 1699 | { | 
 | 1700 | 	return PyParser_SimpleParseFileFlags(fp, filename, start, 0); | 
 | 1701 | } | 
 | 1702 |  | 
| Thomas Heller | 1b04664 | 2006-04-18 18:51:06 +0000 | [diff] [blame] | 1703 | #undef PyParser_SimpleParseString | 
 | 1704 | PyAPI_FUNC(node *) | 
| Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 1705 | PyParser_SimpleParseString(const char *str, int start) | 
 | 1706 | { | 
 | 1707 | 	return PyParser_SimpleParseStringFlags(str, start, 0); | 
 | 1708 | } | 
| Anthony Baxter | ac6bd46 | 2006-04-13 02:06:09 +0000 | [diff] [blame] | 1709 |  | 
| Thomas Heller | 1b04664 | 2006-04-18 18:51:06 +0000 | [diff] [blame] | 1710 | #undef PyRun_AnyFile | 
 | 1711 | PyAPI_FUNC(int) | 
 | 1712 | PyRun_AnyFile(FILE *fp, const char *name) | 
 | 1713 | { | 
 | 1714 | 	return PyRun_AnyFileExFlags(fp, name, 0, NULL); | 
 | 1715 | } | 
 | 1716 |  | 
 | 1717 | #undef PyRun_AnyFileEx | 
 | 1718 | PyAPI_FUNC(int) | 
 | 1719 | PyRun_AnyFileEx(FILE *fp, const char *name, int closeit) | 
 | 1720 | { | 
 | 1721 | 	return PyRun_AnyFileExFlags(fp, name, closeit, NULL); | 
 | 1722 | } | 
 | 1723 |  | 
 | 1724 | #undef PyRun_AnyFileFlags | 
 | 1725 | PyAPI_FUNC(int) | 
 | 1726 | PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags) | 
 | 1727 | { | 
 | 1728 | 	return PyRun_AnyFileExFlags(fp, name, 0, flags); | 
 | 1729 | } | 
 | 1730 |  | 
 | 1731 | #undef PyRun_File | 
 | 1732 | PyAPI_FUNC(PyObject *) | 
 | 1733 | PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l) | 
 | 1734 | { | 
 | 1735 |         return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL); | 
 | 1736 | } | 
 | 1737 |  | 
 | 1738 | #undef PyRun_FileEx | 
 | 1739 | PyAPI_FUNC(PyObject *) | 
 | 1740 | PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c) | 
 | 1741 | { | 
 | 1742 |         return PyRun_FileExFlags(fp, p, s, g, l, c, NULL); | 
 | 1743 | } | 
 | 1744 |  | 
 | 1745 | #undef PyRun_FileFlags | 
 | 1746 | PyAPI_FUNC(PyObject *) | 
 | 1747 | PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, | 
 | 1748 | 		PyCompilerFlags *flags) | 
 | 1749 | { | 
 | 1750 |         return PyRun_FileExFlags(fp, p, s, g, l, 0, flags); | 
 | 1751 | } | 
 | 1752 |  | 
 | 1753 | #undef PyRun_SimpleFile | 
 | 1754 | PyAPI_FUNC(int) | 
 | 1755 | PyRun_SimpleFile(FILE *f, const char *p) | 
 | 1756 | { | 
 | 1757 | 	return PyRun_SimpleFileExFlags(f, p, 0, NULL); | 
 | 1758 | } | 
 | 1759 |  | 
 | 1760 | #undef PyRun_SimpleFileEx | 
 | 1761 | PyAPI_FUNC(int) | 
 | 1762 | PyRun_SimpleFileEx(FILE *f, const char *p, int c) | 
 | 1763 | { | 
 | 1764 | 	return PyRun_SimpleFileExFlags(f, p, c, NULL); | 
 | 1765 | } | 
 | 1766 |  | 
 | 1767 |  | 
 | 1768 | #undef PyRun_String | 
 | 1769 | PyAPI_FUNC(PyObject *) | 
 | 1770 | PyRun_String(const char *str, int s, PyObject *g, PyObject *l) | 
 | 1771 | { | 
 | 1772 | 	return PyRun_StringFlags(str, s, g, l, NULL); | 
 | 1773 | } | 
 | 1774 |  | 
 | 1775 | #undef PyRun_SimpleString | 
 | 1776 | PyAPI_FUNC(int) | 
 | 1777 | PyRun_SimpleString(const char *s) | 
 | 1778 | { | 
 | 1779 | 	return PyRun_SimpleStringFlags(s, NULL); | 
 | 1780 | } | 
 | 1781 |  | 
 | 1782 | #undef Py_CompileString | 
 | 1783 | PyAPI_FUNC(PyObject *) | 
 | 1784 | Py_CompileString(const char *str, const char *p, int s) | 
 | 1785 | { | 
 | 1786 | 	return Py_CompileStringFlags(str, p, s, NULL); | 
 | 1787 | } | 
 | 1788 |  | 
 | 1789 | #undef PyRun_InteractiveOne | 
 | 1790 | PyAPI_FUNC(int) | 
 | 1791 | PyRun_InteractiveOne(FILE *f, const char *p) | 
 | 1792 | { | 
 | 1793 | 	return PyRun_InteractiveOneFlags(f, p, NULL); | 
 | 1794 | } | 
 | 1795 |  | 
 | 1796 | #undef PyRun_InteractiveLoop | 
 | 1797 | PyAPI_FUNC(int) | 
 | 1798 | PyRun_InteractiveLoop(FILE *f, const char *p) | 
 | 1799 | { | 
 | 1800 | 	return PyRun_InteractiveLoopFlags(f, p, NULL); | 
 | 1801 | } | 
 | 1802 |  | 
| Anthony Baxter | ac6bd46 | 2006-04-13 02:06:09 +0000 | [diff] [blame] | 1803 | #ifdef __cplusplus | 
 | 1804 | } | 
 | 1805 | #endif | 
 | 1806 |  |