blob: cdcf9f6f32abffd3b7b5a07e5e5e7ee98c116a0a [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum3f5da241990-12-20 15:06:42 +00002/* Execute compiled code */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003
Guido van Rossum681d79a1995-07-18 14:51:37 +00004/* XXX TO DO:
Guido van Rossum681d79a1995-07-18 14:51:37 +00005 XXX speed up searching for keywords by using a dictionary
Guido van Rossum681d79a1995-07-18 14:51:37 +00006 XXX document it!
7 */
8
Fredrik Lundh7a830892006-05-27 10:39:48 +00009/* enable more aggressive intra-module optimizations, where available */
Fredrik Lundh57640f52006-05-26 11:54:04 +000010#define PY_LOCAL_AGGRESSIVE
11
Guido van Rossumb209a111997-04-29 18:18:01 +000012#include "Python.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000013
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000014#include "code.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000015#include "frameobject.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000016#include "eval.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000017#include "opcode.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000018#include "structmember.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000019
Guido van Rossumc6004111993-11-05 10:22:19 +000020#include <ctype.h>
21
Tim Peters7df5e7f2006-05-26 23:14:37 +000022#ifndef WITH_TSC
Michael W. Hudson75eabd22005-01-18 15:56:11 +000023
24#define READ_TIMESTAMP(var)
25
26#else
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000027
28typedef unsigned long long uint64;
29
Michael W. Hudson800ba232004-08-12 18:19:17 +000030#if defined(__ppc__) /* <- Don't know if this is the correct symbol; this
Thomas Wouterse2176022007-09-20 17:35:10 +000031 section should work for GCC on any PowerPC
32 platform, irrespective of OS.
33 POWER? Who knows :-) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000034
Michael W. Hudson75eabd22005-01-18 15:56:11 +000035#define READ_TIMESTAMP(var) ppc_getcounter(&var)
Michael W. Hudson800ba232004-08-12 18:19:17 +000036
Fredrik Lundh7a830892006-05-27 10:39:48 +000037static void
Michael W. Hudson800ba232004-08-12 18:19:17 +000038ppc_getcounter(uint64 *v)
39{
40 register unsigned long tbu, tb, tbu2;
41
42 loop:
43 asm volatile ("mftbu %0" : "=r" (tbu) );
44 asm volatile ("mftb %0" : "=r" (tb) );
45 asm volatile ("mftbu %0" : "=r" (tbu2));
46 if (__builtin_expect(tbu != tbu2, 0)) goto loop;
47
Tim Peters7df5e7f2006-05-26 23:14:37 +000048 /* The slightly peculiar way of writing the next lines is
Michael W. Hudson800ba232004-08-12 18:19:17 +000049 compiled better by GCC than any other way I tried. */
50 ((long*)(v))[0] = tbu;
51 ((long*)(v))[1] = tb;
52}
53
Michael W. Hudson75eabd22005-01-18 15:56:11 +000054#else /* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000055
Michael W. Hudson75eabd22005-01-18 15:56:11 +000056#define READ_TIMESTAMP(val) \
57 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000058
59#endif
60
Tim Peters7df5e7f2006-05-26 23:14:37 +000061void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000062 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
63{
64 uint64 intr, inst, loop;
65 PyThreadState *tstate = PyThreadState_Get();
66 if (!tstate->interp->tscdump)
67 return;
68 intr = intr1 - intr0;
69 inst = inst1 - inst0 - intr;
70 loop = loop1 - loop0 - intr;
71 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
72 opcode, ticked, inst, loop);
73}
Michael W. Hudson800ba232004-08-12 18:19:17 +000074
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000075#endif
76
Guido van Rossum04691fc1992-08-12 15:35:34 +000077/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000078/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000079
Guido van Rossum408027e1996-12-30 16:17:54 +000080#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000081/* For debugging the interpreter: */
82#define LLTRACE 1 /* Low-level trace feature */
83#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +000084#endif
85
Jeremy Hylton52820442001-01-03 23:52:36 +000086typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +000087
Guido van Rossum374a9221991-04-04 10:40:29 +000088/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000089#ifdef WITH_TSC
Fredrik Lundh7a830892006-05-27 10:39:48 +000090static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091#else
Fredrik Lundh7a830892006-05-27 10:39:48 +000092static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +000094static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
95static PyObject * do_call(PyObject *, PyObject ***, int, int);
96static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouterse2176022007-09-20 17:35:10 +000097static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
98 PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +000099static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
100static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000101#define CALL_FLAG_VAR 1
102#define CALL_FLAG_KW 2
103
Guido van Rossum0a066c01992-03-27 17:29:15 +0000104#ifdef LLTRACE
Fredrik Lundh1b949402006-05-26 12:01:49 +0000105static int lltrace;
Fredrik Lundh7a830892006-05-27 10:39:48 +0000106static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000107#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +0000108static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Fred Drake5755ce62001-06-27 19:19:46 +0000109 int, PyObject *);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000110static int call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000111 PyFrameObject *, int, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000112static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
113static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Armin Rigobf57a142004-03-22 19:24:58 +0000114 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000115
Fredrik Lundh7a830892006-05-27 10:39:48 +0000116static PyObject * apply_slice(PyObject *, PyObject *, PyObject *);
117static int assign_slice(PyObject *, PyObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000118 PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000119static PyObject * cmp_outcome(int, PyObject *, PyObject *);
120static PyObject * import_from(PyObject *, PyObject *);
121static int import_all_from(PyObject *, PyObject *);
122static PyObject * build_class(PyObject *, PyObject *, PyObject *);
123static int exec_statement(PyFrameObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000124 PyObject *, PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000125static void set_exc_info(PyThreadState *, PyObject *, PyObject *, PyObject *);
126static void reset_exc_info(PyThreadState *);
127static void format_exc_check_arg(PyObject *, char *, PyObject *);
128static PyObject * string_concatenate(PyObject *, PyObject *,
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000129 PyFrameObject *, unsigned char *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000130
Paul Prescode68140d2000-08-30 20:25:01 +0000131#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000132 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000133#define GLOBAL_NAME_ERROR_MSG \
134 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000135#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000136 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000137#define UNBOUNDFREE_ERROR_MSG \
138 "free variable '%.200s' referenced before assignment" \
139 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000140
Guido van Rossum950361c1997-01-24 13:49:28 +0000141/* Dynamic execution profile */
142#ifdef DYNAMIC_EXECUTION_PROFILE
143#ifdef DXPAIRS
144static long dxpairs[257][256];
145#define dxp dxpairs[256]
146#else
147static long dxp[256];
148#endif
149#endif
150
Jeremy Hylton985eba52003-02-05 23:13:00 +0000151/* Function call profile */
152#ifdef CALL_PROFILE
153#define PCALL_NUM 11
154static int pcall[PCALL_NUM];
155
156#define PCALL_ALL 0
157#define PCALL_FUNCTION 1
158#define PCALL_FAST_FUNCTION 2
159#define PCALL_FASTER_FUNCTION 3
160#define PCALL_METHOD 4
161#define PCALL_BOUND_METHOD 5
162#define PCALL_CFUNCTION 6
163#define PCALL_TYPE 7
164#define PCALL_GENERATOR 8
165#define PCALL_OTHER 9
166#define PCALL_POP 10
167
168/* Notes about the statistics
169
170 PCALL_FAST stats
171
172 FAST_FUNCTION means no argument tuple needs to be created.
173 FASTER_FUNCTION means that the fast-path frame setup code is used.
174
175 If there is a method call where the call can be optimized by changing
176 the argument tuple and calling the function directly, it gets recorded
177 twice.
178
179 As a result, the relationship among the statistics appears to be
180 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
181 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
182 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
183 PCALL_METHOD > PCALL_BOUND_METHOD
184*/
185
186#define PCALL(POS) pcall[POS]++
187
188PyObject *
189PyEval_GetCallStats(PyObject *self)
190{
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000191 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000192 pcall[0], pcall[1], pcall[2], pcall[3],
193 pcall[4], pcall[5], pcall[6], pcall[7],
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000194 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000195}
196#else
197#define PCALL(O)
198
199PyObject *
200PyEval_GetCallStats(PyObject *self)
201{
202 Py_INCREF(Py_None);
203 return Py_None;
204}
205#endif
206
Tim Peters5ca576e2001-06-18 22:08:13 +0000207
Guido van Rossume59214e1994-08-30 08:01:59 +0000208#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000209
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000210#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000211#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000212#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000213#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000214
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000215static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Guido van Rossuma9672091994-09-14 13:31:22 +0000216static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000217
Tim Peters7f468f22004-10-11 02:40:51 +0000218int
219PyEval_ThreadsInitialized(void)
220{
221 return interpreter_lock != 0;
222}
223
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000224void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000225PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000226{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000227 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000228 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000229 interpreter_lock = PyThread_allocate_lock();
230 PyThread_acquire_lock(interpreter_lock, 1);
231 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000232}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000233
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000234void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000235PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000237 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000238}
239
240void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000241PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000242{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000243 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000244}
245
246void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000247PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000248{
249 if (tstate == NULL)
250 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000251 /* Check someone has called PyEval_InitThreads() to create the lock */
252 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000253 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000254 if (PyThreadState_Swap(tstate) != NULL)
255 Py_FatalError(
256 "PyEval_AcquireThread: non-NULL old thread state");
257}
258
259void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000260PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000261{
262 if (tstate == NULL)
263 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
264 if (PyThreadState_Swap(NULL) != tstate)
265 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000266 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000267}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000268
269/* This function is called from PyOS_AfterFork to ensure that newly
270 created child processes don't hold locks referring to threads which
271 are not running in the child process. (This could also be done using
272 pthread_atfork mechanism, at least for the pthreads implementation.) */
273
274void
275PyEval_ReInitThreads(void)
276{
Jesse Noller5e62ca42008-07-16 20:03:47 +0000277 PyObject *threading, *result;
278 PyThreadState *tstate;
279
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000280 if (!interpreter_lock)
281 return;
282 /*XXX Can't use PyThread_free_lock here because it does too
283 much error-checking. Doing this cleanly would require
284 adding a new function to each thread_*.h. Instead, just
285 create a new lock and waste a little bit of memory */
286 interpreter_lock = PyThread_allocate_lock();
287 PyThread_acquire_lock(interpreter_lock, 1);
288 main_thread = PyThread_get_thread_ident();
Jesse Noller5e62ca42008-07-16 20:03:47 +0000289
290 /* Update the threading module with the new state.
291 */
292 tstate = PyThreadState_GET();
293 threading = PyMapping_GetItemString(tstate->interp->modules,
294 "threading");
295 if (threading == NULL) {
296 /* threading not imported */
297 PyErr_Clear();
298 return;
299 }
300 result = PyObject_CallMethod(threading, "_after_fork", NULL);
301 if (result == NULL)
302 PyErr_WriteUnraisable(threading);
303 else
304 Py_DECREF(result);
305 Py_DECREF(threading);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000306}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000307#endif
308
Guido van Rossumff4949e1992-08-05 19:58:53 +0000309/* Functions save_thread and restore_thread are always defined so
310 dynamically loaded modules needn't be compiled separately for use
311 with and without threads: */
312
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000313PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000314PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000315{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000316 PyThreadState *tstate = PyThreadState_Swap(NULL);
317 if (tstate == NULL)
318 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000319#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000320 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000321 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000322#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000323 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000324}
325
326void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000327PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000328{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000329 if (tstate == NULL)
330 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000331#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000332 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000333 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000334 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000335 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000336 }
337#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000338 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000339}
340
341
Guido van Rossuma9672091994-09-14 13:31:22 +0000342/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
343 signal handlers or Mac I/O completion routines) can schedule calls
344 to a function to be called synchronously.
345 The synchronous function is called with one void* argument.
346 It should return 0 for success or -1 for failure -- failure should
347 be accompanied by an exception.
348
349 If registry succeeds, the registry function returns 0; if it fails
350 (e.g. due to too many pending calls) it returns -1 (without setting
351 an exception condition).
352
353 Note that because registry may occur from within signal handlers,
354 or other asynchronous events, calling malloc() is unsafe!
355
356#ifdef WITH_THREAD
357 Any thread can schedule pending calls, but only the main thread
358 will execute them.
359#endif
360
361 XXX WARNING! ASYNCHRONOUSLY EXECUTING CODE!
362 There are two possible race conditions:
363 (1) nested asynchronous registry calls;
364 (2) registry calls made while pending calls are being processed.
365 While (1) is very unlikely, (2) is a real possibility.
366 The current code is safe against (2), but not against (1).
367 The safety against (2) is derived from the fact that only one
368 thread (the main thread) ever takes things out of the queue.
Guido van Rossuma9672091994-09-14 13:31:22 +0000369
Guido van Rossuma027efa1997-05-05 20:56:21 +0000370 XXX Darn! With the advent of thread state, we should have an array
371 of pending calls per thread in the thread state! Later...
372*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000373
Guido van Rossuma9672091994-09-14 13:31:22 +0000374#define NPENDINGCALLS 32
375static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000376 int (*func)(void *);
377 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000378} pendingcalls[NPENDINGCALLS];
379static volatile int pendingfirst = 0;
380static volatile int pendinglast = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000381static volatile int things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000382
383int
Thomas Wouters334fb892000-07-25 12:56:38 +0000384Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000385{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000386 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000387 int i, j;
388 /* XXX Begin critical section */
389 /* XXX If you want this to be safe against nested
390 XXX asynchronous calls, you'll have to work harder! */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000391 if (busy)
392 return -1;
393 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000394 i = pendinglast;
395 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000396 if (j == pendingfirst) {
397 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000398 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000399 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000400 pendingcalls[i].func = func;
401 pendingcalls[i].arg = arg;
402 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000403
404 _Py_Ticker = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000405 things_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000406 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000407 /* XXX End critical section */
408 return 0;
409}
410
Guido van Rossum180d7b41994-09-29 09:45:57 +0000411int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000412Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000413{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000414 static int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000415#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000416 if (main_thread && PyThread_get_thread_ident() != main_thread)
Guido van Rossuma9672091994-09-14 13:31:22 +0000417 return 0;
418#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000419 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000420 return 0;
421 busy = 1;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000422 things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000423 for (;;) {
424 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000425 int (*func)(void *);
426 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000427 i = pendingfirst;
428 if (i == pendinglast)
429 break; /* Queue empty */
430 func = pendingcalls[i].func;
431 arg = pendingcalls[i].arg;
432 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000433 if (func(arg) < 0) {
434 busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000435 things_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000436 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000437 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000438 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000439 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000440 return 0;
441}
442
443
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000444/* The interpreter's recursion limit */
445
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000446#ifndef Py_DEFAULT_RECURSION_LIMIT
447#define Py_DEFAULT_RECURSION_LIMIT 1000
448#endif
449static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
450int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000451
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000452int
453Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000454{
455 return recursion_limit;
456}
457
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000458void
459Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000460{
461 recursion_limit = new_limit;
Thomas Woutersae406c62007-09-19 17:27:43 +0000462 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000463}
464
Armin Rigo2b3eb402003-10-28 12:05:48 +0000465/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
466 if the recursion_depth reaches _Py_CheckRecursionLimit.
467 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
468 to guarantee that _Py_CheckRecursiveCall() is regularly called.
469 Without USE_STACKCHECK, there is no need for this. */
470int
471_Py_CheckRecursiveCall(char *where)
472{
473 PyThreadState *tstate = PyThreadState_GET();
474
475#ifdef USE_STACKCHECK
476 if (PyOS_CheckStack()) {
477 --tstate->recursion_depth;
478 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
479 return -1;
480 }
481#endif
482 if (tstate->recursion_depth > recursion_limit) {
483 --tstate->recursion_depth;
484 PyErr_Format(PyExc_RuntimeError,
485 "maximum recursion depth exceeded%s",
486 where);
487 return -1;
488 }
Thomas Woutersae406c62007-09-19 17:27:43 +0000489 _Py_CheckRecursionLimit = recursion_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000490 return 0;
491}
492
Guido van Rossum374a9221991-04-04 10:40:29 +0000493/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000494enum why_code {
495 WHY_NOT = 0x0001, /* No error */
496 WHY_EXCEPTION = 0x0002, /* Exception occurred */
497 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
498 WHY_RETURN = 0x0008, /* 'return' statement */
499 WHY_BREAK = 0x0010, /* 'break' statement */
500 WHY_CONTINUE = 0x0020, /* 'continue' statement */
501 WHY_YIELD = 0x0040 /* 'yield' operator */
502};
Guido van Rossum374a9221991-04-04 10:40:29 +0000503
Fredrik Lundh7a830892006-05-27 10:39:48 +0000504static enum why_code do_raise(PyObject *, PyObject *, PyObject *);
505static int unpack_iterable(PyObject *, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000506
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +0000507/* Records whether tracing is on for any thread. Counts the number of
508 threads for which tstate->c_tracefunc is non-NULL, so if the value
509 is 0, we know we don't have to check this thread's c_tracefunc.
510 This speeds up the if statement in PyEval_EvalFrameEx() after
511 fast_next_opcode*/
512static int _Py_TracingPossible = 0;
513
Skip Montanarod581d772002-09-03 20:10:45 +0000514/* for manipulating the thread switch and periodic "stuff" - used to be
515 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000516int _Py_CheckInterval = 100;
517volatile int _Py_Ticker = 100;
Guido van Rossum374a9221991-04-04 10:40:29 +0000518
Guido van Rossumb209a111997-04-29 18:18:01 +0000519PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000520PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000521{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000522 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000523 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000524 (PyObject **)NULL, 0,
525 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000526 (PyObject **)NULL, 0,
527 NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000528}
529
530
531/* Interpreter main loop */
532
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000533PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000534PyEval_EvalFrame(PyFrameObject *f) {
535 /* This is for backward compatibility with extension modules that
Thomas Wouterse2176022007-09-20 17:35:10 +0000536 used this API; core interpreter code should call
537 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000538 return PyEval_EvalFrameEx(f, 0);
539}
540
541PyObject *
Anthony Baxtera863d332006-04-11 07:43:46 +0000542PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000543{
Guido van Rossum950361c1997-01-24 13:49:28 +0000544#ifdef DXPAIRS
545 int lastopcode = 0;
546#endif
Thomas Wouterse2176022007-09-20 17:35:10 +0000547 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000548 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000549 register int opcode; /* Current opcode */
550 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000551 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000552 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000553 register PyObject *x; /* Result object -- NULL if error */
554 register PyObject *v; /* Temporary objects popped off stack */
555 register PyObject *w;
556 register PyObject *u;
557 register PyObject *t;
Barry Warsaw23c9ec82000-08-21 15:44:01 +0000558 register PyObject *stream = NULL; /* for PRINT opcodes */
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000559 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000560 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000561 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000562 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000563
Tim Peters8a5c3c72004-04-05 19:36:21 +0000564 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000565
566 not (instr_lb <= current_bytecode_offset < instr_ub)
567
Tim Peters8a5c3c72004-04-05 19:36:21 +0000568 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000569 initial values are such as to make this false the first
570 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000571 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000572
Guido van Rossumd076c731998-10-07 19:42:25 +0000573 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000574 PyObject *names;
575 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000576#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000577 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000578 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000579#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000580
Neal Norwitza81d2202002-07-14 00:27:26 +0000581/* Tuple access macros */
582
583#ifndef Py_DEBUG
584#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
585#else
586#define GETITEM(v, i) PyTuple_GetItem((v), (i))
587#endif
588
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000589#ifdef WITH_TSC
590/* Use Pentium timestamp counter to mark certain events:
591 inst0 -- beginning of switch statement for opcode dispatch
592 inst1 -- end of switch statement (may be skipped)
593 loop0 -- the top of the mainloop
Tim Peters7df5e7f2006-05-26 23:14:37 +0000594 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000595 (may be skipped)
596 intr1 -- beginning of long interruption
597 intr2 -- end of long interruption
598
599 Many opcodes call out to helper C functions. In some cases, the
600 time in those functions should be counted towards the time for the
601 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
602 calls another Python function; there's no point in charge all the
603 bytecode executed by the called function to the caller.
604
605 It's hard to make a useful judgement statically. In the presence
606 of operator overloading, it's impossible to tell if a call will
607 execute new Python code or not.
608
609 It's a case-by-case judgement. I'll use intr1 for the following
610 cases:
611
612 EXEC_STMT
613 IMPORT_STAR
614 IMPORT_FROM
615 CALL_FUNCTION (and friends)
616
617 */
618 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
619 int ticked = 0;
620
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000621 READ_TIMESTAMP(inst0);
622 READ_TIMESTAMP(inst1);
623 READ_TIMESTAMP(loop0);
624 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000625
626 /* shut up the compiler */
627 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000628#endif
629
Guido van Rossum374a9221991-04-04 10:40:29 +0000630/* Code access macros */
631
Martin v. Löwis18e16552006-02-15 17:27:45 +0000632#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000633#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000634#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000635#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000636#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000637#define JUMPBY(x) (next_instr += (x))
638
Raymond Hettingerf606f872003-03-16 03:11:04 +0000639/* OpCode prediction macros
Thomas Wouterse2176022007-09-20 17:35:10 +0000640 Some opcodes tend to come in pairs thus making it possible to
641 predict the second code when the first is run. For example,
642 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
643 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000644
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000645 Verifying the prediction costs a single high-speed test of a register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000646 variable against a constant. If the pairing was good, then the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000647 processor's own internal branch predication has a high likelihood of
648 success, resulting in a nearly zero-overhead transition to the
649 next opcode. A successful prediction saves a trip through the eval-loop
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000650 including its two unpredictable branches, the HAS_ARG test and the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000651 switch-case. Combined with the processor's internal branch prediction,
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000652 a successful PREDICT has the effect of making the two opcodes run as if
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000653 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000654
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000655 If collecting opcode statistics, your choices are to either keep the
656 predictions turned-on and interpret the results as if some opcodes
657 had been combined or turn-off predictions so that the opcode frequency
658 counter updates for both opcodes.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000659*/
660
Raymond Hettingera7216982004-02-08 19:59:27 +0000661#ifdef DYNAMIC_EXECUTION_PROFILE
662#define PREDICT(op) if (0) goto PRED_##op
663#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000664#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000665#endif
666
Raymond Hettingerf606f872003-03-16 03:11:04 +0000667#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000668#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000669
Guido van Rossum374a9221991-04-04 10:40:29 +0000670/* Stack manipulation macros */
671
Martin v. Löwis18e16552006-02-15 17:27:45 +0000672/* The stack can grow at most MAXINT deep, as co_nlocals and
673 co_stacksize are ints. */
674#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000675#define EMPTY() (STACK_LEVEL() == 0)
676#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000677#define SECOND() (stack_pointer[-2])
678#define THIRD() (stack_pointer[-3])
679#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000680#define SET_TOP(v) (stack_pointer[-1] = (v))
681#define SET_SECOND(v) (stack_pointer[-2] = (v))
682#define SET_THIRD(v) (stack_pointer[-3] = (v))
683#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000684#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000685#define BASIC_PUSH(v) (*stack_pointer++ = (v))
686#define BASIC_POP() (*--stack_pointer)
687
Guido van Rossum96a42c81992-01-12 02:29:51 +0000688#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000689#define PUSH(v) { (void)(BASIC_PUSH(v), \
690 lltrace && prtrace(TOP(), "push")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000691 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouterse2176022007-09-20 17:35:10 +0000692#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
693 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000694#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
695 lltrace && prtrace(TOP(), "stackadj")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000696 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes52729ac2007-12-14 02:33:57 +0000697#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
698 prtrace((STACK_POINTER)[-1], "ext_pop")), \
699 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000700#else
701#define PUSH(v) BASIC_PUSH(v)
702#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000703#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000704#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000705#endif
706
Guido van Rossum681d79a1995-07-18 14:51:37 +0000707/* Local variable macros */
708
709#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000710
711/* The SETLOCAL() macro must not DECREF the local variable in-place and
712 then store the new value; it must copy the old value to a temporary
713 value, then store the new value, and then DECREF the temporary value.
714 This is because it is possible that during the DECREF the frame is
715 accessed by other code (e.g. a __del__ method or gc.collect()) and the
716 variable would be pointing to already-freed memory. */
717#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
718 GETLOCAL(i) = value; \
719 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000720
Guido van Rossuma027efa1997-05-05 20:56:21 +0000721/* Start of code */
722
Tim Peters5ca576e2001-06-18 22:08:13 +0000723 if (f == NULL)
724 return NULL;
725
Armin Rigo1d313ab2003-10-25 14:33:09 +0000726 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000727 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000728 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000729
Tim Peters5ca576e2001-06-18 22:08:13 +0000730 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000731
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000732 if (tstate->use_tracing) {
733 if (tstate->c_tracefunc != NULL) {
734 /* tstate->c_tracefunc, if defined, is a
735 function that will be called on *every* entry
736 to a code block. Its return value, if not
737 None, is a function that will be called at
738 the start of each executed line of code.
739 (Actually, the function must return itself
740 in order to continue tracing.) The trace
741 functions are called with three arguments:
742 a pointer to the current frame, a string
743 indicating why the function is called, and
744 an argument which depends on the situation.
745 The global trace function is also called
746 whenever an exception is detected. */
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000747 if (call_trace_protected(tstate->c_tracefunc,
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000748 tstate->c_traceobj,
749 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000750 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000751 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000752 }
753 }
754 if (tstate->c_profilefunc != NULL) {
755 /* Similar for c_profilefunc, except it needn't
756 return itself and isn't called for "line" events */
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000757 if (call_trace_protected(tstate->c_profilefunc,
758 tstate->c_profileobj,
759 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000760 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000761 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000762 }
763 }
764 }
765
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000766 co = f->f_code;
767 names = co->co_names;
768 consts = co->co_consts;
769 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000770 freevars = f->f_localsplus + co->co_nlocals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000771 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000772 /* An explanation is in order for the next line.
773
774 f->f_lasti now refers to the index of the last instruction
775 executed. You might think this was obvious from the name, but
776 this wasn't always true before 2.3! PyFrame_New now sets
777 f->f_lasti to -1 (i.e. the index *before* the first instruction)
778 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000779 does work. Promise.
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000780
781 When the PREDICT() macros are enabled, some opcode pairs follow in
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000782 direct succession without updating f->f_lasti. A successful
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000783 prediction effectively links the two codes together as if they
784 were a single new opcode; accordingly,f->f_lasti will point to
785 the first code in the pair (for instance, GET_ITER followed by
786 FOR_ITER is effectively a single opcode and f->f_lasti will point
787 at to the beginning of the combined pair.)
788 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000789 next_instr = first_instr + f->f_lasti + 1;
790 stack_pointer = f->f_stacktop;
791 assert(stack_pointer != NULL);
792 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
793
Tim Peters5ca576e2001-06-18 22:08:13 +0000794#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000795 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000796#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000797#if defined(Py_DEBUG) || defined(LLTRACE)
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000798 filename = PyString_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000799#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000800
Guido van Rossum374a9221991-04-04 10:40:29 +0000801 why = WHY_NOT;
802 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000803 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000804 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000805
Anthony Baxtera863d332006-04-11 07:43:46 +0000806 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000807 why = WHY_EXCEPTION;
808 goto on_error;
809 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000810
Guido van Rossum374a9221991-04-04 10:40:29 +0000811 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000812#ifdef WITH_TSC
813 if (inst1 == 0) {
814 /* Almost surely, the opcode executed a break
815 or a continue, preventing inst1 from being set
816 on the way out of the loop.
817 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000818 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000819 loop1 = inst1;
820 }
821 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
822 intr0, intr1);
823 ticked = 0;
824 inst1 = 0;
825 intr0 = 0;
826 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000827 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000828#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000829 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000830 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000831
Guido van Rossuma027efa1997-05-05 20:56:21 +0000832 /* Do periodic things. Doing this every time through
833 the loop would add too much overhead, so we do it
834 only every Nth instruction. We also do it if
835 ``things_to_do'' is set, i.e. when an asynchronous
836 event needs attention (e.g. a signal handler or
837 async I/O handler); see Py_AddPendingCall() and
838 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000839
Skip Montanarod581d772002-09-03 20:10:45 +0000840 if (--_Py_Ticker < 0) {
Thomas Woutersae406c62007-09-19 17:27:43 +0000841 if (*next_instr == SETUP_FINALLY) {
842 /* Make the last opcode before
843 a try: finally: block uninterruptable. */
844 goto fast_next_opcode;
845 }
Skip Montanarod581d772002-09-03 20:10:45 +0000846 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000847 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000848#ifdef WITH_TSC
849 ticked = 1;
850#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000851 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000852 if (Py_MakePendingCalls() < 0) {
853 why = WHY_EXCEPTION;
854 goto on_error;
855 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000856 if (things_to_do)
857 /* MakePendingCalls() didn't succeed.
858 Force early re-execution of this
859 "periodic" code, possibly after
860 a thread switch */
861 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000862 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000863#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864 if (interpreter_lock) {
865 /* Give another thread a chance */
866
Guido van Rossum25ce5661997-08-02 03:10:38 +0000867 if (PyThreadState_Swap(NULL) != tstate)
868 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000869 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000870
871 /* Other threads may run now */
872
Guido van Rossum65d5b571998-12-21 19:32:43 +0000873 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000874 if (PyThreadState_Swap(tstate) != NULL)
875 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000876
877 /* Check for thread interrupts */
878
879 if (tstate->async_exc != NULL) {
880 x = tstate->async_exc;
881 tstate->async_exc = NULL;
882 PyErr_SetNone(x);
883 Py_DECREF(x);
884 why = WHY_EXCEPTION;
885 goto on_error;
886 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000887 }
888#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000889 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000890
Neil Schemenauer63543862002-02-17 19:10:14 +0000891 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000892 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000893
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000894 /* line-by-line tracing support */
895
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +0000896 if (_Py_TracingPossible &&
897 tstate->c_tracefunc != NULL && !tstate->tracing) {
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000898 /* see maybe_call_line_trace
899 for expository comments */
900 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000901
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000902 err = maybe_call_line_trace(tstate->c_tracefunc,
903 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000904 f, &instr_lb, &instr_ub,
905 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000906 /* Reload possibly changed frame fields */
907 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000908 if (f->f_stacktop != NULL) {
909 stack_pointer = f->f_stacktop;
910 f->f_stacktop = NULL;
911 }
912 if (err) {
913 /* trace function raised an exception */
914 goto on_error;
915 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000916 }
917
918 /* Extract opcode and argument */
919
Guido van Rossum374a9221991-04-04 10:40:29 +0000920 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000921 oparg = 0; /* allows oparg to be stored in a register because
922 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000923 if (HAS_ARG(opcode))
924 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000925 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000926#ifdef DYNAMIC_EXECUTION_PROFILE
927#ifdef DXPAIRS
928 dxpairs[lastopcode][opcode]++;
929 lastopcode = opcode;
930#endif
931 dxp[opcode]++;
932#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000933
Guido van Rossum96a42c81992-01-12 02:29:51 +0000934#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000935 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000936
Guido van Rossum96a42c81992-01-12 02:29:51 +0000937 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000938 if (HAS_ARG(opcode)) {
939 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000940 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000941 }
942 else {
943 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000944 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000945 }
946 }
947#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000948
Guido van Rossum374a9221991-04-04 10:40:29 +0000949 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000950 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000951
Guido van Rossum374a9221991-04-04 10:40:29 +0000952 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000953
Guido van Rossum374a9221991-04-04 10:40:29 +0000954 /* BEWARE!
955 It is essential that any operation that fails sets either
956 x to NULL, err to nonzero, or why to anything but WHY_NOT,
957 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000958
Guido van Rossum374a9221991-04-04 10:40:29 +0000959 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000960
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000961 case NOP:
962 goto fast_next_opcode;
963
Neil Schemenauer63543862002-02-17 19:10:14 +0000964 case LOAD_FAST:
965 x = GETLOCAL(oparg);
966 if (x != NULL) {
967 Py_INCREF(x);
968 PUSH(x);
969 goto fast_next_opcode;
970 }
971 format_exc_check_arg(PyExc_UnboundLocalError,
972 UNBOUNDLOCAL_ERROR_MSG,
973 PyTuple_GetItem(co->co_varnames, oparg));
974 break;
975
976 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000977 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000978 Py_INCREF(x);
979 PUSH(x);
980 goto fast_next_opcode;
981
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000982 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000983 case STORE_FAST:
984 v = POP();
985 SETLOCAL(oparg, v);
986 goto fast_next_opcode;
987
Raymond Hettingerf606f872003-03-16 03:11:04 +0000988 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000989 case POP_TOP:
990 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000991 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000992 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000993
Guido van Rossum374a9221991-04-04 10:40:29 +0000994 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000995 v = TOP();
996 w = SECOND();
997 SET_TOP(w);
998 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000999 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001000
Guido van Rossum374a9221991-04-04 10:40:29 +00001001 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001002 v = TOP();
1003 w = SECOND();
1004 x = THIRD();
1005 SET_TOP(w);
1006 SET_SECOND(x);
1007 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001008 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001009
Thomas Wouters434d0822000-08-24 20:11:32 +00001010 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001011 u = TOP();
1012 v = SECOND();
1013 w = THIRD();
1014 x = FOURTH();
1015 SET_TOP(v);
1016 SET_SECOND(w);
1017 SET_THIRD(x);
1018 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001019 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001020
Guido van Rossum374a9221991-04-04 10:40:29 +00001021 case DUP_TOP:
1022 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001023 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001024 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001025 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001026
Thomas Wouters434d0822000-08-24 20:11:32 +00001027 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001028 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001029 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001030 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001031 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001032 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001033 STACKADJ(2);
1034 SET_TOP(x);
1035 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001036 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001037 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001038 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001039 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001040 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001041 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001042 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001043 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001044 STACKADJ(3);
1045 SET_TOP(x);
1046 SET_SECOND(w);
1047 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001048 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001049 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001050 Py_FatalError("invalid argument to DUP_TOPX"
1051 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001052 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001053
Guido van Rossum374a9221991-04-04 10:40:29 +00001054 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001055 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001056 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001057 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001058 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001059 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001060 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001061
Guido van Rossum374a9221991-04-04 10:40:29 +00001062 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001063 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001064 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001065 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001066 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001067 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001068 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001069
Guido van Rossum374a9221991-04-04 10:40:29 +00001070 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001071 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001072 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001073 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001074 if (err == 0) {
1075 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001076 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001077 continue;
1078 }
1079 else if (err > 0) {
1080 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001081 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001082 err = 0;
1083 continue;
1084 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001085 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001086 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001087
Guido van Rossum374a9221991-04-04 10:40:29 +00001088 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001089 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001090 x = PyObject_Repr(v);
1091 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001092 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001093 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001094 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001095
Guido van Rossum7928cd71991-10-24 14:59:31 +00001096 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001097 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001098 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001099 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001100 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001101 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001102 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001103
Guido van Rossum50564e81996-01-12 01:13:16 +00001104 case BINARY_POWER:
1105 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001106 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001107 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001108 Py_DECREF(v);
1109 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001110 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001111 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001112 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001113
Guido van Rossum374a9221991-04-04 10:40:29 +00001114 case BINARY_MULTIPLY:
1115 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001116 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001117 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001118 Py_DECREF(v);
1119 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001120 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001121 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001122 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001123
Guido van Rossum374a9221991-04-04 10:40:29 +00001124 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001125 if (!_Py_QnewFlag) {
1126 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001127 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001128 x = PyNumber_Divide(v, w);
1129 Py_DECREF(v);
1130 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001131 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001132 if (x != NULL) continue;
1133 break;
1134 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001135 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001136 BINARY_TRUE_DIVIDE */
1137 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001138 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001139 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001140 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001141 Py_DECREF(v);
1142 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001143 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001144 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001145 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001146
Guido van Rossum4668b002001-08-08 05:00:18 +00001147 case BINARY_FLOOR_DIVIDE:
1148 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001149 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001150 x = PyNumber_FloorDivide(v, w);
1151 Py_DECREF(v);
1152 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001153 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001154 if (x != NULL) continue;
1155 break;
1156
Guido van Rossum374a9221991-04-04 10:40:29 +00001157 case BINARY_MODULO:
1158 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001159 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001160 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001161 Py_DECREF(v);
1162 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001163 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001164 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001165 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001166
Guido van Rossum374a9221991-04-04 10:40:29 +00001167 case BINARY_ADD:
1168 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001169 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001170 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001171 /* INLINE: int + int */
1172 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001173 a = PyInt_AS_LONG(v);
1174 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001175 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001176 if ((i^a) < 0 && (i^b) < 0)
1177 goto slow_add;
1178 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001179 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001180 else if (PyString_CheckExact(v) &&
1181 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001182 x = string_concatenate(v, w, f, next_instr);
1183 /* string_concatenate consumed the ref to v */
1184 goto skip_decref_vx;
1185 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001186 else {
1187 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001188 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001189 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001190 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001191 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001192 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001193 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001194 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001195 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001196
Guido van Rossum374a9221991-04-04 10:40:29 +00001197 case BINARY_SUBTRACT:
1198 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001199 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001200 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001201 /* INLINE: int - int */
1202 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001203 a = PyInt_AS_LONG(v);
1204 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001205 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001206 if ((i^a) < 0 && (i^~b) < 0)
1207 goto slow_sub;
1208 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001209 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001210 else {
1211 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001212 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001213 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001214 Py_DECREF(v);
1215 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001216 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001217 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001218 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001219
Guido van Rossum374a9221991-04-04 10:40:29 +00001220 case BINARY_SUBSCR:
1221 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001222 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001223 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001224 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001225 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001226 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001227 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001228 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001229 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001230 Py_INCREF(x);
1231 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001232 else
1233 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001234 }
1235 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001236 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001237 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001238 Py_DECREF(v);
1239 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001240 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001241 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001242 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001243
Guido van Rossum7928cd71991-10-24 14:59:31 +00001244 case BINARY_LSHIFT:
1245 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001246 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001247 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001248 Py_DECREF(v);
1249 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001250 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001251 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001252 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001253
Guido van Rossum7928cd71991-10-24 14:59:31 +00001254 case BINARY_RSHIFT:
1255 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001256 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001257 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001258 Py_DECREF(v);
1259 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001260 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001261 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001262 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001263
Guido van Rossum7928cd71991-10-24 14:59:31 +00001264 case BINARY_AND:
1265 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001266 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001267 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001268 Py_DECREF(v);
1269 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001270 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001271 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001272 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001273
Guido van Rossum7928cd71991-10-24 14:59:31 +00001274 case BINARY_XOR:
1275 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001276 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001277 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001278 Py_DECREF(v);
1279 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001280 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001281 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001282 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001283
Guido van Rossum7928cd71991-10-24 14:59:31 +00001284 case BINARY_OR:
1285 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001286 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001287 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001288 Py_DECREF(v);
1289 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001290 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001291 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001292 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001293
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001294 case LIST_APPEND:
1295 w = POP();
1296 v = POP();
1297 err = PyList_Append(v, w);
1298 Py_DECREF(v);
1299 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001300 if (err == 0) {
1301 PREDICT(JUMP_ABSOLUTE);
1302 continue;
1303 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001304 break;
1305
Thomas Wouters434d0822000-08-24 20:11:32 +00001306 case INPLACE_POWER:
1307 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001308 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001309 x = PyNumber_InPlacePower(v, w, Py_None);
1310 Py_DECREF(v);
1311 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001312 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001313 if (x != NULL) continue;
1314 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001315
Thomas Wouters434d0822000-08-24 20:11:32 +00001316 case INPLACE_MULTIPLY:
1317 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001318 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001319 x = PyNumber_InPlaceMultiply(v, w);
1320 Py_DECREF(v);
1321 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001322 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001323 if (x != NULL) continue;
1324 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001325
Thomas Wouters434d0822000-08-24 20:11:32 +00001326 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001327 if (!_Py_QnewFlag) {
1328 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001329 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001330 x = PyNumber_InPlaceDivide(v, w);
1331 Py_DECREF(v);
1332 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001333 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001334 if (x != NULL) continue;
1335 break;
1336 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001337 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001338 INPLACE_TRUE_DIVIDE */
1339 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001340 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001341 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001342 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001343 Py_DECREF(v);
1344 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001345 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001346 if (x != NULL) continue;
1347 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001348
Guido van Rossum4668b002001-08-08 05:00:18 +00001349 case INPLACE_FLOOR_DIVIDE:
1350 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001351 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001352 x = PyNumber_InPlaceFloorDivide(v, w);
1353 Py_DECREF(v);
1354 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001355 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001356 if (x != NULL) continue;
1357 break;
1358
Thomas Wouters434d0822000-08-24 20:11:32 +00001359 case INPLACE_MODULO:
1360 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001361 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001362 x = PyNumber_InPlaceRemainder(v, w);
1363 Py_DECREF(v);
1364 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001365 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001366 if (x != NULL) continue;
1367 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001368
Thomas Wouters434d0822000-08-24 20:11:32 +00001369 case INPLACE_ADD:
1370 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001371 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001372 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001373 /* INLINE: int + int */
1374 register long a, b, i;
1375 a = PyInt_AS_LONG(v);
1376 b = PyInt_AS_LONG(w);
1377 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001378 if ((i^a) < 0 && (i^b) < 0)
1379 goto slow_iadd;
1380 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001381 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001382 else if (PyString_CheckExact(v) &&
1383 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001384 x = string_concatenate(v, w, f, next_instr);
1385 /* string_concatenate consumed the ref to v */
1386 goto skip_decref_v;
1387 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001388 else {
1389 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001390 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001391 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001392 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001393 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001394 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001395 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001396 if (x != NULL) continue;
1397 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001398
Thomas Wouters434d0822000-08-24 20:11:32 +00001399 case INPLACE_SUBTRACT:
1400 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001401 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001402 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001403 /* INLINE: int - int */
1404 register long a, b, i;
1405 a = PyInt_AS_LONG(v);
1406 b = PyInt_AS_LONG(w);
1407 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001408 if ((i^a) < 0 && (i^~b) < 0)
1409 goto slow_isub;
1410 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001411 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001412 else {
1413 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001414 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001415 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001416 Py_DECREF(v);
1417 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001418 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001419 if (x != NULL) continue;
1420 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001421
Thomas Wouters434d0822000-08-24 20:11:32 +00001422 case INPLACE_LSHIFT:
1423 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001424 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001425 x = PyNumber_InPlaceLshift(v, w);
1426 Py_DECREF(v);
1427 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001428 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001429 if (x != NULL) continue;
1430 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001431
Thomas Wouters434d0822000-08-24 20:11:32 +00001432 case INPLACE_RSHIFT:
1433 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001434 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001435 x = PyNumber_InPlaceRshift(v, w);
1436 Py_DECREF(v);
1437 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001438 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001439 if (x != NULL) continue;
1440 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001441
Thomas Wouters434d0822000-08-24 20:11:32 +00001442 case INPLACE_AND:
1443 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001444 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001445 x = PyNumber_InPlaceAnd(v, w);
1446 Py_DECREF(v);
1447 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001448 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001449 if (x != NULL) continue;
1450 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001451
Thomas Wouters434d0822000-08-24 20:11:32 +00001452 case INPLACE_XOR:
1453 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001454 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001455 x = PyNumber_InPlaceXor(v, w);
1456 Py_DECREF(v);
1457 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001458 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001459 if (x != NULL) continue;
1460 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001461
Thomas Wouters434d0822000-08-24 20:11:32 +00001462 case INPLACE_OR:
1463 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001464 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001465 x = PyNumber_InPlaceOr(v, w);
1466 Py_DECREF(v);
1467 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001468 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001469 if (x != NULL) continue;
1470 break;
1471
Guido van Rossum374a9221991-04-04 10:40:29 +00001472 case SLICE+0:
1473 case SLICE+1:
1474 case SLICE+2:
1475 case SLICE+3:
1476 if ((opcode-SLICE) & 2)
1477 w = POP();
1478 else
1479 w = NULL;
1480 if ((opcode-SLICE) & 1)
1481 v = POP();
1482 else
1483 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001484 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001485 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001486 Py_DECREF(u);
1487 Py_XDECREF(v);
1488 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001489 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001490 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001491 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001492
Guido van Rossum374a9221991-04-04 10:40:29 +00001493 case STORE_SLICE+0:
1494 case STORE_SLICE+1:
1495 case STORE_SLICE+2:
1496 case STORE_SLICE+3:
1497 if ((opcode-STORE_SLICE) & 2)
1498 w = POP();
1499 else
1500 w = NULL;
1501 if ((opcode-STORE_SLICE) & 1)
1502 v = POP();
1503 else
1504 v = NULL;
1505 u = POP();
1506 t = POP();
1507 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001508 Py_DECREF(t);
1509 Py_DECREF(u);
1510 Py_XDECREF(v);
1511 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001512 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001513 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001514
Guido van Rossum374a9221991-04-04 10:40:29 +00001515 case DELETE_SLICE+0:
1516 case DELETE_SLICE+1:
1517 case DELETE_SLICE+2:
1518 case DELETE_SLICE+3:
1519 if ((opcode-DELETE_SLICE) & 2)
1520 w = POP();
1521 else
1522 w = NULL;
1523 if ((opcode-DELETE_SLICE) & 1)
1524 v = POP();
1525 else
1526 v = NULL;
1527 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001528 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001529 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001530 Py_DECREF(u);
1531 Py_XDECREF(v);
1532 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001533 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001534 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001535
Guido van Rossum374a9221991-04-04 10:40:29 +00001536 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001537 w = TOP();
1538 v = SECOND();
1539 u = THIRD();
1540 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001541 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001542 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001543 Py_DECREF(u);
1544 Py_DECREF(v);
1545 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001546 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001547 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001548
Guido van Rossum374a9221991-04-04 10:40:29 +00001549 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001550 w = TOP();
1551 v = SECOND();
1552 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001553 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001554 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001555 Py_DECREF(v);
1556 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001557 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001558 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001559
Guido van Rossum374a9221991-04-04 10:40:29 +00001560 case PRINT_EXPR:
1561 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001562 w = PySys_GetObject("displayhook");
1563 if (w == NULL) {
1564 PyErr_SetString(PyExc_RuntimeError,
1565 "lost sys.displayhook");
1566 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001567 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001568 }
1569 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001570 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001571 if (x == NULL)
1572 err = -1;
1573 }
1574 if (err == 0) {
1575 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001576 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001577 if (w == NULL)
1578 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001579 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001580 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001581 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001582 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001583
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001584 case PRINT_ITEM_TO:
1585 w = stream = POP();
1586 /* fall through to PRINT_ITEM */
1587
Guido van Rossum374a9221991-04-04 10:40:29 +00001588 case PRINT_ITEM:
1589 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001590 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001591 w = PySys_GetObject("stdout");
1592 if (w == NULL) {
1593 PyErr_SetString(PyExc_RuntimeError,
1594 "lost sys.stdout");
1595 err = -1;
1596 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001597 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001598 /* PyFile_SoftSpace() can exececute arbitrary code
1599 if sys.stdout is an instance with a __getattr__.
1600 If __getattr__ raises an exception, w will
1601 be freed, so we need to prevent that temporarily. */
1602 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001603 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001604 err = PyFile_WriteString(" ", w);
1605 if (err == 0)
1606 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001607 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001608 /* XXX move into writeobject() ? */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001609 if (PyString_Check(v)) {
1610 char *s = PyString_AS_STRING(v);
1611 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001612 if (len == 0 ||
1613 !isspace(Py_CHARMASK(s[len-1])) ||
1614 s[len-1] == ' ')
1615 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001616 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001617#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001618 else if (PyUnicode_Check(v)) {
1619 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001620 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001621 if (len == 0 ||
1622 !Py_UNICODE_ISSPACE(s[len-1]) ||
1623 s[len-1] == ' ')
1624 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001625 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001626#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001627 else
1628 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001629 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001630 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001631 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001632 Py_XDECREF(stream);
1633 stream = NULL;
1634 if (err == 0)
1635 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001636 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001637
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001638 case PRINT_NEWLINE_TO:
1639 w = stream = POP();
1640 /* fall through to PRINT_NEWLINE */
1641
Guido van Rossum374a9221991-04-04 10:40:29 +00001642 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001643 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001644 w = PySys_GetObject("stdout");
1645 if (w == NULL)
1646 PyErr_SetString(PyExc_RuntimeError,
1647 "lost sys.stdout");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001648 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001649 if (w != NULL) {
Georg Brandlaa76d772008-07-01 20:56:03 +00001650 /* w.write() may replace sys.stdout, so we
1651 * have to keep our reference to it */
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001652 Py_INCREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001653 err = PyFile_WriteString("\n", w);
1654 if (err == 0)
1655 PyFile_SoftSpace(w, 0);
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001656 Py_DECREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001657 }
1658 Py_XDECREF(stream);
1659 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001660 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001661
Thomas Wouters434d0822000-08-24 20:11:32 +00001662
1663#ifdef CASE_TOO_BIG
1664 default: switch (opcode) {
1665#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001666 case RAISE_VARARGS:
1667 u = v = w = NULL;
1668 switch (oparg) {
1669 case 3:
1670 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001671 /* Fallthrough */
1672 case 2:
1673 v = POP(); /* value */
1674 /* Fallthrough */
1675 case 1:
1676 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001677 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001678 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001679 break;
1680 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001681 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001682 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001683 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001684 break;
1685 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001686 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001687
Guido van Rossum374a9221991-04-04 10:40:29 +00001688 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001689 if ((x = f->f_locals) != NULL) {
1690 Py_INCREF(x);
1691 PUSH(x);
1692 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001693 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001694 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001695 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001696
Guido van Rossum374a9221991-04-04 10:40:29 +00001697 case RETURN_VALUE:
1698 retval = POP();
1699 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001700 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001701
Tim Peters5ca576e2001-06-18 22:08:13 +00001702 case YIELD_VALUE:
1703 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001704 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001705 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001706 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001707
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001708 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001709 w = TOP();
1710 v = SECOND();
1711 u = THIRD();
1712 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001713 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001714 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001715 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001716 Py_DECREF(u);
1717 Py_DECREF(v);
1718 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001719 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001720
Guido van Rossum374a9221991-04-04 10:40:29 +00001721 case POP_BLOCK:
1722 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001723 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001724 while (STACK_LEVEL() > b->b_level) {
1725 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001726 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001727 }
1728 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001729 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001730
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00001731 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001732 case END_FINALLY:
1733 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001734 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001735 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001736 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001737 if (why == WHY_RETURN ||
1738 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001739 retval = POP();
1740 }
Thomas Wouterse2176022007-09-20 17:35:10 +00001741 else if (PyExceptionClass_Check(v) ||
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001742 PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001743 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001744 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001745 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001746 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001747 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001748 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001749 else if (v != Py_None) {
1750 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001751 "'finally' pops bad exception");
1752 why = WHY_EXCEPTION;
1753 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001754 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001755 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001756
Guido van Rossum374a9221991-04-04 10:40:29 +00001757 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001758 u = TOP();
1759 v = SECOND();
1760 w = THIRD();
1761 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001762 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001763 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001764 Py_DECREF(u);
1765 Py_DECREF(v);
1766 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001767 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001768
Guido van Rossum374a9221991-04-04 10:40:29 +00001769 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001770 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001771 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001772 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001773 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001774 err = PyDict_SetItem(x, w, v);
1775 else
1776 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001777 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001778 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001779 break;
1780 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001781 PyErr_Format(PyExc_SystemError,
1782 "no locals found when storing %s",
1783 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001784 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001785
Guido van Rossum374a9221991-04-04 10:40:29 +00001786 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001787 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001788 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001789 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001790 format_exc_check_arg(PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001791 NAME_ERROR_MSG,
1792 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001793 break;
1794 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001795 PyErr_Format(PyExc_SystemError,
1796 "no locals when deleting %s",
1797 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001798 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001799
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001800 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001801 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001802 v = POP();
Thomas Wouterse2176022007-09-20 17:35:10 +00001803 if (PyTuple_CheckExact(v) &&
1804 PyTuple_GET_SIZE(v) == oparg) {
1805 PyObject **items = \
1806 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001807 while (oparg--) {
1808 w = items[oparg];
1809 Py_INCREF(w);
1810 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001811 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001812 Py_DECREF(v);
1813 continue;
Thomas Wouterse2176022007-09-20 17:35:10 +00001814 } else if (PyList_CheckExact(v) &&
1815 PyList_GET_SIZE(v) == oparg) {
1816 PyObject **items = \
1817 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001818 while (oparg--) {
1819 w = items[oparg];
1820 Py_INCREF(w);
1821 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001822 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001823 } else if (unpack_iterable(v, oparg,
Thomas Wouterse2176022007-09-20 17:35:10 +00001824 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001825 stack_pointer += oparg;
Georg Brandl5cb76c12007-03-21 09:00:39 +00001826 } else {
1827 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001828 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001829 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001830 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001831 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001832
Guido van Rossum374a9221991-04-04 10:40:29 +00001833 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001834 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001835 v = TOP();
1836 u = SECOND();
1837 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001838 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1839 Py_DECREF(v);
1840 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001841 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001842 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001843
Guido van Rossum374a9221991-04-04 10:40:29 +00001844 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001845 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001846 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001847 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1848 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001849 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001850 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001851
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001852 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001853 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001854 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001855 err = PyDict_SetItem(f->f_globals, w, v);
1856 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001857 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001858 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001859
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001860 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001861 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001862 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001863 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001864 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001865 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001866
Guido van Rossum374a9221991-04-04 10:40:29 +00001867 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001868 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001869 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001870 PyErr_Format(PyExc_SystemError,
1871 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001872 PyObject_REPR(w));
Guido van Rossum681d79a1995-07-18 14:51:37 +00001873 break;
1874 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001875 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001876 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001877 Py_XINCREF(x);
1878 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001879 else {
1880 x = PyObject_GetItem(v, w);
1881 if (x == NULL && PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00001882 if (!PyErr_ExceptionMatches(
1883 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001884 break;
1885 PyErr_Clear();
1886 }
1887 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001888 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001889 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001890 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001891 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001892 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001893 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001894 PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001895 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001896 break;
1897 }
1898 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001899 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001900 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001901 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001902 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001903
Guido van Rossum374a9221991-04-04 10:40:29 +00001904 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001905 w = GETITEM(names, oparg);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001906 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001907 /* Inline the PyDict_GetItem() calls.
1908 WARNING: this is an extreme speed hack.
1909 Do not try this at home. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001910 long hash = ((PyStringObject *)w)->ob_shash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001911 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001912 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00001913 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001914 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00001915 e = d->ma_lookup(d, w, hash);
1916 if (e == NULL) {
1917 x = NULL;
1918 break;
1919 }
1920 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001921 if (x != NULL) {
1922 Py_INCREF(x);
1923 PUSH(x);
1924 continue;
1925 }
1926 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00001927 e = d->ma_lookup(d, w, hash);
1928 if (e == NULL) {
1929 x = NULL;
1930 break;
1931 }
1932 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001933 if (x != NULL) {
1934 Py_INCREF(x);
1935 PUSH(x);
1936 continue;
1937 }
1938 goto load_global_error;
1939 }
1940 }
1941 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001942 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001943 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001944 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001945 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001946 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001947 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001948 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001949 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001950 break;
1951 }
1952 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001953 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001954 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001955 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001956
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001957 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001958 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001959 if (x != NULL) {
1960 SETLOCAL(oparg, NULL);
1961 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001962 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001963 format_exc_check_arg(
1964 PyExc_UnboundLocalError,
1965 UNBOUNDLOCAL_ERROR_MSG,
1966 PyTuple_GetItem(co->co_varnames, oparg)
1967 );
1968 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001969
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001970 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001971 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001972 Py_INCREF(x);
1973 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001974 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001975 break;
1976
1977 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001978 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001979 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001980 if (w != NULL) {
1981 PUSH(w);
1982 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001983 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001984 err = -1;
1985 /* Don't stomp existing exception */
1986 if (PyErr_Occurred())
1987 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00001988 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1989 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001990 oparg);
1991 format_exc_check_arg(
1992 PyExc_UnboundLocalError,
1993 UNBOUNDLOCAL_ERROR_MSG,
1994 v);
1995 } else {
Thomas Wouterse2176022007-09-20 17:35:10 +00001996 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
1997 PyTuple_GET_SIZE(co->co_cellvars));
1998 format_exc_check_arg(PyExc_NameError,
1999 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002000 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002001 break;
2002
2003 case STORE_DEREF:
2004 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002005 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002006 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00002007 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002008 continue;
2009
Guido van Rossum374a9221991-04-04 10:40:29 +00002010 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00002011 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002012 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002013 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002014 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002015 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002016 }
2017 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002018 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002019 }
2020 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002021
Guido van Rossum374a9221991-04-04 10:40:29 +00002022 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00002023 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002024 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002025 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002026 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00002027 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002028 }
2029 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002030 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002031 }
2032 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002033
Guido van Rossum374a9221991-04-04 10:40:29 +00002034 case BUILD_MAP:
Raymond Hettingerfd7ed402007-12-18 21:24:09 +00002035 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002036 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002037 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002038 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002039
Raymond Hettingereffde122007-12-18 18:26:18 +00002040 case STORE_MAP:
2041 w = TOP(); /* key */
2042 u = SECOND(); /* value */
2043 v = THIRD(); /* dict */
2044 STACKADJ(-2);
2045 assert (PyDict_CheckExact(v));
2046 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2047 Py_DECREF(u);
2048 Py_DECREF(w);
2049 if (err == 0) continue;
2050 break;
2051
Guido van Rossum374a9221991-04-04 10:40:29 +00002052 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00002053 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002054 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002055 x = PyObject_GetAttr(v, w);
2056 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002057 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002058 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002059 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002060
Guido van Rossum374a9221991-04-04 10:40:29 +00002061 case COMPARE_OP:
2062 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002063 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00002064 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002065 /* INLINE: cmp(int, int) */
2066 register long a, b;
2067 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002068 a = PyInt_AS_LONG(v);
2069 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002070 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002071 case PyCmp_LT: res = a < b; break;
2072 case PyCmp_LE: res = a <= b; break;
2073 case PyCmp_EQ: res = a == b; break;
2074 case PyCmp_NE: res = a != b; break;
2075 case PyCmp_GT: res = a > b; break;
2076 case PyCmp_GE: res = a >= b; break;
2077 case PyCmp_IS: res = v == w; break;
2078 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002079 default: goto slow_compare;
2080 }
2081 x = res ? Py_True : Py_False;
2082 Py_INCREF(x);
2083 }
2084 else {
2085 slow_compare:
2086 x = cmp_outcome(oparg, v, w);
2087 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002088 Py_DECREF(v);
2089 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002090 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002091 if (x == NULL) break;
2092 PREDICT(JUMP_IF_FALSE);
2093 PREDICT(JUMP_IF_TRUE);
2094 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002095
Guido van Rossum374a9221991-04-04 10:40:29 +00002096 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002097 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002098 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002099 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002100 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002101 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002102 break;
2103 }
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002104 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002105 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002106 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002107 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2108 w = PyTuple_Pack(5,
2109 w,
2110 f->f_globals,
2111 f->f_locals == NULL ?
2112 Py_None : f->f_locals,
2113 v,
2114 u);
2115 else
2116 w = PyTuple_Pack(4,
2117 w,
2118 f->f_globals,
2119 f->f_locals == NULL ?
2120 Py_None : f->f_locals,
2121 v);
2122 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002123 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002124 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002125 u = POP();
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002126 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002127 x = NULL;
2128 break;
2129 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002130 READ_TIMESTAMP(intr0);
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002131 v = x;
2132 x = PyEval_CallObject(v, w);
2133 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002134 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002135 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002136 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002137 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002138 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002139
Thomas Wouters52152252000-08-17 22:55:00 +00002140 case IMPORT_STAR:
2141 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002142 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002143 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002144 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002145 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002146 break;
2147 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002148 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002149 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002150 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002151 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002152 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002153 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002154 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002155
Thomas Wouters52152252000-08-17 22:55:00 +00002156 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002157 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002158 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002159 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002160 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002161 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002162 PUSH(x);
2163 if (x != NULL) continue;
2164 break;
2165
Guido van Rossum374a9221991-04-04 10:40:29 +00002166 case JUMP_FORWARD:
2167 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002168 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002169
Raymond Hettingerf606f872003-03-16 03:11:04 +00002170 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002171 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002172 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002173 if (w == Py_True) {
2174 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002175 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002176 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002177 if (w == Py_False) {
2178 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002179 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002180 }
2181 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002182 if (err > 0)
2183 err = 0;
2184 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002185 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002186 else
2187 break;
2188 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002189
Raymond Hettingerf606f872003-03-16 03:11:04 +00002190 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002191 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002192 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002193 if (w == Py_False) {
2194 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002195 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002196 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002197 if (w == Py_True) {
2198 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002199 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002200 }
2201 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002202 if (err > 0) {
2203 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002204 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002205 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002206 else if (err == 0)
2207 ;
2208 else
2209 break;
2210 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002211
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002212 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002213 case JUMP_ABSOLUTE:
2214 JUMPTO(oparg);
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002215#if FAST_LOOPS
2216 /* Enabling this path speeds-up all while and for-loops by bypassing
2217 the per-loop checks for signals. By default, this should be turned-off
2218 because it prevents detection of a control-break in tight loops like
2219 "while 1: pass". Compile with this option turned-on when you need
2220 the speed-up and do not need break checking inside tight loops (ones
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002221 that contain only instructions ending with goto fast_next_opcode).
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002222 */
2223 goto fast_next_opcode;
2224#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002225 continue;
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002226#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002227
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002228 case GET_ITER:
2229 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002230 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002231 x = PyObject_GetIter(v);
2232 Py_DECREF(v);
2233 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002234 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002235 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002236 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002237 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002238 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002239 break;
2240
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002241 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002242 case FOR_ITER:
2243 /* before: [iter]; after: [iter, iter()] *or* [] */
2244 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002245 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002246 if (x != NULL) {
2247 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002248 PREDICT(STORE_FAST);
2249 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002250 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002251 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002252 if (PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002253 if (!PyErr_ExceptionMatches(
2254 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002255 break;
2256 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002257 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002258 /* iterator ended normally */
2259 x = v = POP();
2260 Py_DECREF(v);
2261 JUMPBY(oparg);
2262 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002263
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002264 case BREAK_LOOP:
2265 why = WHY_BREAK;
2266 goto fast_block_end;
2267
2268 case CONTINUE_LOOP:
2269 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002270 if (!retval) {
2271 x = NULL;
2272 break;
2273 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002274 why = WHY_CONTINUE;
2275 goto fast_block_end;
2276
Guido van Rossum374a9221991-04-04 10:40:29 +00002277 case SETUP_LOOP:
2278 case SETUP_EXCEPT:
2279 case SETUP_FINALLY:
Thomas Wouterse2176022007-09-20 17:35:10 +00002280 /* NOTE: If you add any new block-setup opcodes that
2281 are not try/except/finally handlers, you may need
2282 to update the PyGen_NeedsFinalizing() function.
2283 */
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002284
Guido van Rossumb209a111997-04-29 18:18:01 +00002285 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002286 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002287 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002288
Guido van Rossumc2e20742006-02-27 22:32:47 +00002289 case WITH_CLEANUP:
2290 {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002291 /* At the top of the stack are 1-3 values indicating
2292 how/why we entered the finally clause:
2293 - TOP = None
2294 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2295 - TOP = WHY_*; no retval below it
2296 - (TOP, SECOND, THIRD) = exc_info()
2297 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002298 In the last case, we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002299 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002300 otherwise we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002301 EXIT(None, None, None)
2302
2303 In all cases, we remove EXIT from the stack, leaving
2304 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002305
2306 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002307 *and* the function call returns a 'true' value, we
2308 "zap" this information, to prevent END_FINALLY from
2309 re-raising the exception. (But non-local gotos
2310 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002311 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002312
Nick Coghlan7af53be2008-03-07 14:13:28 +00002313 PyObject *exit_func;
2314
2315 u = POP();
2316 if (u == Py_None) {
2317 exit_func = TOP();
2318 SET_TOP(u);
2319 v = w = Py_None;
2320 }
2321 else if (PyInt_Check(u)) {
2322 switch(PyInt_AS_LONG(u)) {
2323 case WHY_RETURN:
2324 case WHY_CONTINUE:
2325 /* Retval in TOP. */
2326 exit_func = SECOND();
2327 SET_SECOND(TOP());
2328 SET_TOP(u);
2329 break;
2330 default:
2331 exit_func = TOP();
2332 SET_TOP(u);
2333 break;
2334 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002335 u = v = w = Py_None;
2336 }
2337 else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002338 v = TOP();
2339 w = SECOND();
2340 exit_func = THIRD();
2341 SET_TOP(u);
2342 SET_SECOND(v);
2343 SET_THIRD(w);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002344 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002345 /* XXX Not the fastest way to call it... */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002346 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2347 NULL);
2348 if (x == NULL) {
2349 Py_DECREF(exit_func);
Guido van Rossumf6694362006-03-10 02:28:35 +00002350 break; /* Go to error exit */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002351 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002352 if (u != Py_None && PyObject_IsTrue(x)) {
2353 /* There was an exception and a true return */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002354 STACKADJ(-2);
Guido van Rossumf6694362006-03-10 02:28:35 +00002355 Py_INCREF(Py_None);
2356 SET_TOP(Py_None);
Guido van Rossumf6694362006-03-10 02:28:35 +00002357 Py_DECREF(u);
2358 Py_DECREF(v);
2359 Py_DECREF(w);
2360 } else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002361 /* The stack was rearranged to remove EXIT
2362 above. Let END_FINALLY do its thing */
Guido van Rossumf6694362006-03-10 02:28:35 +00002363 }
Nick Coghlan7af53be2008-03-07 14:13:28 +00002364 Py_DECREF(x);
2365 Py_DECREF(exit_func);
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00002366 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002367 break;
2368 }
2369
Guido van Rossumf10570b1995-07-07 22:53:21 +00002370 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002371 {
2372 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002373 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002374 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002375#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002376 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002377#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002378 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002379#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002380 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002381 PUSH(x);
2382 if (x != NULL)
2383 continue;
2384 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002385 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002386
Jeremy Hylton76901512000-03-28 23:49:17 +00002387 case CALL_FUNCTION_VAR:
2388 case CALL_FUNCTION_KW:
2389 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002390 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002391 int na = oparg & 0xff;
2392 int nk = (oparg>>8) & 0xff;
2393 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002394 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002395 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002396 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002397 if (flags & CALL_FLAG_VAR)
2398 n++;
2399 if (flags & CALL_FLAG_KW)
2400 n++;
2401 pfunc = stack_pointer - n - 1;
2402 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002403
Guido van Rossumac7be682001-01-17 15:42:30 +00002404 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002405 && PyMethod_GET_SELF(func) != NULL) {
2406 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002407 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002408 func = PyMethod_GET_FUNCTION(func);
2409 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002410 Py_DECREF(*pfunc);
2411 *pfunc = self;
2412 na++;
2413 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002414 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002415 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002416 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002417 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002418 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002419 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002420 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002421 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002422
Jeremy Hylton76901512000-03-28 23:49:17 +00002423 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002424 w = POP();
2425 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002426 }
2427 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002428 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002429 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002430 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002431 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002432
Guido van Rossum681d79a1995-07-18 14:51:37 +00002433 case MAKE_FUNCTION:
2434 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002435 x = PyFunction_New(v, f->f_globals);
2436 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002437 /* XXX Maybe this should be a separate opcode? */
2438 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002439 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002440 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002441 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002442 x = NULL;
2443 break;
2444 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002445 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002446 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002447 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002448 }
2449 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002450 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002451 }
2452 PUSH(x);
2453 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002454
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002455 case MAKE_CLOSURE:
2456 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002457 v = POP(); /* code object */
2458 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002459 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002460 if (x != NULL) {
2461 v = POP();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002462 err = PyFunction_SetClosure(x, v);
2463 Py_DECREF(v);
2464 }
2465 if (x != NULL && oparg > 0) {
2466 v = PyTuple_New(oparg);
2467 if (v == NULL) {
2468 Py_DECREF(x);
2469 x = NULL;
2470 break;
2471 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002472 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002473 w = POP();
2474 PyTuple_SET_ITEM(v, oparg, w);
2475 }
2476 err = PyFunction_SetDefaults(x, v);
2477 Py_DECREF(v);
2478 }
2479 PUSH(x);
2480 break;
2481 }
2482
Guido van Rossum8861b741996-07-30 16:49:37 +00002483 case BUILD_SLICE:
2484 if (oparg == 3)
2485 w = POP();
2486 else
2487 w = NULL;
2488 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002489 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002490 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002491 Py_DECREF(u);
2492 Py_DECREF(v);
2493 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002494 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002495 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002496 break;
2497
Fred Drakeef8ace32000-08-24 00:32:09 +00002498 case EXTENDED_ARG:
2499 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002500 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002501 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002502
Guido van Rossum374a9221991-04-04 10:40:29 +00002503 default:
2504 fprintf(stderr,
2505 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002506 PyCode_Addr2Line(f->f_code, f->f_lasti),
2507 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002508 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002509 why = WHY_EXCEPTION;
2510 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002511
2512#ifdef CASE_TOO_BIG
2513 }
2514#endif
2515
Guido van Rossum374a9221991-04-04 10:40:29 +00002516 } /* switch */
2517
2518 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002519
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002520 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002521
Guido van Rossum374a9221991-04-04 10:40:29 +00002522 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002523
Guido van Rossum374a9221991-04-04 10:40:29 +00002524 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002525 if (err == 0 && x != NULL) {
2526#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002527 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002528 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002529 fprintf(stderr,
2530 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002531 else {
2532#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002533 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002534 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002535#ifdef CHECKEXC
2536 }
2537#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002538 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002539 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002540 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002541 err = 0;
2542 }
2543
Guido van Rossum374a9221991-04-04 10:40:29 +00002544 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002545
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002546 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002547 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002548 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002549 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002550 why = WHY_EXCEPTION;
2551 }
2552 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002553#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002554 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002555 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002556 if (PyErr_Occurred()) {
Neal Norwitz8250fbe2008-01-27 17:12:15 +00002557 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002558 sprintf(buf, "Stack unwind with exception "
2559 "set and why=%d", why);
2560 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002561 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002562 }
2563#endif
2564
2565 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002566
Guido van Rossum374a9221991-04-04 10:40:29 +00002567 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002568 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002569
Fred Drake8f51f542001-10-04 14:48:42 +00002570 if (tstate->c_tracefunc != NULL)
2571 call_exc_trace(tstate->c_tracefunc,
2572 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002573 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002574
Guido van Rossum374a9221991-04-04 10:40:29 +00002575 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002576
Guido van Rossum374a9221991-04-04 10:40:29 +00002577 if (why == WHY_RERAISE)
2578 why = WHY_EXCEPTION;
2579
2580 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002581
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002582fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002583 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002584 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002585
Tim Peters8a5c3c72004-04-05 19:36:21 +00002586 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002587 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2588 /* For a continue inside a try block,
2589 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002590 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2591 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002592 why = WHY_NOT;
2593 JUMPTO(PyInt_AS_LONG(retval));
2594 Py_DECREF(retval);
2595 break;
2596 }
2597
Guido van Rossum374a9221991-04-04 10:40:29 +00002598 while (STACK_LEVEL() > b->b_level) {
2599 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002600 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002601 }
2602 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2603 why = WHY_NOT;
2604 JUMPTO(b->b_handler);
2605 break;
2606 }
2607 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002608 (b->b_type == SETUP_EXCEPT &&
2609 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002610 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002611 PyObject *exc, *val, *tb;
2612 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002613 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002614 val = Py_None;
2615 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002616 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002617 /* Make the raw exception data
2618 available to the handler,
2619 so a program can emulate the
2620 Python main loop. Don't do
2621 this for 'finally'. */
2622 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002623 PyErr_NormalizeException(
2624 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002625 set_exc_info(tstate,
2626 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002627 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002628 if (tb == NULL) {
2629 Py_INCREF(Py_None);
2630 PUSH(Py_None);
2631 } else
2632 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002633 PUSH(val);
2634 PUSH(exc);
2635 }
2636 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002637 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002638 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002639 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002640 PUSH(v);
2641 }
2642 why = WHY_NOT;
2643 JUMPTO(b->b_handler);
2644 break;
2645 }
2646 } /* unwind stack */
2647
2648 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002649
Guido van Rossum374a9221991-04-04 10:40:29 +00002650 if (why != WHY_NOT)
2651 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002652 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002653
Guido van Rossum374a9221991-04-04 10:40:29 +00002654 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002655
Tim Peters8a5c3c72004-04-05 19:36:21 +00002656 assert(why != WHY_YIELD);
2657 /* Pop remaining stack entries. */
2658 while (!EMPTY()) {
2659 v = POP();
2660 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002661 }
2662
Tim Peters8a5c3c72004-04-05 19:36:21 +00002663 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002664 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002665
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002666fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002667 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002668 if (tstate->c_tracefunc) {
2669 if (why == WHY_RETURN || why == WHY_YIELD) {
2670 if (call_trace(tstate->c_tracefunc,
2671 tstate->c_traceobj, f,
2672 PyTrace_RETURN, retval)) {
2673 Py_XDECREF(retval);
2674 retval = NULL;
2675 why = WHY_EXCEPTION;
2676 }
2677 }
2678 else if (why == WHY_EXCEPTION) {
2679 call_trace_protected(tstate->c_tracefunc,
2680 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002681 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002682 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002683 }
Fred Drake8f51f542001-10-04 14:48:42 +00002684 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002685 if (why == WHY_EXCEPTION)
2686 call_trace_protected(tstate->c_profilefunc,
2687 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002688 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002689 else if (call_trace(tstate->c_profilefunc,
2690 tstate->c_profileobj, f,
2691 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002692 Py_XDECREF(retval);
2693 retval = NULL;
2694 why = WHY_EXCEPTION;
2695 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002696 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002697 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002698
Tim Peters7df5e7f2006-05-26 23:14:37 +00002699 if (tstate->frame->f_exc_type != NULL)
2700 reset_exc_info(tstate);
2701 else {
2702 assert(tstate->frame->f_exc_value == NULL);
2703 assert(tstate->frame->f_exc_traceback == NULL);
2704 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002705
Tim Peters5ca576e2001-06-18 22:08:13 +00002706 /* pop frame */
Thomas Woutersae406c62007-09-19 17:27:43 +00002707exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002708 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002709 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002710
Guido van Rossum96a42c81992-01-12 02:29:51 +00002711 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002712}
2713
Guido van Rossumc2e20742006-02-27 22:32:47 +00002714/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002715 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002716 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002717
Tim Peters6d6c1a32001-08-02 04:15:00 +00002718PyObject *
2719PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002720 PyObject **args, int argcount, PyObject **kws, int kwcount,
2721 PyObject **defs, int defcount, PyObject *closure)
2722{
2723 register PyFrameObject *f;
2724 register PyObject *retval = NULL;
2725 register PyObject **fastlocals, **freevars;
2726 PyThreadState *tstate = PyThreadState_GET();
2727 PyObject *x, *u;
2728
2729 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002730 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002731 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002732 return NULL;
2733 }
2734
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002735 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002736 assert(globals != NULL);
2737 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002738 if (f == NULL)
2739 return NULL;
2740
2741 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002742 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002743
2744 if (co->co_argcount > 0 ||
2745 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2746 int i;
2747 int n = argcount;
2748 PyObject *kwdict = NULL;
2749 if (co->co_flags & CO_VARKEYWORDS) {
2750 kwdict = PyDict_New();
2751 if (kwdict == NULL)
2752 goto fail;
2753 i = co->co_argcount;
2754 if (co->co_flags & CO_VARARGS)
2755 i++;
2756 SETLOCAL(i, kwdict);
2757 }
2758 if (argcount > co->co_argcount) {
2759 if (!(co->co_flags & CO_VARARGS)) {
2760 PyErr_Format(PyExc_TypeError,
2761 "%.200s() takes %s %d "
2762 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002763 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002764 defcount ? "at most" : "exactly",
2765 co->co_argcount,
2766 kwcount ? "non-keyword " : "",
2767 co->co_argcount == 1 ? "" : "s",
2768 argcount);
2769 goto fail;
2770 }
2771 n = co->co_argcount;
2772 }
2773 for (i = 0; i < n; i++) {
2774 x = args[i];
2775 Py_INCREF(x);
2776 SETLOCAL(i, x);
2777 }
2778 if (co->co_flags & CO_VARARGS) {
2779 u = PyTuple_New(argcount - n);
2780 if (u == NULL)
2781 goto fail;
2782 SETLOCAL(co->co_argcount, u);
2783 for (i = n; i < argcount; i++) {
2784 x = args[i];
2785 Py_INCREF(x);
2786 PyTuple_SET_ITEM(u, i-n, x);
2787 }
2788 }
2789 for (i = 0; i < kwcount; i++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002790 PyObject **co_varnames;
Tim Peters5ca576e2001-06-18 22:08:13 +00002791 PyObject *keyword = kws[2*i];
2792 PyObject *value = kws[2*i + 1];
2793 int j;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002794 if (keyword == NULL || !PyString_Check(keyword)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002795 PyErr_Format(PyExc_TypeError,
2796 "%.200s() keywords must be strings",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002797 PyString_AsString(co->co_name));
Tim Peters5ca576e2001-06-18 22:08:13 +00002798 goto fail;
2799 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002800 /* Speed hack: do raw pointer compares. As names are
2801 normally interned this should almost always hit. */
2802 co_varnames = PySequence_Fast_ITEMS(co->co_varnames);
Tim Peters5ca576e2001-06-18 22:08:13 +00002803 for (j = 0; j < co->co_argcount; j++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002804 PyObject *nm = co_varnames[j];
2805 if (nm == keyword)
2806 goto kw_found;
2807 }
2808 /* Slow fallback, just in case */
2809 for (j = 0; j < co->co_argcount; j++) {
2810 PyObject *nm = co_varnames[j];
Tim Peters5ca576e2001-06-18 22:08:13 +00002811 int cmp = PyObject_RichCompareBool(
2812 keyword, nm, Py_EQ);
2813 if (cmp > 0)
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002814 goto kw_found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002815 else if (cmp < 0)
2816 goto fail;
2817 }
2818 /* Check errors from Compare */
2819 if (PyErr_Occurred())
2820 goto fail;
2821 if (j >= co->co_argcount) {
2822 if (kwdict == NULL) {
2823 PyErr_Format(PyExc_TypeError,
2824 "%.200s() got an unexpected "
2825 "keyword argument '%.400s'",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002826 PyString_AsString(co->co_name),
2827 PyString_AsString(keyword));
Tim Peters5ca576e2001-06-18 22:08:13 +00002828 goto fail;
2829 }
2830 PyDict_SetItem(kwdict, keyword, value);
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002831 continue;
Tim Peters5ca576e2001-06-18 22:08:13 +00002832 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002833kw_found:
2834 if (GETLOCAL(j) != NULL) {
2835 PyErr_Format(PyExc_TypeError,
2836 "%.200s() got multiple "
2837 "values for keyword "
2838 "argument '%.400s'",
2839 PyString_AsString(co->co_name),
2840 PyString_AsString(keyword));
2841 goto fail;
Tim Peters5ca576e2001-06-18 22:08:13 +00002842 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002843 Py_INCREF(value);
2844 SETLOCAL(j, value);
Tim Peters5ca576e2001-06-18 22:08:13 +00002845 }
2846 if (argcount < co->co_argcount) {
2847 int m = co->co_argcount - defcount;
2848 for (i = argcount; i < m; i++) {
2849 if (GETLOCAL(i) == NULL) {
2850 PyErr_Format(PyExc_TypeError,
2851 "%.200s() takes %s %d "
2852 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002853 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002854 ((co->co_flags & CO_VARARGS) ||
2855 defcount) ? "at least"
2856 : "exactly",
2857 m, kwcount ? "non-keyword " : "",
2858 m == 1 ? "" : "s", i);
2859 goto fail;
2860 }
2861 }
2862 if (n > m)
2863 i = n - m;
2864 else
2865 i = 0;
2866 for (; i < defcount; i++) {
2867 if (GETLOCAL(m+i) == NULL) {
2868 PyObject *def = defs[i];
2869 Py_INCREF(def);
2870 SETLOCAL(m+i, def);
2871 }
2872 }
2873 }
2874 }
2875 else {
2876 if (argcount > 0 || kwcount > 0) {
2877 PyErr_Format(PyExc_TypeError,
2878 "%.200s() takes no arguments (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002879 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002880 argcount + kwcount);
2881 goto fail;
2882 }
2883 }
2884 /* Allocate and initialize storage for cell vars, and copy free
2885 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00002886 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00002887 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002888 char *cellname, *argname;
2889 PyObject *c;
2890
2891 nargs = co->co_argcount;
2892 if (co->co_flags & CO_VARARGS)
2893 nargs++;
2894 if (co->co_flags & CO_VARKEYWORDS)
2895 nargs++;
2896
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002897 /* Initialize each cell var, taking into account
2898 cell vars that are initialized from arguments.
2899
2900 Should arrange for the compiler to put cellvars
2901 that are arguments at the beginning of the cellvars
2902 list so that we can march over it more efficiently?
2903 */
Richard Jonescebbefc2006-05-23 18:28:17 +00002904 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002905 cellname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00002906 PyTuple_GET_ITEM(co->co_cellvars, i));
2907 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002908 for (j = 0; j < nargs; j++) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002909 argname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00002910 PyTuple_GET_ITEM(co->co_varnames, j));
2911 if (strcmp(cellname, argname) == 0) {
2912 c = PyCell_New(GETLOCAL(j));
2913 if (c == NULL)
2914 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002915 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002916 found = 1;
2917 break;
2918 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002919 }
2920 if (found == 0) {
2921 c = PyCell_New(NULL);
2922 if (c == NULL)
2923 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002924 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002925 }
2926 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002927 }
Richard Jonescebbefc2006-05-23 18:28:17 +00002928 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002929 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00002930 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002931 PyObject *o = PyTuple_GET_ITEM(closure, i);
2932 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00002933 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002934 }
2935 }
2936
Tim Peters5ca576e2001-06-18 22:08:13 +00002937 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002938 /* Don't need to keep the reference to f_back, it will be set
2939 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002940 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002941 f->f_back = NULL;
2942
Jeremy Hylton985eba52003-02-05 23:13:00 +00002943 PCALL(PCALL_GENERATOR);
2944
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002945 /* Create a new generator that owns the ready to run frame
2946 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002947 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002948 }
2949
Thomas Woutersae406c62007-09-19 17:27:43 +00002950 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002951
Thomas Woutersae406c62007-09-19 17:27:43 +00002952fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00002953
Tim Petersb13680b2001-11-27 23:29:29 +00002954 /* decref'ing the frame can cause __del__ methods to get invoked,
2955 which can call back into Python. While we're done with the
2956 current Python frame (f), the associated C stack is still in use,
2957 so recursion_depth must be boosted for the duration.
2958 */
2959 assert(tstate != NULL);
2960 ++tstate->recursion_depth;
Thomas Woutersae406c62007-09-19 17:27:43 +00002961 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002962 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002963 return retval;
2964}
2965
2966
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002967/* Implementation notes for set_exc_info() and reset_exc_info():
2968
2969- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2970 'exc_traceback'. These always travel together.
2971
2972- tstate->curexc_ZZZ is the "hot" exception that is set by
2973 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2974
2975- Once an exception is caught by an except clause, it is transferred
2976 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2977 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00002978 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002979
2980- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2981
2982 Long ago, when none of this existed, there were just a few globals:
2983 one set corresponding to the "hot" exception, and one set
2984 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2985 globals; they were simply stored as sys.exc_ZZZ. For backwards
2986 compatibility, they still are!) The problem was that in code like
2987 this:
2988
2989 try:
2990 "something that may fail"
2991 except "some exception":
2992 "do something else first"
2993 "print the exception from sys.exc_ZZZ."
2994
2995 if "do something else first" invoked something that raised and caught
2996 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2997 cause of subtle bugs. I fixed this by changing the semantics as
2998 follows:
2999
3000 - Within one frame, sys.exc_ZZZ will hold the last exception caught
3001 *in that frame*.
3002
3003 - But initially, and as long as no exception is caught in a given
3004 frame, sys.exc_ZZZ will hold the last exception caught in the
3005 previous frame (or the frame before that, etc.).
3006
3007 The first bullet fixed the bug in the above example. The second
3008 bullet was for backwards compatibility: it was (and is) common to
3009 have a function that is called when an exception is caught, and to
3010 have that function access the caught exception via sys.exc_ZZZ.
3011 (Example: traceback.print_exc()).
3012
3013 At the same time I fixed the problem that sys.exc_ZZZ weren't
3014 thread-safe, by introducing sys.exc_info() which gets it from tstate;
3015 but that's really a separate improvement.
3016
3017 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
3018 variables to what they were before the current frame was called. The
3019 set_exc_info() function saves them on the frame so that
3020 reset_exc_info() can restore them. The invariant is that
3021 frame->f_exc_ZZZ is NULL iff the current frame never caught an
3022 exception (where "catching" an exception applies only to successful
3023 except clauses); and if the current frame ever caught an exception,
3024 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
3025 at the start of the current frame.
3026
3027*/
3028
Fredrik Lundh7a830892006-05-27 10:39:48 +00003029static void
Guido van Rossumac7be682001-01-17 15:42:30 +00003030set_exc_info(PyThreadState *tstate,
3031 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003032{
Tim Peters7df5e7f2006-05-26 23:14:37 +00003033 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003034 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00003035
Tim Peters7df5e7f2006-05-26 23:14:37 +00003036 assert(type != NULL);
3037 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003038 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003039 assert(frame->f_exc_value == NULL);
3040 assert(frame->f_exc_traceback == NULL);
3041 /* This frame didn't catch an exception before. */
3042 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003043 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003044 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003045 Py_INCREF(Py_None);
3046 tstate->exc_type = Py_None;
3047 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003048 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003049 Py_XINCREF(tstate->exc_value);
3050 Py_XINCREF(tstate->exc_traceback);
3051 frame->f_exc_type = tstate->exc_type;
3052 frame->f_exc_value = tstate->exc_value;
3053 frame->f_exc_traceback = tstate->exc_traceback;
3054 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003055 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003056 tmp_type = tstate->exc_type;
3057 tmp_value = tstate->exc_value;
3058 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003059 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003060 Py_XINCREF(value);
3061 Py_XINCREF(tb);
3062 tstate->exc_type = type;
3063 tstate->exc_value = value;
3064 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003065 Py_XDECREF(tmp_type);
3066 Py_XDECREF(tmp_value);
3067 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003068 /* For b/w compatibility */
3069 PySys_SetObject("exc_type", type);
3070 PySys_SetObject("exc_value", value);
3071 PySys_SetObject("exc_traceback", tb);
3072}
3073
Fredrik Lundh7a830892006-05-27 10:39:48 +00003074static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003075reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003076{
3077 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003078 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003079
3080 /* It's a precondition that the thread state's frame caught an
3081 * exception -- verify in a debug build.
3082 */
3083 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003084 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003085 assert(frame != NULL);
3086 assert(frame->f_exc_type != NULL);
3087
3088 /* Copy the frame's exception info back to the thread state. */
3089 tmp_type = tstate->exc_type;
3090 tmp_value = tstate->exc_value;
3091 tmp_tb = tstate->exc_traceback;
3092 Py_INCREF(frame->f_exc_type);
3093 Py_XINCREF(frame->f_exc_value);
3094 Py_XINCREF(frame->f_exc_traceback);
3095 tstate->exc_type = frame->f_exc_type;
3096 tstate->exc_value = frame->f_exc_value;
3097 tstate->exc_traceback = frame->f_exc_traceback;
3098 Py_XDECREF(tmp_type);
3099 Py_XDECREF(tmp_value);
3100 Py_XDECREF(tmp_tb);
3101
3102 /* For b/w compatibility */
3103 PySys_SetObject("exc_type", frame->f_exc_type);
3104 PySys_SetObject("exc_value", frame->f_exc_value);
3105 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
3106
3107 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003108 tmp_type = frame->f_exc_type;
3109 tmp_value = frame->f_exc_value;
3110 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003111 frame->f_exc_type = NULL;
3112 frame->f_exc_value = NULL;
3113 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003114 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003115 Py_XDECREF(tmp_value);
3116 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003117}
3118
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003119/* Logic for the raise statement (too complicated for inlining).
3120 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003121static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003122do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003123{
Guido van Rossumd295f121998-04-09 21:39:57 +00003124 if (type == NULL) {
3125 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003126 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003127 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3128 value = tstate->exc_value;
3129 tb = tstate->exc_traceback;
3130 Py_XINCREF(type);
3131 Py_XINCREF(value);
3132 Py_XINCREF(tb);
3133 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003134
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003135 /* We support the following forms of raise:
3136 raise <class>, <classinstance>
3137 raise <class>, <argument tuple>
3138 raise <class>, None
3139 raise <class>, <argument>
3140 raise <classinstance>, None
3141 raise <string>, <object>
3142 raise <string>, None
3143
3144 An omitted second argument is the same as None.
3145
3146 In addition, raise <tuple>, <anything> is the same as
3147 raising the tuple's first item (and it better have one!);
3148 this rule is applied recursively.
3149
3150 Finally, an optional third argument can be supplied, which
3151 gives the traceback to be substituted (useful when
3152 re-raising an exception after examining it). */
3153
3154 /* First, check the traceback argument, replacing None with
3155 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003156 if (tb == Py_None) {
3157 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003158 tb = NULL;
3159 }
3160 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003161 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003162 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003163 goto raise_error;
3164 }
3165
3166 /* Next, replace a missing value with None */
3167 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003168 value = Py_None;
3169 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003170 }
3171
3172 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003173 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3174 PyObject *tmp = type;
3175 type = PyTuple_GET_ITEM(type, 0);
3176 Py_INCREF(type);
3177 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003178 }
3179
Brett Cannon129bd522007-01-30 21:34:36 +00003180 if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003181 PyErr_NormalizeException(&type, &value, &tb);
3182
Brett Cannonbf364092006-03-01 04:25:17 +00003183 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003184 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003185 if (value != Py_None) {
3186 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003187 "instance exception may not have a separate value");
3188 goto raise_error;
3189 }
3190 else {
3191 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003192 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003193 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003194 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003195 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003196 }
3197 }
3198 else {
3199 /* Not something you can raise. You get an exception
3200 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003201 PyErr_Format(PyExc_TypeError,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003202 "exceptions must be classes or instances, not %s",
3203 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003204 goto raise_error;
3205 }
Guido van Rossum504153d2008-03-18 04:26:48 +00003206
3207 assert(PyExceptionClass_Check(type));
3208 if (Py_Py3kWarningFlag && PyClass_Check(type)) {
Benjamin Peterson9f4f4812008-04-27 03:01:45 +00003209 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003210 "exceptions must derive from BaseException "
3211 "in 3.x", 1) < 0)
Guido van Rossum504153d2008-03-18 04:26:48 +00003212 goto raise_error;
3213 }
3214
Guido van Rossumb209a111997-04-29 18:18:01 +00003215 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003216 if (tb == NULL)
3217 return WHY_EXCEPTION;
3218 else
3219 return WHY_RERAISE;
3220 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003221 Py_XDECREF(value);
3222 Py_XDECREF(type);
3223 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003224 return WHY_EXCEPTION;
3225}
3226
Tim Petersd6d010b2001-06-21 02:49:55 +00003227/* Iterate v argcnt times and store the results on the stack (via decreasing
3228 sp). Return 1 for success, 0 if error. */
3229
Fredrik Lundh7a830892006-05-27 10:39:48 +00003230static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003231unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003232{
Tim Petersd6d010b2001-06-21 02:49:55 +00003233 int i = 0;
3234 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003235 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003236
Tim Petersd6d010b2001-06-21 02:49:55 +00003237 assert(v != NULL);
3238
3239 it = PyObject_GetIter(v);
3240 if (it == NULL)
3241 goto Error;
3242
3243 for (; i < argcnt; i++) {
3244 w = PyIter_Next(it);
3245 if (w == NULL) {
3246 /* Iterator done, via error or exhaustion. */
3247 if (!PyErr_Occurred()) {
3248 PyErr_Format(PyExc_ValueError,
3249 "need more than %d value%s to unpack",
3250 i, i == 1 ? "" : "s");
3251 }
3252 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003253 }
3254 *--sp = w;
3255 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003256
3257 /* We better have exhausted the iterator now. */
3258 w = PyIter_Next(it);
3259 if (w == NULL) {
3260 if (PyErr_Occurred())
3261 goto Error;
3262 Py_DECREF(it);
3263 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003264 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003265 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003266 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003267 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003268Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003269 for (; i > 0; i--, sp++)
3270 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003271 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003272 return 0;
3273}
3274
3275
Guido van Rossum96a42c81992-01-12 02:29:51 +00003276#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003277static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003278prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003279{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003280 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003281 if (PyObject_Print(v, stdout, 0) != 0)
3282 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003283 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003284 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003285}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003286#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003287
Fredrik Lundh7a830892006-05-27 10:39:48 +00003288static void
Fred Drake5755ce62001-06-27 19:19:46 +00003289call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003290{
Guido van Rossumb209a111997-04-29 18:18:01 +00003291 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003292 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003293 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003294 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003295 value = Py_None;
3296 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003297 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003298 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003299 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003300 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003301 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003302 }
Fred Drake5755ce62001-06-27 19:19:46 +00003303 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003304 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003305 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003306 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003307 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003308 Py_XDECREF(type);
3309 Py_XDECREF(value);
3310 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003311 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003312}
3313
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003314static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003315call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003316 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003317{
3318 PyObject *type, *value, *traceback;
3319 int err;
3320 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003321 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003322 if (err == 0)
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003323 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003324 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003325 return 0;
3326 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003327 else {
3328 Py_XDECREF(type);
3329 Py_XDECREF(value);
3330 Py_XDECREF(traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003331 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003332 }
3333}
3334
Fredrik Lundh7a830892006-05-27 10:39:48 +00003335static int
Fred Drake5755ce62001-06-27 19:19:46 +00003336call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3337 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003338{
Fred Drake5755ce62001-06-27 19:19:46 +00003339 register PyThreadState *tstate = frame->f_tstate;
3340 int result;
3341 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003342 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003343 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003344 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003345 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003346 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3347 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003348 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003349 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003350}
3351
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003352PyObject *
3353_PyEval_CallTracing(PyObject *func, PyObject *args)
3354{
3355 PyFrameObject *frame = PyEval_GetFrame();
3356 PyThreadState *tstate = frame->f_tstate;
3357 int save_tracing = tstate->tracing;
3358 int save_use_tracing = tstate->use_tracing;
3359 PyObject *result;
3360
3361 tstate->tracing = 0;
3362 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3363 || (tstate->c_profilefunc != NULL));
3364 result = PyObject_Call(func, args, NULL);
3365 tstate->tracing = save_tracing;
3366 tstate->use_tracing = save_use_tracing;
3367 return result;
3368}
3369
Fredrik Lundh7a830892006-05-27 10:39:48 +00003370static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003371maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003372 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3373 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003374{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003375 int result = 0;
3376
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003377 /* If the last instruction executed isn't in the current
3378 instruction window, reset the window. If the last
3379 instruction happens to fall at the start of a line or if it
3380 represents a jump backwards, call the trace function.
3381 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003382 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003383 int line;
3384 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003385
Thomas Woutersae406c62007-09-19 17:27:43 +00003386 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3387 &bounds);
3388 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003389 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003390 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003391 PyTrace_LINE, Py_None);
Thomas Woutersae406c62007-09-19 17:27:43 +00003392 }
3393 *instr_lb = bounds.ap_lower;
3394 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003395 }
Armin Rigobf57a142004-03-22 19:24:58 +00003396 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003397 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003398 }
3399 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003400 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003401}
3402
Fred Drake5755ce62001-06-27 19:19:46 +00003403void
3404PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003405{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003406 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003407 PyObject *temp = tstate->c_profileobj;
3408 Py_XINCREF(arg);
3409 tstate->c_profilefunc = NULL;
3410 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003411 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003412 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003413 Py_XDECREF(temp);
3414 tstate->c_profilefunc = func;
3415 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003416 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003417 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003418}
3419
3420void
3421PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3422{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003423 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003424 PyObject *temp = tstate->c_traceobj;
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +00003425 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003426 Py_XINCREF(arg);
3427 tstate->c_tracefunc = NULL;
3428 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003429 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003430 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003431 Py_XDECREF(temp);
3432 tstate->c_tracefunc = func;
3433 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003434 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003435 tstate->use_tracing = ((func != NULL)
3436 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003437}
3438
Guido van Rossumb209a111997-04-29 18:18:01 +00003439PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003440PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003441{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003442 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003443 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003444 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003445 else
3446 return current_frame->f_builtins;
3447}
3448
Guido van Rossumb209a111997-04-29 18:18:01 +00003449PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003450PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003451{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003452 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003453 if (current_frame == NULL)
3454 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003455 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003456 return current_frame->f_locals;
3457}
3458
Guido van Rossumb209a111997-04-29 18:18:01 +00003459PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003460PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003461{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003462 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003463 if (current_frame == NULL)
3464 return NULL;
3465 else
3466 return current_frame->f_globals;
3467}
3468
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003469PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003470PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003471{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003472 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003473 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003474}
3475
Guido van Rossum6135a871995-01-09 17:53:26 +00003476int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003477PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003478{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003479 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003480 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003481}
3482
Guido van Rossumbe270261997-05-22 22:26:18 +00003483int
Tim Peters5ba58662001-07-16 02:29:45 +00003484PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003485{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003486 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003487 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003488
3489 if (current_frame != NULL) {
3490 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003491 const int compilerflags = codeflags & PyCF_MASK;
3492 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003493 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003494 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003495 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003496#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003497 if (codeflags & CO_GENERATOR_ALLOWED) {
3498 result = 1;
3499 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3500 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003501#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003502 }
3503 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003504}
3505
3506int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003507Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003508{
Guido van Rossumb209a111997-04-29 18:18:01 +00003509 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003510 if (f == NULL)
3511 return 0;
3512 if (!PyFile_SoftSpace(f, 0))
3513 return 0;
3514 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003515}
3516
Guido van Rossum3f5da241990-12-20 15:06:42 +00003517
Guido van Rossum681d79a1995-07-18 14:51:37 +00003518/* External interface to call any callable object.
3519 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003520
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003521#undef PyEval_CallObject
3522/* for backward compatibility: export this interface */
3523
Guido van Rossumb209a111997-04-29 18:18:01 +00003524PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003525PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003526{
Guido van Rossumb209a111997-04-29 18:18:01 +00003527 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003528}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003529#define PyEval_CallObject(func,arg) \
3530 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003531
Guido van Rossumb209a111997-04-29 18:18:01 +00003532PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003533PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003534{
Jeremy Hylton52820442001-01-03 23:52:36 +00003535 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003536
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003537 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003538 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003539 if (arg == NULL)
3540 return NULL;
3541 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003542 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003543 PyErr_SetString(PyExc_TypeError,
3544 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003545 return NULL;
3546 }
3547 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003548 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003549
Guido van Rossumb209a111997-04-29 18:18:01 +00003550 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003551 PyErr_SetString(PyExc_TypeError,
3552 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003553 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003554 return NULL;
3555 }
3556
Tim Peters6d6c1a32001-08-02 04:15:00 +00003557 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003558 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003559 return result;
3560}
3561
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003562const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003563PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003564{
3565 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003566 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003567 else if (PyFunction_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003568 return PyString_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003569 else if (PyCFunction_Check(func))
3570 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3571 else if (PyClass_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003572 return PyString_AsString(((PyClassObject*)func)->cl_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003573 else if (PyInstance_Check(func)) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003574 return PyString_AsString(
Jeremy Hylton512a2372001-04-11 13:52:29 +00003575 ((PyInstanceObject*)func)->in_class->cl_name);
3576 } else {
3577 return func->ob_type->tp_name;
3578 }
3579}
3580
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003581const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003582PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003583{
3584 if (PyMethod_Check(func))
3585 return "()";
3586 else if (PyFunction_Check(func))
3587 return "()";
3588 else if (PyCFunction_Check(func))
3589 return "()";
3590 else if (PyClass_Check(func))
3591 return " constructor";
3592 else if (PyInstance_Check(func)) {
3593 return " instance";
3594 } else {
3595 return " object";
3596 }
3597}
3598
Fredrik Lundh7a830892006-05-27 10:39:48 +00003599static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003600err_args(PyObject *func, int flags, int nargs)
3601{
3602 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003603 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003604 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003605 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003606 nargs);
3607 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003608 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003609 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003610 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003611 nargs);
3612}
3613
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003614#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003615if (tstate->use_tracing && tstate->c_profilefunc) { \
3616 if (call_trace(tstate->c_profilefunc, \
3617 tstate->c_profileobj, \
3618 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003619 func)) { \
3620 x = NULL; \
3621 } \
3622 else { \
3623 x = call; \
3624 if (tstate->c_profilefunc != NULL) { \
3625 if (x == NULL) { \
3626 call_trace_protected(tstate->c_profilefunc, \
3627 tstate->c_profileobj, \
3628 tstate->frame, PyTrace_C_EXCEPTION, \
3629 func); \
3630 /* XXX should pass (type, value, tb) */ \
3631 } else { \
3632 if (call_trace(tstate->c_profilefunc, \
3633 tstate->c_profileobj, \
3634 tstate->frame, PyTrace_C_RETURN, \
3635 func)) { \
3636 Py_DECREF(x); \
3637 x = NULL; \
3638 } \
3639 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003640 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003641 } \
3642} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003643 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003644 }
3645
Fredrik Lundh7a830892006-05-27 10:39:48 +00003646static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003647call_function(PyObject ***pp_stack, int oparg
3648#ifdef WITH_TSC
3649 , uint64* pintr0, uint64* pintr1
3650#endif
3651 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003652{
3653 int na = oparg & 0xff;
3654 int nk = (oparg>>8) & 0xff;
3655 int n = na + 2 * nk;
3656 PyObject **pfunc = (*pp_stack) - n - 1;
3657 PyObject *func = *pfunc;
3658 PyObject *x, *w;
3659
Jeremy Hylton985eba52003-02-05 23:13:00 +00003660 /* Always dispatch PyCFunction first, because these are
3661 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003662 */
3663 if (PyCFunction_Check(func) && nk == 0) {
3664 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003665 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003666
3667 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003668 if (flags & (METH_NOARGS | METH_O)) {
3669 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3670 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003671 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003672 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003673 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003674 else if (flags & METH_O && na == 1) {
3675 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003676 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003677 Py_DECREF(arg);
3678 }
3679 else {
3680 err_args(func, flags, na);
3681 x = NULL;
3682 }
3683 }
3684 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003685 PyObject *callargs;
3686 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003687 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003688 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003689 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003690 Py_XDECREF(callargs);
3691 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003692 } else {
3693 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3694 /* optimize access to bound methods */
3695 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003696 PCALL(PCALL_METHOD);
3697 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003698 Py_INCREF(self);
3699 func = PyMethod_GET_FUNCTION(func);
3700 Py_INCREF(func);
3701 Py_DECREF(*pfunc);
3702 *pfunc = self;
3703 na++;
3704 n++;
3705 } else
3706 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003707 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003708 if (PyFunction_Check(func))
3709 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003710 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003711 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003712 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003713 Py_DECREF(func);
3714 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003715
Armin Rigod34fa522006-03-28 19:10:40 +00003716 /* Clear the stack of the function object. Also removes
3717 the arguments in case they weren't consumed already
3718 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003719 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003720 while ((*pp_stack) > pfunc) {
3721 w = EXT_POP(*pp_stack);
3722 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003723 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003724 }
3725 return x;
3726}
3727
Jeremy Hylton192690e2002-08-16 18:36:11 +00003728/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003729 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003730 For the simplest case -- a function that takes only positional
3731 arguments and is called with only positional arguments -- it
3732 inlines the most primitive frame setup code from
3733 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3734 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003735*/
3736
Fredrik Lundh7a830892006-05-27 10:39:48 +00003737static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003738fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003739{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003740 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003741 PyObject *globals = PyFunction_GET_GLOBALS(func);
3742 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3743 PyObject **d = NULL;
3744 int nd = 0;
3745
Jeremy Hylton985eba52003-02-05 23:13:00 +00003746 PCALL(PCALL_FUNCTION);
3747 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003748 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003749 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3750 PyFrameObject *f;
3751 PyObject *retval = NULL;
3752 PyThreadState *tstate = PyThreadState_GET();
3753 PyObject **fastlocals, **stack;
3754 int i;
3755
3756 PCALL(PCALL_FASTER_FUNCTION);
3757 assert(globals != NULL);
3758 /* XXX Perhaps we should create a specialized
3759 PyFrame_New() that doesn't take locals, but does
3760 take builtins without sanity checking them.
3761 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003762 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003763 f = PyFrame_New(tstate, co, globals, NULL);
3764 if (f == NULL)
3765 return NULL;
3766
3767 fastlocals = f->f_localsplus;
3768 stack = (*pp_stack) - n;
3769
3770 for (i = 0; i < n; i++) {
3771 Py_INCREF(*stack);
3772 fastlocals[i] = *stack++;
3773 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003774 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003775 ++tstate->recursion_depth;
3776 Py_DECREF(f);
3777 --tstate->recursion_depth;
3778 return retval;
3779 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003780 if (argdefs != NULL) {
3781 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimese93237d2007-12-19 02:37:44 +00003782 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003783 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003784 return PyEval_EvalCodeEx(co, globals,
3785 (PyObject *)NULL, (*pp_stack)-n, na,
3786 (*pp_stack)-2*nk, nk, d, nd,
3787 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003788}
3789
Fredrik Lundh7a830892006-05-27 10:39:48 +00003790static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003791update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3792 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003793{
3794 PyObject *kwdict = NULL;
3795 if (orig_kwdict == NULL)
3796 kwdict = PyDict_New();
3797 else {
3798 kwdict = PyDict_Copy(orig_kwdict);
3799 Py_DECREF(orig_kwdict);
3800 }
3801 if (kwdict == NULL)
3802 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003803 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003804 int err;
3805 PyObject *value = EXT_POP(*pp_stack);
3806 PyObject *key = EXT_POP(*pp_stack);
3807 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003808 PyErr_Format(PyExc_TypeError,
3809 "%.200s%s got multiple values "
3810 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003811 PyEval_GetFuncName(func),
3812 PyEval_GetFuncDesc(func),
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003813 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003814 Py_DECREF(key);
3815 Py_DECREF(value);
3816 Py_DECREF(kwdict);
3817 return NULL;
3818 }
3819 err = PyDict_SetItem(kwdict, key, value);
3820 Py_DECREF(key);
3821 Py_DECREF(value);
3822 if (err) {
3823 Py_DECREF(kwdict);
3824 return NULL;
3825 }
3826 }
3827 return kwdict;
3828}
3829
Fredrik Lundh7a830892006-05-27 10:39:48 +00003830static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003831update_star_args(int nstack, int nstar, PyObject *stararg,
3832 PyObject ***pp_stack)
3833{
3834 PyObject *callargs, *w;
3835
3836 callargs = PyTuple_New(nstack + nstar);
3837 if (callargs == NULL) {
3838 return NULL;
3839 }
3840 if (nstar) {
3841 int i;
3842 for (i = 0; i < nstar; i++) {
3843 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3844 Py_INCREF(a);
3845 PyTuple_SET_ITEM(callargs, nstack + i, a);
3846 }
3847 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003848 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003849 w = EXT_POP(*pp_stack);
3850 PyTuple_SET_ITEM(callargs, nstack, w);
3851 }
3852 return callargs;
3853}
3854
Fredrik Lundh7a830892006-05-27 10:39:48 +00003855static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003856load_args(PyObject ***pp_stack, int na)
3857{
3858 PyObject *args = PyTuple_New(na);
3859 PyObject *w;
3860
3861 if (args == NULL)
3862 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003863 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003864 w = EXT_POP(*pp_stack);
3865 PyTuple_SET_ITEM(args, na, w);
3866 }
3867 return args;
3868}
3869
Fredrik Lundh7a830892006-05-27 10:39:48 +00003870static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003871do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3872{
3873 PyObject *callargs = NULL;
3874 PyObject *kwdict = NULL;
3875 PyObject *result = NULL;
3876
3877 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003878 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003879 if (kwdict == NULL)
3880 goto call_fail;
3881 }
3882 callargs = load_args(pp_stack, na);
3883 if (callargs == NULL)
3884 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003885#ifdef CALL_PROFILE
3886 /* At this point, we have to look at the type of func to
3887 update the call stats properly. Do it here so as to avoid
3888 exposing the call stats machinery outside ceval.c
3889 */
3890 if (PyFunction_Check(func))
3891 PCALL(PCALL_FUNCTION);
3892 else if (PyMethod_Check(func))
3893 PCALL(PCALL_METHOD);
3894 else if (PyType_Check(func))
3895 PCALL(PCALL_TYPE);
3896 else
3897 PCALL(PCALL_OTHER);
3898#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003899 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003900 call_fail:
3901 Py_XDECREF(callargs);
3902 Py_XDECREF(kwdict);
3903 return result;
3904}
3905
Fredrik Lundh7a830892006-05-27 10:39:48 +00003906static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003907ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3908{
3909 int nstar = 0;
3910 PyObject *callargs = NULL;
3911 PyObject *stararg = NULL;
3912 PyObject *kwdict = NULL;
3913 PyObject *result = NULL;
3914
3915 if (flags & CALL_FLAG_KW) {
3916 kwdict = EXT_POP(*pp_stack);
Georg Brandl2134e752007-05-21 20:34:16 +00003917 if (!PyDict_Check(kwdict)) {
3918 PyObject *d;
3919 d = PyDict_New();
3920 if (d == NULL)
3921 goto ext_call_fail;
3922 if (PyDict_Update(d, kwdict) != 0) {
3923 Py_DECREF(d);
3924 /* PyDict_Update raises attribute
3925 * error (percolated from an attempt
3926 * to get 'keys' attribute) instead of
3927 * a type error if its second argument
3928 * is not a mapping.
3929 */
3930 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
3931 PyErr_Format(PyExc_TypeError,
3932 "%.200s%.200s argument after ** "
3933 "must be a mapping, not %.200s",
3934 PyEval_GetFuncName(func),
3935 PyEval_GetFuncDesc(func),
3936 kwdict->ob_type->tp_name);
3937 }
3938 goto ext_call_fail;
3939 }
3940 Py_DECREF(kwdict);
3941 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00003942 }
3943 }
3944 if (flags & CALL_FLAG_VAR) {
3945 stararg = EXT_POP(*pp_stack);
3946 if (!PyTuple_Check(stararg)) {
3947 PyObject *t = NULL;
3948 t = PySequence_Tuple(stararg);
3949 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003950 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3951 PyErr_Format(PyExc_TypeError,
Georg Brandl2134e752007-05-21 20:34:16 +00003952 "%.200s%.200s argument after * "
3953 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003954 PyEval_GetFuncName(func),
Georg Brandl2134e752007-05-21 20:34:16 +00003955 PyEval_GetFuncDesc(func),
3956 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003957 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003958 goto ext_call_fail;
3959 }
3960 Py_DECREF(stararg);
3961 stararg = t;
3962 }
3963 nstar = PyTuple_GET_SIZE(stararg);
3964 }
3965 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003966 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003967 if (kwdict == NULL)
3968 goto ext_call_fail;
3969 }
3970 callargs = update_star_args(na, nstar, stararg, pp_stack);
3971 if (callargs == NULL)
3972 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003973#ifdef CALL_PROFILE
3974 /* At this point, we have to look at the type of func to
3975 update the call stats properly. Do it here so as to avoid
3976 exposing the call stats machinery outside ceval.c
3977 */
3978 if (PyFunction_Check(func))
3979 PCALL(PCALL_FUNCTION);
3980 else if (PyMethod_Check(func))
3981 PCALL(PCALL_METHOD);
3982 else if (PyType_Check(func))
3983 PCALL(PCALL_TYPE);
3984 else
3985 PCALL(PCALL_OTHER);
3986#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003987 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersae406c62007-09-19 17:27:43 +00003988ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003989 Py_XDECREF(callargs);
3990 Py_XDECREF(kwdict);
3991 Py_XDECREF(stararg);
3992 return result;
3993}
3994
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003995/* Extract a slice index from a PyInt or PyLong or an object with the
3996 nb_index slot defined, and store in *pi.
3997 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3998 and silently boost values less than -PY_SSIZE_T_MAX-1 to -PY_SSIZE_T_MAX-1.
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003999 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004000*/
Tim Petersb5196382001-12-16 19:44:20 +00004001/* Note: If v is NULL, return success without storing into *pi. This
4002 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4003 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004004*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004005int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004006_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004007{
Tim Petersb5196382001-12-16 19:44:20 +00004008 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004009 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00004010 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004011 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
4012 however, it looks like it should be AsSsize_t.
4013 There should be a comment here explaining why.
4014 */
4015 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00004016 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004017 else if (PyIndex_Check(v)) {
4018 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004019 if (x == -1 && PyErr_Occurred())
4020 return 0;
4021 }
4022 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00004023 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004024 "slice indices must be integers or "
4025 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00004026 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004027 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00004028 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004029 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00004030 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004031}
4032
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004033#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004034#define ISINDEX(x) ((x) == NULL || \
4035 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00004036
Fredrik Lundh7a830892006-05-27 10:39:48 +00004037static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004038apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004039{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004040 PyTypeObject *tp = u->ob_type;
4041 PySequenceMethods *sq = tp->tp_as_sequence;
4042
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004043 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004044 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004045 if (!_PyEval_SliceIndex(v, &ilow))
4046 return NULL;
4047 if (!_PyEval_SliceIndex(w, &ihigh))
4048 return NULL;
4049 return PySequence_GetSlice(u, ilow, ihigh);
4050 }
4051 else {
4052 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00004053 if (slice != NULL) {
4054 PyObject *res = PyObject_GetItem(u, slice);
4055 Py_DECREF(slice);
4056 return res;
4057 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00004058 else
4059 return NULL;
4060 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004061}
Guido van Rossum3f5da241990-12-20 15:06:42 +00004062
Fredrik Lundh7a830892006-05-27 10:39:48 +00004063static int
Guido van Rossumac7be682001-01-17 15:42:30 +00004064assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
4065 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004066{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004067 PyTypeObject *tp = u->ob_type;
4068 PySequenceMethods *sq = tp->tp_as_sequence;
4069
Georg Brandl0fca97a2007-03-05 22:28:08 +00004070 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004071 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004072 if (!_PyEval_SliceIndex(v, &ilow))
4073 return -1;
4074 if (!_PyEval_SliceIndex(w, &ihigh))
4075 return -1;
4076 if (x == NULL)
4077 return PySequence_DelSlice(u, ilow, ihigh);
4078 else
4079 return PySequence_SetSlice(u, ilow, ihigh, x);
4080 }
4081 else {
4082 PyObject *slice = PySlice_New(v, w, NULL);
4083 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00004084 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004085 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00004086 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00004087 else
Guido van Rossum354797c2001-12-03 19:45:06 +00004088 res = PyObject_DelItem(u, slice);
4089 Py_DECREF(slice);
4090 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004091 }
4092 else
4093 return -1;
4094 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004095}
4096
Guido van Rossum04edb522008-03-18 02:49:46 +00004097#define Py3kExceptionClass_Check(x) \
4098 (PyType_Check((x)) && \
4099 PyType_FastSubclass((PyTypeObject*)(x), Py_TPFLAGS_BASE_EXC_SUBCLASS))
4100
4101#define CANNOT_CATCH_MSG "catching classes that don't inherit from " \
Georg Brandld5b635f2008-03-25 08:29:14 +00004102 "BaseException is not allowed in 3.x"
Guido van Rossum04edb522008-03-18 02:49:46 +00004103
Fredrik Lundh7a830892006-05-27 10:39:48 +00004104static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004105cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004106{
Guido van Rossumac7be682001-01-17 15:42:30 +00004107 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004108 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00004109 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00004110 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004111 break;
4112 case PyCmp_IS_NOT:
4113 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004114 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004115 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004116 res = PySequence_Contains(w, v);
4117 if (res < 0)
4118 return NULL;
4119 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004120 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00004121 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004122 if (res < 0)
4123 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004124 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004125 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004126 case PyCmp_EXC_MATCH:
Brett Cannon129bd522007-01-30 21:34:36 +00004127 if (PyTuple_Check(w)) {
4128 Py_ssize_t i, length;
4129 length = PyTuple_Size(w);
4130 for (i = 0; i < length; i += 1) {
4131 PyObject *exc = PyTuple_GET_ITEM(w, i);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004132 if (PyString_Check(exc)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004133 int ret_val;
4134 ret_val = PyErr_WarnEx(
Thomas Wouterse2176022007-09-20 17:35:10 +00004135 PyExc_DeprecationWarning,
4136 "catching of string "
4137 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004138 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004139 return NULL;
4140 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004141 else if (Py_Py3kWarningFlag &&
4142 !PyTuple_Check(exc) &&
4143 !Py3kExceptionClass_Check(exc))
Guido van Rossum04edb522008-03-18 02:49:46 +00004144 {
4145 int ret_val;
4146 ret_val = PyErr_WarnEx(
4147 PyExc_DeprecationWarning,
4148 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004149 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004150 return NULL;
4151 }
Brett Cannon129bd522007-01-30 21:34:36 +00004152 }
4153 }
4154 else {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004155 if (PyString_Check(w)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004156 int ret_val;
4157 ret_val = PyErr_WarnEx(
4158 PyExc_DeprecationWarning,
4159 "catching of string "
Thomas Wouterse2176022007-09-20 17:35:10 +00004160 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004161 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004162 return NULL;
4163 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004164 else if (Py_Py3kWarningFlag &&
4165 !PyTuple_Check(w) &&
4166 !Py3kExceptionClass_Check(w))
Guido van Rossum04edb522008-03-18 02:49:46 +00004167 {
4168 int ret_val;
4169 ret_val = PyErr_WarnEx(
4170 PyExc_DeprecationWarning,
4171 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004172 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004173 return NULL;
4174 }
Brett Cannon129bd522007-01-30 21:34:36 +00004175 }
Barry Warsaw4249f541997-08-22 21:26:19 +00004176 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004177 break;
4178 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00004179 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004180 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004181 v = res ? Py_True : Py_False;
4182 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004183 return v;
4184}
4185
Fredrik Lundh7a830892006-05-27 10:39:48 +00004186static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00004187import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004188{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004189 PyObject *x;
4190
4191 x = PyObject_GetAttr(v, name);
4192 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004193 PyErr_Format(PyExc_ImportError,
4194 "cannot import name %.230s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004195 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004196 }
Thomas Wouters52152252000-08-17 22:55:00 +00004197 return x;
4198}
Guido van Rossumac7be682001-01-17 15:42:30 +00004199
Fredrik Lundh7a830892006-05-27 10:39:48 +00004200static int
Thomas Wouters52152252000-08-17 22:55:00 +00004201import_all_from(PyObject *locals, PyObject *v)
4202{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004203 PyObject *all = PyObject_GetAttrString(v, "__all__");
4204 PyObject *dict, *name, *value;
4205 int skip_leading_underscores = 0;
4206 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004207
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004208 if (all == NULL) {
4209 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4210 return -1; /* Unexpected error */
4211 PyErr_Clear();
4212 dict = PyObject_GetAttrString(v, "__dict__");
4213 if (dict == NULL) {
4214 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4215 return -1;
4216 PyErr_SetString(PyExc_ImportError,
4217 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004218 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004219 }
4220 all = PyMapping_Keys(dict);
4221 Py_DECREF(dict);
4222 if (all == NULL)
4223 return -1;
4224 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004225 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004226
4227 for (pos = 0, err = 0; ; pos++) {
4228 name = PySequence_GetItem(all, pos);
4229 if (name == NULL) {
4230 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4231 err = -1;
4232 else
4233 PyErr_Clear();
4234 break;
4235 }
4236 if (skip_leading_underscores &&
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004237 PyString_Check(name) &&
4238 PyString_AS_STRING(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004239 {
4240 Py_DECREF(name);
4241 continue;
4242 }
4243 value = PyObject_GetAttr(v, name);
4244 if (value == NULL)
4245 err = -1;
Armin Rigo70370852006-11-29 21:59:22 +00004246 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004247 err = PyDict_SetItem(locals, name, value);
Armin Rigo70370852006-11-29 21:59:22 +00004248 else
4249 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004250 Py_DECREF(name);
4251 Py_XDECREF(value);
4252 if (err != 0)
4253 break;
4254 }
4255 Py_DECREF(all);
4256 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004257}
4258
Fredrik Lundh7a830892006-05-27 10:39:48 +00004259static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004260build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004261{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004262 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004263
4264 if (PyDict_Check(methods))
4265 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004266 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004267 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004268 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4269 base = PyTuple_GET_ITEM(bases, 0);
4270 metaclass = PyObject_GetAttrString(base, "__class__");
4271 if (metaclass == NULL) {
4272 PyErr_Clear();
4273 metaclass = (PyObject *)base->ob_type;
4274 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004275 }
4276 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004277 else {
4278 PyObject *g = PyEval_GetGlobals();
4279 if (g != NULL && PyDict_Check(g))
4280 metaclass = PyDict_GetItemString(g, "__metaclass__");
4281 if (metaclass == NULL)
4282 metaclass = (PyObject *) &PyClass_Type;
4283 Py_INCREF(metaclass);
4284 }
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004285 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods,
Thomas Woutersae406c62007-09-19 17:27:43 +00004286 NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004287 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004288 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004289 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004290 in a base that was not a class (such the random module
4291 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004292 by augmenting the error message with more information.*/
4293
4294 PyObject *ptype, *pvalue, *ptraceback;
4295
4296 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004297 if (PyString_Check(pvalue)) {
Raymond Hettingercfc31922004-09-16 16:41:57 +00004298 PyObject *newmsg;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004299 newmsg = PyString_FromFormat(
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004300 "Error when calling the metaclass bases\n"
Thomas Woutersae406c62007-09-19 17:27:43 +00004301 " %s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004302 PyString_AS_STRING(pvalue));
Raymond Hettingercfc31922004-09-16 16:41:57 +00004303 if (newmsg != NULL) {
4304 Py_DECREF(pvalue);
4305 pvalue = newmsg;
4306 }
4307 }
4308 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004309 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004310 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004311}
4312
Fredrik Lundh7a830892006-05-27 10:39:48 +00004313static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004314exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4315 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004316{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004317 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004318 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004319 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004320
Guido van Rossumb209a111997-04-29 18:18:01 +00004321 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4322 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004323 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004324 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004325 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004326 locals = PyTuple_GetItem(prog, 2);
4327 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004328 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004329 if (globals == Py_None) {
4330 globals = PyEval_GetGlobals();
4331 if (locals == Py_None) {
4332 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004333 plain = 1;
4334 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004335 if (!globals || !locals) {
4336 PyErr_SetString(PyExc_SystemError,
4337 "globals and locals cannot be NULL");
4338 return -1;
4339 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004340 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004341 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004342 locals = globals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004343 if (!PyString_Check(prog) &&
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004344 !PyUnicode_Check(prog) &&
Guido van Rossumb209a111997-04-29 18:18:01 +00004345 !PyCode_Check(prog) &&
4346 !PyFile_Check(prog)) {
4347 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004348 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004349 return -1;
4350 }
Fred Drake661ea262000-10-24 19:57:45 +00004351 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004352 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004353 "exec: arg 2 must be a dictionary or None");
4354 return -1;
4355 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004356 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004357 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004358 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004359 return -1;
4360 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004361 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004362 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004363 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004364 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4365 PyErr_SetString(PyExc_TypeError,
4366 "code object passed to exec may not contain free variables");
4367 return -1;
4368 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004369 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004370 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004371 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004372 FILE *fp = PyFile_AsFile(prog);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004373 char *name = PyString_AsString(PyFile_Name(prog));
Jeremy Hylton714b1122007-02-25 16:01:58 +00004374 PyCompilerFlags cf;
Thomas Woutersae406c62007-09-19 17:27:43 +00004375 if (name == NULL)
4376 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004377 cf.cf_flags = 0;
4378 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004379 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004380 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004381 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004382 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004383 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004384 }
4385 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004386 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004387 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004388 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004389 cf.cf_flags = 0;
4390#ifdef Py_USING_UNICODE
4391 if (PyUnicode_Check(prog)) {
4392 tmp = PyUnicode_AsUTF8String(prog);
4393 if (tmp == NULL)
4394 return -1;
4395 prog = tmp;
4396 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4397 }
4398#endif
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004399 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004400 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004401 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004402 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004403 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004404 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004405 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004406 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004407 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004408 if (plain)
4409 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004410 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004411 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004412 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004413 return 0;
4414}
Guido van Rossum24c13741995-02-14 09:42:43 +00004415
Fredrik Lundh7a830892006-05-27 10:39:48 +00004416static void
Paul Prescode68140d2000-08-30 20:25:01 +00004417format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4418{
4419 char *obj_str;
4420
4421 if (!obj)
4422 return;
4423
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004424 obj_str = PyString_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00004425 if (!obj_str)
4426 return;
4427
4428 PyErr_Format(exc, format_str, obj_str);
4429}
Guido van Rossum950361c1997-01-24 13:49:28 +00004430
Fredrik Lundh7a830892006-05-27 10:39:48 +00004431static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004432string_concatenate(PyObject *v, PyObject *w,
4433 PyFrameObject *f, unsigned char *next_instr)
4434{
4435 /* This function implements 'variable += expr' when both arguments
4436 are strings. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004437 Py_ssize_t v_len = PyString_GET_SIZE(v);
4438 Py_ssize_t w_len = PyString_GET_SIZE(w);
Armin Rigo97ff0472006-08-09 15:37:26 +00004439 Py_ssize_t new_len = v_len + w_len;
4440 if (new_len < 0) {
4441 PyErr_SetString(PyExc_OverflowError,
4442 "strings are too large to concat");
4443 return NULL;
4444 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004445
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004446 if (v->ob_refcnt == 2) {
4447 /* In the common case, there are 2 references to the value
4448 * stored in 'variable' when the += is performed: one on the
Thomas Wouterse2176022007-09-20 17:35:10 +00004449 * value stack (in 'v') and one still stored in the
4450 * 'variable'. We try to delete the variable now to reduce
4451 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004452 */
4453 switch (*next_instr) {
4454 case STORE_FAST:
4455 {
4456 int oparg = PEEKARG();
4457 PyObject **fastlocals = f->f_localsplus;
4458 if (GETLOCAL(oparg) == v)
4459 SETLOCAL(oparg, NULL);
4460 break;
4461 }
4462 case STORE_DEREF:
4463 {
Thomas Wouterse2176022007-09-20 17:35:10 +00004464 PyObject **freevars = (f->f_localsplus +
4465 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004466 PyObject *c = freevars[PEEKARG()];
4467 if (PyCell_GET(c) == v)
4468 PyCell_Set(c, NULL);
4469 break;
4470 }
4471 case STORE_NAME:
4472 {
4473 PyObject *names = f->f_code->co_names;
4474 PyObject *name = GETITEM(names, PEEKARG());
4475 PyObject *locals = f->f_locals;
4476 if (PyDict_CheckExact(locals) &&
4477 PyDict_GetItem(locals, name) == v) {
4478 if (PyDict_DelItem(locals, name) != 0) {
4479 PyErr_Clear();
4480 }
4481 }
4482 break;
4483 }
4484 }
4485 }
4486
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004487 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004488 /* Now we own the last reference to 'v', so we can resize it
4489 * in-place.
4490 */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004491 if (_PyString_Resize(&v, new_len) != 0) {
4492 /* XXX if _PyString_Resize() fails, 'v' has been
Thomas Wouterse2176022007-09-20 17:35:10 +00004493 * deallocated so it cannot be put back into
4494 * 'variable'. The MemoryError is raised when there
4495 * is no value in 'variable', which might (very
4496 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004497 */
4498 return NULL;
4499 }
4500 /* copy 'w' into the newly allocated area of 'v' */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004501 memcpy(PyString_AS_STRING(v) + v_len,
4502 PyString_AS_STRING(w), w_len);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004503 return v;
4504 }
4505 else {
4506 /* When in-place resizing is not an option. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004507 PyString_Concat(&v, w);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004508 return v;
4509 }
4510}
4511
Guido van Rossum950361c1997-01-24 13:49:28 +00004512#ifdef DYNAMIC_EXECUTION_PROFILE
4513
Fredrik Lundh7a830892006-05-27 10:39:48 +00004514static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004515getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004516{
4517 int i;
4518 PyObject *l = PyList_New(256);
4519 if (l == NULL) return NULL;
4520 for (i = 0; i < 256; i++) {
4521 PyObject *x = PyInt_FromLong(a[i]);
4522 if (x == NULL) {
4523 Py_DECREF(l);
4524 return NULL;
4525 }
4526 PyList_SetItem(l, i, x);
4527 }
4528 for (i = 0; i < 256; i++)
4529 a[i] = 0;
4530 return l;
4531}
4532
4533PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004534_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004535{
4536#ifndef DXPAIRS
4537 return getarray(dxp);
4538#else
4539 int i;
4540 PyObject *l = PyList_New(257);
4541 if (l == NULL) return NULL;
4542 for (i = 0; i < 257; i++) {
4543 PyObject *x = getarray(dxpairs[i]);
4544 if (x == NULL) {
4545 Py_DECREF(l);
4546 return NULL;
4547 }
4548 PyList_SetItem(l, i, x);
4549 }
4550 return l;
4551#endif
4552}
4553
4554#endif