blob: 037c86c134a40e56a205939bfb134645bc8c40e0 [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{
277 if (!interpreter_lock)
278 return;
279 /*XXX Can't use PyThread_free_lock here because it does too
280 much error-checking. Doing this cleanly would require
281 adding a new function to each thread_*.h. Instead, just
282 create a new lock and waste a little bit of memory */
283 interpreter_lock = PyThread_allocate_lock();
284 PyThread_acquire_lock(interpreter_lock, 1);
285 main_thread = PyThread_get_thread_ident();
286}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000287#endif
288
Guido van Rossumff4949e1992-08-05 19:58:53 +0000289/* Functions save_thread and restore_thread are always defined so
290 dynamically loaded modules needn't be compiled separately for use
291 with and without threads: */
292
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000293PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000294PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000295{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000296 PyThreadState *tstate = PyThreadState_Swap(NULL);
297 if (tstate == NULL)
298 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000299#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000300 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000301 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000302#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000303 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000304}
305
306void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000307PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000308{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000309 if (tstate == NULL)
310 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000311#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000312 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000313 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000314 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000315 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000316 }
317#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000318 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000319}
320
321
Guido van Rossuma9672091994-09-14 13:31:22 +0000322/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
323 signal handlers or Mac I/O completion routines) can schedule calls
324 to a function to be called synchronously.
325 The synchronous function is called with one void* argument.
326 It should return 0 for success or -1 for failure -- failure should
327 be accompanied by an exception.
328
329 If registry succeeds, the registry function returns 0; if it fails
330 (e.g. due to too many pending calls) it returns -1 (without setting
331 an exception condition).
332
333 Note that because registry may occur from within signal handlers,
334 or other asynchronous events, calling malloc() is unsafe!
335
336#ifdef WITH_THREAD
337 Any thread can schedule pending calls, but only the main thread
338 will execute them.
339#endif
340
341 XXX WARNING! ASYNCHRONOUSLY EXECUTING CODE!
342 There are two possible race conditions:
343 (1) nested asynchronous registry calls;
344 (2) registry calls made while pending calls are being processed.
345 While (1) is very unlikely, (2) is a real possibility.
346 The current code is safe against (2), but not against (1).
347 The safety against (2) is derived from the fact that only one
348 thread (the main thread) ever takes things out of the queue.
Guido van Rossuma9672091994-09-14 13:31:22 +0000349
Guido van Rossuma027efa1997-05-05 20:56:21 +0000350 XXX Darn! With the advent of thread state, we should have an array
351 of pending calls per thread in the thread state! Later...
352*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000353
Guido van Rossuma9672091994-09-14 13:31:22 +0000354#define NPENDINGCALLS 32
355static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000356 int (*func)(void *);
357 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000358} pendingcalls[NPENDINGCALLS];
359static volatile int pendingfirst = 0;
360static volatile int pendinglast = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000361static volatile int things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000362
363int
Thomas Wouters334fb892000-07-25 12:56:38 +0000364Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000365{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000366 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000367 int i, j;
368 /* XXX Begin critical section */
369 /* XXX If you want this to be safe against nested
370 XXX asynchronous calls, you'll have to work harder! */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000371 if (busy)
372 return -1;
373 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000374 i = pendinglast;
375 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000376 if (j == pendingfirst) {
377 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000378 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000379 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000380 pendingcalls[i].func = func;
381 pendingcalls[i].arg = arg;
382 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000383
384 _Py_Ticker = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000385 things_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000386 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000387 /* XXX End critical section */
388 return 0;
389}
390
Guido van Rossum180d7b41994-09-29 09:45:57 +0000391int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000392Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000393{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000394 static int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000395#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000396 if (main_thread && PyThread_get_thread_ident() != main_thread)
Guido van Rossuma9672091994-09-14 13:31:22 +0000397 return 0;
398#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000399 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000400 return 0;
401 busy = 1;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000402 things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000403 for (;;) {
404 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000405 int (*func)(void *);
406 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000407 i = pendingfirst;
408 if (i == pendinglast)
409 break; /* Queue empty */
410 func = pendingcalls[i].func;
411 arg = pendingcalls[i].arg;
412 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000413 if (func(arg) < 0) {
414 busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000415 things_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000416 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000417 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000418 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000419 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000420 return 0;
421}
422
423
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000424/* The interpreter's recursion limit */
425
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000426#ifndef Py_DEFAULT_RECURSION_LIMIT
427#define Py_DEFAULT_RECURSION_LIMIT 1000
428#endif
429static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
430int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000431
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000432int
433Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000434{
435 return recursion_limit;
436}
437
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000438void
439Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000440{
441 recursion_limit = new_limit;
Thomas Woutersae406c62007-09-19 17:27:43 +0000442 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000443}
444
Armin Rigo2b3eb402003-10-28 12:05:48 +0000445/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
446 if the recursion_depth reaches _Py_CheckRecursionLimit.
447 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
448 to guarantee that _Py_CheckRecursiveCall() is regularly called.
449 Without USE_STACKCHECK, there is no need for this. */
450int
451_Py_CheckRecursiveCall(char *where)
452{
453 PyThreadState *tstate = PyThreadState_GET();
454
455#ifdef USE_STACKCHECK
456 if (PyOS_CheckStack()) {
457 --tstate->recursion_depth;
458 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
459 return -1;
460 }
461#endif
462 if (tstate->recursion_depth > recursion_limit) {
463 --tstate->recursion_depth;
464 PyErr_Format(PyExc_RuntimeError,
465 "maximum recursion depth exceeded%s",
466 where);
467 return -1;
468 }
Thomas Woutersae406c62007-09-19 17:27:43 +0000469 _Py_CheckRecursionLimit = recursion_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000470 return 0;
471}
472
Guido van Rossum374a9221991-04-04 10:40:29 +0000473/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000474enum why_code {
475 WHY_NOT = 0x0001, /* No error */
476 WHY_EXCEPTION = 0x0002, /* Exception occurred */
477 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
478 WHY_RETURN = 0x0008, /* 'return' statement */
479 WHY_BREAK = 0x0010, /* 'break' statement */
480 WHY_CONTINUE = 0x0020, /* 'continue' statement */
481 WHY_YIELD = 0x0040 /* 'yield' operator */
482};
Guido van Rossum374a9221991-04-04 10:40:29 +0000483
Fredrik Lundh7a830892006-05-27 10:39:48 +0000484static enum why_code do_raise(PyObject *, PyObject *, PyObject *);
485static int unpack_iterable(PyObject *, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000486
Skip Montanarod581d772002-09-03 20:10:45 +0000487/* for manipulating the thread switch and periodic "stuff" - used to be
488 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000489int _Py_CheckInterval = 100;
490volatile int _Py_Ticker = 100;
Guido van Rossum374a9221991-04-04 10:40:29 +0000491
Guido van Rossumb209a111997-04-29 18:18:01 +0000492PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000493PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000494{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000495 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000496 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000497 (PyObject **)NULL, 0,
498 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000499 (PyObject **)NULL, 0,
500 NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000501}
502
503
504/* Interpreter main loop */
505
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000506PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000507PyEval_EvalFrame(PyFrameObject *f) {
508 /* This is for backward compatibility with extension modules that
Thomas Wouterse2176022007-09-20 17:35:10 +0000509 used this API; core interpreter code should call
510 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000511 return PyEval_EvalFrameEx(f, 0);
512}
513
514PyObject *
Anthony Baxtera863d332006-04-11 07:43:46 +0000515PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000516{
Guido van Rossum950361c1997-01-24 13:49:28 +0000517#ifdef DXPAIRS
518 int lastopcode = 0;
519#endif
Thomas Wouterse2176022007-09-20 17:35:10 +0000520 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000521 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000522 register int opcode; /* Current opcode */
523 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000524 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000525 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000526 register PyObject *x; /* Result object -- NULL if error */
527 register PyObject *v; /* Temporary objects popped off stack */
528 register PyObject *w;
529 register PyObject *u;
530 register PyObject *t;
Barry Warsaw23c9ec82000-08-21 15:44:01 +0000531 register PyObject *stream = NULL; /* for PRINT opcodes */
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000532 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000533 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000534 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000535 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000536
Tim Peters8a5c3c72004-04-05 19:36:21 +0000537 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000538
539 not (instr_lb <= current_bytecode_offset < instr_ub)
540
Tim Peters8a5c3c72004-04-05 19:36:21 +0000541 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000542 initial values are such as to make this false the first
543 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000544 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000545
Guido van Rossumd076c731998-10-07 19:42:25 +0000546 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000547 PyObject *names;
548 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000549#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000550 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000551 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000552#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000553
Neal Norwitza81d2202002-07-14 00:27:26 +0000554/* Tuple access macros */
555
556#ifndef Py_DEBUG
557#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
558#else
559#define GETITEM(v, i) PyTuple_GetItem((v), (i))
560#endif
561
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000562#ifdef WITH_TSC
563/* Use Pentium timestamp counter to mark certain events:
564 inst0 -- beginning of switch statement for opcode dispatch
565 inst1 -- end of switch statement (may be skipped)
566 loop0 -- the top of the mainloop
Tim Peters7df5e7f2006-05-26 23:14:37 +0000567 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000568 (may be skipped)
569 intr1 -- beginning of long interruption
570 intr2 -- end of long interruption
571
572 Many opcodes call out to helper C functions. In some cases, the
573 time in those functions should be counted towards the time for the
574 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
575 calls another Python function; there's no point in charge all the
576 bytecode executed by the called function to the caller.
577
578 It's hard to make a useful judgement statically. In the presence
579 of operator overloading, it's impossible to tell if a call will
580 execute new Python code or not.
581
582 It's a case-by-case judgement. I'll use intr1 for the following
583 cases:
584
585 EXEC_STMT
586 IMPORT_STAR
587 IMPORT_FROM
588 CALL_FUNCTION (and friends)
589
590 */
591 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
592 int ticked = 0;
593
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000594 READ_TIMESTAMP(inst0);
595 READ_TIMESTAMP(inst1);
596 READ_TIMESTAMP(loop0);
597 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000598
599 /* shut up the compiler */
600 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000601#endif
602
Guido van Rossum374a9221991-04-04 10:40:29 +0000603/* Code access macros */
604
Martin v. Löwis18e16552006-02-15 17:27:45 +0000605#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000606#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000607#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000608#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000609#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000610#define JUMPBY(x) (next_instr += (x))
611
Raymond Hettingerf606f872003-03-16 03:11:04 +0000612/* OpCode prediction macros
Thomas Wouterse2176022007-09-20 17:35:10 +0000613 Some opcodes tend to come in pairs thus making it possible to
614 predict the second code when the first is run. For example,
615 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
616 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000617
618 Verifying the prediction costs a single high-speed test of register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000619 variable against a constant. If the pairing was good, then the
Raymond Hettingerf606f872003-03-16 03:11:04 +0000620 processor has a high likelihood of making its own successful branch
621 prediction which results in a nearly zero overhead transition to the
622 next opcode.
623
624 A successful prediction saves a trip through the eval-loop including
Georg Brandl2d3953b2007-12-05 07:02:47 +0000625 its two unpredictable branches, the HAS_ARG test and the switch-case.
Raymond Hettingera7216982004-02-08 19:59:27 +0000626
Tim Peters8a5c3c72004-04-05 19:36:21 +0000627 If collecting opcode statistics, turn off prediction so that
628 statistics are accurately maintained (the predictions bypass
Raymond Hettingera7216982004-02-08 19:59:27 +0000629 the opcode frequency counter updates).
Raymond Hettingerf606f872003-03-16 03:11:04 +0000630*/
631
Raymond Hettingera7216982004-02-08 19:59:27 +0000632#ifdef DYNAMIC_EXECUTION_PROFILE
633#define PREDICT(op) if (0) goto PRED_##op
634#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000635#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000636#endif
637
Raymond Hettingerf606f872003-03-16 03:11:04 +0000638#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000639#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000640
Guido van Rossum374a9221991-04-04 10:40:29 +0000641/* Stack manipulation macros */
642
Martin v. Löwis18e16552006-02-15 17:27:45 +0000643/* The stack can grow at most MAXINT deep, as co_nlocals and
644 co_stacksize are ints. */
645#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000646#define EMPTY() (STACK_LEVEL() == 0)
647#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000648#define SECOND() (stack_pointer[-2])
649#define THIRD() (stack_pointer[-3])
650#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000651#define SET_TOP(v) (stack_pointer[-1] = (v))
652#define SET_SECOND(v) (stack_pointer[-2] = (v))
653#define SET_THIRD(v) (stack_pointer[-3] = (v))
654#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000655#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000656#define BASIC_PUSH(v) (*stack_pointer++ = (v))
657#define BASIC_POP() (*--stack_pointer)
658
Guido van Rossum96a42c81992-01-12 02:29:51 +0000659#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000660#define PUSH(v) { (void)(BASIC_PUSH(v), \
661 lltrace && prtrace(TOP(), "push")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000662 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouterse2176022007-09-20 17:35:10 +0000663#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
664 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000665#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
666 lltrace && prtrace(TOP(), "stackadj")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000667 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes52729ac2007-12-14 02:33:57 +0000668#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
669 prtrace((STACK_POINTER)[-1], "ext_pop")), \
670 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000671#else
672#define PUSH(v) BASIC_PUSH(v)
673#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000674#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000675#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000676#endif
677
Guido van Rossum681d79a1995-07-18 14:51:37 +0000678/* Local variable macros */
679
680#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000681
682/* The SETLOCAL() macro must not DECREF the local variable in-place and
683 then store the new value; it must copy the old value to a temporary
684 value, then store the new value, and then DECREF the temporary value.
685 This is because it is possible that during the DECREF the frame is
686 accessed by other code (e.g. a __del__ method or gc.collect()) and the
687 variable would be pointing to already-freed memory. */
688#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
689 GETLOCAL(i) = value; \
690 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000691
Guido van Rossuma027efa1997-05-05 20:56:21 +0000692/* Start of code */
693
Tim Peters5ca576e2001-06-18 22:08:13 +0000694 if (f == NULL)
695 return NULL;
696
Armin Rigo1d313ab2003-10-25 14:33:09 +0000697 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000698 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000699 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000700
Tim Peters5ca576e2001-06-18 22:08:13 +0000701 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000702
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000703 if (tstate->use_tracing) {
704 if (tstate->c_tracefunc != NULL) {
705 /* tstate->c_tracefunc, if defined, is a
706 function that will be called on *every* entry
707 to a code block. Its return value, if not
708 None, is a function that will be called at
709 the start of each executed line of code.
710 (Actually, the function must return itself
711 in order to continue tracing.) The trace
712 functions are called with three arguments:
713 a pointer to the current frame, a string
714 indicating why the function is called, and
715 an argument which depends on the situation.
716 The global trace function is also called
717 whenever an exception is detected. */
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000718 if (call_trace_protected(tstate->c_tracefunc,
719 tstate->c_traceobj,
720 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000721 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000722 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000723 }
724 }
725 if (tstate->c_profilefunc != NULL) {
726 /* Similar for c_profilefunc, except it needn't
727 return itself and isn't called for "line" events */
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000728 if (call_trace_protected(tstate->c_profilefunc,
729 tstate->c_profileobj,
730 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000731 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000732 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000733 }
734 }
735 }
736
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000737 co = f->f_code;
738 names = co->co_names;
739 consts = co->co_consts;
740 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000741 freevars = f->f_localsplus + co->co_nlocals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000742 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000743 /* An explanation is in order for the next line.
744
745 f->f_lasti now refers to the index of the last instruction
746 executed. You might think this was obvious from the name, but
747 this wasn't always true before 2.3! PyFrame_New now sets
748 f->f_lasti to -1 (i.e. the index *before* the first instruction)
749 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000750 does work. Promise.
751
752 When the PREDICT() macros are enabled, some opcode pairs follow in
753 direct succession without updating f->f_lasti. A successful
754 prediction effectively links the two codes together as if they
755 were a single new opcode; accordingly,f->f_lasti will point to
756 the first code in the pair (for instance, GET_ITER followed by
757 FOR_ITER is effectively a single opcode and f->f_lasti will point
758 at to the beginning of the combined pair.)
759 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000760 next_instr = first_instr + f->f_lasti + 1;
761 stack_pointer = f->f_stacktop;
762 assert(stack_pointer != NULL);
763 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
764
Tim Peters5ca576e2001-06-18 22:08:13 +0000765#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000766 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000767#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000768#if defined(Py_DEBUG) || defined(LLTRACE)
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000769 filename = PyString_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000770#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000771
Guido van Rossum374a9221991-04-04 10:40:29 +0000772 why = WHY_NOT;
773 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000774 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000775 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000776
Anthony Baxtera863d332006-04-11 07:43:46 +0000777 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000778 why = WHY_EXCEPTION;
779 goto on_error;
780 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000781
Guido van Rossum374a9221991-04-04 10:40:29 +0000782 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000783#ifdef WITH_TSC
784 if (inst1 == 0) {
785 /* Almost surely, the opcode executed a break
786 or a continue, preventing inst1 from being set
787 on the way out of the loop.
788 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000789 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000790 loop1 = inst1;
791 }
792 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
793 intr0, intr1);
794 ticked = 0;
795 inst1 = 0;
796 intr0 = 0;
797 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000798 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000799#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000800 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000801 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000802
Guido van Rossuma027efa1997-05-05 20:56:21 +0000803 /* Do periodic things. Doing this every time through
804 the loop would add too much overhead, so we do it
805 only every Nth instruction. We also do it if
806 ``things_to_do'' is set, i.e. when an asynchronous
807 event needs attention (e.g. a signal handler or
808 async I/O handler); see Py_AddPendingCall() and
809 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000810
Skip Montanarod581d772002-09-03 20:10:45 +0000811 if (--_Py_Ticker < 0) {
Thomas Woutersae406c62007-09-19 17:27:43 +0000812 if (*next_instr == SETUP_FINALLY) {
813 /* Make the last opcode before
814 a try: finally: block uninterruptable. */
815 goto fast_next_opcode;
816 }
Skip Montanarod581d772002-09-03 20:10:45 +0000817 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000818 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000819#ifdef WITH_TSC
820 ticked = 1;
821#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000822 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000823 if (Py_MakePendingCalls() < 0) {
824 why = WHY_EXCEPTION;
825 goto on_error;
826 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000827 if (things_to_do)
828 /* MakePendingCalls() didn't succeed.
829 Force early re-execution of this
830 "periodic" code, possibly after
831 a thread switch */
832 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000833 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000834#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000835 if (interpreter_lock) {
836 /* Give another thread a chance */
837
Guido van Rossum25ce5661997-08-02 03:10:38 +0000838 if (PyThreadState_Swap(NULL) != tstate)
839 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000840 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000841
842 /* Other threads may run now */
843
Guido van Rossum65d5b571998-12-21 19:32:43 +0000844 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000845 if (PyThreadState_Swap(tstate) != NULL)
846 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000847
848 /* Check for thread interrupts */
849
850 if (tstate->async_exc != NULL) {
851 x = tstate->async_exc;
852 tstate->async_exc = NULL;
853 PyErr_SetNone(x);
854 Py_DECREF(x);
855 why = WHY_EXCEPTION;
856 goto on_error;
857 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000858 }
859#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000860 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000861
Neil Schemenauer63543862002-02-17 19:10:14 +0000862 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000863 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000864
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000865 /* line-by-line tracing support */
866
867 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
868 /* see maybe_call_line_trace
869 for expository comments */
870 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000871
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000872 err = maybe_call_line_trace(tstate->c_tracefunc,
873 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000874 f, &instr_lb, &instr_ub,
875 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000876 /* Reload possibly changed frame fields */
877 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000878 if (f->f_stacktop != NULL) {
879 stack_pointer = f->f_stacktop;
880 f->f_stacktop = NULL;
881 }
882 if (err) {
883 /* trace function raised an exception */
884 goto on_error;
885 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000886 }
887
888 /* Extract opcode and argument */
889
Guido van Rossum374a9221991-04-04 10:40:29 +0000890 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000891 oparg = 0; /* allows oparg to be stored in a register because
892 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000893 if (HAS_ARG(opcode))
894 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000895 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000896#ifdef DYNAMIC_EXECUTION_PROFILE
897#ifdef DXPAIRS
898 dxpairs[lastopcode][opcode]++;
899 lastopcode = opcode;
900#endif
901 dxp[opcode]++;
902#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000903
Guido van Rossum96a42c81992-01-12 02:29:51 +0000904#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000905 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000906
Guido van Rossum96a42c81992-01-12 02:29:51 +0000907 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000908 if (HAS_ARG(opcode)) {
909 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000910 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000911 }
912 else {
913 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000914 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000915 }
916 }
917#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000918
Guido van Rossum374a9221991-04-04 10:40:29 +0000919 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000920 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000921
Guido van Rossum374a9221991-04-04 10:40:29 +0000922 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000923
Guido van Rossum374a9221991-04-04 10:40:29 +0000924 /* BEWARE!
925 It is essential that any operation that fails sets either
926 x to NULL, err to nonzero, or why to anything but WHY_NOT,
927 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000928
Guido van Rossum374a9221991-04-04 10:40:29 +0000929 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000930
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000931 case NOP:
932 goto fast_next_opcode;
933
Neil Schemenauer63543862002-02-17 19:10:14 +0000934 case LOAD_FAST:
935 x = GETLOCAL(oparg);
936 if (x != NULL) {
937 Py_INCREF(x);
938 PUSH(x);
939 goto fast_next_opcode;
940 }
941 format_exc_check_arg(PyExc_UnboundLocalError,
942 UNBOUNDLOCAL_ERROR_MSG,
943 PyTuple_GetItem(co->co_varnames, oparg));
944 break;
945
946 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000947 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000948 Py_INCREF(x);
949 PUSH(x);
950 goto fast_next_opcode;
951
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000952 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000953 case STORE_FAST:
954 v = POP();
955 SETLOCAL(oparg, v);
956 goto fast_next_opcode;
957
Raymond Hettingerf606f872003-03-16 03:11:04 +0000958 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000959 case POP_TOP:
960 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000961 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000962 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000963
Guido van Rossum374a9221991-04-04 10:40:29 +0000964 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000965 v = TOP();
966 w = SECOND();
967 SET_TOP(w);
968 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000969 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000970
Guido van Rossum374a9221991-04-04 10:40:29 +0000971 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000972 v = TOP();
973 w = SECOND();
974 x = THIRD();
975 SET_TOP(w);
976 SET_SECOND(x);
977 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000978 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000979
Thomas Wouters434d0822000-08-24 20:11:32 +0000980 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000981 u = TOP();
982 v = SECOND();
983 w = THIRD();
984 x = FOURTH();
985 SET_TOP(v);
986 SET_SECOND(w);
987 SET_THIRD(x);
988 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000989 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000990
Guido van Rossum374a9221991-04-04 10:40:29 +0000991 case DUP_TOP:
992 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000993 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000994 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000995 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000996
Thomas Wouters434d0822000-08-24 20:11:32 +0000997 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000998 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +0000999 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001000 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001001 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001002 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001003 STACKADJ(2);
1004 SET_TOP(x);
1005 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001006 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001007 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001008 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001009 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001010 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001011 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001012 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001013 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001014 STACKADJ(3);
1015 SET_TOP(x);
1016 SET_SECOND(w);
1017 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001018 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001019 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001020 Py_FatalError("invalid argument to DUP_TOPX"
1021 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001022 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001023
Guido van Rossum374a9221991-04-04 10:40:29 +00001024 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001025 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001026 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001027 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001028 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001029 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001030 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001031
Guido van Rossum374a9221991-04-04 10:40:29 +00001032 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001033 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001034 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001035 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001036 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001037 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001038 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001039
Guido van Rossum374a9221991-04-04 10:40:29 +00001040 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001041 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001042 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001043 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001044 if (err == 0) {
1045 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001046 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001047 continue;
1048 }
1049 else if (err > 0) {
1050 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001051 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001052 err = 0;
1053 continue;
1054 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001055 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001056 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001057
Guido van Rossum374a9221991-04-04 10:40:29 +00001058 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001059 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001060 x = PyObject_Repr(v);
1061 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001062 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001063 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001064 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001065
Guido van Rossum7928cd71991-10-24 14:59:31 +00001066 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001067 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001068 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001069 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001070 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001071 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001072 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001073
Guido van Rossum50564e81996-01-12 01:13:16 +00001074 case BINARY_POWER:
1075 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001076 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001077 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001078 Py_DECREF(v);
1079 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001080 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001081 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001082 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001083
Guido van Rossum374a9221991-04-04 10:40:29 +00001084 case BINARY_MULTIPLY:
1085 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001086 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001087 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001088 Py_DECREF(v);
1089 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001090 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001091 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001092 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001093
Guido van Rossum374a9221991-04-04 10:40:29 +00001094 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001095 if (!_Py_QnewFlag) {
1096 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001097 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001098 x = PyNumber_Divide(v, w);
1099 Py_DECREF(v);
1100 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001101 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001102 if (x != NULL) continue;
1103 break;
1104 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001105 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001106 BINARY_TRUE_DIVIDE */
1107 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001108 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001109 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001110 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001111 Py_DECREF(v);
1112 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001113 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001114 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001115 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001116
Guido van Rossum4668b002001-08-08 05:00:18 +00001117 case BINARY_FLOOR_DIVIDE:
1118 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001119 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001120 x = PyNumber_FloorDivide(v, w);
1121 Py_DECREF(v);
1122 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001123 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001124 if (x != NULL) continue;
1125 break;
1126
Guido van Rossum374a9221991-04-04 10:40:29 +00001127 case BINARY_MODULO:
1128 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001129 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001130 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001131 Py_DECREF(v);
1132 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001133 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001134 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001135 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001136
Guido van Rossum374a9221991-04-04 10:40:29 +00001137 case BINARY_ADD:
1138 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001139 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001140 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001141 /* INLINE: int + int */
1142 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001143 a = PyInt_AS_LONG(v);
1144 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001145 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001146 if ((i^a) < 0 && (i^b) < 0)
1147 goto slow_add;
1148 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001149 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001150 else if (PyString_CheckExact(v) &&
1151 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001152 x = string_concatenate(v, w, f, next_instr);
1153 /* string_concatenate consumed the ref to v */
1154 goto skip_decref_vx;
1155 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001156 else {
1157 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001158 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001159 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001160 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001161 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001162 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_SUBTRACT:
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_sub;
1178 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001179 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001180 else {
1181 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001182 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001183 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001184 Py_DECREF(v);
1185 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001186 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001187 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001188 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001189
Guido van Rossum374a9221991-04-04 10:40:29 +00001190 case BINARY_SUBSCR:
1191 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001192 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001193 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001194 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001195 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001196 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001197 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001198 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001199 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001200 Py_INCREF(x);
1201 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001202 else
1203 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001204 }
1205 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001206 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001207 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001208 Py_DECREF(v);
1209 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001210 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001211 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001212 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001213
Guido van Rossum7928cd71991-10-24 14:59:31 +00001214 case BINARY_LSHIFT:
1215 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001216 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001217 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001218 Py_DECREF(v);
1219 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001220 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001221 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001222 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001223
Guido van Rossum7928cd71991-10-24 14:59:31 +00001224 case BINARY_RSHIFT:
1225 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001226 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001227 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001228 Py_DECREF(v);
1229 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001230 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001231 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001232 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001233
Guido van Rossum7928cd71991-10-24 14:59:31 +00001234 case BINARY_AND:
1235 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001236 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001237 x = PyNumber_And(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 Rossum7928cd71991-10-24 14:59:31 +00001242 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001243
Guido van Rossum7928cd71991-10-24 14:59:31 +00001244 case BINARY_XOR:
1245 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001246 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001247 x = PyNumber_Xor(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_OR:
1255 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001256 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001257 x = PyNumber_Or(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;
Thomas Wouters434d0822000-08-24 20:11:32 +00001263
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001264 case LIST_APPEND:
1265 w = POP();
1266 v = POP();
1267 err = PyList_Append(v, w);
1268 Py_DECREF(v);
1269 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001270 if (err == 0) {
1271 PREDICT(JUMP_ABSOLUTE);
1272 continue;
1273 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001274 break;
1275
Thomas Wouters434d0822000-08-24 20:11:32 +00001276 case INPLACE_POWER:
1277 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001278 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001279 x = PyNumber_InPlacePower(v, w, Py_None);
1280 Py_DECREF(v);
1281 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001282 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001283 if (x != NULL) continue;
1284 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001285
Thomas Wouters434d0822000-08-24 20:11:32 +00001286 case INPLACE_MULTIPLY:
1287 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001288 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001289 x = PyNumber_InPlaceMultiply(v, w);
1290 Py_DECREF(v);
1291 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001292 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001293 if (x != NULL) continue;
1294 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001295
Thomas Wouters434d0822000-08-24 20:11:32 +00001296 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001297 if (!_Py_QnewFlag) {
1298 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001299 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001300 x = PyNumber_InPlaceDivide(v, w);
1301 Py_DECREF(v);
1302 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001303 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001304 if (x != NULL) continue;
1305 break;
1306 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001307 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001308 INPLACE_TRUE_DIVIDE */
1309 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001310 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001311 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001312 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001313 Py_DECREF(v);
1314 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001315 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001316 if (x != NULL) continue;
1317 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001318
Guido van Rossum4668b002001-08-08 05:00:18 +00001319 case INPLACE_FLOOR_DIVIDE:
1320 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001321 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001322 x = PyNumber_InPlaceFloorDivide(v, w);
1323 Py_DECREF(v);
1324 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001325 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001326 if (x != NULL) continue;
1327 break;
1328
Thomas Wouters434d0822000-08-24 20:11:32 +00001329 case INPLACE_MODULO:
1330 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001331 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001332 x = PyNumber_InPlaceRemainder(v, w);
1333 Py_DECREF(v);
1334 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001335 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001336 if (x != NULL) continue;
1337 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001338
Thomas Wouters434d0822000-08-24 20:11:32 +00001339 case INPLACE_ADD:
1340 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001341 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001342 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001343 /* INLINE: int + int */
1344 register long a, b, i;
1345 a = PyInt_AS_LONG(v);
1346 b = PyInt_AS_LONG(w);
1347 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001348 if ((i^a) < 0 && (i^b) < 0)
1349 goto slow_iadd;
1350 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001351 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001352 else if (PyString_CheckExact(v) &&
1353 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001354 x = string_concatenate(v, w, f, next_instr);
1355 /* string_concatenate consumed the ref to v */
1356 goto skip_decref_v;
1357 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001358 else {
1359 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001360 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001361 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001362 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001363 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001364 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_SUBTRACT:
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_isub;
1380 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001381 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001382 else {
1383 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001384 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001385 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001386 Py_DECREF(v);
1387 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001388 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001389 if (x != NULL) continue;
1390 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001391
Thomas Wouters434d0822000-08-24 20:11:32 +00001392 case INPLACE_LSHIFT:
1393 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001394 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001395 x = PyNumber_InPlaceLshift(v, w);
1396 Py_DECREF(v);
1397 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001398 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001399 if (x != NULL) continue;
1400 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001401
Thomas Wouters434d0822000-08-24 20:11:32 +00001402 case INPLACE_RSHIFT:
1403 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001404 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001405 x = PyNumber_InPlaceRshift(v, w);
1406 Py_DECREF(v);
1407 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001408 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001409 if (x != NULL) continue;
1410 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001411
Thomas Wouters434d0822000-08-24 20:11:32 +00001412 case INPLACE_AND:
1413 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001414 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001415 x = PyNumber_InPlaceAnd(v, w);
1416 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_XOR:
1423 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001424 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001425 x = PyNumber_InPlaceXor(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_OR:
1433 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001434 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001435 x = PyNumber_InPlaceOr(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;
1441
Guido van Rossum374a9221991-04-04 10:40:29 +00001442 case SLICE+0:
1443 case SLICE+1:
1444 case SLICE+2:
1445 case SLICE+3:
1446 if ((opcode-SLICE) & 2)
1447 w = POP();
1448 else
1449 w = NULL;
1450 if ((opcode-SLICE) & 1)
1451 v = POP();
1452 else
1453 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001454 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001455 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001456 Py_DECREF(u);
1457 Py_XDECREF(v);
1458 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001459 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001460 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001461 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001462
Guido van Rossum374a9221991-04-04 10:40:29 +00001463 case STORE_SLICE+0:
1464 case STORE_SLICE+1:
1465 case STORE_SLICE+2:
1466 case STORE_SLICE+3:
1467 if ((opcode-STORE_SLICE) & 2)
1468 w = POP();
1469 else
1470 w = NULL;
1471 if ((opcode-STORE_SLICE) & 1)
1472 v = POP();
1473 else
1474 v = NULL;
1475 u = POP();
1476 t = POP();
1477 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001478 Py_DECREF(t);
1479 Py_DECREF(u);
1480 Py_XDECREF(v);
1481 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001482 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001483 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001484
Guido van Rossum374a9221991-04-04 10:40:29 +00001485 case DELETE_SLICE+0:
1486 case DELETE_SLICE+1:
1487 case DELETE_SLICE+2:
1488 case DELETE_SLICE+3:
1489 if ((opcode-DELETE_SLICE) & 2)
1490 w = POP();
1491 else
1492 w = NULL;
1493 if ((opcode-DELETE_SLICE) & 1)
1494 v = POP();
1495 else
1496 v = NULL;
1497 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001498 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001499 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001500 Py_DECREF(u);
1501 Py_XDECREF(v);
1502 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001503 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001504 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001505
Guido van Rossum374a9221991-04-04 10:40:29 +00001506 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001507 w = TOP();
1508 v = SECOND();
1509 u = THIRD();
1510 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001511 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001512 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001513 Py_DECREF(u);
1514 Py_DECREF(v);
1515 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001516 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001517 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001518
Guido van Rossum374a9221991-04-04 10:40:29 +00001519 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001520 w = TOP();
1521 v = SECOND();
1522 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001523 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001524 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001525 Py_DECREF(v);
1526 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001527 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001528 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001529
Guido van Rossum374a9221991-04-04 10:40:29 +00001530 case PRINT_EXPR:
1531 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001532 w = PySys_GetObject("displayhook");
1533 if (w == NULL) {
1534 PyErr_SetString(PyExc_RuntimeError,
1535 "lost sys.displayhook");
1536 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001537 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001538 }
1539 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001540 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001541 if (x == NULL)
1542 err = -1;
1543 }
1544 if (err == 0) {
1545 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001546 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001547 if (w == NULL)
1548 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001549 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001550 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001551 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001552 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001553
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001554 case PRINT_ITEM_TO:
1555 w = stream = POP();
1556 /* fall through to PRINT_ITEM */
1557
Guido van Rossum374a9221991-04-04 10:40:29 +00001558 case PRINT_ITEM:
1559 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001560 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001561 w = PySys_GetObject("stdout");
1562 if (w == NULL) {
1563 PyErr_SetString(PyExc_RuntimeError,
1564 "lost sys.stdout");
1565 err = -1;
1566 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001567 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001568 /* PyFile_SoftSpace() can exececute arbitrary code
1569 if sys.stdout is an instance with a __getattr__.
1570 If __getattr__ raises an exception, w will
1571 be freed, so we need to prevent that temporarily. */
1572 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001573 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001574 err = PyFile_WriteString(" ", w);
1575 if (err == 0)
1576 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001577 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001578 /* XXX move into writeobject() ? */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001579 if (PyString_Check(v)) {
1580 char *s = PyString_AS_STRING(v);
1581 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001582 if (len == 0 ||
1583 !isspace(Py_CHARMASK(s[len-1])) ||
1584 s[len-1] == ' ')
1585 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001586 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001587#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001588 else if (PyUnicode_Check(v)) {
1589 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001590 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001591 if (len == 0 ||
1592 !Py_UNICODE_ISSPACE(s[len-1]) ||
1593 s[len-1] == ' ')
1594 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001595 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001596#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001597 else
1598 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001599 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001600 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001601 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001602 Py_XDECREF(stream);
1603 stream = NULL;
1604 if (err == 0)
1605 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001606 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001607
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001608 case PRINT_NEWLINE_TO:
1609 w = stream = POP();
1610 /* fall through to PRINT_NEWLINE */
1611
Guido van Rossum374a9221991-04-04 10:40:29 +00001612 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001613 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001614 w = PySys_GetObject("stdout");
1615 if (w == NULL)
1616 PyErr_SetString(PyExc_RuntimeError,
1617 "lost sys.stdout");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001618 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001619 if (w != NULL) {
Georg Brandlaa76d772008-07-01 20:56:03 +00001620 /* w.write() may replace sys.stdout, so we
1621 * have to keep our reference to it */
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001622 Py_INCREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001623 err = PyFile_WriteString("\n", w);
1624 if (err == 0)
1625 PyFile_SoftSpace(w, 0);
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001626 Py_DECREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001627 }
1628 Py_XDECREF(stream);
1629 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001630 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001631
Thomas Wouters434d0822000-08-24 20:11:32 +00001632
1633#ifdef CASE_TOO_BIG
1634 default: switch (opcode) {
1635#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001636 case RAISE_VARARGS:
1637 u = v = w = NULL;
1638 switch (oparg) {
1639 case 3:
1640 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001641 /* Fallthrough */
1642 case 2:
1643 v = POP(); /* value */
1644 /* Fallthrough */
1645 case 1:
1646 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001647 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001648 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001649 break;
1650 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001651 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001652 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001653 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001654 break;
1655 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001656 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001657
Guido van Rossum374a9221991-04-04 10:40:29 +00001658 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001659 if ((x = f->f_locals) != NULL) {
1660 Py_INCREF(x);
1661 PUSH(x);
1662 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001663 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001664 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001665 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001666
Guido van Rossum374a9221991-04-04 10:40:29 +00001667 case RETURN_VALUE:
1668 retval = POP();
1669 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001670 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001671
Tim Peters5ca576e2001-06-18 22:08:13 +00001672 case YIELD_VALUE:
1673 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001674 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001675 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001676 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001677
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001678 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001679 w = TOP();
1680 v = SECOND();
1681 u = THIRD();
1682 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001683 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001684 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001685 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001686 Py_DECREF(u);
1687 Py_DECREF(v);
1688 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001689 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001690
Guido van Rossum374a9221991-04-04 10:40:29 +00001691 case POP_BLOCK:
1692 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001693 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001694 while (STACK_LEVEL() > b->b_level) {
1695 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001696 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001697 }
1698 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001699 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001700
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00001701 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001702 case END_FINALLY:
1703 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001704 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001705 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001706 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001707 if (why == WHY_RETURN ||
1708 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001709 retval = POP();
1710 }
Thomas Wouterse2176022007-09-20 17:35:10 +00001711 else if (PyExceptionClass_Check(v) ||
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001712 PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001713 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001714 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001715 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001716 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001717 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001718 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001719 else if (v != Py_None) {
1720 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001721 "'finally' pops bad exception");
1722 why = WHY_EXCEPTION;
1723 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001724 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001725 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001726
Guido van Rossum374a9221991-04-04 10:40:29 +00001727 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001728 u = TOP();
1729 v = SECOND();
1730 w = THIRD();
1731 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001732 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001733 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001734 Py_DECREF(u);
1735 Py_DECREF(v);
1736 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001737 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001738
Guido van Rossum374a9221991-04-04 10:40:29 +00001739 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001740 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001741 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001742 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001743 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001744 err = PyDict_SetItem(x, w, v);
1745 else
1746 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001747 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001748 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001749 break;
1750 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001751 PyErr_Format(PyExc_SystemError,
1752 "no locals found when storing %s",
1753 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001754 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001755
Guido van Rossum374a9221991-04-04 10:40:29 +00001756 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001757 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001758 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001759 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001760 format_exc_check_arg(PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001761 NAME_ERROR_MSG,
1762 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001763 break;
1764 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001765 PyErr_Format(PyExc_SystemError,
1766 "no locals when deleting %s",
1767 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001768 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001769
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001770 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001771 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001772 v = POP();
Thomas Wouterse2176022007-09-20 17:35:10 +00001773 if (PyTuple_CheckExact(v) &&
1774 PyTuple_GET_SIZE(v) == oparg) {
1775 PyObject **items = \
1776 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001777 while (oparg--) {
1778 w = items[oparg];
1779 Py_INCREF(w);
1780 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001781 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001782 Py_DECREF(v);
1783 continue;
Thomas Wouterse2176022007-09-20 17:35:10 +00001784 } else if (PyList_CheckExact(v) &&
1785 PyList_GET_SIZE(v) == oparg) {
1786 PyObject **items = \
1787 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001788 while (oparg--) {
1789 w = items[oparg];
1790 Py_INCREF(w);
1791 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001792 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001793 } else if (unpack_iterable(v, oparg,
Thomas Wouterse2176022007-09-20 17:35:10 +00001794 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001795 stack_pointer += oparg;
Georg Brandl5cb76c12007-03-21 09:00:39 +00001796 } else {
1797 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001798 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001799 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001800 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001801 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001802
Guido van Rossum374a9221991-04-04 10:40:29 +00001803 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001804 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001805 v = TOP();
1806 u = SECOND();
1807 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001808 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1809 Py_DECREF(v);
1810 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001811 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001812 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001813
Guido van Rossum374a9221991-04-04 10:40:29 +00001814 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001815 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001816 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001817 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1818 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001819 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001820 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001821
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001822 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001823 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001824 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001825 err = PyDict_SetItem(f->f_globals, w, v);
1826 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001827 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001828 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001829
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001830 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001831 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001832 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001833 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001834 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001835 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001836
Guido van Rossum374a9221991-04-04 10:40:29 +00001837 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001838 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001839 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001840 PyErr_Format(PyExc_SystemError,
1841 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001842 PyObject_REPR(w));
Guido van Rossum681d79a1995-07-18 14:51:37 +00001843 break;
1844 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001845 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001846 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001847 Py_XINCREF(x);
1848 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001849 else {
1850 x = PyObject_GetItem(v, w);
1851 if (x == NULL && PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00001852 if (!PyErr_ExceptionMatches(
1853 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001854 break;
1855 PyErr_Clear();
1856 }
1857 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001858 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001859 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001860 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001861 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001862 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001863 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001864 PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001865 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001866 break;
1867 }
1868 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001869 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001870 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001871 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001872 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001873
Guido van Rossum374a9221991-04-04 10:40:29 +00001874 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001875 w = GETITEM(names, oparg);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001876 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001877 /* Inline the PyDict_GetItem() calls.
1878 WARNING: this is an extreme speed hack.
1879 Do not try this at home. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001880 long hash = ((PyStringObject *)w)->ob_shash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001881 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001882 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00001883 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001884 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00001885 e = d->ma_lookup(d, w, hash);
1886 if (e == NULL) {
1887 x = NULL;
1888 break;
1889 }
1890 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001891 if (x != NULL) {
1892 Py_INCREF(x);
1893 PUSH(x);
1894 continue;
1895 }
1896 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00001897 e = d->ma_lookup(d, w, hash);
1898 if (e == NULL) {
1899 x = NULL;
1900 break;
1901 }
1902 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001903 if (x != NULL) {
1904 Py_INCREF(x);
1905 PUSH(x);
1906 continue;
1907 }
1908 goto load_global_error;
1909 }
1910 }
1911 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001912 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001913 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001914 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001915 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001916 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001917 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001918 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001919 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001920 break;
1921 }
1922 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001923 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001924 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001925 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001926
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001927 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001928 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001929 if (x != NULL) {
1930 SETLOCAL(oparg, NULL);
1931 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001932 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001933 format_exc_check_arg(
1934 PyExc_UnboundLocalError,
1935 UNBOUNDLOCAL_ERROR_MSG,
1936 PyTuple_GetItem(co->co_varnames, oparg)
1937 );
1938 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001939
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001940 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001941 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001942 Py_INCREF(x);
1943 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001944 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001945 break;
1946
1947 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001948 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001949 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001950 if (w != NULL) {
1951 PUSH(w);
1952 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001953 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001954 err = -1;
1955 /* Don't stomp existing exception */
1956 if (PyErr_Occurred())
1957 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00001958 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1959 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001960 oparg);
1961 format_exc_check_arg(
1962 PyExc_UnboundLocalError,
1963 UNBOUNDLOCAL_ERROR_MSG,
1964 v);
1965 } else {
Thomas Wouterse2176022007-09-20 17:35:10 +00001966 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
1967 PyTuple_GET_SIZE(co->co_cellvars));
1968 format_exc_check_arg(PyExc_NameError,
1969 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001970 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001971 break;
1972
1973 case STORE_DEREF:
1974 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001975 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001976 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001977 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001978 continue;
1979
Guido van Rossum374a9221991-04-04 10:40:29 +00001980 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001981 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001982 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001983 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001984 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001985 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001986 }
1987 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001988 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001989 }
1990 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001991
Guido van Rossum374a9221991-04-04 10:40:29 +00001992 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001993 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001994 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001995 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001996 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001997 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001998 }
1999 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002000 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002001 }
2002 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002003
Guido van Rossum374a9221991-04-04 10:40:29 +00002004 case BUILD_MAP:
Raymond Hettingerfd7ed402007-12-18 21:24:09 +00002005 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002006 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002007 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002008 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002009
Raymond Hettingereffde122007-12-18 18:26:18 +00002010 case STORE_MAP:
2011 w = TOP(); /* key */
2012 u = SECOND(); /* value */
2013 v = THIRD(); /* dict */
2014 STACKADJ(-2);
2015 assert (PyDict_CheckExact(v));
2016 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2017 Py_DECREF(u);
2018 Py_DECREF(w);
2019 if (err == 0) continue;
2020 break;
2021
Guido van Rossum374a9221991-04-04 10:40:29 +00002022 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00002023 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002024 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002025 x = PyObject_GetAttr(v, w);
2026 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002027 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002028 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002029 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002030
Guido van Rossum374a9221991-04-04 10:40:29 +00002031 case COMPARE_OP:
2032 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002033 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00002034 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002035 /* INLINE: cmp(int, int) */
2036 register long a, b;
2037 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002038 a = PyInt_AS_LONG(v);
2039 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002040 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002041 case PyCmp_LT: res = a < b; break;
2042 case PyCmp_LE: res = a <= b; break;
2043 case PyCmp_EQ: res = a == b; break;
2044 case PyCmp_NE: res = a != b; break;
2045 case PyCmp_GT: res = a > b; break;
2046 case PyCmp_GE: res = a >= b; break;
2047 case PyCmp_IS: res = v == w; break;
2048 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002049 default: goto slow_compare;
2050 }
2051 x = res ? Py_True : Py_False;
2052 Py_INCREF(x);
2053 }
2054 else {
2055 slow_compare:
2056 x = cmp_outcome(oparg, v, w);
2057 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002058 Py_DECREF(v);
2059 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002060 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002061 if (x == NULL) break;
2062 PREDICT(JUMP_IF_FALSE);
2063 PREDICT(JUMP_IF_TRUE);
2064 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002065
Guido van Rossum374a9221991-04-04 10:40:29 +00002066 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002067 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002068 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002069 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002070 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002071 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002072 break;
2073 }
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002074 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002075 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002076 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002077 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2078 w = PyTuple_Pack(5,
2079 w,
2080 f->f_globals,
2081 f->f_locals == NULL ?
2082 Py_None : f->f_locals,
2083 v,
2084 u);
2085 else
2086 w = PyTuple_Pack(4,
2087 w,
2088 f->f_globals,
2089 f->f_locals == NULL ?
2090 Py_None : f->f_locals,
2091 v);
2092 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002093 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002094 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002095 u = POP();
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002096 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002097 x = NULL;
2098 break;
2099 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002100 READ_TIMESTAMP(intr0);
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002101 v = x;
2102 x = PyEval_CallObject(v, w);
2103 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002104 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002105 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002106 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002107 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002108 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002109
Thomas Wouters52152252000-08-17 22:55:00 +00002110 case IMPORT_STAR:
2111 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002112 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002113 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002114 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002115 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002116 break;
2117 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002118 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002119 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002120 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002121 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002122 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002123 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002124 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002125
Thomas Wouters52152252000-08-17 22:55:00 +00002126 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002127 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002128 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002129 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002130 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002131 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002132 PUSH(x);
2133 if (x != NULL) continue;
2134 break;
2135
Guido van Rossum374a9221991-04-04 10:40:29 +00002136 case JUMP_FORWARD:
2137 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002138 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002139
Raymond Hettingerf606f872003-03-16 03:11:04 +00002140 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002141 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002142 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002143 if (w == Py_True) {
2144 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002145 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002146 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002147 if (w == Py_False) {
2148 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002149 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002150 }
2151 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002152 if (err > 0)
2153 err = 0;
2154 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002155 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002156 else
2157 break;
2158 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002159
Raymond Hettingerf606f872003-03-16 03:11:04 +00002160 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002161 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002162 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002163 if (w == Py_False) {
2164 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002165 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002166 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002167 if (w == Py_True) {
2168 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002169 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002170 }
2171 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002172 if (err > 0) {
2173 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002174 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002175 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002176 else if (err == 0)
2177 ;
2178 else
2179 break;
2180 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002181
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002182 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002183 case JUMP_ABSOLUTE:
2184 JUMPTO(oparg);
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002185#if FAST_LOOPS
2186 /* Enabling this path speeds-up all while and for-loops by bypassing
2187 the per-loop checks for signals. By default, this should be turned-off
2188 because it prevents detection of a control-break in tight loops like
2189 "while 1: pass". Compile with this option turned-on when you need
2190 the speed-up and do not need break checking inside tight loops (ones
2191 that contain only instructions ending with goto fast_next_opcode).
2192 */
2193 goto fast_next_opcode;
2194#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002195 continue;
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002196#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002197
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002198 case GET_ITER:
2199 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002200 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002201 x = PyObject_GetIter(v);
2202 Py_DECREF(v);
2203 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002204 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002205 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002206 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002207 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002208 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002209 break;
2210
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002211 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002212 case FOR_ITER:
2213 /* before: [iter]; after: [iter, iter()] *or* [] */
2214 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002215 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002216 if (x != NULL) {
2217 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002218 PREDICT(STORE_FAST);
2219 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002220 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002221 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002222 if (PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002223 if (!PyErr_ExceptionMatches(
2224 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002225 break;
2226 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002227 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002228 /* iterator ended normally */
2229 x = v = POP();
2230 Py_DECREF(v);
2231 JUMPBY(oparg);
2232 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002233
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002234 case BREAK_LOOP:
2235 why = WHY_BREAK;
2236 goto fast_block_end;
2237
2238 case CONTINUE_LOOP:
2239 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002240 if (!retval) {
2241 x = NULL;
2242 break;
2243 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002244 why = WHY_CONTINUE;
2245 goto fast_block_end;
2246
Guido van Rossum374a9221991-04-04 10:40:29 +00002247 case SETUP_LOOP:
2248 case SETUP_EXCEPT:
2249 case SETUP_FINALLY:
Thomas Wouterse2176022007-09-20 17:35:10 +00002250 /* NOTE: If you add any new block-setup opcodes that
2251 are not try/except/finally handlers, you may need
2252 to update the PyGen_NeedsFinalizing() function.
2253 */
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002254
Guido van Rossumb209a111997-04-29 18:18:01 +00002255 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002256 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002257 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002258
Guido van Rossumc2e20742006-02-27 22:32:47 +00002259 case WITH_CLEANUP:
2260 {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002261 /* At the top of the stack are 1-3 values indicating
2262 how/why we entered the finally clause:
2263 - TOP = None
2264 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2265 - TOP = WHY_*; no retval below it
2266 - (TOP, SECOND, THIRD) = exc_info()
2267 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002268 In the last case, we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002269 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002270 otherwise we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002271 EXIT(None, None, None)
2272
2273 In all cases, we remove EXIT from the stack, leaving
2274 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002275
2276 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002277 *and* the function call returns a 'true' value, we
2278 "zap" this information, to prevent END_FINALLY from
2279 re-raising the exception. (But non-local gotos
2280 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002281 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002282
Nick Coghlan7af53be2008-03-07 14:13:28 +00002283 PyObject *exit_func;
2284
2285 u = POP();
2286 if (u == Py_None) {
2287 exit_func = TOP();
2288 SET_TOP(u);
2289 v = w = Py_None;
2290 }
2291 else if (PyInt_Check(u)) {
2292 switch(PyInt_AS_LONG(u)) {
2293 case WHY_RETURN:
2294 case WHY_CONTINUE:
2295 /* Retval in TOP. */
2296 exit_func = SECOND();
2297 SET_SECOND(TOP());
2298 SET_TOP(u);
2299 break;
2300 default:
2301 exit_func = TOP();
2302 SET_TOP(u);
2303 break;
2304 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002305 u = v = w = Py_None;
2306 }
2307 else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002308 v = TOP();
2309 w = SECOND();
2310 exit_func = THIRD();
2311 SET_TOP(u);
2312 SET_SECOND(v);
2313 SET_THIRD(w);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002314 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002315 /* XXX Not the fastest way to call it... */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002316 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2317 NULL);
2318 if (x == NULL) {
2319 Py_DECREF(exit_func);
Guido van Rossumf6694362006-03-10 02:28:35 +00002320 break; /* Go to error exit */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002321 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002322 if (u != Py_None && PyObject_IsTrue(x)) {
2323 /* There was an exception and a true return */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002324 STACKADJ(-2);
Guido van Rossumf6694362006-03-10 02:28:35 +00002325 Py_INCREF(Py_None);
2326 SET_TOP(Py_None);
Guido van Rossumf6694362006-03-10 02:28:35 +00002327 Py_DECREF(u);
2328 Py_DECREF(v);
2329 Py_DECREF(w);
2330 } else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002331 /* The stack was rearranged to remove EXIT
2332 above. Let END_FINALLY do its thing */
Guido van Rossumf6694362006-03-10 02:28:35 +00002333 }
Nick Coghlan7af53be2008-03-07 14:13:28 +00002334 Py_DECREF(x);
2335 Py_DECREF(exit_func);
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00002336 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002337 break;
2338 }
2339
Guido van Rossumf10570b1995-07-07 22:53:21 +00002340 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002341 {
2342 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002343 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002344 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002345#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002346 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002347#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002348 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002349#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002350 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002351 PUSH(x);
2352 if (x != NULL)
2353 continue;
2354 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002355 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002356
Jeremy Hylton76901512000-03-28 23:49:17 +00002357 case CALL_FUNCTION_VAR:
2358 case CALL_FUNCTION_KW:
2359 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002360 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002361 int na = oparg & 0xff;
2362 int nk = (oparg>>8) & 0xff;
2363 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002364 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002365 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002366 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002367 if (flags & CALL_FLAG_VAR)
2368 n++;
2369 if (flags & CALL_FLAG_KW)
2370 n++;
2371 pfunc = stack_pointer - n - 1;
2372 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002373
Guido van Rossumac7be682001-01-17 15:42:30 +00002374 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002375 && PyMethod_GET_SELF(func) != NULL) {
2376 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002377 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002378 func = PyMethod_GET_FUNCTION(func);
2379 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002380 Py_DECREF(*pfunc);
2381 *pfunc = self;
2382 na++;
2383 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002384 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002385 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002386 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002387 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002388 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002389 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002390 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002391 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002392
Jeremy Hylton76901512000-03-28 23:49:17 +00002393 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002394 w = POP();
2395 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002396 }
2397 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002398 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002399 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002400 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002401 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002402
Guido van Rossum681d79a1995-07-18 14:51:37 +00002403 case MAKE_FUNCTION:
2404 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002405 x = PyFunction_New(v, f->f_globals);
2406 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002407 /* XXX Maybe this should be a separate opcode? */
2408 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002409 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002410 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002411 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002412 x = NULL;
2413 break;
2414 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002415 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002416 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002417 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002418 }
2419 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002420 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002421 }
2422 PUSH(x);
2423 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002424
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002425 case MAKE_CLOSURE:
2426 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002427 v = POP(); /* code object */
2428 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002429 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002430 if (x != NULL) {
2431 v = POP();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002432 err = PyFunction_SetClosure(x, v);
2433 Py_DECREF(v);
2434 }
2435 if (x != NULL && oparg > 0) {
2436 v = PyTuple_New(oparg);
2437 if (v == NULL) {
2438 Py_DECREF(x);
2439 x = NULL;
2440 break;
2441 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002442 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002443 w = POP();
2444 PyTuple_SET_ITEM(v, oparg, w);
2445 }
2446 err = PyFunction_SetDefaults(x, v);
2447 Py_DECREF(v);
2448 }
2449 PUSH(x);
2450 break;
2451 }
2452
Guido van Rossum8861b741996-07-30 16:49:37 +00002453 case BUILD_SLICE:
2454 if (oparg == 3)
2455 w = POP();
2456 else
2457 w = NULL;
2458 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002459 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002460 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002461 Py_DECREF(u);
2462 Py_DECREF(v);
2463 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002464 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002465 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002466 break;
2467
Fred Drakeef8ace32000-08-24 00:32:09 +00002468 case EXTENDED_ARG:
2469 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002470 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002471 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002472
Guido van Rossum374a9221991-04-04 10:40:29 +00002473 default:
2474 fprintf(stderr,
2475 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002476 PyCode_Addr2Line(f->f_code, f->f_lasti),
2477 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002478 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002479 why = WHY_EXCEPTION;
2480 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002481
2482#ifdef CASE_TOO_BIG
2483 }
2484#endif
2485
Guido van Rossum374a9221991-04-04 10:40:29 +00002486 } /* switch */
2487
2488 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002489
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002490 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002491
Guido van Rossum374a9221991-04-04 10:40:29 +00002492 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002493
Guido van Rossum374a9221991-04-04 10:40:29 +00002494 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002495 if (err == 0 && x != NULL) {
2496#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002497 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002498 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002499 fprintf(stderr,
2500 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002501 else {
2502#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002503 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002504 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002505#ifdef CHECKEXC
2506 }
2507#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002508 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002509 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002510 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002511 err = 0;
2512 }
2513
Guido van Rossum374a9221991-04-04 10:40:29 +00002514 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002515
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002516 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002517 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002518 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002519 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002520 why = WHY_EXCEPTION;
2521 }
2522 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002523#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002524 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002525 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002526 if (PyErr_Occurred()) {
Neal Norwitz8250fbe2008-01-27 17:12:15 +00002527 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002528 sprintf(buf, "Stack unwind with exception "
2529 "set and why=%d", why);
2530 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002531 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002532 }
2533#endif
2534
2535 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002536
Guido van Rossum374a9221991-04-04 10:40:29 +00002537 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002538 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002539
Fred Drake8f51f542001-10-04 14:48:42 +00002540 if (tstate->c_tracefunc != NULL)
2541 call_exc_trace(tstate->c_tracefunc,
2542 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002543 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002544
Guido van Rossum374a9221991-04-04 10:40:29 +00002545 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002546
Guido van Rossum374a9221991-04-04 10:40:29 +00002547 if (why == WHY_RERAISE)
2548 why = WHY_EXCEPTION;
2549
2550 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002551
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002552fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002553 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002554 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002555
Tim Peters8a5c3c72004-04-05 19:36:21 +00002556 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002557 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2558 /* For a continue inside a try block,
2559 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002560 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2561 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002562 why = WHY_NOT;
2563 JUMPTO(PyInt_AS_LONG(retval));
2564 Py_DECREF(retval);
2565 break;
2566 }
2567
Guido van Rossum374a9221991-04-04 10:40:29 +00002568 while (STACK_LEVEL() > b->b_level) {
2569 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002570 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002571 }
2572 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2573 why = WHY_NOT;
2574 JUMPTO(b->b_handler);
2575 break;
2576 }
2577 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002578 (b->b_type == SETUP_EXCEPT &&
2579 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002580 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002581 PyObject *exc, *val, *tb;
2582 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002583 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002584 val = Py_None;
2585 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002586 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002587 /* Make the raw exception data
2588 available to the handler,
2589 so a program can emulate the
2590 Python main loop. Don't do
2591 this for 'finally'. */
2592 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002593 PyErr_NormalizeException(
2594 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002595 set_exc_info(tstate,
2596 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002597 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002598 if (tb == NULL) {
2599 Py_INCREF(Py_None);
2600 PUSH(Py_None);
2601 } else
2602 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002603 PUSH(val);
2604 PUSH(exc);
2605 }
2606 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002607 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002608 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002609 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002610 PUSH(v);
2611 }
2612 why = WHY_NOT;
2613 JUMPTO(b->b_handler);
2614 break;
2615 }
2616 } /* unwind stack */
2617
2618 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002619
Guido van Rossum374a9221991-04-04 10:40:29 +00002620 if (why != WHY_NOT)
2621 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002622 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002623
Guido van Rossum374a9221991-04-04 10:40:29 +00002624 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002625
Tim Peters8a5c3c72004-04-05 19:36:21 +00002626 assert(why != WHY_YIELD);
2627 /* Pop remaining stack entries. */
2628 while (!EMPTY()) {
2629 v = POP();
2630 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002631 }
2632
Tim Peters8a5c3c72004-04-05 19:36:21 +00002633 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002634 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002635
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002636fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002637 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002638 if (tstate->c_tracefunc) {
2639 if (why == WHY_RETURN || why == WHY_YIELD) {
2640 if (call_trace(tstate->c_tracefunc,
2641 tstate->c_traceobj, f,
2642 PyTrace_RETURN, retval)) {
2643 Py_XDECREF(retval);
2644 retval = NULL;
2645 why = WHY_EXCEPTION;
2646 }
2647 }
2648 else if (why == WHY_EXCEPTION) {
2649 call_trace_protected(tstate->c_tracefunc,
2650 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002651 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002652 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002653 }
Fred Drake8f51f542001-10-04 14:48:42 +00002654 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002655 if (why == WHY_EXCEPTION)
2656 call_trace_protected(tstate->c_profilefunc,
2657 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002658 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002659 else if (call_trace(tstate->c_profilefunc,
2660 tstate->c_profileobj, f,
2661 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002662 Py_XDECREF(retval);
2663 retval = NULL;
2664 why = WHY_EXCEPTION;
2665 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002666 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002667 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002668
Tim Peters7df5e7f2006-05-26 23:14:37 +00002669 if (tstate->frame->f_exc_type != NULL)
2670 reset_exc_info(tstate);
2671 else {
2672 assert(tstate->frame->f_exc_value == NULL);
2673 assert(tstate->frame->f_exc_traceback == NULL);
2674 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002675
Tim Peters5ca576e2001-06-18 22:08:13 +00002676 /* pop frame */
Thomas Woutersae406c62007-09-19 17:27:43 +00002677exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002678 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002679 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002680
Guido van Rossum96a42c81992-01-12 02:29:51 +00002681 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002682}
2683
Guido van Rossumc2e20742006-02-27 22:32:47 +00002684/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002685 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002686 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002687
Tim Peters6d6c1a32001-08-02 04:15:00 +00002688PyObject *
2689PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002690 PyObject **args, int argcount, PyObject **kws, int kwcount,
2691 PyObject **defs, int defcount, PyObject *closure)
2692{
2693 register PyFrameObject *f;
2694 register PyObject *retval = NULL;
2695 register PyObject **fastlocals, **freevars;
2696 PyThreadState *tstate = PyThreadState_GET();
2697 PyObject *x, *u;
2698
2699 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002700 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002701 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002702 return NULL;
2703 }
2704
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002705 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002706 assert(globals != NULL);
2707 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002708 if (f == NULL)
2709 return NULL;
2710
2711 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002712 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002713
2714 if (co->co_argcount > 0 ||
2715 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2716 int i;
2717 int n = argcount;
2718 PyObject *kwdict = NULL;
2719 if (co->co_flags & CO_VARKEYWORDS) {
2720 kwdict = PyDict_New();
2721 if (kwdict == NULL)
2722 goto fail;
2723 i = co->co_argcount;
2724 if (co->co_flags & CO_VARARGS)
2725 i++;
2726 SETLOCAL(i, kwdict);
2727 }
2728 if (argcount > co->co_argcount) {
2729 if (!(co->co_flags & CO_VARARGS)) {
2730 PyErr_Format(PyExc_TypeError,
2731 "%.200s() takes %s %d "
2732 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002733 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002734 defcount ? "at most" : "exactly",
2735 co->co_argcount,
2736 kwcount ? "non-keyword " : "",
2737 co->co_argcount == 1 ? "" : "s",
2738 argcount);
2739 goto fail;
2740 }
2741 n = co->co_argcount;
2742 }
2743 for (i = 0; i < n; i++) {
2744 x = args[i];
2745 Py_INCREF(x);
2746 SETLOCAL(i, x);
2747 }
2748 if (co->co_flags & CO_VARARGS) {
2749 u = PyTuple_New(argcount - n);
2750 if (u == NULL)
2751 goto fail;
2752 SETLOCAL(co->co_argcount, u);
2753 for (i = n; i < argcount; i++) {
2754 x = args[i];
2755 Py_INCREF(x);
2756 PyTuple_SET_ITEM(u, i-n, x);
2757 }
2758 }
2759 for (i = 0; i < kwcount; i++) {
2760 PyObject *keyword = kws[2*i];
2761 PyObject *value = kws[2*i + 1];
2762 int j;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002763 if (keyword == NULL || !PyString_Check(keyword)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002764 PyErr_Format(PyExc_TypeError,
2765 "%.200s() keywords must be strings",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002766 PyString_AsString(co->co_name));
Tim Peters5ca576e2001-06-18 22:08:13 +00002767 goto fail;
2768 }
2769 /* XXX slow -- speed up using dictionary? */
2770 for (j = 0; j < co->co_argcount; j++) {
2771 PyObject *nm = PyTuple_GET_ITEM(
2772 co->co_varnames, j);
2773 int cmp = PyObject_RichCompareBool(
2774 keyword, nm, Py_EQ);
2775 if (cmp > 0)
2776 break;
2777 else if (cmp < 0)
2778 goto fail;
2779 }
2780 /* Check errors from Compare */
2781 if (PyErr_Occurred())
2782 goto fail;
2783 if (j >= co->co_argcount) {
2784 if (kwdict == NULL) {
2785 PyErr_Format(PyExc_TypeError,
2786 "%.200s() got an unexpected "
2787 "keyword argument '%.400s'",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002788 PyString_AsString(co->co_name),
2789 PyString_AsString(keyword));
Tim Peters5ca576e2001-06-18 22:08:13 +00002790 goto fail;
2791 }
2792 PyDict_SetItem(kwdict, keyword, value);
2793 }
2794 else {
2795 if (GETLOCAL(j) != NULL) {
2796 PyErr_Format(PyExc_TypeError,
2797 "%.200s() got multiple "
2798 "values for keyword "
2799 "argument '%.400s'",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002800 PyString_AsString(co->co_name),
2801 PyString_AsString(keyword));
Tim Peters5ca576e2001-06-18 22:08:13 +00002802 goto fail;
2803 }
2804 Py_INCREF(value);
2805 SETLOCAL(j, value);
2806 }
2807 }
2808 if (argcount < co->co_argcount) {
2809 int m = co->co_argcount - defcount;
2810 for (i = argcount; i < m; i++) {
2811 if (GETLOCAL(i) == NULL) {
2812 PyErr_Format(PyExc_TypeError,
2813 "%.200s() takes %s %d "
2814 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002815 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002816 ((co->co_flags & CO_VARARGS) ||
2817 defcount) ? "at least"
2818 : "exactly",
2819 m, kwcount ? "non-keyword " : "",
2820 m == 1 ? "" : "s", i);
2821 goto fail;
2822 }
2823 }
2824 if (n > m)
2825 i = n - m;
2826 else
2827 i = 0;
2828 for (; i < defcount; i++) {
2829 if (GETLOCAL(m+i) == NULL) {
2830 PyObject *def = defs[i];
2831 Py_INCREF(def);
2832 SETLOCAL(m+i, def);
2833 }
2834 }
2835 }
2836 }
2837 else {
2838 if (argcount > 0 || kwcount > 0) {
2839 PyErr_Format(PyExc_TypeError,
2840 "%.200s() takes no arguments (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002841 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002842 argcount + kwcount);
2843 goto fail;
2844 }
2845 }
2846 /* Allocate and initialize storage for cell vars, and copy free
2847 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00002848 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00002849 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002850 char *cellname, *argname;
2851 PyObject *c;
2852
2853 nargs = co->co_argcount;
2854 if (co->co_flags & CO_VARARGS)
2855 nargs++;
2856 if (co->co_flags & CO_VARKEYWORDS)
2857 nargs++;
2858
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002859 /* Initialize each cell var, taking into account
2860 cell vars that are initialized from arguments.
2861
2862 Should arrange for the compiler to put cellvars
2863 that are arguments at the beginning of the cellvars
2864 list so that we can march over it more efficiently?
2865 */
Richard Jonescebbefc2006-05-23 18:28:17 +00002866 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002867 cellname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00002868 PyTuple_GET_ITEM(co->co_cellvars, i));
2869 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002870 for (j = 0; j < nargs; j++) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002871 argname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00002872 PyTuple_GET_ITEM(co->co_varnames, j));
2873 if (strcmp(cellname, argname) == 0) {
2874 c = PyCell_New(GETLOCAL(j));
2875 if (c == NULL)
2876 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002877 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002878 found = 1;
2879 break;
2880 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002881 }
2882 if (found == 0) {
2883 c = PyCell_New(NULL);
2884 if (c == NULL)
2885 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002886 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002887 }
2888 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002889 }
Richard Jonescebbefc2006-05-23 18:28:17 +00002890 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002891 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00002892 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002893 PyObject *o = PyTuple_GET_ITEM(closure, i);
2894 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00002895 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002896 }
2897 }
2898
Tim Peters5ca576e2001-06-18 22:08:13 +00002899 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002900 /* Don't need to keep the reference to f_back, it will be set
2901 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002902 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002903 f->f_back = NULL;
2904
Jeremy Hylton985eba52003-02-05 23:13:00 +00002905 PCALL(PCALL_GENERATOR);
2906
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002907 /* Create a new generator that owns the ready to run frame
2908 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002909 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002910 }
2911
Thomas Woutersae406c62007-09-19 17:27:43 +00002912 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002913
Thomas Woutersae406c62007-09-19 17:27:43 +00002914fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00002915
Tim Petersb13680b2001-11-27 23:29:29 +00002916 /* decref'ing the frame can cause __del__ methods to get invoked,
2917 which can call back into Python. While we're done with the
2918 current Python frame (f), the associated C stack is still in use,
2919 so recursion_depth must be boosted for the duration.
2920 */
2921 assert(tstate != NULL);
2922 ++tstate->recursion_depth;
Thomas Woutersae406c62007-09-19 17:27:43 +00002923 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002924 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002925 return retval;
2926}
2927
2928
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002929/* Implementation notes for set_exc_info() and reset_exc_info():
2930
2931- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2932 'exc_traceback'. These always travel together.
2933
2934- tstate->curexc_ZZZ is the "hot" exception that is set by
2935 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2936
2937- Once an exception is caught by an except clause, it is transferred
2938 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2939 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00002940 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002941
2942- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2943
2944 Long ago, when none of this existed, there were just a few globals:
2945 one set corresponding to the "hot" exception, and one set
2946 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2947 globals; they were simply stored as sys.exc_ZZZ. For backwards
2948 compatibility, they still are!) The problem was that in code like
2949 this:
2950
2951 try:
2952 "something that may fail"
2953 except "some exception":
2954 "do something else first"
2955 "print the exception from sys.exc_ZZZ."
2956
2957 if "do something else first" invoked something that raised and caught
2958 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2959 cause of subtle bugs. I fixed this by changing the semantics as
2960 follows:
2961
2962 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2963 *in that frame*.
2964
2965 - But initially, and as long as no exception is caught in a given
2966 frame, sys.exc_ZZZ will hold the last exception caught in the
2967 previous frame (or the frame before that, etc.).
2968
2969 The first bullet fixed the bug in the above example. The second
2970 bullet was for backwards compatibility: it was (and is) common to
2971 have a function that is called when an exception is caught, and to
2972 have that function access the caught exception via sys.exc_ZZZ.
2973 (Example: traceback.print_exc()).
2974
2975 At the same time I fixed the problem that sys.exc_ZZZ weren't
2976 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2977 but that's really a separate improvement.
2978
2979 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2980 variables to what they were before the current frame was called. The
2981 set_exc_info() function saves them on the frame so that
2982 reset_exc_info() can restore them. The invariant is that
2983 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2984 exception (where "catching" an exception applies only to successful
2985 except clauses); and if the current frame ever caught an exception,
2986 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2987 at the start of the current frame.
2988
2989*/
2990
Fredrik Lundh7a830892006-05-27 10:39:48 +00002991static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002992set_exc_info(PyThreadState *tstate,
2993 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002994{
Tim Peters7df5e7f2006-05-26 23:14:37 +00002995 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002996 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002997
Tim Peters7df5e7f2006-05-26 23:14:37 +00002998 assert(type != NULL);
2999 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003000 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003001 assert(frame->f_exc_value == NULL);
3002 assert(frame->f_exc_traceback == NULL);
3003 /* This frame didn't catch an exception before. */
3004 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003005 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003006 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003007 Py_INCREF(Py_None);
3008 tstate->exc_type = Py_None;
3009 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003010 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003011 Py_XINCREF(tstate->exc_value);
3012 Py_XINCREF(tstate->exc_traceback);
3013 frame->f_exc_type = tstate->exc_type;
3014 frame->f_exc_value = tstate->exc_value;
3015 frame->f_exc_traceback = tstate->exc_traceback;
3016 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003017 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003018 tmp_type = tstate->exc_type;
3019 tmp_value = tstate->exc_value;
3020 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003021 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003022 Py_XINCREF(value);
3023 Py_XINCREF(tb);
3024 tstate->exc_type = type;
3025 tstate->exc_value = value;
3026 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003027 Py_XDECREF(tmp_type);
3028 Py_XDECREF(tmp_value);
3029 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003030 /* For b/w compatibility */
3031 PySys_SetObject("exc_type", type);
3032 PySys_SetObject("exc_value", value);
3033 PySys_SetObject("exc_traceback", tb);
3034}
3035
Fredrik Lundh7a830892006-05-27 10:39:48 +00003036static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003037reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003038{
3039 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003040 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003041
3042 /* It's a precondition that the thread state's frame caught an
3043 * exception -- verify in a debug build.
3044 */
3045 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003046 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003047 assert(frame != NULL);
3048 assert(frame->f_exc_type != NULL);
3049
3050 /* Copy the frame's exception info back to the thread state. */
3051 tmp_type = tstate->exc_type;
3052 tmp_value = tstate->exc_value;
3053 tmp_tb = tstate->exc_traceback;
3054 Py_INCREF(frame->f_exc_type);
3055 Py_XINCREF(frame->f_exc_value);
3056 Py_XINCREF(frame->f_exc_traceback);
3057 tstate->exc_type = frame->f_exc_type;
3058 tstate->exc_value = frame->f_exc_value;
3059 tstate->exc_traceback = frame->f_exc_traceback;
3060 Py_XDECREF(tmp_type);
3061 Py_XDECREF(tmp_value);
3062 Py_XDECREF(tmp_tb);
3063
3064 /* For b/w compatibility */
3065 PySys_SetObject("exc_type", frame->f_exc_type);
3066 PySys_SetObject("exc_value", frame->f_exc_value);
3067 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
3068
3069 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003070 tmp_type = frame->f_exc_type;
3071 tmp_value = frame->f_exc_value;
3072 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003073 frame->f_exc_type = NULL;
3074 frame->f_exc_value = NULL;
3075 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003076 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003077 Py_XDECREF(tmp_value);
3078 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003079}
3080
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003081/* Logic for the raise statement (too complicated for inlining).
3082 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003083static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003084do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003085{
Guido van Rossumd295f121998-04-09 21:39:57 +00003086 if (type == NULL) {
3087 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003088 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003089 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3090 value = tstate->exc_value;
3091 tb = tstate->exc_traceback;
3092 Py_XINCREF(type);
3093 Py_XINCREF(value);
3094 Py_XINCREF(tb);
3095 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003096
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003097 /* We support the following forms of raise:
3098 raise <class>, <classinstance>
3099 raise <class>, <argument tuple>
3100 raise <class>, None
3101 raise <class>, <argument>
3102 raise <classinstance>, None
3103 raise <string>, <object>
3104 raise <string>, None
3105
3106 An omitted second argument is the same as None.
3107
3108 In addition, raise <tuple>, <anything> is the same as
3109 raising the tuple's first item (and it better have one!);
3110 this rule is applied recursively.
3111
3112 Finally, an optional third argument can be supplied, which
3113 gives the traceback to be substituted (useful when
3114 re-raising an exception after examining it). */
3115
3116 /* First, check the traceback argument, replacing None with
3117 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003118 if (tb == Py_None) {
3119 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003120 tb = NULL;
3121 }
3122 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003123 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003124 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003125 goto raise_error;
3126 }
3127
3128 /* Next, replace a missing value with None */
3129 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003130 value = Py_None;
3131 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003132 }
3133
3134 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003135 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3136 PyObject *tmp = type;
3137 type = PyTuple_GET_ITEM(type, 0);
3138 Py_INCREF(type);
3139 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003140 }
3141
Brett Cannon129bd522007-01-30 21:34:36 +00003142 if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003143 PyErr_NormalizeException(&type, &value, &tb);
3144
Brett Cannonbf364092006-03-01 04:25:17 +00003145 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003146 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003147 if (value != Py_None) {
3148 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003149 "instance exception may not have a separate value");
3150 goto raise_error;
3151 }
3152 else {
3153 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003154 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003155 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003156 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003157 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003158 }
3159 }
3160 else {
3161 /* Not something you can raise. You get an exception
3162 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003163 PyErr_Format(PyExc_TypeError,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003164 "exceptions must be classes or instances, not %s",
3165 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003166 goto raise_error;
3167 }
Guido van Rossum504153d2008-03-18 04:26:48 +00003168
3169 assert(PyExceptionClass_Check(type));
3170 if (Py_Py3kWarningFlag && PyClass_Check(type)) {
Benjamin Peterson9f4f4812008-04-27 03:01:45 +00003171 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003172 "exceptions must derive from BaseException "
3173 "in 3.x", 1) < 0)
Guido van Rossum504153d2008-03-18 04:26:48 +00003174 goto raise_error;
3175 }
3176
Guido van Rossumb209a111997-04-29 18:18:01 +00003177 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003178 if (tb == NULL)
3179 return WHY_EXCEPTION;
3180 else
3181 return WHY_RERAISE;
3182 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003183 Py_XDECREF(value);
3184 Py_XDECREF(type);
3185 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003186 return WHY_EXCEPTION;
3187}
3188
Tim Petersd6d010b2001-06-21 02:49:55 +00003189/* Iterate v argcnt times and store the results on the stack (via decreasing
3190 sp). Return 1 for success, 0 if error. */
3191
Fredrik Lundh7a830892006-05-27 10:39:48 +00003192static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003193unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003194{
Tim Petersd6d010b2001-06-21 02:49:55 +00003195 int i = 0;
3196 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003197 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003198
Tim Petersd6d010b2001-06-21 02:49:55 +00003199 assert(v != NULL);
3200
3201 it = PyObject_GetIter(v);
3202 if (it == NULL)
3203 goto Error;
3204
3205 for (; i < argcnt; i++) {
3206 w = PyIter_Next(it);
3207 if (w == NULL) {
3208 /* Iterator done, via error or exhaustion. */
3209 if (!PyErr_Occurred()) {
3210 PyErr_Format(PyExc_ValueError,
3211 "need more than %d value%s to unpack",
3212 i, i == 1 ? "" : "s");
3213 }
3214 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003215 }
3216 *--sp = w;
3217 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003218
3219 /* We better have exhausted the iterator now. */
3220 w = PyIter_Next(it);
3221 if (w == NULL) {
3222 if (PyErr_Occurred())
3223 goto Error;
3224 Py_DECREF(it);
3225 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003226 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003227 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003228 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003229 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003230Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003231 for (; i > 0; i--, sp++)
3232 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003233 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003234 return 0;
3235}
3236
3237
Guido van Rossum96a42c81992-01-12 02:29:51 +00003238#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003239static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003240prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003241{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003242 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003243 if (PyObject_Print(v, stdout, 0) != 0)
3244 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003245 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003246 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003247}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003248#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003249
Fredrik Lundh7a830892006-05-27 10:39:48 +00003250static void
Fred Drake5755ce62001-06-27 19:19:46 +00003251call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003252{
Guido van Rossumb209a111997-04-29 18:18:01 +00003253 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003254 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003255 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003256 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003257 value = Py_None;
3258 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003259 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003260 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003261 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003262 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003263 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003264 }
Fred Drake5755ce62001-06-27 19:19:46 +00003265 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003266 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003267 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003268 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003269 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003270 Py_XDECREF(type);
3271 Py_XDECREF(value);
3272 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003273 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003274}
3275
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003276static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003277call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003278 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003279{
3280 PyObject *type, *value, *traceback;
3281 int err;
3282 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003283 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003284 if (err == 0)
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003285 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003286 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003287 return 0;
3288 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003289 else {
3290 Py_XDECREF(type);
3291 Py_XDECREF(value);
3292 Py_XDECREF(traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003293 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003294 }
3295}
3296
Fredrik Lundh7a830892006-05-27 10:39:48 +00003297static int
Fred Drake5755ce62001-06-27 19:19:46 +00003298call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3299 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003300{
Fred Drake5755ce62001-06-27 19:19:46 +00003301 register PyThreadState *tstate = frame->f_tstate;
3302 int result;
3303 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003304 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003305 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003306 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003307 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003308 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3309 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003310 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003311 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003312}
3313
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003314PyObject *
3315_PyEval_CallTracing(PyObject *func, PyObject *args)
3316{
3317 PyFrameObject *frame = PyEval_GetFrame();
3318 PyThreadState *tstate = frame->f_tstate;
3319 int save_tracing = tstate->tracing;
3320 int save_use_tracing = tstate->use_tracing;
3321 PyObject *result;
3322
3323 tstate->tracing = 0;
3324 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3325 || (tstate->c_profilefunc != NULL));
3326 result = PyObject_Call(func, args, NULL);
3327 tstate->tracing = save_tracing;
3328 tstate->use_tracing = save_use_tracing;
3329 return result;
3330}
3331
Fredrik Lundh7a830892006-05-27 10:39:48 +00003332static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003333maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003334 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3335 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003336{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003337 int result = 0;
3338
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003339 /* If the last instruction executed isn't in the current
3340 instruction window, reset the window. If the last
3341 instruction happens to fall at the start of a line or if it
3342 represents a jump backwards, call the trace function.
3343 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003344 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003345 int line;
3346 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003347
Thomas Woutersae406c62007-09-19 17:27:43 +00003348 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3349 &bounds);
3350 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003351 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003352 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003353 PyTrace_LINE, Py_None);
Thomas Woutersae406c62007-09-19 17:27:43 +00003354 }
3355 *instr_lb = bounds.ap_lower;
3356 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003357 }
Armin Rigobf57a142004-03-22 19:24:58 +00003358 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003359 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003360 }
3361 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003362 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003363}
3364
Fred Drake5755ce62001-06-27 19:19:46 +00003365void
3366PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003367{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003368 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003369 PyObject *temp = tstate->c_profileobj;
3370 Py_XINCREF(arg);
3371 tstate->c_profilefunc = NULL;
3372 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003373 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003374 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003375 Py_XDECREF(temp);
3376 tstate->c_profilefunc = func;
3377 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003378 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003379 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003380}
3381
3382void
3383PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3384{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003385 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003386 PyObject *temp = tstate->c_traceobj;
3387 Py_XINCREF(arg);
3388 tstate->c_tracefunc = NULL;
3389 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003390 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003391 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003392 Py_XDECREF(temp);
3393 tstate->c_tracefunc = func;
3394 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003395 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003396 tstate->use_tracing = ((func != NULL)
3397 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003398}
3399
Guido van Rossumb209a111997-04-29 18:18:01 +00003400PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003401PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003402{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003403 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003404 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003405 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003406 else
3407 return current_frame->f_builtins;
3408}
3409
Guido van Rossumb209a111997-04-29 18:18:01 +00003410PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003411PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003412{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003413 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003414 if (current_frame == NULL)
3415 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003416 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003417 return current_frame->f_locals;
3418}
3419
Guido van Rossumb209a111997-04-29 18:18:01 +00003420PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003421PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003422{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003423 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003424 if (current_frame == NULL)
3425 return NULL;
3426 else
3427 return current_frame->f_globals;
3428}
3429
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003430PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003431PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003432{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003433 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003434 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003435}
3436
Guido van Rossum6135a871995-01-09 17:53:26 +00003437int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003438PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003439{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003440 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003441 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003442}
3443
Guido van Rossumbe270261997-05-22 22:26:18 +00003444int
Tim Peters5ba58662001-07-16 02:29:45 +00003445PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003446{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003447 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003448 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003449
3450 if (current_frame != NULL) {
3451 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003452 const int compilerflags = codeflags & PyCF_MASK;
3453 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003454 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003455 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003456 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003457#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003458 if (codeflags & CO_GENERATOR_ALLOWED) {
3459 result = 1;
3460 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3461 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003462#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003463 }
3464 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003465}
3466
3467int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003468Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003469{
Guido van Rossumb209a111997-04-29 18:18:01 +00003470 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003471 if (f == NULL)
3472 return 0;
3473 if (!PyFile_SoftSpace(f, 0))
3474 return 0;
3475 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003476}
3477
Guido van Rossum3f5da241990-12-20 15:06:42 +00003478
Guido van Rossum681d79a1995-07-18 14:51:37 +00003479/* External interface to call any callable object.
3480 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003481
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003482#undef PyEval_CallObject
3483/* for backward compatibility: export this interface */
3484
Guido van Rossumb209a111997-04-29 18:18:01 +00003485PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003486PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003487{
Guido van Rossumb209a111997-04-29 18:18:01 +00003488 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003489}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003490#define PyEval_CallObject(func,arg) \
3491 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003492
Guido van Rossumb209a111997-04-29 18:18:01 +00003493PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003494PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003495{
Jeremy Hylton52820442001-01-03 23:52:36 +00003496 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003497
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003498 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003499 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003500 if (arg == NULL)
3501 return NULL;
3502 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003503 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003504 PyErr_SetString(PyExc_TypeError,
3505 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003506 return NULL;
3507 }
3508 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003509 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003510
Guido van Rossumb209a111997-04-29 18:18:01 +00003511 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003512 PyErr_SetString(PyExc_TypeError,
3513 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003514 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003515 return NULL;
3516 }
3517
Tim Peters6d6c1a32001-08-02 04:15:00 +00003518 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003519 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003520 return result;
3521}
3522
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003523const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003524PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003525{
3526 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003527 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003528 else if (PyFunction_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003529 return PyString_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003530 else if (PyCFunction_Check(func))
3531 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3532 else if (PyClass_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003533 return PyString_AsString(((PyClassObject*)func)->cl_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003534 else if (PyInstance_Check(func)) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003535 return PyString_AsString(
Jeremy Hylton512a2372001-04-11 13:52:29 +00003536 ((PyInstanceObject*)func)->in_class->cl_name);
3537 } else {
3538 return func->ob_type->tp_name;
3539 }
3540}
3541
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003542const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003543PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003544{
3545 if (PyMethod_Check(func))
3546 return "()";
3547 else if (PyFunction_Check(func))
3548 return "()";
3549 else if (PyCFunction_Check(func))
3550 return "()";
3551 else if (PyClass_Check(func))
3552 return " constructor";
3553 else if (PyInstance_Check(func)) {
3554 return " instance";
3555 } else {
3556 return " object";
3557 }
3558}
3559
Fredrik Lundh7a830892006-05-27 10:39:48 +00003560static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003561err_args(PyObject *func, int flags, int nargs)
3562{
3563 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003564 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003565 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003566 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003567 nargs);
3568 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003569 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003570 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003571 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003572 nargs);
3573}
3574
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003575#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003576if (tstate->use_tracing && tstate->c_profilefunc) { \
3577 if (call_trace(tstate->c_profilefunc, \
3578 tstate->c_profileobj, \
3579 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003580 func)) { \
3581 x = NULL; \
3582 } \
3583 else { \
3584 x = call; \
3585 if (tstate->c_profilefunc != NULL) { \
3586 if (x == NULL) { \
3587 call_trace_protected(tstate->c_profilefunc, \
3588 tstate->c_profileobj, \
3589 tstate->frame, PyTrace_C_EXCEPTION, \
3590 func); \
3591 /* XXX should pass (type, value, tb) */ \
3592 } else { \
3593 if (call_trace(tstate->c_profilefunc, \
3594 tstate->c_profileobj, \
3595 tstate->frame, PyTrace_C_RETURN, \
3596 func)) { \
3597 Py_DECREF(x); \
3598 x = NULL; \
3599 } \
3600 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003601 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003602 } \
3603} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003604 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003605 }
3606
Fredrik Lundh7a830892006-05-27 10:39:48 +00003607static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003608call_function(PyObject ***pp_stack, int oparg
3609#ifdef WITH_TSC
3610 , uint64* pintr0, uint64* pintr1
3611#endif
3612 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003613{
3614 int na = oparg & 0xff;
3615 int nk = (oparg>>8) & 0xff;
3616 int n = na + 2 * nk;
3617 PyObject **pfunc = (*pp_stack) - n - 1;
3618 PyObject *func = *pfunc;
3619 PyObject *x, *w;
3620
Jeremy Hylton985eba52003-02-05 23:13:00 +00003621 /* Always dispatch PyCFunction first, because these are
3622 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003623 */
3624 if (PyCFunction_Check(func) && nk == 0) {
3625 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003626 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003627
3628 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003629 if (flags & (METH_NOARGS | METH_O)) {
3630 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3631 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003632 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003633 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003634 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003635 else if (flags & METH_O && na == 1) {
3636 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003637 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003638 Py_DECREF(arg);
3639 }
3640 else {
3641 err_args(func, flags, na);
3642 x = NULL;
3643 }
3644 }
3645 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003646 PyObject *callargs;
3647 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003648 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003649 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003650 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003651 Py_XDECREF(callargs);
3652 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003653 } else {
3654 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3655 /* optimize access to bound methods */
3656 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003657 PCALL(PCALL_METHOD);
3658 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003659 Py_INCREF(self);
3660 func = PyMethod_GET_FUNCTION(func);
3661 Py_INCREF(func);
3662 Py_DECREF(*pfunc);
3663 *pfunc = self;
3664 na++;
3665 n++;
3666 } else
3667 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003668 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003669 if (PyFunction_Check(func))
3670 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003671 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003672 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003673 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003674 Py_DECREF(func);
3675 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003676
Armin Rigod34fa522006-03-28 19:10:40 +00003677 /* Clear the stack of the function object. Also removes
3678 the arguments in case they weren't consumed already
3679 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003680 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003681 while ((*pp_stack) > pfunc) {
3682 w = EXT_POP(*pp_stack);
3683 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003684 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003685 }
3686 return x;
3687}
3688
Jeremy Hylton192690e2002-08-16 18:36:11 +00003689/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003690 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003691 For the simplest case -- a function that takes only positional
3692 arguments and is called with only positional arguments -- it
3693 inlines the most primitive frame setup code from
3694 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3695 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003696*/
3697
Fredrik Lundh7a830892006-05-27 10:39:48 +00003698static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003699fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003700{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003701 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003702 PyObject *globals = PyFunction_GET_GLOBALS(func);
3703 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3704 PyObject **d = NULL;
3705 int nd = 0;
3706
Jeremy Hylton985eba52003-02-05 23:13:00 +00003707 PCALL(PCALL_FUNCTION);
3708 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003709 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003710 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3711 PyFrameObject *f;
3712 PyObject *retval = NULL;
3713 PyThreadState *tstate = PyThreadState_GET();
3714 PyObject **fastlocals, **stack;
3715 int i;
3716
3717 PCALL(PCALL_FASTER_FUNCTION);
3718 assert(globals != NULL);
3719 /* XXX Perhaps we should create a specialized
3720 PyFrame_New() that doesn't take locals, but does
3721 take builtins without sanity checking them.
3722 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003723 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003724 f = PyFrame_New(tstate, co, globals, NULL);
3725 if (f == NULL)
3726 return NULL;
3727
3728 fastlocals = f->f_localsplus;
3729 stack = (*pp_stack) - n;
3730
3731 for (i = 0; i < n; i++) {
3732 Py_INCREF(*stack);
3733 fastlocals[i] = *stack++;
3734 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003735 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003736 ++tstate->recursion_depth;
3737 Py_DECREF(f);
3738 --tstate->recursion_depth;
3739 return retval;
3740 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003741 if (argdefs != NULL) {
3742 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimese93237d2007-12-19 02:37:44 +00003743 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003744 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003745 return PyEval_EvalCodeEx(co, globals,
3746 (PyObject *)NULL, (*pp_stack)-n, na,
3747 (*pp_stack)-2*nk, nk, d, nd,
3748 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003749}
3750
Fredrik Lundh7a830892006-05-27 10:39:48 +00003751static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003752update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3753 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003754{
3755 PyObject *kwdict = NULL;
3756 if (orig_kwdict == NULL)
3757 kwdict = PyDict_New();
3758 else {
3759 kwdict = PyDict_Copy(orig_kwdict);
3760 Py_DECREF(orig_kwdict);
3761 }
3762 if (kwdict == NULL)
3763 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003764 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003765 int err;
3766 PyObject *value = EXT_POP(*pp_stack);
3767 PyObject *key = EXT_POP(*pp_stack);
3768 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003769 PyErr_Format(PyExc_TypeError,
3770 "%.200s%s got multiple values "
3771 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003772 PyEval_GetFuncName(func),
3773 PyEval_GetFuncDesc(func),
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003774 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003775 Py_DECREF(key);
3776 Py_DECREF(value);
3777 Py_DECREF(kwdict);
3778 return NULL;
3779 }
3780 err = PyDict_SetItem(kwdict, key, value);
3781 Py_DECREF(key);
3782 Py_DECREF(value);
3783 if (err) {
3784 Py_DECREF(kwdict);
3785 return NULL;
3786 }
3787 }
3788 return kwdict;
3789}
3790
Fredrik Lundh7a830892006-05-27 10:39:48 +00003791static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003792update_star_args(int nstack, int nstar, PyObject *stararg,
3793 PyObject ***pp_stack)
3794{
3795 PyObject *callargs, *w;
3796
3797 callargs = PyTuple_New(nstack + nstar);
3798 if (callargs == NULL) {
3799 return NULL;
3800 }
3801 if (nstar) {
3802 int i;
3803 for (i = 0; i < nstar; i++) {
3804 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3805 Py_INCREF(a);
3806 PyTuple_SET_ITEM(callargs, nstack + i, a);
3807 }
3808 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003809 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003810 w = EXT_POP(*pp_stack);
3811 PyTuple_SET_ITEM(callargs, nstack, w);
3812 }
3813 return callargs;
3814}
3815
Fredrik Lundh7a830892006-05-27 10:39:48 +00003816static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003817load_args(PyObject ***pp_stack, int na)
3818{
3819 PyObject *args = PyTuple_New(na);
3820 PyObject *w;
3821
3822 if (args == NULL)
3823 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003824 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003825 w = EXT_POP(*pp_stack);
3826 PyTuple_SET_ITEM(args, na, w);
3827 }
3828 return args;
3829}
3830
Fredrik Lundh7a830892006-05-27 10:39:48 +00003831static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003832do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3833{
3834 PyObject *callargs = NULL;
3835 PyObject *kwdict = NULL;
3836 PyObject *result = NULL;
3837
3838 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003839 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003840 if (kwdict == NULL)
3841 goto call_fail;
3842 }
3843 callargs = load_args(pp_stack, na);
3844 if (callargs == NULL)
3845 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003846#ifdef CALL_PROFILE
3847 /* At this point, we have to look at the type of func to
3848 update the call stats properly. Do it here so as to avoid
3849 exposing the call stats machinery outside ceval.c
3850 */
3851 if (PyFunction_Check(func))
3852 PCALL(PCALL_FUNCTION);
3853 else if (PyMethod_Check(func))
3854 PCALL(PCALL_METHOD);
3855 else if (PyType_Check(func))
3856 PCALL(PCALL_TYPE);
3857 else
3858 PCALL(PCALL_OTHER);
3859#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003860 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003861 call_fail:
3862 Py_XDECREF(callargs);
3863 Py_XDECREF(kwdict);
3864 return result;
3865}
3866
Fredrik Lundh7a830892006-05-27 10:39:48 +00003867static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003868ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3869{
3870 int nstar = 0;
3871 PyObject *callargs = NULL;
3872 PyObject *stararg = NULL;
3873 PyObject *kwdict = NULL;
3874 PyObject *result = NULL;
3875
3876 if (flags & CALL_FLAG_KW) {
3877 kwdict = EXT_POP(*pp_stack);
Georg Brandl2134e752007-05-21 20:34:16 +00003878 if (!PyDict_Check(kwdict)) {
3879 PyObject *d;
3880 d = PyDict_New();
3881 if (d == NULL)
3882 goto ext_call_fail;
3883 if (PyDict_Update(d, kwdict) != 0) {
3884 Py_DECREF(d);
3885 /* PyDict_Update raises attribute
3886 * error (percolated from an attempt
3887 * to get 'keys' attribute) instead of
3888 * a type error if its second argument
3889 * is not a mapping.
3890 */
3891 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
3892 PyErr_Format(PyExc_TypeError,
3893 "%.200s%.200s argument after ** "
3894 "must be a mapping, not %.200s",
3895 PyEval_GetFuncName(func),
3896 PyEval_GetFuncDesc(func),
3897 kwdict->ob_type->tp_name);
3898 }
3899 goto ext_call_fail;
3900 }
3901 Py_DECREF(kwdict);
3902 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00003903 }
3904 }
3905 if (flags & CALL_FLAG_VAR) {
3906 stararg = EXT_POP(*pp_stack);
3907 if (!PyTuple_Check(stararg)) {
3908 PyObject *t = NULL;
3909 t = PySequence_Tuple(stararg);
3910 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003911 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3912 PyErr_Format(PyExc_TypeError,
Georg Brandl2134e752007-05-21 20:34:16 +00003913 "%.200s%.200s argument after * "
3914 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003915 PyEval_GetFuncName(func),
Georg Brandl2134e752007-05-21 20:34:16 +00003916 PyEval_GetFuncDesc(func),
3917 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003918 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003919 goto ext_call_fail;
3920 }
3921 Py_DECREF(stararg);
3922 stararg = t;
3923 }
3924 nstar = PyTuple_GET_SIZE(stararg);
3925 }
3926 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003927 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003928 if (kwdict == NULL)
3929 goto ext_call_fail;
3930 }
3931 callargs = update_star_args(na, nstar, stararg, pp_stack);
3932 if (callargs == NULL)
3933 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003934#ifdef CALL_PROFILE
3935 /* At this point, we have to look at the type of func to
3936 update the call stats properly. Do it here so as to avoid
3937 exposing the call stats machinery outside ceval.c
3938 */
3939 if (PyFunction_Check(func))
3940 PCALL(PCALL_FUNCTION);
3941 else if (PyMethod_Check(func))
3942 PCALL(PCALL_METHOD);
3943 else if (PyType_Check(func))
3944 PCALL(PCALL_TYPE);
3945 else
3946 PCALL(PCALL_OTHER);
3947#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003948 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersae406c62007-09-19 17:27:43 +00003949ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003950 Py_XDECREF(callargs);
3951 Py_XDECREF(kwdict);
3952 Py_XDECREF(stararg);
3953 return result;
3954}
3955
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003956/* Extract a slice index from a PyInt or PyLong or an object with the
3957 nb_index slot defined, and store in *pi.
3958 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3959 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 +00003960 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003961*/
Tim Petersb5196382001-12-16 19:44:20 +00003962/* Note: If v is NULL, return success without storing into *pi. This
3963 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3964 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003965*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003966int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003967_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003968{
Tim Petersb5196382001-12-16 19:44:20 +00003969 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003970 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00003971 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003972 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
3973 however, it looks like it should be AsSsize_t.
3974 There should be a comment here explaining why.
3975 */
3976 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00003977 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003978 else if (PyIndex_Check(v)) {
3979 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003980 if (x == -1 && PyErr_Occurred())
3981 return 0;
3982 }
3983 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003984 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003985 "slice indices must be integers or "
3986 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003987 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003988 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003989 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003990 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003991 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003992}
3993
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003994#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003995#define ISINDEX(x) ((x) == NULL || \
3996 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00003997
Fredrik Lundh7a830892006-05-27 10:39:48 +00003998static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003999apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004000{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004001 PyTypeObject *tp = u->ob_type;
4002 PySequenceMethods *sq = tp->tp_as_sequence;
4003
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004004 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004005 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004006 if (!_PyEval_SliceIndex(v, &ilow))
4007 return NULL;
4008 if (!_PyEval_SliceIndex(w, &ihigh))
4009 return NULL;
4010 return PySequence_GetSlice(u, ilow, ihigh);
4011 }
4012 else {
4013 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00004014 if (slice != NULL) {
4015 PyObject *res = PyObject_GetItem(u, slice);
4016 Py_DECREF(slice);
4017 return res;
4018 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00004019 else
4020 return NULL;
4021 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004022}
Guido van Rossum3f5da241990-12-20 15:06:42 +00004023
Fredrik Lundh7a830892006-05-27 10:39:48 +00004024static int
Guido van Rossumac7be682001-01-17 15:42:30 +00004025assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
4026 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004027{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004028 PyTypeObject *tp = u->ob_type;
4029 PySequenceMethods *sq = tp->tp_as_sequence;
4030
Georg Brandl0fca97a2007-03-05 22:28:08 +00004031 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004032 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004033 if (!_PyEval_SliceIndex(v, &ilow))
4034 return -1;
4035 if (!_PyEval_SliceIndex(w, &ihigh))
4036 return -1;
4037 if (x == NULL)
4038 return PySequence_DelSlice(u, ilow, ihigh);
4039 else
4040 return PySequence_SetSlice(u, ilow, ihigh, x);
4041 }
4042 else {
4043 PyObject *slice = PySlice_New(v, w, NULL);
4044 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00004045 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004046 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00004047 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00004048 else
Guido van Rossum354797c2001-12-03 19:45:06 +00004049 res = PyObject_DelItem(u, slice);
4050 Py_DECREF(slice);
4051 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004052 }
4053 else
4054 return -1;
4055 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004056}
4057
Guido van Rossum04edb522008-03-18 02:49:46 +00004058#define Py3kExceptionClass_Check(x) \
4059 (PyType_Check((x)) && \
4060 PyType_FastSubclass((PyTypeObject*)(x), Py_TPFLAGS_BASE_EXC_SUBCLASS))
4061
4062#define CANNOT_CATCH_MSG "catching classes that don't inherit from " \
Georg Brandld5b635f2008-03-25 08:29:14 +00004063 "BaseException is not allowed in 3.x"
Guido van Rossum04edb522008-03-18 02:49:46 +00004064
Fredrik Lundh7a830892006-05-27 10:39:48 +00004065static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004066cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004067{
Guido van Rossumac7be682001-01-17 15:42:30 +00004068 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004069 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00004070 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00004071 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004072 break;
4073 case PyCmp_IS_NOT:
4074 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004075 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004076 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004077 res = PySequence_Contains(w, v);
4078 if (res < 0)
4079 return NULL;
4080 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004081 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00004082 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004083 if (res < 0)
4084 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004085 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004086 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004087 case PyCmp_EXC_MATCH:
Brett Cannon129bd522007-01-30 21:34:36 +00004088 if (PyTuple_Check(w)) {
4089 Py_ssize_t i, length;
4090 length = PyTuple_Size(w);
4091 for (i = 0; i < length; i += 1) {
4092 PyObject *exc = PyTuple_GET_ITEM(w, i);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004093 if (PyString_Check(exc)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004094 int ret_val;
4095 ret_val = PyErr_WarnEx(
Thomas Wouterse2176022007-09-20 17:35:10 +00004096 PyExc_DeprecationWarning,
4097 "catching of string "
4098 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004099 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004100 return NULL;
4101 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004102 else if (Py_Py3kWarningFlag &&
4103 !PyTuple_Check(exc) &&
4104 !Py3kExceptionClass_Check(exc))
Guido van Rossum04edb522008-03-18 02:49:46 +00004105 {
4106 int ret_val;
4107 ret_val = PyErr_WarnEx(
4108 PyExc_DeprecationWarning,
4109 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004110 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004111 return NULL;
4112 }
Brett Cannon129bd522007-01-30 21:34:36 +00004113 }
4114 }
4115 else {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004116 if (PyString_Check(w)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004117 int ret_val;
4118 ret_val = PyErr_WarnEx(
4119 PyExc_DeprecationWarning,
4120 "catching of string "
Thomas Wouterse2176022007-09-20 17:35:10 +00004121 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004122 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004123 return NULL;
4124 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004125 else if (Py_Py3kWarningFlag &&
4126 !PyTuple_Check(w) &&
4127 !Py3kExceptionClass_Check(w))
Guido van Rossum04edb522008-03-18 02:49:46 +00004128 {
4129 int ret_val;
4130 ret_val = PyErr_WarnEx(
4131 PyExc_DeprecationWarning,
4132 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004133 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004134 return NULL;
4135 }
Brett Cannon129bd522007-01-30 21:34:36 +00004136 }
Barry Warsaw4249f541997-08-22 21:26:19 +00004137 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004138 break;
4139 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00004140 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004141 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004142 v = res ? Py_True : Py_False;
4143 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004144 return v;
4145}
4146
Fredrik Lundh7a830892006-05-27 10:39:48 +00004147static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00004148import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004149{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004150 PyObject *x;
4151
4152 x = PyObject_GetAttr(v, name);
4153 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004154 PyErr_Format(PyExc_ImportError,
4155 "cannot import name %.230s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004156 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004157 }
Thomas Wouters52152252000-08-17 22:55:00 +00004158 return x;
4159}
Guido van Rossumac7be682001-01-17 15:42:30 +00004160
Fredrik Lundh7a830892006-05-27 10:39:48 +00004161static int
Thomas Wouters52152252000-08-17 22:55:00 +00004162import_all_from(PyObject *locals, PyObject *v)
4163{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004164 PyObject *all = PyObject_GetAttrString(v, "__all__");
4165 PyObject *dict, *name, *value;
4166 int skip_leading_underscores = 0;
4167 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004168
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004169 if (all == NULL) {
4170 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4171 return -1; /* Unexpected error */
4172 PyErr_Clear();
4173 dict = PyObject_GetAttrString(v, "__dict__");
4174 if (dict == NULL) {
4175 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4176 return -1;
4177 PyErr_SetString(PyExc_ImportError,
4178 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004179 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004180 }
4181 all = PyMapping_Keys(dict);
4182 Py_DECREF(dict);
4183 if (all == NULL)
4184 return -1;
4185 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004186 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004187
4188 for (pos = 0, err = 0; ; pos++) {
4189 name = PySequence_GetItem(all, pos);
4190 if (name == NULL) {
4191 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4192 err = -1;
4193 else
4194 PyErr_Clear();
4195 break;
4196 }
4197 if (skip_leading_underscores &&
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004198 PyString_Check(name) &&
4199 PyString_AS_STRING(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004200 {
4201 Py_DECREF(name);
4202 continue;
4203 }
4204 value = PyObject_GetAttr(v, name);
4205 if (value == NULL)
4206 err = -1;
Armin Rigo70370852006-11-29 21:59:22 +00004207 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004208 err = PyDict_SetItem(locals, name, value);
Armin Rigo70370852006-11-29 21:59:22 +00004209 else
4210 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004211 Py_DECREF(name);
4212 Py_XDECREF(value);
4213 if (err != 0)
4214 break;
4215 }
4216 Py_DECREF(all);
4217 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004218}
4219
Fredrik Lundh7a830892006-05-27 10:39:48 +00004220static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004221build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004222{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004223 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004224
4225 if (PyDict_Check(methods))
4226 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004227 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004228 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004229 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4230 base = PyTuple_GET_ITEM(bases, 0);
4231 metaclass = PyObject_GetAttrString(base, "__class__");
4232 if (metaclass == NULL) {
4233 PyErr_Clear();
4234 metaclass = (PyObject *)base->ob_type;
4235 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004236 }
4237 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004238 else {
4239 PyObject *g = PyEval_GetGlobals();
4240 if (g != NULL && PyDict_Check(g))
4241 metaclass = PyDict_GetItemString(g, "__metaclass__");
4242 if (metaclass == NULL)
4243 metaclass = (PyObject *) &PyClass_Type;
4244 Py_INCREF(metaclass);
4245 }
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004246 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods,
Thomas Woutersae406c62007-09-19 17:27:43 +00004247 NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004248 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004249 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004250 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004251 in a base that was not a class (such the random module
4252 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004253 by augmenting the error message with more information.*/
4254
4255 PyObject *ptype, *pvalue, *ptraceback;
4256
4257 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004258 if (PyString_Check(pvalue)) {
Raymond Hettingercfc31922004-09-16 16:41:57 +00004259 PyObject *newmsg;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004260 newmsg = PyString_FromFormat(
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004261 "Error when calling the metaclass bases\n"
Thomas Woutersae406c62007-09-19 17:27:43 +00004262 " %s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004263 PyString_AS_STRING(pvalue));
Raymond Hettingercfc31922004-09-16 16:41:57 +00004264 if (newmsg != NULL) {
4265 Py_DECREF(pvalue);
4266 pvalue = newmsg;
4267 }
4268 }
4269 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004270 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004271 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004272}
4273
Fredrik Lundh7a830892006-05-27 10:39:48 +00004274static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004275exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4276 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004277{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004278 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004279 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004280 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004281
Guido van Rossumb209a111997-04-29 18:18:01 +00004282 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4283 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004284 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004285 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004286 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004287 locals = PyTuple_GetItem(prog, 2);
4288 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004289 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004290 if (globals == Py_None) {
4291 globals = PyEval_GetGlobals();
4292 if (locals == Py_None) {
4293 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004294 plain = 1;
4295 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004296 if (!globals || !locals) {
4297 PyErr_SetString(PyExc_SystemError,
4298 "globals and locals cannot be NULL");
4299 return -1;
4300 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004301 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004302 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004303 locals = globals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004304 if (!PyString_Check(prog) &&
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004305 !PyUnicode_Check(prog) &&
Guido van Rossumb209a111997-04-29 18:18:01 +00004306 !PyCode_Check(prog) &&
4307 !PyFile_Check(prog)) {
4308 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004309 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004310 return -1;
4311 }
Fred Drake661ea262000-10-24 19:57:45 +00004312 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004313 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004314 "exec: arg 2 must be a dictionary or None");
4315 return -1;
4316 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004317 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004318 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004319 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004320 return -1;
4321 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004322 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004323 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004324 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004325 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4326 PyErr_SetString(PyExc_TypeError,
4327 "code object passed to exec may not contain free variables");
4328 return -1;
4329 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004330 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004331 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004332 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004333 FILE *fp = PyFile_AsFile(prog);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004334 char *name = PyString_AsString(PyFile_Name(prog));
Jeremy Hylton714b1122007-02-25 16:01:58 +00004335 PyCompilerFlags cf;
Thomas Woutersae406c62007-09-19 17:27:43 +00004336 if (name == NULL)
4337 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004338 cf.cf_flags = 0;
4339 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004340 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004341 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004342 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004343 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004344 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004345 }
4346 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004347 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004348 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004349 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004350 cf.cf_flags = 0;
4351#ifdef Py_USING_UNICODE
4352 if (PyUnicode_Check(prog)) {
4353 tmp = PyUnicode_AsUTF8String(prog);
4354 if (tmp == NULL)
4355 return -1;
4356 prog = tmp;
4357 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4358 }
4359#endif
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004360 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004361 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004362 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004363 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004364 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004365 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004366 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004367 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004368 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004369 if (plain)
4370 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004371 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004372 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004373 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004374 return 0;
4375}
Guido van Rossum24c13741995-02-14 09:42:43 +00004376
Fredrik Lundh7a830892006-05-27 10:39:48 +00004377static void
Paul Prescode68140d2000-08-30 20:25:01 +00004378format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4379{
4380 char *obj_str;
4381
4382 if (!obj)
4383 return;
4384
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004385 obj_str = PyString_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00004386 if (!obj_str)
4387 return;
4388
4389 PyErr_Format(exc, format_str, obj_str);
4390}
Guido van Rossum950361c1997-01-24 13:49:28 +00004391
Fredrik Lundh7a830892006-05-27 10:39:48 +00004392static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004393string_concatenate(PyObject *v, PyObject *w,
4394 PyFrameObject *f, unsigned char *next_instr)
4395{
4396 /* This function implements 'variable += expr' when both arguments
4397 are strings. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004398 Py_ssize_t v_len = PyString_GET_SIZE(v);
4399 Py_ssize_t w_len = PyString_GET_SIZE(w);
Armin Rigo97ff0472006-08-09 15:37:26 +00004400 Py_ssize_t new_len = v_len + w_len;
4401 if (new_len < 0) {
4402 PyErr_SetString(PyExc_OverflowError,
4403 "strings are too large to concat");
4404 return NULL;
4405 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004406
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004407 if (v->ob_refcnt == 2) {
4408 /* In the common case, there are 2 references to the value
4409 * stored in 'variable' when the += is performed: one on the
Thomas Wouterse2176022007-09-20 17:35:10 +00004410 * value stack (in 'v') and one still stored in the
4411 * 'variable'. We try to delete the variable now to reduce
4412 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004413 */
4414 switch (*next_instr) {
4415 case STORE_FAST:
4416 {
4417 int oparg = PEEKARG();
4418 PyObject **fastlocals = f->f_localsplus;
4419 if (GETLOCAL(oparg) == v)
4420 SETLOCAL(oparg, NULL);
4421 break;
4422 }
4423 case STORE_DEREF:
4424 {
Thomas Wouterse2176022007-09-20 17:35:10 +00004425 PyObject **freevars = (f->f_localsplus +
4426 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004427 PyObject *c = freevars[PEEKARG()];
4428 if (PyCell_GET(c) == v)
4429 PyCell_Set(c, NULL);
4430 break;
4431 }
4432 case STORE_NAME:
4433 {
4434 PyObject *names = f->f_code->co_names;
4435 PyObject *name = GETITEM(names, PEEKARG());
4436 PyObject *locals = f->f_locals;
4437 if (PyDict_CheckExact(locals) &&
4438 PyDict_GetItem(locals, name) == v) {
4439 if (PyDict_DelItem(locals, name) != 0) {
4440 PyErr_Clear();
4441 }
4442 }
4443 break;
4444 }
4445 }
4446 }
4447
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004448 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004449 /* Now we own the last reference to 'v', so we can resize it
4450 * in-place.
4451 */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004452 if (_PyString_Resize(&v, new_len) != 0) {
4453 /* XXX if _PyString_Resize() fails, 'v' has been
Thomas Wouterse2176022007-09-20 17:35:10 +00004454 * deallocated so it cannot be put back into
4455 * 'variable'. The MemoryError is raised when there
4456 * is no value in 'variable', which might (very
4457 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004458 */
4459 return NULL;
4460 }
4461 /* copy 'w' into the newly allocated area of 'v' */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004462 memcpy(PyString_AS_STRING(v) + v_len,
4463 PyString_AS_STRING(w), w_len);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004464 return v;
4465 }
4466 else {
4467 /* When in-place resizing is not an option. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004468 PyString_Concat(&v, w);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004469 return v;
4470 }
4471}
4472
Guido van Rossum950361c1997-01-24 13:49:28 +00004473#ifdef DYNAMIC_EXECUTION_PROFILE
4474
Fredrik Lundh7a830892006-05-27 10:39:48 +00004475static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004476getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004477{
4478 int i;
4479 PyObject *l = PyList_New(256);
4480 if (l == NULL) return NULL;
4481 for (i = 0; i < 256; i++) {
4482 PyObject *x = PyInt_FromLong(a[i]);
4483 if (x == NULL) {
4484 Py_DECREF(l);
4485 return NULL;
4486 }
4487 PyList_SetItem(l, i, x);
4488 }
4489 for (i = 0; i < 256; i++)
4490 a[i] = 0;
4491 return l;
4492}
4493
4494PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004495_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004496{
4497#ifndef DXPAIRS
4498 return getarray(dxp);
4499#else
4500 int i;
4501 PyObject *l = PyList_New(257);
4502 if (l == NULL) return NULL;
4503 for (i = 0; i < 257; i++) {
4504 PyObject *x = getarray(dxpairs[i]);
4505 if (x == NULL) {
4506 Py_DECREF(l);
4507 return NULL;
4508 }
4509 PyList_SetItem(l, i, x);
4510 }
4511 return l;
4512#endif
4513}
4514
4515#endif