blob: e184ef6a4b899966eeb94478534b2aa089069765 [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 Rossum10dc2e81990-11-18 17:27:39 +000016#include "opcode.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000017#include "structmember.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000018
Guido van Rossumc6004111993-11-05 10:22:19 +000019#include <ctype.h>
20
Thomas Wouters477c8d52006-05-27 19:21:47 +000021#ifndef WITH_TSC
Michael W. Hudson75eabd22005-01-18 15:56:11 +000022
23#define READ_TIMESTAMP(var)
24
25#else
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000026
27typedef unsigned long long uint64;
28
Ezio Melotti13925002011-03-16 11:05:33 +020029/* PowerPC support.
David Malcolmf1397ad2011-01-06 17:01:36 +000030 "__ppc__" appears to be the preprocessor definition to detect on OS X, whereas
31 "__powerpc__" appears to be the correct one for Linux with GCC
32*/
33#if defined(__ppc__) || defined (__powerpc__)
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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000040 register unsigned long tbu, tb, tbu2;
Michael W. Hudson800ba232004-08-12 18:19:17 +000041
42 loop:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000043 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;
Michael W. Hudson800ba232004-08-12 18:19:17 +000047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000048 /* The slightly peculiar way of writing the next lines is
49 compiled better by GCC than any other way I tried. */
50 ((long*)(v))[0] = tbu;
51 ((long*)(v))[1] = tb;
Michael W. Hudson800ba232004-08-12 18:19:17 +000052}
53
Mark Dickinsona25b1312009-10-31 10:18:44 +000054#elif defined(__i386__)
55
56/* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000057
Michael W. Hudson75eabd22005-01-18 15:56:11 +000058#define READ_TIMESTAMP(val) \
59 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000060
Mark Dickinsona25b1312009-10-31 10:18:44 +000061#elif defined(__x86_64__)
62
63/* for gcc/x86_64, the "A" constraint in DI mode means *either* rax *or* rdx;
64 not edx:eax as it does for i386. Since rdtsc puts its result in edx:eax
65 even in 64-bit mode, we need to use "a" and "d" for the lower and upper
66 32-bit pieces of the result. */
67
68#define READ_TIMESTAMP(val) \
69 __asm__ __volatile__("rdtsc" : \
70 "=a" (((int*)&(val))[0]), "=d" (((int*)&(val))[1]));
71
72
73#else
74
75#error "Don't know how to implement timestamp counter for this architecture"
76
Michael W. Hudson800ba232004-08-12 18:19:17 +000077#endif
78
Thomas Wouters477c8d52006-05-27 19:21:47 +000079void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000080 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000081{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082 uint64 intr, inst, loop;
83 PyThreadState *tstate = PyThreadState_Get();
84 if (!tstate->interp->tscdump)
85 return;
86 intr = intr1 - intr0;
87 inst = inst1 - inst0 - intr;
88 loop = loop1 - loop0 - intr;
89 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
Stefan Krahb7e10102010-06-23 18:42:39 +000090 opcode, ticked, inst, loop);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091}
Michael W. Hudson800ba232004-08-12 18:19:17 +000092
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
94
Guido van Rossum04691fc1992-08-12 15:35:34 +000095/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000096/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000097
Guido van Rossum408027e1996-12-30 16:17:54 +000098#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000099/* For debugging the interpreter: */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000100#define LLTRACE 1 /* Low-level trace feature */
101#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000102#endif
103
Jeremy Hylton52820442001-01-03 23:52:36 +0000104typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +0000105
Guido van Rossum374a9221991-04-04 10:40:29 +0000106/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000107#ifdef WITH_TSC
Thomas Wouters477c8d52006-05-27 19:21:47 +0000108static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000109#else
Thomas Wouters477c8d52006-05-27 19:21:47 +0000110static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000111#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000112static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
113static PyObject * do_call(PyObject *, PyObject ***, int, int);
114static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000115static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000116 PyObject *);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000117static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
118static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000119#define CALL_FLAG_VAR 1
120#define CALL_FLAG_KW 2
121
Guido van Rossum0a066c01992-03-27 17:29:15 +0000122#ifdef LLTRACE
Guido van Rossumc2e20742006-02-27 22:32:47 +0000123static int lltrace;
Tim Petersdbd9ba62000-07-09 03:09:57 +0000124static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000125#endif
Fred Drake5755ce62001-06-27 19:19:46 +0000126static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000127 int, PyObject *);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000128static int call_trace_protected(Py_tracefunc, PyObject *,
Stefan Krahb7e10102010-06-23 18:42:39 +0000129 PyFrameObject *, int, PyObject *);
Fred Drake5755ce62001-06-27 19:19:46 +0000130static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
Tim Peters8a5c3c72004-04-05 19:36:21 +0000131static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Stefan Krahb7e10102010-06-23 18:42:39 +0000132 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000133
Thomas Wouters477c8d52006-05-27 19:21:47 +0000134static PyObject * cmp_outcome(int, PyObject *, PyObject *);
135static PyObject * import_from(PyObject *, PyObject *);
Thomas Wouters52152252000-08-17 22:55:00 +0000136static int import_all_from(PyObject *, PyObject *);
Neal Norwitzda059e32007-08-26 05:33:45 +0000137static void format_exc_check_arg(PyObject *, const char *, PyObject *);
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000138static void format_exc_unbound(PyCodeObject *co, int oparg);
Victor Stinnerd2a915d2011-10-02 20:34:20 +0200139static PyObject * unicode_concatenate(PyObject *, PyObject *,
140 PyFrameObject *, unsigned char *);
Benjamin Petersonce798522012-01-22 11:24:29 -0500141static PyObject * special_lookup(PyObject *, _Py_Identifier *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000142
Paul Prescode68140d2000-08-30 20:25:01 +0000143#define NAME_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000144 "name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000145#define UNBOUNDLOCAL_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000146 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000147#define UNBOUNDFREE_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000148 "free variable '%.200s' referenced before assignment" \
149 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000150
Guido van Rossum950361c1997-01-24 13:49:28 +0000151/* Dynamic execution profile */
152#ifdef DYNAMIC_EXECUTION_PROFILE
153#ifdef DXPAIRS
154static long dxpairs[257][256];
155#define dxp dxpairs[256]
156#else
157static long dxp[256];
158#endif
159#endif
160
Jeremy Hylton985eba52003-02-05 23:13:00 +0000161/* Function call profile */
162#ifdef CALL_PROFILE
163#define PCALL_NUM 11
164static int pcall[PCALL_NUM];
165
166#define PCALL_ALL 0
167#define PCALL_FUNCTION 1
168#define PCALL_FAST_FUNCTION 2
169#define PCALL_FASTER_FUNCTION 3
170#define PCALL_METHOD 4
171#define PCALL_BOUND_METHOD 5
172#define PCALL_CFUNCTION 6
173#define PCALL_TYPE 7
174#define PCALL_GENERATOR 8
175#define PCALL_OTHER 9
176#define PCALL_POP 10
177
178/* Notes about the statistics
179
180 PCALL_FAST stats
181
182 FAST_FUNCTION means no argument tuple needs to be created.
183 FASTER_FUNCTION means that the fast-path frame setup code is used.
184
185 If there is a method call where the call can be optimized by changing
186 the argument tuple and calling the function directly, it gets recorded
187 twice.
188
189 As a result, the relationship among the statistics appears to be
190 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
191 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
192 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
193 PCALL_METHOD > PCALL_BOUND_METHOD
194*/
195
196#define PCALL(POS) pcall[POS]++
197
198PyObject *
199PyEval_GetCallStats(PyObject *self)
200{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000201 return Py_BuildValue("iiiiiiiiiii",
202 pcall[0], pcall[1], pcall[2], pcall[3],
203 pcall[4], pcall[5], pcall[6], pcall[7],
204 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000205}
206#else
207#define PCALL(O)
208
209PyObject *
210PyEval_GetCallStats(PyObject *self)
211{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000212 Py_INCREF(Py_None);
213 return Py_None;
Jeremy Hylton985eba52003-02-05 23:13:00 +0000214}
215#endif
216
Tim Peters5ca576e2001-06-18 22:08:13 +0000217
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000218#ifdef WITH_THREAD
219#define GIL_REQUEST _Py_atomic_load_relaxed(&gil_drop_request)
220#else
221#define GIL_REQUEST 0
222#endif
223
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000224/* This can set eval_breaker to 0 even though gil_drop_request became
225 1. We believe this is all right because the eval loop will release
226 the GIL eventually anyway. */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000227#define COMPUTE_EVAL_BREAKER() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000228 _Py_atomic_store_relaxed( \
229 &eval_breaker, \
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000230 GIL_REQUEST | \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000231 _Py_atomic_load_relaxed(&pendingcalls_to_do) | \
232 pending_async_exc)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000233
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000234#ifdef WITH_THREAD
235
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000236#define SET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000237 do { \
238 _Py_atomic_store_relaxed(&gil_drop_request, 1); \
239 _Py_atomic_store_relaxed(&eval_breaker, 1); \
240 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000241
242#define RESET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000243 do { \
244 _Py_atomic_store_relaxed(&gil_drop_request, 0); \
245 COMPUTE_EVAL_BREAKER(); \
246 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000247
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000248#endif
249
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000250/* Pending calls are only modified under pending_lock */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000251#define SIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000252 do { \
253 _Py_atomic_store_relaxed(&pendingcalls_to_do, 1); \
254 _Py_atomic_store_relaxed(&eval_breaker, 1); \
255 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000256
257#define UNSIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000258 do { \
259 _Py_atomic_store_relaxed(&pendingcalls_to_do, 0); \
260 COMPUTE_EVAL_BREAKER(); \
261 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000262
263#define SIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000264 do { \
265 pending_async_exc = 1; \
266 _Py_atomic_store_relaxed(&eval_breaker, 1); \
267 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000268
269#define UNSIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000270 do { pending_async_exc = 0; COMPUTE_EVAL_BREAKER(); } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000271
272
Guido van Rossume59214e1994-08-30 08:01:59 +0000273#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000274
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000275#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000276#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000277#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000278#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000279
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000280static PyThread_type_lock pending_lock = 0; /* for pending calls */
Guido van Rossuma9672091994-09-14 13:31:22 +0000281static long main_thread = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000282/* This single variable consolidates all requests to break out of the fast path
283 in the eval loop. */
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000284static _Py_atomic_int eval_breaker = {0};
285/* Request for dropping the GIL */
286static _Py_atomic_int gil_drop_request = {0};
287/* Request for running pending calls. */
288static _Py_atomic_int pendingcalls_to_do = {0};
289/* Request for looking at the `async_exc` field of the current thread state.
290 Guarded by the GIL. */
291static int pending_async_exc = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000292
293#include "ceval_gil.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000294
Tim Peters7f468f22004-10-11 02:40:51 +0000295int
296PyEval_ThreadsInitialized(void)
297{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000298 return gil_created();
Tim Peters7f468f22004-10-11 02:40:51 +0000299}
300
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000301void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000302PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000303{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000304 if (gil_created())
305 return;
306 create_gil();
307 take_gil(PyThreadState_GET());
308 main_thread = PyThread_get_thread_ident();
309 if (!pending_lock)
310 pending_lock = PyThread_allocate_lock();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000311}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000312
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000313void
Antoine Pitrou1df15362010-09-13 14:16:46 +0000314_PyEval_FiniThreads(void)
315{
316 if (!gil_created())
317 return;
318 destroy_gil();
319 assert(!gil_created());
320}
321
322void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000323PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000324{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000325 PyThreadState *tstate = PyThreadState_GET();
326 if (tstate == NULL)
327 Py_FatalError("PyEval_AcquireLock: current thread state is NULL");
328 take_gil(tstate);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000329}
330
331void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000332PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000333{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000334 /* This function must succeed when the current thread state is NULL.
335 We therefore avoid PyThreadState_GET() which dumps a fatal error
336 in debug mode.
337 */
338 drop_gil((PyThreadState*)_Py_atomic_load_relaxed(
339 &_PyThreadState_Current));
Guido van Rossum25ce5661997-08-02 03:10:38 +0000340}
341
342void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000343PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000344{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000345 if (tstate == NULL)
346 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
347 /* Check someone has called PyEval_InitThreads() to create the lock */
348 assert(gil_created());
349 take_gil(tstate);
350 if (PyThreadState_Swap(tstate) != NULL)
351 Py_FatalError(
352 "PyEval_AcquireThread: non-NULL old thread state");
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000353}
354
355void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000356PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000357{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000358 if (tstate == NULL)
359 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
360 if (PyThreadState_Swap(NULL) != tstate)
361 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
362 drop_gil(tstate);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000363}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000364
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200365/* This function is called from PyOS_AfterFork to destroy all threads which are
366 * not running in the child process, and clear internal locks which might be
367 * held by those threads. (This could also be done using pthread_atfork
368 * mechanism, at least for the pthreads implementation.) */
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000369
370void
371PyEval_ReInitThreads(void)
372{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200373 _Py_IDENTIFIER(_after_fork);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000374 PyObject *threading, *result;
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200375 PyThreadState *current_tstate = PyThreadState_GET();
Jesse Nollera8513972008-07-17 16:49:17 +0000376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000377 if (!gil_created())
378 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000379 recreate_gil();
380 pending_lock = PyThread_allocate_lock();
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200381 take_gil(current_tstate);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000382 main_thread = PyThread_get_thread_ident();
Jesse Nollera8513972008-07-17 16:49:17 +0000383
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000384 /* Update the threading module with the new state.
385 */
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200386 threading = PyMapping_GetItemString(current_tstate->interp->modules,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000387 "threading");
388 if (threading == NULL) {
389 /* threading not imported */
390 PyErr_Clear();
391 return;
392 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200393 result = _PyObject_CallMethodId(threading, &PyId__after_fork, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000394 if (result == NULL)
395 PyErr_WriteUnraisable(threading);
396 else
397 Py_DECREF(result);
398 Py_DECREF(threading);
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200399
400 /* Destroy all threads except the current one */
401 _PyThreadState_DeleteExcept(current_tstate);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000402}
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000403
404#else
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000405static _Py_atomic_int eval_breaker = {0};
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000406static int pending_async_exc = 0;
407#endif /* WITH_THREAD */
408
409/* This function is used to signal that async exceptions are waiting to be
410 raised, therefore it is also useful in non-threaded builds. */
411
412void
413_PyEval_SignalAsyncExc(void)
414{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 SIGNAL_ASYNC_EXC();
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000416}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000417
Guido van Rossumff4949e1992-08-05 19:58:53 +0000418/* Functions save_thread and restore_thread are always defined so
419 dynamically loaded modules needn't be compiled separately for use
420 with and without threads: */
421
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000422PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000423PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000424{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000425 PyThreadState *tstate = PyThreadState_Swap(NULL);
426 if (tstate == NULL)
427 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000428#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000429 if (gil_created())
430 drop_gil(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000431#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000432 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000433}
434
435void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000436PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000437{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000438 if (tstate == NULL)
439 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000440#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000441 if (gil_created()) {
442 int err = errno;
443 take_gil(tstate);
Antoine Pitrou0d5e52d2011-05-04 20:02:30 +0200444 /* _Py_Finalizing is protected by the GIL */
445 if (_Py_Finalizing && tstate != _Py_Finalizing) {
446 drop_gil(tstate);
447 PyThread_exit_thread();
448 assert(0); /* unreachable */
449 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000450 errno = err;
451 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000452#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000453 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000454}
455
456
Guido van Rossuma9672091994-09-14 13:31:22 +0000457/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
458 signal handlers or Mac I/O completion routines) can schedule calls
459 to a function to be called synchronously.
460 The synchronous function is called with one void* argument.
461 It should return 0 for success or -1 for failure -- failure should
462 be accompanied by an exception.
463
464 If registry succeeds, the registry function returns 0; if it fails
465 (e.g. due to too many pending calls) it returns -1 (without setting
466 an exception condition).
467
468 Note that because registry may occur from within signal handlers,
469 or other asynchronous events, calling malloc() is unsafe!
470
471#ifdef WITH_THREAD
472 Any thread can schedule pending calls, but only the main thread
473 will execute them.
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000474 There is no facility to schedule calls to a particular thread, but
475 that should be easy to change, should that ever be required. In
476 that case, the static variables here should go into the python
477 threadstate.
Guido van Rossuma9672091994-09-14 13:31:22 +0000478#endif
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000479*/
Guido van Rossuma9672091994-09-14 13:31:22 +0000480
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000481#ifdef WITH_THREAD
482
483/* The WITH_THREAD implementation is thread-safe. It allows
484 scheduling to be made from any thread, and even from an executing
485 callback.
486 */
487
488#define NPENDINGCALLS 32
489static struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000490 int (*func)(void *);
491 void *arg;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000492} pendingcalls[NPENDINGCALLS];
493static int pendingfirst = 0;
494static int pendinglast = 0;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000495
496int
497Py_AddPendingCall(int (*func)(void *), void *arg)
498{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000499 int i, j, result=0;
500 PyThread_type_lock lock = pending_lock;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000502 /* try a few times for the lock. Since this mechanism is used
503 * for signal handling (on the main thread), there is a (slim)
504 * chance that a signal is delivered on the same thread while we
505 * hold the lock during the Py_MakePendingCalls() function.
506 * This avoids a deadlock in that case.
507 * Note that signals can be delivered on any thread. In particular,
508 * on Windows, a SIGINT is delivered on a system-created worker
509 * thread.
510 * We also check for lock being NULL, in the unlikely case that
511 * this function is called before any bytecode evaluation takes place.
512 */
513 if (lock != NULL) {
514 for (i = 0; i<100; i++) {
515 if (PyThread_acquire_lock(lock, NOWAIT_LOCK))
516 break;
517 }
518 if (i == 100)
519 return -1;
520 }
521
522 i = pendinglast;
523 j = (i + 1) % NPENDINGCALLS;
524 if (j == pendingfirst) {
525 result = -1; /* Queue full */
526 } else {
527 pendingcalls[i].func = func;
528 pendingcalls[i].arg = arg;
529 pendinglast = j;
530 }
531 /* signal main loop */
532 SIGNAL_PENDING_CALLS();
533 if (lock != NULL)
534 PyThread_release_lock(lock);
535 return result;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000536}
537
538int
539Py_MakePendingCalls(void)
540{
Charles-François Natalif23339a2011-07-23 18:15:43 +0200541 static int busy = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000542 int i;
543 int r = 0;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 if (!pending_lock) {
546 /* initial allocation of the lock */
547 pending_lock = PyThread_allocate_lock();
548 if (pending_lock == NULL)
549 return -1;
550 }
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000551
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000552 /* only service pending calls on main thread */
553 if (main_thread && PyThread_get_thread_ident() != main_thread)
554 return 0;
555 /* don't perform recursive pending calls */
Charles-François Natalif23339a2011-07-23 18:15:43 +0200556 if (busy)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 return 0;
Charles-François Natalif23339a2011-07-23 18:15:43 +0200558 busy = 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000559 /* perform a bounded number of calls, in case of recursion */
560 for (i=0; i<NPENDINGCALLS; i++) {
561 int j;
562 int (*func)(void *);
563 void *arg = NULL;
564
565 /* pop one item off the queue while holding the lock */
566 PyThread_acquire_lock(pending_lock, WAIT_LOCK);
567 j = pendingfirst;
568 if (j == pendinglast) {
569 func = NULL; /* Queue empty */
570 } else {
571 func = pendingcalls[j].func;
572 arg = pendingcalls[j].arg;
573 pendingfirst = (j + 1) % NPENDINGCALLS;
574 }
575 if (pendingfirst != pendinglast)
576 SIGNAL_PENDING_CALLS();
577 else
578 UNSIGNAL_PENDING_CALLS();
579 PyThread_release_lock(pending_lock);
580 /* having released the lock, perform the callback */
581 if (func == NULL)
582 break;
583 r = func(arg);
584 if (r)
585 break;
586 }
Charles-François Natalif23339a2011-07-23 18:15:43 +0200587 busy = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000588 return r;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000589}
590
591#else /* if ! defined WITH_THREAD */
592
593/*
594 WARNING! ASYNCHRONOUSLY EXECUTING CODE!
595 This code is used for signal handling in python that isn't built
596 with WITH_THREAD.
597 Don't use this implementation when Py_AddPendingCalls() can happen
598 on a different thread!
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599
Guido van Rossuma9672091994-09-14 13:31:22 +0000600 There are two possible race conditions:
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000601 (1) nested asynchronous calls to Py_AddPendingCall()
602 (2) AddPendingCall() calls made while pending calls are being processed.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000603
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000604 (1) is very unlikely because typically signal delivery
605 is blocked during signal handling. So it should be impossible.
606 (2) is a real possibility.
Guido van Rossuma9672091994-09-14 13:31:22 +0000607 The current code is safe against (2), but not against (1).
608 The safety against (2) is derived from the fact that only one
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000609 thread is present, interrupted by signals, and that the critical
610 section is protected with the "busy" variable. On Windows, which
611 delivers SIGINT on a system thread, this does not hold and therefore
612 Windows really shouldn't use this version.
613 The two threads could theoretically wiggle around the "busy" variable.
Guido van Rossuma027efa1997-05-05 20:56:21 +0000614*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000615
Guido van Rossuma9672091994-09-14 13:31:22 +0000616#define NPENDINGCALLS 32
617static struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 int (*func)(void *);
619 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000620} pendingcalls[NPENDINGCALLS];
621static volatile int pendingfirst = 0;
622static volatile int pendinglast = 0;
Benjamin Peterson08ec84c2010-05-30 14:49:32 +0000623static _Py_atomic_int pendingcalls_to_do = {0};
Guido van Rossuma9672091994-09-14 13:31:22 +0000624
625int
Thomas Wouters334fb892000-07-25 12:56:38 +0000626Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000627{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 static volatile int busy = 0;
629 int i, j;
630 /* XXX Begin critical section */
631 if (busy)
632 return -1;
633 busy = 1;
634 i = pendinglast;
635 j = (i + 1) % NPENDINGCALLS;
636 if (j == pendingfirst) {
637 busy = 0;
638 return -1; /* Queue full */
639 }
640 pendingcalls[i].func = func;
641 pendingcalls[i].arg = arg;
642 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 SIGNAL_PENDING_CALLS();
645 busy = 0;
646 /* XXX End critical section */
647 return 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000648}
649
Guido van Rossum180d7b41994-09-29 09:45:57 +0000650int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000651Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000652{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000653 static int busy = 0;
654 if (busy)
655 return 0;
656 busy = 1;
657 UNSIGNAL_PENDING_CALLS();
658 for (;;) {
659 int i;
660 int (*func)(void *);
661 void *arg;
662 i = pendingfirst;
663 if (i == pendinglast)
664 break; /* Queue empty */
665 func = pendingcalls[i].func;
666 arg = pendingcalls[i].arg;
667 pendingfirst = (i + 1) % NPENDINGCALLS;
668 if (func(arg) < 0) {
669 busy = 0;
670 SIGNAL_PENDING_CALLS(); /* We're not done yet */
671 return -1;
672 }
673 }
674 busy = 0;
675 return 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000676}
677
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000678#endif /* WITH_THREAD */
679
Guido van Rossuma9672091994-09-14 13:31:22 +0000680
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000681/* The interpreter's recursion limit */
682
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000683#ifndef Py_DEFAULT_RECURSION_LIMIT
684#define Py_DEFAULT_RECURSION_LIMIT 1000
685#endif
686static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
687int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000688
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000689int
690Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000691{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000692 return recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000693}
694
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000695void
696Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000697{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000698 recursion_limit = new_limit;
699 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000700}
701
Armin Rigo2b3eb402003-10-28 12:05:48 +0000702/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
703 if the recursion_depth reaches _Py_CheckRecursionLimit.
704 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
705 to guarantee that _Py_CheckRecursiveCall() is regularly called.
706 Without USE_STACKCHECK, there is no need for this. */
707int
708_Py_CheckRecursiveCall(char *where)
709{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 PyThreadState *tstate = PyThreadState_GET();
Armin Rigo2b3eb402003-10-28 12:05:48 +0000711
712#ifdef USE_STACKCHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000713 if (PyOS_CheckStack()) {
714 --tstate->recursion_depth;
715 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
716 return -1;
717 }
Armin Rigo2b3eb402003-10-28 12:05:48 +0000718#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719 _Py_CheckRecursionLimit = recursion_limit;
720 if (tstate->recursion_critical)
721 /* Somebody asked that we don't check for recursion. */
722 return 0;
723 if (tstate->overflowed) {
724 if (tstate->recursion_depth > recursion_limit + 50) {
725 /* Overflowing while handling an overflow. Give up. */
726 Py_FatalError("Cannot recover from stack overflow.");
727 }
728 return 0;
729 }
730 if (tstate->recursion_depth > recursion_limit) {
731 --tstate->recursion_depth;
732 tstate->overflowed = 1;
733 PyErr_Format(PyExc_RuntimeError,
734 "maximum recursion depth exceeded%s",
735 where);
736 return -1;
737 }
738 return 0;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000739}
740
Guido van Rossum374a9221991-04-04 10:40:29 +0000741/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000742enum why_code {
Stefan Krahb7e10102010-06-23 18:42:39 +0000743 WHY_NOT = 0x0001, /* No error */
744 WHY_EXCEPTION = 0x0002, /* Exception occurred */
Stefan Krahb7e10102010-06-23 18:42:39 +0000745 WHY_RETURN = 0x0008, /* 'return' statement */
746 WHY_BREAK = 0x0010, /* 'break' statement */
747 WHY_CONTINUE = 0x0020, /* 'continue' statement */
748 WHY_YIELD = 0x0040, /* 'yield' operator */
749 WHY_SILENCED = 0x0080 /* Exception silenced by 'with' */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000750};
Guido van Rossum374a9221991-04-04 10:40:29 +0000751
Benjamin Peterson87880242011-07-03 16:48:31 -0500752static void save_exc_state(PyThreadState *, PyFrameObject *);
753static void swap_exc_state(PyThreadState *, PyFrameObject *);
754static void restore_and_clear_exc_state(PyThreadState *, PyFrameObject *);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -0400755static int do_raise(PyObject *, PyObject *);
Guido van Rossum0368b722007-05-11 16:50:42 +0000756static int unpack_iterable(PyObject *, int, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000757
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +0000758/* Records whether tracing is on for any thread. Counts the number of
759 threads for which tstate->c_tracefunc is non-NULL, so if the value
760 is 0, we know we don't have to check this thread's c_tracefunc.
761 This speeds up the if statement in PyEval_EvalFrameEx() after
762 fast_next_opcode*/
763static int _Py_TracingPossible = 0;
764
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000765
Guido van Rossum374a9221991-04-04 10:40:29 +0000766
Guido van Rossumb209a111997-04-29 18:18:01 +0000767PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000768PyEval_EvalCode(PyObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000769{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000770 return PyEval_EvalCodeEx(co,
771 globals, locals,
772 (PyObject **)NULL, 0,
773 (PyObject **)NULL, 0,
774 (PyObject **)NULL, 0,
775 NULL, NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000776}
777
778
779/* Interpreter main loop */
780
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000781PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000782PyEval_EvalFrame(PyFrameObject *f) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000783 /* This is for backward compatibility with extension modules that
784 used this API; core interpreter code should call
785 PyEval_EvalFrameEx() */
786 return PyEval_EvalFrameEx(f, 0);
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000787}
788
789PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000790PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000791{
Guido van Rossum950361c1997-01-24 13:49:28 +0000792#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000793 int lastopcode = 0;
Guido van Rossum950361c1997-01-24 13:49:28 +0000794#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000795 register PyObject **stack_pointer; /* Next free slot in value stack */
796 register unsigned char *next_instr;
797 register int opcode; /* Current opcode */
798 register int oparg; /* Current opcode argument, if any */
799 register enum why_code why; /* Reason for block stack unwind */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000800 register PyObject **fastlocals, **freevars;
801 PyObject *retval = NULL; /* Return value */
802 PyThreadState *tstate = PyThreadState_GET();
803 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000804
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000806
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000807 not (instr_lb <= current_bytecode_offset < instr_ub)
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000808
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000809 is true when the line being executed has changed. The
810 initial values are such as to make this false the first
811 time it is tested. */
812 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000813
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 unsigned char *first_instr;
815 PyObject *names;
816 PyObject *consts;
Guido van Rossum374a9221991-04-04 10:40:29 +0000817
Brett Cannon368b4b72012-04-02 12:17:59 -0400818#ifdef LLTRACE
Victor Stinner3c1e4812012-03-26 22:10:51 +0200819 _Py_IDENTIFIER(__ltrace__);
Brett Cannon368b4b72012-04-02 12:17:59 -0400820#endif
Victor Stinner3c1e4812012-03-26 22:10:51 +0200821
Antoine Pitroub52ec782009-01-25 16:34:23 +0000822/* Computed GOTOs, or
823 the-optimization-commonly-but-improperly-known-as-"threaded code"
824 using gcc's labels-as-values extension
825 (http://gcc.gnu.org/onlinedocs/gcc/Labels-as-Values.html).
826
827 The traditional bytecode evaluation loop uses a "switch" statement, which
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000828 decent compilers will optimize as a single indirect branch instruction
Antoine Pitroub52ec782009-01-25 16:34:23 +0000829 combined with a lookup table of jump addresses. However, since the
830 indirect jump instruction is shared by all opcodes, the CPU will have a
831 hard time making the right prediction for where to jump next (actually,
832 it will be always wrong except in the uncommon case of a sequence of
833 several identical opcodes).
834
835 "Threaded code" in contrast, uses an explicit jump table and an explicit
836 indirect jump instruction at the end of each opcode. Since the jump
837 instruction is at a different address for each opcode, the CPU will make a
838 separate prediction for each of these instructions, which is equivalent to
839 predicting the second opcode of each opcode pair. These predictions have
840 a much better chance to turn out valid, especially in small bytecode loops.
841
842 A mispredicted branch on a modern CPU flushes the whole pipeline and
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000843 can cost several CPU cycles (depending on the pipeline depth),
Antoine Pitroub52ec782009-01-25 16:34:23 +0000844 and potentially many more instructions (depending on the pipeline width).
845 A correctly predicted branch, however, is nearly free.
846
847 At the time of this writing, the "threaded code" version is up to 15-20%
848 faster than the normal "switch" version, depending on the compiler and the
849 CPU architecture.
850
851 We disable the optimization if DYNAMIC_EXECUTION_PROFILE is defined,
852 because it would render the measurements invalid.
853
854
855 NOTE: care must be taken that the compiler doesn't try to "optimize" the
856 indirect jumps by sharing them between all opcodes. Such optimizations
857 can be disabled on gcc by using the -fno-gcse flag (or possibly
858 -fno-crossjumping).
859*/
860
Antoine Pitrou042b1282010-08-13 21:15:58 +0000861#ifdef DYNAMIC_EXECUTION_PROFILE
Antoine Pitroub52ec782009-01-25 16:34:23 +0000862#undef USE_COMPUTED_GOTOS
Antoine Pitrou042b1282010-08-13 21:15:58 +0000863#define USE_COMPUTED_GOTOS 0
Antoine Pitroub52ec782009-01-25 16:34:23 +0000864#endif
865
Antoine Pitrou042b1282010-08-13 21:15:58 +0000866#ifdef HAVE_COMPUTED_GOTOS
867 #ifndef USE_COMPUTED_GOTOS
868 #define USE_COMPUTED_GOTOS 1
869 #endif
870#else
871 #if defined(USE_COMPUTED_GOTOS) && USE_COMPUTED_GOTOS
872 #error "Computed gotos are not supported on this compiler."
873 #endif
874 #undef USE_COMPUTED_GOTOS
875 #define USE_COMPUTED_GOTOS 0
876#endif
877
878#if USE_COMPUTED_GOTOS
Antoine Pitroub52ec782009-01-25 16:34:23 +0000879/* Import the static jump table */
880#include "opcode_targets.h"
881
882/* This macro is used when several opcodes defer to the same implementation
883 (e.g. SETUP_LOOP, SETUP_FINALLY) */
884#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000885 TARGET_##op: \
886 opcode = op; \
887 if (HAS_ARG(op)) \
888 oparg = NEXTARG(); \
889 case op: \
890 goto impl; \
Antoine Pitroub52ec782009-01-25 16:34:23 +0000891
892#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000893 TARGET_##op: \
894 opcode = op; \
895 if (HAS_ARG(op)) \
896 oparg = NEXTARG(); \
897 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000898
899
900#define DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000901 { \
902 if (!_Py_atomic_load_relaxed(&eval_breaker)) { \
903 FAST_DISPATCH(); \
904 } \
905 continue; \
906 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000907
908#ifdef LLTRACE
909#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000910 { \
911 if (!lltrace && !_Py_TracingPossible) { \
912 f->f_lasti = INSTR_OFFSET(); \
913 goto *opcode_targets[*next_instr++]; \
914 } \
915 goto fast_next_opcode; \
916 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000917#else
918#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000919 { \
920 if (!_Py_TracingPossible) { \
921 f->f_lasti = INSTR_OFFSET(); \
922 goto *opcode_targets[*next_instr++]; \
923 } \
924 goto fast_next_opcode; \
925 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000926#endif
927
928#else
929#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000930 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000931#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000932 /* silence compiler warnings about `impl` unused */ \
933 if (0) goto impl; \
934 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000935#define DISPATCH() continue
936#define FAST_DISPATCH() goto fast_next_opcode
937#endif
938
939
Neal Norwitza81d2202002-07-14 00:27:26 +0000940/* Tuple access macros */
941
942#ifndef Py_DEBUG
943#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
944#else
945#define GETITEM(v, i) PyTuple_GetItem((v), (i))
946#endif
947
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000948#ifdef WITH_TSC
949/* Use Pentium timestamp counter to mark certain events:
950 inst0 -- beginning of switch statement for opcode dispatch
951 inst1 -- end of switch statement (may be skipped)
952 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000953 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000954 (may be skipped)
955 intr1 -- beginning of long interruption
956 intr2 -- end of long interruption
957
958 Many opcodes call out to helper C functions. In some cases, the
959 time in those functions should be counted towards the time for the
960 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
961 calls another Python function; there's no point in charge all the
962 bytecode executed by the called function to the caller.
963
964 It's hard to make a useful judgement statically. In the presence
965 of operator overloading, it's impossible to tell if a call will
966 execute new Python code or not.
967
968 It's a case-by-case judgement. I'll use intr1 for the following
969 cases:
970
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000971 IMPORT_STAR
972 IMPORT_FROM
973 CALL_FUNCTION (and friends)
974
975 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000976 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
977 int ticked = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000978
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979 READ_TIMESTAMP(inst0);
980 READ_TIMESTAMP(inst1);
981 READ_TIMESTAMP(loop0);
982 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000983
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000984 /* shut up the compiler */
985 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000986#endif
987
Guido van Rossum374a9221991-04-04 10:40:29 +0000988/* Code access macros */
989
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990#define INSTR_OFFSET() ((int)(next_instr - first_instr))
991#define NEXTOP() (*next_instr++)
992#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
993#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
994#define JUMPTO(x) (next_instr = first_instr + (x))
995#define JUMPBY(x) (next_instr += (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000996
Raymond Hettingerf606f872003-03-16 03:11:04 +0000997/* OpCode prediction macros
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000998 Some opcodes tend to come in pairs thus making it possible to
999 predict the second code when the first is run. For example,
1000 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
1001 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001002
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 Verifying the prediction costs a single high-speed test of a register
1004 variable against a constant. If the pairing was good, then the
1005 processor's own internal branch predication has a high likelihood of
1006 success, resulting in a nearly zero-overhead transition to the
1007 next opcode. A successful prediction saves a trip through the eval-loop
1008 including its two unpredictable branches, the HAS_ARG test and the
1009 switch-case. Combined with the processor's internal branch prediction,
1010 a successful PREDICT has the effect of making the two opcodes run as if
1011 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001012
Georg Brandl86b2fb92008-07-16 03:43:04 +00001013 If collecting opcode statistics, your choices are to either keep the
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001014 predictions turned-on and interpret the results as if some opcodes
1015 had been combined or turn-off predictions so that the opcode frequency
1016 counter updates for both opcodes.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001017
1018 Opcode prediction is disabled with threaded code, since the latter allows
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001019 the CPU to record separate branch prediction information for each
1020 opcode.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001021
Raymond Hettingerf606f872003-03-16 03:11:04 +00001022*/
1023
Antoine Pitrou042b1282010-08-13 21:15:58 +00001024#if defined(DYNAMIC_EXECUTION_PROFILE) || USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001025#define PREDICT(op) if (0) goto PRED_##op
1026#define PREDICTED(op) PRED_##op:
1027#define PREDICTED_WITH_ARG(op) PRED_##op:
Raymond Hettingera7216982004-02-08 19:59:27 +00001028#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001029#define PREDICT(op) if (*next_instr == op) goto PRED_##op
1030#define PREDICTED(op) PRED_##op: next_instr++
1031#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Antoine Pitroub52ec782009-01-25 16:34:23 +00001032#endif
1033
Raymond Hettingerf606f872003-03-16 03:11:04 +00001034
Guido van Rossum374a9221991-04-04 10:40:29 +00001035/* Stack manipulation macros */
1036
Martin v. Löwis18e16552006-02-15 17:27:45 +00001037/* The stack can grow at most MAXINT deep, as co_nlocals and
1038 co_stacksize are ints. */
Stefan Krahb7e10102010-06-23 18:42:39 +00001039#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
1040#define EMPTY() (STACK_LEVEL() == 0)
1041#define TOP() (stack_pointer[-1])
1042#define SECOND() (stack_pointer[-2])
1043#define THIRD() (stack_pointer[-3])
1044#define FOURTH() (stack_pointer[-4])
1045#define PEEK(n) (stack_pointer[-(n)])
1046#define SET_TOP(v) (stack_pointer[-1] = (v))
1047#define SET_SECOND(v) (stack_pointer[-2] = (v))
1048#define SET_THIRD(v) (stack_pointer[-3] = (v))
1049#define SET_FOURTH(v) (stack_pointer[-4] = (v))
1050#define SET_VALUE(n, v) (stack_pointer[-(n)] = (v))
1051#define BASIC_STACKADJ(n) (stack_pointer += n)
1052#define BASIC_PUSH(v) (*stack_pointer++ = (v))
1053#define BASIC_POP() (*--stack_pointer)
Guido van Rossum374a9221991-04-04 10:40:29 +00001054
Guido van Rossum96a42c81992-01-12 02:29:51 +00001055#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001056#define PUSH(v) { (void)(BASIC_PUSH(v), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001057 lltrace && prtrace(TOP(), "push")); \
1058 assert(STACK_LEVEL() <= co->co_stacksize); }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001059#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001060 BASIC_POP())
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001061#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001062 lltrace && prtrace(TOP(), "stackadj")); \
1063 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes0449f632007-12-15 01:27:15 +00001064#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
Stefan Krahb7e10102010-06-23 18:42:39 +00001065 prtrace((STACK_POINTER)[-1], "ext_pop")), \
1066 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001067#else
Stefan Krahb7e10102010-06-23 18:42:39 +00001068#define PUSH(v) BASIC_PUSH(v)
1069#define POP() BASIC_POP()
1070#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001071#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001072#endif
1073
Guido van Rossum681d79a1995-07-18 14:51:37 +00001074/* Local variable macros */
1075
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001076#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +00001077
1078/* The SETLOCAL() macro must not DECREF the local variable in-place and
1079 then store the new value; it must copy the old value to a temporary
1080 value, then store the new value, and then DECREF the temporary value.
1081 This is because it is possible that during the DECREF the frame is
1082 accessed by other code (e.g. a __del__ method or gc.collect()) and the
1083 variable would be pointing to already-freed memory. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001084#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
Stefan Krahb7e10102010-06-23 18:42:39 +00001085 GETLOCAL(i) = value; \
1086 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +00001087
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001088
1089#define UNWIND_BLOCK(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001090 while (STACK_LEVEL() > (b)->b_level) { \
1091 PyObject *v = POP(); \
1092 Py_XDECREF(v); \
1093 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001094
1095#define UNWIND_EXCEPT_HANDLER(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001096 { \
1097 PyObject *type, *value, *traceback; \
1098 assert(STACK_LEVEL() >= (b)->b_level + 3); \
1099 while (STACK_LEVEL() > (b)->b_level + 3) { \
1100 value = POP(); \
1101 Py_XDECREF(value); \
1102 } \
1103 type = tstate->exc_type; \
1104 value = tstate->exc_value; \
1105 traceback = tstate->exc_traceback; \
1106 tstate->exc_type = POP(); \
1107 tstate->exc_value = POP(); \
1108 tstate->exc_traceback = POP(); \
1109 Py_XDECREF(type); \
1110 Py_XDECREF(value); \
1111 Py_XDECREF(traceback); \
1112 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001113
Guido van Rossuma027efa1997-05-05 20:56:21 +00001114/* Start of code */
1115
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001116 /* push frame */
1117 if (Py_EnterRecursiveCall(""))
1118 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +00001119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001120 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +00001121
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 if (tstate->use_tracing) {
1123 if (tstate->c_tracefunc != NULL) {
1124 /* tstate->c_tracefunc, if defined, is a
1125 function that will be called on *every* entry
1126 to a code block. Its return value, if not
1127 None, is a function that will be called at
1128 the start of each executed line of code.
1129 (Actually, the function must return itself
1130 in order to continue tracing.) The trace
1131 functions are called with three arguments:
1132 a pointer to the current frame, a string
1133 indicating why the function is called, and
1134 an argument which depends on the situation.
1135 The global trace function is also called
1136 whenever an exception is detected. */
1137 if (call_trace_protected(tstate->c_tracefunc,
1138 tstate->c_traceobj,
1139 f, PyTrace_CALL, Py_None)) {
1140 /* Trace function raised an error */
1141 goto exit_eval_frame;
1142 }
1143 }
1144 if (tstate->c_profilefunc != NULL) {
1145 /* Similar for c_profilefunc, except it needn't
1146 return itself and isn't called for "line" events */
1147 if (call_trace_protected(tstate->c_profilefunc,
1148 tstate->c_profileobj,
1149 f, PyTrace_CALL, Py_None)) {
1150 /* Profile function raised an error */
1151 goto exit_eval_frame;
1152 }
1153 }
1154 }
Neil Schemenauer6c0f2002001-09-04 19:03:35 +00001155
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001156 co = f->f_code;
1157 names = co->co_names;
1158 consts = co->co_consts;
1159 fastlocals = f->f_localsplus;
1160 freevars = f->f_localsplus + co->co_nlocals;
1161 first_instr = (unsigned char*) PyBytes_AS_STRING(co->co_code);
1162 /* An explanation is in order for the next line.
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 f->f_lasti now refers to the index of the last instruction
1165 executed. You might think this was obvious from the name, but
1166 this wasn't always true before 2.3! PyFrame_New now sets
1167 f->f_lasti to -1 (i.e. the index *before* the first instruction)
1168 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
1169 does work. Promise.
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001170 YIELD_FROM sets f_lasti to itself, in order to repeated yield
1171 multiple values.
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001172
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001173 When the PREDICT() macros are enabled, some opcode pairs follow in
1174 direct succession without updating f->f_lasti. A successful
1175 prediction effectively links the two codes together as if they
1176 were a single new opcode; accordingly,f->f_lasti will point to
1177 the first code in the pair (for instance, GET_ITER followed by
1178 FOR_ITER is effectively a single opcode and f->f_lasti will point
1179 at to the beginning of the combined pair.)
1180 */
1181 next_instr = first_instr + f->f_lasti + 1;
1182 stack_pointer = f->f_stacktop;
1183 assert(stack_pointer != NULL);
1184 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001186 if (co->co_flags & CO_GENERATOR && !throwflag) {
1187 if (f->f_exc_type != NULL && f->f_exc_type != Py_None) {
1188 /* We were in an except handler when we left,
1189 restore the exception state which was put aside
1190 (see YIELD_VALUE). */
Benjamin Peterson87880242011-07-03 16:48:31 -05001191 swap_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001192 }
Benjamin Peterson87880242011-07-03 16:48:31 -05001193 else
1194 save_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001195 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001196
Tim Peters5ca576e2001-06-18 22:08:13 +00001197#ifdef LLTRACE
Victor Stinner3c1e4812012-03-26 22:10:51 +02001198 lltrace = _PyDict_GetItemId(f->f_globals, &PyId___ltrace__) != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00001199#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00001200
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001201 why = WHY_NOT;
Guido van Rossumac7be682001-01-17 15:42:30 +00001202
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001203 if (throwflag) /* support for generator.throw() */
1204 goto error;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001206 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001207#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 if (inst1 == 0) {
1209 /* Almost surely, the opcode executed a break
1210 or a continue, preventing inst1 from being set
1211 on the way out of the loop.
1212 */
1213 READ_TIMESTAMP(inst1);
1214 loop1 = inst1;
1215 }
1216 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
1217 intr0, intr1);
1218 ticked = 0;
1219 inst1 = 0;
1220 intr0 = 0;
1221 intr1 = 0;
1222 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001223#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001224 assert(stack_pointer >= f->f_valuestack); /* else underflow */
1225 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001226
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 /* Do periodic things. Doing this every time through
1228 the loop would add too much overhead, so we do it
1229 only every Nth instruction. We also do it if
1230 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
1231 event needs attention (e.g. a signal handler or
1232 async I/O handler); see Py_AddPendingCall() and
1233 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +00001234
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001235 if (_Py_atomic_load_relaxed(&eval_breaker)) {
1236 if (*next_instr == SETUP_FINALLY) {
1237 /* Make the last opcode before
Ezio Melotti13925002011-03-16 11:05:33 +02001238 a try: finally: block uninterruptible. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001239 goto fast_next_opcode;
1240 }
1241 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001242#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001243 ticked = 1;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001244#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001245 if (_Py_atomic_load_relaxed(&pendingcalls_to_do)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001246 if (Py_MakePendingCalls() < 0)
1247 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001248 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001249#ifdef WITH_THREAD
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001250 if (_Py_atomic_load_relaxed(&gil_drop_request)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 /* Give another thread a chance */
1252 if (PyThreadState_Swap(NULL) != tstate)
1253 Py_FatalError("ceval: tstate mix-up");
1254 drop_gil(tstate);
1255
1256 /* Other threads may run now */
1257
1258 take_gil(tstate);
1259 if (PyThreadState_Swap(tstate) != NULL)
1260 Py_FatalError("ceval: orphan tstate");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001261 }
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001262#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001263 /* Check for asynchronous exceptions. */
1264 if (tstate->async_exc != NULL) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001265 PyObject *exc = tstate->async_exc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001266 tstate->async_exc = NULL;
1267 UNSIGNAL_ASYNC_EXC();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001268 PyErr_SetNone(exc);
1269 Py_DECREF(exc);
1270 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001271 }
1272 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001273
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001274 fast_next_opcode:
1275 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001276
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001277 /* line-by-line tracing support */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001278
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001279 if (_Py_TracingPossible &&
Benjamin Peterson51f46162013-01-23 08:38:47 -05001280 tstate->c_tracefunc != NULL && !tstate->tracing) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001281 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001282 /* see maybe_call_line_trace
1283 for expository comments */
1284 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 err = maybe_call_line_trace(tstate->c_tracefunc,
1287 tstate->c_traceobj,
1288 f, &instr_lb, &instr_ub,
1289 &instr_prev);
1290 /* Reload possibly changed frame fields */
1291 JUMPTO(f->f_lasti);
1292 if (f->f_stacktop != NULL) {
1293 stack_pointer = f->f_stacktop;
1294 f->f_stacktop = NULL;
1295 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001296 if (err)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001297 /* trace function raised an exception */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001298 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001299 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001301 /* Extract opcode and argument */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001302
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001303 opcode = NEXTOP();
1304 oparg = 0; /* allows oparg to be stored in a register because
1305 it doesn't have to be remembered across a full loop */
1306 if (HAS_ARG(opcode))
1307 oparg = NEXTARG();
Stefan Krahb7e10102010-06-23 18:42:39 +00001308 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001309#ifdef DYNAMIC_EXECUTION_PROFILE
1310#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001311 dxpairs[lastopcode][opcode]++;
1312 lastopcode = opcode;
Guido van Rossum950361c1997-01-24 13:49:28 +00001313#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001314 dxp[opcode]++;
Guido van Rossum950361c1997-01-24 13:49:28 +00001315#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001316
Guido van Rossum96a42c81992-01-12 02:29:51 +00001317#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001318 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001319
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001320 if (lltrace) {
1321 if (HAS_ARG(opcode)) {
1322 printf("%d: %d, %d\n",
1323 f->f_lasti, opcode, oparg);
1324 }
1325 else {
1326 printf("%d: %d\n",
1327 f->f_lasti, opcode);
1328 }
1329 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001330#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001331
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001332 /* Main switch on opcode */
1333 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001334
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001335 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001336
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001337 /* BEWARE!
1338 It is essential that any operation that fails sets either
1339 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1340 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001341
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001342 TARGET(NOP)
1343 FAST_DISPATCH();
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001344
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001345 TARGET(LOAD_FAST) {
1346 PyObject *value = GETLOCAL(oparg);
1347 if (value == NULL) {
1348 format_exc_check_arg(PyExc_UnboundLocalError,
1349 UNBOUNDLOCAL_ERROR_MSG,
1350 PyTuple_GetItem(co->co_varnames, oparg));
1351 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001352 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001353 Py_INCREF(value);
1354 PUSH(value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001355 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001356 }
1357
1358 TARGET(LOAD_CONST) {
1359 PyObject *value = GETITEM(consts, oparg);
1360 Py_INCREF(value);
1361 PUSH(value);
1362 FAST_DISPATCH();
1363 }
Neil Schemenauer63543862002-02-17 19:10:14 +00001364
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001365 PREDICTED_WITH_ARG(STORE_FAST);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001366 TARGET(STORE_FAST) {
1367 PyObject *value = POP();
1368 SETLOCAL(oparg, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001369 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001370 }
Neil Schemenauer63543862002-02-17 19:10:14 +00001371
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001372 TARGET(POP_TOP) {
1373 PyObject *value = POP();
1374 Py_DECREF(value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001375 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001376 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001377
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001378 TARGET(ROT_TWO) {
1379 PyObject *top = TOP();
1380 PyObject *second = SECOND();
1381 SET_TOP(second);
1382 SET_SECOND(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001384 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001385
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001386 TARGET(ROT_THREE) {
1387 PyObject *top = TOP();
1388 PyObject *second = SECOND();
1389 PyObject *third = THIRD();
1390 SET_TOP(second);
1391 SET_SECOND(third);
1392 SET_THIRD(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001393 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001394 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001395
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001396 TARGET(DUP_TOP) {
1397 PyObject *top = TOP();
1398 Py_INCREF(top);
1399 PUSH(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001400 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001401 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001402
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001403 TARGET(DUP_TOP_TWO) {
1404 PyObject *top = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001405 PyObject *second = SECOND();
Benjamin Petersonf208df32012-10-12 11:37:56 -04001406 Py_INCREF(top);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001407 Py_INCREF(second);
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001408 STACKADJ(2);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001409 SET_TOP(top);
1410 SET_SECOND(second);
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001411 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001412 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001413
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001414 TARGET(UNARY_POSITIVE) {
1415 PyObject *value = TOP();
1416 PyObject *res = PyNumber_Positive(value);
1417 Py_DECREF(value);
1418 SET_TOP(res);
1419 if (res == NULL)
1420 goto error;
1421 DISPATCH();
1422 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001423
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001424 TARGET(UNARY_NEGATIVE) {
1425 PyObject *value = TOP();
1426 PyObject *res = PyNumber_Negative(value);
1427 Py_DECREF(value);
1428 SET_TOP(res);
1429 if (res == NULL)
1430 goto error;
1431 DISPATCH();
1432 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001433
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001434 TARGET(UNARY_NOT) {
1435 PyObject *value = TOP();
1436 int err = PyObject_IsTrue(value);
1437 Py_DECREF(value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001438 if (err == 0) {
1439 Py_INCREF(Py_True);
1440 SET_TOP(Py_True);
1441 DISPATCH();
1442 }
1443 else if (err > 0) {
1444 Py_INCREF(Py_False);
1445 SET_TOP(Py_False);
1446 err = 0;
1447 DISPATCH();
1448 }
1449 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001450 goto error;
1451 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001452
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001453 TARGET(UNARY_INVERT) {
1454 PyObject *value = TOP();
1455 PyObject *res = PyNumber_Invert(value);
1456 Py_DECREF(value);
1457 SET_TOP(res);
1458 if (res == NULL)
1459 goto error;
1460 DISPATCH();
1461 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001462
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001463 TARGET(BINARY_POWER) {
1464 PyObject *exp = POP();
1465 PyObject *base = TOP();
1466 PyObject *res = PyNumber_Power(base, exp, Py_None);
1467 Py_DECREF(base);
1468 Py_DECREF(exp);
1469 SET_TOP(res);
1470 if (res == NULL)
1471 goto error;
1472 DISPATCH();
1473 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001474
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001475 TARGET(BINARY_MULTIPLY) {
1476 PyObject *right = POP();
1477 PyObject *left = TOP();
1478 PyObject *res = PyNumber_Multiply(left, right);
1479 Py_DECREF(left);
1480 Py_DECREF(right);
1481 SET_TOP(res);
1482 if (res == NULL)
1483 goto error;
1484 DISPATCH();
1485 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001486
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001487 TARGET(BINARY_TRUE_DIVIDE) {
1488 PyObject *divisor = POP();
1489 PyObject *dividend = TOP();
1490 PyObject *quotient = PyNumber_TrueDivide(dividend, divisor);
1491 Py_DECREF(dividend);
1492 Py_DECREF(divisor);
1493 SET_TOP(quotient);
1494 if (quotient == NULL)
1495 goto error;
1496 DISPATCH();
1497 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001498
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001499 TARGET(BINARY_FLOOR_DIVIDE) {
1500 PyObject *divisor = POP();
1501 PyObject *dividend = TOP();
1502 PyObject *quotient = PyNumber_FloorDivide(dividend, divisor);
1503 Py_DECREF(dividend);
1504 Py_DECREF(divisor);
1505 SET_TOP(quotient);
1506 if (quotient == NULL)
1507 goto error;
1508 DISPATCH();
1509 }
Guido van Rossum4668b002001-08-08 05:00:18 +00001510
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001511 TARGET(BINARY_MODULO) {
1512 PyObject *divisor = POP();
1513 PyObject *dividend = TOP();
1514 PyObject *res = PyUnicode_CheckExact(dividend) ?
1515 PyUnicode_Format(dividend, divisor) :
1516 PyNumber_Remainder(dividend, divisor);
1517 Py_DECREF(divisor);
1518 Py_DECREF(dividend);
1519 SET_TOP(res);
1520 if (res == NULL)
1521 goto error;
1522 DISPATCH();
1523 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001524
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001525 TARGET(BINARY_ADD) {
1526 PyObject *right = POP();
1527 PyObject *left = TOP();
1528 PyObject *sum;
1529 if (PyUnicode_CheckExact(left) &&
1530 PyUnicode_CheckExact(right)) {
1531 sum = unicode_concatenate(left, right, f, next_instr);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001532 /* unicode_concatenate consumed the ref to v */
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001533 }
1534 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001535 sum = PyNumber_Add(left, right);
1536 Py_DECREF(left);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001537 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001538 Py_DECREF(right);
1539 SET_TOP(sum);
1540 if (sum == NULL)
1541 goto error;
1542 DISPATCH();
1543 }
1544
1545 TARGET(BINARY_SUBTRACT) {
1546 PyObject *right = POP();
1547 PyObject *left = TOP();
1548 PyObject *diff = PyNumber_Subtract(left, right);
1549 Py_DECREF(right);
1550 Py_DECREF(left);
1551 SET_TOP(diff);
1552 if (diff == NULL)
1553 goto error;
1554 DISPATCH();
1555 }
1556
1557 TARGET(BINARY_SUBSCR) {
1558 PyObject *sub = POP();
1559 PyObject *container = TOP();
1560 PyObject *res = PyObject_GetItem(container, sub);
1561 Py_DECREF(container);
1562 Py_DECREF(sub);
1563 SET_TOP(res);
1564 if (res == NULL)
1565 goto error;
1566 DISPATCH();
1567 }
1568
1569 TARGET(BINARY_LSHIFT) {
1570 PyObject *right = POP();
1571 PyObject *left = TOP();
1572 PyObject *res = PyNumber_Lshift(left, right);
1573 Py_DECREF(left);
1574 Py_DECREF(right);
1575 SET_TOP(res);
1576 if (res == NULL)
1577 goto error;
1578 DISPATCH();
1579 }
1580
1581 TARGET(BINARY_RSHIFT) {
1582 PyObject *right = POP();
1583 PyObject *left = TOP();
1584 PyObject *res = PyNumber_Rshift(left, right);
1585 Py_DECREF(left);
1586 Py_DECREF(right);
1587 SET_TOP(res);
1588 if (res == NULL)
1589 goto error;
1590 DISPATCH();
1591 }
1592
1593 TARGET(BINARY_AND) {
1594 PyObject *right = POP();
1595 PyObject *left = TOP();
1596 PyObject *res = PyNumber_And(left, right);
1597 Py_DECREF(left);
1598 Py_DECREF(right);
1599 SET_TOP(res);
1600 if (res == NULL)
1601 goto error;
1602 DISPATCH();
1603 }
1604
1605 TARGET(BINARY_XOR) {
1606 PyObject *right = POP();
1607 PyObject *left = TOP();
1608 PyObject *res = PyNumber_Xor(left, right);
1609 Py_DECREF(left);
1610 Py_DECREF(right);
1611 SET_TOP(res);
1612 if (res == NULL)
1613 goto error;
1614 DISPATCH();
1615 }
1616
1617 TARGET(BINARY_OR) {
1618 PyObject *right = POP();
1619 PyObject *left = TOP();
1620 PyObject *res = PyNumber_Or(left, right);
1621 Py_DECREF(left);
1622 Py_DECREF(right);
1623 SET_TOP(res);
1624 if (res == NULL)
1625 goto error;
1626 DISPATCH();
1627 }
1628
1629 TARGET(LIST_APPEND) {
1630 PyObject *v = POP();
1631 PyObject *list = PEEK(oparg);
1632 int err;
1633 err = PyList_Append(list, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001634 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001635 if (err != 0)
1636 goto error;
1637 PREDICT(JUMP_ABSOLUTE);
1638 DISPATCH();
1639 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001640
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001641 TARGET(SET_ADD) {
1642 PyObject *v = POP();
1643 PyObject *set = stack_pointer[-oparg];
1644 int err;
1645 err = PySet_Add(set, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001646 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001647 if (err != 0)
1648 goto error;
1649 PREDICT(JUMP_ABSOLUTE);
1650 DISPATCH();
1651 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001652
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001653 TARGET(INPLACE_POWER) {
1654 PyObject *exp = POP();
1655 PyObject *base = TOP();
1656 PyObject *res = PyNumber_InPlacePower(base, exp, Py_None);
1657 Py_DECREF(base);
1658 Py_DECREF(exp);
1659 SET_TOP(res);
1660 if (res == NULL)
1661 goto error;
1662 DISPATCH();
1663 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001664
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001665 TARGET(INPLACE_MULTIPLY) {
1666 PyObject *right = POP();
1667 PyObject *left = TOP();
1668 PyObject *res = PyNumber_InPlaceMultiply(left, right);
1669 Py_DECREF(left);
1670 Py_DECREF(right);
1671 SET_TOP(res);
1672 if (res == NULL)
1673 goto error;
1674 DISPATCH();
1675 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001676
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001677 TARGET(INPLACE_TRUE_DIVIDE) {
1678 PyObject *divisor = POP();
1679 PyObject *dividend = TOP();
1680 PyObject *quotient = PyNumber_InPlaceTrueDivide(dividend, divisor);
1681 Py_DECREF(dividend);
1682 Py_DECREF(divisor);
1683 SET_TOP(quotient);
1684 if (quotient == NULL)
1685 goto error;
1686 DISPATCH();
1687 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001688
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001689 TARGET(INPLACE_FLOOR_DIVIDE) {
1690 PyObject *divisor = POP();
1691 PyObject *dividend = TOP();
1692 PyObject *quotient = PyNumber_InPlaceFloorDivide(dividend, divisor);
1693 Py_DECREF(dividend);
1694 Py_DECREF(divisor);
1695 SET_TOP(quotient);
1696 if (quotient == NULL)
1697 goto error;
1698 DISPATCH();
1699 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001700
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001701 TARGET(INPLACE_MODULO) {
1702 PyObject *right = POP();
1703 PyObject *left = TOP();
1704 PyObject *mod = PyNumber_InPlaceRemainder(left, right);
1705 Py_DECREF(left);
1706 Py_DECREF(right);
1707 SET_TOP(mod);
1708 if (mod == NULL)
1709 goto error;
1710 DISPATCH();
1711 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001712
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001713 TARGET(INPLACE_ADD) {
1714 PyObject *right = POP();
1715 PyObject *left = TOP();
1716 PyObject *sum;
1717 if (PyUnicode_CheckExact(left) && PyUnicode_CheckExact(right)) {
1718 sum = unicode_concatenate(left, right, f, next_instr);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001719 /* unicode_concatenate consumed the ref to v */
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001720 }
1721 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001722 sum = PyNumber_InPlaceAdd(left, right);
1723 Py_DECREF(left);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001724 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001725 Py_DECREF(right);
1726 SET_TOP(sum);
1727 if (sum == NULL)
1728 goto error;
1729 DISPATCH();
1730 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001731
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001732 TARGET(INPLACE_SUBTRACT) {
1733 PyObject *right = POP();
1734 PyObject *left = TOP();
1735 PyObject *diff = PyNumber_InPlaceSubtract(left, right);
1736 Py_DECREF(left);
1737 Py_DECREF(right);
1738 SET_TOP(diff);
1739 if (diff == NULL)
1740 goto error;
1741 DISPATCH();
1742 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001743
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001744 TARGET(INPLACE_LSHIFT) {
1745 PyObject *right = POP();
1746 PyObject *left = TOP();
1747 PyObject *res = PyNumber_InPlaceLshift(left, right);
1748 Py_DECREF(left);
1749 Py_DECREF(right);
1750 SET_TOP(res);
1751 if (res == NULL)
1752 goto error;
1753 DISPATCH();
1754 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001755
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001756 TARGET(INPLACE_RSHIFT) {
1757 PyObject *right = POP();
1758 PyObject *left = TOP();
1759 PyObject *res = PyNumber_InPlaceRshift(left, right);
1760 Py_DECREF(left);
1761 Py_DECREF(right);
1762 SET_TOP(res);
1763 if (res == NULL)
1764 goto error;
1765 DISPATCH();
1766 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001767
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001768 TARGET(INPLACE_AND) {
1769 PyObject *right = POP();
1770 PyObject *left = TOP();
1771 PyObject *res = PyNumber_InPlaceAnd(left, right);
1772 Py_DECREF(left);
1773 Py_DECREF(right);
1774 SET_TOP(res);
1775 if (res == NULL)
1776 goto error;
1777 DISPATCH();
1778 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001779
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001780 TARGET(INPLACE_XOR) {
1781 PyObject *right = POP();
1782 PyObject *left = TOP();
1783 PyObject *res = PyNumber_InPlaceXor(left, right);
1784 Py_DECREF(left);
1785 Py_DECREF(right);
1786 SET_TOP(res);
1787 if (res == NULL)
1788 goto error;
1789 DISPATCH();
1790 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001791
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001792 TARGET(INPLACE_OR) {
1793 PyObject *right = POP();
1794 PyObject *left = TOP();
1795 PyObject *res = PyNumber_InPlaceOr(left, right);
1796 Py_DECREF(left);
1797 Py_DECREF(right);
1798 SET_TOP(res);
1799 if (res == NULL)
1800 goto error;
1801 DISPATCH();
1802 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001803
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001804 TARGET(STORE_SUBSCR) {
1805 PyObject *sub = TOP();
1806 PyObject *container = SECOND();
1807 PyObject *v = THIRD();
1808 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001809 STACKADJ(-3);
1810 /* v[w] = u */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001811 err = PyObject_SetItem(container, sub, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001812 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001813 Py_DECREF(container);
1814 Py_DECREF(sub);
1815 if (err != 0)
1816 goto error;
1817 DISPATCH();
1818 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001819
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001820 TARGET(DELETE_SUBSCR) {
1821 PyObject *sub = TOP();
1822 PyObject *container = SECOND();
1823 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001824 STACKADJ(-2);
1825 /* del v[w] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001826 err = PyObject_DelItem(container, sub);
1827 Py_DECREF(container);
1828 Py_DECREF(sub);
1829 if (err != 0)
1830 goto error;
1831 DISPATCH();
1832 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001833
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001834 TARGET(PRINT_EXPR) {
1835 PyObject *value = POP();
1836 PyObject *hook = PySys_GetObject("displayhook");
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04001837 PyObject *res;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001838 if (hook == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001839 PyErr_SetString(PyExc_RuntimeError,
1840 "lost sys.displayhook");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001841 Py_DECREF(value);
1842 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001843 }
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04001844 res = PyObject_CallFunctionObjArgs(hook, value, NULL);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001845 Py_DECREF(value);
1846 if (res == NULL)
1847 goto error;
1848 Py_DECREF(res);
1849 DISPATCH();
1850 }
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001851
Thomas Wouters434d0822000-08-24 20:11:32 +00001852#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001853 default: switch (opcode) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001854#endif
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001855 TARGET(RAISE_VARARGS) {
1856 PyObject *cause = NULL, *exc = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001857 switch (oparg) {
1858 case 2:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001859 cause = POP(); /* cause */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001860 case 1:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001861 exc = POP(); /* exc */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001862 case 0: /* Fallthrough */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001863 if (do_raise(exc, cause)) {
1864 why = WHY_EXCEPTION;
1865 goto fast_block_end;
1866 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001867 break;
1868 default:
1869 PyErr_SetString(PyExc_SystemError,
1870 "bad RAISE_VARARGS oparg");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001871 break;
1872 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001873 goto error;
1874 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001875
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001876 TARGET(RETURN_VALUE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 retval = POP();
1878 why = WHY_RETURN;
1879 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001880 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001881
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001882 TARGET(YIELD_FROM) {
1883 PyObject *v = POP();
1884 PyObject *reciever = TOP();
1885 int err;
1886 if (PyGen_CheckExact(reciever)) {
1887 retval = _PyGen_Send((PyGenObject *)reciever, v);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001888 } else {
Benjamin Peterson302e7902012-03-20 23:17:04 -04001889 _Py_IDENTIFIER(send);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001890 if (v == Py_None)
1891 retval = Py_TYPE(reciever)->tp_iternext(reciever);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001892 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001893 retval = _PyObject_CallMethodId(reciever, &PyId_send, "O", v);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001894 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001895 Py_DECREF(v);
1896 if (retval == NULL) {
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001897 PyObject *val;
Nick Coghlanc40bc092012-06-17 15:15:49 +10001898 err = _PyGen_FetchStopIterationValue(&val);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001899 if (err < 0)
1900 goto error;
1901 Py_DECREF(reciever);
1902 SET_TOP(val);
1903 DISPATCH();
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001904 }
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001905 /* x remains on stack, retval is value to be yielded */
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001906 f->f_stacktop = stack_pointer;
1907 why = WHY_YIELD;
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001908 /* and repeat... */
1909 f->f_lasti--;
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001910 goto fast_yield;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001911 }
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001912
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001913 TARGET(YIELD_VALUE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001914 retval = POP();
1915 f->f_stacktop = stack_pointer;
1916 why = WHY_YIELD;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001917 goto fast_yield;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001918 }
Tim Peters5ca576e2001-06-18 22:08:13 +00001919
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001920 TARGET(POP_EXCEPT) {
1921 PyTryBlock *b = PyFrame_BlockPop(f);
1922 if (b->b_type != EXCEPT_HANDLER) {
1923 PyErr_SetString(PyExc_SystemError,
1924 "popped block is not an except handler");
1925 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001926 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001927 UNWIND_EXCEPT_HANDLER(b);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001928 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001929 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001930
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001931 TARGET(POP_BLOCK) {
1932 PyTryBlock *b = PyFrame_BlockPop(f);
1933 UNWIND_BLOCK(b);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001934 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001935 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001936
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001937 PREDICTED(END_FINALLY);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001938 TARGET(END_FINALLY) {
1939 PyObject *status = POP();
1940 if (PyLong_Check(status)) {
1941 why = (enum why_code) PyLong_AS_LONG(status);
1942 assert(why != WHY_YIELD && why != WHY_EXCEPTION);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001943 if (why == WHY_RETURN ||
1944 why == WHY_CONTINUE)
1945 retval = POP();
1946 if (why == WHY_SILENCED) {
1947 /* An exception was silenced by 'with', we must
1948 manually unwind the EXCEPT_HANDLER block which was
1949 created when the exception was caught, otherwise
1950 the stack will be in an inconsistent state. */
1951 PyTryBlock *b = PyFrame_BlockPop(f);
1952 assert(b->b_type == EXCEPT_HANDLER);
1953 UNWIND_EXCEPT_HANDLER(b);
1954 why = WHY_NOT;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001955 Py_DECREF(status);
1956 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001957 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001958 Py_DECREF(status);
1959 goto fast_block_end;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001960 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001961 else if (PyExceptionClass_Check(status)) {
1962 PyObject *exc = POP();
1963 PyObject *tb = POP();
1964 PyErr_Restore(status, exc, tb);
1965 why = WHY_EXCEPTION;
1966 goto fast_block_end;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001967 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001968 else if (status != Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001969 PyErr_SetString(PyExc_SystemError,
1970 "'finally' pops bad exception");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001971 Py_DECREF(status);
1972 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001973 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001974 Py_DECREF(status);
1975 DISPATCH();
1976 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001977
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001978 TARGET(LOAD_BUILD_CLASS) {
Victor Stinner3c1e4812012-03-26 22:10:51 +02001979 _Py_IDENTIFIER(__build_class__);
Victor Stinnerb0b22422012-04-19 00:57:45 +02001980
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001981 PyObject *bc;
Victor Stinnerb0b22422012-04-19 00:57:45 +02001982 if (PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001983 bc = _PyDict_GetItemId(f->f_builtins, &PyId___build_class__);
1984 if (bc == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02001985 PyErr_SetString(PyExc_NameError,
1986 "__build_class__ not found");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001987 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02001988 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001989 Py_INCREF(bc);
Victor Stinnerb0b22422012-04-19 00:57:45 +02001990 }
1991 else {
1992 PyObject *build_class_str = _PyUnicode_FromId(&PyId___build_class__);
1993 if (build_class_str == NULL)
1994 break;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001995 bc = PyObject_GetItem(f->f_builtins, build_class_str);
1996 if (bc == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02001997 if (PyErr_ExceptionMatches(PyExc_KeyError))
1998 PyErr_SetString(PyExc_NameError,
1999 "__build_class__ not found");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002000 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002001 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002002 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002003 PUSH(bc);
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002004 DISPATCH();
Victor Stinner3c1e4812012-03-26 22:10:51 +02002005 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002006
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002007 TARGET(STORE_NAME) {
2008 PyObject *name = GETITEM(names, oparg);
2009 PyObject *v = POP();
2010 PyObject *ns = f->f_locals;
2011 int err;
2012 if (ns == NULL) {
2013 PyErr_Format(PyExc_SystemError,
2014 "no locals found when storing %R", name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002015 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002016 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002017 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002018 if (PyDict_CheckExact(ns))
2019 err = PyDict_SetItem(ns, name, v);
2020 else
2021 err = PyObject_SetItem(ns, name, v);
2022 Py_DECREF(v);
2023 if (err != 0)
2024 goto error;
2025 DISPATCH();
2026 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002027
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002028 TARGET(DELETE_NAME) {
2029 PyObject *name = GETITEM(names, oparg);
2030 PyObject *ns = f->f_locals;
2031 int err;
2032 if (ns == NULL) {
2033 PyErr_Format(PyExc_SystemError,
2034 "no locals when deleting %R", name);
2035 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002036 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002037 err = PyObject_DelItem(ns, name);
2038 if (err != 0) {
2039 format_exc_check_arg(PyExc_NameError,
2040 NAME_ERROR_MSG,
2041 name);
2042 goto error;
2043 }
2044 DISPATCH();
2045 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00002046
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002047 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002048 TARGET(UNPACK_SEQUENCE) {
2049 PyObject *seq = POP(), *item, **items;
2050 if (PyTuple_CheckExact(seq) &&
2051 PyTuple_GET_SIZE(seq) == oparg) {
2052 items = ((PyTupleObject *)seq)->ob_item;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002053 while (oparg--) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002054 item = items[oparg];
2055 Py_INCREF(item);
2056 PUSH(item);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002058 } else if (PyList_CheckExact(seq) &&
2059 PyList_GET_SIZE(seq) == oparg) {
2060 items = ((PyListObject *)seq)->ob_item;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002061 while (oparg--) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002062 item = items[oparg];
2063 Py_INCREF(item);
2064 PUSH(item);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002065 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002066 } else if (unpack_iterable(seq, oparg, -1,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002067 stack_pointer + oparg)) {
2068 STACKADJ(oparg);
2069 } else {
2070 /* unpack_iterable() raised an exception */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002071 Py_DECREF(seq);
2072 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002073 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002074 Py_DECREF(seq);
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002075 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002076 }
Guido van Rossum0368b722007-05-11 16:50:42 +00002077
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002078 TARGET(UNPACK_EX) {
2079 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
2080 PyObject *seq = POP();
2081
2082 if (unpack_iterable(seq, oparg & 0xFF, oparg >> 8,
2083 stack_pointer + totalargs)) {
2084 stack_pointer += totalargs;
2085 } else {
2086 Py_DECREF(seq);
2087 goto error;
2088 }
2089 Py_DECREF(seq);
2090 DISPATCH();
2091 }
2092
2093 TARGET(STORE_ATTR) {
2094 PyObject *name = GETITEM(names, oparg);
2095 PyObject *owner = TOP();
2096 PyObject *v = SECOND();
2097 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002098 STACKADJ(-2);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002099 err = PyObject_SetAttr(owner, name, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002100 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002101 Py_DECREF(owner);
2102 if (err != 0)
2103 goto error;
2104 DISPATCH();
2105 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002106
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002107 TARGET(DELETE_ATTR) {
2108 PyObject *name = GETITEM(names, oparg);
2109 PyObject *owner = POP();
2110 int err;
2111 err = PyObject_SetAttr(owner, name, (PyObject *)NULL);
2112 Py_DECREF(owner);
2113 if (err != 0)
2114 goto error;
2115 DISPATCH();
2116 }
2117
2118 TARGET(STORE_GLOBAL) {
2119 PyObject *name = GETITEM(names, oparg);
2120 PyObject *v = POP();
2121 int err;
2122 err = PyDict_SetItem(f->f_globals, name, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002123 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002124 if (err != 0)
2125 goto error;
2126 DISPATCH();
2127 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002128
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002129 TARGET(DELETE_GLOBAL) {
2130 PyObject *name = GETITEM(names, oparg);
2131 int err;
2132 err = PyDict_DelItem(f->f_globals, name);
2133 if (err != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002134 format_exc_check_arg(
Ezio Melotti04a29552013-03-03 15:12:44 +02002135 PyExc_NameError, NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002136 goto error;
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002137 }
2138 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002139 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002140
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002141 TARGET(LOAD_NAME) {
2142 PyObject *name = GETITEM(names, oparg);
2143 PyObject *locals = f->f_locals;
2144 PyObject *v;
2145 if (locals == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002146 PyErr_Format(PyExc_SystemError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002147 "no locals when loading %R", name);
2148 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002149 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002150 if (PyDict_CheckExact(locals)) {
2151 v = PyDict_GetItem(locals, name);
2152 Py_XINCREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002153 }
2154 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002155 v = PyObject_GetItem(locals, name);
2156 if (v == NULL && PyErr_Occurred()) {
Benjamin Peterson92722792012-12-15 12:51:05 -05002157 if (!PyErr_ExceptionMatches(PyExc_KeyError))
2158 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002159 PyErr_Clear();
2160 }
2161 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002162 if (v == NULL) {
2163 v = PyDict_GetItem(f->f_globals, name);
2164 Py_XINCREF(v);
2165 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002166 if (PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002167 v = PyDict_GetItem(f->f_builtins, name);
2168 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002169 format_exc_check_arg(
2170 PyExc_NameError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002171 NAME_ERROR_MSG, name);
2172 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002173 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002174 Py_INCREF(v);
Victor Stinnerb0b22422012-04-19 00:57:45 +02002175 }
2176 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002177 v = PyObject_GetItem(f->f_builtins, name);
2178 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002179 if (PyErr_ExceptionMatches(PyExc_KeyError))
2180 format_exc_check_arg(
2181 PyExc_NameError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002182 NAME_ERROR_MSG, name);
2183 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002184 }
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002185 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002186 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002187 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002188 PUSH(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002189 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002190 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002191
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002192 TARGET(LOAD_GLOBAL) {
2193 PyObject *name = GETITEM(names, oparg);
2194 PyObject *v;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002195 if (PyDict_CheckExact(f->f_globals)
2196 && PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002197 v = _PyDict_LoadGlobal((PyDictObject *)f->f_globals,
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002198 (PyDictObject *)f->f_builtins,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002199 name);
2200 if (v == NULL) {
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002201 if (!PyErr_Occurred())
2202 format_exc_check_arg(PyExc_NameError,
Ezio Melotti04a29552013-03-03 15:12:44 +02002203 NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002204 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002205 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002206 Py_INCREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002207 }
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002208 else {
2209 /* Slow-path if globals or builtins is not a dict */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002210 v = PyObject_GetItem(f->f_globals, name);
2211 if (v == NULL) {
2212 v = PyObject_GetItem(f->f_builtins, name);
2213 if (v == NULL) {
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002214 if (PyErr_ExceptionMatches(PyExc_KeyError))
2215 format_exc_check_arg(
2216 PyExc_NameError,
Ezio Melotti04a29552013-03-03 15:12:44 +02002217 NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002218 goto error;
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002219 }
2220 }
2221 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002222 PUSH(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002223 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002224 }
Guido van Rossum681d79a1995-07-18 14:51:37 +00002225
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002226 TARGET(DELETE_FAST) {
2227 PyObject *v = GETLOCAL(oparg);
2228 if (v != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002229 SETLOCAL(oparg, NULL);
2230 DISPATCH();
2231 }
2232 format_exc_check_arg(
2233 PyExc_UnboundLocalError,
2234 UNBOUNDLOCAL_ERROR_MSG,
2235 PyTuple_GetItem(co->co_varnames, oparg)
2236 );
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002237 goto error;
2238 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002239
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002240 TARGET(DELETE_DEREF) {
2241 PyObject *cell = freevars[oparg];
2242 if (PyCell_GET(cell) != NULL) {
2243 PyCell_Set(cell, NULL);
Benjamin Peterson00ebe2c2010-09-10 22:02:31 +00002244 DISPATCH();
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002245 }
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002246 format_exc_unbound(co, oparg);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002247 goto error;
2248 }
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002249
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002250 TARGET(LOAD_CLOSURE) {
2251 PyObject *cell = freevars[oparg];
2252 Py_INCREF(cell);
2253 PUSH(cell);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002254 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002255 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002256
Benjamin Peterson3b0431d2013-04-30 09:41:40 -04002257 TARGET(LOAD_CLASSDEREF) {
2258 PyObject *name, *value, *locals = f->f_locals;
Victor Stinnerd3dfd0e2013-05-16 23:48:01 +02002259 Py_ssize_t idx;
Benjamin Peterson3b0431d2013-04-30 09:41:40 -04002260 assert(locals);
2261 assert(oparg >= PyTuple_GET_SIZE(co->co_cellvars));
2262 idx = oparg - PyTuple_GET_SIZE(co->co_cellvars);
2263 assert(idx >= 0 && idx < PyTuple_GET_SIZE(co->co_freevars));
2264 name = PyTuple_GET_ITEM(co->co_freevars, idx);
2265 if (PyDict_CheckExact(locals)) {
2266 value = PyDict_GetItem(locals, name);
2267 Py_XINCREF(value);
2268 }
2269 else {
2270 value = PyObject_GetItem(locals, name);
2271 if (value == NULL && PyErr_Occurred()) {
2272 if (!PyErr_ExceptionMatches(PyExc_KeyError))
2273 goto error;
2274 PyErr_Clear();
2275 }
2276 }
2277 if (!value) {
2278 PyObject *cell = freevars[oparg];
2279 value = PyCell_GET(cell);
2280 if (value == NULL) {
2281 format_exc_unbound(co, oparg);
2282 goto error;
2283 }
2284 Py_INCREF(value);
2285 }
2286 PUSH(value);
2287 DISPATCH();
2288 }
2289
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002290 TARGET(LOAD_DEREF) {
2291 PyObject *cell = freevars[oparg];
2292 PyObject *value = PyCell_GET(cell);
2293 if (value == NULL) {
2294 format_exc_unbound(co, oparg);
2295 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002296 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002297 Py_INCREF(value);
2298 PUSH(value);
2299 DISPATCH();
2300 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002301
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002302 TARGET(STORE_DEREF) {
2303 PyObject *v = POP();
2304 PyObject *cell = freevars[oparg];
2305 PyCell_Set(cell, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002306 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002307 DISPATCH();
2308 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002309
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002310 TARGET(BUILD_TUPLE) {
2311 PyObject *tup = PyTuple_New(oparg);
2312 if (tup == NULL)
2313 goto error;
2314 while (--oparg >= 0) {
2315 PyObject *item = POP();
2316 PyTuple_SET_ITEM(tup, oparg, item);
2317 }
2318 PUSH(tup);
2319 DISPATCH();
2320 }
2321
2322 TARGET(BUILD_LIST) {
2323 PyObject *list = PyList_New(oparg);
2324 if (list == NULL)
2325 goto error;
2326 while (--oparg >= 0) {
2327 PyObject *item = POP();
2328 PyList_SET_ITEM(list, oparg, item);
2329 }
2330 PUSH(list);
2331 DISPATCH();
2332 }
2333
2334 TARGET(BUILD_SET) {
2335 PyObject *set = PySet_New(NULL);
2336 int err = 0;
2337 if (set == NULL)
2338 goto error;
2339 while (--oparg >= 0) {
2340 PyObject *item = POP();
2341 if (err == 0)
2342 err = PySet_Add(set, item);
2343 Py_DECREF(item);
2344 }
2345 if (err != 0) {
2346 Py_DECREF(set);
2347 goto error;
2348 }
2349 PUSH(set);
2350 DISPATCH();
2351 }
2352
2353 TARGET(BUILD_MAP) {
2354 PyObject *map = _PyDict_NewPresized((Py_ssize_t)oparg);
2355 if (map == NULL)
2356 goto error;
2357 PUSH(map);
2358 DISPATCH();
2359 }
2360
2361 TARGET(STORE_MAP) {
2362 PyObject *key = TOP();
2363 PyObject *value = SECOND();
2364 PyObject *map = THIRD();
2365 int err;
2366 STACKADJ(-2);
2367 assert(PyDict_CheckExact(map));
2368 err = PyDict_SetItem(map, key, value);
2369 Py_DECREF(value);
2370 Py_DECREF(key);
2371 if (err != 0)
2372 goto error;
2373 DISPATCH();
2374 }
2375
2376 TARGET(MAP_ADD) {
2377 PyObject *key = TOP();
2378 PyObject *value = SECOND();
2379 PyObject *map;
2380 int err;
2381 STACKADJ(-2);
2382 map = stack_pointer[-oparg]; /* dict */
2383 assert(PyDict_CheckExact(map));
2384 err = PyDict_SetItem(map, key, value); /* v[w] = u */
2385 Py_DECREF(value);
2386 Py_DECREF(key);
2387 if (err != 0)
2388 goto error;
2389 PREDICT(JUMP_ABSOLUTE);
2390 DISPATCH();
2391 }
2392
2393 TARGET(LOAD_ATTR) {
2394 PyObject *name = GETITEM(names, oparg);
2395 PyObject *owner = TOP();
2396 PyObject *res = PyObject_GetAttr(owner, name);
2397 Py_DECREF(owner);
2398 SET_TOP(res);
2399 if (res == NULL)
2400 goto error;
2401 DISPATCH();
2402 }
2403
2404 TARGET(COMPARE_OP) {
2405 PyObject *right = POP();
2406 PyObject *left = TOP();
2407 PyObject *res = cmp_outcome(oparg, left, right);
2408 Py_DECREF(left);
2409 Py_DECREF(right);
2410 SET_TOP(res);
2411 if (res == NULL)
2412 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002413 PREDICT(POP_JUMP_IF_FALSE);
2414 PREDICT(POP_JUMP_IF_TRUE);
2415 DISPATCH();
Victor Stinner3c1e4812012-03-26 22:10:51 +02002416 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002417
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002418 TARGET(IMPORT_NAME) {
2419 _Py_IDENTIFIER(__import__);
2420 PyObject *name = GETITEM(names, oparg);
2421 PyObject *func = _PyDict_GetItemId(f->f_builtins, &PyId___import__);
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04002422 PyObject *from, *level, *args, *res;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002423 if (func == NULL) {
2424 PyErr_SetString(PyExc_ImportError,
2425 "__import__ not found");
2426 goto error;
2427 }
2428 Py_INCREF(func);
2429 from = POP();
2430 level = TOP();
2431 if (PyLong_AsLong(level) != -1 || PyErr_Occurred())
2432 args = PyTuple_Pack(5,
2433 name,
2434 f->f_globals,
2435 f->f_locals == NULL ?
2436 Py_None : f->f_locals,
2437 from,
2438 level);
2439 else
2440 args = PyTuple_Pack(4,
2441 name,
2442 f->f_globals,
2443 f->f_locals == NULL ?
2444 Py_None : f->f_locals,
2445 from);
2446 Py_DECREF(level);
2447 Py_DECREF(from);
2448 if (args == NULL) {
2449 Py_DECREF(func);
2450 STACKADJ(-1);
2451 goto error;
2452 }
2453 READ_TIMESTAMP(intr0);
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04002454 res = PyEval_CallObject(func, args);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002455 READ_TIMESTAMP(intr1);
2456 Py_DECREF(args);
2457 Py_DECREF(func);
2458 SET_TOP(res);
2459 if (res == NULL)
2460 goto error;
2461 DISPATCH();
2462 }
2463
2464 TARGET(IMPORT_STAR) {
2465 PyObject *from = POP(), *locals;
2466 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002467 PyFrame_FastToLocals(f);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002468 locals = f->f_locals;
2469 if (locals == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002470 PyErr_SetString(PyExc_SystemError,
2471 "no locals found during 'import *'");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002472 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002473 }
2474 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002475 err = import_all_from(locals, from);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002476 READ_TIMESTAMP(intr1);
2477 PyFrame_LocalsToFast(f, 0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002478 Py_DECREF(from);
2479 if (err != 0)
2480 goto error;
2481 DISPATCH();
2482 }
Guido van Rossum25831651993-05-19 14:50:45 +00002483
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002484 TARGET(IMPORT_FROM) {
2485 PyObject *name = GETITEM(names, oparg);
2486 PyObject *from = TOP();
2487 PyObject *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002488 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002489 res = import_from(from, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002490 READ_TIMESTAMP(intr1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002491 PUSH(res);
2492 if (res == NULL)
2493 goto error;
2494 DISPATCH();
2495 }
Thomas Wouters52152252000-08-17 22:55:00 +00002496
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002497 TARGET(JUMP_FORWARD) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002498 JUMPBY(oparg);
2499 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002500 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002502 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002503 TARGET(POP_JUMP_IF_FALSE) {
2504 PyObject *cond = POP();
2505 int err;
2506 if (cond == Py_True) {
2507 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002508 FAST_DISPATCH();
2509 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002510 if (cond == Py_False) {
2511 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002512 JUMPTO(oparg);
2513 FAST_DISPATCH();
2514 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002515 err = PyObject_IsTrue(cond);
2516 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002517 if (err > 0)
2518 err = 0;
2519 else if (err == 0)
2520 JUMPTO(oparg);
2521 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002522 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002524 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002527 TARGET(POP_JUMP_IF_TRUE) {
2528 PyObject *cond = POP();
2529 int err;
2530 if (cond == Py_False) {
2531 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002532 FAST_DISPATCH();
2533 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002534 if (cond == Py_True) {
2535 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002536 JUMPTO(oparg);
2537 FAST_DISPATCH();
2538 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002539 err = PyObject_IsTrue(cond);
2540 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 if (err > 0) {
2542 err = 0;
2543 JUMPTO(oparg);
2544 }
2545 else if (err == 0)
2546 ;
2547 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002548 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002549 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002550 }
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002551
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002552 TARGET(JUMP_IF_FALSE_OR_POP) {
2553 PyObject *cond = TOP();
2554 int err;
2555 if (cond == Py_True) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002556 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002557 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002558 FAST_DISPATCH();
2559 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002560 if (cond == Py_False) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002561 JUMPTO(oparg);
2562 FAST_DISPATCH();
2563 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002564 err = PyObject_IsTrue(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002565 if (err > 0) {
2566 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002567 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002568 err = 0;
2569 }
2570 else if (err == 0)
2571 JUMPTO(oparg);
2572 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002573 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002574 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002575 }
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002576
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002577 TARGET(JUMP_IF_TRUE_OR_POP) {
2578 PyObject *cond = TOP();
2579 int err;
2580 if (cond == Py_False) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002581 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002582 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002583 FAST_DISPATCH();
2584 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002585 if (cond == Py_True) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002586 JUMPTO(oparg);
2587 FAST_DISPATCH();
2588 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002589 err = PyObject_IsTrue(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002590 if (err > 0) {
2591 err = 0;
2592 JUMPTO(oparg);
2593 }
2594 else if (err == 0) {
2595 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002596 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002597 }
2598 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002599 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002600 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002601 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002602
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002603 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002604 TARGET(JUMP_ABSOLUTE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002605 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002606#if FAST_LOOPS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002607 /* Enabling this path speeds-up all while and for-loops by bypassing
2608 the per-loop checks for signals. By default, this should be turned-off
2609 because it prevents detection of a control-break in tight loops like
2610 "while 1: pass". Compile with this option turned-on when you need
2611 the speed-up and do not need break checking inside tight loops (ones
2612 that contain only instructions ending with FAST_DISPATCH).
2613 */
2614 FAST_DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002615#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002616 DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002617#endif
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002618 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002619
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002620 TARGET(GET_ITER) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002621 /* before: [obj]; after [getiter(obj)] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002622 PyObject *iterable = TOP();
2623 PyObject *iter = PyObject_GetIter(iterable);
2624 Py_DECREF(iterable);
2625 SET_TOP(iter);
2626 if (iter == NULL)
2627 goto error;
2628 PREDICT(FOR_ITER);
2629 DISPATCH();
2630 }
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002631
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002632 PREDICTED_WITH_ARG(FOR_ITER);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002633 TARGET(FOR_ITER) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002634 /* before: [iter]; after: [iter, iter()] *or* [] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002635 PyObject *iter = TOP();
2636 PyObject *next = (*iter->ob_type->tp_iternext)(iter);
2637 if (next != NULL) {
2638 PUSH(next);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002639 PREDICT(STORE_FAST);
2640 PREDICT(UNPACK_SEQUENCE);
2641 DISPATCH();
2642 }
2643 if (PyErr_Occurred()) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002644 if (!PyErr_ExceptionMatches(PyExc_StopIteration))
2645 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002646 PyErr_Clear();
2647 }
2648 /* iterator ended normally */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002649 STACKADJ(-1);
2650 Py_DECREF(iter);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002651 JUMPBY(oparg);
2652 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002653 }
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002654
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002655 TARGET(BREAK_LOOP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002656 why = WHY_BREAK;
2657 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002658 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002659
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002660 TARGET(CONTINUE_LOOP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002661 retval = PyLong_FromLong(oparg);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002662 if (retval == NULL)
2663 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002664 why = WHY_CONTINUE;
2665 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002666 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002667
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002668 TARGET_WITH_IMPL(SETUP_LOOP, _setup_finally)
2669 TARGET_WITH_IMPL(SETUP_EXCEPT, _setup_finally)
2670 TARGET(SETUP_FINALLY)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002671 _setup_finally: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002672 /* NOTE: If you add any new block-setup opcodes that
2673 are not try/except/finally handlers, you may need
2674 to update the PyGen_NeedsFinalizing() function.
2675 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002677 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
2678 STACK_LEVEL());
2679 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002680 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002681
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002682 TARGET(SETUP_WITH) {
Benjamin Petersonce798522012-01-22 11:24:29 -05002683 _Py_IDENTIFIER(__exit__);
2684 _Py_IDENTIFIER(__enter__);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002685 PyObject *mgr = TOP();
2686 PyObject *exit = special_lookup(mgr, &PyId___exit__), *enter;
2687 PyObject *res;
2688 if (exit == NULL)
2689 goto error;
2690 SET_TOP(exit);
2691 enter = special_lookup(mgr, &PyId___enter__);
2692 Py_DECREF(mgr);
2693 if (enter == NULL)
2694 goto error;
2695 res = PyObject_CallFunctionObjArgs(enter, NULL);
2696 Py_DECREF(enter);
2697 if (res == NULL)
2698 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 /* Setup the finally block before pushing the result
2700 of __enter__ on the stack. */
2701 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2702 STACK_LEVEL());
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002703
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002704 PUSH(res);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002705 DISPATCH();
2706 }
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002707
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002708 TARGET(WITH_CLEANUP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002709 /* At the top of the stack are 1-3 values indicating
2710 how/why we entered the finally clause:
2711 - TOP = None
2712 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2713 - TOP = WHY_*; no retval below it
2714 - (TOP, SECOND, THIRD) = exc_info()
2715 (FOURTH, FITH, SIXTH) = previous exception for EXCEPT_HANDLER
2716 Below them is EXIT, the context.__exit__ bound method.
2717 In the last case, we must call
2718 EXIT(TOP, SECOND, THIRD)
2719 otherwise we must call
2720 EXIT(None, None, None)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002721
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002722 In the first two cases, we remove EXIT from the
2723 stack, leaving the rest in the same order. In the
2724 third case, we shift the bottom 3 values of the
2725 stack down, and replace the empty spot with NULL.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002726
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002727 In addition, if the stack represents an exception,
2728 *and* the function call returns a 'true' value, we
2729 push WHY_SILENCED onto the stack. END_FINALLY will
2730 then not re-raise the exception. (But non-local
2731 gotos should still be resumed.)
2732 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002733
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002734 PyObject *exit_func;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002735 PyObject *exc = TOP(), *val = Py_None, *tb = Py_None, *res;
2736 int err;
2737 if (exc == Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002738 (void)POP();
2739 exit_func = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002740 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002741 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002742 else if (PyLong_Check(exc)) {
2743 STACKADJ(-1);
2744 switch (PyLong_AsLong(exc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002745 case WHY_RETURN:
2746 case WHY_CONTINUE:
2747 /* Retval in TOP. */
2748 exit_func = SECOND();
2749 SET_SECOND(TOP());
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002750 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002751 break;
2752 default:
2753 exit_func = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002754 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002755 break;
2756 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002757 exc = Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002758 }
2759 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002760 PyObject *tp2, *exc2, *tb2;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002761 PyTryBlock *block;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002762 val = SECOND();
2763 tb = THIRD();
2764 tp2 = FOURTH();
2765 exc2 = PEEK(5);
2766 tb2 = PEEK(6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002767 exit_func = PEEK(7);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002768 SET_VALUE(7, tb2);
2769 SET_VALUE(6, exc2);
2770 SET_VALUE(5, tp2);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002771 /* UNWIND_EXCEPT_HANDLER will pop this off. */
2772 SET_FOURTH(NULL);
2773 /* We just shifted the stack down, so we have
2774 to tell the except handler block that the
2775 values are lower than it expects. */
2776 block = &f->f_blockstack[f->f_iblock - 1];
2777 assert(block->b_type == EXCEPT_HANDLER);
2778 block->b_level--;
2779 }
2780 /* XXX Not the fastest way to call it... */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002781 res = PyObject_CallFunctionObjArgs(exit_func, exc, val, tb, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002782 Py_DECREF(exit_func);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002783 if (res == NULL)
2784 goto error;
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002785
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002786 if (exc != Py_None)
2787 err = PyObject_IsTrue(res);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002788 else
2789 err = 0;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002790 Py_DECREF(res);
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002792 if (err < 0)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002793 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002794 else if (err > 0) {
2795 err = 0;
2796 /* There was an exception and a True return */
2797 PUSH(PyLong_FromLong((long) WHY_SILENCED));
2798 }
2799 PREDICT(END_FINALLY);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002800 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002801 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002802
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002803 TARGET(CALL_FUNCTION) {
2804 PyObject **sp, *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002805 PCALL(PCALL_ALL);
2806 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002807#ifdef WITH_TSC
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002808 res = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002809#else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002810 res = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002811#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002812 stack_pointer = sp;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002813 PUSH(res);
2814 if (res == NULL)
2815 goto error;
2816 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002817 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002819 TARGET_WITH_IMPL(CALL_FUNCTION_VAR, _call_function_var_kw)
2820 TARGET_WITH_IMPL(CALL_FUNCTION_KW, _call_function_var_kw)
2821 TARGET(CALL_FUNCTION_VAR_KW)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002822 _call_function_var_kw: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002823 int na = oparg & 0xff;
2824 int nk = (oparg>>8) & 0xff;
2825 int flags = (opcode - CALL_FUNCTION) & 3;
2826 int n = na + 2 * nk;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002827 PyObject **pfunc, *func, **sp, *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002828 PCALL(PCALL_ALL);
2829 if (flags & CALL_FLAG_VAR)
2830 n++;
2831 if (flags & CALL_FLAG_KW)
2832 n++;
2833 pfunc = stack_pointer - n - 1;
2834 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002835
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002836 if (PyMethod_Check(func)
Stefan Krahb7e10102010-06-23 18:42:39 +00002837 && PyMethod_GET_SELF(func) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002838 PyObject *self = PyMethod_GET_SELF(func);
2839 Py_INCREF(self);
2840 func = PyMethod_GET_FUNCTION(func);
2841 Py_INCREF(func);
2842 Py_DECREF(*pfunc);
2843 *pfunc = self;
2844 na++;
Brett Cannonb94767f2011-02-22 20:15:44 +00002845 /* n++; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002846 } else
2847 Py_INCREF(func);
2848 sp = stack_pointer;
2849 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002850 res = ext_do_call(func, &sp, flags, na, nk);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002851 READ_TIMESTAMP(intr1);
2852 stack_pointer = sp;
2853 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002854
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002855 while (stack_pointer > pfunc) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002856 PyObject *o = POP();
2857 Py_DECREF(o);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002858 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002859 PUSH(res);
2860 if (res == NULL)
2861 goto error;
2862 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002863 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002864
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002865 TARGET_WITH_IMPL(MAKE_CLOSURE, _make_function)
2866 TARGET(MAKE_FUNCTION)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002867 _make_function: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002868 int posdefaults = oparg & 0xff;
2869 int kwdefaults = (oparg>>8) & 0xff;
2870 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002871
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002872 PyObject *qualname = POP(); /* qualname */
2873 PyObject *code = POP(); /* code object */
2874 PyObject *func = PyFunction_NewWithQualName(code, f->f_globals, qualname);
2875 Py_DECREF(code);
2876 Py_DECREF(qualname);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002877
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002878 if (func == NULL)
2879 goto error;
2880
2881 if (opcode == MAKE_CLOSURE) {
2882 PyObject *closure = POP();
2883 if (PyFunction_SetClosure(func, closure) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002884 /* Can't happen unless bytecode is corrupt. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002885 Py_DECREF(func);
2886 Py_DECREF(closure);
2887 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002888 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002889 Py_DECREF(closure);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002890 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002891
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002892 if (num_annotations > 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002893 Py_ssize_t name_ix;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002894 PyObject *names = POP(); /* names of args with annotations */
2895 PyObject *anns = PyDict_New();
2896 if (anns == NULL) {
2897 Py_DECREF(func);
2898 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002899 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002900 name_ix = PyTuple_Size(names);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002901 assert(num_annotations == name_ix+1);
2902 while (name_ix > 0) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002903 PyObject *name, *value;
2904 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002905 --name_ix;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002906 name = PyTuple_GET_ITEM(names, name_ix);
2907 value = POP();
2908 err = PyDict_SetItem(anns, name, value);
2909 Py_DECREF(value);
2910 if (err != 0) {
2911 Py_DECREF(anns);
2912 Py_DECREF(func);
2913 goto error;
2914 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002915 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002916
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002917 if (PyFunction_SetAnnotations(func, anns) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002918 /* Can't happen unless
2919 PyFunction_SetAnnotations changes. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002920 Py_DECREF(anns);
2921 Py_DECREF(func);
2922 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002923 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002924 Py_DECREF(anns);
2925 Py_DECREF(names);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002926 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002927
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002928 /* XXX Maybe this should be a separate opcode? */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002929 if (kwdefaults > 0) {
2930 PyObject *defs = PyDict_New();
2931 if (defs == NULL) {
2932 Py_DECREF(func);
2933 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002934 }
2935 while (--kwdefaults >= 0) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002936 PyObject *v = POP(); /* default value */
2937 PyObject *key = POP(); /* kw only arg name */
2938 int err = PyDict_SetItem(defs, key, v);
2939 Py_DECREF(v);
2940 Py_DECREF(key);
2941 if (err != 0) {
2942 Py_DECREF(defs);
2943 Py_DECREF(func);
2944 goto error;
2945 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002946 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002947 if (PyFunction_SetKwDefaults(func, defs) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002948 /* Can't happen unless
2949 PyFunction_SetKwDefaults changes. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002950 Py_DECREF(func);
2951 Py_DECREF(defs);
2952 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002953 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002954 Py_DECREF(defs);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002955 }
Benjamin Peterson1ef876c2013-02-10 09:29:59 -05002956 if (posdefaults > 0) {
2957 PyObject *defs = PyTuple_New(posdefaults);
2958 if (defs == NULL) {
2959 Py_DECREF(func);
2960 goto error;
2961 }
2962 while (--posdefaults >= 0)
2963 PyTuple_SET_ITEM(defs, posdefaults, POP());
2964 if (PyFunction_SetDefaults(func, defs) != 0) {
2965 /* Can't happen unless
2966 PyFunction_SetDefaults changes. */
2967 Py_DECREF(defs);
2968 Py_DECREF(func);
2969 goto error;
2970 }
2971 Py_DECREF(defs);
2972 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002973 PUSH(func);
2974 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002975 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002976
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002977 TARGET(BUILD_SLICE) {
2978 PyObject *start, *stop, *step, *slice;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002979 if (oparg == 3)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002980 step = POP();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002981 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002982 step = NULL;
2983 stop = POP();
2984 start = TOP();
2985 slice = PySlice_New(start, stop, step);
2986 Py_DECREF(start);
2987 Py_DECREF(stop);
2988 Py_XDECREF(step);
2989 SET_TOP(slice);
2990 if (slice == NULL)
2991 goto error;
2992 DISPATCH();
2993 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002994
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002995 TARGET(EXTENDED_ARG) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002996 opcode = NEXTOP();
2997 oparg = oparg<<16 | NEXTARG();
2998 goto dispatch_opcode;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002999 }
Guido van Rossum8861b741996-07-30 16:49:37 +00003000
Antoine Pitrou042b1282010-08-13 21:15:58 +00003001#if USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003002 _unknown_opcode:
Antoine Pitroub52ec782009-01-25 16:34:23 +00003003#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003004 default:
3005 fprintf(stderr,
3006 "XXX lineno: %d, opcode: %d\n",
3007 PyFrame_GetLineNumber(f),
3008 opcode);
3009 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003010 goto error;
Guido van Rossum04691fc1992-08-12 15:35:34 +00003011
3012#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003013 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00003014#endif
3015
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003016 } /* switch */
Guido van Rossum374a9221991-04-04 10:40:29 +00003017
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003018 /* This should never be reached. Every opcode should end with DISPATCH()
3019 or goto error. */
3020 assert(0);
Guido van Rossumac7be682001-01-17 15:42:30 +00003021
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003022error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003023 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003024
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003025 assert(why == WHY_NOT);
3026 why = WHY_EXCEPTION;
Guido van Rossumac7be682001-01-17 15:42:30 +00003027
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003028 /* Double-check exception status. */
3029 if (!PyErr_Occurred())
3030 PyErr_SetString(PyExc_SystemError,
3031 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00003032
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003033 /* Log traceback info. */
3034 PyTraceBack_Here(f);
Guido van Rossumac7be682001-01-17 15:42:30 +00003035
Benjamin Peterson51f46162013-01-23 08:38:47 -05003036 if (tstate->c_tracefunc != NULL)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003037 call_exc_trace(tstate->c_tracefunc, tstate->c_traceobj, f);
Guido van Rossumac7be682001-01-17 15:42:30 +00003038
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003039fast_block_end:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003040 assert(why != WHY_NOT);
3041
3042 /* Unwind stacks if a (pseudo) exception occurred */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003043 while (why != WHY_NOT && f->f_iblock > 0) {
3044 /* Peek at the current block. */
3045 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00003046
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003047 assert(why != WHY_YIELD);
3048 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
3049 why = WHY_NOT;
3050 JUMPTO(PyLong_AS_LONG(retval));
3051 Py_DECREF(retval);
3052 break;
3053 }
3054 /* Now we have to pop the block. */
3055 f->f_iblock--;
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00003056
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003057 if (b->b_type == EXCEPT_HANDLER) {
3058 UNWIND_EXCEPT_HANDLER(b);
3059 continue;
3060 }
3061 UNWIND_BLOCK(b);
3062 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
3063 why = WHY_NOT;
3064 JUMPTO(b->b_handler);
3065 break;
3066 }
3067 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
3068 || b->b_type == SETUP_FINALLY)) {
3069 PyObject *exc, *val, *tb;
3070 int handler = b->b_handler;
3071 /* Beware, this invalidates all b->b_* fields */
3072 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
3073 PUSH(tstate->exc_traceback);
3074 PUSH(tstate->exc_value);
3075 if (tstate->exc_type != NULL) {
3076 PUSH(tstate->exc_type);
3077 }
3078 else {
3079 Py_INCREF(Py_None);
3080 PUSH(Py_None);
3081 }
3082 PyErr_Fetch(&exc, &val, &tb);
3083 /* Make the raw exception data
3084 available to the handler,
3085 so a program can emulate the
3086 Python main loop. */
3087 PyErr_NormalizeException(
3088 &exc, &val, &tb);
3089 PyException_SetTraceback(val, tb);
3090 Py_INCREF(exc);
3091 tstate->exc_type = exc;
3092 Py_INCREF(val);
3093 tstate->exc_value = val;
3094 tstate->exc_traceback = tb;
3095 if (tb == NULL)
3096 tb = Py_None;
3097 Py_INCREF(tb);
3098 PUSH(tb);
3099 PUSH(val);
3100 PUSH(exc);
3101 why = WHY_NOT;
3102 JUMPTO(handler);
3103 break;
3104 }
3105 if (b->b_type == SETUP_FINALLY) {
3106 if (why & (WHY_RETURN | WHY_CONTINUE))
3107 PUSH(retval);
3108 PUSH(PyLong_FromLong((long)why));
3109 why = WHY_NOT;
3110 JUMPTO(b->b_handler);
3111 break;
3112 }
3113 } /* unwind stack */
Guido van Rossum374a9221991-04-04 10:40:29 +00003114
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003115 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00003116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003117 if (why != WHY_NOT)
3118 break;
3119 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00003120
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003121 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00003122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003123 assert(why != WHY_YIELD);
3124 /* Pop remaining stack entries. */
3125 while (!EMPTY()) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003126 PyObject *o = POP();
3127 Py_XDECREF(o);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003128 }
Guido van Rossum35974fb2001-12-06 21:28:18 +00003129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003130 if (why != WHY_RETURN)
3131 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00003132
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003133fast_yield:
Benjamin Petersonac913412011-07-03 16:25:11 -05003134 if (co->co_flags & CO_GENERATOR && (why == WHY_YIELD || why == WHY_RETURN)) {
3135 /* The purpose of this block is to put aside the generator's exception
3136 state and restore that of the calling frame. If the current
3137 exception state is from the caller, we clear the exception values
3138 on the generator frame, so they are not swapped back in latter. The
3139 origin of the current exception state is determined by checking for
3140 except handler blocks, which we must be in iff a new exception
3141 state came into existence in this frame. (An uncaught exception
3142 would have why == WHY_EXCEPTION, and we wouldn't be here). */
3143 int i;
3144 for (i = 0; i < f->f_iblock; i++)
3145 if (f->f_blockstack[i].b_type == EXCEPT_HANDLER)
3146 break;
3147 if (i == f->f_iblock)
3148 /* We did not create this exception. */
Benjamin Peterson87880242011-07-03 16:48:31 -05003149 restore_and_clear_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003150 else
Benjamin Peterson87880242011-07-03 16:48:31 -05003151 swap_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003152 }
Benjamin Peterson83195c32011-07-03 13:44:00 -05003153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003154 if (tstate->use_tracing) {
Benjamin Peterson51f46162013-01-23 08:38:47 -05003155 if (tstate->c_tracefunc) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003156 if (why == WHY_RETURN || why == WHY_YIELD) {
3157 if (call_trace(tstate->c_tracefunc,
3158 tstate->c_traceobj, f,
3159 PyTrace_RETURN, retval)) {
3160 Py_XDECREF(retval);
3161 retval = NULL;
3162 why = WHY_EXCEPTION;
3163 }
3164 }
3165 else if (why == WHY_EXCEPTION) {
3166 call_trace_protected(tstate->c_tracefunc,
3167 tstate->c_traceobj, f,
3168 PyTrace_RETURN, NULL);
3169 }
3170 }
3171 if (tstate->c_profilefunc) {
3172 if (why == WHY_EXCEPTION)
3173 call_trace_protected(tstate->c_profilefunc,
3174 tstate->c_profileobj, f,
3175 PyTrace_RETURN, NULL);
3176 else if (call_trace(tstate->c_profilefunc,
3177 tstate->c_profileobj, f,
3178 PyTrace_RETURN, retval)) {
3179 Py_XDECREF(retval);
3180 retval = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003181 /* why = WHY_EXCEPTION; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003182 }
3183 }
3184 }
Guido van Rossuma4240131997-01-21 21:18:36 +00003185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003186 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00003187exit_eval_frame:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003188 Py_LeaveRecursiveCall();
3189 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00003190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003191 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00003192}
3193
Benjamin Petersonb204a422011-06-05 22:04:07 -05003194static void
Benjamin Petersone109c702011-06-24 09:37:26 -05003195format_missing(const char *kind, PyCodeObject *co, PyObject *names)
3196{
3197 int err;
3198 Py_ssize_t len = PyList_GET_SIZE(names);
3199 PyObject *name_str, *comma, *tail, *tmp;
3200
3201 assert(PyList_CheckExact(names));
3202 assert(len >= 1);
3203 /* Deal with the joys of natural language. */
3204 switch (len) {
3205 case 1:
3206 name_str = PyList_GET_ITEM(names, 0);
3207 Py_INCREF(name_str);
3208 break;
3209 case 2:
3210 name_str = PyUnicode_FromFormat("%U and %U",
3211 PyList_GET_ITEM(names, len - 2),
3212 PyList_GET_ITEM(names, len - 1));
3213 break;
3214 default:
3215 tail = PyUnicode_FromFormat(", %U, and %U",
3216 PyList_GET_ITEM(names, len - 2),
3217 PyList_GET_ITEM(names, len - 1));
Benjamin Petersond1ab6082012-06-01 11:18:22 -07003218 if (tail == NULL)
3219 return;
Benjamin Petersone109c702011-06-24 09:37:26 -05003220 /* Chop off the last two objects in the list. This shouldn't actually
3221 fail, but we can't be too careful. */
3222 err = PyList_SetSlice(names, len - 2, len, NULL);
3223 if (err == -1) {
3224 Py_DECREF(tail);
3225 return;
3226 }
3227 /* Stitch everything up into a nice comma-separated list. */
3228 comma = PyUnicode_FromString(", ");
3229 if (comma == NULL) {
3230 Py_DECREF(tail);
3231 return;
3232 }
3233 tmp = PyUnicode_Join(comma, names);
3234 Py_DECREF(comma);
3235 if (tmp == NULL) {
3236 Py_DECREF(tail);
3237 return;
3238 }
3239 name_str = PyUnicode_Concat(tmp, tail);
3240 Py_DECREF(tmp);
3241 Py_DECREF(tail);
3242 break;
3243 }
3244 if (name_str == NULL)
3245 return;
3246 PyErr_Format(PyExc_TypeError,
3247 "%U() missing %i required %s argument%s: %U",
3248 co->co_name,
3249 len,
3250 kind,
3251 len == 1 ? "" : "s",
3252 name_str);
3253 Py_DECREF(name_str);
3254}
3255
3256static void
3257missing_arguments(PyCodeObject *co, int missing, int defcount,
3258 PyObject **fastlocals)
3259{
3260 int i, j = 0;
3261 int start, end;
3262 int positional = defcount != -1;
3263 const char *kind = positional ? "positional" : "keyword-only";
3264 PyObject *missing_names;
3265
3266 /* Compute the names of the arguments that are missing. */
3267 missing_names = PyList_New(missing);
3268 if (missing_names == NULL)
3269 return;
3270 if (positional) {
3271 start = 0;
3272 end = co->co_argcount - defcount;
3273 }
3274 else {
3275 start = co->co_argcount;
3276 end = start + co->co_kwonlyargcount;
3277 }
3278 for (i = start; i < end; i++) {
3279 if (GETLOCAL(i) == NULL) {
3280 PyObject *raw = PyTuple_GET_ITEM(co->co_varnames, i);
3281 PyObject *name = PyObject_Repr(raw);
3282 if (name == NULL) {
3283 Py_DECREF(missing_names);
3284 return;
3285 }
3286 PyList_SET_ITEM(missing_names, j++, name);
3287 }
3288 }
3289 assert(j == missing);
3290 format_missing(kind, co, missing_names);
3291 Py_DECREF(missing_names);
3292}
3293
3294static void
3295too_many_positional(PyCodeObject *co, int given, int defcount, PyObject **fastlocals)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003296{
3297 int plural;
3298 int kwonly_given = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003299 int i;
3300 PyObject *sig, *kwonly_sig;
3301
Benjamin Petersone109c702011-06-24 09:37:26 -05003302 assert((co->co_flags & CO_VARARGS) == 0);
3303 /* Count missing keyword-only args. */
Benjamin Petersonb204a422011-06-05 22:04:07 -05003304 for (i = co->co_argcount; i < co->co_argcount + co->co_kwonlyargcount; i++)
Benjamin Petersone109c702011-06-24 09:37:26 -05003305 if (GETLOCAL(i) != NULL)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003306 kwonly_given++;
Benjamin Petersone109c702011-06-24 09:37:26 -05003307 if (defcount) {
3308 int atleast = co->co_argcount - defcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003309 plural = 1;
3310 sig = PyUnicode_FromFormat("from %d to %d", atleast, co->co_argcount);
3311 }
3312 else {
3313 plural = co->co_argcount != 1;
3314 sig = PyUnicode_FromFormat("%d", co->co_argcount);
3315 }
3316 if (sig == NULL)
3317 return;
3318 if (kwonly_given) {
3319 const char *format = " positional argument%s (and %d keyword-only argument%s)";
3320 kwonly_sig = PyUnicode_FromFormat(format, given != 1 ? "s" : "", kwonly_given,
3321 kwonly_given != 1 ? "s" : "");
3322 if (kwonly_sig == NULL) {
3323 Py_DECREF(sig);
3324 return;
3325 }
3326 }
3327 else {
3328 /* This will not fail. */
3329 kwonly_sig = PyUnicode_FromString("");
Benjamin Petersone109c702011-06-24 09:37:26 -05003330 assert(kwonly_sig != NULL);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003331 }
3332 PyErr_Format(PyExc_TypeError,
3333 "%U() takes %U positional argument%s but %d%U %s given",
3334 co->co_name,
3335 sig,
3336 plural ? "s" : "",
3337 given,
3338 kwonly_sig,
3339 given == 1 && !kwonly_given ? "was" : "were");
3340 Py_DECREF(sig);
3341 Py_DECREF(kwonly_sig);
3342}
3343
Guido van Rossumc2e20742006-02-27 22:32:47 +00003344/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00003345 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00003346 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00003347
Tim Peters6d6c1a32001-08-02 04:15:00 +00003348PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003349PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003350 PyObject **args, int argcount, PyObject **kws, int kwcount,
3351 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00003352{
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003353 PyCodeObject* co = (PyCodeObject*)_co;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003354 register PyFrameObject *f;
3355 register PyObject *retval = NULL;
3356 register PyObject **fastlocals, **freevars;
3357 PyThreadState *tstate = PyThreadState_GET();
3358 PyObject *x, *u;
3359 int total_args = co->co_argcount + co->co_kwonlyargcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003360 int i;
3361 int n = argcount;
3362 PyObject *kwdict = NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003363
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003364 if (globals == NULL) {
3365 PyErr_SetString(PyExc_SystemError,
3366 "PyEval_EvalCodeEx: NULL globals");
3367 return NULL;
3368 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003370 assert(tstate != NULL);
3371 assert(globals != NULL);
3372 f = PyFrame_New(tstate, co, globals, locals);
3373 if (f == NULL)
3374 return NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003376 fastlocals = f->f_localsplus;
3377 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00003378
Benjamin Petersonb204a422011-06-05 22:04:07 -05003379 /* Parse arguments. */
3380 if (co->co_flags & CO_VARKEYWORDS) {
3381 kwdict = PyDict_New();
3382 if (kwdict == NULL)
3383 goto fail;
3384 i = total_args;
3385 if (co->co_flags & CO_VARARGS)
3386 i++;
3387 SETLOCAL(i, kwdict);
3388 }
3389 if (argcount > co->co_argcount)
3390 n = co->co_argcount;
3391 for (i = 0; i < n; i++) {
3392 x = args[i];
3393 Py_INCREF(x);
3394 SETLOCAL(i, x);
3395 }
3396 if (co->co_flags & CO_VARARGS) {
3397 u = PyTuple_New(argcount - n);
3398 if (u == NULL)
3399 goto fail;
3400 SETLOCAL(total_args, u);
3401 for (i = n; i < argcount; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003402 x = args[i];
3403 Py_INCREF(x);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003404 PyTuple_SET_ITEM(u, i-n, x);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003405 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003406 }
3407 for (i = 0; i < kwcount; i++) {
3408 PyObject **co_varnames;
3409 PyObject *keyword = kws[2*i];
3410 PyObject *value = kws[2*i + 1];
3411 int j;
3412 if (keyword == NULL || !PyUnicode_Check(keyword)) {
3413 PyErr_Format(PyExc_TypeError,
3414 "%U() keywords must be strings",
3415 co->co_name);
3416 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003417 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003418 /* Speed hack: do raw pointer compares. As names are
3419 normally interned this should almost always hit. */
3420 co_varnames = ((PyTupleObject *)(co->co_varnames))->ob_item;
3421 for (j = 0; j < total_args; j++) {
3422 PyObject *nm = co_varnames[j];
3423 if (nm == keyword)
3424 goto kw_found;
3425 }
3426 /* Slow fallback, just in case */
3427 for (j = 0; j < total_args; j++) {
3428 PyObject *nm = co_varnames[j];
3429 int cmp = PyObject_RichCompareBool(
3430 keyword, nm, Py_EQ);
3431 if (cmp > 0)
3432 goto kw_found;
3433 else if (cmp < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003434 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003435 }
3436 if (j >= total_args && kwdict == NULL) {
3437 PyErr_Format(PyExc_TypeError,
3438 "%U() got an unexpected "
3439 "keyword argument '%S'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003440 co->co_name,
3441 keyword);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003442 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003443 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003444 PyDict_SetItem(kwdict, keyword, value);
3445 continue;
3446 kw_found:
3447 if (GETLOCAL(j) != NULL) {
3448 PyErr_Format(PyExc_TypeError,
3449 "%U() got multiple "
3450 "values for argument '%S'",
3451 co->co_name,
3452 keyword);
3453 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003454 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003455 Py_INCREF(value);
3456 SETLOCAL(j, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003457 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003458 if (argcount > co->co_argcount && !(co->co_flags & CO_VARARGS)) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003459 too_many_positional(co, argcount, defcount, fastlocals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003460 goto fail;
3461 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003462 if (argcount < co->co_argcount) {
3463 int m = co->co_argcount - defcount;
Benjamin Petersone109c702011-06-24 09:37:26 -05003464 int missing = 0;
3465 for (i = argcount; i < m; i++)
3466 if (GETLOCAL(i) == NULL)
3467 missing++;
3468 if (missing) {
3469 missing_arguments(co, missing, defcount, fastlocals);
3470 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003471 }
3472 if (n > m)
3473 i = n - m;
3474 else
3475 i = 0;
3476 for (; i < defcount; i++) {
3477 if (GETLOCAL(m+i) == NULL) {
3478 PyObject *def = defs[i];
3479 Py_INCREF(def);
3480 SETLOCAL(m+i, def);
3481 }
3482 }
3483 }
3484 if (co->co_kwonlyargcount > 0) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003485 int missing = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003486 for (i = co->co_argcount; i < total_args; i++) {
3487 PyObject *name;
3488 if (GETLOCAL(i) != NULL)
3489 continue;
3490 name = PyTuple_GET_ITEM(co->co_varnames, i);
3491 if (kwdefs != NULL) {
3492 PyObject *def = PyDict_GetItem(kwdefs, name);
3493 if (def) {
3494 Py_INCREF(def);
3495 SETLOCAL(i, def);
3496 continue;
3497 }
3498 }
Benjamin Petersone109c702011-06-24 09:37:26 -05003499 missing++;
3500 }
3501 if (missing) {
3502 missing_arguments(co, missing, -1, fastlocals);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003503 goto fail;
3504 }
3505 }
3506
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003507 /* Allocate and initialize storage for cell vars, and copy free
Benjamin Peterson90037602011-06-25 22:54:45 -05003508 vars into frame. */
3509 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003510 PyObject *c;
Benjamin Peterson90037602011-06-25 22:54:45 -05003511 int arg;
3512 /* Possibly account for the cell variable being an argument. */
3513 if (co->co_cell2arg != NULL &&
Guido van Rossum6832c812013-05-10 08:47:42 -07003514 (arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG) {
Benjamin Peterson90037602011-06-25 22:54:45 -05003515 c = PyCell_New(GETLOCAL(arg));
Benjamin Peterson159ae412013-05-12 18:16:06 -05003516 /* Clear the local copy. */
3517 SETLOCAL(arg, NULL);
Guido van Rossum6832c812013-05-10 08:47:42 -07003518 }
3519 else {
Benjamin Peterson90037602011-06-25 22:54:45 -05003520 c = PyCell_New(NULL);
Guido van Rossum6832c812013-05-10 08:47:42 -07003521 }
Benjamin Peterson159ae412013-05-12 18:16:06 -05003522 if (c == NULL)
3523 goto fail;
Benjamin Peterson90037602011-06-25 22:54:45 -05003524 SETLOCAL(co->co_nlocals + i, c);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003525 }
Benjamin Peterson90037602011-06-25 22:54:45 -05003526 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
3527 PyObject *o = PyTuple_GET_ITEM(closure, i);
3528 Py_INCREF(o);
3529 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003530 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003531
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003532 if (co->co_flags & CO_GENERATOR) {
3533 /* Don't need to keep the reference to f_back, it will be set
3534 * when the generator is resumed. */
3535 Py_XDECREF(f->f_back);
3536 f->f_back = NULL;
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003538 PCALL(PCALL_GENERATOR);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003540 /* Create a new generator that owns the ready to run frame
3541 * and return that as the value. */
3542 return PyGen_New(f);
3543 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003545 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003546
Thomas Woutersce272b62007-09-19 21:19:28 +00003547fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003548
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003549 /* decref'ing the frame can cause __del__ methods to get invoked,
3550 which can call back into Python. While we're done with the
3551 current Python frame (f), the associated C stack is still in use,
3552 so recursion_depth must be boosted for the duration.
3553 */
3554 assert(tstate != NULL);
3555 ++tstate->recursion_depth;
3556 Py_DECREF(f);
3557 --tstate->recursion_depth;
3558 return retval;
Tim Peters5ca576e2001-06-18 22:08:13 +00003559}
3560
3561
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003562static PyObject *
Benjamin Petersonce798522012-01-22 11:24:29 -05003563special_lookup(PyObject *o, _Py_Identifier *id)
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003564{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003565 PyObject *res;
Benjamin Petersonce798522012-01-22 11:24:29 -05003566 res = _PyObject_LookupSpecial(o, id);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003567 if (res == NULL && !PyErr_Occurred()) {
Benjamin Petersonce798522012-01-22 11:24:29 -05003568 PyErr_SetObject(PyExc_AttributeError, id->object);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003569 return NULL;
3570 }
3571 return res;
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003572}
3573
3574
Benjamin Peterson87880242011-07-03 16:48:31 -05003575/* These 3 functions deal with the exception state of generators. */
3576
3577static void
3578save_exc_state(PyThreadState *tstate, PyFrameObject *f)
3579{
3580 PyObject *type, *value, *traceback;
3581 Py_XINCREF(tstate->exc_type);
3582 Py_XINCREF(tstate->exc_value);
3583 Py_XINCREF(tstate->exc_traceback);
3584 type = f->f_exc_type;
3585 value = f->f_exc_value;
3586 traceback = f->f_exc_traceback;
3587 f->f_exc_type = tstate->exc_type;
3588 f->f_exc_value = tstate->exc_value;
3589 f->f_exc_traceback = tstate->exc_traceback;
3590 Py_XDECREF(type);
3591 Py_XDECREF(value);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02003592 Py_XDECREF(traceback);
Benjamin Peterson87880242011-07-03 16:48:31 -05003593}
3594
3595static void
3596swap_exc_state(PyThreadState *tstate, PyFrameObject *f)
3597{
3598 PyObject *tmp;
3599 tmp = tstate->exc_type;
3600 tstate->exc_type = f->f_exc_type;
3601 f->f_exc_type = tmp;
3602 tmp = tstate->exc_value;
3603 tstate->exc_value = f->f_exc_value;
3604 f->f_exc_value = tmp;
3605 tmp = tstate->exc_traceback;
3606 tstate->exc_traceback = f->f_exc_traceback;
3607 f->f_exc_traceback = tmp;
3608}
3609
3610static void
3611restore_and_clear_exc_state(PyThreadState *tstate, PyFrameObject *f)
3612{
3613 PyObject *type, *value, *tb;
3614 type = tstate->exc_type;
3615 value = tstate->exc_value;
3616 tb = tstate->exc_traceback;
3617 tstate->exc_type = f->f_exc_type;
3618 tstate->exc_value = f->f_exc_value;
3619 tstate->exc_traceback = f->f_exc_traceback;
3620 f->f_exc_type = NULL;
3621 f->f_exc_value = NULL;
3622 f->f_exc_traceback = NULL;
3623 Py_XDECREF(type);
3624 Py_XDECREF(value);
3625 Py_XDECREF(tb);
3626}
3627
3628
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003629/* Logic for the raise statement (too complicated for inlining).
3630 This *consumes* a reference count to each of its arguments. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003631static int
Collin Winter828f04a2007-08-31 00:04:24 +00003632do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003633{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003634 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00003635
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003636 if (exc == NULL) {
3637 /* Reraise */
3638 PyThreadState *tstate = PyThreadState_GET();
3639 PyObject *tb;
3640 type = tstate->exc_type;
3641 value = tstate->exc_value;
3642 tb = tstate->exc_traceback;
3643 if (type == Py_None) {
3644 PyErr_SetString(PyExc_RuntimeError,
3645 "No active exception to reraise");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003646 return 0;
3647 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003648 Py_XINCREF(type);
3649 Py_XINCREF(value);
3650 Py_XINCREF(tb);
3651 PyErr_Restore(type, value, tb);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003652 return 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003653 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003655 /* We support the following forms of raise:
3656 raise
Collin Winter828f04a2007-08-31 00:04:24 +00003657 raise <instance>
3658 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003660 if (PyExceptionClass_Check(exc)) {
3661 type = exc;
3662 value = PyObject_CallObject(exc, NULL);
3663 if (value == NULL)
3664 goto raise_error;
Benjamin Peterson5afa03a2011-07-15 14:09:26 -05003665 if (!PyExceptionInstance_Check(value)) {
3666 PyErr_Format(PyExc_TypeError,
3667 "calling %R should have returned an instance of "
3668 "BaseException, not %R",
3669 type, Py_TYPE(value));
3670 goto raise_error;
3671 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003672 }
3673 else if (PyExceptionInstance_Check(exc)) {
3674 value = exc;
3675 type = PyExceptionInstance_Class(exc);
3676 Py_INCREF(type);
3677 }
3678 else {
3679 /* Not something you can raise. You get an exception
3680 anyway, just not what you specified :-) */
3681 Py_DECREF(exc);
3682 PyErr_SetString(PyExc_TypeError,
3683 "exceptions must derive from BaseException");
3684 goto raise_error;
3685 }
Collin Winter828f04a2007-08-31 00:04:24 +00003686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003687 if (cause) {
3688 PyObject *fixed_cause;
3689 if (PyExceptionClass_Check(cause)) {
3690 fixed_cause = PyObject_CallObject(cause, NULL);
3691 if (fixed_cause == NULL)
3692 goto raise_error;
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003693 Py_DECREF(cause);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003694 }
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003695 else if (PyExceptionInstance_Check(cause)) {
3696 fixed_cause = cause;
3697 }
3698 else if (cause == Py_None) {
3699 Py_DECREF(cause);
3700 fixed_cause = NULL;
3701 }
3702 else {
3703 PyErr_SetString(PyExc_TypeError,
3704 "exception causes must derive from "
3705 "BaseException");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003706 goto raise_error;
3707 }
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003708 PyException_SetCause(value, fixed_cause);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003709 }
Collin Winter828f04a2007-08-31 00:04:24 +00003710
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003711 PyErr_SetObject(type, value);
3712 /* PyErr_SetObject incref's its arguments */
3713 Py_XDECREF(value);
3714 Py_XDECREF(type);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003715 return 0;
Collin Winter828f04a2007-08-31 00:04:24 +00003716
3717raise_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003718 Py_XDECREF(value);
3719 Py_XDECREF(type);
3720 Py_XDECREF(cause);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003721 return 0;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003722}
3723
Tim Petersd6d010b2001-06-21 02:49:55 +00003724/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003725 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00003726
Guido van Rossum0368b722007-05-11 16:50:42 +00003727 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3728 with a variable target.
3729*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003730
Barry Warsawe42b18f1997-08-25 22:13:04 +00003731static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003732unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003733{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003734 int i = 0, j = 0;
3735 Py_ssize_t ll = 0;
3736 PyObject *it; /* iter(v) */
3737 PyObject *w;
3738 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003739
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003740 assert(v != NULL);
Tim Petersd6d010b2001-06-21 02:49:55 +00003741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003742 it = PyObject_GetIter(v);
3743 if (it == NULL)
3744 goto Error;
Tim Petersd6d010b2001-06-21 02:49:55 +00003745
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003746 for (; i < argcnt; i++) {
3747 w = PyIter_Next(it);
3748 if (w == NULL) {
3749 /* Iterator done, via error or exhaustion. */
3750 if (!PyErr_Occurred()) {
3751 PyErr_Format(PyExc_ValueError,
3752 "need more than %d value%s to unpack",
3753 i, i == 1 ? "" : "s");
3754 }
3755 goto Error;
3756 }
3757 *--sp = w;
3758 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003759
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003760 if (argcntafter == -1) {
3761 /* We better have exhausted the iterator now. */
3762 w = PyIter_Next(it);
3763 if (w == NULL) {
3764 if (PyErr_Occurred())
3765 goto Error;
3766 Py_DECREF(it);
3767 return 1;
3768 }
3769 Py_DECREF(w);
Georg Brandl0310a832010-07-10 10:32:36 +00003770 PyErr_Format(PyExc_ValueError, "too many values to unpack "
3771 "(expected %d)", argcnt);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003772 goto Error;
3773 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003774
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003775 l = PySequence_List(it);
3776 if (l == NULL)
3777 goto Error;
3778 *--sp = l;
3779 i++;
Guido van Rossum0368b722007-05-11 16:50:42 +00003780
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003781 ll = PyList_GET_SIZE(l);
3782 if (ll < argcntafter) {
3783 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3784 argcnt + ll);
3785 goto Error;
3786 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003787
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003788 /* Pop the "after-variable" args off the list. */
3789 for (j = argcntafter; j > 0; j--, i++) {
3790 *--sp = PyList_GET_ITEM(l, ll - j);
3791 }
3792 /* Resize the list. */
3793 Py_SIZE(l) = ll - argcntafter;
3794 Py_DECREF(it);
3795 return 1;
Guido van Rossum0368b722007-05-11 16:50:42 +00003796
Tim Petersd6d010b2001-06-21 02:49:55 +00003797Error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003798 for (; i > 0; i--, sp++)
3799 Py_DECREF(*sp);
3800 Py_XDECREF(it);
3801 return 0;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003802}
3803
3804
Guido van Rossum96a42c81992-01-12 02:29:51 +00003805#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003806static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003807prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003808{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003809 printf("%s ", str);
3810 if (PyObject_Print(v, stdout, 0) != 0)
3811 PyErr_Clear(); /* Don't know what else to do */
3812 printf("\n");
3813 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003814}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003815#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003816
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003817static void
Fred Drake5755ce62001-06-27 19:19:46 +00003818call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003819{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003820 PyObject *type, *value, *traceback, *arg;
3821 int err;
3822 PyErr_Fetch(&type, &value, &traceback);
3823 if (value == NULL) {
3824 value = Py_None;
3825 Py_INCREF(value);
3826 }
R David Murray35837612013-04-19 12:56:57 -04003827 PyErr_NormalizeException(&type, &value, &traceback);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003828 arg = PyTuple_Pack(3, type, value, traceback);
3829 if (arg == NULL) {
3830 PyErr_Restore(type, value, traceback);
3831 return;
3832 }
3833 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
3834 Py_DECREF(arg);
3835 if (err == 0)
3836 PyErr_Restore(type, value, traceback);
3837 else {
3838 Py_XDECREF(type);
3839 Py_XDECREF(value);
3840 Py_XDECREF(traceback);
3841 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003842}
3843
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003844static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003845call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003846 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003847{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003848 PyObject *type, *value, *traceback;
3849 int err;
3850 PyErr_Fetch(&type, &value, &traceback);
3851 err = call_trace(func, obj, frame, what, arg);
3852 if (err == 0)
3853 {
3854 PyErr_Restore(type, value, traceback);
3855 return 0;
3856 }
3857 else {
3858 Py_XDECREF(type);
3859 Py_XDECREF(value);
3860 Py_XDECREF(traceback);
3861 return -1;
3862 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003863}
3864
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003865static int
Fred Drake5755ce62001-06-27 19:19:46 +00003866call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003867 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003868{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003869 register PyThreadState *tstate = frame->f_tstate;
3870 int result;
3871 if (tstate->tracing)
3872 return 0;
3873 tstate->tracing++;
3874 tstate->use_tracing = 0;
3875 result = func(obj, frame, what, arg);
3876 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3877 || (tstate->c_profilefunc != NULL));
3878 tstate->tracing--;
3879 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003880}
3881
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003882PyObject *
3883_PyEval_CallTracing(PyObject *func, PyObject *args)
3884{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003885 PyFrameObject *frame = PyEval_GetFrame();
3886 PyThreadState *tstate = frame->f_tstate;
3887 int save_tracing = tstate->tracing;
3888 int save_use_tracing = tstate->use_tracing;
3889 PyObject *result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003890
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003891 tstate->tracing = 0;
3892 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3893 || (tstate->c_profilefunc != NULL));
3894 result = PyObject_Call(func, args, NULL);
3895 tstate->tracing = save_tracing;
3896 tstate->use_tracing = save_use_tracing;
3897 return result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003898}
3899
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00003900/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Michael W. Hudson006c7522002-11-08 13:08:46 +00003901static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003902maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003903 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3904 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003905{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003906 int result = 0;
3907 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003908
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003909 /* If the last instruction executed isn't in the current
3910 instruction window, reset the window.
3911 */
3912 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
3913 PyAddrPair bounds;
3914 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3915 &bounds);
3916 *instr_lb = bounds.ap_lower;
3917 *instr_ub = bounds.ap_upper;
3918 }
3919 /* If the last instruction falls at the start of a line or if
3920 it represents a jump backwards, update the frame's line
3921 number and call the trace function. */
3922 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3923 frame->f_lineno = line;
3924 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
3925 }
3926 *instr_prev = frame->f_lasti;
3927 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003928}
3929
Fred Drake5755ce62001-06-27 19:19:46 +00003930void
3931PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003932{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003933 PyThreadState *tstate = PyThreadState_GET();
3934 PyObject *temp = tstate->c_profileobj;
3935 Py_XINCREF(arg);
3936 tstate->c_profilefunc = NULL;
3937 tstate->c_profileobj = NULL;
3938 /* Must make sure that tracing is not ignored if 'temp' is freed */
3939 tstate->use_tracing = tstate->c_tracefunc != NULL;
3940 Py_XDECREF(temp);
3941 tstate->c_profilefunc = func;
3942 tstate->c_profileobj = arg;
3943 /* Flag that tracing or profiling is turned on */
3944 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003945}
3946
3947void
3948PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003950 PyThreadState *tstate = PyThreadState_GET();
3951 PyObject *temp = tstate->c_traceobj;
3952 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
3953 Py_XINCREF(arg);
3954 tstate->c_tracefunc = NULL;
3955 tstate->c_traceobj = NULL;
3956 /* Must make sure that profiling is not ignored if 'temp' is freed */
3957 tstate->use_tracing = tstate->c_profilefunc != NULL;
3958 Py_XDECREF(temp);
3959 tstate->c_tracefunc = func;
3960 tstate->c_traceobj = arg;
3961 /* Flag that tracing or profiling is turned on */
3962 tstate->use_tracing = ((func != NULL)
3963 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003964}
3965
Guido van Rossumb209a111997-04-29 18:18:01 +00003966PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003967PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003968{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003969 PyFrameObject *current_frame = PyEval_GetFrame();
3970 if (current_frame == NULL)
3971 return PyThreadState_GET()->interp->builtins;
3972 else
3973 return current_frame->f_builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003974}
3975
Guido van Rossumb209a111997-04-29 18:18:01 +00003976PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003977PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003978{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003979 PyFrameObject *current_frame = PyEval_GetFrame();
3980 if (current_frame == NULL)
3981 return NULL;
3982 PyFrame_FastToLocals(current_frame);
3983 return current_frame->f_locals;
Guido van Rossum5b722181993-03-30 17:46:03 +00003984}
3985
Guido van Rossumb209a111997-04-29 18:18:01 +00003986PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003987PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003988{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003989 PyFrameObject *current_frame = PyEval_GetFrame();
3990 if (current_frame == NULL)
3991 return NULL;
3992 else
3993 return current_frame->f_globals;
Guido van Rossum3f5da241990-12-20 15:06:42 +00003994}
3995
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003996PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003997PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003998{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003999 PyThreadState *tstate = PyThreadState_GET();
4000 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00004001}
4002
Guido van Rossum6135a871995-01-09 17:53:26 +00004003int
Tim Peters5ba58662001-07-16 02:29:45 +00004004PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00004005{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004006 PyFrameObject *current_frame = PyEval_GetFrame();
4007 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00004008
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004009 if (current_frame != NULL) {
4010 const int codeflags = current_frame->f_code->co_flags;
4011 const int compilerflags = codeflags & PyCF_MASK;
4012 if (compilerflags) {
4013 result = 1;
4014 cf->cf_flags |= compilerflags;
4015 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00004016#if 0 /* future keyword */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004017 if (codeflags & CO_GENERATOR_ALLOWED) {
4018 result = 1;
4019 cf->cf_flags |= CO_GENERATOR_ALLOWED;
4020 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00004021#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004022 }
4023 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00004024}
4025
Guido van Rossum3f5da241990-12-20 15:06:42 +00004026
Guido van Rossum681d79a1995-07-18 14:51:37 +00004027/* External interface to call any callable object.
Antoine Pitrou8689a102010-04-01 16:53:15 +00004028 The arg must be a tuple or NULL. The kw must be a dict or NULL. */
Guido van Rossume59214e1994-08-30 08:01:59 +00004029
Guido van Rossumb209a111997-04-29 18:18:01 +00004030PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004031PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00004032{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004033 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004035 if (arg == NULL) {
4036 arg = PyTuple_New(0);
4037 if (arg == NULL)
4038 return NULL;
4039 }
4040 else if (!PyTuple_Check(arg)) {
4041 PyErr_SetString(PyExc_TypeError,
4042 "argument list must be a tuple");
4043 return NULL;
4044 }
4045 else
4046 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004048 if (kw != NULL && !PyDict_Check(kw)) {
4049 PyErr_SetString(PyExc_TypeError,
4050 "keyword list must be a dictionary");
4051 Py_DECREF(arg);
4052 return NULL;
4053 }
Guido van Rossume3e61c11995-08-04 04:14:47 +00004054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004055 result = PyObject_Call(func, arg, kw);
4056 Py_DECREF(arg);
4057 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004058}
4059
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00004060const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00004061PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00004062{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004063 if (PyMethod_Check(func))
4064 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
4065 else if (PyFunction_Check(func))
4066 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
4067 else if (PyCFunction_Check(func))
4068 return ((PyCFunctionObject*)func)->m_ml->ml_name;
4069 else
4070 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00004071}
4072
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00004073const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00004074PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00004075{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004076 if (PyMethod_Check(func))
4077 return "()";
4078 else if (PyFunction_Check(func))
4079 return "()";
4080 else if (PyCFunction_Check(func))
4081 return "()";
4082 else
4083 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00004084}
4085
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00004086static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00004087err_args(PyObject *func, int flags, int nargs)
4088{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004089 if (flags & METH_NOARGS)
4090 PyErr_Format(PyExc_TypeError,
4091 "%.200s() takes no arguments (%d given)",
4092 ((PyCFunctionObject *)func)->m_ml->ml_name,
4093 nargs);
4094 else
4095 PyErr_Format(PyExc_TypeError,
4096 "%.200s() takes exactly one argument (%d given)",
4097 ((PyCFunctionObject *)func)->m_ml->ml_name,
4098 nargs);
Jeremy Hylton192690e2002-08-16 18:36:11 +00004099}
4100
Armin Rigo1c2d7e52005-09-20 18:34:01 +00004101#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00004102if (tstate->use_tracing && tstate->c_profilefunc) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004103 if (call_trace(tstate->c_profilefunc, \
4104 tstate->c_profileobj, \
4105 tstate->frame, PyTrace_C_CALL, \
4106 func)) { \
4107 x = NULL; \
4108 } \
4109 else { \
4110 x = call; \
4111 if (tstate->c_profilefunc != NULL) { \
4112 if (x == NULL) { \
4113 call_trace_protected(tstate->c_profilefunc, \
4114 tstate->c_profileobj, \
4115 tstate->frame, PyTrace_C_EXCEPTION, \
4116 func); \
4117 /* XXX should pass (type, value, tb) */ \
4118 } else { \
4119 if (call_trace(tstate->c_profilefunc, \
4120 tstate->c_profileobj, \
4121 tstate->frame, PyTrace_C_RETURN, \
4122 func)) { \
4123 Py_DECREF(x); \
4124 x = NULL; \
4125 } \
4126 } \
4127 } \
4128 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00004129} else { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004130 x = call; \
4131 }
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00004132
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004133static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004134call_function(PyObject ***pp_stack, int oparg
4135#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004136 , uint64* pintr0, uint64* pintr1
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004137#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004138 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004139{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004140 int na = oparg & 0xff;
4141 int nk = (oparg>>8) & 0xff;
4142 int n = na + 2 * nk;
4143 PyObject **pfunc = (*pp_stack) - n - 1;
4144 PyObject *func = *pfunc;
4145 PyObject *x, *w;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004146
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004147 /* Always dispatch PyCFunction first, because these are
4148 presumed to be the most frequent callable object.
4149 */
4150 if (PyCFunction_Check(func) && nk == 0) {
4151 int flags = PyCFunction_GET_FLAGS(func);
4152 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00004153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004154 PCALL(PCALL_CFUNCTION);
4155 if (flags & (METH_NOARGS | METH_O)) {
4156 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
4157 PyObject *self = PyCFunction_GET_SELF(func);
4158 if (flags & METH_NOARGS && na == 0) {
4159 C_TRACE(x, (*meth)(self,NULL));
4160 }
4161 else if (flags & METH_O && na == 1) {
4162 PyObject *arg = EXT_POP(*pp_stack);
4163 C_TRACE(x, (*meth)(self,arg));
4164 Py_DECREF(arg);
4165 }
4166 else {
4167 err_args(func, flags, na);
4168 x = NULL;
4169 }
4170 }
4171 else {
4172 PyObject *callargs;
4173 callargs = load_args(pp_stack, na);
4174 READ_TIMESTAMP(*pintr0);
4175 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
4176 READ_TIMESTAMP(*pintr1);
4177 Py_XDECREF(callargs);
4178 }
4179 } else {
4180 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
4181 /* optimize access to bound methods */
4182 PyObject *self = PyMethod_GET_SELF(func);
4183 PCALL(PCALL_METHOD);
4184 PCALL(PCALL_BOUND_METHOD);
4185 Py_INCREF(self);
4186 func = PyMethod_GET_FUNCTION(func);
4187 Py_INCREF(func);
4188 Py_DECREF(*pfunc);
4189 *pfunc = self;
4190 na++;
4191 n++;
4192 } else
4193 Py_INCREF(func);
4194 READ_TIMESTAMP(*pintr0);
4195 if (PyFunction_Check(func))
4196 x = fast_function(func, pp_stack, n, na, nk);
4197 else
4198 x = do_call(func, pp_stack, na, nk);
4199 READ_TIMESTAMP(*pintr1);
4200 Py_DECREF(func);
4201 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00004202
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004203 /* Clear the stack of the function object. Also removes
4204 the arguments in case they weren't consumed already
4205 (fast_function() and err_args() leave them on the stack).
4206 */
4207 while ((*pp_stack) > pfunc) {
4208 w = EXT_POP(*pp_stack);
4209 Py_DECREF(w);
4210 PCALL(PCALL_POP);
4211 }
4212 return x;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004213}
4214
Jeremy Hylton192690e2002-08-16 18:36:11 +00004215/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004216 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004217 For the simplest case -- a function that takes only positional
4218 arguments and is called with only positional arguments -- it
4219 inlines the most primitive frame setup code from
4220 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4221 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004222*/
4223
4224static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004225fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004226{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004227 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
4228 PyObject *globals = PyFunction_GET_GLOBALS(func);
4229 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4230 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
4231 PyObject **d = NULL;
4232 int nd = 0;
Jeremy Hylton52820442001-01-03 23:52:36 +00004233
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004234 PCALL(PCALL_FUNCTION);
4235 PCALL(PCALL_FAST_FUNCTION);
4236 if (argdefs == NULL && co->co_argcount == n &&
4237 co->co_kwonlyargcount == 0 && nk==0 &&
4238 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4239 PyFrameObject *f;
4240 PyObject *retval = NULL;
4241 PyThreadState *tstate = PyThreadState_GET();
4242 PyObject **fastlocals, **stack;
4243 int i;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004245 PCALL(PCALL_FASTER_FUNCTION);
4246 assert(globals != NULL);
4247 /* XXX Perhaps we should create a specialized
4248 PyFrame_New() that doesn't take locals, but does
4249 take builtins without sanity checking them.
4250 */
4251 assert(tstate != NULL);
4252 f = PyFrame_New(tstate, co, globals, NULL);
4253 if (f == NULL)
4254 return NULL;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004255
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004256 fastlocals = f->f_localsplus;
4257 stack = (*pp_stack) - n;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004259 for (i = 0; i < n; i++) {
4260 Py_INCREF(*stack);
4261 fastlocals[i] = *stack++;
4262 }
4263 retval = PyEval_EvalFrameEx(f,0);
4264 ++tstate->recursion_depth;
4265 Py_DECREF(f);
4266 --tstate->recursion_depth;
4267 return retval;
4268 }
4269 if (argdefs != NULL) {
4270 d = &PyTuple_GET_ITEM(argdefs, 0);
4271 nd = Py_SIZE(argdefs);
4272 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004273 return PyEval_EvalCodeEx((PyObject*)co, globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004274 (PyObject *)NULL, (*pp_stack)-n, na,
4275 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
4276 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004277}
4278
4279static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004280update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4281 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004282{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004283 PyObject *kwdict = NULL;
4284 if (orig_kwdict == NULL)
4285 kwdict = PyDict_New();
4286 else {
4287 kwdict = PyDict_Copy(orig_kwdict);
4288 Py_DECREF(orig_kwdict);
4289 }
4290 if (kwdict == NULL)
4291 return NULL;
4292 while (--nk >= 0) {
4293 int err;
4294 PyObject *value = EXT_POP(*pp_stack);
4295 PyObject *key = EXT_POP(*pp_stack);
4296 if (PyDict_GetItem(kwdict, key) != NULL) {
4297 PyErr_Format(PyExc_TypeError,
4298 "%.200s%s got multiple values "
4299 "for keyword argument '%U'",
4300 PyEval_GetFuncName(func),
4301 PyEval_GetFuncDesc(func),
4302 key);
4303 Py_DECREF(key);
4304 Py_DECREF(value);
4305 Py_DECREF(kwdict);
4306 return NULL;
4307 }
4308 err = PyDict_SetItem(kwdict, key, value);
4309 Py_DECREF(key);
4310 Py_DECREF(value);
4311 if (err) {
4312 Py_DECREF(kwdict);
4313 return NULL;
4314 }
4315 }
4316 return kwdict;
Jeremy Hylton52820442001-01-03 23:52:36 +00004317}
4318
4319static PyObject *
4320update_star_args(int nstack, int nstar, PyObject *stararg,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004321 PyObject ***pp_stack)
Jeremy Hylton52820442001-01-03 23:52:36 +00004322{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004323 PyObject *callargs, *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004325 callargs = PyTuple_New(nstack + nstar);
4326 if (callargs == NULL) {
4327 return NULL;
4328 }
4329 if (nstar) {
4330 int i;
4331 for (i = 0; i < nstar; i++) {
4332 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4333 Py_INCREF(a);
4334 PyTuple_SET_ITEM(callargs, nstack + i, a);
4335 }
4336 }
4337 while (--nstack >= 0) {
4338 w = EXT_POP(*pp_stack);
4339 PyTuple_SET_ITEM(callargs, nstack, w);
4340 }
4341 return callargs;
Jeremy Hylton52820442001-01-03 23:52:36 +00004342}
4343
4344static PyObject *
4345load_args(PyObject ***pp_stack, int na)
4346{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004347 PyObject *args = PyTuple_New(na);
4348 PyObject *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004349
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004350 if (args == NULL)
4351 return NULL;
4352 while (--na >= 0) {
4353 w = EXT_POP(*pp_stack);
4354 PyTuple_SET_ITEM(args, na, w);
4355 }
4356 return args;
Jeremy Hylton52820442001-01-03 23:52:36 +00004357}
4358
4359static PyObject *
4360do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4361{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004362 PyObject *callargs = NULL;
4363 PyObject *kwdict = NULL;
4364 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 if (nk > 0) {
4367 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
4368 if (kwdict == NULL)
4369 goto call_fail;
4370 }
4371 callargs = load_args(pp_stack, na);
4372 if (callargs == NULL)
4373 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004374#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004375 /* At this point, we have to look at the type of func to
4376 update the call stats properly. Do it here so as to avoid
4377 exposing the call stats machinery outside ceval.c
4378 */
4379 if (PyFunction_Check(func))
4380 PCALL(PCALL_FUNCTION);
4381 else if (PyMethod_Check(func))
4382 PCALL(PCALL_METHOD);
4383 else if (PyType_Check(func))
4384 PCALL(PCALL_TYPE);
4385 else if (PyCFunction_Check(func))
4386 PCALL(PCALL_CFUNCTION);
4387 else
4388 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004389#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004390 if (PyCFunction_Check(func)) {
4391 PyThreadState *tstate = PyThreadState_GET();
4392 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4393 }
4394 else
4395 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00004396call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004397 Py_XDECREF(callargs);
4398 Py_XDECREF(kwdict);
4399 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004400}
4401
4402static PyObject *
4403ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4404{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004405 int nstar = 0;
4406 PyObject *callargs = NULL;
4407 PyObject *stararg = NULL;
4408 PyObject *kwdict = NULL;
4409 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004410
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004411 if (flags & CALL_FLAG_KW) {
4412 kwdict = EXT_POP(*pp_stack);
4413 if (!PyDict_Check(kwdict)) {
4414 PyObject *d;
4415 d = PyDict_New();
4416 if (d == NULL)
4417 goto ext_call_fail;
4418 if (PyDict_Update(d, kwdict) != 0) {
4419 Py_DECREF(d);
4420 /* PyDict_Update raises attribute
4421 * error (percolated from an attempt
4422 * to get 'keys' attribute) instead of
4423 * a type error if its second argument
4424 * is not a mapping.
4425 */
4426 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4427 PyErr_Format(PyExc_TypeError,
4428 "%.200s%.200s argument after ** "
4429 "must be a mapping, not %.200s",
4430 PyEval_GetFuncName(func),
4431 PyEval_GetFuncDesc(func),
4432 kwdict->ob_type->tp_name);
4433 }
4434 goto ext_call_fail;
4435 }
4436 Py_DECREF(kwdict);
4437 kwdict = d;
4438 }
4439 }
4440 if (flags & CALL_FLAG_VAR) {
4441 stararg = EXT_POP(*pp_stack);
4442 if (!PyTuple_Check(stararg)) {
4443 PyObject *t = NULL;
4444 t = PySequence_Tuple(stararg);
4445 if (t == NULL) {
4446 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4447 PyErr_Format(PyExc_TypeError,
4448 "%.200s%.200s argument after * "
Victor Stinner0a5f65a2011-03-22 01:09:21 +01004449 "must be a sequence, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004450 PyEval_GetFuncName(func),
4451 PyEval_GetFuncDesc(func),
4452 stararg->ob_type->tp_name);
4453 }
4454 goto ext_call_fail;
4455 }
4456 Py_DECREF(stararg);
4457 stararg = t;
4458 }
4459 nstar = PyTuple_GET_SIZE(stararg);
4460 }
4461 if (nk > 0) {
4462 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
4463 if (kwdict == NULL)
4464 goto ext_call_fail;
4465 }
4466 callargs = update_star_args(na, nstar, stararg, pp_stack);
4467 if (callargs == NULL)
4468 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004469#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004470 /* At this point, we have to look at the type of func to
4471 update the call stats properly. Do it here so as to avoid
4472 exposing the call stats machinery outside ceval.c
4473 */
4474 if (PyFunction_Check(func))
4475 PCALL(PCALL_FUNCTION);
4476 else if (PyMethod_Check(func))
4477 PCALL(PCALL_METHOD);
4478 else if (PyType_Check(func))
4479 PCALL(PCALL_TYPE);
4480 else if (PyCFunction_Check(func))
4481 PCALL(PCALL_CFUNCTION);
4482 else
4483 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004484#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004485 if (PyCFunction_Check(func)) {
4486 PyThreadState *tstate = PyThreadState_GET();
4487 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4488 }
4489 else
4490 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00004491ext_call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004492 Py_XDECREF(callargs);
4493 Py_XDECREF(kwdict);
4494 Py_XDECREF(stararg);
4495 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004496}
4497
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004498/* Extract a slice index from a PyInt or PyLong or an object with the
4499 nb_index slot defined, and store in *pi.
4500 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4501 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 +00004502 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004503*/
Tim Petersb5196382001-12-16 19:44:20 +00004504/* Note: If v is NULL, return success without storing into *pi. This
4505 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4506 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004507*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004508int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004509_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004510{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004511 if (v != NULL) {
4512 Py_ssize_t x;
4513 if (PyIndex_Check(v)) {
4514 x = PyNumber_AsSsize_t(v, NULL);
4515 if (x == -1 && PyErr_Occurred())
4516 return 0;
4517 }
4518 else {
4519 PyErr_SetString(PyExc_TypeError,
4520 "slice indices must be integers or "
4521 "None or have an __index__ method");
4522 return 0;
4523 }
4524 *pi = x;
4525 }
4526 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004527}
4528
Guido van Rossum486364b2007-06-30 05:01:58 +00004529#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004530 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00004531
Guido van Rossumb209a111997-04-29 18:18:01 +00004532static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004533cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004534{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004535 int res = 0;
4536 switch (op) {
4537 case PyCmp_IS:
4538 res = (v == w);
4539 break;
4540 case PyCmp_IS_NOT:
4541 res = (v != w);
4542 break;
4543 case PyCmp_IN:
4544 res = PySequence_Contains(w, v);
4545 if (res < 0)
4546 return NULL;
4547 break;
4548 case PyCmp_NOT_IN:
4549 res = PySequence_Contains(w, v);
4550 if (res < 0)
4551 return NULL;
4552 res = !res;
4553 break;
4554 case PyCmp_EXC_MATCH:
4555 if (PyTuple_Check(w)) {
4556 Py_ssize_t i, length;
4557 length = PyTuple_Size(w);
4558 for (i = 0; i < length; i += 1) {
4559 PyObject *exc = PyTuple_GET_ITEM(w, i);
4560 if (!PyExceptionClass_Check(exc)) {
4561 PyErr_SetString(PyExc_TypeError,
4562 CANNOT_CATCH_MSG);
4563 return NULL;
4564 }
4565 }
4566 }
4567 else {
4568 if (!PyExceptionClass_Check(w)) {
4569 PyErr_SetString(PyExc_TypeError,
4570 CANNOT_CATCH_MSG);
4571 return NULL;
4572 }
4573 }
4574 res = PyErr_GivenExceptionMatches(v, w);
4575 break;
4576 default:
4577 return PyObject_RichCompare(v, w, op);
4578 }
4579 v = res ? Py_True : Py_False;
4580 Py_INCREF(v);
4581 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004582}
4583
Thomas Wouters52152252000-08-17 22:55:00 +00004584static PyObject *
4585import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004586{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004587 PyObject *x;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004589 x = PyObject_GetAttr(v, name);
4590 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
4591 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
4592 }
4593 return x;
Thomas Wouters52152252000-08-17 22:55:00 +00004594}
Guido van Rossumac7be682001-01-17 15:42:30 +00004595
Thomas Wouters52152252000-08-17 22:55:00 +00004596static int
4597import_all_from(PyObject *locals, PyObject *v)
4598{
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004599 _Py_IDENTIFIER(__all__);
4600 _Py_IDENTIFIER(__dict__);
4601 PyObject *all = _PyObject_GetAttrId(v, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004602 PyObject *dict, *name, *value;
4603 int skip_leading_underscores = 0;
4604 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004606 if (all == NULL) {
4607 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4608 return -1; /* Unexpected error */
4609 PyErr_Clear();
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004610 dict = _PyObject_GetAttrId(v, &PyId___dict__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004611 if (dict == NULL) {
4612 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4613 return -1;
4614 PyErr_SetString(PyExc_ImportError,
4615 "from-import-* object has no __dict__ and no __all__");
4616 return -1;
4617 }
4618 all = PyMapping_Keys(dict);
4619 Py_DECREF(dict);
4620 if (all == NULL)
4621 return -1;
4622 skip_leading_underscores = 1;
4623 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004624
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004625 for (pos = 0, err = 0; ; pos++) {
4626 name = PySequence_GetItem(all, pos);
4627 if (name == NULL) {
4628 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4629 err = -1;
4630 else
4631 PyErr_Clear();
4632 break;
4633 }
4634 if (skip_leading_underscores &&
4635 PyUnicode_Check(name) &&
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004636 PyUnicode_READY(name) != -1 &&
4637 PyUnicode_READ_CHAR(name, 0) == '_')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004638 {
4639 Py_DECREF(name);
4640 continue;
4641 }
4642 value = PyObject_GetAttr(v, name);
4643 if (value == NULL)
4644 err = -1;
4645 else if (PyDict_CheckExact(locals))
4646 err = PyDict_SetItem(locals, name, value);
4647 else
4648 err = PyObject_SetItem(locals, name, value);
4649 Py_DECREF(name);
4650 Py_XDECREF(value);
4651 if (err != 0)
4652 break;
4653 }
4654 Py_DECREF(all);
4655 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004656}
4657
Guido van Rossumac7be682001-01-17 15:42:30 +00004658static void
Neal Norwitzda059e32007-08-26 05:33:45 +00004659format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00004660{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004661 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00004662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004663 if (!obj)
4664 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004666 obj_str = _PyUnicode_AsString(obj);
4667 if (!obj_str)
4668 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004669
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004670 PyErr_Format(exc, format_str, obj_str);
Paul Prescode68140d2000-08-30 20:25:01 +00004671}
Guido van Rossum950361c1997-01-24 13:49:28 +00004672
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00004673static void
4674format_exc_unbound(PyCodeObject *co, int oparg)
4675{
4676 PyObject *name;
4677 /* Don't stomp existing exception */
4678 if (PyErr_Occurred())
4679 return;
4680 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
4681 name = PyTuple_GET_ITEM(co->co_cellvars,
4682 oparg);
4683 format_exc_check_arg(
4684 PyExc_UnboundLocalError,
4685 UNBOUNDLOCAL_ERROR_MSG,
4686 name);
4687 } else {
4688 name = PyTuple_GET_ITEM(co->co_freevars, oparg -
4689 PyTuple_GET_SIZE(co->co_cellvars));
4690 format_exc_check_arg(PyExc_NameError,
4691 UNBOUNDFREE_ERROR_MSG, name);
4692 }
4693}
4694
Victor Stinnerd2a915d2011-10-02 20:34:20 +02004695static PyObject *
4696unicode_concatenate(PyObject *v, PyObject *w,
4697 PyFrameObject *f, unsigned char *next_instr)
4698{
4699 PyObject *res;
4700 if (Py_REFCNT(v) == 2) {
4701 /* In the common case, there are 2 references to the value
4702 * stored in 'variable' when the += is performed: one on the
4703 * value stack (in 'v') and one still stored in the
4704 * 'variable'. We try to delete the variable now to reduce
4705 * the refcnt to 1.
4706 */
4707 switch (*next_instr) {
4708 case STORE_FAST:
4709 {
4710 int oparg = PEEKARG();
4711 PyObject **fastlocals = f->f_localsplus;
4712 if (GETLOCAL(oparg) == v)
4713 SETLOCAL(oparg, NULL);
4714 break;
4715 }
4716 case STORE_DEREF:
4717 {
4718 PyObject **freevars = (f->f_localsplus +
4719 f->f_code->co_nlocals);
4720 PyObject *c = freevars[PEEKARG()];
4721 if (PyCell_GET(c) == v)
4722 PyCell_Set(c, NULL);
4723 break;
4724 }
4725 case STORE_NAME:
4726 {
4727 PyObject *names = f->f_code->co_names;
4728 PyObject *name = GETITEM(names, PEEKARG());
4729 PyObject *locals = f->f_locals;
4730 if (PyDict_CheckExact(locals) &&
4731 PyDict_GetItem(locals, name) == v) {
4732 if (PyDict_DelItem(locals, name) != 0) {
4733 PyErr_Clear();
4734 }
4735 }
4736 break;
4737 }
4738 }
4739 }
4740 res = v;
4741 PyUnicode_Append(&res, w);
4742 return res;
4743}
4744
Guido van Rossum950361c1997-01-24 13:49:28 +00004745#ifdef DYNAMIC_EXECUTION_PROFILE
4746
Skip Montanarof118cb12001-10-15 20:51:38 +00004747static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004748getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004749{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004750 int i;
4751 PyObject *l = PyList_New(256);
4752 if (l == NULL) return NULL;
4753 for (i = 0; i < 256; i++) {
4754 PyObject *x = PyLong_FromLong(a[i]);
4755 if (x == NULL) {
4756 Py_DECREF(l);
4757 return NULL;
4758 }
4759 PyList_SetItem(l, i, x);
4760 }
4761 for (i = 0; i < 256; i++)
4762 a[i] = 0;
4763 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004764}
4765
4766PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004767_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004768{
4769#ifndef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770 return getarray(dxp);
Guido van Rossum950361c1997-01-24 13:49:28 +00004771#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004772 int i;
4773 PyObject *l = PyList_New(257);
4774 if (l == NULL) return NULL;
4775 for (i = 0; i < 257; i++) {
4776 PyObject *x = getarray(dxpairs[i]);
4777 if (x == NULL) {
4778 Py_DECREF(l);
4779 return NULL;
4780 }
4781 PyList_SetItem(l, i, x);
4782 }
4783 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004784#endif
4785}
4786
4787#endif