blob: defd00299f127d8686a391a126538b87eebe60c7 [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 *);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000110static int 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
Christian Heimesb9eccbf2007-12-05 20:18:38 +0000628 its two unpredictable branches, the HAS_ARG 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); }
Christian Heimes0449f632007-12-15 01:27:15 +0000671#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
672 prtrace((STACK_POINTER)[-1], "ext_pop")), \
673 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000674#else
675#define PUSH(v) BASIC_PUSH(v)
676#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000677#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000678#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000679#endif
680
Guido van Rossum681d79a1995-07-18 14:51:37 +0000681/* Local variable macros */
682
683#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000684
685/* The SETLOCAL() macro must not DECREF the local variable in-place and
686 then store the new value; it must copy the old value to a temporary
687 value, then store the new value, and then DECREF the temporary value.
688 This is because it is possible that during the DECREF the frame is
689 accessed by other code (e.g. a __del__ method or gc.collect()) and the
690 variable would be pointing to already-freed memory. */
691#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
692 GETLOCAL(i) = value; \
693 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000694
Guido van Rossuma027efa1997-05-05 20:56:21 +0000695/* Start of code */
696
Tim Peters5ca576e2001-06-18 22:08:13 +0000697 if (f == NULL)
698 return NULL;
699
Armin Rigo1d313ab2003-10-25 14:33:09 +0000700 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000701 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000702 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000703
Tim Peters5ca576e2001-06-18 22:08:13 +0000704 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000705
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000706 if (tstate->use_tracing) {
707 if (tstate->c_tracefunc != NULL) {
708 /* tstate->c_tracefunc, if defined, is a
709 function that will be called on *every* entry
710 to a code block. Its return value, if not
711 None, is a function that will be called at
712 the start of each executed line of code.
713 (Actually, the function must return itself
714 in order to continue tracing.) The trace
715 functions are called with three arguments:
716 a pointer to the current frame, a string
717 indicating why the function is called, and
718 an argument which depends on the situation.
719 The global trace function is also called
720 whenever an exception is detected. */
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000721 if (call_trace_protected(tstate->c_tracefunc,
722 tstate->c_traceobj,
723 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000724 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000725 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000726 }
727 }
728 if (tstate->c_profilefunc != NULL) {
729 /* Similar for c_profilefunc, except it needn't
730 return itself and isn't called for "line" events */
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000731 if (call_trace_protected(tstate->c_profilefunc,
732 tstate->c_profileobj,
733 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000734 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000735 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000736 }
737 }
738 }
739
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000740 co = f->f_code;
741 names = co->co_names;
742 consts = co->co_consts;
743 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000744 freevars = f->f_localsplus + co->co_nlocals;
Brett Cannonc9371d42005-06-25 08:23:41 +0000745 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000746 /* An explanation is in order for the next line.
747
748 f->f_lasti now refers to the index of the last instruction
749 executed. You might think this was obvious from the name, but
750 this wasn't always true before 2.3! PyFrame_New now sets
751 f->f_lasti to -1 (i.e. the index *before* the first instruction)
752 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000753 does work. Promise.
754
755 When the PREDICT() macros are enabled, some opcode pairs follow in
756 direct succession without updating f->f_lasti. A successful
757 prediction effectively links the two codes together as if they
758 were a single new opcode; accordingly,f->f_lasti will point to
759 the first code in the pair (for instance, GET_ITER followed by
760 FOR_ITER is effectively a single opcode and f->f_lasti will point
761 at to the beginning of the combined pair.)
762 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000763 next_instr = first_instr + f->f_lasti + 1;
764 stack_pointer = f->f_stacktop;
765 assert(stack_pointer != NULL);
766 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
767
Tim Peters5ca576e2001-06-18 22:08:13 +0000768#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000769 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000770#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000771#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum00bc0e02007-10-15 02:52:41 +0000772 filename = PyUnicode_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000773#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000774
Guido van Rossum374a9221991-04-04 10:40:29 +0000775 why = WHY_NOT;
776 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000777 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000778 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000779
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000780 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000781 why = WHY_EXCEPTION;
782 goto on_error;
783 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000784
Guido van Rossum374a9221991-04-04 10:40:29 +0000785 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000786#ifdef WITH_TSC
787 if (inst1 == 0) {
788 /* Almost surely, the opcode executed a break
789 or a continue, preventing inst1 from being set
790 on the way out of the loop.
791 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000792 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000793 loop1 = inst1;
794 }
795 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
796 intr0, intr1);
797 ticked = 0;
798 inst1 = 0;
799 intr0 = 0;
800 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000801 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000802#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000803 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000804 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000805
Guido van Rossuma027efa1997-05-05 20:56:21 +0000806 /* Do periodic things. Doing this every time through
807 the loop would add too much overhead, so we do it
808 only every Nth instruction. We also do it if
809 ``things_to_do'' is set, i.e. when an asynchronous
810 event needs attention (e.g. a signal handler or
811 async I/O handler); see Py_AddPendingCall() and
812 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000813
Skip Montanarod581d772002-09-03 20:10:45 +0000814 if (--_Py_Ticker < 0) {
Thomas Woutersce272b62007-09-19 21:19:28 +0000815 if (*next_instr == SETUP_FINALLY) {
816 /* Make the last opcode before
817 a try: finally: block uninterruptable. */
818 goto fast_next_opcode;
819 }
Skip Montanarod581d772002-09-03 20:10:45 +0000820 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000821 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000822#ifdef WITH_TSC
823 ticked = 1;
824#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000825 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000826 if (Py_MakePendingCalls() < 0) {
827 why = WHY_EXCEPTION;
828 goto on_error;
829 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000830 if (things_to_do)
831 /* MakePendingCalls() didn't succeed.
832 Force early re-execution of this
833 "periodic" code, possibly after
834 a thread switch */
835 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000836 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000837#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000838 if (interpreter_lock) {
839 /* Give another thread a chance */
840
Guido van Rossum25ce5661997-08-02 03:10:38 +0000841 if (PyThreadState_Swap(NULL) != tstate)
842 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000843 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000844
845 /* Other threads may run now */
846
Guido van Rossum65d5b571998-12-21 19:32:43 +0000847 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000848 if (PyThreadState_Swap(tstate) != NULL)
849 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000850
851 /* Check for thread interrupts */
852
853 if (tstate->async_exc != NULL) {
854 x = tstate->async_exc;
855 tstate->async_exc = NULL;
856 PyErr_SetNone(x);
857 Py_DECREF(x);
858 why = WHY_EXCEPTION;
859 goto on_error;
860 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000861 }
862#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000863 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864
Neil Schemenauer63543862002-02-17 19:10:14 +0000865 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000866 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000867
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000868 /* line-by-line tracing support */
869
870 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
871 /* see maybe_call_line_trace
872 for expository comments */
873 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000874
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000875 err = maybe_call_line_trace(tstate->c_tracefunc,
876 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000877 f, &instr_lb, &instr_ub,
878 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000879 /* Reload possibly changed frame fields */
880 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000881 if (f->f_stacktop != NULL) {
882 stack_pointer = f->f_stacktop;
883 f->f_stacktop = NULL;
884 }
885 if (err) {
886 /* trace function raised an exception */
887 goto on_error;
888 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000889 }
890
891 /* Extract opcode and argument */
892
Guido van Rossum374a9221991-04-04 10:40:29 +0000893 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000894 oparg = 0; /* allows oparg to be stored in a register because
895 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000896 if (HAS_ARG(opcode))
897 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000898 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000899#ifdef DYNAMIC_EXECUTION_PROFILE
900#ifdef DXPAIRS
901 dxpairs[lastopcode][opcode]++;
902 lastopcode = opcode;
903#endif
904 dxp[opcode]++;
905#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000906
Guido van Rossum96a42c81992-01-12 02:29:51 +0000907#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000908 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000909
Guido van Rossum96a42c81992-01-12 02:29:51 +0000910 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000911 if (HAS_ARG(opcode)) {
912 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000913 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000914 }
915 else {
916 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000917 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000918 }
919 }
920#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000921
Guido van Rossum374a9221991-04-04 10:40:29 +0000922 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000923 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000924
Guido van Rossum374a9221991-04-04 10:40:29 +0000925 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000926
Guido van Rossum374a9221991-04-04 10:40:29 +0000927 /* BEWARE!
928 It is essential that any operation that fails sets either
929 x to NULL, err to nonzero, or why to anything but WHY_NOT,
930 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000931
Guido van Rossum374a9221991-04-04 10:40:29 +0000932 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000933
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000934 case NOP:
935 goto fast_next_opcode;
936
Neil Schemenauer63543862002-02-17 19:10:14 +0000937 case LOAD_FAST:
938 x = GETLOCAL(oparg);
939 if (x != NULL) {
940 Py_INCREF(x);
941 PUSH(x);
942 goto fast_next_opcode;
943 }
944 format_exc_check_arg(PyExc_UnboundLocalError,
945 UNBOUNDLOCAL_ERROR_MSG,
946 PyTuple_GetItem(co->co_varnames, oparg));
947 break;
948
949 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000950 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000951 Py_INCREF(x);
952 PUSH(x);
953 goto fast_next_opcode;
954
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000955 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000956 case STORE_FAST:
957 v = POP();
958 SETLOCAL(oparg, v);
959 goto fast_next_opcode;
960
Raymond Hettingerf606f872003-03-16 03:11:04 +0000961 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000962 case POP_TOP:
963 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000964 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000965 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000966
Guido van Rossum374a9221991-04-04 10:40:29 +0000967 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000968 v = TOP();
969 w = SECOND();
970 SET_TOP(w);
971 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000972 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000973
Guido van Rossum374a9221991-04-04 10:40:29 +0000974 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000975 v = TOP();
976 w = SECOND();
977 x = THIRD();
978 SET_TOP(w);
979 SET_SECOND(x);
980 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000981 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000982
Thomas Wouters434d0822000-08-24 20:11:32 +0000983 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000984 u = TOP();
985 v = SECOND();
986 w = THIRD();
987 x = FOURTH();
988 SET_TOP(v);
989 SET_SECOND(w);
990 SET_THIRD(x);
991 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000992 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000993
Guido van Rossum374a9221991-04-04 10:40:29 +0000994 case DUP_TOP:
995 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000996 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000997 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000998 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000999
Thomas Wouters434d0822000-08-24 20:11:32 +00001000 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001001 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001002 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001003 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001004 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001005 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001006 STACKADJ(2);
1007 SET_TOP(x);
1008 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001009 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001010 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001011 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001012 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001013 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001014 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001015 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001016 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001017 STACKADJ(3);
1018 SET_TOP(x);
1019 SET_SECOND(w);
1020 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001021 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001022 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001023 Py_FatalError("invalid argument to DUP_TOPX"
1024 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001025 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001026
Guido van Rossum374a9221991-04-04 10:40:29 +00001027 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001028 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001029 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001030 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001031 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001032 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001033 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001034
Guido van Rossum374a9221991-04-04 10:40:29 +00001035 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001036 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001037 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001038 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001039 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001040 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001041 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001042
Guido van Rossum374a9221991-04-04 10:40:29 +00001043 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001044 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001045 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001046 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001047 if (err == 0) {
1048 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001049 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001050 continue;
1051 }
1052 else if (err > 0) {
1053 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001054 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001055 err = 0;
1056 continue;
1057 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001058 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001059 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001060
Guido van Rossum7928cd71991-10-24 14:59:31 +00001061 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001062 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001063 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001064 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001065 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001066 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001067 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001068
Guido van Rossum50564e81996-01-12 01:13:16 +00001069 case BINARY_POWER:
1070 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001071 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001072 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001073 Py_DECREF(v);
1074 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001075 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001076 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001077 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001078
Guido van Rossum374a9221991-04-04 10:40:29 +00001079 case BINARY_MULTIPLY:
1080 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001081 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001082 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001083 Py_DECREF(v);
1084 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001085 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001086 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001087 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001088
Tim Peters3caca232001-12-06 06:23:26 +00001089 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001090 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001091 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001092 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001093 Py_DECREF(v);
1094 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001095 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001096 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001097 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001098
Guido van Rossum4668b002001-08-08 05:00:18 +00001099 case BINARY_FLOOR_DIVIDE:
1100 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001101 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001102 x = PyNumber_FloorDivide(v, w);
1103 Py_DECREF(v);
1104 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001105 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001106 if (x != NULL) continue;
1107 break;
1108
Guido van Rossum374a9221991-04-04 10:40:29 +00001109 case BINARY_MODULO:
1110 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001111 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001112 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001113 Py_DECREF(v);
1114 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001115 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001116 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001117 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001118
Guido van Rossum374a9221991-04-04 10:40:29 +00001119 case BINARY_ADD:
1120 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001121 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001122 if (PyUnicode_CheckExact(v) &&
Guido van Rossum98297ee2007-11-06 21:34:58 +00001123 PyUnicode_CheckExact(w)) {
1124 x = unicode_concatenate(v, w, f, next_instr);
1125 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001126 goto skip_decref_vx;
1127 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001128 else {
Guido van Rossumc12da691997-07-17 23:12:42 +00001129 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001130 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001131 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001132 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001133 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001134 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001135 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001136 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001137
Guido van Rossum374a9221991-04-04 10:40:29 +00001138 case BINARY_SUBTRACT:
1139 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001140 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001141 x = PyNumber_Subtract(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001142 Py_DECREF(v);
1143 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001144 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001145 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001146 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001147
Guido van Rossum374a9221991-04-04 10:40:29 +00001148 case BINARY_SUBSCR:
1149 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001150 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001151 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001152 Py_DECREF(v);
1153 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001154 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001155 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001156 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001157
Guido van Rossum7928cd71991-10-24 14:59:31 +00001158 case BINARY_LSHIFT:
1159 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001160 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001161 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001162 Py_DECREF(v);
1163 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001164 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001165 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001166 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001167
Guido van Rossum7928cd71991-10-24 14:59:31 +00001168 case BINARY_RSHIFT:
1169 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001170 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001171 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001172 Py_DECREF(v);
1173 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001174 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001175 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001176 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001177
Guido van Rossum7928cd71991-10-24 14:59:31 +00001178 case BINARY_AND:
1179 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001180 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001181 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001182 Py_DECREF(v);
1183 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001184 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001185 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001186 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001187
Guido van Rossum7928cd71991-10-24 14:59:31 +00001188 case BINARY_XOR:
1189 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001190 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001191 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001192 Py_DECREF(v);
1193 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001194 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001195 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001196 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001197
Guido van Rossum7928cd71991-10-24 14:59:31 +00001198 case BINARY_OR:
1199 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001200 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001201 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001202 Py_DECREF(v);
1203 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001204 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001205 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001206 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001207
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001208 case LIST_APPEND:
1209 w = POP();
1210 v = POP();
1211 err = PyList_Append(v, w);
1212 Py_DECREF(v);
1213 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001214 if (err == 0) {
1215 PREDICT(JUMP_ABSOLUTE);
1216 continue;
1217 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001218 break;
1219
Nick Coghlan650f0d02007-04-15 12:05:43 +00001220 case SET_ADD:
1221 w = POP();
1222 v = POP();
1223 err = PySet_Add(v, w);
1224 Py_DECREF(v);
1225 Py_DECREF(w);
1226 if (err == 0) {
1227 PREDICT(JUMP_ABSOLUTE);
1228 continue;
1229 }
1230 break;
1231
Thomas Wouters434d0822000-08-24 20:11:32 +00001232 case INPLACE_POWER:
1233 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001234 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001235 x = PyNumber_InPlacePower(v, w, Py_None);
1236 Py_DECREF(v);
1237 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001238 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001239 if (x != NULL) continue;
1240 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001241
Thomas Wouters434d0822000-08-24 20:11:32 +00001242 case INPLACE_MULTIPLY:
1243 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001244 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001245 x = PyNumber_InPlaceMultiply(v, w);
1246 Py_DECREF(v);
1247 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001248 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001249 if (x != NULL) continue;
1250 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001251
Tim Peters54b11912001-12-25 18:49:11 +00001252 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001253 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001254 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001255 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001256 Py_DECREF(v);
1257 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001258 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001259 if (x != NULL) continue;
1260 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001261
Guido van Rossum4668b002001-08-08 05:00:18 +00001262 case INPLACE_FLOOR_DIVIDE:
1263 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001264 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001265 x = PyNumber_InPlaceFloorDivide(v, w);
1266 Py_DECREF(v);
1267 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001268 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001269 if (x != NULL) continue;
1270 break;
1271
Thomas Wouters434d0822000-08-24 20:11:32 +00001272 case INPLACE_MODULO:
1273 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001274 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001275 x = PyNumber_InPlaceRemainder(v, w);
1276 Py_DECREF(v);
1277 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001278 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001279 if (x != NULL) continue;
1280 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001281
Thomas Wouters434d0822000-08-24 20:11:32 +00001282 case INPLACE_ADD:
1283 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001284 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001285 if (PyUnicode_CheckExact(v) &&
Guido van Rossum98297ee2007-11-06 21:34:58 +00001286 PyUnicode_CheckExact(w)) {
1287 x = unicode_concatenate(v, w, f, next_instr);
1288 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001289 goto skip_decref_v;
1290 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001291 else {
Thomas Wouters434d0822000-08-24 20:11:32 +00001292 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001293 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001294 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001295 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001296 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001297 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001298 if (x != NULL) continue;
1299 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001300
Thomas Wouters434d0822000-08-24 20:11:32 +00001301 case INPLACE_SUBTRACT:
1302 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001303 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001304 x = PyNumber_InPlaceSubtract(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001305 Py_DECREF(v);
1306 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001307 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001308 if (x != NULL) continue;
1309 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001310
Thomas Wouters434d0822000-08-24 20:11:32 +00001311 case INPLACE_LSHIFT:
1312 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001313 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001314 x = PyNumber_InPlaceLshift(v, w);
1315 Py_DECREF(v);
1316 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001317 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001318 if (x != NULL) continue;
1319 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001320
Thomas Wouters434d0822000-08-24 20:11:32 +00001321 case INPLACE_RSHIFT:
1322 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001323 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001324 x = PyNumber_InPlaceRshift(v, w);
1325 Py_DECREF(v);
1326 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001327 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001328 if (x != NULL) continue;
1329 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001330
Thomas Wouters434d0822000-08-24 20:11:32 +00001331 case INPLACE_AND:
1332 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001333 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001334 x = PyNumber_InPlaceAnd(v, w);
1335 Py_DECREF(v);
1336 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001337 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001338 if (x != NULL) continue;
1339 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001340
Thomas Wouters434d0822000-08-24 20:11:32 +00001341 case INPLACE_XOR:
1342 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001343 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001344 x = PyNumber_InPlaceXor(v, w);
1345 Py_DECREF(v);
1346 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001347 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001348 if (x != NULL) continue;
1349 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001350
Thomas Wouters434d0822000-08-24 20:11:32 +00001351 case INPLACE_OR:
1352 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001353 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001354 x = PyNumber_InPlaceOr(v, w);
1355 Py_DECREF(v);
1356 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001357 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001358 if (x != NULL) continue;
1359 break;
1360
Guido van Rossum374a9221991-04-04 10:40:29 +00001361 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001362 w = TOP();
1363 v = SECOND();
1364 u = THIRD();
1365 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001366 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001367 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001368 Py_DECREF(u);
1369 Py_DECREF(v);
1370 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001371 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001372 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001373
Guido van Rossum374a9221991-04-04 10:40:29 +00001374 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001375 w = TOP();
1376 v = SECOND();
1377 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001378 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001379 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001380 Py_DECREF(v);
1381 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001382 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001383 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001384
Guido van Rossum374a9221991-04-04 10:40:29 +00001385 case PRINT_EXPR:
1386 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001387 w = PySys_GetObject("displayhook");
1388 if (w == NULL) {
1389 PyErr_SetString(PyExc_RuntimeError,
1390 "lost sys.displayhook");
1391 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001392 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001393 }
1394 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001395 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001396 if (x == NULL)
1397 err = -1;
1398 }
1399 if (err == 0) {
1400 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001401 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001402 if (w == NULL)
1403 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001404 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001405 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001406 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001407 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001408
Thomas Wouters434d0822000-08-24 20:11:32 +00001409#ifdef CASE_TOO_BIG
1410 default: switch (opcode) {
1411#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001412 case RAISE_VARARGS:
Collin Winter828f04a2007-08-31 00:04:24 +00001413 v = w = NULL;
Guido van Rossumf10570b1995-07-07 22:53:21 +00001414 switch (oparg) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00001415 case 2:
1416 v = POP(); /* cause */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001417 case 1:
1418 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001419 case 0: /* Fallthrough */
Collin Winter828f04a2007-08-31 00:04:24 +00001420 why = do_raise(w, v);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001421 break;
1422 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001423 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001424 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001425 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001426 break;
1427 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001428 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001429
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001430 case STORE_LOCALS:
1431 x = POP();
1432 v = f->f_locals;
1433 Py_XDECREF(v);
1434 f->f_locals = x;
1435 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001436
Guido van Rossum374a9221991-04-04 10:40:29 +00001437 case RETURN_VALUE:
1438 retval = POP();
1439 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001440 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001441
Tim Peters5ca576e2001-06-18 22:08:13 +00001442 case YIELD_VALUE:
1443 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001444 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001445 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001446 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001447
Guido van Rossum374a9221991-04-04 10:40:29 +00001448 case POP_BLOCK:
1449 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001450 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001451 while (STACK_LEVEL() > b->b_level) {
1452 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001453 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001454 }
1455 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001456 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001457
Christian Heimes180510d2008-03-03 19:15:45 +00001458 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001459 case END_FINALLY:
1460 v = POP();
Christian Heimes217cfd12007-12-02 14:31:20 +00001461 if (PyLong_Check(v)) {
1462 why = (enum why_code) PyLong_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001463 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001464 if (why == WHY_RETURN ||
1465 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001466 retval = POP();
1467 }
Brett Cannonf74225d2007-02-26 21:10:16 +00001468 else if (PyExceptionClass_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001469 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001470 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001471 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001472 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001473 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001474 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001475 else if (v != Py_None) {
1476 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001477 "'finally' pops bad exception");
1478 why = WHY_EXCEPTION;
1479 }
Barry Warsaw8d109cb2008-05-08 04:26:35 +00001480 /*
1481 Make sure the exception state is cleaned up before
1482 the end of an except block. This ensures objects
1483 referenced by the exception state are not kept
1484 alive too long.
1485 See #2507.
1486 */
1487 if (tstate->frame->f_exc_type != NULL)
1488 reset_exc_info(tstate);
1489 else {
1490 assert(tstate->frame->f_exc_value == NULL);
1491 assert(tstate->frame->f_exc_traceback == NULL);
1492 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001493 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001494 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001495
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001496 case LOAD_BUILD_CLASS:
1497 x = PyDict_GetItemString(f->f_builtins,
1498 "__build_class__");
1499 if (x == NULL) {
1500 PyErr_SetString(PyExc_ImportError,
1501 "__build_class__ not found");
1502 break;
1503 }
1504 Py_INCREF(x);
1505 PUSH(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001506 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001507
Guido van Rossum374a9221991-04-04 10:40:29 +00001508 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001509 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001510 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001511 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001512 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001513 err = PyDict_SetItem(x, w, v);
1514 else
1515 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001516 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001517 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001518 break;
1519 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001520 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001521 "no locals found when storing %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001522 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001523
Guido van Rossum374a9221991-04-04 10:40:29 +00001524 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001525 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001526 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001527 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001528 format_exc_check_arg(PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001529 NAME_ERROR_MSG,
1530 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001531 break;
1532 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001533 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001534 "no locals when deleting %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001535 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001536
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001537 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001538 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001539 v = POP();
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001540 if (PyTuple_CheckExact(v) &&
1541 PyTuple_GET_SIZE(v) == oparg) {
1542 PyObject **items = \
1543 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001544 while (oparg--) {
1545 w = items[oparg];
1546 Py_INCREF(w);
1547 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001548 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001549 Py_DECREF(v);
1550 continue;
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001551 } else if (PyList_CheckExact(v) &&
1552 PyList_GET_SIZE(v) == oparg) {
1553 PyObject **items = \
1554 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001555 while (oparg--) {
1556 w = items[oparg];
1557 Py_INCREF(w);
1558 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001559 }
Guido van Rossum0368b722007-05-11 16:50:42 +00001560 } else if (unpack_iterable(v, oparg, -1,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001561 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001562 stack_pointer += oparg;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001563 } else {
1564 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001565 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001566 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001567 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001568 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001569
Guido van Rossum0368b722007-05-11 16:50:42 +00001570 case UNPACK_EX:
1571 {
1572 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
1573 v = POP();
1574
1575 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
1576 stack_pointer + totalargs)) {
1577 stack_pointer += totalargs;
1578 } else {
1579 why = WHY_EXCEPTION;
1580 }
1581 Py_DECREF(v);
1582 break;
1583 }
1584
Guido van Rossum374a9221991-04-04 10:40:29 +00001585 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001586 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001587 v = TOP();
1588 u = SECOND();
1589 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001590 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1591 Py_DECREF(v);
1592 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001593 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001594 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001595
Guido van Rossum374a9221991-04-04 10:40:29 +00001596 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001597 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001598 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001599 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1600 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001601 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001602 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001603
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001604 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001605 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001606 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001607 err = PyDict_SetItem(f->f_globals, w, v);
1608 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001609 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001610 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001611
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001612 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001613 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001614 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001615 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001616 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001617 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001618
Guido van Rossum374a9221991-04-04 10:40:29 +00001619 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001620 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001621 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001622 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001623 "no locals when loading %R", w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001624 break;
1625 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001626 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001627 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001628 Py_XINCREF(x);
1629 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001630 else {
1631 x = PyObject_GetItem(v, w);
1632 if (x == NULL && PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001633 if (!PyErr_ExceptionMatches(
1634 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001635 break;
1636 PyErr_Clear();
1637 }
1638 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001639 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001640 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001641 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001642 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001643 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001644 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001645 PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001646 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001647 break;
1648 }
1649 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001650 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001651 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001652 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001653 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001654
Guido van Rossum374a9221991-04-04 10:40:29 +00001655 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001656 w = GETITEM(names, oparg);
Neal Norwitzda059e32007-08-26 05:33:45 +00001657 if (PyUnicode_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001658 /* Inline the PyDict_GetItem() calls.
1659 WARNING: this is an extreme speed hack.
1660 Do not try this at home. */
Neal Norwitzda059e32007-08-26 05:33:45 +00001661 long hash = ((PyUnicodeObject *)w)->hash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001662 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001663 PyDictObject *d;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001664 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001665 d = (PyDictObject *)(f->f_globals);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001666 e = d->ma_lookup(d, w, hash);
1667 if (e == NULL) {
1668 x = NULL;
1669 break;
1670 }
1671 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001672 if (x != NULL) {
1673 Py_INCREF(x);
1674 PUSH(x);
1675 continue;
1676 }
1677 d = (PyDictObject *)(f->f_builtins);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001678 e = d->ma_lookup(d, w, hash);
1679 if (e == NULL) {
1680 x = NULL;
1681 break;
1682 }
1683 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001684 if (x != NULL) {
1685 Py_INCREF(x);
1686 PUSH(x);
1687 continue;
1688 }
1689 goto load_global_error;
1690 }
1691 }
1692 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001693 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001694 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001695 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001696 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001697 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001698 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001699 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001700 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001701 break;
1702 }
1703 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001704 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001705 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001706 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001707
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001708 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001709 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001710 if (x != NULL) {
1711 SETLOCAL(oparg, NULL);
1712 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001713 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001714 format_exc_check_arg(
1715 PyExc_UnboundLocalError,
1716 UNBOUNDLOCAL_ERROR_MSG,
1717 PyTuple_GetItem(co->co_varnames, oparg)
1718 );
1719 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001720
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001721 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001722 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001723 Py_INCREF(x);
1724 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001725 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001726 break;
1727
1728 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001729 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001730 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001731 if (w != NULL) {
1732 PUSH(w);
1733 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001734 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001735 err = -1;
1736 /* Don't stomp existing exception */
1737 if (PyErr_Occurred())
1738 break;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001739 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1740 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001741 oparg);
1742 format_exc_check_arg(
1743 PyExc_UnboundLocalError,
1744 UNBOUNDLOCAL_ERROR_MSG,
1745 v);
1746 } else {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001747 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
1748 PyTuple_GET_SIZE(co->co_cellvars));
1749 format_exc_check_arg(PyExc_NameError,
1750 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001751 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001752 break;
1753
1754 case STORE_DEREF:
1755 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001756 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001757 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001758 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001759 continue;
1760
Guido van Rossum374a9221991-04-04 10:40:29 +00001761 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001762 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001763 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001764 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001765 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001766 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001767 }
1768 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001769 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001770 }
1771 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001772
Guido van Rossum374a9221991-04-04 10:40:29 +00001773 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001774 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001775 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001776 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001777 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001778 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001779 }
1780 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001781 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001782 }
1783 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001784
Guido van Rossum86e58e22006-08-28 15:27:34 +00001785 case BUILD_SET:
1786 x = PySet_New(NULL);
1787 if (x != NULL) {
1788 for (; --oparg >= 0;) {
1789 w = POP();
1790 if (err == 0)
1791 err = PySet_Add(x, w);
1792 Py_DECREF(w);
1793 }
1794 if (err != 0) {
1795 Py_DECREF(x);
1796 break;
1797 }
1798 PUSH(x);
1799 continue;
1800 }
1801 break;
1802
Guido van Rossum374a9221991-04-04 10:40:29 +00001803 case BUILD_MAP:
Christian Heimes99170a52007-12-19 02:07:34 +00001804 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001805 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001806 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001807 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001808
Christian Heimes99170a52007-12-19 02:07:34 +00001809 case STORE_MAP:
1810 w = TOP(); /* key */
1811 u = SECOND(); /* value */
1812 v = THIRD(); /* dict */
1813 STACKADJ(-2);
1814 assert (PyDict_CheckExact(v));
1815 err = PyDict_SetItem(v, w, u); /* v[w] = u */
1816 Py_DECREF(u);
1817 Py_DECREF(w);
1818 if (err == 0) continue;
1819 break;
1820
Guido van Rossum374a9221991-04-04 10:40:29 +00001821 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001822 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001823 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001824 x = PyObject_GetAttr(v, w);
1825 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001826 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001827 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001828 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001829
Guido van Rossum374a9221991-04-04 10:40:29 +00001830 case COMPARE_OP:
1831 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001832 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001833 x = cmp_outcome(oparg, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001834 Py_DECREF(v);
1835 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001836 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001837 if (x == NULL) break;
1838 PREDICT(JUMP_IF_FALSE);
1839 PREDICT(JUMP_IF_TRUE);
1840 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001841
Guido van Rossum374a9221991-04-04 10:40:29 +00001842 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001843 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001844 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001845 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001846 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00001847 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001848 break;
1849 }
Christian Heimesbbffeb62008-01-24 09:42:52 +00001850 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001851 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001852 u = TOP();
Christian Heimes217cfd12007-12-02 14:31:20 +00001853 if (PyLong_AsLong(u) != -1 || PyErr_Occurred())
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001854 w = PyTuple_Pack(5,
1855 w,
1856 f->f_globals,
1857 f->f_locals == NULL ?
1858 Py_None : f->f_locals,
1859 v,
1860 u);
1861 else
1862 w = PyTuple_Pack(4,
1863 w,
1864 f->f_globals,
1865 f->f_locals == NULL ?
1866 Py_None : f->f_locals,
1867 v);
1868 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001869 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001870 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001871 u = POP();
Christian Heimesbbffeb62008-01-24 09:42:52 +00001872 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001873 x = NULL;
1874 break;
1875 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001876 READ_TIMESTAMP(intr0);
Christian Heimesbbffeb62008-01-24 09:42:52 +00001877 v = x;
1878 x = PyEval_CallObject(v, w);
1879 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001880 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001881 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001882 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001883 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001884 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001885
Thomas Wouters52152252000-08-17 22:55:00 +00001886 case IMPORT_STAR:
1887 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001888 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001889 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00001890 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001891 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00001892 break;
1893 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001894 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00001895 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001896 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001897 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00001898 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001899 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001900 break;
Guido van Rossum25831651993-05-19 14:50:45 +00001901
Thomas Wouters52152252000-08-17 22:55:00 +00001902 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00001903 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00001904 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001905 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00001906 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001907 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00001908 PUSH(x);
1909 if (x != NULL) continue;
1910 break;
1911
Guido van Rossum374a9221991-04-04 10:40:29 +00001912 case JUMP_FORWARD:
1913 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001914 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001915
Raymond Hettingerf606f872003-03-16 03:11:04 +00001916 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00001917 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00001918 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00001919 if (w == Py_True) {
1920 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001921 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00001922 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00001923 if (w == Py_False) {
1924 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001925 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00001926 }
1927 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001928 if (err > 0)
1929 err = 0;
1930 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00001931 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001932 else
1933 break;
1934 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001935
Raymond Hettingerf606f872003-03-16 03:11:04 +00001936 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00001937 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00001938 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00001939 if (w == Py_False) {
1940 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001941 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00001942 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00001943 if (w == Py_True) {
1944 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00001945 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00001946 }
1947 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001948 if (err > 0) {
1949 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00001950 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001951 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001952 else if (err == 0)
1953 ;
1954 else
1955 break;
1956 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001957
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001958 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00001959 case JUMP_ABSOLUTE:
1960 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00001961#if FAST_LOOPS
1962 /* Enabling this path speeds-up all while and for-loops by bypassing
1963 the per-loop checks for signals. By default, this should be turned-off
1964 because it prevents detection of a control-break in tight loops like
1965 "while 1: pass". Compile with this option turned-on when you need
1966 the speed-up and do not need break checking inside tight loops (ones
1967 that contain only instructions ending with goto fast_next_opcode).
1968 */
1969 goto fast_next_opcode;
1970#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00001971 continue;
Guido van Rossum58da9312007-11-10 23:39:45 +00001972#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00001973
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001974 case GET_ITER:
1975 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00001976 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001977 x = PyObject_GetIter(v);
1978 Py_DECREF(v);
1979 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001980 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001981 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00001982 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001983 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001984 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001985 break;
1986
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001987 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001988 case FOR_ITER:
1989 /* before: [iter]; after: [iter, iter()] *or* [] */
1990 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00001991 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00001992 if (x != NULL) {
1993 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001994 PREDICT(STORE_FAST);
1995 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00001996 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001997 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00001998 if (PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001999 if (!PyErr_ExceptionMatches(
2000 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002001 break;
2002 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002003 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002004 /* iterator ended normally */
2005 x = v = POP();
2006 Py_DECREF(v);
2007 JUMPBY(oparg);
2008 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002009
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002010 case BREAK_LOOP:
2011 why = WHY_BREAK;
2012 goto fast_block_end;
2013
2014 case CONTINUE_LOOP:
Christian Heimes217cfd12007-12-02 14:31:20 +00002015 retval = PyLong_FromLong(oparg);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002016 if (!retval) {
2017 x = NULL;
2018 break;
2019 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002020 why = WHY_CONTINUE;
2021 goto fast_block_end;
2022
Guido van Rossum374a9221991-04-04 10:40:29 +00002023 case SETUP_LOOP:
2024 case SETUP_EXCEPT:
2025 case SETUP_FINALLY:
Thomas Wouters8ce81f72007-09-20 18:22:40 +00002026 /* NOTE: If you add any new block-setup opcodes that
2027 are not try/except/finally handlers, you may need
2028 to update the PyGen_NeedsFinalizing() function.
2029 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002030
Guido van Rossumb209a111997-04-29 18:18:01 +00002031 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002032 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002033 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002034
Guido van Rossumc2e20742006-02-27 22:32:47 +00002035 case WITH_CLEANUP:
2036 {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002037 /* At the top of the stack are 1-3 values indicating
2038 how/why we entered the finally clause:
2039 - TOP = None
2040 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2041 - TOP = WHY_*; no retval below it
2042 - (TOP, SECOND, THIRD) = exc_info()
2043 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002044 In the last case, we must call
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002045 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002046 otherwise we must call
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002047 EXIT(None, None, None)
2048
2049 In all cases, we remove EXIT from the stack, leaving
2050 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002051
2052 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002053 *and* the function call returns a 'true' value, we
2054 "zap" this information, to prevent END_FINALLY from
2055 re-raising the exception. (But non-local gotos
2056 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002057 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002058
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002059 PyObject *exit_func;
2060
2061 u = POP();
2062 if (u == Py_None) {
2063 exit_func = TOP();
2064 SET_TOP(u);
2065 v = w = Py_None;
2066 }
2067 else if (PyLong_Check(u)) {
2068 switch(PyLong_AS_LONG(u)) {
2069 case WHY_RETURN:
2070 case WHY_CONTINUE:
2071 /* Retval in TOP. */
2072 exit_func = SECOND();
2073 SET_SECOND(TOP());
2074 SET_TOP(u);
2075 break;
2076 default:
2077 exit_func = TOP();
2078 SET_TOP(u);
2079 break;
2080 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002081 u = v = w = Py_None;
2082 }
2083 else {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002084 v = TOP();
2085 w = SECOND();
2086 exit_func = THIRD();
2087 SET_TOP(u);
2088 SET_SECOND(v);
2089 SET_THIRD(w);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002090 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002091 /* XXX Not the fastest way to call it... */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002092 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2093 NULL);
2094 if (x == NULL) {
2095 Py_DECREF(exit_func);
Guido van Rossumf6694362006-03-10 02:28:35 +00002096 break; /* Go to error exit */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002097 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002098 if (u != Py_None && PyObject_IsTrue(x)) {
2099 /* There was an exception and a true return */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002100 STACKADJ(-2);
Guido van Rossumf6694362006-03-10 02:28:35 +00002101 Py_INCREF(Py_None);
2102 SET_TOP(Py_None);
Guido van Rossumf6694362006-03-10 02:28:35 +00002103 Py_DECREF(u);
2104 Py_DECREF(v);
2105 Py_DECREF(w);
2106 } else {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002107 /* The stack was rearranged to remove EXIT
2108 above. Let END_FINALLY do its thing */
Guido van Rossumf6694362006-03-10 02:28:35 +00002109 }
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002110 Py_DECREF(x);
2111 Py_DECREF(exit_func);
Christian Heimes180510d2008-03-03 19:15:45 +00002112 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002113 break;
2114 }
2115
Guido van Rossumf10570b1995-07-07 22:53:21 +00002116 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002117 {
2118 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002119 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002120 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002121#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002122 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002123#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002124 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002125#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002126 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002127 PUSH(x);
2128 if (x != NULL)
2129 continue;
2130 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002131 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002132
Jeremy Hylton76901512000-03-28 23:49:17 +00002133 case CALL_FUNCTION_VAR:
2134 case CALL_FUNCTION_KW:
2135 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002136 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002137 int na = oparg & 0xff;
2138 int nk = (oparg>>8) & 0xff;
2139 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002140 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002141 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002142 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002143 if (flags & CALL_FLAG_VAR)
2144 n++;
2145 if (flags & CALL_FLAG_KW)
2146 n++;
2147 pfunc = stack_pointer - n - 1;
2148 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002149
Guido van Rossumac7be682001-01-17 15:42:30 +00002150 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002151 && PyMethod_GET_SELF(func) != NULL) {
2152 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002153 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002154 func = PyMethod_GET_FUNCTION(func);
2155 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002156 Py_DECREF(*pfunc);
2157 *pfunc = self;
2158 na++;
2159 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002160 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002161 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002162 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002163 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002164 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002165 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002166 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002167 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002168
Jeremy Hylton76901512000-03-28 23:49:17 +00002169 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002170 w = POP();
2171 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002172 }
2173 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002174 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002175 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002176 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002177 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002178
Guido van Rossum0240b922007-02-26 21:23:50 +00002179 case MAKE_CLOSURE:
Guido van Rossum681d79a1995-07-18 14:51:37 +00002180 case MAKE_FUNCTION:
Guido van Rossum4f72a782006-10-27 23:31:49 +00002181 {
2182 int posdefaults = oparg & 0xff;
2183 int kwdefaults = (oparg>>8) & 0xff;
Neal Norwitzc1505362006-12-28 06:47:50 +00002184 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002185
Guido van Rossum681d79a1995-07-18 14:51:37 +00002186 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002187 x = PyFunction_New(v, f->f_globals);
2188 Py_DECREF(v);
Guido van Rossum0240b922007-02-26 21:23:50 +00002189
2190 if (x != NULL && opcode == MAKE_CLOSURE) {
2191 v = POP();
2192 err = PyFunction_SetClosure(x, v);
2193 Py_DECREF(v);
2194 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002195
2196 if (x != NULL && num_annotations > 0) {
2197 Py_ssize_t name_ix;
2198 u = POP(); /* names of args with annotations */
2199 v = PyDict_New();
2200 if (v == NULL) {
2201 Py_DECREF(x);
2202 x = NULL;
2203 break;
2204 }
2205 name_ix = PyTuple_Size(u);
2206 assert(num_annotations == name_ix+1);
2207 while (name_ix > 0) {
2208 --name_ix;
2209 t = PyTuple_GET_ITEM(u, name_ix);
2210 w = POP();
2211 /* XXX(nnorwitz): check for errors */
2212 PyDict_SetItem(v, t, w);
2213 Py_DECREF(w);
2214 }
2215
2216 err = PyFunction_SetAnnotations(x, v);
2217 Py_DECREF(v);
2218 Py_DECREF(u);
2219 }
2220
Guido van Rossum681d79a1995-07-18 14:51:37 +00002221 /* XXX Maybe this should be a separate opcode? */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002222 if (x != NULL && posdefaults > 0) {
2223 v = PyTuple_New(posdefaults);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002224 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002225 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002226 x = NULL;
2227 break;
2228 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002229 while (--posdefaults >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002230 w = POP();
Guido van Rossum4f72a782006-10-27 23:31:49 +00002231 PyTuple_SET_ITEM(v, posdefaults, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002232 }
2233 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002234 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002235 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002236 if (x != NULL && kwdefaults > 0) {
2237 v = PyDict_New();
2238 if (v == NULL) {
2239 Py_DECREF(x);
2240 x = NULL;
2241 break;
2242 }
2243 while (--kwdefaults >= 0) {
2244 w = POP(); /* default value */
2245 u = POP(); /* kw only arg name */
Neal Norwitzc1505362006-12-28 06:47:50 +00002246 /* XXX(nnorwitz): check for errors */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002247 PyDict_SetItem(v, u, w);
Georg Brandl94ab0002007-02-26 13:58:18 +00002248 Py_DECREF(w);
2249 Py_DECREF(u);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002250 }
2251 err = PyFunction_SetKwDefaults(x, v);
2252 Py_DECREF(v);
2253 }
Guido van Rossum681d79a1995-07-18 14:51:37 +00002254 PUSH(x);
2255 break;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002256 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002257
2258 case BUILD_SLICE:
2259 if (oparg == 3)
2260 w = POP();
2261 else
2262 w = NULL;
2263 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002264 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002265 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002266 Py_DECREF(u);
2267 Py_DECREF(v);
2268 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002269 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002270 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002271 break;
2272
Fred Drakeef8ace32000-08-24 00:32:09 +00002273 case EXTENDED_ARG:
2274 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002275 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002276 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002277
Guido van Rossum374a9221991-04-04 10:40:29 +00002278 default:
2279 fprintf(stderr,
2280 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002281 PyCode_Addr2Line(f->f_code, f->f_lasti),
2282 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002283 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002284 why = WHY_EXCEPTION;
2285 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002286
2287#ifdef CASE_TOO_BIG
2288 }
2289#endif
2290
Guido van Rossum374a9221991-04-04 10:40:29 +00002291 } /* switch */
2292
2293 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002294
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002295 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002296
Guido van Rossum374a9221991-04-04 10:40:29 +00002297 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002298
Guido van Rossum374a9221991-04-04 10:40:29 +00002299 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002300 if (err == 0 && x != NULL) {
2301#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002302 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002303 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002304 fprintf(stderr,
2305 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002306 else {
2307#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002308 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002309 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002310#ifdef CHECKEXC
2311 }
2312#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002313 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002314 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002315 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002316 err = 0;
2317 }
2318
Guido van Rossum374a9221991-04-04 10:40:29 +00002319 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002320
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002321 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002322 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002323 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002324 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002325 why = WHY_EXCEPTION;
2326 }
2327 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002328#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002329 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002330 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002331 if (PyErr_Occurred()) {
Christian Heimes412dc9c2008-01-27 18:55:54 +00002332 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002333 sprintf(buf, "Stack unwind with exception "
2334 "set and why=%d", why);
2335 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002336 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002337 }
2338#endif
2339
2340 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002341
Guido van Rossum374a9221991-04-04 10:40:29 +00002342 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002343 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002344
Fred Drake8f51f542001-10-04 14:48:42 +00002345 if (tstate->c_tracefunc != NULL)
2346 call_exc_trace(tstate->c_tracefunc,
2347 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002348 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002349
Guido van Rossum374a9221991-04-04 10:40:29 +00002350 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002351
Guido van Rossum374a9221991-04-04 10:40:29 +00002352 if (why == WHY_RERAISE)
2353 why = WHY_EXCEPTION;
2354
2355 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002356
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002357fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002358 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002359 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002360
Tim Peters8a5c3c72004-04-05 19:36:21 +00002361 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002362 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2363 /* For a continue inside a try block,
2364 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002365 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2366 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002367 why = WHY_NOT;
Christian Heimes217cfd12007-12-02 14:31:20 +00002368 JUMPTO(PyLong_AS_LONG(retval));
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002369 Py_DECREF(retval);
2370 break;
2371 }
2372
Guido van Rossum374a9221991-04-04 10:40:29 +00002373 while (STACK_LEVEL() > b->b_level) {
2374 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002375 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002376 }
2377 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2378 why = WHY_NOT;
2379 JUMPTO(b->b_handler);
2380 break;
2381 }
2382 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002383 (b->b_type == SETUP_EXCEPT &&
2384 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002385 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002386 PyObject *exc, *val, *tb;
2387 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002388 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002389 val = Py_None;
2390 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002391 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002392 /* Make the raw exception data
2393 available to the handler,
2394 so a program can emulate the
2395 Python main loop. Don't do
2396 this for 'finally'. */
2397 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002398 PyErr_NormalizeException(
2399 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002400 set_exc_info(tstate,
2401 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002402 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002403 if (tb == NULL) {
2404 Py_INCREF(Py_None);
2405 PUSH(Py_None);
2406 } else
2407 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002408 PUSH(val);
2409 PUSH(exc);
2410 }
2411 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002412 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002413 PUSH(retval);
Christian Heimes217cfd12007-12-02 14:31:20 +00002414 v = PyLong_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002415 PUSH(v);
2416 }
2417 why = WHY_NOT;
2418 JUMPTO(b->b_handler);
2419 break;
2420 }
2421 } /* unwind stack */
2422
2423 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002424
Guido van Rossum374a9221991-04-04 10:40:29 +00002425 if (why != WHY_NOT)
2426 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002427 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002428
Guido van Rossum374a9221991-04-04 10:40:29 +00002429 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002430
Tim Peters8a5c3c72004-04-05 19:36:21 +00002431 assert(why != WHY_YIELD);
2432 /* Pop remaining stack entries. */
2433 while (!EMPTY()) {
2434 v = POP();
2435 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002436 }
2437
Tim Peters8a5c3c72004-04-05 19:36:21 +00002438 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002439 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002440
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002441fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002442 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002443 if (tstate->c_tracefunc) {
2444 if (why == WHY_RETURN || why == WHY_YIELD) {
2445 if (call_trace(tstate->c_tracefunc,
2446 tstate->c_traceobj, f,
2447 PyTrace_RETURN, retval)) {
2448 Py_XDECREF(retval);
2449 retval = NULL;
2450 why = WHY_EXCEPTION;
2451 }
2452 }
2453 else if (why == WHY_EXCEPTION) {
2454 call_trace_protected(tstate->c_tracefunc,
2455 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002456 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002457 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002458 }
Fred Drake8f51f542001-10-04 14:48:42 +00002459 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002460 if (why == WHY_EXCEPTION)
2461 call_trace_protected(tstate->c_profilefunc,
2462 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002463 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002464 else if (call_trace(tstate->c_profilefunc,
2465 tstate->c_profileobj, f,
2466 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002467 Py_XDECREF(retval);
2468 retval = NULL;
2469 why = WHY_EXCEPTION;
2470 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002471 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002472 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002473
Thomas Wouters477c8d52006-05-27 19:21:47 +00002474 if (tstate->frame->f_exc_type != NULL)
2475 reset_exc_info(tstate);
2476 else {
2477 assert(tstate->frame->f_exc_value == NULL);
2478 assert(tstate->frame->f_exc_traceback == NULL);
2479 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002480
Tim Peters5ca576e2001-06-18 22:08:13 +00002481 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00002482exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002483 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002484 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002485
Guido van Rossum96a42c81992-01-12 02:29:51 +00002486 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002487}
2488
Guido van Rossumc2e20742006-02-27 22:32:47 +00002489/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002490 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002491 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002492
Tim Peters6d6c1a32001-08-02 04:15:00 +00002493PyObject *
2494PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002495 PyObject **args, int argcount, PyObject **kws, int kwcount,
Guido van Rossum4f72a782006-10-27 23:31:49 +00002496 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00002497{
2498 register PyFrameObject *f;
2499 register PyObject *retval = NULL;
2500 register PyObject **fastlocals, **freevars;
2501 PyThreadState *tstate = PyThreadState_GET();
2502 PyObject *x, *u;
2503
2504 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002505 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002506 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002507 return NULL;
2508 }
2509
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002510 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002511 assert(globals != NULL);
2512 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002513 if (f == NULL)
2514 return NULL;
2515
2516 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002517 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002518
2519 if (co->co_argcount > 0 ||
Guido van Rossum4f72a782006-10-27 23:31:49 +00002520 co->co_kwonlyargcount > 0 ||
Tim Peters5ca576e2001-06-18 22:08:13 +00002521 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2522 int i;
2523 int n = argcount;
2524 PyObject *kwdict = NULL;
2525 if (co->co_flags & CO_VARKEYWORDS) {
2526 kwdict = PyDict_New();
2527 if (kwdict == NULL)
2528 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002529 i = co->co_argcount + co->co_kwonlyargcount;
Tim Peters5ca576e2001-06-18 22:08:13 +00002530 if (co->co_flags & CO_VARARGS)
2531 i++;
2532 SETLOCAL(i, kwdict);
2533 }
2534 if (argcount > co->co_argcount) {
2535 if (!(co->co_flags & CO_VARARGS)) {
2536 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002537 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002538 "%spositional argument%s (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002539 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002540 defcount ? "at most" : "exactly",
2541 co->co_argcount,
2542 kwcount ? "non-keyword " : "",
2543 co->co_argcount == 1 ? "" : "s",
2544 argcount);
2545 goto fail;
2546 }
2547 n = co->co_argcount;
2548 }
2549 for (i = 0; i < n; i++) {
2550 x = args[i];
2551 Py_INCREF(x);
2552 SETLOCAL(i, x);
2553 }
2554 if (co->co_flags & CO_VARARGS) {
2555 u = PyTuple_New(argcount - n);
2556 if (u == NULL)
2557 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002558 SETLOCAL(co->co_argcount + co->co_kwonlyargcount, u);
Tim Peters5ca576e2001-06-18 22:08:13 +00002559 for (i = n; i < argcount; i++) {
2560 x = args[i];
2561 Py_INCREF(x);
2562 PyTuple_SET_ITEM(u, i-n, x);
2563 }
2564 }
2565 for (i = 0; i < kwcount; i++) {
2566 PyObject *keyword = kws[2*i];
2567 PyObject *value = kws[2*i + 1];
2568 int j;
Neal Norwitzda059e32007-08-26 05:33:45 +00002569 if (keyword == NULL || !PyUnicode_Check(keyword)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002570 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002571 "%U() keywords must be strings",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002572 co->co_name);
Tim Peters5ca576e2001-06-18 22:08:13 +00002573 goto fail;
2574 }
2575 /* XXX slow -- speed up using dictionary? */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002576 for (j = 0;
2577 j < co->co_argcount + co->co_kwonlyargcount;
2578 j++) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002579 PyObject *nm = PyTuple_GET_ITEM(
2580 co->co_varnames, j);
2581 int cmp = PyObject_RichCompareBool(
2582 keyword, nm, Py_EQ);
2583 if (cmp > 0)
2584 break;
2585 else if (cmp < 0)
2586 goto fail;
2587 }
2588 /* Check errors from Compare */
2589 if (PyErr_Occurred())
2590 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002591 if (j >= co->co_argcount + co->co_kwonlyargcount) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002592 if (kwdict == NULL) {
2593 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002594 "%U() got an unexpected "
Walter Dörwald573c08c2007-05-25 15:46:59 +00002595 "keyword argument '%S'",
2596 co->co_name,
2597 keyword);
Tim Peters5ca576e2001-06-18 22:08:13 +00002598 goto fail;
2599 }
2600 PyDict_SetItem(kwdict, keyword, value);
2601 }
2602 else {
2603 if (GETLOCAL(j) != NULL) {
2604 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002605 "%U() got multiple "
Tim Peters5ca576e2001-06-18 22:08:13 +00002606 "values for keyword "
Walter Dörwald573c08c2007-05-25 15:46:59 +00002607 "argument '%S'",
2608 co->co_name,
2609 keyword);
Tim Peters5ca576e2001-06-18 22:08:13 +00002610 goto fail;
2611 }
2612 Py_INCREF(value);
2613 SETLOCAL(j, value);
2614 }
2615 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002616 if (co->co_kwonlyargcount > 0) {
2617 for (i = co->co_argcount;
2618 i < co->co_argcount + co->co_kwonlyargcount;
2619 i++) {
Guido van Rossum29602e42006-11-22 04:45:33 +00002620 PyObject *name, *def;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002621 if (GETLOCAL(i) != NULL)
2622 continue;
Guido van Rossum29602e42006-11-22 04:45:33 +00002623 name = PyTuple_GET_ITEM(co->co_varnames, i);
2624 def = NULL;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002625 if (kwdefs != NULL)
2626 def = PyDict_GetItem(kwdefs, name);
2627 if (def != NULL) {
2628 Py_INCREF(def);
2629 SETLOCAL(i, def);
2630 continue;
2631 }
2632 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002633 "%U() needs keyword-only argument %S",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002634 co->co_name, name);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002635 goto fail;
2636 }
2637 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002638 if (argcount < co->co_argcount) {
2639 int m = co->co_argcount - defcount;
2640 for (i = argcount; i < m; i++) {
2641 if (GETLOCAL(i) == NULL) {
2642 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002643 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002644 "%spositional argument%s "
2645 "(%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002646 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002647 ((co->co_flags & CO_VARARGS) ||
2648 defcount) ? "at least"
2649 : "exactly",
2650 m, kwcount ? "non-keyword " : "",
2651 m == 1 ? "" : "s", i);
2652 goto fail;
2653 }
2654 }
2655 if (n > m)
2656 i = n - m;
2657 else
2658 i = 0;
2659 for (; i < defcount; i++) {
2660 if (GETLOCAL(m+i) == NULL) {
2661 PyObject *def = defs[i];
2662 Py_INCREF(def);
2663 SETLOCAL(m+i, def);
2664 }
2665 }
2666 }
2667 }
2668 else {
2669 if (argcount > 0 || kwcount > 0) {
2670 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002671 "%U() takes no arguments (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002672 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002673 argcount + kwcount);
2674 goto fail;
2675 }
2676 }
2677 /* Allocate and initialize storage for cell vars, and copy free
2678 vars into frame. This isn't too efficient right now. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002679 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002680 int i, j, nargs, found;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002681 Py_UNICODE *cellname, *argname;
Tim Peters5ca576e2001-06-18 22:08:13 +00002682 PyObject *c;
2683
Amaury Forgeot d'Arce670bd42007-11-24 00:29:24 +00002684 nargs = co->co_argcount + co->co_kwonlyargcount;
Tim Peters5ca576e2001-06-18 22:08:13 +00002685 if (co->co_flags & CO_VARARGS)
2686 nargs++;
2687 if (co->co_flags & CO_VARKEYWORDS)
2688 nargs++;
2689
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002690 /* Initialize each cell var, taking into account
2691 cell vars that are initialized from arguments.
2692
2693 Should arrange for the compiler to put cellvars
2694 that are arguments at the beginning of the cellvars
2695 list so that we can march over it more efficiently?
2696 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002697 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002698 cellname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002699 PyTuple_GET_ITEM(co->co_cellvars, i));
2700 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002701 for (j = 0; j < nargs; j++) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002702 argname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002703 PyTuple_GET_ITEM(co->co_varnames, j));
Martin v. Löwis5b222132007-06-10 09:51:05 +00002704 if (Py_UNICODE_strcmp(cellname, argname) == 0) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002705 c = PyCell_New(GETLOCAL(j));
2706 if (c == NULL)
2707 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002708 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002709 found = 1;
2710 break;
2711 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002712 }
2713 if (found == 0) {
2714 c = PyCell_New(NULL);
2715 if (c == NULL)
2716 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002717 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002718 }
2719 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002720 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002721 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002722 int i;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002723 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002724 PyObject *o = PyTuple_GET_ITEM(closure, i);
2725 Py_INCREF(o);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002726 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002727 }
2728 }
2729
Tim Peters5ca576e2001-06-18 22:08:13 +00002730 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002731 /* Don't need to keep the reference to f_back, it will be set
2732 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002733 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002734 f->f_back = NULL;
2735
Jeremy Hylton985eba52003-02-05 23:13:00 +00002736 PCALL(PCALL_GENERATOR);
2737
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002738 /* Create a new generator that owns the ready to run frame
2739 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002740 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002741 }
2742
Thomas Woutersce272b62007-09-19 21:19:28 +00002743 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002744
Thomas Woutersce272b62007-09-19 21:19:28 +00002745fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00002746
Tim Petersb13680b2001-11-27 23:29:29 +00002747 /* decref'ing the frame can cause __del__ methods to get invoked,
2748 which can call back into Python. While we're done with the
2749 current Python frame (f), the associated C stack is still in use,
2750 so recursion_depth must be boosted for the duration.
2751 */
2752 assert(tstate != NULL);
2753 ++tstate->recursion_depth;
Thomas Woutersce272b62007-09-19 21:19:28 +00002754 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002755 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002756 return retval;
2757}
2758
2759
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002760/* Implementation notes for set_exc_info() and reset_exc_info():
2761
2762- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2763 'exc_traceback'. These always travel together.
2764
2765- tstate->curexc_ZZZ is the "hot" exception that is set by
2766 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2767
2768- Once an exception is caught by an except clause, it is transferred
2769 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2770 can pick it up. This is the primary task of set_exc_info().
Thomas Wouters477c8d52006-05-27 19:21:47 +00002771 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002772
2773- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2774
2775 Long ago, when none of this existed, there were just a few globals:
2776 one set corresponding to the "hot" exception, and one set
2777 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2778 globals; they were simply stored as sys.exc_ZZZ. For backwards
2779 compatibility, they still are!) The problem was that in code like
2780 this:
2781
2782 try:
2783 "something that may fail"
2784 except "some exception":
2785 "do something else first"
2786 "print the exception from sys.exc_ZZZ."
2787
2788 if "do something else first" invoked something that raised and caught
2789 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2790 cause of subtle bugs. I fixed this by changing the semantics as
2791 follows:
2792
2793 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2794 *in that frame*.
2795
2796 - But initially, and as long as no exception is caught in a given
2797 frame, sys.exc_ZZZ will hold the last exception caught in the
2798 previous frame (or the frame before that, etc.).
2799
2800 The first bullet fixed the bug in the above example. The second
2801 bullet was for backwards compatibility: it was (and is) common to
2802 have a function that is called when an exception is caught, and to
2803 have that function access the caught exception via sys.exc_ZZZ.
2804 (Example: traceback.print_exc()).
2805
2806 At the same time I fixed the problem that sys.exc_ZZZ weren't
2807 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2808 but that's really a separate improvement.
2809
2810 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2811 variables to what they were before the current frame was called. The
2812 set_exc_info() function saves them on the frame so that
2813 reset_exc_info() can restore them. The invariant is that
2814 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2815 exception (where "catching" an exception applies only to successful
2816 except clauses); and if the current frame ever caught an exception,
2817 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2818 at the start of the current frame.
2819
2820*/
2821
Guido van Rossuma027efa1997-05-05 20:56:21 +00002822static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002823set_exc_info(PyThreadState *tstate,
2824 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002825{
Thomas Wouters477c8d52006-05-27 19:21:47 +00002826 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002827 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002828
Thomas Wouters477c8d52006-05-27 19:21:47 +00002829 assert(type != NULL);
2830 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002831 if (frame->f_exc_type == NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002832 assert(frame->f_exc_value == NULL);
2833 assert(frame->f_exc_traceback == NULL);
2834 /* This frame didn't catch an exception before. */
2835 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002836 if (tstate->exc_type == NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002837 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002838 Py_INCREF(Py_None);
2839 tstate->exc_type = Py_None;
2840 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002841 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002842 Py_XINCREF(tstate->exc_value);
2843 Py_XINCREF(tstate->exc_traceback);
2844 frame->f_exc_type = tstate->exc_type;
2845 frame->f_exc_value = tstate->exc_value;
2846 frame->f_exc_traceback = tstate->exc_traceback;
2847 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002848 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002849 tmp_type = tstate->exc_type;
2850 tmp_value = tstate->exc_value;
2851 tmp_tb = tstate->exc_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002852 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002853 Py_XINCREF(value);
2854 Py_XINCREF(tb);
2855 tstate->exc_type = type;
2856 tstate->exc_value = value;
2857 tstate->exc_traceback = tb;
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002858 PyException_SetTraceback(value, tb);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002859 Py_XDECREF(tmp_type);
2860 Py_XDECREF(tmp_value);
2861 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002862}
2863
2864static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002865reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002866{
2867 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002868 PyObject *tmp_type, *tmp_value, *tmp_tb;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002869
2870 /* It's a precondition that the thread state's frame caught an
2871 * exception -- verify in a debug build.
2872 */
2873 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002874 frame = tstate->frame;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002875 assert(frame != NULL);
2876 assert(frame->f_exc_type != NULL);
2877
2878 /* Copy the frame's exception info back to the thread state. */
2879 tmp_type = tstate->exc_type;
2880 tmp_value = tstate->exc_value;
2881 tmp_tb = tstate->exc_traceback;
2882 Py_INCREF(frame->f_exc_type);
2883 Py_XINCREF(frame->f_exc_value);
2884 Py_XINCREF(frame->f_exc_traceback);
2885 tstate->exc_type = frame->f_exc_type;
2886 tstate->exc_value = frame->f_exc_value;
2887 tstate->exc_traceback = frame->f_exc_traceback;
2888 Py_XDECREF(tmp_type);
2889 Py_XDECREF(tmp_value);
2890 Py_XDECREF(tmp_tb);
2891
Thomas Wouters477c8d52006-05-27 19:21:47 +00002892 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002893 tmp_type = frame->f_exc_type;
2894 tmp_value = frame->f_exc_value;
2895 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00002896 frame->f_exc_type = NULL;
2897 frame->f_exc_value = NULL;
2898 frame->f_exc_traceback = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002899 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002900 Py_XDECREF(tmp_value);
2901 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002902}
2903
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002904/* Logic for the raise statement (too complicated for inlining).
2905 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00002906static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00002907do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002908{
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002909 PyObject *type = NULL, *value = NULL, *tb = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00002910
2911 if (exc == NULL) {
Guido van Rossumd295f121998-04-09 21:39:57 +00002912 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00002913 PyThreadState *tstate = PyThreadState_GET();
Collin Winter828f04a2007-08-31 00:04:24 +00002914 type = tstate->exc_type;
Guido van Rossumd295f121998-04-09 21:39:57 +00002915 value = tstate->exc_value;
2916 tb = tstate->exc_traceback;
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002917 if (type == Py_None) {
2918 PyErr_SetString(PyExc_RuntimeError,
2919 "No active exception to reraise");
2920 return WHY_EXCEPTION;
2921 }
2922 Py_XINCREF(type);
Guido van Rossumd295f121998-04-09 21:39:57 +00002923 Py_XINCREF(value);
2924 Py_XINCREF(tb);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002925 PyErr_Restore(type, value, tb);
2926 return WHY_RERAISE;
Guido van Rossumd295f121998-04-09 21:39:57 +00002927 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002928
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002929 /* We support the following forms of raise:
Collin Winter828f04a2007-08-31 00:04:24 +00002930 raise
2931 raise <instance>
2932 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002933
Collin Winter828f04a2007-08-31 00:04:24 +00002934 if (PyExceptionClass_Check(exc)) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002935 type = exc;
2936 value = PyObject_CallObject(exc, NULL);
Collin Winter828f04a2007-08-31 00:04:24 +00002937 if (value == NULL)
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002938 goto raise_error;
2939 }
Collin Winter828f04a2007-08-31 00:04:24 +00002940 else if (PyExceptionInstance_Check(exc)) {
2941 value = exc;
2942 type = PyExceptionInstance_Class(exc);
Guido van Rossumb209a111997-04-29 18:18:01 +00002943 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002944 }
2945 else {
2946 /* Not something you can raise. You get an exception
2947 anyway, just not what you specified :-) */
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002948 Py_DECREF(exc);
Guido van Rossum45aecf42006-03-15 04:58:47 +00002949 PyErr_SetString(PyExc_TypeError,
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002950 "exceptions must derive from BaseException");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002951 goto raise_error;
2952 }
Collin Winter828f04a2007-08-31 00:04:24 +00002953
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002954 tb = PyException_GetTraceback(value);
2955 if (cause) {
2956 PyObject *fixed_cause;
2957 if (PyExceptionClass_Check(cause)) {
2958 fixed_cause = PyObject_CallObject(cause, NULL);
2959 if (fixed_cause == NULL)
2960 goto raise_error;
2961 Py_DECREF(cause);
2962 }
2963 else if (PyExceptionInstance_Check(cause)) {
2964 fixed_cause = cause;
2965 }
2966 else {
2967 PyErr_SetString(PyExc_TypeError,
2968 "exception causes must derive from "
2969 "BaseException");
2970 goto raise_error;
2971 }
2972 PyException_SetCause(value, fixed_cause);
2973 }
Collin Winter828f04a2007-08-31 00:04:24 +00002974
Guido van Rossumb209a111997-04-29 18:18:01 +00002975 PyErr_Restore(type, value, tb);
Collin Winter828f04a2007-08-31 00:04:24 +00002976 return WHY_EXCEPTION;
2977
2978raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00002979 Py_XDECREF(value);
2980 Py_XDECREF(type);
2981 Py_XDECREF(tb);
Collin Winter1966f1c2007-09-01 20:26:44 +00002982 Py_XDECREF(cause);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002983 return WHY_EXCEPTION;
2984}
2985
Tim Petersd6d010b2001-06-21 02:49:55 +00002986/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00002987 sp). Return 1 for success, 0 if error.
2988
2989 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
2990 with a variable target.
2991*/
Tim Petersd6d010b2001-06-21 02:49:55 +00002992
Barry Warsawe42b18f1997-08-25 22:13:04 +00002993static int
Guido van Rossum0368b722007-05-11 16:50:42 +00002994unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00002995{
Guido van Rossum0368b722007-05-11 16:50:42 +00002996 int i = 0, j = 0;
2997 Py_ssize_t ll = 0;
Tim Petersd6d010b2001-06-21 02:49:55 +00002998 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00002999 PyObject *w;
Guido van Rossum0368b722007-05-11 16:50:42 +00003000 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003001
Tim Petersd6d010b2001-06-21 02:49:55 +00003002 assert(v != NULL);
3003
3004 it = PyObject_GetIter(v);
3005 if (it == NULL)
3006 goto Error;
3007
3008 for (; i < argcnt; i++) {
3009 w = PyIter_Next(it);
3010 if (w == NULL) {
3011 /* Iterator done, via error or exhaustion. */
3012 if (!PyErr_Occurred()) {
3013 PyErr_Format(PyExc_ValueError,
3014 "need more than %d value%s to unpack",
3015 i, i == 1 ? "" : "s");
3016 }
3017 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003018 }
3019 *--sp = w;
3020 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003021
Guido van Rossum0368b722007-05-11 16:50:42 +00003022 if (argcntafter == -1) {
3023 /* We better have exhausted the iterator now. */
3024 w = PyIter_Next(it);
3025 if (w == NULL) {
3026 if (PyErr_Occurred())
3027 goto Error;
3028 Py_DECREF(it);
3029 return 1;
3030 }
3031 Py_DECREF(w);
3032 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
3033 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003034 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003035
3036 l = PySequence_List(it);
3037 if (l == NULL)
3038 goto Error;
3039 *--sp = l;
3040 i++;
3041
3042 ll = PyList_GET_SIZE(l);
3043 if (ll < argcntafter) {
3044 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3045 argcnt + ll);
3046 goto Error;
3047 }
3048
3049 /* Pop the "after-variable" args off the list. */
3050 for (j = argcntafter; j > 0; j--, i++) {
3051 *--sp = PyList_GET_ITEM(l, ll - j);
3052 }
3053 /* Resize the list. */
Christian Heimes90aa7642007-12-19 02:45:37 +00003054 Py_SIZE(l) = ll - argcntafter;
Guido van Rossum0368b722007-05-11 16:50:42 +00003055 Py_DECREF(it);
3056 return 1;
3057
Tim Petersd6d010b2001-06-21 02:49:55 +00003058Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003059 for (; i > 0; i--, sp++)
3060 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003061 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003062 return 0;
3063}
3064
3065
Guido van Rossum96a42c81992-01-12 02:29:51 +00003066#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003067static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003068prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003069{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003070 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003071 if (PyObject_Print(v, stdout, 0) != 0)
3072 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003073 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003074 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003075}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003076#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003077
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003078static void
Fred Drake5755ce62001-06-27 19:19:46 +00003079call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003080{
Guido van Rossumb209a111997-04-29 18:18:01 +00003081 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003082 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003083 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003084 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003085 value = Py_None;
3086 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003087 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003088 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003089 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003090 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003091 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003092 }
Fred Drake5755ce62001-06-27 19:19:46 +00003093 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003094 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003095 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003096 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003097 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003098 Py_XDECREF(type);
3099 Py_XDECREF(value);
3100 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003101 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003102}
3103
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003104static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003105call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003106 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003107{
3108 PyObject *type, *value, *traceback;
3109 int err;
3110 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003111 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003112 if (err == 0)
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003113 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003114 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003115 return 0;
3116 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003117 else {
3118 Py_XDECREF(type);
3119 Py_XDECREF(value);
3120 Py_XDECREF(traceback);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003121 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003122 }
3123}
3124
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003125static int
Fred Drake5755ce62001-06-27 19:19:46 +00003126call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3127 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003128{
Fred Drake5755ce62001-06-27 19:19:46 +00003129 register PyThreadState *tstate = frame->f_tstate;
3130 int result;
3131 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003132 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003133 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003134 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003135 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003136 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3137 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003138 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003139 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003140}
3141
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003142PyObject *
3143_PyEval_CallTracing(PyObject *func, PyObject *args)
3144{
3145 PyFrameObject *frame = PyEval_GetFrame();
3146 PyThreadState *tstate = frame->f_tstate;
3147 int save_tracing = tstate->tracing;
3148 int save_use_tracing = tstate->use_tracing;
3149 PyObject *result;
3150
3151 tstate->tracing = 0;
3152 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3153 || (tstate->c_profilefunc != NULL));
3154 result = PyObject_Call(func, args, NULL);
3155 tstate->tracing = save_tracing;
3156 tstate->use_tracing = save_use_tracing;
3157 return result;
3158}
3159
Michael W. Hudson006c7522002-11-08 13:08:46 +00003160static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003161maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003162 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3163 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003164{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003165 int result = 0;
3166
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003167 /* If the last instruction executed isn't in the current
3168 instruction window, reset the window. If the last
3169 instruction happens to fall at the start of a line or if it
3170 represents a jump backwards, call the trace function.
3171 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003172 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003173 int line;
3174 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003175
Thomas Woutersce272b62007-09-19 21:19:28 +00003176 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3177 &bounds);
3178 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003179 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003180 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003181 PyTrace_LINE, Py_None);
Thomas Woutersce272b62007-09-19 21:19:28 +00003182 }
3183 *instr_lb = bounds.ap_lower;
3184 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003185 }
Armin Rigobf57a142004-03-22 19:24:58 +00003186 else if (frame->f_lasti <= *instr_prev) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003187 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003188 }
3189 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003190 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003191}
3192
Fred Drake5755ce62001-06-27 19:19:46 +00003193void
3194PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003195{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003196 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003197 PyObject *temp = tstate->c_profileobj;
3198 Py_XINCREF(arg);
3199 tstate->c_profilefunc = NULL;
3200 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003201 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003202 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003203 Py_XDECREF(temp);
3204 tstate->c_profilefunc = func;
3205 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003206 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003207 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003208}
3209
3210void
3211PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3212{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003213 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003214 PyObject *temp = tstate->c_traceobj;
3215 Py_XINCREF(arg);
3216 tstate->c_tracefunc = NULL;
3217 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003218 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003219 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003220 Py_XDECREF(temp);
3221 tstate->c_tracefunc = func;
3222 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003223 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003224 tstate->use_tracing = ((func != NULL)
3225 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003226}
3227
Guido van Rossumb209a111997-04-29 18:18:01 +00003228PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003229PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003230{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003231 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003232 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003233 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003234 else
3235 return current_frame->f_builtins;
3236}
3237
Guido van Rossumb209a111997-04-29 18:18:01 +00003238PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003239PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003240{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003241 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003242 if (current_frame == NULL)
3243 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003244 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003245 return current_frame->f_locals;
3246}
3247
Guido van Rossumb209a111997-04-29 18:18:01 +00003248PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003249PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003250{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003251 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003252 if (current_frame == NULL)
3253 return NULL;
3254 else
3255 return current_frame->f_globals;
3256}
3257
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003258PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003259PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003260{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003261 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003262 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003263}
3264
Guido van Rossum6135a871995-01-09 17:53:26 +00003265int
Tim Peters5ba58662001-07-16 02:29:45 +00003266PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003267{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003268 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003269 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003270
3271 if (current_frame != NULL) {
3272 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003273 const int compilerflags = codeflags & PyCF_MASK;
3274 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003275 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003276 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003277 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003278#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003279 if (codeflags & CO_GENERATOR_ALLOWED) {
3280 result = 1;
3281 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3282 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003283#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003284 }
3285 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003286}
3287
Guido van Rossum3f5da241990-12-20 15:06:42 +00003288
Guido van Rossum681d79a1995-07-18 14:51:37 +00003289/* External interface to call any callable object.
3290 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003291
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003292#undef PyEval_CallObject
3293/* for backward compatibility: export this interface */
3294
Guido van Rossumb209a111997-04-29 18:18:01 +00003295PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003296PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003297{
Guido van Rossumb209a111997-04-29 18:18:01 +00003298 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003299}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003300#define PyEval_CallObject(func,arg) \
3301 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003302
Guido van Rossumb209a111997-04-29 18:18:01 +00003303PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003304PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003305{
Jeremy Hylton52820442001-01-03 23:52:36 +00003306 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003307
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003308 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003309 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003310 if (arg == NULL)
3311 return NULL;
3312 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003313 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003314 PyErr_SetString(PyExc_TypeError,
3315 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003316 return NULL;
3317 }
3318 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003319 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003320
Guido van Rossumb209a111997-04-29 18:18:01 +00003321 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003322 PyErr_SetString(PyExc_TypeError,
3323 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003324 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003325 return NULL;
3326 }
3327
Tim Peters6d6c1a32001-08-02 04:15:00 +00003328 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003329 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003330 return result;
3331}
3332
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003333const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003334PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003335{
3336 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003337 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003338 else if (PyFunction_Check(func))
Martin v. Löwis5b222132007-06-10 09:51:05 +00003339 return PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003340 else if (PyCFunction_Check(func))
3341 return ((PyCFunctionObject*)func)->m_ml->ml_name;
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003342 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003343 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003344}
3345
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003346const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003347PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003348{
3349 if (PyMethod_Check(func))
3350 return "()";
3351 else if (PyFunction_Check(func))
3352 return "()";
3353 else if (PyCFunction_Check(func))
3354 return "()";
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003355 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003356 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00003357}
3358
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00003359static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003360err_args(PyObject *func, int flags, int nargs)
3361{
3362 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003363 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003364 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003365 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003366 nargs);
3367 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003368 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003369 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003370 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003371 nargs);
3372}
3373
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003374#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003375if (tstate->use_tracing && tstate->c_profilefunc) { \
3376 if (call_trace(tstate->c_profilefunc, \
3377 tstate->c_profileobj, \
3378 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003379 func)) { \
3380 x = NULL; \
3381 } \
3382 else { \
3383 x = call; \
3384 if (tstate->c_profilefunc != NULL) { \
3385 if (x == NULL) { \
3386 call_trace_protected(tstate->c_profilefunc, \
3387 tstate->c_profileobj, \
3388 tstate->frame, PyTrace_C_EXCEPTION, \
3389 func); \
3390 /* XXX should pass (type, value, tb) */ \
3391 } else { \
3392 if (call_trace(tstate->c_profilefunc, \
3393 tstate->c_profileobj, \
3394 tstate->frame, PyTrace_C_RETURN, \
3395 func)) { \
3396 Py_DECREF(x); \
3397 x = NULL; \
3398 } \
3399 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003400 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003401 } \
3402} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003403 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003404 }
3405
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003406static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003407call_function(PyObject ***pp_stack, int oparg
3408#ifdef WITH_TSC
3409 , uint64* pintr0, uint64* pintr1
3410#endif
3411 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003412{
3413 int na = oparg & 0xff;
3414 int nk = (oparg>>8) & 0xff;
3415 int n = na + 2 * nk;
3416 PyObject **pfunc = (*pp_stack) - n - 1;
3417 PyObject *func = *pfunc;
3418 PyObject *x, *w;
3419
Jeremy Hylton985eba52003-02-05 23:13:00 +00003420 /* Always dispatch PyCFunction first, because these are
3421 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003422 */
3423 if (PyCFunction_Check(func) && nk == 0) {
3424 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003425 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003426
3427 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003428 if (flags & (METH_NOARGS | METH_O)) {
3429 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3430 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003431 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003432 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003433 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003434 else if (flags & METH_O && na == 1) {
3435 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003436 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003437 Py_DECREF(arg);
3438 }
3439 else {
3440 err_args(func, flags, na);
3441 x = NULL;
3442 }
3443 }
3444 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003445 PyObject *callargs;
3446 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003447 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003448 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003449 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003450 Py_XDECREF(callargs);
3451 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003452 } else {
3453 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3454 /* optimize access to bound methods */
3455 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003456 PCALL(PCALL_METHOD);
3457 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003458 Py_INCREF(self);
3459 func = PyMethod_GET_FUNCTION(func);
3460 Py_INCREF(func);
3461 Py_DECREF(*pfunc);
3462 *pfunc = self;
3463 na++;
3464 n++;
3465 } else
3466 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003467 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003468 if (PyFunction_Check(func))
3469 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003470 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003471 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003472 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003473 Py_DECREF(func);
3474 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003475
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003476 /* Clear the stack of the function object. Also removes
3477 the arguments in case they weren't consumed already
3478 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003479 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003480 while ((*pp_stack) > pfunc) {
3481 w = EXT_POP(*pp_stack);
3482 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003483 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003484 }
3485 return x;
3486}
3487
Jeremy Hylton192690e2002-08-16 18:36:11 +00003488/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003489 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003490 For the simplest case -- a function that takes only positional
3491 arguments and is called with only positional arguments -- it
3492 inlines the most primitive frame setup code from
3493 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3494 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003495*/
3496
3497static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003498fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003499{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003500 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003501 PyObject *globals = PyFunction_GET_GLOBALS(func);
3502 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003503 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003504 PyObject **d = NULL;
3505 int nd = 0;
3506
Jeremy Hylton985eba52003-02-05 23:13:00 +00003507 PCALL(PCALL_FUNCTION);
3508 PCALL(PCALL_FAST_FUNCTION);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003509 if (argdefs == NULL && co->co_argcount == n &&
3510 co->co_kwonlyargcount == 0 && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003511 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3512 PyFrameObject *f;
3513 PyObject *retval = NULL;
3514 PyThreadState *tstate = PyThreadState_GET();
3515 PyObject **fastlocals, **stack;
3516 int i;
3517
3518 PCALL(PCALL_FASTER_FUNCTION);
3519 assert(globals != NULL);
3520 /* XXX Perhaps we should create a specialized
3521 PyFrame_New() that doesn't take locals, but does
3522 take builtins without sanity checking them.
3523 */
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003524 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003525 f = PyFrame_New(tstate, co, globals, NULL);
3526 if (f == NULL)
3527 return NULL;
3528
3529 fastlocals = f->f_localsplus;
3530 stack = (*pp_stack) - n;
3531
3532 for (i = 0; i < n; i++) {
3533 Py_INCREF(*stack);
3534 fastlocals[i] = *stack++;
3535 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003536 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003537 ++tstate->recursion_depth;
3538 Py_DECREF(f);
3539 --tstate->recursion_depth;
3540 return retval;
3541 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003542 if (argdefs != NULL) {
3543 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimes90aa7642007-12-19 02:45:37 +00003544 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003545 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003546 return PyEval_EvalCodeEx(co, globals,
3547 (PyObject *)NULL, (*pp_stack)-n, na,
Guido van Rossum4f72a782006-10-27 23:31:49 +00003548 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
Jeremy Hylton985eba52003-02-05 23:13:00 +00003549 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003550}
3551
3552static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003553update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3554 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003555{
3556 PyObject *kwdict = NULL;
3557 if (orig_kwdict == NULL)
3558 kwdict = PyDict_New();
3559 else {
3560 kwdict = PyDict_Copy(orig_kwdict);
3561 Py_DECREF(orig_kwdict);
3562 }
3563 if (kwdict == NULL)
3564 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003565 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003566 int err;
3567 PyObject *value = EXT_POP(*pp_stack);
3568 PyObject *key = EXT_POP(*pp_stack);
3569 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003570 PyErr_Format(PyExc_TypeError,
3571 "%.200s%s got multiple values "
3572 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003573 PyEval_GetFuncName(func),
3574 PyEval_GetFuncDesc(func),
Neal Norwitzda059e32007-08-26 05:33:45 +00003575 PyUnicode_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003576 Py_DECREF(key);
3577 Py_DECREF(value);
3578 Py_DECREF(kwdict);
3579 return NULL;
3580 }
3581 err = PyDict_SetItem(kwdict, key, value);
3582 Py_DECREF(key);
3583 Py_DECREF(value);
3584 if (err) {
3585 Py_DECREF(kwdict);
3586 return NULL;
3587 }
3588 }
3589 return kwdict;
3590}
3591
3592static PyObject *
3593update_star_args(int nstack, int nstar, PyObject *stararg,
3594 PyObject ***pp_stack)
3595{
3596 PyObject *callargs, *w;
3597
3598 callargs = PyTuple_New(nstack + nstar);
3599 if (callargs == NULL) {
3600 return NULL;
3601 }
3602 if (nstar) {
3603 int i;
3604 for (i = 0; i < nstar; i++) {
3605 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3606 Py_INCREF(a);
3607 PyTuple_SET_ITEM(callargs, nstack + i, a);
3608 }
3609 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003610 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003611 w = EXT_POP(*pp_stack);
3612 PyTuple_SET_ITEM(callargs, nstack, w);
3613 }
3614 return callargs;
3615}
3616
3617static PyObject *
3618load_args(PyObject ***pp_stack, int na)
3619{
3620 PyObject *args = PyTuple_New(na);
3621 PyObject *w;
3622
3623 if (args == NULL)
3624 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003625 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003626 w = EXT_POP(*pp_stack);
3627 PyTuple_SET_ITEM(args, na, w);
3628 }
3629 return args;
3630}
3631
3632static PyObject *
3633do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3634{
3635 PyObject *callargs = NULL;
3636 PyObject *kwdict = NULL;
3637 PyObject *result = NULL;
3638
3639 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003640 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003641 if (kwdict == NULL)
3642 goto call_fail;
3643 }
3644 callargs = load_args(pp_stack, na);
3645 if (callargs == NULL)
3646 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003647#ifdef CALL_PROFILE
3648 /* At this point, we have to look at the type of func to
3649 update the call stats properly. Do it here so as to avoid
3650 exposing the call stats machinery outside ceval.c
3651 */
3652 if (PyFunction_Check(func))
3653 PCALL(PCALL_FUNCTION);
3654 else if (PyMethod_Check(func))
3655 PCALL(PCALL_METHOD);
3656 else if (PyType_Check(func))
3657 PCALL(PCALL_TYPE);
3658 else
3659 PCALL(PCALL_OTHER);
3660#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003661 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00003662call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003663 Py_XDECREF(callargs);
3664 Py_XDECREF(kwdict);
3665 return result;
3666}
3667
3668static PyObject *
3669ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3670{
3671 int nstar = 0;
3672 PyObject *callargs = NULL;
3673 PyObject *stararg = NULL;
3674 PyObject *kwdict = NULL;
3675 PyObject *result = NULL;
3676
3677 if (flags & CALL_FLAG_KW) {
3678 kwdict = EXT_POP(*pp_stack);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003679 if (!PyDict_Check(kwdict)) {
3680 PyObject *d;
3681 d = PyDict_New();
3682 if (d == NULL)
3683 goto ext_call_fail;
3684 if (PyDict_Update(d, kwdict) != 0) {
3685 Py_DECREF(d);
3686 /* PyDict_Update raises attribute
3687 * error (percolated from an attempt
3688 * to get 'keys' attribute) instead of
3689 * a type error if its second argument
3690 * is not a mapping.
3691 */
3692 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
3693 PyErr_Format(PyExc_TypeError,
3694 "%.200s%.200s argument after ** "
3695 "must be a mapping, not %.200s",
3696 PyEval_GetFuncName(func),
3697 PyEval_GetFuncDesc(func),
3698 kwdict->ob_type->tp_name);
3699 }
3700 goto ext_call_fail;
3701 }
3702 Py_DECREF(kwdict);
3703 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00003704 }
3705 }
3706 if (flags & CALL_FLAG_VAR) {
3707 stararg = EXT_POP(*pp_stack);
3708 if (!PyTuple_Check(stararg)) {
3709 PyObject *t = NULL;
3710 t = PySequence_Tuple(stararg);
3711 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003712 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3713 PyErr_Format(PyExc_TypeError,
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003714 "%.200s%.200s argument after * "
3715 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003716 PyEval_GetFuncName(func),
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003717 PyEval_GetFuncDesc(func),
3718 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003719 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003720 goto ext_call_fail;
3721 }
3722 Py_DECREF(stararg);
3723 stararg = t;
3724 }
3725 nstar = PyTuple_GET_SIZE(stararg);
3726 }
3727 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003728 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003729 if (kwdict == NULL)
3730 goto ext_call_fail;
3731 }
3732 callargs = update_star_args(na, nstar, stararg, pp_stack);
3733 if (callargs == NULL)
3734 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003735#ifdef CALL_PROFILE
3736 /* At this point, we have to look at the type of func to
3737 update the call stats properly. Do it here so as to avoid
3738 exposing the call stats machinery outside ceval.c
3739 */
3740 if (PyFunction_Check(func))
3741 PCALL(PCALL_FUNCTION);
3742 else if (PyMethod_Check(func))
3743 PCALL(PCALL_METHOD);
3744 else if (PyType_Check(func))
3745 PCALL(PCALL_TYPE);
3746 else
3747 PCALL(PCALL_OTHER);
3748#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003749 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00003750ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003751 Py_XDECREF(callargs);
3752 Py_XDECREF(kwdict);
3753 Py_XDECREF(stararg);
3754 return result;
3755}
3756
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003757/* Extract a slice index from a PyInt or PyLong or an object with the
3758 nb_index slot defined, and store in *pi.
3759 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3760 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 +00003761 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003762*/
Tim Petersb5196382001-12-16 19:44:20 +00003763/* Note: If v is NULL, return success without storing into *pi. This
3764 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3765 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003766*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003767int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003768_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003769{
Tim Petersb5196382001-12-16 19:44:20 +00003770 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003771 Py_ssize_t x;
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +00003772 if (PyIndex_Check(v)) {
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003773 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003774 if (x == -1 && PyErr_Occurred())
3775 return 0;
3776 }
3777 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003778 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003779 "slice indices must be integers or "
3780 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003781 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003782 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003783 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003784 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003785 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003786}
3787
Guido van Rossum486364b2007-06-30 05:01:58 +00003788#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Brett Cannon39590462007-02-26 22:01:14 +00003789 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00003790
Guido van Rossumb209a111997-04-29 18:18:01 +00003791static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003792cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003793{
Guido van Rossumac7be682001-01-17 15:42:30 +00003794 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003795 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003796 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003797 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003798 break;
3799 case PyCmp_IS_NOT:
3800 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003801 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003802 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003803 res = PySequence_Contains(w, v);
3804 if (res < 0)
3805 return NULL;
3806 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003807 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003808 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003809 if (res < 0)
3810 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003811 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003812 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003813 case PyCmp_EXC_MATCH:
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003814 if (PyTuple_Check(w)) {
3815 Py_ssize_t i, length;
3816 length = PyTuple_Size(w);
3817 for (i = 0; i < length; i += 1) {
3818 PyObject *exc = PyTuple_GET_ITEM(w, i);
Brett Cannon39590462007-02-26 22:01:14 +00003819 if (!PyExceptionClass_Check(exc)) {
3820 PyErr_SetString(PyExc_TypeError,
3821 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003822 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003823 }
3824 }
3825 }
3826 else {
Brett Cannon39590462007-02-26 22:01:14 +00003827 if (!PyExceptionClass_Check(w)) {
3828 PyErr_SetString(PyExc_TypeError,
3829 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003830 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003831 }
3832 }
Barry Warsaw4249f541997-08-22 21:26:19 +00003833 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003834 break;
3835 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003836 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003837 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003838 v = res ? Py_True : Py_False;
3839 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003840 return v;
3841}
3842
Thomas Wouters52152252000-08-17 22:55:00 +00003843static PyObject *
3844import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00003845{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003846 PyObject *x;
3847
3848 x = PyObject_GetAttr(v, name);
3849 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Walter Dörwald573c08c2007-05-25 15:46:59 +00003850 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003851 }
Thomas Wouters52152252000-08-17 22:55:00 +00003852 return x;
3853}
Guido van Rossumac7be682001-01-17 15:42:30 +00003854
Thomas Wouters52152252000-08-17 22:55:00 +00003855static int
3856import_all_from(PyObject *locals, PyObject *v)
3857{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003858 PyObject *all = PyObject_GetAttrString(v, "__all__");
3859 PyObject *dict, *name, *value;
3860 int skip_leading_underscores = 0;
3861 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00003862
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003863 if (all == NULL) {
3864 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3865 return -1; /* Unexpected error */
3866 PyErr_Clear();
3867 dict = PyObject_GetAttrString(v, "__dict__");
3868 if (dict == NULL) {
3869 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3870 return -1;
3871 PyErr_SetString(PyExc_ImportError,
3872 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00003873 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003874 }
3875 all = PyMapping_Keys(dict);
3876 Py_DECREF(dict);
3877 if (all == NULL)
3878 return -1;
3879 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003880 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003881
3882 for (pos = 0, err = 0; ; pos++) {
3883 name = PySequence_GetItem(all, pos);
3884 if (name == NULL) {
3885 if (!PyErr_ExceptionMatches(PyExc_IndexError))
3886 err = -1;
3887 else
3888 PyErr_Clear();
3889 break;
3890 }
3891 if (skip_leading_underscores &&
Martin v. Löwis5b222132007-06-10 09:51:05 +00003892 PyUnicode_Check(name) &&
3893 PyUnicode_AS_UNICODE(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003894 {
3895 Py_DECREF(name);
3896 continue;
3897 }
3898 value = PyObject_GetAttr(v, name);
3899 if (value == NULL)
3900 err = -1;
Thomas Wouters89f507f2006-12-13 04:49:30 +00003901 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003902 err = PyDict_SetItem(locals, name, value);
Thomas Wouters89f507f2006-12-13 04:49:30 +00003903 else
3904 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003905 Py_DECREF(name);
3906 Py_XDECREF(value);
3907 if (err != 0)
3908 break;
3909 }
3910 Py_DECREF(all);
3911 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003912}
3913
Guido van Rossumac7be682001-01-17 15:42:30 +00003914static void
Neal Norwitzda059e32007-08-26 05:33:45 +00003915format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00003916{
Neal Norwitzda059e32007-08-26 05:33:45 +00003917 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00003918
3919 if (!obj)
3920 return;
3921
Neal Norwitzda059e32007-08-26 05:33:45 +00003922 obj_str = PyUnicode_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00003923 if (!obj_str)
3924 return;
3925
3926 PyErr_Format(exc, format_str, obj_str);
3927}
Guido van Rossum950361c1997-01-24 13:49:28 +00003928
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003929static PyObject *
Guido van Rossum98297ee2007-11-06 21:34:58 +00003930unicode_concatenate(PyObject *v, PyObject *w,
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003931 PyFrameObject *f, unsigned char *next_instr)
3932{
3933 /* This function implements 'variable += expr' when both arguments
Guido van Rossum98297ee2007-11-06 21:34:58 +00003934 are (Unicode) strings. */
3935 Py_ssize_t v_len = PyUnicode_GET_SIZE(v);
3936 Py_ssize_t w_len = PyUnicode_GET_SIZE(w);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003937 Py_ssize_t new_len = v_len + w_len;
3938 if (new_len < 0) {
3939 PyErr_SetString(PyExc_OverflowError,
3940 "strings are too large to concat");
3941 return NULL;
3942 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003943
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003944 if (v->ob_refcnt == 2) {
3945 /* In the common case, there are 2 references to the value
3946 * stored in 'variable' when the += is performed: one on the
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003947 * value stack (in 'v') and one still stored in the
3948 * 'variable'. We try to delete the variable now to reduce
3949 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003950 */
3951 switch (*next_instr) {
3952 case STORE_FAST:
3953 {
3954 int oparg = PEEKARG();
3955 PyObject **fastlocals = f->f_localsplus;
3956 if (GETLOCAL(oparg) == v)
3957 SETLOCAL(oparg, NULL);
3958 break;
3959 }
3960 case STORE_DEREF:
3961 {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003962 PyObject **freevars = (f->f_localsplus +
3963 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003964 PyObject *c = freevars[PEEKARG()];
3965 if (PyCell_GET(c) == v)
3966 PyCell_Set(c, NULL);
3967 break;
3968 }
3969 case STORE_NAME:
3970 {
3971 PyObject *names = f->f_code->co_names;
3972 PyObject *name = GETITEM(names, PEEKARG());
3973 PyObject *locals = f->f_locals;
3974 if (PyDict_CheckExact(locals) &&
3975 PyDict_GetItem(locals, name) == v) {
3976 if (PyDict_DelItem(locals, name) != 0) {
3977 PyErr_Clear();
3978 }
3979 }
3980 break;
3981 }
3982 }
3983 }
3984
Guido van Rossum98297ee2007-11-06 21:34:58 +00003985 if (v->ob_refcnt == 1 && !PyUnicode_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003986 /* Now we own the last reference to 'v', so we can resize it
3987 * in-place.
3988 */
Guido van Rossum98297ee2007-11-06 21:34:58 +00003989 if (PyUnicode_Resize(&v, new_len) != 0) {
3990 /* XXX if PyUnicode_Resize() fails, 'v' has been
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003991 * deallocated so it cannot be put back into
3992 * 'variable'. The MemoryError is raised when there
3993 * is no value in 'variable', which might (very
3994 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003995 */
3996 return NULL;
3997 }
3998 /* copy 'w' into the newly allocated area of 'v' */
Guido van Rossum98297ee2007-11-06 21:34:58 +00003999 memcpy(PyUnicode_AS_UNICODE(v) + v_len,
4000 PyUnicode_AS_UNICODE(w), w_len*sizeof(Py_UNICODE));
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004001 return v;
4002 }
4003 else {
4004 /* When in-place resizing is not an option. */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004005 w = PyUnicode_Concat(v, w);
4006 Py_DECREF(v);
4007 return w;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004008 }
4009}
4010
Guido van Rossum950361c1997-01-24 13:49:28 +00004011#ifdef DYNAMIC_EXECUTION_PROFILE
4012
Skip Montanarof118cb12001-10-15 20:51:38 +00004013static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004014getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004015{
4016 int i;
4017 PyObject *l = PyList_New(256);
4018 if (l == NULL) return NULL;
4019 for (i = 0; i < 256; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00004020 PyObject *x = PyLong_FromLong(a[i]);
Guido van Rossum950361c1997-01-24 13:49:28 +00004021 if (x == NULL) {
4022 Py_DECREF(l);
4023 return NULL;
4024 }
4025 PyList_SetItem(l, i, x);
4026 }
4027 for (i = 0; i < 256; i++)
4028 a[i] = 0;
4029 return l;
4030}
4031
4032PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004033_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004034{
4035#ifndef DXPAIRS
4036 return getarray(dxp);
4037#else
4038 int i;
4039 PyObject *l = PyList_New(257);
4040 if (l == NULL) return NULL;
4041 for (i = 0; i < 257; i++) {
4042 PyObject *x = getarray(dxpairs[i]);
4043 if (x == NULL) {
4044 Py_DECREF(l);
4045 return NULL;
4046 }
4047 PyList_SetItem(l, i, x);
4048 }
4049 return l;
4050#endif
4051}
4052
4053#endif