blob: b42444da82db882821ee1e9c2cd946a25ed2ee6a [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
Thomas Wouters477c8d52006-05-27 19:21:47 +00009/* enable more aggressive intra-module optimizations, where available */
10#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
Thomas Wouters477c8d52006-05-27 19:21:47 +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 Wouters8ce81f72007-09-20 18:22:40 +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
37static void
38ppc_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
Thomas Wouters477c8d52006-05-27 19:21:47 +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
Thomas Wouters477c8d52006-05-27 19:21:47 +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
Thomas Wouters477c8d52006-05-27 19:21:47 +000090static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091#else
Thomas Wouters477c8d52006-05-27 19:21:47 +000092static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +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 Wouters8ce81f72007-09-20 18:22:40 +000097static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
98 PyObject *);
Thomas Wouters477c8d52006-05-27 19:21:47 +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
Guido van Rossumc2e20742006-02-27 22:32:47 +0000105static int lltrace;
Tim Petersdbd9ba62000-07-09 03:09:57 +0000106static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000107#endif
Fred Drake5755ce62001-06-27 19:19:46 +0000108static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
109 int, PyObject *);
Fred Drake4ec5d562001-10-04 19:26:43 +0000110static void call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000111 PyFrameObject *, int, PyObject *);
Fred Drake5755ce62001-06-27 19:19:46 +0000112static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
Tim Peters8a5c3c72004-04-05 19:36:21 +0000113static 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
Thomas Wouters477c8d52006-05-27 19:21:47 +0000116static PyObject * cmp_outcome(int, PyObject *, PyObject *);
117static PyObject * import_from(PyObject *, PyObject *);
Thomas Wouters52152252000-08-17 22:55:00 +0000118static int import_all_from(PyObject *, PyObject *);
Tim Petersdbd9ba62000-07-09 03:09:57 +0000119static void set_exc_info(PyThreadState *, PyObject *, PyObject *, PyObject *);
120static void reset_exc_info(PyThreadState *);
Neal Norwitzda059e32007-08-26 05:33:45 +0000121static void format_exc_check_arg(PyObject *, const char *, PyObject *);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000122static PyObject * unicode_concatenate(PyObject *, PyObject *,
123 PyFrameObject *, unsigned char *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000124
Paul Prescode68140d2000-08-30 20:25:01 +0000125#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000126 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000127#define GLOBAL_NAME_ERROR_MSG \
128 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000129#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000130 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000131#define UNBOUNDFREE_ERROR_MSG \
132 "free variable '%.200s' referenced before assignment" \
133 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000134
Guido van Rossum950361c1997-01-24 13:49:28 +0000135/* Dynamic execution profile */
136#ifdef DYNAMIC_EXECUTION_PROFILE
137#ifdef DXPAIRS
138static long dxpairs[257][256];
139#define dxp dxpairs[256]
140#else
141static long dxp[256];
142#endif
143#endif
144
Jeremy Hylton985eba52003-02-05 23:13:00 +0000145/* Function call profile */
146#ifdef CALL_PROFILE
147#define PCALL_NUM 11
148static int pcall[PCALL_NUM];
149
150#define PCALL_ALL 0
151#define PCALL_FUNCTION 1
152#define PCALL_FAST_FUNCTION 2
153#define PCALL_FASTER_FUNCTION 3
154#define PCALL_METHOD 4
155#define PCALL_BOUND_METHOD 5
156#define PCALL_CFUNCTION 6
157#define PCALL_TYPE 7
158#define PCALL_GENERATOR 8
159#define PCALL_OTHER 9
160#define PCALL_POP 10
161
162/* Notes about the statistics
163
164 PCALL_FAST stats
165
166 FAST_FUNCTION means no argument tuple needs to be created.
167 FASTER_FUNCTION means that the fast-path frame setup code is used.
168
169 If there is a method call where the call can be optimized by changing
170 the argument tuple and calling the function directly, it gets recorded
171 twice.
172
173 As a result, the relationship among the statistics appears to be
174 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
175 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
176 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
177 PCALL_METHOD > PCALL_BOUND_METHOD
178*/
179
180#define PCALL(POS) pcall[POS]++
181
182PyObject *
183PyEval_GetCallStats(PyObject *self)
184{
Thomas Wouters89f507f2006-12-13 04:49:30 +0000185 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000186 pcall[0], pcall[1], pcall[2], pcall[3],
187 pcall[4], pcall[5], pcall[6], pcall[7],
Thomas Wouters89f507f2006-12-13 04:49:30 +0000188 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000189}
190#else
191#define PCALL(O)
192
193PyObject *
194PyEval_GetCallStats(PyObject *self)
195{
196 Py_INCREF(Py_None);
197 return Py_None;
198}
199#endif
200
Tim Peters5ca576e2001-06-18 22:08:13 +0000201
Guido van Rossume59214e1994-08-30 08:01:59 +0000202#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000203
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000204#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000205#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000206#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000207#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000208
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000209static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Guido van Rossuma9672091994-09-14 13:31:22 +0000210static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000211
Tim Peters7f468f22004-10-11 02:40:51 +0000212int
213PyEval_ThreadsInitialized(void)
214{
215 return interpreter_lock != 0;
216}
217
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000218void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000219PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000220{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000221 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000222 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000223 interpreter_lock = PyThread_allocate_lock();
224 PyThread_acquire_lock(interpreter_lock, 1);
225 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000226}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000227
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000228void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000229PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000230{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000231 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000232}
233
234void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000235PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000237 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000238}
239
240void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000241PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000242{
243 if (tstate == NULL)
244 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000245 /* Check someone has called PyEval_InitThreads() to create the lock */
246 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000247 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000248 if (PyThreadState_Swap(tstate) != NULL)
249 Py_FatalError(
250 "PyEval_AcquireThread: non-NULL old thread state");
251}
252
253void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000254PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000255{
256 if (tstate == NULL)
257 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
258 if (PyThreadState_Swap(NULL) != tstate)
259 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000260 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000261}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000262
263/* This function is called from PyOS_AfterFork to ensure that newly
264 created child processes don't hold locks referring to threads which
265 are not running in the child process. (This could also be done using
266 pthread_atfork mechanism, at least for the pthreads implementation.) */
267
268void
269PyEval_ReInitThreads(void)
270{
271 if (!interpreter_lock)
272 return;
273 /*XXX Can't use PyThread_free_lock here because it does too
274 much error-checking. Doing this cleanly would require
275 adding a new function to each thread_*.h. Instead, just
276 create a new lock and waste a little bit of memory */
277 interpreter_lock = PyThread_allocate_lock();
278 PyThread_acquire_lock(interpreter_lock, 1);
279 main_thread = PyThread_get_thread_ident();
280}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000281#endif
282
Guido van Rossumff4949e1992-08-05 19:58:53 +0000283/* Functions save_thread and restore_thread are always defined so
284 dynamically loaded modules needn't be compiled separately for use
285 with and without threads: */
286
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000287PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000288PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000289{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000290 PyThreadState *tstate = PyThreadState_Swap(NULL);
291 if (tstate == NULL)
292 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000293#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000294 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000295 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000296#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000297 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000298}
299
300void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000301PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000302{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000303 if (tstate == NULL)
304 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000305#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000306 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000307 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000308 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000309 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000310 }
311#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000312 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000313}
314
315
Guido van Rossuma9672091994-09-14 13:31:22 +0000316/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
317 signal handlers or Mac I/O completion routines) can schedule calls
318 to a function to be called synchronously.
319 The synchronous function is called with one void* argument.
320 It should return 0 for success or -1 for failure -- failure should
321 be accompanied by an exception.
322
323 If registry succeeds, the registry function returns 0; if it fails
324 (e.g. due to too many pending calls) it returns -1 (without setting
325 an exception condition).
326
327 Note that because registry may occur from within signal handlers,
328 or other asynchronous events, calling malloc() is unsafe!
329
330#ifdef WITH_THREAD
331 Any thread can schedule pending calls, but only the main thread
332 will execute them.
333#endif
334
335 XXX WARNING! ASYNCHRONOUSLY EXECUTING CODE!
336 There are two possible race conditions:
337 (1) nested asynchronous registry calls;
338 (2) registry calls made while pending calls are being processed.
339 While (1) is very unlikely, (2) is a real possibility.
340 The current code is safe against (2), but not against (1).
341 The safety against (2) is derived from the fact that only one
342 thread (the main thread) ever takes things out of the queue.
Guido van Rossuma9672091994-09-14 13:31:22 +0000343
Guido van Rossuma027efa1997-05-05 20:56:21 +0000344 XXX Darn! With the advent of thread state, we should have an array
345 of pending calls per thread in the thread state! Later...
346*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000347
Guido van Rossuma9672091994-09-14 13:31:22 +0000348#define NPENDINGCALLS 32
349static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000350 int (*func)(void *);
351 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000352} pendingcalls[NPENDINGCALLS];
353static volatile int pendingfirst = 0;
354static volatile int pendinglast = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000355static volatile int things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000356
357int
Thomas Wouters334fb892000-07-25 12:56:38 +0000358Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000359{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000360 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000361 int i, j;
362 /* XXX Begin critical section */
363 /* XXX If you want this to be safe against nested
364 XXX asynchronous calls, you'll have to work harder! */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000365 if (busy)
366 return -1;
367 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000368 i = pendinglast;
369 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000370 if (j == pendingfirst) {
371 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000372 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000373 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000374 pendingcalls[i].func = func;
375 pendingcalls[i].arg = arg;
376 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000377
378 _Py_Ticker = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000379 things_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000380 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000381 /* XXX End critical section */
382 return 0;
383}
384
Guido van Rossum180d7b41994-09-29 09:45:57 +0000385int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000386Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000387{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000388 static int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000389#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000390 if (main_thread && PyThread_get_thread_ident() != main_thread)
Guido van Rossuma9672091994-09-14 13:31:22 +0000391 return 0;
392#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000393 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000394 return 0;
395 busy = 1;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000396 things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000397 for (;;) {
398 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000399 int (*func)(void *);
400 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000401 i = pendingfirst;
402 if (i == pendinglast)
403 break; /* Queue empty */
404 func = pendingcalls[i].func;
405 arg = pendingcalls[i].arg;
406 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000407 if (func(arg) < 0) {
408 busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000409 things_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000410 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000411 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000412 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000413 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000414 return 0;
415}
416
417
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000418/* The interpreter's recursion limit */
419
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000420#ifndef Py_DEFAULT_RECURSION_LIMIT
421#define Py_DEFAULT_RECURSION_LIMIT 1000
422#endif
423static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
424int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000425
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000426int
427Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000428{
429 return recursion_limit;
430}
431
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000432void
433Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000434{
435 recursion_limit = new_limit;
Thomas Woutersce272b62007-09-19 21:19:28 +0000436 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000437}
438
Armin Rigo2b3eb402003-10-28 12:05:48 +0000439/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
440 if the recursion_depth reaches _Py_CheckRecursionLimit.
441 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
442 to guarantee that _Py_CheckRecursiveCall() is regularly called.
443 Without USE_STACKCHECK, there is no need for this. */
444int
445_Py_CheckRecursiveCall(char *where)
446{
447 PyThreadState *tstate = PyThreadState_GET();
448
449#ifdef USE_STACKCHECK
450 if (PyOS_CheckStack()) {
451 --tstate->recursion_depth;
452 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
453 return -1;
454 }
455#endif
Martin v. Löwis5b222132007-06-10 09:51:05 +0000456 if (tstate->recursion_critical)
457 /* Somebody asked that we don't check for recursion. */
458 return 0;
459 if (tstate->overflowed) {
460 if (tstate->recursion_depth > recursion_limit + 50) {
461 /* Overflowing while handling an overflow. Give up. */
462 Py_FatalError("Cannot recover from stack overflow.");
463 }
464 return 0;
465 }
Armin Rigo2b3eb402003-10-28 12:05:48 +0000466 if (tstate->recursion_depth > recursion_limit) {
467 --tstate->recursion_depth;
Martin v. Löwis5b222132007-06-10 09:51:05 +0000468 tstate->overflowed = 1;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000469 PyErr_Format(PyExc_RuntimeError,
470 "maximum recursion depth exceeded%s",
471 where);
472 return -1;
473 }
Thomas Woutersce272b62007-09-19 21:19:28 +0000474 _Py_CheckRecursionLimit = recursion_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000475 return 0;
476}
477
Guido van Rossum374a9221991-04-04 10:40:29 +0000478/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000479enum why_code {
480 WHY_NOT = 0x0001, /* No error */
481 WHY_EXCEPTION = 0x0002, /* Exception occurred */
482 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
483 WHY_RETURN = 0x0008, /* 'return' statement */
484 WHY_BREAK = 0x0010, /* 'break' statement */
485 WHY_CONTINUE = 0x0020, /* 'continue' statement */
486 WHY_YIELD = 0x0040 /* 'yield' operator */
487};
Guido van Rossum374a9221991-04-04 10:40:29 +0000488
Collin Winter828f04a2007-08-31 00:04:24 +0000489static enum why_code do_raise(PyObject *, PyObject *);
Guido van Rossum0368b722007-05-11 16:50:42 +0000490static int unpack_iterable(PyObject *, int, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000491
Skip Montanarod581d772002-09-03 20:10:45 +0000492/* for manipulating the thread switch and periodic "stuff" - used to be
493 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000494int _Py_CheckInterval = 100;
495volatile int _Py_Ticker = 100;
Guido van Rossum374a9221991-04-04 10:40:29 +0000496
Guido van Rossumb209a111997-04-29 18:18:01 +0000497PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000498PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000499{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000500 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000501 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000502 (PyObject **)NULL, 0,
503 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000504 (PyObject **)NULL, 0,
Guido van Rossum4f72a782006-10-27 23:31:49 +0000505 NULL, NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000506}
507
508
509/* Interpreter main loop */
510
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000511PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000512PyEval_EvalFrame(PyFrameObject *f) {
513 /* This is for backward compatibility with extension modules that
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000514 used this API; core interpreter code should call
515 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000516 return PyEval_EvalFrameEx(f, 0);
517}
518
519PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000520PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000521{
Guido van Rossum950361c1997-01-24 13:49:28 +0000522#ifdef DXPAIRS
523 int lastopcode = 0;
524#endif
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000525 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000526 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000527 register int opcode; /* Current opcode */
528 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000529 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000530 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000531 register PyObject *x; /* Result object -- NULL if error */
532 register PyObject *v; /* Temporary objects popped off stack */
533 register PyObject *w;
534 register PyObject *u;
535 register PyObject *t;
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000536 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000537 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000538 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000539 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000540
Tim Peters8a5c3c72004-04-05 19:36:21 +0000541 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000542
543 not (instr_lb <= current_bytecode_offset < instr_ub)
544
Tim Peters8a5c3c72004-04-05 19:36:21 +0000545 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000546 initial values are such as to make this false the first
547 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000548 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000549
Guido van Rossumd076c731998-10-07 19:42:25 +0000550 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000551 PyObject *names;
552 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000553#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000554 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000555 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000556#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000557
Neal Norwitza81d2202002-07-14 00:27:26 +0000558/* Tuple access macros */
559
560#ifndef Py_DEBUG
561#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
562#else
563#define GETITEM(v, i) PyTuple_GetItem((v), (i))
564#endif
565
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000566#ifdef WITH_TSC
567/* Use Pentium timestamp counter to mark certain events:
568 inst0 -- beginning of switch statement for opcode dispatch
569 inst1 -- end of switch statement (may be skipped)
570 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000571 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000572 (may be skipped)
573 intr1 -- beginning of long interruption
574 intr2 -- end of long interruption
575
576 Many opcodes call out to helper C functions. In some cases, the
577 time in those functions should be counted towards the time for the
578 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
579 calls another Python function; there's no point in charge all the
580 bytecode executed by the called function to the caller.
581
582 It's hard to make a useful judgement statically. In the presence
583 of operator overloading, it's impossible to tell if a call will
584 execute new Python code or not.
585
586 It's a case-by-case judgement. I'll use intr1 for the following
587 cases:
588
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000589 IMPORT_STAR
590 IMPORT_FROM
591 CALL_FUNCTION (and friends)
592
593 */
594 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
595 int ticked = 0;
596
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000597 READ_TIMESTAMP(inst0);
598 READ_TIMESTAMP(inst1);
599 READ_TIMESTAMP(loop0);
600 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000601
602 /* shut up the compiler */
603 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000604#endif
605
Guido van Rossum374a9221991-04-04 10:40:29 +0000606/* Code access macros */
607
Martin v. Löwis18e16552006-02-15 17:27:45 +0000608#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000609#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000610#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000611#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000612#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000613#define JUMPBY(x) (next_instr += (x))
614
Raymond Hettingerf606f872003-03-16 03:11:04 +0000615/* OpCode prediction macros
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000616 Some opcodes tend to come in pairs thus making it possible to
617 predict the second code when the first is run. For example,
618 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
619 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000620
621 Verifying the prediction costs a single high-speed test of register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000622 variable against a constant. If the pairing was good, then the
Raymond Hettingerf606f872003-03-16 03:11:04 +0000623 processor has a high likelihood of making its own successful branch
624 prediction which results in a nearly zero overhead transition to the
625 next opcode.
626
627 A successful prediction saves a trip through the eval-loop including
628 its two unpredictable branches, the HASARG test and the switch-case.
Raymond Hettingera7216982004-02-08 19:59:27 +0000629
Tim Peters8a5c3c72004-04-05 19:36:21 +0000630 If collecting opcode statistics, turn off prediction so that
631 statistics are accurately maintained (the predictions bypass
Raymond Hettingera7216982004-02-08 19:59:27 +0000632 the opcode frequency counter updates).
Raymond Hettingerf606f872003-03-16 03:11:04 +0000633*/
634
Raymond Hettingera7216982004-02-08 19:59:27 +0000635#ifdef DYNAMIC_EXECUTION_PROFILE
636#define PREDICT(op) if (0) goto PRED_##op
637#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000638#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000639#endif
640
Raymond Hettingerf606f872003-03-16 03:11:04 +0000641#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000642#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000643
Guido van Rossum374a9221991-04-04 10:40:29 +0000644/* Stack manipulation macros */
645
Martin v. Löwis18e16552006-02-15 17:27:45 +0000646/* The stack can grow at most MAXINT deep, as co_nlocals and
647 co_stacksize are ints. */
648#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000649#define EMPTY() (STACK_LEVEL() == 0)
650#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000651#define SECOND() (stack_pointer[-2])
652#define THIRD() (stack_pointer[-3])
653#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000654#define SET_TOP(v) (stack_pointer[-1] = (v))
655#define SET_SECOND(v) (stack_pointer[-2] = (v))
656#define SET_THIRD(v) (stack_pointer[-3] = (v))
657#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000658#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000659#define BASIC_PUSH(v) (*stack_pointer++ = (v))
660#define BASIC_POP() (*--stack_pointer)
661
Guido van Rossum96a42c81992-01-12 02:29:51 +0000662#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000663#define PUSH(v) { (void)(BASIC_PUSH(v), \
664 lltrace && prtrace(TOP(), "push")); \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000665 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000666#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
667 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000668#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
669 lltrace && prtrace(TOP(), "stackadj")); \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000670 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000671#define EXT_POP(STACK_POINTER) (lltrace && prtrace((STACK_POINTER)[-1], \
672 "ext_pop"), *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000673#else
674#define PUSH(v) BASIC_PUSH(v)
675#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000676#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000677#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000678#endif
679
Guido van Rossum681d79a1995-07-18 14:51:37 +0000680/* Local variable macros */
681
682#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000683
684/* The SETLOCAL() macro must not DECREF the local variable in-place and
685 then store the new value; it must copy the old value to a temporary
686 value, then store the new value, and then DECREF the temporary value.
687 This is because it is possible that during the DECREF the frame is
688 accessed by other code (e.g. a __del__ method or gc.collect()) and the
689 variable would be pointing to already-freed memory. */
690#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
691 GETLOCAL(i) = value; \
692 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000693
Guido van Rossuma027efa1997-05-05 20:56:21 +0000694/* Start of code */
695
Tim Peters5ca576e2001-06-18 22:08:13 +0000696 if (f == NULL)
697 return NULL;
698
Armin Rigo1d313ab2003-10-25 14:33:09 +0000699 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000700 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000701 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000702
Tim Peters5ca576e2001-06-18 22:08:13 +0000703 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000704
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000705 if (tstate->use_tracing) {
706 if (tstate->c_tracefunc != NULL) {
707 /* tstate->c_tracefunc, if defined, is a
708 function that will be called on *every* entry
709 to a code block. Its return value, if not
710 None, is a function that will be called at
711 the start of each executed line of code.
712 (Actually, the function must return itself
713 in order to continue tracing.) The trace
714 functions are called with three arguments:
715 a pointer to the current frame, a string
716 indicating why the function is called, and
717 an argument which depends on the situation.
718 The global trace function is also called
719 whenever an exception is detected. */
720 if (call_trace(tstate->c_tracefunc, tstate->c_traceobj,
721 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000722 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000723 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000724 }
725 }
726 if (tstate->c_profilefunc != NULL) {
727 /* Similar for c_profilefunc, except it needn't
728 return itself and isn't called for "line" events */
729 if (call_trace(tstate->c_profilefunc,
730 tstate->c_profileobj,
731 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000732 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000733 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000734 }
735 }
736 }
737
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000738 co = f->f_code;
739 names = co->co_names;
740 consts = co->co_consts;
741 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000742 freevars = f->f_localsplus + co->co_nlocals;
Brett Cannonc9371d42005-06-25 08:23:41 +0000743 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000744 /* An explanation is in order for the next line.
745
746 f->f_lasti now refers to the index of the last instruction
747 executed. You might think this was obvious from the name, but
748 this wasn't always true before 2.3! PyFrame_New now sets
749 f->f_lasti to -1 (i.e. the index *before* the first instruction)
750 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000751 does work. Promise.
752
753 When the PREDICT() macros are enabled, some opcode pairs follow in
754 direct succession without updating f->f_lasti. A successful
755 prediction effectively links the two codes together as if they
756 were a single new opcode; accordingly,f->f_lasti will point to
757 the first code in the pair (for instance, GET_ITER followed by
758 FOR_ITER is effectively a single opcode and f->f_lasti will point
759 at to the beginning of the combined pair.)
760 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000761 next_instr = first_instr + f->f_lasti + 1;
762 stack_pointer = f->f_stacktop;
763 assert(stack_pointer != NULL);
764 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
765
Tim Peters5ca576e2001-06-18 22:08:13 +0000766#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000767 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000768#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000769#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum00bc0e02007-10-15 02:52:41 +0000770 filename = PyUnicode_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000771#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000772
Guido van Rossum374a9221991-04-04 10:40:29 +0000773 why = WHY_NOT;
774 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000775 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000776 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000777
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000778 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000779 why = WHY_EXCEPTION;
780 goto on_error;
781 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000782
Guido van Rossum374a9221991-04-04 10:40:29 +0000783 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000784#ifdef WITH_TSC
785 if (inst1 == 0) {
786 /* Almost surely, the opcode executed a break
787 or a continue, preventing inst1 from being set
788 on the way out of the loop.
789 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000790 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000791 loop1 = inst1;
792 }
793 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
794 intr0, intr1);
795 ticked = 0;
796 inst1 = 0;
797 intr0 = 0;
798 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000799 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000800#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000801 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000802 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000803
Guido van Rossuma027efa1997-05-05 20:56:21 +0000804 /* Do periodic things. Doing this every time through
805 the loop would add too much overhead, so we do it
806 only every Nth instruction. We also do it if
807 ``things_to_do'' is set, i.e. when an asynchronous
808 event needs attention (e.g. a signal handler or
809 async I/O handler); see Py_AddPendingCall() and
810 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000811
Skip Montanarod581d772002-09-03 20:10:45 +0000812 if (--_Py_Ticker < 0) {
Thomas Woutersce272b62007-09-19 21:19:28 +0000813 if (*next_instr == SETUP_FINALLY) {
814 /* Make the last opcode before
815 a try: finally: block uninterruptable. */
816 goto fast_next_opcode;
817 }
Skip Montanarod581d772002-09-03 20:10:45 +0000818 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000819 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000820#ifdef WITH_TSC
821 ticked = 1;
822#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000823 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000824 if (Py_MakePendingCalls() < 0) {
825 why = WHY_EXCEPTION;
826 goto on_error;
827 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000828 if (things_to_do)
829 /* MakePendingCalls() didn't succeed.
830 Force early re-execution of this
831 "periodic" code, possibly after
832 a thread switch */
833 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000834 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000835#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000836 if (interpreter_lock) {
837 /* Give another thread a chance */
838
Guido van Rossum25ce5661997-08-02 03:10:38 +0000839 if (PyThreadState_Swap(NULL) != tstate)
840 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000841 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000842
843 /* Other threads may run now */
844
Guido van Rossum65d5b571998-12-21 19:32:43 +0000845 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000846 if (PyThreadState_Swap(tstate) != NULL)
847 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000848
849 /* Check for thread interrupts */
850
851 if (tstate->async_exc != NULL) {
852 x = tstate->async_exc;
853 tstate->async_exc = NULL;
854 PyErr_SetNone(x);
855 Py_DECREF(x);
856 why = WHY_EXCEPTION;
857 goto on_error;
858 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000859 }
860#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000861 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000862
Neil Schemenauer63543862002-02-17 19:10:14 +0000863 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000864 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000865
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000866 /* line-by-line tracing support */
867
868 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
869 /* see maybe_call_line_trace
870 for expository comments */
871 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000872
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000873 err = maybe_call_line_trace(tstate->c_tracefunc,
874 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000875 f, &instr_lb, &instr_ub,
876 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000877 /* Reload possibly changed frame fields */
878 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000879 if (f->f_stacktop != NULL) {
880 stack_pointer = f->f_stacktop;
881 f->f_stacktop = NULL;
882 }
883 if (err) {
884 /* trace function raised an exception */
885 goto on_error;
886 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000887 }
888
889 /* Extract opcode and argument */
890
Guido van Rossum374a9221991-04-04 10:40:29 +0000891 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000892 oparg = 0; /* allows oparg to be stored in a register because
893 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000894 if (HAS_ARG(opcode))
895 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000896 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000897#ifdef DYNAMIC_EXECUTION_PROFILE
898#ifdef DXPAIRS
899 dxpairs[lastopcode][opcode]++;
900 lastopcode = opcode;
901#endif
902 dxp[opcode]++;
903#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000904
Guido van Rossum96a42c81992-01-12 02:29:51 +0000905#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000906 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000907
Guido van Rossum96a42c81992-01-12 02:29:51 +0000908 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000909 if (HAS_ARG(opcode)) {
910 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000911 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000912 }
913 else {
914 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000915 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000916 }
917 }
918#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000919
Guido van Rossum374a9221991-04-04 10:40:29 +0000920 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000921 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000922
Guido van Rossum374a9221991-04-04 10:40:29 +0000923 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000924
Guido van Rossum374a9221991-04-04 10:40:29 +0000925 /* BEWARE!
926 It is essential that any operation that fails sets either
927 x to NULL, err to nonzero, or why to anything but WHY_NOT,
928 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000929
Guido van Rossum374a9221991-04-04 10:40:29 +0000930 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000931
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000932 case NOP:
933 goto fast_next_opcode;
934
Neil Schemenauer63543862002-02-17 19:10:14 +0000935 case LOAD_FAST:
936 x = GETLOCAL(oparg);
937 if (x != NULL) {
938 Py_INCREF(x);
939 PUSH(x);
940 goto fast_next_opcode;
941 }
942 format_exc_check_arg(PyExc_UnboundLocalError,
943 UNBOUNDLOCAL_ERROR_MSG,
944 PyTuple_GetItem(co->co_varnames, oparg));
945 break;
946
947 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000948 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000949 Py_INCREF(x);
950 PUSH(x);
951 goto fast_next_opcode;
952
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000953 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000954 case STORE_FAST:
955 v = POP();
956 SETLOCAL(oparg, v);
957 goto fast_next_opcode;
958
Raymond Hettingerf606f872003-03-16 03:11:04 +0000959 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000960 case POP_TOP:
961 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000962 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000963 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000964
Guido van Rossum374a9221991-04-04 10:40:29 +0000965 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000966 v = TOP();
967 w = SECOND();
968 SET_TOP(w);
969 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000970 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000971
Guido van Rossum374a9221991-04-04 10:40:29 +0000972 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000973 v = TOP();
974 w = SECOND();
975 x = THIRD();
976 SET_TOP(w);
977 SET_SECOND(x);
978 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000979 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000980
Thomas Wouters434d0822000-08-24 20:11:32 +0000981 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000982 u = TOP();
983 v = SECOND();
984 w = THIRD();
985 x = FOURTH();
986 SET_TOP(v);
987 SET_SECOND(w);
988 SET_THIRD(x);
989 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000990 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000991
Guido van Rossum374a9221991-04-04 10:40:29 +0000992 case DUP_TOP:
993 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000994 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000995 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000996 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000997
Thomas Wouters434d0822000-08-24 20:11:32 +0000998 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000999 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001000 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001001 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001002 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001003 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001004 STACKADJ(2);
1005 SET_TOP(x);
1006 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001007 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001008 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001009 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001010 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001011 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001012 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001013 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001014 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001015 STACKADJ(3);
1016 SET_TOP(x);
1017 SET_SECOND(w);
1018 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001019 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001020 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001021 Py_FatalError("invalid argument to DUP_TOPX"
1022 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001023 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001024
Guido van Rossum374a9221991-04-04 10:40:29 +00001025 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001026 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001027 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001028 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001029 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001030 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001031 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001032
Guido van Rossum374a9221991-04-04 10:40:29 +00001033 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001034 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001035 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001036 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001037 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001038 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001039 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001040
Guido van Rossum374a9221991-04-04 10:40:29 +00001041 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001042 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001043 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001044 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001045 if (err == 0) {
1046 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001047 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001048 continue;
1049 }
1050 else if (err > 0) {
1051 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001052 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001053 err = 0;
1054 continue;
1055 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001056 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001057 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001058
Guido van Rossum7928cd71991-10-24 14:59:31 +00001059 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001060 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001061 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001062 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001063 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001064 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001065 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001066
Guido van Rossum50564e81996-01-12 01:13:16 +00001067 case BINARY_POWER:
1068 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001069 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001070 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001071 Py_DECREF(v);
1072 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001073 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001074 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001075 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001076
Guido van Rossum374a9221991-04-04 10:40:29 +00001077 case BINARY_MULTIPLY:
1078 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001079 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001080 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001081 Py_DECREF(v);
1082 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001083 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001084 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001085 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001086
Tim Peters3caca232001-12-06 06:23:26 +00001087 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001088 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001089 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001090 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001091 Py_DECREF(v);
1092 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001093 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001094 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001095 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001096
Guido van Rossum4668b002001-08-08 05:00:18 +00001097 case BINARY_FLOOR_DIVIDE:
1098 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001099 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001100 x = PyNumber_FloorDivide(v, w);
1101 Py_DECREF(v);
1102 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001103 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001104 if (x != NULL) continue;
1105 break;
1106
Guido van Rossum374a9221991-04-04 10:40:29 +00001107 case BINARY_MODULO:
1108 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001109 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001110 x = PyNumber_Remainder(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 Rossum374a9221991-04-04 10:40:29 +00001117 case BINARY_ADD:
1118 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001119 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001120 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001121 /* INLINE: int + int */
1122 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001123 a = PyInt_AS_LONG(v);
1124 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001125 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001126 if ((i^a) < 0 && (i^b) < 0)
1127 goto slow_add;
1128 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001129 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00001130 else if (PyUnicode_CheckExact(v) &&
1131 PyUnicode_CheckExact(w)) {
1132 x = unicode_concatenate(v, w, f, next_instr);
1133 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001134 goto skip_decref_vx;
1135 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001136 else {
1137 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001138 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001139 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001140 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001141 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001142 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001143 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001144 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001145 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001146
Guido van Rossum374a9221991-04-04 10:40:29 +00001147 case BINARY_SUBTRACT:
1148 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001149 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001150 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001151 /* INLINE: int - int */
1152 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001153 a = PyInt_AS_LONG(v);
1154 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001155 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001156 if ((i^a) < 0 && (i^~b) < 0)
1157 goto slow_sub;
1158 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001159 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001160 else {
1161 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001162 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001163 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001164 Py_DECREF(v);
1165 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001166 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001167 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001168 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001169
Guido van Rossum374a9221991-04-04 10:40:29 +00001170 case BINARY_SUBSCR:
1171 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001172 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001173 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001174 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001175 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001176 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001177 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001178 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001179 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001180 Py_INCREF(x);
1181 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001182 else
1183 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001184 }
1185 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001186 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001187 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001188 Py_DECREF(v);
1189 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001190 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001191 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001192 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001193
Guido van Rossum7928cd71991-10-24 14:59:31 +00001194 case BINARY_LSHIFT:
1195 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001196 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001197 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001198 Py_DECREF(v);
1199 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001200 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001201 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001202 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001203
Guido van Rossum7928cd71991-10-24 14:59:31 +00001204 case BINARY_RSHIFT:
1205 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001206 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001207 x = PyNumber_Rshift(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 Rossum7928cd71991-10-24 14:59:31 +00001212 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001213
Guido van Rossum7928cd71991-10-24 14:59:31 +00001214 case BINARY_AND:
1215 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001216 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001217 x = PyNumber_And(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_XOR:
1225 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001226 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001227 x = PyNumber_Xor(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_OR:
1235 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001236 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001237 x = PyNumber_Or(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;
Thomas Wouters434d0822000-08-24 20:11:32 +00001243
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001244 case LIST_APPEND:
1245 w = POP();
1246 v = POP();
1247 err = PyList_Append(v, w);
1248 Py_DECREF(v);
1249 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001250 if (err == 0) {
1251 PREDICT(JUMP_ABSOLUTE);
1252 continue;
1253 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001254 break;
1255
Nick Coghlan650f0d02007-04-15 12:05:43 +00001256 case SET_ADD:
1257 w = POP();
1258 v = POP();
1259 err = PySet_Add(v, w);
1260 Py_DECREF(v);
1261 Py_DECREF(w);
1262 if (err == 0) {
1263 PREDICT(JUMP_ABSOLUTE);
1264 continue;
1265 }
1266 break;
1267
Thomas Wouters434d0822000-08-24 20:11:32 +00001268 case INPLACE_POWER:
1269 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001270 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001271 x = PyNumber_InPlacePower(v, w, Py_None);
1272 Py_DECREF(v);
1273 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001274 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001275 if (x != NULL) continue;
1276 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001277
Thomas Wouters434d0822000-08-24 20:11:32 +00001278 case INPLACE_MULTIPLY:
1279 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001280 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001281 x = PyNumber_InPlaceMultiply(v, w);
1282 Py_DECREF(v);
1283 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001284 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001285 if (x != NULL) continue;
1286 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001287
Tim Peters54b11912001-12-25 18:49:11 +00001288 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001289 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001290 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001291 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001292 Py_DECREF(v);
1293 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001294 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001295 if (x != NULL) continue;
1296 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001297
Guido van Rossum4668b002001-08-08 05:00:18 +00001298 case INPLACE_FLOOR_DIVIDE:
1299 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001300 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001301 x = PyNumber_InPlaceFloorDivide(v, w);
1302 Py_DECREF(v);
1303 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001304 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001305 if (x != NULL) continue;
1306 break;
1307
Thomas Wouters434d0822000-08-24 20:11:32 +00001308 case INPLACE_MODULO:
1309 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001310 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001311 x = PyNumber_InPlaceRemainder(v, w);
1312 Py_DECREF(v);
1313 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001314 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001315 if (x != NULL) continue;
1316 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001317
Thomas Wouters434d0822000-08-24 20:11:32 +00001318 case INPLACE_ADD:
1319 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001320 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001321 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001322 /* INLINE: int + int */
1323 register long a, b, i;
1324 a = PyInt_AS_LONG(v);
1325 b = PyInt_AS_LONG(w);
1326 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001327 if ((i^a) < 0 && (i^b) < 0)
1328 goto slow_iadd;
1329 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001330 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00001331 else if (PyUnicode_CheckExact(v) &&
1332 PyUnicode_CheckExact(w)) {
1333 x = unicode_concatenate(v, w, f, next_instr);
1334 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001335 goto skip_decref_v;
1336 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001337 else {
1338 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001339 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001340 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001341 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001342 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001343 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001344 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001345 if (x != NULL) continue;
1346 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001347
Thomas Wouters434d0822000-08-24 20:11:32 +00001348 case INPLACE_SUBTRACT:
1349 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001350 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001351 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001352 /* INLINE: int - int */
1353 register long a, b, i;
1354 a = PyInt_AS_LONG(v);
1355 b = PyInt_AS_LONG(w);
1356 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001357 if ((i^a) < 0 && (i^~b) < 0)
1358 goto slow_isub;
1359 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001360 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001361 else {
1362 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001363 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001364 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001365 Py_DECREF(v);
1366 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001367 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001368 if (x != NULL) continue;
1369 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001370
Thomas Wouters434d0822000-08-24 20:11:32 +00001371 case INPLACE_LSHIFT:
1372 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001373 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001374 x = PyNumber_InPlaceLshift(v, w);
1375 Py_DECREF(v);
1376 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001377 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001378 if (x != NULL) continue;
1379 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001380
Thomas Wouters434d0822000-08-24 20:11:32 +00001381 case INPLACE_RSHIFT:
1382 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001383 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001384 x = PyNumber_InPlaceRshift(v, w);
1385 Py_DECREF(v);
1386 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001387 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001388 if (x != NULL) continue;
1389 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001390
Thomas Wouters434d0822000-08-24 20:11:32 +00001391 case INPLACE_AND:
1392 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001393 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001394 x = PyNumber_InPlaceAnd(v, w);
1395 Py_DECREF(v);
1396 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001397 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001398 if (x != NULL) continue;
1399 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001400
Thomas Wouters434d0822000-08-24 20:11:32 +00001401 case INPLACE_XOR:
1402 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001403 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001404 x = PyNumber_InPlaceXor(v, w);
1405 Py_DECREF(v);
1406 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001407 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001408 if (x != NULL) continue;
1409 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001410
Thomas Wouters434d0822000-08-24 20:11:32 +00001411 case INPLACE_OR:
1412 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001413 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001414 x = PyNumber_InPlaceOr(v, w);
1415 Py_DECREF(v);
1416 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001417 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001418 if (x != NULL) continue;
1419 break;
1420
Guido van Rossum374a9221991-04-04 10:40:29 +00001421 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001422 w = TOP();
1423 v = SECOND();
1424 u = THIRD();
1425 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001426 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001427 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001428 Py_DECREF(u);
1429 Py_DECREF(v);
1430 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001431 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001432 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001433
Guido van Rossum374a9221991-04-04 10:40:29 +00001434 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001435 w = TOP();
1436 v = SECOND();
1437 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001438 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001439 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001440 Py_DECREF(v);
1441 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001442 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001443 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001444
Guido van Rossum374a9221991-04-04 10:40:29 +00001445 case PRINT_EXPR:
1446 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001447 w = PySys_GetObject("displayhook");
1448 if (w == NULL) {
1449 PyErr_SetString(PyExc_RuntimeError,
1450 "lost sys.displayhook");
1451 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001452 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001453 }
1454 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001455 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001456 if (x == NULL)
1457 err = -1;
1458 }
1459 if (err == 0) {
1460 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001461 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001462 if (w == NULL)
1463 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001464 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001465 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001466 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001467 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001468
Thomas Wouters434d0822000-08-24 20:11:32 +00001469#ifdef CASE_TOO_BIG
1470 default: switch (opcode) {
1471#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001472 case RAISE_VARARGS:
Collin Winter828f04a2007-08-31 00:04:24 +00001473 v = w = NULL;
Guido van Rossumf10570b1995-07-07 22:53:21 +00001474 switch (oparg) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00001475 case 2:
1476 v = POP(); /* cause */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001477 case 1:
1478 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001479 case 0: /* Fallthrough */
Collin Winter828f04a2007-08-31 00:04:24 +00001480 why = do_raise(w, v);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001481 break;
1482 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001483 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001484 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001485 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001486 break;
1487 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001488 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001489
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001490 case STORE_LOCALS:
1491 x = POP();
1492 v = f->f_locals;
1493 Py_XDECREF(v);
1494 f->f_locals = x;
1495 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001496
Guido van Rossum374a9221991-04-04 10:40:29 +00001497 case RETURN_VALUE:
1498 retval = POP();
1499 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001500 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001501
Tim Peters5ca576e2001-06-18 22:08:13 +00001502 case YIELD_VALUE:
1503 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001504 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001505 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001506 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001507
Guido van Rossum374a9221991-04-04 10:40:29 +00001508 case POP_BLOCK:
1509 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001510 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001511 while (STACK_LEVEL() > b->b_level) {
1512 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001513 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001514 }
1515 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001516 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001517
Guido van Rossum374a9221991-04-04 10:40:29 +00001518 case END_FINALLY:
1519 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001520 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001521 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001522 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001523 if (why == WHY_RETURN ||
1524 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001525 retval = POP();
1526 }
Brett Cannonf74225d2007-02-26 21:10:16 +00001527 else if (PyExceptionClass_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001528 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001529 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001530 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001531 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001532 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001533 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001534 else if (v != Py_None) {
1535 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001536 "'finally' pops bad exception");
1537 why = WHY_EXCEPTION;
1538 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001539 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001540 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001541
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001542 case LOAD_BUILD_CLASS:
1543 x = PyDict_GetItemString(f->f_builtins,
1544 "__build_class__");
1545 if (x == NULL) {
1546 PyErr_SetString(PyExc_ImportError,
1547 "__build_class__ not found");
1548 break;
1549 }
1550 Py_INCREF(x);
1551 PUSH(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001552 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001553
Guido van Rossum374a9221991-04-04 10:40:29 +00001554 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001555 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001556 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001557 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001558 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001559 err = PyDict_SetItem(x, w, v);
1560 else
1561 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001562 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001563 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001564 break;
1565 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001566 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001567 "no locals found when storing %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001568 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001569
Guido van Rossum374a9221991-04-04 10:40:29 +00001570 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001571 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001572 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001573 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001574 format_exc_check_arg(PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001575 NAME_ERROR_MSG,
1576 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001577 break;
1578 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001579 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001580 "no locals when deleting %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001581 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001582
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001583 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001584 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001585 v = POP();
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001586 if (PyTuple_CheckExact(v) &&
1587 PyTuple_GET_SIZE(v) == oparg) {
1588 PyObject **items = \
1589 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001590 while (oparg--) {
1591 w = items[oparg];
1592 Py_INCREF(w);
1593 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001594 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001595 Py_DECREF(v);
1596 continue;
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001597 } else if (PyList_CheckExact(v) &&
1598 PyList_GET_SIZE(v) == oparg) {
1599 PyObject **items = \
1600 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001601 while (oparg--) {
1602 w = items[oparg];
1603 Py_INCREF(w);
1604 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001605 }
Guido van Rossum0368b722007-05-11 16:50:42 +00001606 } else if (unpack_iterable(v, oparg, -1,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001607 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001608 stack_pointer += oparg;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001609 } else {
1610 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001611 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001612 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001613 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001614 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001615
Guido van Rossum0368b722007-05-11 16:50:42 +00001616 case UNPACK_EX:
1617 {
1618 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
1619 v = POP();
1620
1621 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
1622 stack_pointer + totalargs)) {
1623 stack_pointer += totalargs;
1624 } else {
1625 why = WHY_EXCEPTION;
1626 }
1627 Py_DECREF(v);
1628 break;
1629 }
1630
Guido van Rossum374a9221991-04-04 10:40:29 +00001631 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001632 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001633 v = TOP();
1634 u = SECOND();
1635 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001636 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1637 Py_DECREF(v);
1638 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001639 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001640 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001641
Guido van Rossum374a9221991-04-04 10:40:29 +00001642 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001643 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001644 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001645 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1646 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001647 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001648 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001649
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001650 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001651 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001652 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001653 err = PyDict_SetItem(f->f_globals, w, v);
1654 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001655 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001656 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001657
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001658 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001659 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001660 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001661 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001662 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001663 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001664
Guido van Rossum374a9221991-04-04 10:40:29 +00001665 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001666 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001667 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001668 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001669 "no locals when loading %R", w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001670 break;
1671 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001672 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001673 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001674 Py_XINCREF(x);
1675 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001676 else {
1677 x = PyObject_GetItem(v, w);
1678 if (x == NULL && PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001679 if (!PyErr_ExceptionMatches(
1680 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001681 break;
1682 PyErr_Clear();
1683 }
1684 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001685 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001686 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001687 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001688 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001689 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001690 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001691 PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001692 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001693 break;
1694 }
1695 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001696 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001697 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001698 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001699 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001700
Guido van Rossum374a9221991-04-04 10:40:29 +00001701 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001702 w = GETITEM(names, oparg);
Neal Norwitzda059e32007-08-26 05:33:45 +00001703 if (PyUnicode_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001704 /* Inline the PyDict_GetItem() calls.
1705 WARNING: this is an extreme speed hack.
1706 Do not try this at home. */
Neal Norwitzda059e32007-08-26 05:33:45 +00001707 long hash = ((PyUnicodeObject *)w)->hash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001708 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001709 PyDictObject *d;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001710 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001711 d = (PyDictObject *)(f->f_globals);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001712 e = d->ma_lookup(d, w, hash);
1713 if (e == NULL) {
1714 x = NULL;
1715 break;
1716 }
1717 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001718 if (x != NULL) {
1719 Py_INCREF(x);
1720 PUSH(x);
1721 continue;
1722 }
1723 d = (PyDictObject *)(f->f_builtins);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001724 e = d->ma_lookup(d, w, hash);
1725 if (e == NULL) {
1726 x = NULL;
1727 break;
1728 }
1729 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001730 if (x != NULL) {
1731 Py_INCREF(x);
1732 PUSH(x);
1733 continue;
1734 }
1735 goto load_global_error;
1736 }
1737 }
1738 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001739 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001740 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001741 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001742 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001743 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001744 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001745 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001746 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001747 break;
1748 }
1749 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001750 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001751 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001752 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001753
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001754 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001755 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001756 if (x != NULL) {
1757 SETLOCAL(oparg, NULL);
1758 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001759 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001760 format_exc_check_arg(
1761 PyExc_UnboundLocalError,
1762 UNBOUNDLOCAL_ERROR_MSG,
1763 PyTuple_GetItem(co->co_varnames, oparg)
1764 );
1765 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001766
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001767 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001768 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001769 Py_INCREF(x);
1770 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001771 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001772 break;
1773
1774 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001775 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001776 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001777 if (w != NULL) {
1778 PUSH(w);
1779 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001780 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001781 err = -1;
1782 /* Don't stomp existing exception */
1783 if (PyErr_Occurred())
1784 break;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001785 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1786 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001787 oparg);
1788 format_exc_check_arg(
1789 PyExc_UnboundLocalError,
1790 UNBOUNDLOCAL_ERROR_MSG,
1791 v);
1792 } else {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001793 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
1794 PyTuple_GET_SIZE(co->co_cellvars));
1795 format_exc_check_arg(PyExc_NameError,
1796 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001797 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001798 break;
1799
1800 case STORE_DEREF:
1801 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001802 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001803 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001804 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001805 continue;
1806
Guido van Rossum374a9221991-04-04 10:40:29 +00001807 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001808 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001809 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001810 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001811 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001812 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001813 }
1814 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001815 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001816 }
1817 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001818
Guido van Rossum374a9221991-04-04 10:40:29 +00001819 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001820 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001821 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001822 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001823 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001824 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001825 }
1826 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001827 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001828 }
1829 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001830
Guido van Rossum86e58e22006-08-28 15:27:34 +00001831 case BUILD_SET:
1832 x = PySet_New(NULL);
1833 if (x != NULL) {
1834 for (; --oparg >= 0;) {
1835 w = POP();
1836 if (err == 0)
1837 err = PySet_Add(x, w);
1838 Py_DECREF(w);
1839 }
1840 if (err != 0) {
1841 Py_DECREF(x);
1842 break;
1843 }
1844 PUSH(x);
1845 continue;
1846 }
1847 break;
1848
Guido van Rossum374a9221991-04-04 10:40:29 +00001849 case BUILD_MAP:
Guido van Rossumb209a111997-04-29 18:18:01 +00001850 x = PyDict_New();
Guido van Rossum374a9221991-04-04 10:40:29 +00001851 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001852 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001853 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001854
Guido van Rossum374a9221991-04-04 10:40:29 +00001855 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001856 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001857 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001858 x = PyObject_GetAttr(v, w);
1859 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001860 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001861 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001862 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001863
Guido van Rossum374a9221991-04-04 10:40:29 +00001864 case COMPARE_OP:
1865 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001866 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001867 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001868 /* INLINE: cmp(int, int) */
1869 register long a, b;
1870 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001871 a = PyInt_AS_LONG(v);
1872 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001873 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00001874 case PyCmp_LT: res = a < b; break;
1875 case PyCmp_LE: res = a <= b; break;
1876 case PyCmp_EQ: res = a == b; break;
1877 case PyCmp_NE: res = a != b; break;
1878 case PyCmp_GT: res = a > b; break;
1879 case PyCmp_GE: res = a >= b; break;
1880 case PyCmp_IS: res = v == w; break;
1881 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00001882 default: goto slow_compare;
1883 }
1884 x = res ? Py_True : Py_False;
1885 Py_INCREF(x);
1886 }
1887 else {
1888 slow_compare:
1889 x = cmp_outcome(oparg, v, w);
1890 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001891 Py_DECREF(v);
1892 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001893 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001894 if (x == NULL) break;
1895 PREDICT(JUMP_IF_FALSE);
1896 PREDICT(JUMP_IF_TRUE);
1897 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001898
Guido van Rossum374a9221991-04-04 10:40:29 +00001899 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001900 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001901 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001902 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001903 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00001904 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001905 break;
1906 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001907 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001908 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001909 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
1910 w = PyTuple_Pack(5,
1911 w,
1912 f->f_globals,
1913 f->f_locals == NULL ?
1914 Py_None : f->f_locals,
1915 v,
1916 u);
1917 else
1918 w = PyTuple_Pack(4,
1919 w,
1920 f->f_globals,
1921 f->f_locals == NULL ?
1922 Py_None : f->f_locals,
1923 v);
1924 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001925 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001926 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001927 u = POP();
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001928 x = NULL;
1929 break;
1930 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001931 READ_TIMESTAMP(intr0);
Guido van Rossumb209a111997-04-29 18:18:01 +00001932 x = PyEval_CallObject(x, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001933 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001934 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001935 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001936 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001937 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001938
Thomas Wouters52152252000-08-17 22:55:00 +00001939 case IMPORT_STAR:
1940 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001941 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001942 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00001943 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001944 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00001945 break;
1946 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001947 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00001948 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001949 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001950 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00001951 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001952 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001953 break;
Guido van Rossum25831651993-05-19 14:50:45 +00001954
Thomas Wouters52152252000-08-17 22:55:00 +00001955 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00001956 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00001957 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001958 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00001959 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001960 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00001961 PUSH(x);
1962 if (x != NULL) continue;
1963 break;
1964
Guido van Rossum374a9221991-04-04 10:40:29 +00001965 case JUMP_FORWARD:
1966 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001967 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001968
Raymond Hettingerf606f872003-03-16 03:11:04 +00001969 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00001970 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00001971 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00001972 if (w == Py_True) {
1973 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001974 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00001975 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00001976 if (w == Py_False) {
1977 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001978 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00001979 }
1980 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001981 if (err > 0)
1982 err = 0;
1983 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00001984 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001985 else
1986 break;
1987 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001988
Raymond Hettingerf606f872003-03-16 03:11:04 +00001989 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00001990 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00001991 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00001992 if (w == Py_False) {
1993 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001994 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00001995 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00001996 if (w == Py_True) {
1997 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001998 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00001999 }
2000 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002001 if (err > 0) {
2002 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002003 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002004 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002005 else if (err == 0)
2006 ;
2007 else
2008 break;
2009 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002010
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002011 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002012 case JUMP_ABSOLUTE:
2013 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002014#if FAST_LOOPS
2015 /* Enabling this path speeds-up all while and for-loops by bypassing
2016 the per-loop checks for signals. By default, this should be turned-off
2017 because it prevents detection of a control-break in tight loops like
2018 "while 1: pass". Compile with this option turned-on when you need
2019 the speed-up and do not need break checking inside tight loops (ones
2020 that contain only instructions ending with goto fast_next_opcode).
2021 */
2022 goto fast_next_opcode;
2023#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002024 continue;
Guido van Rossum58da9312007-11-10 23:39:45 +00002025#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002026
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002027 case GET_ITER:
2028 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002029 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002030 x = PyObject_GetIter(v);
2031 Py_DECREF(v);
2032 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002033 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002034 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002035 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002036 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002037 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002038 break;
2039
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002040 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002041 case FOR_ITER:
2042 /* before: [iter]; after: [iter, iter()] *or* [] */
2043 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002044 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002045 if (x != NULL) {
2046 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002047 PREDICT(STORE_FAST);
2048 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002049 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002050 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002051 if (PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00002052 if (!PyErr_ExceptionMatches(
2053 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002054 break;
2055 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002056 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002057 /* iterator ended normally */
2058 x = v = POP();
2059 Py_DECREF(v);
2060 JUMPBY(oparg);
2061 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002062
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002063 case BREAK_LOOP:
2064 why = WHY_BREAK;
2065 goto fast_block_end;
2066
2067 case CONTINUE_LOOP:
2068 retval = PyInt_FromLong(oparg);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002069 if (!retval) {
2070 x = NULL;
2071 break;
2072 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002073 why = WHY_CONTINUE;
2074 goto fast_block_end;
2075
Guido van Rossum374a9221991-04-04 10:40:29 +00002076 case SETUP_LOOP:
2077 case SETUP_EXCEPT:
2078 case SETUP_FINALLY:
Thomas Wouters8ce81f72007-09-20 18:22:40 +00002079 /* NOTE: If you add any new block-setup opcodes that
2080 are not try/except/finally handlers, you may need
2081 to update the PyGen_NeedsFinalizing() function.
2082 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002083
Guido van Rossumb209a111997-04-29 18:18:01 +00002084 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002085 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002086 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002087
Guido van Rossumc2e20742006-02-27 22:32:47 +00002088 case WITH_CLEANUP:
2089 {
2090 /* TOP is the context.__exit__ bound method.
2091 Below that are 1-3 values indicating how/why
2092 we entered the finally clause:
2093 - SECOND = None
Guido van Rossumf6694362006-03-10 02:28:35 +00002094 - (SECOND, THIRD) = (WHY_{RETURN,CONTINUE}), retval
Guido van Rossumc2e20742006-02-27 22:32:47 +00002095 - SECOND = WHY_*; no retval below it
2096 - (SECOND, THIRD, FOURTH) = exc_info()
2097 In the last case, we must call
2098 TOP(SECOND, THIRD, FOURTH)
2099 otherwise we must call
2100 TOP(None, None, None)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002101
2102 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002103 *and* the function call returns a 'true' value, we
2104 "zap" this information, to prevent END_FINALLY from
2105 re-raising the exception. (But non-local gotos
2106 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002107 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002108
Guido van Rossumc2e20742006-02-27 22:32:47 +00002109 x = TOP();
2110 u = SECOND();
2111 if (PyInt_Check(u) || u == Py_None) {
2112 u = v = w = Py_None;
2113 }
2114 else {
2115 v = THIRD();
2116 w = FOURTH();
2117 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002118 /* XXX Not the fastest way to call it... */
2119 x = PyObject_CallFunctionObjArgs(x, u, v, w, NULL);
2120 if (x == NULL)
2121 break; /* Go to error exit */
2122 if (u != Py_None && PyObject_IsTrue(x)) {
2123 /* There was an exception and a true return */
2124 Py_DECREF(x);
2125 x = TOP(); /* Again */
2126 STACKADJ(-3);
2127 Py_INCREF(Py_None);
2128 SET_TOP(Py_None);
2129 Py_DECREF(x);
2130 Py_DECREF(u);
2131 Py_DECREF(v);
2132 Py_DECREF(w);
2133 } else {
2134 /* Let END_FINALLY do its thing */
2135 Py_DECREF(x);
2136 x = POP();
2137 Py_DECREF(x);
2138 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002139 break;
2140 }
2141
Guido van Rossumf10570b1995-07-07 22:53:21 +00002142 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002143 {
2144 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002145 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002146 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002147#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002148 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002149#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002150 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002151#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002152 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002153 PUSH(x);
2154 if (x != NULL)
2155 continue;
2156 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002157 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002158
Jeremy Hylton76901512000-03-28 23:49:17 +00002159 case CALL_FUNCTION_VAR:
2160 case CALL_FUNCTION_KW:
2161 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002162 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002163 int na = oparg & 0xff;
2164 int nk = (oparg>>8) & 0xff;
2165 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002166 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002167 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002168 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002169 if (flags & CALL_FLAG_VAR)
2170 n++;
2171 if (flags & CALL_FLAG_KW)
2172 n++;
2173 pfunc = stack_pointer - n - 1;
2174 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002175
Guido van Rossumac7be682001-01-17 15:42:30 +00002176 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002177 && PyMethod_GET_SELF(func) != NULL) {
2178 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002179 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002180 func = PyMethod_GET_FUNCTION(func);
2181 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002182 Py_DECREF(*pfunc);
2183 *pfunc = self;
2184 na++;
2185 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002186 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002187 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002188 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002189 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002190 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002191 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002192 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002193 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002194
Jeremy Hylton76901512000-03-28 23:49:17 +00002195 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002196 w = POP();
2197 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002198 }
2199 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002200 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002201 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002202 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002203 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002204
Guido van Rossum0240b922007-02-26 21:23:50 +00002205 case MAKE_CLOSURE:
Guido van Rossum681d79a1995-07-18 14:51:37 +00002206 case MAKE_FUNCTION:
Guido van Rossum4f72a782006-10-27 23:31:49 +00002207 {
2208 int posdefaults = oparg & 0xff;
2209 int kwdefaults = (oparg>>8) & 0xff;
Neal Norwitzc1505362006-12-28 06:47:50 +00002210 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002211
Guido van Rossum681d79a1995-07-18 14:51:37 +00002212 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002213 x = PyFunction_New(v, f->f_globals);
2214 Py_DECREF(v);
Guido van Rossum0240b922007-02-26 21:23:50 +00002215
2216 if (x != NULL && opcode == MAKE_CLOSURE) {
2217 v = POP();
2218 err = PyFunction_SetClosure(x, v);
2219 Py_DECREF(v);
2220 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002221
2222 if (x != NULL && num_annotations > 0) {
2223 Py_ssize_t name_ix;
2224 u = POP(); /* names of args with annotations */
2225 v = PyDict_New();
2226 if (v == NULL) {
2227 Py_DECREF(x);
2228 x = NULL;
2229 break;
2230 }
2231 name_ix = PyTuple_Size(u);
2232 assert(num_annotations == name_ix+1);
2233 while (name_ix > 0) {
2234 --name_ix;
2235 t = PyTuple_GET_ITEM(u, name_ix);
2236 w = POP();
2237 /* XXX(nnorwitz): check for errors */
2238 PyDict_SetItem(v, t, w);
2239 Py_DECREF(w);
2240 }
2241
2242 err = PyFunction_SetAnnotations(x, v);
2243 Py_DECREF(v);
2244 Py_DECREF(u);
2245 }
2246
Guido van Rossum681d79a1995-07-18 14:51:37 +00002247 /* XXX Maybe this should be a separate opcode? */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002248 if (x != NULL && posdefaults > 0) {
2249 v = PyTuple_New(posdefaults);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002250 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002251 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002252 x = NULL;
2253 break;
2254 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002255 while (--posdefaults >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002256 w = POP();
Guido van Rossum4f72a782006-10-27 23:31:49 +00002257 PyTuple_SET_ITEM(v, posdefaults, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002258 }
2259 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002260 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002261 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002262 if (x != NULL && kwdefaults > 0) {
2263 v = PyDict_New();
2264 if (v == NULL) {
2265 Py_DECREF(x);
2266 x = NULL;
2267 break;
2268 }
2269 while (--kwdefaults >= 0) {
2270 w = POP(); /* default value */
2271 u = POP(); /* kw only arg name */
Neal Norwitzc1505362006-12-28 06:47:50 +00002272 /* XXX(nnorwitz): check for errors */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002273 PyDict_SetItem(v, u, w);
Georg Brandl94ab0002007-02-26 13:58:18 +00002274 Py_DECREF(w);
2275 Py_DECREF(u);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002276 }
2277 err = PyFunction_SetKwDefaults(x, v);
2278 Py_DECREF(v);
2279 }
Guido van Rossum681d79a1995-07-18 14:51:37 +00002280 PUSH(x);
2281 break;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002282 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002283
2284 case BUILD_SLICE:
2285 if (oparg == 3)
2286 w = POP();
2287 else
2288 w = NULL;
2289 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002290 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002291 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002292 Py_DECREF(u);
2293 Py_DECREF(v);
2294 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002295 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002296 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002297 break;
2298
Fred Drakeef8ace32000-08-24 00:32:09 +00002299 case EXTENDED_ARG:
2300 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002301 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002302 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002303
Guido van Rossum374a9221991-04-04 10:40:29 +00002304 default:
2305 fprintf(stderr,
2306 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002307 PyCode_Addr2Line(f->f_code, f->f_lasti),
2308 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002309 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002310 why = WHY_EXCEPTION;
2311 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002312
2313#ifdef CASE_TOO_BIG
2314 }
2315#endif
2316
Guido van Rossum374a9221991-04-04 10:40:29 +00002317 } /* switch */
2318
2319 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002320
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002321 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002322
Guido van Rossum374a9221991-04-04 10:40:29 +00002323 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002324
Guido van Rossum374a9221991-04-04 10:40:29 +00002325 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002326 if (err == 0 && x != NULL) {
2327#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002328 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002329 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002330 fprintf(stderr,
2331 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002332 else {
2333#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002334 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002335 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002336#ifdef CHECKEXC
2337 }
2338#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002339 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002340 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002341 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002342 err = 0;
2343 }
2344
Guido van Rossum374a9221991-04-04 10:40:29 +00002345 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002346
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002347 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002348 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002349 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002350 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002351 why = WHY_EXCEPTION;
2352 }
2353 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002354#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002355 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002356 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002357 if (PyErr_Occurred()) {
Jeremy Hylton904ed862003-11-05 17:29:35 +00002358 char buf[1024];
2359 sprintf(buf, "Stack unwind with exception "
2360 "set and why=%d", why);
2361 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002362 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002363 }
2364#endif
2365
2366 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002367
Guido van Rossum374a9221991-04-04 10:40:29 +00002368 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002369 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002370
Fred Drake8f51f542001-10-04 14:48:42 +00002371 if (tstate->c_tracefunc != NULL)
2372 call_exc_trace(tstate->c_tracefunc,
2373 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002374 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002375
Guido van Rossum374a9221991-04-04 10:40:29 +00002376 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002377
Guido van Rossum374a9221991-04-04 10:40:29 +00002378 if (why == WHY_RERAISE)
2379 why = WHY_EXCEPTION;
2380
2381 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002382
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002383fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002384 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002385 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002386
Tim Peters8a5c3c72004-04-05 19:36:21 +00002387 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002388 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2389 /* For a continue inside a try block,
2390 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002391 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2392 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002393 why = WHY_NOT;
2394 JUMPTO(PyInt_AS_LONG(retval));
2395 Py_DECREF(retval);
2396 break;
2397 }
2398
Guido van Rossum374a9221991-04-04 10:40:29 +00002399 while (STACK_LEVEL() > b->b_level) {
2400 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002401 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002402 }
2403 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2404 why = WHY_NOT;
2405 JUMPTO(b->b_handler);
2406 break;
2407 }
2408 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002409 (b->b_type == SETUP_EXCEPT &&
2410 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002411 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002412 PyObject *exc, *val, *tb;
2413 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002414 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002415 val = Py_None;
2416 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002417 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002418 /* Make the raw exception data
2419 available to the handler,
2420 so a program can emulate the
2421 Python main loop. Don't do
2422 this for 'finally'. */
2423 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002424 PyErr_NormalizeException(
2425 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002426 set_exc_info(tstate,
2427 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002428 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002429 if (tb == NULL) {
2430 Py_INCREF(Py_None);
2431 PUSH(Py_None);
2432 } else
2433 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002434 PUSH(val);
2435 PUSH(exc);
2436 }
2437 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002438 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002439 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002440 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002441 PUSH(v);
2442 }
2443 why = WHY_NOT;
2444 JUMPTO(b->b_handler);
2445 break;
2446 }
2447 } /* unwind stack */
2448
2449 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002450
Guido van Rossum374a9221991-04-04 10:40:29 +00002451 if (why != WHY_NOT)
2452 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002453 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002454
Guido van Rossum374a9221991-04-04 10:40:29 +00002455 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002456
Tim Peters8a5c3c72004-04-05 19:36:21 +00002457 assert(why != WHY_YIELD);
2458 /* Pop remaining stack entries. */
2459 while (!EMPTY()) {
2460 v = POP();
2461 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002462 }
2463
Tim Peters8a5c3c72004-04-05 19:36:21 +00002464 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002465 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002466
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002467fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002468 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002469 if (tstate->c_tracefunc) {
2470 if (why == WHY_RETURN || why == WHY_YIELD) {
2471 if (call_trace(tstate->c_tracefunc,
2472 tstate->c_traceobj, f,
2473 PyTrace_RETURN, retval)) {
2474 Py_XDECREF(retval);
2475 retval = NULL;
2476 why = WHY_EXCEPTION;
2477 }
2478 }
2479 else if (why == WHY_EXCEPTION) {
2480 call_trace_protected(tstate->c_tracefunc,
2481 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002482 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002483 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002484 }
Fred Drake8f51f542001-10-04 14:48:42 +00002485 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002486 if (why == WHY_EXCEPTION)
2487 call_trace_protected(tstate->c_profilefunc,
2488 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002489 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002490 else if (call_trace(tstate->c_profilefunc,
2491 tstate->c_profileobj, f,
2492 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002493 Py_XDECREF(retval);
2494 retval = NULL;
2495 why = WHY_EXCEPTION;
2496 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002497 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002498 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002499
Thomas Wouters477c8d52006-05-27 19:21:47 +00002500 if (tstate->frame->f_exc_type != NULL)
2501 reset_exc_info(tstate);
2502 else {
2503 assert(tstate->frame->f_exc_value == NULL);
2504 assert(tstate->frame->f_exc_traceback == NULL);
2505 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002506
Tim Peters5ca576e2001-06-18 22:08:13 +00002507 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00002508exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002509 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002510 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002511
Guido van Rossum96a42c81992-01-12 02:29:51 +00002512 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002513}
2514
Guido van Rossumc2e20742006-02-27 22:32:47 +00002515/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002516 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002517 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002518
Tim Peters6d6c1a32001-08-02 04:15:00 +00002519PyObject *
2520PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002521 PyObject **args, int argcount, PyObject **kws, int kwcount,
Guido van Rossum4f72a782006-10-27 23:31:49 +00002522 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00002523{
2524 register PyFrameObject *f;
2525 register PyObject *retval = NULL;
2526 register PyObject **fastlocals, **freevars;
2527 PyThreadState *tstate = PyThreadState_GET();
2528 PyObject *x, *u;
2529
2530 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002531 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002532 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002533 return NULL;
2534 }
2535
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002536 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002537 assert(globals != NULL);
2538 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002539 if (f == NULL)
2540 return NULL;
2541
2542 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002543 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002544
2545 if (co->co_argcount > 0 ||
Guido van Rossum4f72a782006-10-27 23:31:49 +00002546 co->co_kwonlyargcount > 0 ||
Tim Peters5ca576e2001-06-18 22:08:13 +00002547 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2548 int i;
2549 int n = argcount;
2550 PyObject *kwdict = NULL;
2551 if (co->co_flags & CO_VARKEYWORDS) {
2552 kwdict = PyDict_New();
2553 if (kwdict == NULL)
2554 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002555 i = co->co_argcount + co->co_kwonlyargcount;
Tim Peters5ca576e2001-06-18 22:08:13 +00002556 if (co->co_flags & CO_VARARGS)
2557 i++;
2558 SETLOCAL(i, kwdict);
2559 }
2560 if (argcount > co->co_argcount) {
2561 if (!(co->co_flags & CO_VARARGS)) {
2562 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002563 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002564 "%spositional argument%s (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002565 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002566 defcount ? "at most" : "exactly",
2567 co->co_argcount,
2568 kwcount ? "non-keyword " : "",
2569 co->co_argcount == 1 ? "" : "s",
2570 argcount);
2571 goto fail;
2572 }
2573 n = co->co_argcount;
2574 }
2575 for (i = 0; i < n; i++) {
2576 x = args[i];
2577 Py_INCREF(x);
2578 SETLOCAL(i, x);
2579 }
2580 if (co->co_flags & CO_VARARGS) {
2581 u = PyTuple_New(argcount - n);
2582 if (u == NULL)
2583 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002584 SETLOCAL(co->co_argcount + co->co_kwonlyargcount, u);
Tim Peters5ca576e2001-06-18 22:08:13 +00002585 for (i = n; i < argcount; i++) {
2586 x = args[i];
2587 Py_INCREF(x);
2588 PyTuple_SET_ITEM(u, i-n, x);
2589 }
2590 }
2591 for (i = 0; i < kwcount; i++) {
2592 PyObject *keyword = kws[2*i];
2593 PyObject *value = kws[2*i + 1];
2594 int j;
Neal Norwitzda059e32007-08-26 05:33:45 +00002595 if (keyword == NULL || !PyUnicode_Check(keyword)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002596 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002597 "%U() keywords must be strings",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002598 co->co_name);
Tim Peters5ca576e2001-06-18 22:08:13 +00002599 goto fail;
2600 }
2601 /* XXX slow -- speed up using dictionary? */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002602 for (j = 0;
2603 j < co->co_argcount + co->co_kwonlyargcount;
2604 j++) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002605 PyObject *nm = PyTuple_GET_ITEM(
2606 co->co_varnames, j);
2607 int cmp = PyObject_RichCompareBool(
2608 keyword, nm, Py_EQ);
2609 if (cmp > 0)
2610 break;
2611 else if (cmp < 0)
2612 goto fail;
2613 }
2614 /* Check errors from Compare */
2615 if (PyErr_Occurred())
2616 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002617 if (j >= co->co_argcount + co->co_kwonlyargcount) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002618 if (kwdict == NULL) {
2619 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002620 "%U() got an unexpected "
Walter Dörwald573c08c2007-05-25 15:46:59 +00002621 "keyword argument '%S'",
2622 co->co_name,
2623 keyword);
Tim Peters5ca576e2001-06-18 22:08:13 +00002624 goto fail;
2625 }
2626 PyDict_SetItem(kwdict, keyword, value);
2627 }
2628 else {
2629 if (GETLOCAL(j) != NULL) {
2630 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002631 "%U() got multiple "
Tim Peters5ca576e2001-06-18 22:08:13 +00002632 "values for keyword "
Walter Dörwald573c08c2007-05-25 15:46:59 +00002633 "argument '%S'",
2634 co->co_name,
2635 keyword);
Tim Peters5ca576e2001-06-18 22:08:13 +00002636 goto fail;
2637 }
2638 Py_INCREF(value);
2639 SETLOCAL(j, value);
2640 }
2641 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002642 if (co->co_kwonlyargcount > 0) {
2643 for (i = co->co_argcount;
2644 i < co->co_argcount + co->co_kwonlyargcount;
2645 i++) {
Guido van Rossum29602e42006-11-22 04:45:33 +00002646 PyObject *name, *def;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002647 if (GETLOCAL(i) != NULL)
2648 continue;
Guido van Rossum29602e42006-11-22 04:45:33 +00002649 name = PyTuple_GET_ITEM(co->co_varnames, i);
2650 def = NULL;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002651 if (kwdefs != NULL)
2652 def = PyDict_GetItem(kwdefs, name);
2653 if (def != NULL) {
2654 Py_INCREF(def);
2655 SETLOCAL(i, def);
2656 continue;
2657 }
2658 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002659 "%U() needs keyword-only argument %S",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002660 co->co_name, name);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002661 goto fail;
2662 }
2663 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002664 if (argcount < co->co_argcount) {
2665 int m = co->co_argcount - defcount;
2666 for (i = argcount; i < m; i++) {
2667 if (GETLOCAL(i) == NULL) {
2668 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002669 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002670 "%spositional argument%s "
2671 "(%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002672 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002673 ((co->co_flags & CO_VARARGS) ||
2674 defcount) ? "at least"
2675 : "exactly",
2676 m, kwcount ? "non-keyword " : "",
2677 m == 1 ? "" : "s", i);
2678 goto fail;
2679 }
2680 }
2681 if (n > m)
2682 i = n - m;
2683 else
2684 i = 0;
2685 for (; i < defcount; i++) {
2686 if (GETLOCAL(m+i) == NULL) {
2687 PyObject *def = defs[i];
2688 Py_INCREF(def);
2689 SETLOCAL(m+i, def);
2690 }
2691 }
2692 }
2693 }
2694 else {
2695 if (argcount > 0 || kwcount > 0) {
2696 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002697 "%U() takes no arguments (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002698 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002699 argcount + kwcount);
2700 goto fail;
2701 }
2702 }
2703 /* Allocate and initialize storage for cell vars, and copy free
2704 vars into frame. This isn't too efficient right now. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002705 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002706 int i, j, nargs, found;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002707 Py_UNICODE *cellname, *argname;
Tim Peters5ca576e2001-06-18 22:08:13 +00002708 PyObject *c;
2709
2710 nargs = co->co_argcount;
2711 if (co->co_flags & CO_VARARGS)
2712 nargs++;
2713 if (co->co_flags & CO_VARKEYWORDS)
2714 nargs++;
2715
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002716 /* Initialize each cell var, taking into account
2717 cell vars that are initialized from arguments.
2718
2719 Should arrange for the compiler to put cellvars
2720 that are arguments at the beginning of the cellvars
2721 list so that we can march over it more efficiently?
2722 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002723 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002724 cellname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002725 PyTuple_GET_ITEM(co->co_cellvars, i));
2726 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002727 for (j = 0; j < nargs; j++) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002728 argname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002729 PyTuple_GET_ITEM(co->co_varnames, j));
Martin v. Löwis5b222132007-06-10 09:51:05 +00002730 if (Py_UNICODE_strcmp(cellname, argname) == 0) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002731 c = PyCell_New(GETLOCAL(j));
2732 if (c == NULL)
2733 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002734 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002735 found = 1;
2736 break;
2737 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002738 }
2739 if (found == 0) {
2740 c = PyCell_New(NULL);
2741 if (c == NULL)
2742 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002743 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002744 }
2745 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002746 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002747 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002748 int i;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002749 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002750 PyObject *o = PyTuple_GET_ITEM(closure, i);
2751 Py_INCREF(o);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002752 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002753 }
2754 }
2755
Tim Peters5ca576e2001-06-18 22:08:13 +00002756 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002757 /* Don't need to keep the reference to f_back, it will be set
2758 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002759 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002760 f->f_back = NULL;
2761
Jeremy Hylton985eba52003-02-05 23:13:00 +00002762 PCALL(PCALL_GENERATOR);
2763
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002764 /* Create a new generator that owns the ready to run frame
2765 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002766 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002767 }
2768
Thomas Woutersce272b62007-09-19 21:19:28 +00002769 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002770
Thomas Woutersce272b62007-09-19 21:19:28 +00002771fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00002772
Tim Petersb13680b2001-11-27 23:29:29 +00002773 /* decref'ing the frame can cause __del__ methods to get invoked,
2774 which can call back into Python. While we're done with the
2775 current Python frame (f), the associated C stack is still in use,
2776 so recursion_depth must be boosted for the duration.
2777 */
2778 assert(tstate != NULL);
2779 ++tstate->recursion_depth;
Thomas Woutersce272b62007-09-19 21:19:28 +00002780 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002781 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002782 return retval;
2783}
2784
2785
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002786/* Implementation notes for set_exc_info() and reset_exc_info():
2787
2788- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2789 'exc_traceback'. These always travel together.
2790
2791- tstate->curexc_ZZZ is the "hot" exception that is set by
2792 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2793
2794- Once an exception is caught by an except clause, it is transferred
2795 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2796 can pick it up. This is the primary task of set_exc_info().
Thomas Wouters477c8d52006-05-27 19:21:47 +00002797 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002798
2799- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2800
2801 Long ago, when none of this existed, there were just a few globals:
2802 one set corresponding to the "hot" exception, and one set
2803 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2804 globals; they were simply stored as sys.exc_ZZZ. For backwards
2805 compatibility, they still are!) The problem was that in code like
2806 this:
2807
2808 try:
2809 "something that may fail"
2810 except "some exception":
2811 "do something else first"
2812 "print the exception from sys.exc_ZZZ."
2813
2814 if "do something else first" invoked something that raised and caught
2815 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2816 cause of subtle bugs. I fixed this by changing the semantics as
2817 follows:
2818
2819 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2820 *in that frame*.
2821
2822 - But initially, and as long as no exception is caught in a given
2823 frame, sys.exc_ZZZ will hold the last exception caught in the
2824 previous frame (or the frame before that, etc.).
2825
2826 The first bullet fixed the bug in the above example. The second
2827 bullet was for backwards compatibility: it was (and is) common to
2828 have a function that is called when an exception is caught, and to
2829 have that function access the caught exception via sys.exc_ZZZ.
2830 (Example: traceback.print_exc()).
2831
2832 At the same time I fixed the problem that sys.exc_ZZZ weren't
2833 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2834 but that's really a separate improvement.
2835
2836 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2837 variables to what they were before the current frame was called. The
2838 set_exc_info() function saves them on the frame so that
2839 reset_exc_info() can restore them. The invariant is that
2840 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2841 exception (where "catching" an exception applies only to successful
2842 except clauses); and if the current frame ever caught an exception,
2843 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2844 at the start of the current frame.
2845
2846*/
2847
Guido van Rossuma027efa1997-05-05 20:56:21 +00002848static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002849set_exc_info(PyThreadState *tstate,
2850 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002851{
Thomas Wouters477c8d52006-05-27 19:21:47 +00002852 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002853 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002854
Thomas Wouters477c8d52006-05-27 19:21:47 +00002855 assert(type != NULL);
2856 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002857 if (frame->f_exc_type == NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002858 assert(frame->f_exc_value == NULL);
2859 assert(frame->f_exc_traceback == NULL);
2860 /* This frame didn't catch an exception before. */
2861 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002862 if (tstate->exc_type == NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002863 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002864 Py_INCREF(Py_None);
2865 tstate->exc_type = Py_None;
2866 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002867 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002868 Py_XINCREF(tstate->exc_value);
2869 Py_XINCREF(tstate->exc_traceback);
2870 frame->f_exc_type = tstate->exc_type;
2871 frame->f_exc_value = tstate->exc_value;
2872 frame->f_exc_traceback = tstate->exc_traceback;
2873 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002874 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002875 tmp_type = tstate->exc_type;
2876 tmp_value = tstate->exc_value;
2877 tmp_tb = tstate->exc_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002878 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002879 Py_XINCREF(value);
2880 Py_XINCREF(tb);
2881 tstate->exc_type = type;
2882 tstate->exc_value = value;
2883 tstate->exc_traceback = tb;
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002884 PyException_SetTraceback(value, tb);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002885 Py_XDECREF(tmp_type);
2886 Py_XDECREF(tmp_value);
2887 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002888}
2889
2890static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002891reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002892{
2893 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002894 PyObject *tmp_type, *tmp_value, *tmp_tb;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002895
2896 /* It's a precondition that the thread state's frame caught an
2897 * exception -- verify in a debug build.
2898 */
2899 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002900 frame = tstate->frame;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002901 assert(frame != NULL);
2902 assert(frame->f_exc_type != NULL);
2903
2904 /* Copy the frame's exception info back to the thread state. */
2905 tmp_type = tstate->exc_type;
2906 tmp_value = tstate->exc_value;
2907 tmp_tb = tstate->exc_traceback;
2908 Py_INCREF(frame->f_exc_type);
2909 Py_XINCREF(frame->f_exc_value);
2910 Py_XINCREF(frame->f_exc_traceback);
2911 tstate->exc_type = frame->f_exc_type;
2912 tstate->exc_value = frame->f_exc_value;
2913 tstate->exc_traceback = frame->f_exc_traceback;
2914 Py_XDECREF(tmp_type);
2915 Py_XDECREF(tmp_value);
2916 Py_XDECREF(tmp_tb);
2917
Thomas Wouters477c8d52006-05-27 19:21:47 +00002918 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002919 tmp_type = frame->f_exc_type;
2920 tmp_value = frame->f_exc_value;
2921 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00002922 frame->f_exc_type = NULL;
2923 frame->f_exc_value = NULL;
2924 frame->f_exc_traceback = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002925 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002926 Py_XDECREF(tmp_value);
2927 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002928}
2929
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002930/* Logic for the raise statement (too complicated for inlining).
2931 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00002932static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00002933do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002934{
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002935 PyObject *type = NULL, *value = NULL, *tb = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00002936
2937 if (exc == NULL) {
Guido van Rossumd295f121998-04-09 21:39:57 +00002938 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00002939 PyThreadState *tstate = PyThreadState_GET();
Collin Winter828f04a2007-08-31 00:04:24 +00002940 type = tstate->exc_type;
Guido van Rossumd295f121998-04-09 21:39:57 +00002941 value = tstate->exc_value;
2942 tb = tstate->exc_traceback;
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002943 if (type == Py_None) {
2944 PyErr_SetString(PyExc_RuntimeError,
2945 "No active exception to reraise");
2946 return WHY_EXCEPTION;
2947 }
2948 Py_XINCREF(type);
Guido van Rossumd295f121998-04-09 21:39:57 +00002949 Py_XINCREF(value);
2950 Py_XINCREF(tb);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002951 PyErr_Restore(type, value, tb);
2952 return WHY_RERAISE;
Guido van Rossumd295f121998-04-09 21:39:57 +00002953 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002954
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002955 /* We support the following forms of raise:
Collin Winter828f04a2007-08-31 00:04:24 +00002956 raise
2957 raise <instance>
2958 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002959
Collin Winter828f04a2007-08-31 00:04:24 +00002960 if (PyExceptionClass_Check(exc)) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002961 type = exc;
2962 value = PyObject_CallObject(exc, NULL);
Collin Winter828f04a2007-08-31 00:04:24 +00002963 if (value == NULL)
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002964 goto raise_error;
2965 }
Collin Winter828f04a2007-08-31 00:04:24 +00002966 else if (PyExceptionInstance_Check(exc)) {
2967 value = exc;
2968 type = PyExceptionInstance_Class(exc);
Guido van Rossumb209a111997-04-29 18:18:01 +00002969 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002970 }
2971 else {
2972 /* Not something you can raise. You get an exception
2973 anyway, just not what you specified :-) */
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002974 Py_DECREF(exc);
Guido van Rossum45aecf42006-03-15 04:58:47 +00002975 PyErr_SetString(PyExc_TypeError,
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002976 "exceptions must derive from BaseException");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002977 goto raise_error;
2978 }
Collin Winter828f04a2007-08-31 00:04:24 +00002979
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002980 tb = PyException_GetTraceback(value);
2981 if (cause) {
2982 PyObject *fixed_cause;
2983 if (PyExceptionClass_Check(cause)) {
2984 fixed_cause = PyObject_CallObject(cause, NULL);
2985 if (fixed_cause == NULL)
2986 goto raise_error;
2987 Py_DECREF(cause);
2988 }
2989 else if (PyExceptionInstance_Check(cause)) {
2990 fixed_cause = cause;
2991 }
2992 else {
2993 PyErr_SetString(PyExc_TypeError,
2994 "exception causes must derive from "
2995 "BaseException");
2996 goto raise_error;
2997 }
2998 PyException_SetCause(value, fixed_cause);
2999 }
Collin Winter828f04a2007-08-31 00:04:24 +00003000
Guido van Rossumb209a111997-04-29 18:18:01 +00003001 PyErr_Restore(type, value, tb);
Collin Winter828f04a2007-08-31 00:04:24 +00003002 return WHY_EXCEPTION;
3003
3004raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003005 Py_XDECREF(value);
3006 Py_XDECREF(type);
3007 Py_XDECREF(tb);
Collin Winter1966f1c2007-09-01 20:26:44 +00003008 Py_XDECREF(cause);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003009 return WHY_EXCEPTION;
3010}
3011
Tim Petersd6d010b2001-06-21 02:49:55 +00003012/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003013 sp). Return 1 for success, 0 if error.
3014
3015 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3016 with a variable target.
3017*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003018
Barry Warsawe42b18f1997-08-25 22:13:04 +00003019static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003020unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003021{
Guido van Rossum0368b722007-05-11 16:50:42 +00003022 int i = 0, j = 0;
3023 Py_ssize_t ll = 0;
Tim Petersd6d010b2001-06-21 02:49:55 +00003024 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003025 PyObject *w;
Guido van Rossum0368b722007-05-11 16:50:42 +00003026 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003027
Tim Petersd6d010b2001-06-21 02:49:55 +00003028 assert(v != NULL);
3029
3030 it = PyObject_GetIter(v);
3031 if (it == NULL)
3032 goto Error;
3033
3034 for (; i < argcnt; i++) {
3035 w = PyIter_Next(it);
3036 if (w == NULL) {
3037 /* Iterator done, via error or exhaustion. */
3038 if (!PyErr_Occurred()) {
3039 PyErr_Format(PyExc_ValueError,
3040 "need more than %d value%s to unpack",
3041 i, i == 1 ? "" : "s");
3042 }
3043 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003044 }
3045 *--sp = w;
3046 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003047
Guido van Rossum0368b722007-05-11 16:50:42 +00003048 if (argcntafter == -1) {
3049 /* We better have exhausted the iterator now. */
3050 w = PyIter_Next(it);
3051 if (w == NULL) {
3052 if (PyErr_Occurred())
3053 goto Error;
3054 Py_DECREF(it);
3055 return 1;
3056 }
3057 Py_DECREF(w);
3058 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
3059 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003060 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003061
3062 l = PySequence_List(it);
3063 if (l == NULL)
3064 goto Error;
3065 *--sp = l;
3066 i++;
3067
3068 ll = PyList_GET_SIZE(l);
3069 if (ll < argcntafter) {
3070 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3071 argcnt + ll);
3072 goto Error;
3073 }
3074
3075 /* Pop the "after-variable" args off the list. */
3076 for (j = argcntafter; j > 0; j--, i++) {
3077 *--sp = PyList_GET_ITEM(l, ll - j);
3078 }
3079 /* Resize the list. */
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00003080 Py_Size(l) = ll - argcntafter;
Guido van Rossum0368b722007-05-11 16:50:42 +00003081 Py_DECREF(it);
3082 return 1;
3083
Tim Petersd6d010b2001-06-21 02:49:55 +00003084Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003085 for (; i > 0; i--, sp++)
3086 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003087 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003088 return 0;
3089}
3090
3091
Guido van Rossum96a42c81992-01-12 02:29:51 +00003092#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003093static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003094prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003095{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003096 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003097 if (PyObject_Print(v, stdout, 0) != 0)
3098 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003099 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003100 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003101}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003102#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003103
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003104static void
Fred Drake5755ce62001-06-27 19:19:46 +00003105call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003106{
Guido van Rossumb209a111997-04-29 18:18:01 +00003107 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003108 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003109 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003110 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003111 value = Py_None;
3112 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003113 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003114 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003115 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003116 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003117 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003118 }
Fred Drake5755ce62001-06-27 19:19:46 +00003119 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003120 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003121 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003122 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003123 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003124 Py_XDECREF(type);
3125 Py_XDECREF(value);
3126 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003127 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003128}
3129
Fred Drake4ec5d562001-10-04 19:26:43 +00003130static void
3131call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003132 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003133{
3134 PyObject *type, *value, *traceback;
3135 int err;
3136 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003137 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003138 if (err == 0)
3139 PyErr_Restore(type, value, traceback);
3140 else {
3141 Py_XDECREF(type);
3142 Py_XDECREF(value);
3143 Py_XDECREF(traceback);
3144 }
3145}
3146
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003147static int
Fred Drake5755ce62001-06-27 19:19:46 +00003148call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3149 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003150{
Fred Drake5755ce62001-06-27 19:19:46 +00003151 register PyThreadState *tstate = frame->f_tstate;
3152 int result;
3153 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003154 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003155 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003156 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003157 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003158 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3159 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003160 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003161 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003162}
3163
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003164PyObject *
3165_PyEval_CallTracing(PyObject *func, PyObject *args)
3166{
3167 PyFrameObject *frame = PyEval_GetFrame();
3168 PyThreadState *tstate = frame->f_tstate;
3169 int save_tracing = tstate->tracing;
3170 int save_use_tracing = tstate->use_tracing;
3171 PyObject *result;
3172
3173 tstate->tracing = 0;
3174 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3175 || (tstate->c_profilefunc != NULL));
3176 result = PyObject_Call(func, args, NULL);
3177 tstate->tracing = save_tracing;
3178 tstate->use_tracing = save_use_tracing;
3179 return result;
3180}
3181
Michael W. Hudson006c7522002-11-08 13:08:46 +00003182static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003183maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003184 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3185 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003186{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003187 int result = 0;
3188
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003189 /* If the last instruction executed isn't in the current
3190 instruction window, reset the window. If the last
3191 instruction happens to fall at the start of a line or if it
3192 represents a jump backwards, call the trace function.
3193 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003194 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003195 int line;
3196 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003197
Thomas Woutersce272b62007-09-19 21:19:28 +00003198 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3199 &bounds);
3200 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003201 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003202 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003203 PyTrace_LINE, Py_None);
Thomas Woutersce272b62007-09-19 21:19:28 +00003204 }
3205 *instr_lb = bounds.ap_lower;
3206 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003207 }
Armin Rigobf57a142004-03-22 19:24:58 +00003208 else if (frame->f_lasti <= *instr_prev) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003209 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003210 }
3211 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003212 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003213}
3214
Fred Drake5755ce62001-06-27 19:19:46 +00003215void
3216PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003217{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003218 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003219 PyObject *temp = tstate->c_profileobj;
3220 Py_XINCREF(arg);
3221 tstate->c_profilefunc = NULL;
3222 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003223 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003224 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003225 Py_XDECREF(temp);
3226 tstate->c_profilefunc = func;
3227 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003228 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003229 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003230}
3231
3232void
3233PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3234{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003235 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003236 PyObject *temp = tstate->c_traceobj;
3237 Py_XINCREF(arg);
3238 tstate->c_tracefunc = NULL;
3239 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003240 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003241 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003242 Py_XDECREF(temp);
3243 tstate->c_tracefunc = func;
3244 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003245 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003246 tstate->use_tracing = ((func != NULL)
3247 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003248}
3249
Guido van Rossumb209a111997-04-29 18:18:01 +00003250PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003251PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003252{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003253 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003254 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003255 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003256 else
3257 return current_frame->f_builtins;
3258}
3259
Guido van Rossumb209a111997-04-29 18:18:01 +00003260PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003261PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003262{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003263 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003264 if (current_frame == NULL)
3265 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003266 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003267 return current_frame->f_locals;
3268}
3269
Guido van Rossumb209a111997-04-29 18:18:01 +00003270PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003271PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003272{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003273 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003274 if (current_frame == NULL)
3275 return NULL;
3276 else
3277 return current_frame->f_globals;
3278}
3279
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003280PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003281PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003282{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003283 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003284 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003285}
3286
Guido van Rossum6135a871995-01-09 17:53:26 +00003287int
Tim Peters5ba58662001-07-16 02:29:45 +00003288PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003289{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003290 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003291 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003292
3293 if (current_frame != NULL) {
3294 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003295 const int compilerflags = codeflags & PyCF_MASK;
3296 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003297 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003298 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003299 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003300#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003301 if (codeflags & CO_GENERATOR_ALLOWED) {
3302 result = 1;
3303 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3304 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003305#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003306 }
3307 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003308}
3309
Guido van Rossum3f5da241990-12-20 15:06:42 +00003310
Guido van Rossum681d79a1995-07-18 14:51:37 +00003311/* External interface to call any callable object.
3312 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003313
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003314#undef PyEval_CallObject
3315/* for backward compatibility: export this interface */
3316
Guido van Rossumb209a111997-04-29 18:18:01 +00003317PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003318PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003319{
Guido van Rossumb209a111997-04-29 18:18:01 +00003320 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003321}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003322#define PyEval_CallObject(func,arg) \
3323 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003324
Guido van Rossumb209a111997-04-29 18:18:01 +00003325PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003326PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003327{
Jeremy Hylton52820442001-01-03 23:52:36 +00003328 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003329
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003330 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003331 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003332 if (arg == NULL)
3333 return NULL;
3334 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003335 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003336 PyErr_SetString(PyExc_TypeError,
3337 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003338 return NULL;
3339 }
3340 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003341 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003342
Guido van Rossumb209a111997-04-29 18:18:01 +00003343 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003344 PyErr_SetString(PyExc_TypeError,
3345 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003346 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003347 return NULL;
3348 }
3349
Tim Peters6d6c1a32001-08-02 04:15:00 +00003350 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003351 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003352 return result;
3353}
3354
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003355const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003356PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003357{
3358 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003359 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003360 else if (PyFunction_Check(func))
Martin v. Löwis5b222132007-06-10 09:51:05 +00003361 return PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003362 else if (PyCFunction_Check(func))
3363 return ((PyCFunctionObject*)func)->m_ml->ml_name;
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003364 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003365 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003366}
3367
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003368const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003369PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003370{
3371 if (PyMethod_Check(func))
3372 return "()";
3373 else if (PyFunction_Check(func))
3374 return "()";
3375 else if (PyCFunction_Check(func))
3376 return "()";
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003377 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003378 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00003379}
3380
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00003381static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003382err_args(PyObject *func, int flags, int nargs)
3383{
3384 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003385 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003386 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003387 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003388 nargs);
3389 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003390 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003391 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003392 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003393 nargs);
3394}
3395
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003396#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003397if (tstate->use_tracing && tstate->c_profilefunc) { \
3398 if (call_trace(tstate->c_profilefunc, \
3399 tstate->c_profileobj, \
3400 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003401 func)) { \
3402 x = NULL; \
3403 } \
3404 else { \
3405 x = call; \
3406 if (tstate->c_profilefunc != NULL) { \
3407 if (x == NULL) { \
3408 call_trace_protected(tstate->c_profilefunc, \
3409 tstate->c_profileobj, \
3410 tstate->frame, PyTrace_C_EXCEPTION, \
3411 func); \
3412 /* XXX should pass (type, value, tb) */ \
3413 } else { \
3414 if (call_trace(tstate->c_profilefunc, \
3415 tstate->c_profileobj, \
3416 tstate->frame, PyTrace_C_RETURN, \
3417 func)) { \
3418 Py_DECREF(x); \
3419 x = NULL; \
3420 } \
3421 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003422 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003423 } \
3424} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003425 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003426 }
3427
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003428static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003429call_function(PyObject ***pp_stack, int oparg
3430#ifdef WITH_TSC
3431 , uint64* pintr0, uint64* pintr1
3432#endif
3433 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003434{
3435 int na = oparg & 0xff;
3436 int nk = (oparg>>8) & 0xff;
3437 int n = na + 2 * nk;
3438 PyObject **pfunc = (*pp_stack) - n - 1;
3439 PyObject *func = *pfunc;
3440 PyObject *x, *w;
3441
Jeremy Hylton985eba52003-02-05 23:13:00 +00003442 /* Always dispatch PyCFunction first, because these are
3443 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003444 */
3445 if (PyCFunction_Check(func) && nk == 0) {
3446 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003447 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003448
3449 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003450 if (flags & (METH_NOARGS | METH_O)) {
3451 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3452 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003453 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003454 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003455 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003456 else if (flags & METH_O && na == 1) {
3457 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003458 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003459 Py_DECREF(arg);
3460 }
3461 else {
3462 err_args(func, flags, na);
3463 x = NULL;
3464 }
3465 }
3466 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003467 PyObject *callargs;
3468 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003469 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003470 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003471 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003472 Py_XDECREF(callargs);
3473 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003474 } else {
3475 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3476 /* optimize access to bound methods */
3477 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003478 PCALL(PCALL_METHOD);
3479 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003480 Py_INCREF(self);
3481 func = PyMethod_GET_FUNCTION(func);
3482 Py_INCREF(func);
3483 Py_DECREF(*pfunc);
3484 *pfunc = self;
3485 na++;
3486 n++;
3487 } else
3488 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003489 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003490 if (PyFunction_Check(func))
3491 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003492 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003493 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003494 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003495 Py_DECREF(func);
3496 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003497
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003498 /* Clear the stack of the function object. Also removes
3499 the arguments in case they weren't consumed already
3500 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003501 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003502 while ((*pp_stack) > pfunc) {
3503 w = EXT_POP(*pp_stack);
3504 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003505 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003506 }
3507 return x;
3508}
3509
Jeremy Hylton192690e2002-08-16 18:36:11 +00003510/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003511 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003512 For the simplest case -- a function that takes only positional
3513 arguments and is called with only positional arguments -- it
3514 inlines the most primitive frame setup code from
3515 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3516 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003517*/
3518
3519static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003520fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003521{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003522 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003523 PyObject *globals = PyFunction_GET_GLOBALS(func);
3524 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003525 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003526 PyObject **d = NULL;
3527 int nd = 0;
3528
Jeremy Hylton985eba52003-02-05 23:13:00 +00003529 PCALL(PCALL_FUNCTION);
3530 PCALL(PCALL_FAST_FUNCTION);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003531 if (argdefs == NULL && co->co_argcount == n &&
3532 co->co_kwonlyargcount == 0 && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003533 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3534 PyFrameObject *f;
3535 PyObject *retval = NULL;
3536 PyThreadState *tstate = PyThreadState_GET();
3537 PyObject **fastlocals, **stack;
3538 int i;
3539
3540 PCALL(PCALL_FASTER_FUNCTION);
3541 assert(globals != NULL);
3542 /* XXX Perhaps we should create a specialized
3543 PyFrame_New() that doesn't take locals, but does
3544 take builtins without sanity checking them.
3545 */
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003546 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003547 f = PyFrame_New(tstate, co, globals, NULL);
3548 if (f == NULL)
3549 return NULL;
3550
3551 fastlocals = f->f_localsplus;
3552 stack = (*pp_stack) - n;
3553
3554 for (i = 0; i < n; i++) {
3555 Py_INCREF(*stack);
3556 fastlocals[i] = *stack++;
3557 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003558 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003559 ++tstate->recursion_depth;
3560 Py_DECREF(f);
3561 --tstate->recursion_depth;
3562 return retval;
3563 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003564 if (argdefs != NULL) {
3565 d = &PyTuple_GET_ITEM(argdefs, 0);
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00003566 nd = Py_Size(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003567 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003568 return PyEval_EvalCodeEx(co, globals,
3569 (PyObject *)NULL, (*pp_stack)-n, na,
Guido van Rossum4f72a782006-10-27 23:31:49 +00003570 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
Jeremy Hylton985eba52003-02-05 23:13:00 +00003571 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003572}
3573
3574static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003575update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3576 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003577{
3578 PyObject *kwdict = NULL;
3579 if (orig_kwdict == NULL)
3580 kwdict = PyDict_New();
3581 else {
3582 kwdict = PyDict_Copy(orig_kwdict);
3583 Py_DECREF(orig_kwdict);
3584 }
3585 if (kwdict == NULL)
3586 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003587 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003588 int err;
3589 PyObject *value = EXT_POP(*pp_stack);
3590 PyObject *key = EXT_POP(*pp_stack);
3591 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003592 PyErr_Format(PyExc_TypeError,
3593 "%.200s%s got multiple values "
3594 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003595 PyEval_GetFuncName(func),
3596 PyEval_GetFuncDesc(func),
Neal Norwitzda059e32007-08-26 05:33:45 +00003597 PyUnicode_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003598 Py_DECREF(key);
3599 Py_DECREF(value);
3600 Py_DECREF(kwdict);
3601 return NULL;
3602 }
3603 err = PyDict_SetItem(kwdict, key, value);
3604 Py_DECREF(key);
3605 Py_DECREF(value);
3606 if (err) {
3607 Py_DECREF(kwdict);
3608 return NULL;
3609 }
3610 }
3611 return kwdict;
3612}
3613
3614static PyObject *
3615update_star_args(int nstack, int nstar, PyObject *stararg,
3616 PyObject ***pp_stack)
3617{
3618 PyObject *callargs, *w;
3619
3620 callargs = PyTuple_New(nstack + nstar);
3621 if (callargs == NULL) {
3622 return NULL;
3623 }
3624 if (nstar) {
3625 int i;
3626 for (i = 0; i < nstar; i++) {
3627 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3628 Py_INCREF(a);
3629 PyTuple_SET_ITEM(callargs, nstack + i, a);
3630 }
3631 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003632 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003633 w = EXT_POP(*pp_stack);
3634 PyTuple_SET_ITEM(callargs, nstack, w);
3635 }
3636 return callargs;
3637}
3638
3639static PyObject *
3640load_args(PyObject ***pp_stack, int na)
3641{
3642 PyObject *args = PyTuple_New(na);
3643 PyObject *w;
3644
3645 if (args == NULL)
3646 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003647 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003648 w = EXT_POP(*pp_stack);
3649 PyTuple_SET_ITEM(args, na, w);
3650 }
3651 return args;
3652}
3653
3654static PyObject *
3655do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3656{
3657 PyObject *callargs = NULL;
3658 PyObject *kwdict = NULL;
3659 PyObject *result = NULL;
3660
3661 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003662 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003663 if (kwdict == NULL)
3664 goto call_fail;
3665 }
3666 callargs = load_args(pp_stack, na);
3667 if (callargs == NULL)
3668 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003669#ifdef CALL_PROFILE
3670 /* At this point, we have to look at the type of func to
3671 update the call stats properly. Do it here so as to avoid
3672 exposing the call stats machinery outside ceval.c
3673 */
3674 if (PyFunction_Check(func))
3675 PCALL(PCALL_FUNCTION);
3676 else if (PyMethod_Check(func))
3677 PCALL(PCALL_METHOD);
3678 else if (PyType_Check(func))
3679 PCALL(PCALL_TYPE);
3680 else
3681 PCALL(PCALL_OTHER);
3682#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003683 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00003684call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003685 Py_XDECREF(callargs);
3686 Py_XDECREF(kwdict);
3687 return result;
3688}
3689
3690static PyObject *
3691ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3692{
3693 int nstar = 0;
3694 PyObject *callargs = NULL;
3695 PyObject *stararg = NULL;
3696 PyObject *kwdict = NULL;
3697 PyObject *result = NULL;
3698
3699 if (flags & CALL_FLAG_KW) {
3700 kwdict = EXT_POP(*pp_stack);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003701 if (!PyDict_Check(kwdict)) {
3702 PyObject *d;
3703 d = PyDict_New();
3704 if (d == NULL)
3705 goto ext_call_fail;
3706 if (PyDict_Update(d, kwdict) != 0) {
3707 Py_DECREF(d);
3708 /* PyDict_Update raises attribute
3709 * error (percolated from an attempt
3710 * to get 'keys' attribute) instead of
3711 * a type error if its second argument
3712 * is not a mapping.
3713 */
3714 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
3715 PyErr_Format(PyExc_TypeError,
3716 "%.200s%.200s argument after ** "
3717 "must be a mapping, not %.200s",
3718 PyEval_GetFuncName(func),
3719 PyEval_GetFuncDesc(func),
3720 kwdict->ob_type->tp_name);
3721 }
3722 goto ext_call_fail;
3723 }
3724 Py_DECREF(kwdict);
3725 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00003726 }
3727 }
3728 if (flags & CALL_FLAG_VAR) {
3729 stararg = EXT_POP(*pp_stack);
3730 if (!PyTuple_Check(stararg)) {
3731 PyObject *t = NULL;
3732 t = PySequence_Tuple(stararg);
3733 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003734 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3735 PyErr_Format(PyExc_TypeError,
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003736 "%.200s%.200s argument after * "
3737 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003738 PyEval_GetFuncName(func),
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003739 PyEval_GetFuncDesc(func),
3740 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003741 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003742 goto ext_call_fail;
3743 }
3744 Py_DECREF(stararg);
3745 stararg = t;
3746 }
3747 nstar = PyTuple_GET_SIZE(stararg);
3748 }
3749 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003750 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003751 if (kwdict == NULL)
3752 goto ext_call_fail;
3753 }
3754 callargs = update_star_args(na, nstar, stararg, pp_stack);
3755 if (callargs == NULL)
3756 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003757#ifdef CALL_PROFILE
3758 /* At this point, we have to look at the type of func to
3759 update the call stats properly. Do it here so as to avoid
3760 exposing the call stats machinery outside ceval.c
3761 */
3762 if (PyFunction_Check(func))
3763 PCALL(PCALL_FUNCTION);
3764 else if (PyMethod_Check(func))
3765 PCALL(PCALL_METHOD);
3766 else if (PyType_Check(func))
3767 PCALL(PCALL_TYPE);
3768 else
3769 PCALL(PCALL_OTHER);
3770#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003771 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00003772ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003773 Py_XDECREF(callargs);
3774 Py_XDECREF(kwdict);
3775 Py_XDECREF(stararg);
3776 return result;
3777}
3778
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003779/* Extract a slice index from a PyInt or PyLong or an object with the
3780 nb_index slot defined, and store in *pi.
3781 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3782 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 +00003783 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003784*/
Tim Petersb5196382001-12-16 19:44:20 +00003785/* Note: If v is NULL, return success without storing into *pi. This
3786 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3787 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003788*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003789int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003790_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003791{
Tim Petersb5196382001-12-16 19:44:20 +00003792 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003793 Py_ssize_t x;
Guido van Rossumddefaf32007-01-14 03:31:43 +00003794 if (PyInt_CheckExact(v)) {
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003795 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
3796 however, it looks like it should be AsSsize_t.
3797 There should be a comment here explaining why.
3798 */
3799 x = PyInt_AS_LONG(v);
Thomas Wouters477c8d52006-05-27 19:21:47 +00003800 }
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003801 else if (PyIndex_Check(v)) {
3802 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003803 if (x == -1 && PyErr_Occurred())
3804 return 0;
3805 }
3806 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003807 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003808 "slice indices must be integers or "
3809 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003810 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003811 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003812 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003813 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003814 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003815}
3816
Guido van Rossum486364b2007-06-30 05:01:58 +00003817#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Brett Cannon39590462007-02-26 22:01:14 +00003818 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00003819
Guido van Rossumb209a111997-04-29 18:18:01 +00003820static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003821cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003822{
Guido van Rossumac7be682001-01-17 15:42:30 +00003823 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003824 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003825 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003826 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003827 break;
3828 case PyCmp_IS_NOT:
3829 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003830 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003831 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003832 res = PySequence_Contains(w, v);
3833 if (res < 0)
3834 return NULL;
3835 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003836 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003837 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003838 if (res < 0)
3839 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003840 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003841 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003842 case PyCmp_EXC_MATCH:
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003843 if (PyTuple_Check(w)) {
3844 Py_ssize_t i, length;
3845 length = PyTuple_Size(w);
3846 for (i = 0; i < length; i += 1) {
3847 PyObject *exc = PyTuple_GET_ITEM(w, i);
Brett Cannon39590462007-02-26 22:01:14 +00003848 if (!PyExceptionClass_Check(exc)) {
3849 PyErr_SetString(PyExc_TypeError,
3850 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003851 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003852 }
3853 }
3854 }
3855 else {
Brett Cannon39590462007-02-26 22:01:14 +00003856 if (!PyExceptionClass_Check(w)) {
3857 PyErr_SetString(PyExc_TypeError,
3858 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003859 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003860 }
3861 }
Barry Warsaw4249f541997-08-22 21:26:19 +00003862 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003863 break;
3864 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003865 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003866 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003867 v = res ? Py_True : Py_False;
3868 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003869 return v;
3870}
3871
Thomas Wouters52152252000-08-17 22:55:00 +00003872static PyObject *
3873import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00003874{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003875 PyObject *x;
3876
3877 x = PyObject_GetAttr(v, name);
3878 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Walter Dörwald573c08c2007-05-25 15:46:59 +00003879 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003880 }
Thomas Wouters52152252000-08-17 22:55:00 +00003881 return x;
3882}
Guido van Rossumac7be682001-01-17 15:42:30 +00003883
Thomas Wouters52152252000-08-17 22:55:00 +00003884static int
3885import_all_from(PyObject *locals, PyObject *v)
3886{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003887 PyObject *all = PyObject_GetAttrString(v, "__all__");
3888 PyObject *dict, *name, *value;
3889 int skip_leading_underscores = 0;
3890 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00003891
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003892 if (all == NULL) {
3893 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3894 return -1; /* Unexpected error */
3895 PyErr_Clear();
3896 dict = PyObject_GetAttrString(v, "__dict__");
3897 if (dict == NULL) {
3898 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3899 return -1;
3900 PyErr_SetString(PyExc_ImportError,
3901 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00003902 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003903 }
3904 all = PyMapping_Keys(dict);
3905 Py_DECREF(dict);
3906 if (all == NULL)
3907 return -1;
3908 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003909 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003910
3911 for (pos = 0, err = 0; ; pos++) {
3912 name = PySequence_GetItem(all, pos);
3913 if (name == NULL) {
3914 if (!PyErr_ExceptionMatches(PyExc_IndexError))
3915 err = -1;
3916 else
3917 PyErr_Clear();
3918 break;
3919 }
3920 if (skip_leading_underscores &&
Martin v. Löwis5b222132007-06-10 09:51:05 +00003921 PyUnicode_Check(name) &&
3922 PyUnicode_AS_UNICODE(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003923 {
3924 Py_DECREF(name);
3925 continue;
3926 }
3927 value = PyObject_GetAttr(v, name);
3928 if (value == NULL)
3929 err = -1;
Thomas Wouters89f507f2006-12-13 04:49:30 +00003930 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003931 err = PyDict_SetItem(locals, name, value);
Thomas Wouters89f507f2006-12-13 04:49:30 +00003932 else
3933 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003934 Py_DECREF(name);
3935 Py_XDECREF(value);
3936 if (err != 0)
3937 break;
3938 }
3939 Py_DECREF(all);
3940 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003941}
3942
Guido van Rossumac7be682001-01-17 15:42:30 +00003943static void
Neal Norwitzda059e32007-08-26 05:33:45 +00003944format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00003945{
Neal Norwitzda059e32007-08-26 05:33:45 +00003946 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00003947
3948 if (!obj)
3949 return;
3950
Neal Norwitzda059e32007-08-26 05:33:45 +00003951 obj_str = PyUnicode_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00003952 if (!obj_str)
3953 return;
3954
3955 PyErr_Format(exc, format_str, obj_str);
3956}
Guido van Rossum950361c1997-01-24 13:49:28 +00003957
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003958static PyObject *
Guido van Rossum98297ee2007-11-06 21:34:58 +00003959unicode_concatenate(PyObject *v, PyObject *w,
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003960 PyFrameObject *f, unsigned char *next_instr)
3961{
3962 /* This function implements 'variable += expr' when both arguments
Guido van Rossum98297ee2007-11-06 21:34:58 +00003963 are (Unicode) strings. */
3964 Py_ssize_t v_len = PyUnicode_GET_SIZE(v);
3965 Py_ssize_t w_len = PyUnicode_GET_SIZE(w);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003966 Py_ssize_t new_len = v_len + w_len;
3967 if (new_len < 0) {
3968 PyErr_SetString(PyExc_OverflowError,
3969 "strings are too large to concat");
3970 return NULL;
3971 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003972
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003973 if (v->ob_refcnt == 2) {
3974 /* In the common case, there are 2 references to the value
3975 * stored in 'variable' when the += is performed: one on the
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003976 * value stack (in 'v') and one still stored in the
3977 * 'variable'. We try to delete the variable now to reduce
3978 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003979 */
3980 switch (*next_instr) {
3981 case STORE_FAST:
3982 {
3983 int oparg = PEEKARG();
3984 PyObject **fastlocals = f->f_localsplus;
3985 if (GETLOCAL(oparg) == v)
3986 SETLOCAL(oparg, NULL);
3987 break;
3988 }
3989 case STORE_DEREF:
3990 {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003991 PyObject **freevars = (f->f_localsplus +
3992 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003993 PyObject *c = freevars[PEEKARG()];
3994 if (PyCell_GET(c) == v)
3995 PyCell_Set(c, NULL);
3996 break;
3997 }
3998 case STORE_NAME:
3999 {
4000 PyObject *names = f->f_code->co_names;
4001 PyObject *name = GETITEM(names, PEEKARG());
4002 PyObject *locals = f->f_locals;
4003 if (PyDict_CheckExact(locals) &&
4004 PyDict_GetItem(locals, name) == v) {
4005 if (PyDict_DelItem(locals, name) != 0) {
4006 PyErr_Clear();
4007 }
4008 }
4009 break;
4010 }
4011 }
4012 }
4013
Guido van Rossum98297ee2007-11-06 21:34:58 +00004014 if (v->ob_refcnt == 1 && !PyUnicode_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004015 /* Now we own the last reference to 'v', so we can resize it
4016 * in-place.
4017 */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004018 if (PyUnicode_Resize(&v, new_len) != 0) {
4019 /* XXX if PyUnicode_Resize() fails, 'v' has been
Thomas Wouters8ce81f72007-09-20 18:22:40 +00004020 * deallocated so it cannot be put back into
4021 * 'variable'. The MemoryError is raised when there
4022 * is no value in 'variable', which might (very
4023 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004024 */
4025 return NULL;
4026 }
4027 /* copy 'w' into the newly allocated area of 'v' */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004028 memcpy(PyUnicode_AS_UNICODE(v) + v_len,
4029 PyUnicode_AS_UNICODE(w), w_len*sizeof(Py_UNICODE));
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004030 return v;
4031 }
4032 else {
4033 /* When in-place resizing is not an option. */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004034 w = PyUnicode_Concat(v, w);
4035 Py_DECREF(v);
4036 return w;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004037 }
4038}
4039
Guido van Rossum950361c1997-01-24 13:49:28 +00004040#ifdef DYNAMIC_EXECUTION_PROFILE
4041
Skip Montanarof118cb12001-10-15 20:51:38 +00004042static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004043getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004044{
4045 int i;
4046 PyObject *l = PyList_New(256);
4047 if (l == NULL) return NULL;
4048 for (i = 0; i < 256; i++) {
4049 PyObject *x = PyInt_FromLong(a[i]);
4050 if (x == NULL) {
4051 Py_DECREF(l);
4052 return NULL;
4053 }
4054 PyList_SetItem(l, i, x);
4055 }
4056 for (i = 0; i < 256; i++)
4057 a[i] = 0;
4058 return l;
4059}
4060
4061PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004062_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004063{
4064#ifndef DXPAIRS
4065 return getarray(dxp);
4066#else
4067 int i;
4068 PyObject *l = PyList_New(257);
4069 if (l == NULL) return NULL;
4070 for (i = 0; i < 257; i++) {
4071 PyObject *x = getarray(dxpairs[i]);
4072 if (x == NULL) {
4073 Py_DECREF(l);
4074 return NULL;
4075 }
4076 PyList_SetItem(l, i, x);
4077 }
4078 return l;
4079#endif
4080}
4081
4082#endif