blob: 8a732a0f12516a2d1c9783f536b5dab4326eee48 [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
Victor Stinnerace47d72013-07-18 01:41:08 +02001206#ifdef Py_DEBUG
1207 /* PyEval_EvalFrameEx() must not be called with an exception set,
1208 because it may clear it (directly or indirectly) and so the
1209 caller looses its exception */
1210 assert(!PyErr_Occurred());
1211#endif
1212
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001214#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001215 if (inst1 == 0) {
1216 /* Almost surely, the opcode executed a break
1217 or a continue, preventing inst1 from being set
1218 on the way out of the loop.
1219 */
1220 READ_TIMESTAMP(inst1);
1221 loop1 = inst1;
1222 }
1223 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
1224 intr0, intr1);
1225 ticked = 0;
1226 inst1 = 0;
1227 intr0 = 0;
1228 intr1 = 0;
1229 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001230#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001231 assert(stack_pointer >= f->f_valuestack); /* else underflow */
1232 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Victor Stinnerace47d72013-07-18 01:41:08 +02001233 assert(!PyErr_Occurred());
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001234
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001235 /* Do periodic things. Doing this every time through
1236 the loop would add too much overhead, so we do it
1237 only every Nth instruction. We also do it if
1238 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
1239 event needs attention (e.g. a signal handler or
1240 async I/O handler); see Py_AddPendingCall() and
1241 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +00001242
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001243 if (_Py_atomic_load_relaxed(&eval_breaker)) {
1244 if (*next_instr == SETUP_FINALLY) {
1245 /* Make the last opcode before
Ezio Melotti13925002011-03-16 11:05:33 +02001246 a try: finally: block uninterruptible. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001247 goto fast_next_opcode;
1248 }
1249 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001250#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 ticked = 1;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001252#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001253 if (_Py_atomic_load_relaxed(&pendingcalls_to_do)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001254 if (Py_MakePendingCalls() < 0)
1255 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001256 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001257#ifdef WITH_THREAD
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001258 if (_Py_atomic_load_relaxed(&gil_drop_request)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 /* Give another thread a chance */
1260 if (PyThreadState_Swap(NULL) != tstate)
1261 Py_FatalError("ceval: tstate mix-up");
1262 drop_gil(tstate);
1263
1264 /* Other threads may run now */
1265
1266 take_gil(tstate);
1267 if (PyThreadState_Swap(tstate) != NULL)
1268 Py_FatalError("ceval: orphan tstate");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001269 }
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001270#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001271 /* Check for asynchronous exceptions. */
1272 if (tstate->async_exc != NULL) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001273 PyObject *exc = tstate->async_exc;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001274 tstate->async_exc = NULL;
1275 UNSIGNAL_ASYNC_EXC();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001276 PyErr_SetNone(exc);
1277 Py_DECREF(exc);
1278 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001279 }
1280 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001281
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001282 fast_next_opcode:
1283 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001284
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001285 /* line-by-line tracing support */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001287 if (_Py_TracingPossible &&
Benjamin Peterson51f46162013-01-23 08:38:47 -05001288 tstate->c_tracefunc != NULL && !tstate->tracing) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001289 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001290 /* see maybe_call_line_trace
1291 for expository comments */
1292 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001294 err = maybe_call_line_trace(tstate->c_tracefunc,
1295 tstate->c_traceobj,
1296 f, &instr_lb, &instr_ub,
1297 &instr_prev);
1298 /* Reload possibly changed frame fields */
1299 JUMPTO(f->f_lasti);
1300 if (f->f_stacktop != NULL) {
1301 stack_pointer = f->f_stacktop;
1302 f->f_stacktop = NULL;
1303 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001304 if (err)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001305 /* trace function raised an exception */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001306 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001307 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001308
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001309 /* Extract opcode and argument */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001310
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001311 opcode = NEXTOP();
1312 oparg = 0; /* allows oparg to be stored in a register because
1313 it doesn't have to be remembered across a full loop */
1314 if (HAS_ARG(opcode))
1315 oparg = NEXTARG();
Stefan Krahb7e10102010-06-23 18:42:39 +00001316 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001317#ifdef DYNAMIC_EXECUTION_PROFILE
1318#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001319 dxpairs[lastopcode][opcode]++;
1320 lastopcode = opcode;
Guido van Rossum950361c1997-01-24 13:49:28 +00001321#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001322 dxp[opcode]++;
Guido van Rossum950361c1997-01-24 13:49:28 +00001323#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001324
Guido van Rossum96a42c81992-01-12 02:29:51 +00001325#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001326 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001327
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001328 if (lltrace) {
1329 if (HAS_ARG(opcode)) {
1330 printf("%d: %d, %d\n",
1331 f->f_lasti, opcode, oparg);
1332 }
1333 else {
1334 printf("%d: %d\n",
1335 f->f_lasti, opcode);
1336 }
1337 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001338#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001339
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001340 /* Main switch on opcode */
1341 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001342
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001343 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001344
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001345 /* BEWARE!
1346 It is essential that any operation that fails sets either
1347 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1348 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001349
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001350 TARGET(NOP)
1351 FAST_DISPATCH();
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001352
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001353 TARGET(LOAD_FAST) {
1354 PyObject *value = GETLOCAL(oparg);
1355 if (value == NULL) {
1356 format_exc_check_arg(PyExc_UnboundLocalError,
1357 UNBOUNDLOCAL_ERROR_MSG,
1358 PyTuple_GetItem(co->co_varnames, oparg));
1359 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001360 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001361 Py_INCREF(value);
1362 PUSH(value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001363 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001364 }
1365
1366 TARGET(LOAD_CONST) {
1367 PyObject *value = GETITEM(consts, oparg);
1368 Py_INCREF(value);
1369 PUSH(value);
1370 FAST_DISPATCH();
1371 }
Neil Schemenauer63543862002-02-17 19:10:14 +00001372
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001373 PREDICTED_WITH_ARG(STORE_FAST);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001374 TARGET(STORE_FAST) {
1375 PyObject *value = POP();
1376 SETLOCAL(oparg, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001377 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001378 }
Neil Schemenauer63543862002-02-17 19:10:14 +00001379
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001380 TARGET(POP_TOP) {
1381 PyObject *value = POP();
1382 Py_DECREF(value);
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_TWO) {
1387 PyObject *top = TOP();
1388 PyObject *second = SECOND();
1389 SET_TOP(second);
1390 SET_SECOND(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001391 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001392 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001393
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001394 TARGET(ROT_THREE) {
1395 PyObject *top = TOP();
1396 PyObject *second = SECOND();
1397 PyObject *third = THIRD();
1398 SET_TOP(second);
1399 SET_SECOND(third);
1400 SET_THIRD(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001401 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001402 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001403
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001404 TARGET(DUP_TOP) {
1405 PyObject *top = TOP();
1406 Py_INCREF(top);
1407 PUSH(top);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001408 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001409 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001410
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001411 TARGET(DUP_TOP_TWO) {
1412 PyObject *top = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001413 PyObject *second = SECOND();
Benjamin Petersonf208df32012-10-12 11:37:56 -04001414 Py_INCREF(top);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001415 Py_INCREF(second);
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001416 STACKADJ(2);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001417 SET_TOP(top);
1418 SET_SECOND(second);
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001419 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001420 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001421
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001422 TARGET(UNARY_POSITIVE) {
1423 PyObject *value = TOP();
1424 PyObject *res = PyNumber_Positive(value);
1425 Py_DECREF(value);
1426 SET_TOP(res);
1427 if (res == NULL)
1428 goto error;
1429 DISPATCH();
1430 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001431
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001432 TARGET(UNARY_NEGATIVE) {
1433 PyObject *value = TOP();
1434 PyObject *res = PyNumber_Negative(value);
1435 Py_DECREF(value);
1436 SET_TOP(res);
1437 if (res == NULL)
1438 goto error;
1439 DISPATCH();
1440 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001441
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001442 TARGET(UNARY_NOT) {
1443 PyObject *value = TOP();
1444 int err = PyObject_IsTrue(value);
1445 Py_DECREF(value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 if (err == 0) {
1447 Py_INCREF(Py_True);
1448 SET_TOP(Py_True);
1449 DISPATCH();
1450 }
1451 else if (err > 0) {
1452 Py_INCREF(Py_False);
1453 SET_TOP(Py_False);
1454 err = 0;
1455 DISPATCH();
1456 }
1457 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001458 goto error;
1459 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001460
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001461 TARGET(UNARY_INVERT) {
1462 PyObject *value = TOP();
1463 PyObject *res = PyNumber_Invert(value);
1464 Py_DECREF(value);
1465 SET_TOP(res);
1466 if (res == NULL)
1467 goto error;
1468 DISPATCH();
1469 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001470
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001471 TARGET(BINARY_POWER) {
1472 PyObject *exp = POP();
1473 PyObject *base = TOP();
1474 PyObject *res = PyNumber_Power(base, exp, Py_None);
1475 Py_DECREF(base);
1476 Py_DECREF(exp);
1477 SET_TOP(res);
1478 if (res == NULL)
1479 goto error;
1480 DISPATCH();
1481 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001482
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001483 TARGET(BINARY_MULTIPLY) {
1484 PyObject *right = POP();
1485 PyObject *left = TOP();
1486 PyObject *res = PyNumber_Multiply(left, right);
1487 Py_DECREF(left);
1488 Py_DECREF(right);
1489 SET_TOP(res);
1490 if (res == NULL)
1491 goto error;
1492 DISPATCH();
1493 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001494
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001495 TARGET(BINARY_TRUE_DIVIDE) {
1496 PyObject *divisor = POP();
1497 PyObject *dividend = TOP();
1498 PyObject *quotient = PyNumber_TrueDivide(dividend, divisor);
1499 Py_DECREF(dividend);
1500 Py_DECREF(divisor);
1501 SET_TOP(quotient);
1502 if (quotient == NULL)
1503 goto error;
1504 DISPATCH();
1505 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001506
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001507 TARGET(BINARY_FLOOR_DIVIDE) {
1508 PyObject *divisor = POP();
1509 PyObject *dividend = TOP();
1510 PyObject *quotient = PyNumber_FloorDivide(dividend, divisor);
1511 Py_DECREF(dividend);
1512 Py_DECREF(divisor);
1513 SET_TOP(quotient);
1514 if (quotient == NULL)
1515 goto error;
1516 DISPATCH();
1517 }
Guido van Rossum4668b002001-08-08 05:00:18 +00001518
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001519 TARGET(BINARY_MODULO) {
1520 PyObject *divisor = POP();
1521 PyObject *dividend = TOP();
1522 PyObject *res = PyUnicode_CheckExact(dividend) ?
1523 PyUnicode_Format(dividend, divisor) :
1524 PyNumber_Remainder(dividend, divisor);
1525 Py_DECREF(divisor);
1526 Py_DECREF(dividend);
1527 SET_TOP(res);
1528 if (res == NULL)
1529 goto error;
1530 DISPATCH();
1531 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001532
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001533 TARGET(BINARY_ADD) {
1534 PyObject *right = POP();
1535 PyObject *left = TOP();
1536 PyObject *sum;
1537 if (PyUnicode_CheckExact(left) &&
1538 PyUnicode_CheckExact(right)) {
1539 sum = unicode_concatenate(left, right, f, next_instr);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001540 /* unicode_concatenate consumed the ref to v */
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001541 }
1542 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001543 sum = PyNumber_Add(left, right);
1544 Py_DECREF(left);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001545 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001546 Py_DECREF(right);
1547 SET_TOP(sum);
1548 if (sum == NULL)
1549 goto error;
1550 DISPATCH();
1551 }
1552
1553 TARGET(BINARY_SUBTRACT) {
1554 PyObject *right = POP();
1555 PyObject *left = TOP();
1556 PyObject *diff = PyNumber_Subtract(left, right);
1557 Py_DECREF(right);
1558 Py_DECREF(left);
1559 SET_TOP(diff);
1560 if (diff == NULL)
1561 goto error;
1562 DISPATCH();
1563 }
1564
1565 TARGET(BINARY_SUBSCR) {
1566 PyObject *sub = POP();
1567 PyObject *container = TOP();
1568 PyObject *res = PyObject_GetItem(container, sub);
1569 Py_DECREF(container);
1570 Py_DECREF(sub);
1571 SET_TOP(res);
1572 if (res == NULL)
1573 goto error;
1574 DISPATCH();
1575 }
1576
1577 TARGET(BINARY_LSHIFT) {
1578 PyObject *right = POP();
1579 PyObject *left = TOP();
1580 PyObject *res = PyNumber_Lshift(left, right);
1581 Py_DECREF(left);
1582 Py_DECREF(right);
1583 SET_TOP(res);
1584 if (res == NULL)
1585 goto error;
1586 DISPATCH();
1587 }
1588
1589 TARGET(BINARY_RSHIFT) {
1590 PyObject *right = POP();
1591 PyObject *left = TOP();
1592 PyObject *res = PyNumber_Rshift(left, right);
1593 Py_DECREF(left);
1594 Py_DECREF(right);
1595 SET_TOP(res);
1596 if (res == NULL)
1597 goto error;
1598 DISPATCH();
1599 }
1600
1601 TARGET(BINARY_AND) {
1602 PyObject *right = POP();
1603 PyObject *left = TOP();
1604 PyObject *res = PyNumber_And(left, right);
1605 Py_DECREF(left);
1606 Py_DECREF(right);
1607 SET_TOP(res);
1608 if (res == NULL)
1609 goto error;
1610 DISPATCH();
1611 }
1612
1613 TARGET(BINARY_XOR) {
1614 PyObject *right = POP();
1615 PyObject *left = TOP();
1616 PyObject *res = PyNumber_Xor(left, right);
1617 Py_DECREF(left);
1618 Py_DECREF(right);
1619 SET_TOP(res);
1620 if (res == NULL)
1621 goto error;
1622 DISPATCH();
1623 }
1624
1625 TARGET(BINARY_OR) {
1626 PyObject *right = POP();
1627 PyObject *left = TOP();
1628 PyObject *res = PyNumber_Or(left, right);
1629 Py_DECREF(left);
1630 Py_DECREF(right);
1631 SET_TOP(res);
1632 if (res == NULL)
1633 goto error;
1634 DISPATCH();
1635 }
1636
1637 TARGET(LIST_APPEND) {
1638 PyObject *v = POP();
1639 PyObject *list = PEEK(oparg);
1640 int err;
1641 err = PyList_Append(list, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001643 if (err != 0)
1644 goto error;
1645 PREDICT(JUMP_ABSOLUTE);
1646 DISPATCH();
1647 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001648
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001649 TARGET(SET_ADD) {
1650 PyObject *v = POP();
1651 PyObject *set = stack_pointer[-oparg];
1652 int err;
1653 err = PySet_Add(set, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001654 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001655 if (err != 0)
1656 goto error;
1657 PREDICT(JUMP_ABSOLUTE);
1658 DISPATCH();
1659 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001660
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001661 TARGET(INPLACE_POWER) {
1662 PyObject *exp = POP();
1663 PyObject *base = TOP();
1664 PyObject *res = PyNumber_InPlacePower(base, exp, Py_None);
1665 Py_DECREF(base);
1666 Py_DECREF(exp);
1667 SET_TOP(res);
1668 if (res == NULL)
1669 goto error;
1670 DISPATCH();
1671 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001672
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001673 TARGET(INPLACE_MULTIPLY) {
1674 PyObject *right = POP();
1675 PyObject *left = TOP();
1676 PyObject *res = PyNumber_InPlaceMultiply(left, right);
1677 Py_DECREF(left);
1678 Py_DECREF(right);
1679 SET_TOP(res);
1680 if (res == NULL)
1681 goto error;
1682 DISPATCH();
1683 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001684
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001685 TARGET(INPLACE_TRUE_DIVIDE) {
1686 PyObject *divisor = POP();
1687 PyObject *dividend = TOP();
1688 PyObject *quotient = PyNumber_InPlaceTrueDivide(dividend, divisor);
1689 Py_DECREF(dividend);
1690 Py_DECREF(divisor);
1691 SET_TOP(quotient);
1692 if (quotient == NULL)
1693 goto error;
1694 DISPATCH();
1695 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001696
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001697 TARGET(INPLACE_FLOOR_DIVIDE) {
1698 PyObject *divisor = POP();
1699 PyObject *dividend = TOP();
1700 PyObject *quotient = PyNumber_InPlaceFloorDivide(dividend, divisor);
1701 Py_DECREF(dividend);
1702 Py_DECREF(divisor);
1703 SET_TOP(quotient);
1704 if (quotient == NULL)
1705 goto error;
1706 DISPATCH();
1707 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001708
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001709 TARGET(INPLACE_MODULO) {
1710 PyObject *right = POP();
1711 PyObject *left = TOP();
1712 PyObject *mod = PyNumber_InPlaceRemainder(left, right);
1713 Py_DECREF(left);
1714 Py_DECREF(right);
1715 SET_TOP(mod);
1716 if (mod == NULL)
1717 goto error;
1718 DISPATCH();
1719 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001720
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001721 TARGET(INPLACE_ADD) {
1722 PyObject *right = POP();
1723 PyObject *left = TOP();
1724 PyObject *sum;
1725 if (PyUnicode_CheckExact(left) && PyUnicode_CheckExact(right)) {
1726 sum = unicode_concatenate(left, right, f, next_instr);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001727 /* unicode_concatenate consumed the ref to v */
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001728 }
1729 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001730 sum = PyNumber_InPlaceAdd(left, right);
1731 Py_DECREF(left);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001732 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001733 Py_DECREF(right);
1734 SET_TOP(sum);
1735 if (sum == NULL)
1736 goto error;
1737 DISPATCH();
1738 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001739
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001740 TARGET(INPLACE_SUBTRACT) {
1741 PyObject *right = POP();
1742 PyObject *left = TOP();
1743 PyObject *diff = PyNumber_InPlaceSubtract(left, right);
1744 Py_DECREF(left);
1745 Py_DECREF(right);
1746 SET_TOP(diff);
1747 if (diff == NULL)
1748 goto error;
1749 DISPATCH();
1750 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001751
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001752 TARGET(INPLACE_LSHIFT) {
1753 PyObject *right = POP();
1754 PyObject *left = TOP();
1755 PyObject *res = PyNumber_InPlaceLshift(left, right);
1756 Py_DECREF(left);
1757 Py_DECREF(right);
1758 SET_TOP(res);
1759 if (res == NULL)
1760 goto error;
1761 DISPATCH();
1762 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001763
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001764 TARGET(INPLACE_RSHIFT) {
1765 PyObject *right = POP();
1766 PyObject *left = TOP();
1767 PyObject *res = PyNumber_InPlaceRshift(left, right);
1768 Py_DECREF(left);
1769 Py_DECREF(right);
1770 SET_TOP(res);
1771 if (res == NULL)
1772 goto error;
1773 DISPATCH();
1774 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001775
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001776 TARGET(INPLACE_AND) {
1777 PyObject *right = POP();
1778 PyObject *left = TOP();
1779 PyObject *res = PyNumber_InPlaceAnd(left, right);
1780 Py_DECREF(left);
1781 Py_DECREF(right);
1782 SET_TOP(res);
1783 if (res == NULL)
1784 goto error;
1785 DISPATCH();
1786 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001787
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001788 TARGET(INPLACE_XOR) {
1789 PyObject *right = POP();
1790 PyObject *left = TOP();
1791 PyObject *res = PyNumber_InPlaceXor(left, right);
1792 Py_DECREF(left);
1793 Py_DECREF(right);
1794 SET_TOP(res);
1795 if (res == NULL)
1796 goto error;
1797 DISPATCH();
1798 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001799
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001800 TARGET(INPLACE_OR) {
1801 PyObject *right = POP();
1802 PyObject *left = TOP();
1803 PyObject *res = PyNumber_InPlaceOr(left, right);
1804 Py_DECREF(left);
1805 Py_DECREF(right);
1806 SET_TOP(res);
1807 if (res == NULL)
1808 goto error;
1809 DISPATCH();
1810 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001811
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001812 TARGET(STORE_SUBSCR) {
1813 PyObject *sub = TOP();
1814 PyObject *container = SECOND();
1815 PyObject *v = THIRD();
1816 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001817 STACKADJ(-3);
1818 /* v[w] = u */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001819 err = PyObject_SetItem(container, sub, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001820 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001821 Py_DECREF(container);
1822 Py_DECREF(sub);
1823 if (err != 0)
1824 goto error;
1825 DISPATCH();
1826 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001827
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001828 TARGET(DELETE_SUBSCR) {
1829 PyObject *sub = TOP();
1830 PyObject *container = SECOND();
1831 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001832 STACKADJ(-2);
1833 /* del v[w] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001834 err = PyObject_DelItem(container, sub);
1835 Py_DECREF(container);
1836 Py_DECREF(sub);
1837 if (err != 0)
1838 goto error;
1839 DISPATCH();
1840 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001841
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001842 TARGET(PRINT_EXPR) {
1843 PyObject *value = POP();
1844 PyObject *hook = PySys_GetObject("displayhook");
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04001845 PyObject *res;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001846 if (hook == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001847 PyErr_SetString(PyExc_RuntimeError,
1848 "lost sys.displayhook");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001849 Py_DECREF(value);
1850 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001851 }
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04001852 res = PyObject_CallFunctionObjArgs(hook, value, NULL);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001853 Py_DECREF(value);
1854 if (res == NULL)
1855 goto error;
1856 Py_DECREF(res);
1857 DISPATCH();
1858 }
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001859
Thomas Wouters434d0822000-08-24 20:11:32 +00001860#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001861 default: switch (opcode) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001862#endif
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001863 TARGET(RAISE_VARARGS) {
1864 PyObject *cause = NULL, *exc = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001865 switch (oparg) {
1866 case 2:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001867 cause = POP(); /* cause */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001868 case 1:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001869 exc = POP(); /* exc */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001870 case 0: /* Fallthrough */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001871 if (do_raise(exc, cause)) {
1872 why = WHY_EXCEPTION;
1873 goto fast_block_end;
1874 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001875 break;
1876 default:
1877 PyErr_SetString(PyExc_SystemError,
1878 "bad RAISE_VARARGS oparg");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001879 break;
1880 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001881 goto error;
1882 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001883
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001884 TARGET(RETURN_VALUE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001885 retval = POP();
1886 why = WHY_RETURN;
1887 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001888 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001889
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001890 TARGET(YIELD_FROM) {
1891 PyObject *v = POP();
1892 PyObject *reciever = TOP();
1893 int err;
1894 if (PyGen_CheckExact(reciever)) {
1895 retval = _PyGen_Send((PyGenObject *)reciever, v);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001896 } else {
Benjamin Peterson302e7902012-03-20 23:17:04 -04001897 _Py_IDENTIFIER(send);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001898 if (v == Py_None)
1899 retval = Py_TYPE(reciever)->tp_iternext(reciever);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001900 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001901 retval = _PyObject_CallMethodId(reciever, &PyId_send, "O", v);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001902 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001903 Py_DECREF(v);
1904 if (retval == NULL) {
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001905 PyObject *val;
Nick Coghlanc40bc092012-06-17 15:15:49 +10001906 err = _PyGen_FetchStopIterationValue(&val);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001907 if (err < 0)
1908 goto error;
1909 Py_DECREF(reciever);
1910 SET_TOP(val);
1911 DISPATCH();
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001912 }
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001913 /* x remains on stack, retval is value to be yielded */
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001914 f->f_stacktop = stack_pointer;
1915 why = WHY_YIELD;
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001916 /* and repeat... */
1917 f->f_lasti--;
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001918 goto fast_yield;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001919 }
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001920
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001921 TARGET(YIELD_VALUE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001922 retval = POP();
1923 f->f_stacktop = stack_pointer;
1924 why = WHY_YIELD;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001925 goto fast_yield;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001926 }
Tim Peters5ca576e2001-06-18 22:08:13 +00001927
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001928 TARGET(POP_EXCEPT) {
1929 PyTryBlock *b = PyFrame_BlockPop(f);
1930 if (b->b_type != EXCEPT_HANDLER) {
1931 PyErr_SetString(PyExc_SystemError,
1932 "popped block is not an except handler");
1933 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001934 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001935 UNWIND_EXCEPT_HANDLER(b);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001936 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001937 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001938
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001939 TARGET(POP_BLOCK) {
1940 PyTryBlock *b = PyFrame_BlockPop(f);
1941 UNWIND_BLOCK(b);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001942 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001943 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001944
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001945 PREDICTED(END_FINALLY);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001946 TARGET(END_FINALLY) {
1947 PyObject *status = POP();
1948 if (PyLong_Check(status)) {
1949 why = (enum why_code) PyLong_AS_LONG(status);
1950 assert(why != WHY_YIELD && why != WHY_EXCEPTION);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001951 if (why == WHY_RETURN ||
1952 why == WHY_CONTINUE)
1953 retval = POP();
1954 if (why == WHY_SILENCED) {
1955 /* An exception was silenced by 'with', we must
1956 manually unwind the EXCEPT_HANDLER block which was
1957 created when the exception was caught, otherwise
1958 the stack will be in an inconsistent state. */
1959 PyTryBlock *b = PyFrame_BlockPop(f);
1960 assert(b->b_type == EXCEPT_HANDLER);
1961 UNWIND_EXCEPT_HANDLER(b);
1962 why = WHY_NOT;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001963 Py_DECREF(status);
1964 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001965 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001966 Py_DECREF(status);
1967 goto fast_block_end;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001968 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001969 else if (PyExceptionClass_Check(status)) {
1970 PyObject *exc = POP();
1971 PyObject *tb = POP();
1972 PyErr_Restore(status, exc, tb);
1973 why = WHY_EXCEPTION;
1974 goto fast_block_end;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001975 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001976 else if (status != Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001977 PyErr_SetString(PyExc_SystemError,
1978 "'finally' pops bad exception");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001979 Py_DECREF(status);
1980 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001981 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001982 Py_DECREF(status);
1983 DISPATCH();
1984 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001985
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001986 TARGET(LOAD_BUILD_CLASS) {
Victor Stinner3c1e4812012-03-26 22:10:51 +02001987 _Py_IDENTIFIER(__build_class__);
Victor Stinnerb0b22422012-04-19 00:57:45 +02001988
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001989 PyObject *bc;
Victor Stinnerb0b22422012-04-19 00:57:45 +02001990 if (PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001991 bc = _PyDict_GetItemId(f->f_builtins, &PyId___build_class__);
1992 if (bc == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02001993 PyErr_SetString(PyExc_NameError,
1994 "__build_class__ not found");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001995 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02001996 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04001997 Py_INCREF(bc);
Victor Stinnerb0b22422012-04-19 00:57:45 +02001998 }
1999 else {
2000 PyObject *build_class_str = _PyUnicode_FromId(&PyId___build_class__);
2001 if (build_class_str == NULL)
2002 break;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002003 bc = PyObject_GetItem(f->f_builtins, build_class_str);
2004 if (bc == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002005 if (PyErr_ExceptionMatches(PyExc_KeyError))
2006 PyErr_SetString(PyExc_NameError,
2007 "__build_class__ not found");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002008 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002009 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002010 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002011 PUSH(bc);
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002012 DISPATCH();
Victor Stinner3c1e4812012-03-26 22:10:51 +02002013 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002014
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002015 TARGET(STORE_NAME) {
2016 PyObject *name = GETITEM(names, oparg);
2017 PyObject *v = POP();
2018 PyObject *ns = f->f_locals;
2019 int err;
2020 if (ns == NULL) {
2021 PyErr_Format(PyExc_SystemError,
2022 "no locals found when storing %R", name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002023 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002024 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002025 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002026 if (PyDict_CheckExact(ns))
2027 err = PyDict_SetItem(ns, name, v);
2028 else
2029 err = PyObject_SetItem(ns, name, v);
2030 Py_DECREF(v);
2031 if (err != 0)
2032 goto error;
2033 DISPATCH();
2034 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002035
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002036 TARGET(DELETE_NAME) {
2037 PyObject *name = GETITEM(names, oparg);
2038 PyObject *ns = f->f_locals;
2039 int err;
2040 if (ns == NULL) {
2041 PyErr_Format(PyExc_SystemError,
2042 "no locals when deleting %R", name);
2043 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002044 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002045 err = PyObject_DelItem(ns, name);
2046 if (err != 0) {
2047 format_exc_check_arg(PyExc_NameError,
2048 NAME_ERROR_MSG,
2049 name);
2050 goto error;
2051 }
2052 DISPATCH();
2053 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00002054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002055 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002056 TARGET(UNPACK_SEQUENCE) {
2057 PyObject *seq = POP(), *item, **items;
2058 if (PyTuple_CheckExact(seq) &&
2059 PyTuple_GET_SIZE(seq) == oparg) {
2060 items = ((PyTupleObject *)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 (PyList_CheckExact(seq) &&
2067 PyList_GET_SIZE(seq) == oparg) {
2068 items = ((PyListObject *)seq)->ob_item;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002069 while (oparg--) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002070 item = items[oparg];
2071 Py_INCREF(item);
2072 PUSH(item);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002073 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002074 } else if (unpack_iterable(seq, oparg, -1,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002075 stack_pointer + oparg)) {
2076 STACKADJ(oparg);
2077 } else {
2078 /* unpack_iterable() raised an exception */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002079 Py_DECREF(seq);
2080 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002081 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002082 Py_DECREF(seq);
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002083 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002084 }
Guido van Rossum0368b722007-05-11 16:50:42 +00002085
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002086 TARGET(UNPACK_EX) {
2087 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
2088 PyObject *seq = POP();
2089
2090 if (unpack_iterable(seq, oparg & 0xFF, oparg >> 8,
2091 stack_pointer + totalargs)) {
2092 stack_pointer += totalargs;
2093 } else {
2094 Py_DECREF(seq);
2095 goto error;
2096 }
2097 Py_DECREF(seq);
2098 DISPATCH();
2099 }
2100
2101 TARGET(STORE_ATTR) {
2102 PyObject *name = GETITEM(names, oparg);
2103 PyObject *owner = TOP();
2104 PyObject *v = SECOND();
2105 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002106 STACKADJ(-2);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002107 err = PyObject_SetAttr(owner, name, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002108 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002109 Py_DECREF(owner);
2110 if (err != 0)
2111 goto error;
2112 DISPATCH();
2113 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002114
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002115 TARGET(DELETE_ATTR) {
2116 PyObject *name = GETITEM(names, oparg);
2117 PyObject *owner = POP();
2118 int err;
2119 err = PyObject_SetAttr(owner, name, (PyObject *)NULL);
2120 Py_DECREF(owner);
2121 if (err != 0)
2122 goto error;
2123 DISPATCH();
2124 }
2125
2126 TARGET(STORE_GLOBAL) {
2127 PyObject *name = GETITEM(names, oparg);
2128 PyObject *v = POP();
2129 int err;
2130 err = PyDict_SetItem(f->f_globals, name, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002131 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002132 if (err != 0)
2133 goto error;
2134 DISPATCH();
2135 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002136
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002137 TARGET(DELETE_GLOBAL) {
2138 PyObject *name = GETITEM(names, oparg);
2139 int err;
2140 err = PyDict_DelItem(f->f_globals, name);
2141 if (err != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002142 format_exc_check_arg(
Ezio Melotti04a29552013-03-03 15:12:44 +02002143 PyExc_NameError, NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002144 goto error;
Benjamin Peterson00f86f22012-10-10 14:10:33 -04002145 }
2146 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002147 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002148
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002149 TARGET(LOAD_NAME) {
2150 PyObject *name = GETITEM(names, oparg);
2151 PyObject *locals = f->f_locals;
2152 PyObject *v;
2153 if (locals == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002154 PyErr_Format(PyExc_SystemError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002155 "no locals when loading %R", name);
2156 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002157 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002158 if (PyDict_CheckExact(locals)) {
2159 v = PyDict_GetItem(locals, name);
2160 Py_XINCREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002161 }
2162 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002163 v = PyObject_GetItem(locals, name);
2164 if (v == NULL && PyErr_Occurred()) {
Benjamin Peterson92722792012-12-15 12:51:05 -05002165 if (!PyErr_ExceptionMatches(PyExc_KeyError))
2166 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002167 PyErr_Clear();
2168 }
2169 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002170 if (v == NULL) {
2171 v = PyDict_GetItem(f->f_globals, name);
2172 Py_XINCREF(v);
2173 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002174 if (PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002175 v = PyDict_GetItem(f->f_builtins, name);
2176 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002177 format_exc_check_arg(
2178 PyExc_NameError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002179 NAME_ERROR_MSG, name);
2180 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002181 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002182 Py_INCREF(v);
Victor Stinnerb0b22422012-04-19 00:57:45 +02002183 }
2184 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002185 v = PyObject_GetItem(f->f_builtins, name);
2186 if (v == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002187 if (PyErr_ExceptionMatches(PyExc_KeyError))
2188 format_exc_check_arg(
2189 PyExc_NameError,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002190 NAME_ERROR_MSG, name);
2191 goto error;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002192 }
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002193 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002194 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002195 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002196 PUSH(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002197 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002198 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002199
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002200 TARGET(LOAD_GLOBAL) {
2201 PyObject *name = GETITEM(names, oparg);
2202 PyObject *v;
Victor Stinnerb0b22422012-04-19 00:57:45 +02002203 if (PyDict_CheckExact(f->f_globals)
2204 && PyDict_CheckExact(f->f_builtins)) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002205 v = _PyDict_LoadGlobal((PyDictObject *)f->f_globals,
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002206 (PyDictObject *)f->f_builtins,
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002207 name);
2208 if (v == NULL) {
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002209 if (!PyErr_Occurred())
2210 format_exc_check_arg(PyExc_NameError,
Ezio Melotti04a29552013-03-03 15:12:44 +02002211 NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002212 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002213 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002214 Py_INCREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002215 }
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002216 else {
2217 /* Slow-path if globals or builtins is not a dict */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002218 v = PyObject_GetItem(f->f_globals, name);
2219 if (v == NULL) {
2220 v = PyObject_GetItem(f->f_builtins, name);
2221 if (v == NULL) {
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002222 if (PyErr_ExceptionMatches(PyExc_KeyError))
2223 format_exc_check_arg(
2224 PyExc_NameError,
Ezio Melotti04a29552013-03-03 15:12:44 +02002225 NAME_ERROR_MSG, name);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002226 goto error;
Benjamin Peterson7d95e402012-04-23 11:24:50 -04002227 }
2228 }
2229 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002230 PUSH(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002231 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002232 }
Guido van Rossum681d79a1995-07-18 14:51:37 +00002233
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002234 TARGET(DELETE_FAST) {
2235 PyObject *v = GETLOCAL(oparg);
2236 if (v != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002237 SETLOCAL(oparg, NULL);
2238 DISPATCH();
2239 }
2240 format_exc_check_arg(
2241 PyExc_UnboundLocalError,
2242 UNBOUNDLOCAL_ERROR_MSG,
2243 PyTuple_GetItem(co->co_varnames, oparg)
2244 );
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002245 goto error;
2246 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002247
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002248 TARGET(DELETE_DEREF) {
2249 PyObject *cell = freevars[oparg];
2250 if (PyCell_GET(cell) != NULL) {
2251 PyCell_Set(cell, NULL);
Benjamin Peterson00ebe2c2010-09-10 22:02:31 +00002252 DISPATCH();
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002253 }
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002254 format_exc_unbound(co, oparg);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002255 goto error;
2256 }
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002257
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002258 TARGET(LOAD_CLOSURE) {
2259 PyObject *cell = freevars[oparg];
2260 Py_INCREF(cell);
2261 PUSH(cell);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002262 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002263 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002264
Benjamin Peterson3b0431d2013-04-30 09:41:40 -04002265 TARGET(LOAD_CLASSDEREF) {
2266 PyObject *name, *value, *locals = f->f_locals;
Victor Stinnerd3dfd0e2013-05-16 23:48:01 +02002267 Py_ssize_t idx;
Benjamin Peterson3b0431d2013-04-30 09:41:40 -04002268 assert(locals);
2269 assert(oparg >= PyTuple_GET_SIZE(co->co_cellvars));
2270 idx = oparg - PyTuple_GET_SIZE(co->co_cellvars);
2271 assert(idx >= 0 && idx < PyTuple_GET_SIZE(co->co_freevars));
2272 name = PyTuple_GET_ITEM(co->co_freevars, idx);
2273 if (PyDict_CheckExact(locals)) {
2274 value = PyDict_GetItem(locals, name);
2275 Py_XINCREF(value);
2276 }
2277 else {
2278 value = PyObject_GetItem(locals, name);
2279 if (value == NULL && PyErr_Occurred()) {
2280 if (!PyErr_ExceptionMatches(PyExc_KeyError))
2281 goto error;
2282 PyErr_Clear();
2283 }
2284 }
2285 if (!value) {
2286 PyObject *cell = freevars[oparg];
2287 value = PyCell_GET(cell);
2288 if (value == NULL) {
2289 format_exc_unbound(co, oparg);
2290 goto error;
2291 }
2292 Py_INCREF(value);
2293 }
2294 PUSH(value);
2295 DISPATCH();
2296 }
2297
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002298 TARGET(LOAD_DEREF) {
2299 PyObject *cell = freevars[oparg];
2300 PyObject *value = PyCell_GET(cell);
2301 if (value == NULL) {
2302 format_exc_unbound(co, oparg);
2303 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002304 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002305 Py_INCREF(value);
2306 PUSH(value);
2307 DISPATCH();
2308 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002309
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002310 TARGET(STORE_DEREF) {
2311 PyObject *v = POP();
2312 PyObject *cell = freevars[oparg];
2313 PyCell_Set(cell, v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002314 Py_DECREF(v);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002315 DISPATCH();
2316 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002317
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002318 TARGET(BUILD_TUPLE) {
2319 PyObject *tup = PyTuple_New(oparg);
2320 if (tup == NULL)
2321 goto error;
2322 while (--oparg >= 0) {
2323 PyObject *item = POP();
2324 PyTuple_SET_ITEM(tup, oparg, item);
2325 }
2326 PUSH(tup);
2327 DISPATCH();
2328 }
2329
2330 TARGET(BUILD_LIST) {
2331 PyObject *list = PyList_New(oparg);
2332 if (list == NULL)
2333 goto error;
2334 while (--oparg >= 0) {
2335 PyObject *item = POP();
2336 PyList_SET_ITEM(list, oparg, item);
2337 }
2338 PUSH(list);
2339 DISPATCH();
2340 }
2341
2342 TARGET(BUILD_SET) {
2343 PyObject *set = PySet_New(NULL);
2344 int err = 0;
2345 if (set == NULL)
2346 goto error;
2347 while (--oparg >= 0) {
2348 PyObject *item = POP();
2349 if (err == 0)
2350 err = PySet_Add(set, item);
2351 Py_DECREF(item);
2352 }
2353 if (err != 0) {
2354 Py_DECREF(set);
2355 goto error;
2356 }
2357 PUSH(set);
2358 DISPATCH();
2359 }
2360
2361 TARGET(BUILD_MAP) {
2362 PyObject *map = _PyDict_NewPresized((Py_ssize_t)oparg);
2363 if (map == NULL)
2364 goto error;
2365 PUSH(map);
2366 DISPATCH();
2367 }
2368
2369 TARGET(STORE_MAP) {
2370 PyObject *key = TOP();
2371 PyObject *value = SECOND();
2372 PyObject *map = THIRD();
2373 int err;
2374 STACKADJ(-2);
2375 assert(PyDict_CheckExact(map));
2376 err = PyDict_SetItem(map, key, value);
2377 Py_DECREF(value);
2378 Py_DECREF(key);
2379 if (err != 0)
2380 goto error;
2381 DISPATCH();
2382 }
2383
2384 TARGET(MAP_ADD) {
2385 PyObject *key = TOP();
2386 PyObject *value = SECOND();
2387 PyObject *map;
2388 int err;
2389 STACKADJ(-2);
2390 map = stack_pointer[-oparg]; /* dict */
2391 assert(PyDict_CheckExact(map));
2392 err = PyDict_SetItem(map, key, value); /* v[w] = u */
2393 Py_DECREF(value);
2394 Py_DECREF(key);
2395 if (err != 0)
2396 goto error;
2397 PREDICT(JUMP_ABSOLUTE);
2398 DISPATCH();
2399 }
2400
2401 TARGET(LOAD_ATTR) {
2402 PyObject *name = GETITEM(names, oparg);
2403 PyObject *owner = TOP();
2404 PyObject *res = PyObject_GetAttr(owner, name);
2405 Py_DECREF(owner);
2406 SET_TOP(res);
2407 if (res == NULL)
2408 goto error;
2409 DISPATCH();
2410 }
2411
2412 TARGET(COMPARE_OP) {
2413 PyObject *right = POP();
2414 PyObject *left = TOP();
2415 PyObject *res = cmp_outcome(oparg, left, right);
2416 Py_DECREF(left);
2417 Py_DECREF(right);
2418 SET_TOP(res);
2419 if (res == NULL)
2420 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002421 PREDICT(POP_JUMP_IF_FALSE);
2422 PREDICT(POP_JUMP_IF_TRUE);
2423 DISPATCH();
Victor Stinner3c1e4812012-03-26 22:10:51 +02002424 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002425
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002426 TARGET(IMPORT_NAME) {
2427 _Py_IDENTIFIER(__import__);
2428 PyObject *name = GETITEM(names, oparg);
2429 PyObject *func = _PyDict_GetItemId(f->f_builtins, &PyId___import__);
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04002430 PyObject *from, *level, *args, *res;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002431 if (func == NULL) {
2432 PyErr_SetString(PyExc_ImportError,
2433 "__import__ not found");
2434 goto error;
2435 }
2436 Py_INCREF(func);
2437 from = POP();
2438 level = TOP();
2439 if (PyLong_AsLong(level) != -1 || PyErr_Occurred())
2440 args = PyTuple_Pack(5,
2441 name,
2442 f->f_globals,
2443 f->f_locals == NULL ?
2444 Py_None : f->f_locals,
2445 from,
2446 level);
2447 else
2448 args = PyTuple_Pack(4,
2449 name,
2450 f->f_globals,
2451 f->f_locals == NULL ?
2452 Py_None : f->f_locals,
2453 from);
2454 Py_DECREF(level);
2455 Py_DECREF(from);
2456 if (args == NULL) {
2457 Py_DECREF(func);
2458 STACKADJ(-1);
2459 goto error;
2460 }
2461 READ_TIMESTAMP(intr0);
Benjamin Petersonfe1bcb62012-10-12 11:40:01 -04002462 res = PyEval_CallObject(func, args);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002463 READ_TIMESTAMP(intr1);
2464 Py_DECREF(args);
2465 Py_DECREF(func);
2466 SET_TOP(res);
2467 if (res == NULL)
2468 goto error;
2469 DISPATCH();
2470 }
2471
2472 TARGET(IMPORT_STAR) {
2473 PyObject *from = POP(), *locals;
2474 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002475 PyFrame_FastToLocals(f);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002476 locals = f->f_locals;
2477 if (locals == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002478 PyErr_SetString(PyExc_SystemError,
2479 "no locals found during 'import *'");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002480 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002481 }
2482 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002483 err = import_all_from(locals, from);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002484 READ_TIMESTAMP(intr1);
2485 PyFrame_LocalsToFast(f, 0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002486 Py_DECREF(from);
2487 if (err != 0)
2488 goto error;
2489 DISPATCH();
2490 }
Guido van Rossum25831651993-05-19 14:50:45 +00002491
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002492 TARGET(IMPORT_FROM) {
2493 PyObject *name = GETITEM(names, oparg);
2494 PyObject *from = TOP();
2495 PyObject *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002496 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002497 res = import_from(from, name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002498 READ_TIMESTAMP(intr1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002499 PUSH(res);
2500 if (res == NULL)
2501 goto error;
2502 DISPATCH();
2503 }
Thomas Wouters52152252000-08-17 22:55:00 +00002504
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002505 TARGET(JUMP_FORWARD) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002506 JUMPBY(oparg);
2507 FAST_DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002508 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002510 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002511 TARGET(POP_JUMP_IF_FALSE) {
2512 PyObject *cond = POP();
2513 int err;
2514 if (cond == Py_True) {
2515 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002516 FAST_DISPATCH();
2517 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002518 if (cond == Py_False) {
2519 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002520 JUMPTO(oparg);
2521 FAST_DISPATCH();
2522 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002523 err = PyObject_IsTrue(cond);
2524 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002525 if (err > 0)
2526 err = 0;
2527 else if (err == 0)
2528 JUMPTO(oparg);
2529 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002530 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002531 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002532 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002533
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002534 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002535 TARGET(POP_JUMP_IF_TRUE) {
2536 PyObject *cond = POP();
2537 int err;
2538 if (cond == Py_False) {
2539 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002540 FAST_DISPATCH();
2541 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002542 if (cond == Py_True) {
2543 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002544 JUMPTO(oparg);
2545 FAST_DISPATCH();
2546 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002547 err = PyObject_IsTrue(cond);
2548 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002549 if (err > 0) {
2550 err = 0;
2551 JUMPTO(oparg);
2552 }
2553 else if (err == 0)
2554 ;
2555 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002556 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002557 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002558 }
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002559
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002560 TARGET(JUMP_IF_FALSE_OR_POP) {
2561 PyObject *cond = TOP();
2562 int err;
2563 if (cond == Py_True) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002564 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002565 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002566 FAST_DISPATCH();
2567 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002568 if (cond == Py_False) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002569 JUMPTO(oparg);
2570 FAST_DISPATCH();
2571 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002572 err = PyObject_IsTrue(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002573 if (err > 0) {
2574 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002575 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002576 err = 0;
2577 }
2578 else if (err == 0)
2579 JUMPTO(oparg);
2580 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002581 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002582 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002583 }
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002584
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002585 TARGET(JUMP_IF_TRUE_OR_POP) {
2586 PyObject *cond = TOP();
2587 int err;
2588 if (cond == Py_False) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002589 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002590 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002591 FAST_DISPATCH();
2592 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002593 if (cond == Py_True) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002594 JUMPTO(oparg);
2595 FAST_DISPATCH();
2596 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002597 err = PyObject_IsTrue(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002598 if (err > 0) {
2599 err = 0;
2600 JUMPTO(oparg);
2601 }
2602 else if (err == 0) {
2603 STACKADJ(-1);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002604 Py_DECREF(cond);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002605 }
2606 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002607 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002608 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002609 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002610
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002611 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002612 TARGET(JUMP_ABSOLUTE) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002613 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002614#if FAST_LOOPS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002615 /* Enabling this path speeds-up all while and for-loops by bypassing
2616 the per-loop checks for signals. By default, this should be turned-off
2617 because it prevents detection of a control-break in tight loops like
2618 "while 1: pass". Compile with this option turned-on when you need
2619 the speed-up and do not need break checking inside tight loops (ones
2620 that contain only instructions ending with FAST_DISPATCH).
2621 */
2622 FAST_DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002623#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002624 DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002625#endif
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002626 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002627
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002628 TARGET(GET_ITER) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002629 /* before: [obj]; after [getiter(obj)] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002630 PyObject *iterable = TOP();
2631 PyObject *iter = PyObject_GetIter(iterable);
2632 Py_DECREF(iterable);
2633 SET_TOP(iter);
2634 if (iter == NULL)
2635 goto error;
2636 PREDICT(FOR_ITER);
2637 DISPATCH();
2638 }
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002639
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002640 PREDICTED_WITH_ARG(FOR_ITER);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002641 TARGET(FOR_ITER) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002642 /* before: [iter]; after: [iter, iter()] *or* [] */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002643 PyObject *iter = TOP();
2644 PyObject *next = (*iter->ob_type->tp_iternext)(iter);
2645 if (next != NULL) {
2646 PUSH(next);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002647 PREDICT(STORE_FAST);
2648 PREDICT(UNPACK_SEQUENCE);
2649 DISPATCH();
2650 }
2651 if (PyErr_Occurred()) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002652 if (!PyErr_ExceptionMatches(PyExc_StopIteration))
2653 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002654 PyErr_Clear();
2655 }
2656 /* iterator ended normally */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002657 STACKADJ(-1);
2658 Py_DECREF(iter);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002659 JUMPBY(oparg);
2660 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002661 }
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002662
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002663 TARGET(BREAK_LOOP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002664 why = WHY_BREAK;
2665 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002666 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002667
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002668 TARGET(CONTINUE_LOOP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002669 retval = PyLong_FromLong(oparg);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002670 if (retval == NULL)
2671 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002672 why = WHY_CONTINUE;
2673 goto fast_block_end;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002674 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002675
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002676 TARGET_WITH_IMPL(SETUP_LOOP, _setup_finally)
2677 TARGET_WITH_IMPL(SETUP_EXCEPT, _setup_finally)
2678 TARGET(SETUP_FINALLY)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002679 _setup_finally: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002680 /* NOTE: If you add any new block-setup opcodes that
2681 are not try/except/finally handlers, you may need
2682 to update the PyGen_NeedsFinalizing() function.
2683 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002684
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002685 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
2686 STACK_LEVEL());
2687 DISPATCH();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002688 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002689
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002690 TARGET(SETUP_WITH) {
Benjamin Petersonce798522012-01-22 11:24:29 -05002691 _Py_IDENTIFIER(__exit__);
2692 _Py_IDENTIFIER(__enter__);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002693 PyObject *mgr = TOP();
2694 PyObject *exit = special_lookup(mgr, &PyId___exit__), *enter;
2695 PyObject *res;
2696 if (exit == NULL)
2697 goto error;
2698 SET_TOP(exit);
2699 enter = special_lookup(mgr, &PyId___enter__);
2700 Py_DECREF(mgr);
2701 if (enter == NULL)
2702 goto error;
2703 res = PyObject_CallFunctionObjArgs(enter, NULL);
2704 Py_DECREF(enter);
2705 if (res == NULL)
2706 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002707 /* Setup the finally block before pushing the result
2708 of __enter__ on the stack. */
2709 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2710 STACK_LEVEL());
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002711
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002712 PUSH(res);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002713 DISPATCH();
2714 }
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002715
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002716 TARGET(WITH_CLEANUP) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002717 /* At the top of the stack are 1-3 values indicating
2718 how/why we entered the finally clause:
2719 - TOP = None
2720 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2721 - TOP = WHY_*; no retval below it
2722 - (TOP, SECOND, THIRD) = exc_info()
2723 (FOURTH, FITH, SIXTH) = previous exception for EXCEPT_HANDLER
2724 Below them is EXIT, the context.__exit__ bound method.
2725 In the last case, we must call
2726 EXIT(TOP, SECOND, THIRD)
2727 otherwise we must call
2728 EXIT(None, None, None)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002729
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002730 In the first two cases, we remove EXIT from the
2731 stack, leaving the rest in the same order. In the
2732 third case, we shift the bottom 3 values of the
2733 stack down, and replace the empty spot with NULL.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002735 In addition, if the stack represents an exception,
2736 *and* the function call returns a 'true' value, we
2737 push WHY_SILENCED onto the stack. END_FINALLY will
2738 then not re-raise the exception. (But non-local
2739 gotos should still be resumed.)
2740 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002742 PyObject *exit_func;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002743 PyObject *exc = TOP(), *val = Py_None, *tb = Py_None, *res;
2744 int err;
2745 if (exc == Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002746 (void)POP();
2747 exit_func = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002748 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002749 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002750 else if (PyLong_Check(exc)) {
2751 STACKADJ(-1);
2752 switch (PyLong_AsLong(exc)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002753 case WHY_RETURN:
2754 case WHY_CONTINUE:
2755 /* Retval in TOP. */
2756 exit_func = SECOND();
2757 SET_SECOND(TOP());
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002758 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002759 break;
2760 default:
2761 exit_func = TOP();
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002762 SET_TOP(exc);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002763 break;
2764 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002765 exc = Py_None;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002766 }
2767 else {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002768 PyObject *tp2, *exc2, *tb2;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002769 PyTryBlock *block;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002770 val = SECOND();
2771 tb = THIRD();
2772 tp2 = FOURTH();
2773 exc2 = PEEK(5);
2774 tb2 = PEEK(6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002775 exit_func = PEEK(7);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002776 SET_VALUE(7, tb2);
2777 SET_VALUE(6, exc2);
2778 SET_VALUE(5, tp2);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002779 /* UNWIND_EXCEPT_HANDLER will pop this off. */
2780 SET_FOURTH(NULL);
2781 /* We just shifted the stack down, so we have
2782 to tell the except handler block that the
2783 values are lower than it expects. */
2784 block = &f->f_blockstack[f->f_iblock - 1];
2785 assert(block->b_type == EXCEPT_HANDLER);
2786 block->b_level--;
2787 }
2788 /* XXX Not the fastest way to call it... */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002789 res = PyObject_CallFunctionObjArgs(exit_func, exc, val, tb, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002790 Py_DECREF(exit_func);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002791 if (res == NULL)
2792 goto error;
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002793
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002794 if (exc != Py_None)
2795 err = PyObject_IsTrue(res);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002796 else
2797 err = 0;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002798 Py_DECREF(res);
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002799
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002800 if (err < 0)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002801 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002802 else if (err > 0) {
2803 err = 0;
2804 /* There was an exception and a True return */
2805 PUSH(PyLong_FromLong((long) WHY_SILENCED));
2806 }
2807 PREDICT(END_FINALLY);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002808 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002809 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002810
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002811 TARGET(CALL_FUNCTION) {
2812 PyObject **sp, *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002813 PCALL(PCALL_ALL);
2814 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002815#ifdef WITH_TSC
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002816 res = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002817#else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002818 res = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002819#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002820 stack_pointer = sp;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002821 PUSH(res);
2822 if (res == NULL)
2823 goto error;
2824 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002825 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002826
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002827 TARGET_WITH_IMPL(CALL_FUNCTION_VAR, _call_function_var_kw)
2828 TARGET_WITH_IMPL(CALL_FUNCTION_KW, _call_function_var_kw)
2829 TARGET(CALL_FUNCTION_VAR_KW)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002830 _call_function_var_kw: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002831 int na = oparg & 0xff;
2832 int nk = (oparg>>8) & 0xff;
2833 int flags = (opcode - CALL_FUNCTION) & 3;
2834 int n = na + 2 * nk;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002835 PyObject **pfunc, *func, **sp, *res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002836 PCALL(PCALL_ALL);
2837 if (flags & CALL_FLAG_VAR)
2838 n++;
2839 if (flags & CALL_FLAG_KW)
2840 n++;
2841 pfunc = stack_pointer - n - 1;
2842 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002843
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002844 if (PyMethod_Check(func)
Stefan Krahb7e10102010-06-23 18:42:39 +00002845 && PyMethod_GET_SELF(func) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002846 PyObject *self = PyMethod_GET_SELF(func);
2847 Py_INCREF(self);
2848 func = PyMethod_GET_FUNCTION(func);
2849 Py_INCREF(func);
2850 Py_DECREF(*pfunc);
2851 *pfunc = self;
2852 na++;
Brett Cannonb94767f2011-02-22 20:15:44 +00002853 /* n++; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002854 } else
2855 Py_INCREF(func);
2856 sp = stack_pointer;
2857 READ_TIMESTAMP(intr0);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002858 res = ext_do_call(func, &sp, flags, na, nk);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002859 READ_TIMESTAMP(intr1);
2860 stack_pointer = sp;
2861 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002863 while (stack_pointer > pfunc) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002864 PyObject *o = POP();
2865 Py_DECREF(o);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002866 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002867 PUSH(res);
2868 if (res == NULL)
2869 goto error;
2870 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002871 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002872
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002873 TARGET_WITH_IMPL(MAKE_CLOSURE, _make_function)
2874 TARGET(MAKE_FUNCTION)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002875 _make_function: {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002876 int posdefaults = oparg & 0xff;
2877 int kwdefaults = (oparg>>8) & 0xff;
2878 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002879
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002880 PyObject *qualname = POP(); /* qualname */
2881 PyObject *code = POP(); /* code object */
2882 PyObject *func = PyFunction_NewWithQualName(code, f->f_globals, qualname);
2883 Py_DECREF(code);
2884 Py_DECREF(qualname);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002885
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002886 if (func == NULL)
2887 goto error;
2888
2889 if (opcode == MAKE_CLOSURE) {
2890 PyObject *closure = POP();
2891 if (PyFunction_SetClosure(func, closure) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002892 /* Can't happen unless bytecode is corrupt. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002893 Py_DECREF(func);
2894 Py_DECREF(closure);
2895 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002896 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002897 Py_DECREF(closure);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002898 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002899
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002900 if (num_annotations > 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002901 Py_ssize_t name_ix;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002902 PyObject *names = POP(); /* names of args with annotations */
2903 PyObject *anns = PyDict_New();
2904 if (anns == NULL) {
2905 Py_DECREF(func);
2906 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002907 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002908 name_ix = PyTuple_Size(names);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002909 assert(num_annotations == name_ix+1);
2910 while (name_ix > 0) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002911 PyObject *name, *value;
2912 int err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002913 --name_ix;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002914 name = PyTuple_GET_ITEM(names, name_ix);
2915 value = POP();
2916 err = PyDict_SetItem(anns, name, value);
2917 Py_DECREF(value);
2918 if (err != 0) {
2919 Py_DECREF(anns);
2920 Py_DECREF(func);
2921 goto error;
2922 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002923 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002924
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002925 if (PyFunction_SetAnnotations(func, anns) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002926 /* Can't happen unless
2927 PyFunction_SetAnnotations changes. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002928 Py_DECREF(anns);
2929 Py_DECREF(func);
2930 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002931 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002932 Py_DECREF(anns);
2933 Py_DECREF(names);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002934 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002935
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002936 /* XXX Maybe this should be a separate opcode? */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002937 if (kwdefaults > 0) {
2938 PyObject *defs = PyDict_New();
2939 if (defs == NULL) {
2940 Py_DECREF(func);
2941 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002942 }
2943 while (--kwdefaults >= 0) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002944 PyObject *v = POP(); /* default value */
2945 PyObject *key = POP(); /* kw only arg name */
2946 int err = PyDict_SetItem(defs, key, v);
2947 Py_DECREF(v);
2948 Py_DECREF(key);
2949 if (err != 0) {
2950 Py_DECREF(defs);
2951 Py_DECREF(func);
2952 goto error;
2953 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002954 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002955 if (PyFunction_SetKwDefaults(func, defs) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002956 /* Can't happen unless
2957 PyFunction_SetKwDefaults changes. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002958 Py_DECREF(func);
2959 Py_DECREF(defs);
2960 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002961 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002962 Py_DECREF(defs);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002963 }
Benjamin Peterson1ef876c2013-02-10 09:29:59 -05002964 if (posdefaults > 0) {
2965 PyObject *defs = PyTuple_New(posdefaults);
2966 if (defs == NULL) {
2967 Py_DECREF(func);
2968 goto error;
2969 }
2970 while (--posdefaults >= 0)
2971 PyTuple_SET_ITEM(defs, posdefaults, POP());
2972 if (PyFunction_SetDefaults(func, defs) != 0) {
2973 /* Can't happen unless
2974 PyFunction_SetDefaults changes. */
2975 Py_DECREF(defs);
2976 Py_DECREF(func);
2977 goto error;
2978 }
2979 Py_DECREF(defs);
2980 }
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002981 PUSH(func);
2982 DISPATCH();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002983 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002984
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002985 TARGET(BUILD_SLICE) {
2986 PyObject *start, *stop, *step, *slice;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002987 if (oparg == 3)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002988 step = POP();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002989 else
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04002990 step = NULL;
2991 stop = POP();
2992 start = TOP();
2993 slice = PySlice_New(start, stop, step);
2994 Py_DECREF(start);
2995 Py_DECREF(stop);
2996 Py_XDECREF(step);
2997 SET_TOP(slice);
2998 if (slice == NULL)
2999 goto error;
3000 DISPATCH();
3001 }
Guido van Rossum8861b741996-07-30 16:49:37 +00003002
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003003 TARGET(EXTENDED_ARG) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003004 opcode = NEXTOP();
3005 oparg = oparg<<16 | NEXTARG();
3006 goto dispatch_opcode;
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003007 }
Guido van Rossum8861b741996-07-30 16:49:37 +00003008
Antoine Pitrou042b1282010-08-13 21:15:58 +00003009#if USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003010 _unknown_opcode:
Antoine Pitroub52ec782009-01-25 16:34:23 +00003011#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003012 default:
3013 fprintf(stderr,
3014 "XXX lineno: %d, opcode: %d\n",
3015 PyFrame_GetLineNumber(f),
3016 opcode);
3017 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003018 goto error;
Guido van Rossum04691fc1992-08-12 15:35:34 +00003019
3020#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003021 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00003022#endif
3023
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003024 } /* switch */
Guido van Rossum374a9221991-04-04 10:40:29 +00003025
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003026 /* This should never be reached. Every opcode should end with DISPATCH()
3027 or goto error. */
3028 assert(0);
Guido van Rossumac7be682001-01-17 15:42:30 +00003029
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003030error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003031 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003032
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003033 assert(why == WHY_NOT);
3034 why = WHY_EXCEPTION;
Guido van Rossumac7be682001-01-17 15:42:30 +00003035
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003036 /* Double-check exception status. */
Victor Stinner365b6932013-07-12 00:11:58 +02003037#ifdef NDEBUG
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003038 if (!PyErr_Occurred())
3039 PyErr_SetString(PyExc_SystemError,
3040 "error return without exception set");
Victor Stinner365b6932013-07-12 00:11:58 +02003041#else
3042 assert(PyErr_Occurred());
3043#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00003044
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003045 /* Log traceback info. */
3046 PyTraceBack_Here(f);
Guido van Rossumac7be682001-01-17 15:42:30 +00003047
Benjamin Peterson51f46162013-01-23 08:38:47 -05003048 if (tstate->c_tracefunc != NULL)
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003049 call_exc_trace(tstate->c_tracefunc, tstate->c_traceobj, f);
Guido van Rossumac7be682001-01-17 15:42:30 +00003050
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003051fast_block_end:
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003052 assert(why != WHY_NOT);
3053
3054 /* Unwind stacks if a (pseudo) exception occurred */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003055 while (why != WHY_NOT && f->f_iblock > 0) {
3056 /* Peek at the current block. */
3057 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00003058
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003059 assert(why != WHY_YIELD);
3060 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
3061 why = WHY_NOT;
3062 JUMPTO(PyLong_AS_LONG(retval));
3063 Py_DECREF(retval);
3064 break;
3065 }
3066 /* Now we have to pop the block. */
3067 f->f_iblock--;
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00003068
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003069 if (b->b_type == EXCEPT_HANDLER) {
3070 UNWIND_EXCEPT_HANDLER(b);
3071 continue;
3072 }
3073 UNWIND_BLOCK(b);
3074 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
3075 why = WHY_NOT;
3076 JUMPTO(b->b_handler);
3077 break;
3078 }
3079 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
3080 || b->b_type == SETUP_FINALLY)) {
3081 PyObject *exc, *val, *tb;
3082 int handler = b->b_handler;
3083 /* Beware, this invalidates all b->b_* fields */
3084 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
3085 PUSH(tstate->exc_traceback);
3086 PUSH(tstate->exc_value);
3087 if (tstate->exc_type != NULL) {
3088 PUSH(tstate->exc_type);
3089 }
3090 else {
3091 Py_INCREF(Py_None);
3092 PUSH(Py_None);
3093 }
3094 PyErr_Fetch(&exc, &val, &tb);
3095 /* Make the raw exception data
3096 available to the handler,
3097 so a program can emulate the
3098 Python main loop. */
3099 PyErr_NormalizeException(
3100 &exc, &val, &tb);
Victor Stinner7eab0d02013-07-15 21:16:27 +02003101 if (tb != NULL)
3102 PyException_SetTraceback(val, tb);
3103 else
3104 PyException_SetTraceback(val, Py_None);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003105 Py_INCREF(exc);
3106 tstate->exc_type = exc;
3107 Py_INCREF(val);
3108 tstate->exc_value = val;
3109 tstate->exc_traceback = tb;
3110 if (tb == NULL)
3111 tb = Py_None;
3112 Py_INCREF(tb);
3113 PUSH(tb);
3114 PUSH(val);
3115 PUSH(exc);
3116 why = WHY_NOT;
3117 JUMPTO(handler);
3118 break;
3119 }
3120 if (b->b_type == SETUP_FINALLY) {
3121 if (why & (WHY_RETURN | WHY_CONTINUE))
3122 PUSH(retval);
3123 PUSH(PyLong_FromLong((long)why));
3124 why = WHY_NOT;
3125 JUMPTO(b->b_handler);
3126 break;
3127 }
3128 } /* unwind stack */
Guido van Rossum374a9221991-04-04 10:40:29 +00003129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003130 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00003131
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003132 if (why != WHY_NOT)
3133 break;
3134 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00003135
Victor Stinnerace47d72013-07-18 01:41:08 +02003136 assert(!PyErr_Occurred());
3137
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003138 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00003139
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003140 assert(why != WHY_YIELD);
3141 /* Pop remaining stack entries. */
3142 while (!EMPTY()) {
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003143 PyObject *o = POP();
3144 Py_XDECREF(o);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003145 }
Guido van Rossum35974fb2001-12-06 21:28:18 +00003146
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003147 if (why != WHY_RETURN)
3148 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00003149
Victor Stinnerace47d72013-07-18 01:41:08 +02003150 assert((retval != NULL && !PyErr_Occurred())
3151 || (retval == NULL && PyErr_Occurred()));
3152
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003153fast_yield:
Benjamin Petersonac913412011-07-03 16:25:11 -05003154 if (co->co_flags & CO_GENERATOR && (why == WHY_YIELD || why == WHY_RETURN)) {
3155 /* The purpose of this block is to put aside the generator's exception
3156 state and restore that of the calling frame. If the current
3157 exception state is from the caller, we clear the exception values
3158 on the generator frame, so they are not swapped back in latter. The
3159 origin of the current exception state is determined by checking for
3160 except handler blocks, which we must be in iff a new exception
3161 state came into existence in this frame. (An uncaught exception
3162 would have why == WHY_EXCEPTION, and we wouldn't be here). */
3163 int i;
3164 for (i = 0; i < f->f_iblock; i++)
3165 if (f->f_blockstack[i].b_type == EXCEPT_HANDLER)
3166 break;
3167 if (i == f->f_iblock)
3168 /* We did not create this exception. */
Benjamin Peterson87880242011-07-03 16:48:31 -05003169 restore_and_clear_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003170 else
Benjamin Peterson87880242011-07-03 16:48:31 -05003171 swap_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003172 }
Benjamin Peterson83195c32011-07-03 13:44:00 -05003173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003174 if (tstate->use_tracing) {
Benjamin Peterson51f46162013-01-23 08:38:47 -05003175 if (tstate->c_tracefunc) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003176 if (why == WHY_RETURN || why == WHY_YIELD) {
3177 if (call_trace(tstate->c_tracefunc,
3178 tstate->c_traceobj, f,
3179 PyTrace_RETURN, retval)) {
3180 Py_XDECREF(retval);
3181 retval = NULL;
3182 why = WHY_EXCEPTION;
3183 }
3184 }
3185 else if (why == WHY_EXCEPTION) {
3186 call_trace_protected(tstate->c_tracefunc,
3187 tstate->c_traceobj, f,
3188 PyTrace_RETURN, NULL);
3189 }
3190 }
3191 if (tstate->c_profilefunc) {
3192 if (why == WHY_EXCEPTION)
3193 call_trace_protected(tstate->c_profilefunc,
3194 tstate->c_profileobj, f,
3195 PyTrace_RETURN, NULL);
3196 else if (call_trace(tstate->c_profilefunc,
3197 tstate->c_profileobj, f,
3198 PyTrace_RETURN, retval)) {
3199 Py_XDECREF(retval);
3200 retval = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003201 /* why = WHY_EXCEPTION; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003202 }
3203 }
3204 }
Guido van Rossuma4240131997-01-21 21:18:36 +00003205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003206 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00003207exit_eval_frame:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003208 Py_LeaveRecursiveCall();
3209 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00003210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003211 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00003212}
3213
Benjamin Petersonb204a422011-06-05 22:04:07 -05003214static void
Benjamin Petersone109c702011-06-24 09:37:26 -05003215format_missing(const char *kind, PyCodeObject *co, PyObject *names)
3216{
3217 int err;
3218 Py_ssize_t len = PyList_GET_SIZE(names);
3219 PyObject *name_str, *comma, *tail, *tmp;
3220
3221 assert(PyList_CheckExact(names));
3222 assert(len >= 1);
3223 /* Deal with the joys of natural language. */
3224 switch (len) {
3225 case 1:
3226 name_str = PyList_GET_ITEM(names, 0);
3227 Py_INCREF(name_str);
3228 break;
3229 case 2:
3230 name_str = PyUnicode_FromFormat("%U and %U",
3231 PyList_GET_ITEM(names, len - 2),
3232 PyList_GET_ITEM(names, len - 1));
3233 break;
3234 default:
3235 tail = PyUnicode_FromFormat(", %U, and %U",
3236 PyList_GET_ITEM(names, len - 2),
3237 PyList_GET_ITEM(names, len - 1));
Benjamin Petersond1ab6082012-06-01 11:18:22 -07003238 if (tail == NULL)
3239 return;
Benjamin Petersone109c702011-06-24 09:37:26 -05003240 /* Chop off the last two objects in the list. This shouldn't actually
3241 fail, but we can't be too careful. */
3242 err = PyList_SetSlice(names, len - 2, len, NULL);
3243 if (err == -1) {
3244 Py_DECREF(tail);
3245 return;
3246 }
3247 /* Stitch everything up into a nice comma-separated list. */
3248 comma = PyUnicode_FromString(", ");
3249 if (comma == NULL) {
3250 Py_DECREF(tail);
3251 return;
3252 }
3253 tmp = PyUnicode_Join(comma, names);
3254 Py_DECREF(comma);
3255 if (tmp == NULL) {
3256 Py_DECREF(tail);
3257 return;
3258 }
3259 name_str = PyUnicode_Concat(tmp, tail);
3260 Py_DECREF(tmp);
3261 Py_DECREF(tail);
3262 break;
3263 }
3264 if (name_str == NULL)
3265 return;
3266 PyErr_Format(PyExc_TypeError,
3267 "%U() missing %i required %s argument%s: %U",
3268 co->co_name,
3269 len,
3270 kind,
3271 len == 1 ? "" : "s",
3272 name_str);
3273 Py_DECREF(name_str);
3274}
3275
3276static void
3277missing_arguments(PyCodeObject *co, int missing, int defcount,
3278 PyObject **fastlocals)
3279{
3280 int i, j = 0;
3281 int start, end;
3282 int positional = defcount != -1;
3283 const char *kind = positional ? "positional" : "keyword-only";
3284 PyObject *missing_names;
3285
3286 /* Compute the names of the arguments that are missing. */
3287 missing_names = PyList_New(missing);
3288 if (missing_names == NULL)
3289 return;
3290 if (positional) {
3291 start = 0;
3292 end = co->co_argcount - defcount;
3293 }
3294 else {
3295 start = co->co_argcount;
3296 end = start + co->co_kwonlyargcount;
3297 }
3298 for (i = start; i < end; i++) {
3299 if (GETLOCAL(i) == NULL) {
3300 PyObject *raw = PyTuple_GET_ITEM(co->co_varnames, i);
3301 PyObject *name = PyObject_Repr(raw);
3302 if (name == NULL) {
3303 Py_DECREF(missing_names);
3304 return;
3305 }
3306 PyList_SET_ITEM(missing_names, j++, name);
3307 }
3308 }
3309 assert(j == missing);
3310 format_missing(kind, co, missing_names);
3311 Py_DECREF(missing_names);
3312}
3313
3314static void
3315too_many_positional(PyCodeObject *co, int given, int defcount, PyObject **fastlocals)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003316{
3317 int plural;
3318 int kwonly_given = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003319 int i;
3320 PyObject *sig, *kwonly_sig;
3321
Benjamin Petersone109c702011-06-24 09:37:26 -05003322 assert((co->co_flags & CO_VARARGS) == 0);
3323 /* Count missing keyword-only args. */
Benjamin Petersonb204a422011-06-05 22:04:07 -05003324 for (i = co->co_argcount; i < co->co_argcount + co->co_kwonlyargcount; i++)
Benjamin Petersone109c702011-06-24 09:37:26 -05003325 if (GETLOCAL(i) != NULL)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003326 kwonly_given++;
Benjamin Petersone109c702011-06-24 09:37:26 -05003327 if (defcount) {
3328 int atleast = co->co_argcount - defcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003329 plural = 1;
3330 sig = PyUnicode_FromFormat("from %d to %d", atleast, co->co_argcount);
3331 }
3332 else {
3333 plural = co->co_argcount != 1;
3334 sig = PyUnicode_FromFormat("%d", co->co_argcount);
3335 }
3336 if (sig == NULL)
3337 return;
3338 if (kwonly_given) {
3339 const char *format = " positional argument%s (and %d keyword-only argument%s)";
3340 kwonly_sig = PyUnicode_FromFormat(format, given != 1 ? "s" : "", kwonly_given,
3341 kwonly_given != 1 ? "s" : "");
3342 if (kwonly_sig == NULL) {
3343 Py_DECREF(sig);
3344 return;
3345 }
3346 }
3347 else {
3348 /* This will not fail. */
3349 kwonly_sig = PyUnicode_FromString("");
Benjamin Petersone109c702011-06-24 09:37:26 -05003350 assert(kwonly_sig != NULL);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003351 }
3352 PyErr_Format(PyExc_TypeError,
3353 "%U() takes %U positional argument%s but %d%U %s given",
3354 co->co_name,
3355 sig,
3356 plural ? "s" : "",
3357 given,
3358 kwonly_sig,
3359 given == 1 && !kwonly_given ? "was" : "were");
3360 Py_DECREF(sig);
3361 Py_DECREF(kwonly_sig);
3362}
3363
Guido van Rossumc2e20742006-02-27 22:32:47 +00003364/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00003365 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00003366 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00003367
Tim Peters6d6c1a32001-08-02 04:15:00 +00003368PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003369PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003370 PyObject **args, int argcount, PyObject **kws, int kwcount,
3371 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00003372{
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003373 PyCodeObject* co = (PyCodeObject*)_co;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003374 register PyFrameObject *f;
3375 register PyObject *retval = NULL;
3376 register PyObject **fastlocals, **freevars;
3377 PyThreadState *tstate = PyThreadState_GET();
3378 PyObject *x, *u;
3379 int total_args = co->co_argcount + co->co_kwonlyargcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003380 int i;
3381 int n = argcount;
3382 PyObject *kwdict = NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003383
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003384 if (globals == NULL) {
3385 PyErr_SetString(PyExc_SystemError,
3386 "PyEval_EvalCodeEx: NULL globals");
3387 return NULL;
3388 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003390 assert(tstate != NULL);
3391 assert(globals != NULL);
3392 f = PyFrame_New(tstate, co, globals, locals);
3393 if (f == NULL)
3394 return NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003395
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003396 fastlocals = f->f_localsplus;
3397 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00003398
Benjamin Petersonb204a422011-06-05 22:04:07 -05003399 /* Parse arguments. */
3400 if (co->co_flags & CO_VARKEYWORDS) {
3401 kwdict = PyDict_New();
3402 if (kwdict == NULL)
3403 goto fail;
3404 i = total_args;
3405 if (co->co_flags & CO_VARARGS)
3406 i++;
3407 SETLOCAL(i, kwdict);
3408 }
3409 if (argcount > co->co_argcount)
3410 n = co->co_argcount;
3411 for (i = 0; i < n; i++) {
3412 x = args[i];
3413 Py_INCREF(x);
3414 SETLOCAL(i, x);
3415 }
3416 if (co->co_flags & CO_VARARGS) {
3417 u = PyTuple_New(argcount - n);
3418 if (u == NULL)
3419 goto fail;
3420 SETLOCAL(total_args, u);
3421 for (i = n; i < argcount; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003422 x = args[i];
3423 Py_INCREF(x);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003424 PyTuple_SET_ITEM(u, i-n, x);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003425 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003426 }
3427 for (i = 0; i < kwcount; i++) {
3428 PyObject **co_varnames;
3429 PyObject *keyword = kws[2*i];
3430 PyObject *value = kws[2*i + 1];
3431 int j;
3432 if (keyword == NULL || !PyUnicode_Check(keyword)) {
3433 PyErr_Format(PyExc_TypeError,
3434 "%U() keywords must be strings",
3435 co->co_name);
3436 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003437 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003438 /* Speed hack: do raw pointer compares. As names are
3439 normally interned this should almost always hit. */
3440 co_varnames = ((PyTupleObject *)(co->co_varnames))->ob_item;
3441 for (j = 0; j < total_args; j++) {
3442 PyObject *nm = co_varnames[j];
3443 if (nm == keyword)
3444 goto kw_found;
3445 }
3446 /* Slow fallback, just in case */
3447 for (j = 0; j < total_args; j++) {
3448 PyObject *nm = co_varnames[j];
3449 int cmp = PyObject_RichCompareBool(
3450 keyword, nm, Py_EQ);
3451 if (cmp > 0)
3452 goto kw_found;
3453 else if (cmp < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003454 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003455 }
3456 if (j >= total_args && kwdict == NULL) {
3457 PyErr_Format(PyExc_TypeError,
3458 "%U() got an unexpected "
3459 "keyword argument '%S'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003460 co->co_name,
3461 keyword);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003462 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003463 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003464 PyDict_SetItem(kwdict, keyword, value);
3465 continue;
3466 kw_found:
3467 if (GETLOCAL(j) != NULL) {
3468 PyErr_Format(PyExc_TypeError,
3469 "%U() got multiple "
3470 "values for argument '%S'",
3471 co->co_name,
3472 keyword);
3473 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003474 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003475 Py_INCREF(value);
3476 SETLOCAL(j, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003477 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003478 if (argcount > co->co_argcount && !(co->co_flags & CO_VARARGS)) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003479 too_many_positional(co, argcount, defcount, fastlocals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003480 goto fail;
3481 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003482 if (argcount < co->co_argcount) {
3483 int m = co->co_argcount - defcount;
Benjamin Petersone109c702011-06-24 09:37:26 -05003484 int missing = 0;
3485 for (i = argcount; i < m; i++)
3486 if (GETLOCAL(i) == NULL)
3487 missing++;
3488 if (missing) {
3489 missing_arguments(co, missing, defcount, fastlocals);
3490 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003491 }
3492 if (n > m)
3493 i = n - m;
3494 else
3495 i = 0;
3496 for (; i < defcount; i++) {
3497 if (GETLOCAL(m+i) == NULL) {
3498 PyObject *def = defs[i];
3499 Py_INCREF(def);
3500 SETLOCAL(m+i, def);
3501 }
3502 }
3503 }
3504 if (co->co_kwonlyargcount > 0) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003505 int missing = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003506 for (i = co->co_argcount; i < total_args; i++) {
3507 PyObject *name;
3508 if (GETLOCAL(i) != NULL)
3509 continue;
3510 name = PyTuple_GET_ITEM(co->co_varnames, i);
3511 if (kwdefs != NULL) {
3512 PyObject *def = PyDict_GetItem(kwdefs, name);
3513 if (def) {
3514 Py_INCREF(def);
3515 SETLOCAL(i, def);
3516 continue;
3517 }
3518 }
Benjamin Petersone109c702011-06-24 09:37:26 -05003519 missing++;
3520 }
3521 if (missing) {
3522 missing_arguments(co, missing, -1, fastlocals);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003523 goto fail;
3524 }
3525 }
3526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003527 /* Allocate and initialize storage for cell vars, and copy free
Benjamin Peterson90037602011-06-25 22:54:45 -05003528 vars into frame. */
3529 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003530 PyObject *c;
Benjamin Peterson90037602011-06-25 22:54:45 -05003531 int arg;
3532 /* Possibly account for the cell variable being an argument. */
3533 if (co->co_cell2arg != NULL &&
Guido van Rossum6832c812013-05-10 08:47:42 -07003534 (arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG) {
Benjamin Peterson90037602011-06-25 22:54:45 -05003535 c = PyCell_New(GETLOCAL(arg));
Benjamin Peterson159ae412013-05-12 18:16:06 -05003536 /* Clear the local copy. */
3537 SETLOCAL(arg, NULL);
Guido van Rossum6832c812013-05-10 08:47:42 -07003538 }
3539 else {
Benjamin Peterson90037602011-06-25 22:54:45 -05003540 c = PyCell_New(NULL);
Guido van Rossum6832c812013-05-10 08:47:42 -07003541 }
Benjamin Peterson159ae412013-05-12 18:16:06 -05003542 if (c == NULL)
3543 goto fail;
Benjamin Peterson90037602011-06-25 22:54:45 -05003544 SETLOCAL(co->co_nlocals + i, c);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003545 }
Benjamin Peterson90037602011-06-25 22:54:45 -05003546 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
3547 PyObject *o = PyTuple_GET_ITEM(closure, i);
3548 Py_INCREF(o);
3549 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003550 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003551
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003552 if (co->co_flags & CO_GENERATOR) {
3553 /* Don't need to keep the reference to f_back, it will be set
3554 * when the generator is resumed. */
3555 Py_XDECREF(f->f_back);
3556 f->f_back = NULL;
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003558 PCALL(PCALL_GENERATOR);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003560 /* Create a new generator that owns the ready to run frame
3561 * and return that as the value. */
3562 return PyGen_New(f);
3563 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003564
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003565 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003566
Thomas Woutersce272b62007-09-19 21:19:28 +00003567fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003569 /* decref'ing the frame can cause __del__ methods to get invoked,
3570 which can call back into Python. While we're done with the
3571 current Python frame (f), the associated C stack is still in use,
3572 so recursion_depth must be boosted for the duration.
3573 */
3574 assert(tstate != NULL);
3575 ++tstate->recursion_depth;
3576 Py_DECREF(f);
3577 --tstate->recursion_depth;
3578 return retval;
Tim Peters5ca576e2001-06-18 22:08:13 +00003579}
3580
3581
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003582static PyObject *
Benjamin Petersonce798522012-01-22 11:24:29 -05003583special_lookup(PyObject *o, _Py_Identifier *id)
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003584{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 PyObject *res;
Benjamin Petersonce798522012-01-22 11:24:29 -05003586 res = _PyObject_LookupSpecial(o, id);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003587 if (res == NULL && !PyErr_Occurred()) {
Benjamin Petersonce798522012-01-22 11:24:29 -05003588 PyErr_SetObject(PyExc_AttributeError, id->object);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003589 return NULL;
3590 }
3591 return res;
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003592}
3593
3594
Benjamin Peterson87880242011-07-03 16:48:31 -05003595/* These 3 functions deal with the exception state of generators. */
3596
3597static void
3598save_exc_state(PyThreadState *tstate, PyFrameObject *f)
3599{
3600 PyObject *type, *value, *traceback;
3601 Py_XINCREF(tstate->exc_type);
3602 Py_XINCREF(tstate->exc_value);
3603 Py_XINCREF(tstate->exc_traceback);
3604 type = f->f_exc_type;
3605 value = f->f_exc_value;
3606 traceback = f->f_exc_traceback;
3607 f->f_exc_type = tstate->exc_type;
3608 f->f_exc_value = tstate->exc_value;
3609 f->f_exc_traceback = tstate->exc_traceback;
3610 Py_XDECREF(type);
3611 Py_XDECREF(value);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02003612 Py_XDECREF(traceback);
Benjamin Peterson87880242011-07-03 16:48:31 -05003613}
3614
3615static void
3616swap_exc_state(PyThreadState *tstate, PyFrameObject *f)
3617{
3618 PyObject *tmp;
3619 tmp = tstate->exc_type;
3620 tstate->exc_type = f->f_exc_type;
3621 f->f_exc_type = tmp;
3622 tmp = tstate->exc_value;
3623 tstate->exc_value = f->f_exc_value;
3624 f->f_exc_value = tmp;
3625 tmp = tstate->exc_traceback;
3626 tstate->exc_traceback = f->f_exc_traceback;
3627 f->f_exc_traceback = tmp;
3628}
3629
3630static void
3631restore_and_clear_exc_state(PyThreadState *tstate, PyFrameObject *f)
3632{
3633 PyObject *type, *value, *tb;
3634 type = tstate->exc_type;
3635 value = tstate->exc_value;
3636 tb = tstate->exc_traceback;
3637 tstate->exc_type = f->f_exc_type;
3638 tstate->exc_value = f->f_exc_value;
3639 tstate->exc_traceback = f->f_exc_traceback;
3640 f->f_exc_type = NULL;
3641 f->f_exc_value = NULL;
3642 f->f_exc_traceback = NULL;
3643 Py_XDECREF(type);
3644 Py_XDECREF(value);
3645 Py_XDECREF(tb);
3646}
3647
3648
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003649/* Logic for the raise statement (too complicated for inlining).
3650 This *consumes* a reference count to each of its arguments. */
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003651static int
Collin Winter828f04a2007-08-31 00:04:24 +00003652do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003653{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003654 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00003655
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003656 if (exc == NULL) {
3657 /* Reraise */
3658 PyThreadState *tstate = PyThreadState_GET();
3659 PyObject *tb;
3660 type = tstate->exc_type;
3661 value = tstate->exc_value;
3662 tb = tstate->exc_traceback;
3663 if (type == Py_None) {
3664 PyErr_SetString(PyExc_RuntimeError,
3665 "No active exception to reraise");
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003666 return 0;
3667 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003668 Py_XINCREF(type);
3669 Py_XINCREF(value);
3670 Py_XINCREF(tb);
3671 PyErr_Restore(type, value, tb);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003672 return 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003673 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003674
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003675 /* We support the following forms of raise:
3676 raise
Collin Winter828f04a2007-08-31 00:04:24 +00003677 raise <instance>
3678 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003679
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003680 if (PyExceptionClass_Check(exc)) {
3681 type = exc;
3682 value = PyObject_CallObject(exc, NULL);
3683 if (value == NULL)
3684 goto raise_error;
Benjamin Peterson5afa03a2011-07-15 14:09:26 -05003685 if (!PyExceptionInstance_Check(value)) {
3686 PyErr_Format(PyExc_TypeError,
3687 "calling %R should have returned an instance of "
3688 "BaseException, not %R",
3689 type, Py_TYPE(value));
3690 goto raise_error;
3691 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003692 }
3693 else if (PyExceptionInstance_Check(exc)) {
3694 value = exc;
3695 type = PyExceptionInstance_Class(exc);
3696 Py_INCREF(type);
3697 }
3698 else {
3699 /* Not something you can raise. You get an exception
3700 anyway, just not what you specified :-) */
3701 Py_DECREF(exc);
3702 PyErr_SetString(PyExc_TypeError,
3703 "exceptions must derive from BaseException");
3704 goto raise_error;
3705 }
Collin Winter828f04a2007-08-31 00:04:24 +00003706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003707 if (cause) {
3708 PyObject *fixed_cause;
3709 if (PyExceptionClass_Check(cause)) {
3710 fixed_cause = PyObject_CallObject(cause, NULL);
3711 if (fixed_cause == NULL)
3712 goto raise_error;
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003713 Py_DECREF(cause);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003714 }
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003715 else if (PyExceptionInstance_Check(cause)) {
3716 fixed_cause = cause;
3717 }
3718 else if (cause == Py_None) {
3719 Py_DECREF(cause);
3720 fixed_cause = NULL;
3721 }
3722 else {
3723 PyErr_SetString(PyExc_TypeError,
3724 "exception causes must derive from "
3725 "BaseException");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003726 goto raise_error;
3727 }
Benjamin Petersond5a1c442012-05-14 22:09:31 -07003728 PyException_SetCause(value, fixed_cause);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003729 }
Collin Winter828f04a2007-08-31 00:04:24 +00003730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003731 PyErr_SetObject(type, value);
3732 /* PyErr_SetObject incref's its arguments */
3733 Py_XDECREF(value);
3734 Py_XDECREF(type);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003735 return 0;
Collin Winter828f04a2007-08-31 00:04:24 +00003736
3737raise_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003738 Py_XDECREF(value);
3739 Py_XDECREF(type);
3740 Py_XDECREF(cause);
Benjamin Peterson31a58ff2012-10-12 11:34:51 -04003741 return 0;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003742}
3743
Tim Petersd6d010b2001-06-21 02:49:55 +00003744/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003745 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00003746
Guido van Rossum0368b722007-05-11 16:50:42 +00003747 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3748 with a variable target.
3749*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003750
Barry Warsawe42b18f1997-08-25 22:13:04 +00003751static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003752unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003753{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003754 int i = 0, j = 0;
3755 Py_ssize_t ll = 0;
3756 PyObject *it; /* iter(v) */
3757 PyObject *w;
3758 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003759
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003760 assert(v != NULL);
Tim Petersd6d010b2001-06-21 02:49:55 +00003761
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003762 it = PyObject_GetIter(v);
3763 if (it == NULL)
3764 goto Error;
Tim Petersd6d010b2001-06-21 02:49:55 +00003765
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003766 for (; i < argcnt; i++) {
3767 w = PyIter_Next(it);
3768 if (w == NULL) {
3769 /* Iterator done, via error or exhaustion. */
3770 if (!PyErr_Occurred()) {
3771 PyErr_Format(PyExc_ValueError,
3772 "need more than %d value%s to unpack",
3773 i, i == 1 ? "" : "s");
3774 }
3775 goto Error;
3776 }
3777 *--sp = w;
3778 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003779
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003780 if (argcntafter == -1) {
3781 /* We better have exhausted the iterator now. */
3782 w = PyIter_Next(it);
3783 if (w == NULL) {
3784 if (PyErr_Occurred())
3785 goto Error;
3786 Py_DECREF(it);
3787 return 1;
3788 }
3789 Py_DECREF(w);
Georg Brandl0310a832010-07-10 10:32:36 +00003790 PyErr_Format(PyExc_ValueError, "too many values to unpack "
3791 "(expected %d)", argcnt);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003792 goto Error;
3793 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003794
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003795 l = PySequence_List(it);
3796 if (l == NULL)
3797 goto Error;
3798 *--sp = l;
3799 i++;
Guido van Rossum0368b722007-05-11 16:50:42 +00003800
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003801 ll = PyList_GET_SIZE(l);
3802 if (ll < argcntafter) {
3803 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3804 argcnt + ll);
3805 goto Error;
3806 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003807
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003808 /* Pop the "after-variable" args off the list. */
3809 for (j = argcntafter; j > 0; j--, i++) {
3810 *--sp = PyList_GET_ITEM(l, ll - j);
3811 }
3812 /* Resize the list. */
3813 Py_SIZE(l) = ll - argcntafter;
3814 Py_DECREF(it);
3815 return 1;
Guido van Rossum0368b722007-05-11 16:50:42 +00003816
Tim Petersd6d010b2001-06-21 02:49:55 +00003817Error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003818 for (; i > 0; i--, sp++)
3819 Py_DECREF(*sp);
3820 Py_XDECREF(it);
3821 return 0;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003822}
3823
3824
Guido van Rossum96a42c81992-01-12 02:29:51 +00003825#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003826static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003827prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003828{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003829 printf("%s ", str);
3830 if (PyObject_Print(v, stdout, 0) != 0)
3831 PyErr_Clear(); /* Don't know what else to do */
3832 printf("\n");
3833 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003834}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003835#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003836
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003837static void
Fred Drake5755ce62001-06-27 19:19:46 +00003838call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003839{
Victor Stinneraaa8ed82013-07-10 13:57:55 +02003840 PyObject *type, *value, *traceback, *orig_traceback, *arg;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003841 int err;
3842 PyErr_Fetch(&type, &value, &traceback);
3843 if (value == NULL) {
3844 value = Py_None;
3845 Py_INCREF(value);
3846 }
R David Murray35837612013-04-19 12:56:57 -04003847 PyErr_NormalizeException(&type, &value, &traceback);
Victor Stinneraaa8ed82013-07-10 13:57:55 +02003848 orig_traceback = traceback;
3849 if (traceback == NULL) {
3850 Py_INCREF(Py_None);
3851 traceback = Py_None;
3852 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003853 arg = PyTuple_Pack(3, type, value, traceback);
3854 if (arg == NULL) {
3855 PyErr_Restore(type, value, traceback);
3856 return;
3857 }
3858 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
3859 Py_DECREF(arg);
3860 if (err == 0)
Victor Stinneraaa8ed82013-07-10 13:57:55 +02003861 PyErr_Restore(type, value, orig_traceback);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003862 else {
3863 Py_XDECREF(type);
3864 Py_XDECREF(value);
Victor Stinneraaa8ed82013-07-10 13:57:55 +02003865 Py_XDECREF(orig_traceback);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003866 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003867}
3868
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003869static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003870call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003871 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003872{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003873 PyObject *type, *value, *traceback;
3874 int err;
3875 PyErr_Fetch(&type, &value, &traceback);
3876 err = call_trace(func, obj, frame, what, arg);
3877 if (err == 0)
3878 {
3879 PyErr_Restore(type, value, traceback);
3880 return 0;
3881 }
3882 else {
3883 Py_XDECREF(type);
3884 Py_XDECREF(value);
3885 Py_XDECREF(traceback);
3886 return -1;
3887 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003888}
3889
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003890static int
Fred Drake5755ce62001-06-27 19:19:46 +00003891call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003892 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003893{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003894 register PyThreadState *tstate = frame->f_tstate;
3895 int result;
3896 if (tstate->tracing)
3897 return 0;
3898 tstate->tracing++;
3899 tstate->use_tracing = 0;
3900 result = func(obj, frame, what, arg);
3901 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3902 || (tstate->c_profilefunc != NULL));
3903 tstate->tracing--;
3904 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003905}
3906
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003907PyObject *
3908_PyEval_CallTracing(PyObject *func, PyObject *args)
3909{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003910 PyFrameObject *frame = PyEval_GetFrame();
3911 PyThreadState *tstate = frame->f_tstate;
3912 int save_tracing = tstate->tracing;
3913 int save_use_tracing = tstate->use_tracing;
3914 PyObject *result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003915
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003916 tstate->tracing = 0;
3917 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3918 || (tstate->c_profilefunc != NULL));
3919 result = PyObject_Call(func, args, NULL);
3920 tstate->tracing = save_tracing;
3921 tstate->use_tracing = save_use_tracing;
3922 return result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003923}
3924
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00003925/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Michael W. Hudson006c7522002-11-08 13:08:46 +00003926static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003927maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003928 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3929 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003930{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003931 int result = 0;
3932 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003933
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003934 /* If the last instruction executed isn't in the current
3935 instruction window, reset the window.
3936 */
3937 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
3938 PyAddrPair bounds;
3939 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3940 &bounds);
3941 *instr_lb = bounds.ap_lower;
3942 *instr_ub = bounds.ap_upper;
3943 }
3944 /* If the last instruction falls at the start of a line or if
3945 it represents a jump backwards, update the frame's line
3946 number and call the trace function. */
3947 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3948 frame->f_lineno = line;
3949 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
3950 }
3951 *instr_prev = frame->f_lasti;
3952 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003953}
3954
Fred Drake5755ce62001-06-27 19:19:46 +00003955void
3956PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003957{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003958 PyThreadState *tstate = PyThreadState_GET();
3959 PyObject *temp = tstate->c_profileobj;
3960 Py_XINCREF(arg);
3961 tstate->c_profilefunc = NULL;
3962 tstate->c_profileobj = NULL;
3963 /* Must make sure that tracing is not ignored if 'temp' is freed */
3964 tstate->use_tracing = tstate->c_tracefunc != NULL;
3965 Py_XDECREF(temp);
3966 tstate->c_profilefunc = func;
3967 tstate->c_profileobj = arg;
3968 /* Flag that tracing or profiling is turned on */
3969 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003970}
3971
3972void
3973PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3974{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003975 PyThreadState *tstate = PyThreadState_GET();
3976 PyObject *temp = tstate->c_traceobj;
3977 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
3978 Py_XINCREF(arg);
3979 tstate->c_tracefunc = NULL;
3980 tstate->c_traceobj = NULL;
3981 /* Must make sure that profiling is not ignored if 'temp' is freed */
3982 tstate->use_tracing = tstate->c_profilefunc != NULL;
3983 Py_XDECREF(temp);
3984 tstate->c_tracefunc = func;
3985 tstate->c_traceobj = arg;
3986 /* Flag that tracing or profiling is turned on */
3987 tstate->use_tracing = ((func != NULL)
3988 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003989}
3990
Guido van Rossumb209a111997-04-29 18:18:01 +00003991PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003992PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003993{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003994 PyFrameObject *current_frame = PyEval_GetFrame();
3995 if (current_frame == NULL)
3996 return PyThreadState_GET()->interp->builtins;
3997 else
3998 return current_frame->f_builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003999}
4000
Guido van Rossumb209a111997-04-29 18:18:01 +00004001PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004002PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00004003{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004004 PyFrameObject *current_frame = PyEval_GetFrame();
4005 if (current_frame == NULL)
4006 return NULL;
4007 PyFrame_FastToLocals(current_frame);
4008 return current_frame->f_locals;
Guido van Rossum5b722181993-03-30 17:46:03 +00004009}
4010
Guido van Rossumb209a111997-04-29 18:18:01 +00004011PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004012PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00004013{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004014 PyFrameObject *current_frame = PyEval_GetFrame();
4015 if (current_frame == NULL)
4016 return NULL;
4017 else
4018 return current_frame->f_globals;
Guido van Rossum3f5da241990-12-20 15:06:42 +00004019}
4020
Guido van Rossum6297a7a2003-02-19 15:53:17 +00004021PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004022PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00004023{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004024 PyThreadState *tstate = PyThreadState_GET();
4025 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00004026}
4027
Guido van Rossum6135a871995-01-09 17:53:26 +00004028int
Tim Peters5ba58662001-07-16 02:29:45 +00004029PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00004030{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004031 PyFrameObject *current_frame = PyEval_GetFrame();
4032 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00004033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004034 if (current_frame != NULL) {
4035 const int codeflags = current_frame->f_code->co_flags;
4036 const int compilerflags = codeflags & PyCF_MASK;
4037 if (compilerflags) {
4038 result = 1;
4039 cf->cf_flags |= compilerflags;
4040 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00004041#if 0 /* future keyword */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004042 if (codeflags & CO_GENERATOR_ALLOWED) {
4043 result = 1;
4044 cf->cf_flags |= CO_GENERATOR_ALLOWED;
4045 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00004046#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004047 }
4048 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00004049}
4050
Guido van Rossum3f5da241990-12-20 15:06:42 +00004051
Guido van Rossum681d79a1995-07-18 14:51:37 +00004052/* External interface to call any callable object.
Antoine Pitrou8689a102010-04-01 16:53:15 +00004053 The arg must be a tuple or NULL. The kw must be a dict or NULL. */
Guido van Rossume59214e1994-08-30 08:01:59 +00004054
Guido van Rossumb209a111997-04-29 18:18:01 +00004055PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004056PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00004057{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004058 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004059
Victor Stinnerace47d72013-07-18 01:41:08 +02004060#ifdef Py_DEBUG
4061 /* PyEval_CallObjectWithKeywords() must not be called with an exception
4062 set, because it may clear it (directly or indirectly)
4063 and so the caller looses its exception */
4064 assert(!PyErr_Occurred());
4065#endif
4066
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004067 if (arg == NULL) {
4068 arg = PyTuple_New(0);
4069 if (arg == NULL)
4070 return NULL;
4071 }
4072 else if (!PyTuple_Check(arg)) {
4073 PyErr_SetString(PyExc_TypeError,
4074 "argument list must be a tuple");
4075 return NULL;
4076 }
4077 else
4078 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004079
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004080 if (kw != NULL && !PyDict_Check(kw)) {
4081 PyErr_SetString(PyExc_TypeError,
4082 "keyword list must be a dictionary");
4083 Py_DECREF(arg);
4084 return NULL;
4085 }
Guido van Rossume3e61c11995-08-04 04:14:47 +00004086
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004087 result = PyObject_Call(func, arg, kw);
4088 Py_DECREF(arg);
Victor Stinnerace47d72013-07-18 01:41:08 +02004089
4090 assert((result != NULL && !PyErr_Occurred())
4091 || (result == NULL && PyErr_Occurred()));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004092 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004093}
4094
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00004095const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00004096PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00004097{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004098 if (PyMethod_Check(func))
4099 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
4100 else if (PyFunction_Check(func))
4101 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
4102 else if (PyCFunction_Check(func))
4103 return ((PyCFunctionObject*)func)->m_ml->ml_name;
4104 else
4105 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00004106}
4107
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00004108const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00004109PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00004110{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004111 if (PyMethod_Check(func))
4112 return "()";
4113 else if (PyFunction_Check(func))
4114 return "()";
4115 else if (PyCFunction_Check(func))
4116 return "()";
4117 else
4118 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00004119}
4120
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00004121static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00004122err_args(PyObject *func, int flags, int nargs)
4123{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004124 if (flags & METH_NOARGS)
4125 PyErr_Format(PyExc_TypeError,
4126 "%.200s() takes no arguments (%d given)",
4127 ((PyCFunctionObject *)func)->m_ml->ml_name,
4128 nargs);
4129 else
4130 PyErr_Format(PyExc_TypeError,
4131 "%.200s() takes exactly one argument (%d given)",
4132 ((PyCFunctionObject *)func)->m_ml->ml_name,
4133 nargs);
Jeremy Hylton192690e2002-08-16 18:36:11 +00004134}
4135
Armin Rigo1c2d7e52005-09-20 18:34:01 +00004136#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00004137if (tstate->use_tracing && tstate->c_profilefunc) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004138 if (call_trace(tstate->c_profilefunc, \
4139 tstate->c_profileobj, \
4140 tstate->frame, PyTrace_C_CALL, \
4141 func)) { \
4142 x = NULL; \
4143 } \
4144 else { \
4145 x = call; \
4146 if (tstate->c_profilefunc != NULL) { \
4147 if (x == NULL) { \
4148 call_trace_protected(tstate->c_profilefunc, \
4149 tstate->c_profileobj, \
4150 tstate->frame, PyTrace_C_EXCEPTION, \
4151 func); \
4152 /* XXX should pass (type, value, tb) */ \
4153 } else { \
4154 if (call_trace(tstate->c_profilefunc, \
4155 tstate->c_profileobj, \
4156 tstate->frame, PyTrace_C_RETURN, \
4157 func)) { \
4158 Py_DECREF(x); \
4159 x = NULL; \
4160 } \
4161 } \
4162 } \
4163 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00004164} else { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004165 x = call; \
4166 }
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00004167
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004168static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004169call_function(PyObject ***pp_stack, int oparg
4170#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004171 , uint64* pintr0, uint64* pintr1
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004172#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004173 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004174{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004175 int na = oparg & 0xff;
4176 int nk = (oparg>>8) & 0xff;
4177 int n = na + 2 * nk;
4178 PyObject **pfunc = (*pp_stack) - n - 1;
4179 PyObject *func = *pfunc;
4180 PyObject *x, *w;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004181
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004182 /* Always dispatch PyCFunction first, because these are
4183 presumed to be the most frequent callable object.
4184 */
4185 if (PyCFunction_Check(func) && nk == 0) {
4186 int flags = PyCFunction_GET_FLAGS(func);
4187 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00004188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004189 PCALL(PCALL_CFUNCTION);
4190 if (flags & (METH_NOARGS | METH_O)) {
4191 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
4192 PyObject *self = PyCFunction_GET_SELF(func);
4193 if (flags & METH_NOARGS && na == 0) {
4194 C_TRACE(x, (*meth)(self,NULL));
4195 }
4196 else if (flags & METH_O && na == 1) {
4197 PyObject *arg = EXT_POP(*pp_stack);
4198 C_TRACE(x, (*meth)(self,arg));
4199 Py_DECREF(arg);
4200 }
4201 else {
4202 err_args(func, flags, na);
4203 x = NULL;
4204 }
4205 }
4206 else {
4207 PyObject *callargs;
4208 callargs = load_args(pp_stack, na);
Victor Stinner0ff0f542013-07-08 22:27:42 +02004209 if (callargs != NULL) {
4210 READ_TIMESTAMP(*pintr0);
4211 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
4212 READ_TIMESTAMP(*pintr1);
4213 Py_XDECREF(callargs);
4214 }
4215 else {
4216 x = NULL;
4217 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004218 }
4219 } else {
4220 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
4221 /* optimize access to bound methods */
4222 PyObject *self = PyMethod_GET_SELF(func);
4223 PCALL(PCALL_METHOD);
4224 PCALL(PCALL_BOUND_METHOD);
4225 Py_INCREF(self);
4226 func = PyMethod_GET_FUNCTION(func);
4227 Py_INCREF(func);
4228 Py_DECREF(*pfunc);
4229 *pfunc = self;
4230 na++;
4231 n++;
4232 } else
4233 Py_INCREF(func);
4234 READ_TIMESTAMP(*pintr0);
4235 if (PyFunction_Check(func))
4236 x = fast_function(func, pp_stack, n, na, nk);
4237 else
4238 x = do_call(func, pp_stack, na, nk);
4239 READ_TIMESTAMP(*pintr1);
4240 Py_DECREF(func);
4241 }
Victor Stinnerf243ee42013-07-16 01:02:12 +02004242 assert((x != NULL && !PyErr_Occurred())
4243 || (x == NULL && PyErr_Occurred()));
Tim Peters8a5c3c72004-04-05 19:36:21 +00004244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004245 /* Clear the stack of the function object. Also removes
4246 the arguments in case they weren't consumed already
4247 (fast_function() and err_args() leave them on the stack).
4248 */
4249 while ((*pp_stack) > pfunc) {
4250 w = EXT_POP(*pp_stack);
4251 Py_DECREF(w);
4252 PCALL(PCALL_POP);
4253 }
Victor Stinnerace47d72013-07-18 01:41:08 +02004254
4255 assert((x != NULL && !PyErr_Occurred())
4256 || (x == NULL && PyErr_Occurred()));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004257 return x;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004258}
4259
Jeremy Hylton192690e2002-08-16 18:36:11 +00004260/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004261 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004262 For the simplest case -- a function that takes only positional
4263 arguments and is called with only positional arguments -- it
4264 inlines the most primitive frame setup code from
4265 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4266 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004267*/
4268
4269static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004270fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004271{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004272 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
4273 PyObject *globals = PyFunction_GET_GLOBALS(func);
4274 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4275 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
4276 PyObject **d = NULL;
4277 int nd = 0;
Jeremy Hylton52820442001-01-03 23:52:36 +00004278
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004279 PCALL(PCALL_FUNCTION);
4280 PCALL(PCALL_FAST_FUNCTION);
4281 if (argdefs == NULL && co->co_argcount == n &&
4282 co->co_kwonlyargcount == 0 && nk==0 &&
4283 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4284 PyFrameObject *f;
4285 PyObject *retval = NULL;
4286 PyThreadState *tstate = PyThreadState_GET();
4287 PyObject **fastlocals, **stack;
4288 int i;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004289
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004290 PCALL(PCALL_FASTER_FUNCTION);
4291 assert(globals != NULL);
4292 /* XXX Perhaps we should create a specialized
4293 PyFrame_New() that doesn't take locals, but does
4294 take builtins without sanity checking them.
4295 */
4296 assert(tstate != NULL);
4297 f = PyFrame_New(tstate, co, globals, NULL);
4298 if (f == NULL)
4299 return NULL;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004301 fastlocals = f->f_localsplus;
4302 stack = (*pp_stack) - n;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004303
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004304 for (i = 0; i < n; i++) {
4305 Py_INCREF(*stack);
4306 fastlocals[i] = *stack++;
4307 }
4308 retval = PyEval_EvalFrameEx(f,0);
4309 ++tstate->recursion_depth;
4310 Py_DECREF(f);
4311 --tstate->recursion_depth;
4312 return retval;
4313 }
4314 if (argdefs != NULL) {
4315 d = &PyTuple_GET_ITEM(argdefs, 0);
4316 nd = Py_SIZE(argdefs);
4317 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004318 return PyEval_EvalCodeEx((PyObject*)co, globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004319 (PyObject *)NULL, (*pp_stack)-n, na,
4320 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
4321 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004322}
4323
4324static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004325update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4326 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004327{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004328 PyObject *kwdict = NULL;
4329 if (orig_kwdict == NULL)
4330 kwdict = PyDict_New();
4331 else {
4332 kwdict = PyDict_Copy(orig_kwdict);
4333 Py_DECREF(orig_kwdict);
4334 }
4335 if (kwdict == NULL)
4336 return NULL;
4337 while (--nk >= 0) {
4338 int err;
4339 PyObject *value = EXT_POP(*pp_stack);
4340 PyObject *key = EXT_POP(*pp_stack);
4341 if (PyDict_GetItem(kwdict, key) != NULL) {
4342 PyErr_Format(PyExc_TypeError,
4343 "%.200s%s got multiple values "
4344 "for keyword argument '%U'",
4345 PyEval_GetFuncName(func),
4346 PyEval_GetFuncDesc(func),
4347 key);
4348 Py_DECREF(key);
4349 Py_DECREF(value);
4350 Py_DECREF(kwdict);
4351 return NULL;
4352 }
4353 err = PyDict_SetItem(kwdict, key, value);
4354 Py_DECREF(key);
4355 Py_DECREF(value);
4356 if (err) {
4357 Py_DECREF(kwdict);
4358 return NULL;
4359 }
4360 }
4361 return kwdict;
Jeremy Hylton52820442001-01-03 23:52:36 +00004362}
4363
4364static PyObject *
4365update_star_args(int nstack, int nstar, PyObject *stararg,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 PyObject ***pp_stack)
Jeremy Hylton52820442001-01-03 23:52:36 +00004367{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004368 PyObject *callargs, *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004370 callargs = PyTuple_New(nstack + nstar);
4371 if (callargs == NULL) {
4372 return NULL;
4373 }
4374 if (nstar) {
4375 int i;
4376 for (i = 0; i < nstar; i++) {
4377 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4378 Py_INCREF(a);
4379 PyTuple_SET_ITEM(callargs, nstack + i, a);
4380 }
4381 }
4382 while (--nstack >= 0) {
4383 w = EXT_POP(*pp_stack);
4384 PyTuple_SET_ITEM(callargs, nstack, w);
4385 }
4386 return callargs;
Jeremy Hylton52820442001-01-03 23:52:36 +00004387}
4388
4389static PyObject *
4390load_args(PyObject ***pp_stack, int na)
4391{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 PyObject *args = PyTuple_New(na);
4393 PyObject *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004394
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004395 if (args == NULL)
4396 return NULL;
4397 while (--na >= 0) {
4398 w = EXT_POP(*pp_stack);
4399 PyTuple_SET_ITEM(args, na, w);
4400 }
4401 return args;
Jeremy Hylton52820442001-01-03 23:52:36 +00004402}
4403
4404static PyObject *
4405do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4406{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004407 PyObject *callargs = 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 (nk > 0) {
4412 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
4413 if (kwdict == NULL)
4414 goto call_fail;
4415 }
4416 callargs = load_args(pp_stack, na);
4417 if (callargs == NULL)
4418 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004419#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004420 /* At this point, we have to look at the type of func to
4421 update the call stats properly. Do it here so as to avoid
4422 exposing the call stats machinery outside ceval.c
4423 */
4424 if (PyFunction_Check(func))
4425 PCALL(PCALL_FUNCTION);
4426 else if (PyMethod_Check(func))
4427 PCALL(PCALL_METHOD);
4428 else if (PyType_Check(func))
4429 PCALL(PCALL_TYPE);
4430 else if (PyCFunction_Check(func))
4431 PCALL(PCALL_CFUNCTION);
4432 else
4433 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004434#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004435 if (PyCFunction_Check(func)) {
4436 PyThreadState *tstate = PyThreadState_GET();
4437 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4438 }
4439 else
4440 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00004441call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004442 Py_XDECREF(callargs);
4443 Py_XDECREF(kwdict);
4444 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004445}
4446
4447static PyObject *
4448ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4449{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004450 int nstar = 0;
4451 PyObject *callargs = NULL;
4452 PyObject *stararg = NULL;
4453 PyObject *kwdict = NULL;
4454 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004455
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004456 if (flags & CALL_FLAG_KW) {
4457 kwdict = EXT_POP(*pp_stack);
4458 if (!PyDict_Check(kwdict)) {
4459 PyObject *d;
4460 d = PyDict_New();
4461 if (d == NULL)
4462 goto ext_call_fail;
4463 if (PyDict_Update(d, kwdict) != 0) {
4464 Py_DECREF(d);
4465 /* PyDict_Update raises attribute
4466 * error (percolated from an attempt
4467 * to get 'keys' attribute) instead of
4468 * a type error if its second argument
4469 * is not a mapping.
4470 */
4471 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4472 PyErr_Format(PyExc_TypeError,
4473 "%.200s%.200s argument after ** "
4474 "must be a mapping, not %.200s",
4475 PyEval_GetFuncName(func),
4476 PyEval_GetFuncDesc(func),
4477 kwdict->ob_type->tp_name);
4478 }
4479 goto ext_call_fail;
4480 }
4481 Py_DECREF(kwdict);
4482 kwdict = d;
4483 }
4484 }
4485 if (flags & CALL_FLAG_VAR) {
4486 stararg = EXT_POP(*pp_stack);
4487 if (!PyTuple_Check(stararg)) {
4488 PyObject *t = NULL;
4489 t = PySequence_Tuple(stararg);
4490 if (t == NULL) {
4491 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4492 PyErr_Format(PyExc_TypeError,
4493 "%.200s%.200s argument after * "
Victor Stinner0a5f65a2011-03-22 01:09:21 +01004494 "must be a sequence, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004495 PyEval_GetFuncName(func),
4496 PyEval_GetFuncDesc(func),
4497 stararg->ob_type->tp_name);
4498 }
4499 goto ext_call_fail;
4500 }
4501 Py_DECREF(stararg);
4502 stararg = t;
4503 }
4504 nstar = PyTuple_GET_SIZE(stararg);
4505 }
4506 if (nk > 0) {
4507 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
4508 if (kwdict == NULL)
4509 goto ext_call_fail;
4510 }
4511 callargs = update_star_args(na, nstar, stararg, pp_stack);
4512 if (callargs == NULL)
4513 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004514#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004515 /* At this point, we have to look at the type of func to
4516 update the call stats properly. Do it here so as to avoid
4517 exposing the call stats machinery outside ceval.c
4518 */
4519 if (PyFunction_Check(func))
4520 PCALL(PCALL_FUNCTION);
4521 else if (PyMethod_Check(func))
4522 PCALL(PCALL_METHOD);
4523 else if (PyType_Check(func))
4524 PCALL(PCALL_TYPE);
4525 else if (PyCFunction_Check(func))
4526 PCALL(PCALL_CFUNCTION);
4527 else
4528 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004529#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004530 if (PyCFunction_Check(func)) {
4531 PyThreadState *tstate = PyThreadState_GET();
4532 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4533 }
4534 else
4535 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00004536ext_call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004537 Py_XDECREF(callargs);
4538 Py_XDECREF(kwdict);
4539 Py_XDECREF(stararg);
Victor Stinnerf243ee42013-07-16 01:02:12 +02004540 assert((result != NULL && !PyErr_Occurred())
4541 || (result == NULL && PyErr_Occurred()));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004542 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004543}
4544
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004545/* Extract a slice index from a PyInt or PyLong or an object with the
4546 nb_index slot defined, and store in *pi.
4547 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4548 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 +00004549 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004550*/
Tim Petersb5196382001-12-16 19:44:20 +00004551/* Note: If v is NULL, return success without storing into *pi. This
4552 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4553 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004554*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004555int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004556_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004557{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004558 if (v != NULL) {
4559 Py_ssize_t x;
4560 if (PyIndex_Check(v)) {
4561 x = PyNumber_AsSsize_t(v, NULL);
4562 if (x == -1 && PyErr_Occurred())
4563 return 0;
4564 }
4565 else {
4566 PyErr_SetString(PyExc_TypeError,
4567 "slice indices must be integers or "
4568 "None or have an __index__ method");
4569 return 0;
4570 }
4571 *pi = x;
4572 }
4573 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004574}
4575
Guido van Rossum486364b2007-06-30 05:01:58 +00004576#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004577 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00004578
Guido van Rossumb209a111997-04-29 18:18:01 +00004579static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004580cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004581{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582 int res = 0;
4583 switch (op) {
4584 case PyCmp_IS:
4585 res = (v == w);
4586 break;
4587 case PyCmp_IS_NOT:
4588 res = (v != w);
4589 break;
4590 case PyCmp_IN:
4591 res = PySequence_Contains(w, v);
4592 if (res < 0)
4593 return NULL;
4594 break;
4595 case PyCmp_NOT_IN:
4596 res = PySequence_Contains(w, v);
4597 if (res < 0)
4598 return NULL;
4599 res = !res;
4600 break;
4601 case PyCmp_EXC_MATCH:
4602 if (PyTuple_Check(w)) {
4603 Py_ssize_t i, length;
4604 length = PyTuple_Size(w);
4605 for (i = 0; i < length; i += 1) {
4606 PyObject *exc = PyTuple_GET_ITEM(w, i);
4607 if (!PyExceptionClass_Check(exc)) {
4608 PyErr_SetString(PyExc_TypeError,
4609 CANNOT_CATCH_MSG);
4610 return NULL;
4611 }
4612 }
4613 }
4614 else {
4615 if (!PyExceptionClass_Check(w)) {
4616 PyErr_SetString(PyExc_TypeError,
4617 CANNOT_CATCH_MSG);
4618 return NULL;
4619 }
4620 }
4621 res = PyErr_GivenExceptionMatches(v, w);
4622 break;
4623 default:
4624 return PyObject_RichCompare(v, w, op);
4625 }
4626 v = res ? Py_True : Py_False;
4627 Py_INCREF(v);
4628 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004629}
4630
Thomas Wouters52152252000-08-17 22:55:00 +00004631static PyObject *
4632import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004633{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004634 PyObject *x;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004635
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004636 x = PyObject_GetAttr(v, name);
4637 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Brett Cannona79e4fb2013-07-12 11:22:26 -04004638 PyErr_Format(PyExc_ImportError, "cannot import name %R", name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004639 }
4640 return x;
Thomas Wouters52152252000-08-17 22:55:00 +00004641}
Guido van Rossumac7be682001-01-17 15:42:30 +00004642
Thomas Wouters52152252000-08-17 22:55:00 +00004643static int
4644import_all_from(PyObject *locals, PyObject *v)
4645{
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004646 _Py_IDENTIFIER(__all__);
4647 _Py_IDENTIFIER(__dict__);
4648 PyObject *all = _PyObject_GetAttrId(v, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004649 PyObject *dict, *name, *value;
4650 int skip_leading_underscores = 0;
4651 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004652
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004653 if (all == NULL) {
4654 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4655 return -1; /* Unexpected error */
4656 PyErr_Clear();
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004657 dict = _PyObject_GetAttrId(v, &PyId___dict__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004658 if (dict == NULL) {
4659 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4660 return -1;
4661 PyErr_SetString(PyExc_ImportError,
4662 "from-import-* object has no __dict__ and no __all__");
4663 return -1;
4664 }
4665 all = PyMapping_Keys(dict);
4666 Py_DECREF(dict);
4667 if (all == NULL)
4668 return -1;
4669 skip_leading_underscores = 1;
4670 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004671
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004672 for (pos = 0, err = 0; ; pos++) {
4673 name = PySequence_GetItem(all, pos);
4674 if (name == NULL) {
4675 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4676 err = -1;
4677 else
4678 PyErr_Clear();
4679 break;
4680 }
4681 if (skip_leading_underscores &&
4682 PyUnicode_Check(name) &&
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004683 PyUnicode_READY(name) != -1 &&
4684 PyUnicode_READ_CHAR(name, 0) == '_')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004685 {
4686 Py_DECREF(name);
4687 continue;
4688 }
4689 value = PyObject_GetAttr(v, name);
4690 if (value == NULL)
4691 err = -1;
4692 else if (PyDict_CheckExact(locals))
4693 err = PyDict_SetItem(locals, name, value);
4694 else
4695 err = PyObject_SetItem(locals, name, value);
4696 Py_DECREF(name);
4697 Py_XDECREF(value);
4698 if (err != 0)
4699 break;
4700 }
4701 Py_DECREF(all);
4702 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004703}
4704
Guido van Rossumac7be682001-01-17 15:42:30 +00004705static void
Neal Norwitzda059e32007-08-26 05:33:45 +00004706format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00004707{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004708 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00004709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004710 if (!obj)
4711 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004712
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004713 obj_str = _PyUnicode_AsString(obj);
4714 if (!obj_str)
4715 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004716
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004717 PyErr_Format(exc, format_str, obj_str);
Paul Prescode68140d2000-08-30 20:25:01 +00004718}
Guido van Rossum950361c1997-01-24 13:49:28 +00004719
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00004720static void
4721format_exc_unbound(PyCodeObject *co, int oparg)
4722{
4723 PyObject *name;
4724 /* Don't stomp existing exception */
4725 if (PyErr_Occurred())
4726 return;
4727 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
4728 name = PyTuple_GET_ITEM(co->co_cellvars,
4729 oparg);
4730 format_exc_check_arg(
4731 PyExc_UnboundLocalError,
4732 UNBOUNDLOCAL_ERROR_MSG,
4733 name);
4734 } else {
4735 name = PyTuple_GET_ITEM(co->co_freevars, oparg -
4736 PyTuple_GET_SIZE(co->co_cellvars));
4737 format_exc_check_arg(PyExc_NameError,
4738 UNBOUNDFREE_ERROR_MSG, name);
4739 }
4740}
4741
Victor Stinnerd2a915d2011-10-02 20:34:20 +02004742static PyObject *
4743unicode_concatenate(PyObject *v, PyObject *w,
4744 PyFrameObject *f, unsigned char *next_instr)
4745{
4746 PyObject *res;
4747 if (Py_REFCNT(v) == 2) {
4748 /* In the common case, there are 2 references to the value
4749 * stored in 'variable' when the += is performed: one on the
4750 * value stack (in 'v') and one still stored in the
4751 * 'variable'. We try to delete the variable now to reduce
4752 * the refcnt to 1.
4753 */
4754 switch (*next_instr) {
4755 case STORE_FAST:
4756 {
4757 int oparg = PEEKARG();
4758 PyObject **fastlocals = f->f_localsplus;
4759 if (GETLOCAL(oparg) == v)
4760 SETLOCAL(oparg, NULL);
4761 break;
4762 }
4763 case STORE_DEREF:
4764 {
4765 PyObject **freevars = (f->f_localsplus +
4766 f->f_code->co_nlocals);
4767 PyObject *c = freevars[PEEKARG()];
4768 if (PyCell_GET(c) == v)
4769 PyCell_Set(c, NULL);
4770 break;
4771 }
4772 case STORE_NAME:
4773 {
4774 PyObject *names = f->f_code->co_names;
4775 PyObject *name = GETITEM(names, PEEKARG());
4776 PyObject *locals = f->f_locals;
4777 if (PyDict_CheckExact(locals) &&
4778 PyDict_GetItem(locals, name) == v) {
4779 if (PyDict_DelItem(locals, name) != 0) {
4780 PyErr_Clear();
4781 }
4782 }
4783 break;
4784 }
4785 }
4786 }
4787 res = v;
4788 PyUnicode_Append(&res, w);
4789 return res;
4790}
4791
Guido van Rossum950361c1997-01-24 13:49:28 +00004792#ifdef DYNAMIC_EXECUTION_PROFILE
4793
Skip Montanarof118cb12001-10-15 20:51:38 +00004794static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004795getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004796{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004797 int i;
4798 PyObject *l = PyList_New(256);
4799 if (l == NULL) return NULL;
4800 for (i = 0; i < 256; i++) {
4801 PyObject *x = PyLong_FromLong(a[i]);
4802 if (x == NULL) {
4803 Py_DECREF(l);
4804 return NULL;
4805 }
4806 PyList_SetItem(l, i, x);
4807 }
4808 for (i = 0; i < 256; i++)
4809 a[i] = 0;
4810 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004811}
4812
4813PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004814_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004815{
4816#ifndef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004817 return getarray(dxp);
Guido van Rossum950361c1997-01-24 13:49:28 +00004818#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004819 int i;
4820 PyObject *l = PyList_New(257);
4821 if (l == NULL) return NULL;
4822 for (i = 0; i < 257; i++) {
4823 PyObject *x = getarray(dxpairs[i]);
4824 if (x == NULL) {
4825 Py_DECREF(l);
4826 return NULL;
4827 }
4828 PyList_SetItem(l, i, x);
4829 }
4830 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004831#endif
4832}
4833
4834#endif