blob: a32d685f08cf447449d5d7353b186ec7403118c8 [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"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000145#define GLOBAL_NAME_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000146 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000147#define UNBOUNDLOCAL_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000148 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000149#define UNBOUNDFREE_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000150 "free variable '%.200s' referenced before assignment" \
151 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000152
Guido van Rossum950361c1997-01-24 13:49:28 +0000153/* Dynamic execution profile */
154#ifdef DYNAMIC_EXECUTION_PROFILE
155#ifdef DXPAIRS
156static long dxpairs[257][256];
157#define dxp dxpairs[256]
158#else
159static long dxp[256];
160#endif
161#endif
162
Jeremy Hylton985eba52003-02-05 23:13:00 +0000163/* Function call profile */
164#ifdef CALL_PROFILE
165#define PCALL_NUM 11
166static int pcall[PCALL_NUM];
167
168#define PCALL_ALL 0
169#define PCALL_FUNCTION 1
170#define PCALL_FAST_FUNCTION 2
171#define PCALL_FASTER_FUNCTION 3
172#define PCALL_METHOD 4
173#define PCALL_BOUND_METHOD 5
174#define PCALL_CFUNCTION 6
175#define PCALL_TYPE 7
176#define PCALL_GENERATOR 8
177#define PCALL_OTHER 9
178#define PCALL_POP 10
179
180/* Notes about the statistics
181
182 PCALL_FAST stats
183
184 FAST_FUNCTION means no argument tuple needs to be created.
185 FASTER_FUNCTION means that the fast-path frame setup code is used.
186
187 If there is a method call where the call can be optimized by changing
188 the argument tuple and calling the function directly, it gets recorded
189 twice.
190
191 As a result, the relationship among the statistics appears to be
192 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
193 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
194 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
195 PCALL_METHOD > PCALL_BOUND_METHOD
196*/
197
198#define PCALL(POS) pcall[POS]++
199
200PyObject *
201PyEval_GetCallStats(PyObject *self)
202{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000203 return Py_BuildValue("iiiiiiiiiii",
204 pcall[0], pcall[1], pcall[2], pcall[3],
205 pcall[4], pcall[5], pcall[6], pcall[7],
206 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000207}
208#else
209#define PCALL(O)
210
211PyObject *
212PyEval_GetCallStats(PyObject *self)
213{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000214 Py_INCREF(Py_None);
215 return Py_None;
Jeremy Hylton985eba52003-02-05 23:13:00 +0000216}
217#endif
218
Tim Peters5ca576e2001-06-18 22:08:13 +0000219
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000220#ifdef WITH_THREAD
221#define GIL_REQUEST _Py_atomic_load_relaxed(&gil_drop_request)
222#else
223#define GIL_REQUEST 0
224#endif
225
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000226/* This can set eval_breaker to 0 even though gil_drop_request became
227 1. We believe this is all right because the eval loop will release
228 the GIL eventually anyway. */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000229#define COMPUTE_EVAL_BREAKER() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000230 _Py_atomic_store_relaxed( \
231 &eval_breaker, \
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000232 GIL_REQUEST | \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000233 _Py_atomic_load_relaxed(&pendingcalls_to_do) | \
234 pending_async_exc)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000235
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000236#ifdef WITH_THREAD
237
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000238#define SET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000239 do { \
240 _Py_atomic_store_relaxed(&gil_drop_request, 1); \
241 _Py_atomic_store_relaxed(&eval_breaker, 1); \
242 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000243
244#define RESET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 do { \
246 _Py_atomic_store_relaxed(&gil_drop_request, 0); \
247 COMPUTE_EVAL_BREAKER(); \
248 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000249
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000250#endif
251
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000252/* Pending calls are only modified under pending_lock */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000253#define SIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000254 do { \
255 _Py_atomic_store_relaxed(&pendingcalls_to_do, 1); \
256 _Py_atomic_store_relaxed(&eval_breaker, 1); \
257 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000258
259#define UNSIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000260 do { \
261 _Py_atomic_store_relaxed(&pendingcalls_to_do, 0); \
262 COMPUTE_EVAL_BREAKER(); \
263 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000264
265#define SIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000266 do { \
267 pending_async_exc = 1; \
268 _Py_atomic_store_relaxed(&eval_breaker, 1); \
269 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000270
271#define UNSIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000272 do { pending_async_exc = 0; COMPUTE_EVAL_BREAKER(); } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000273
274
Guido van Rossume59214e1994-08-30 08:01:59 +0000275#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000276
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000278#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000279#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000280#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000281
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000282static PyThread_type_lock pending_lock = 0; /* for pending calls */
Guido van Rossuma9672091994-09-14 13:31:22 +0000283static long main_thread = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000284/* This single variable consolidates all requests to break out of the fast path
285 in the eval loop. */
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000286static _Py_atomic_int eval_breaker = {0};
287/* Request for dropping the GIL */
288static _Py_atomic_int gil_drop_request = {0};
289/* Request for running pending calls. */
290static _Py_atomic_int pendingcalls_to_do = {0};
291/* Request for looking at the `async_exc` field of the current thread state.
292 Guarded by the GIL. */
293static int pending_async_exc = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000294
295#include "ceval_gil.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000296
Tim Peters7f468f22004-10-11 02:40:51 +0000297int
298PyEval_ThreadsInitialized(void)
299{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000300 return gil_created();
Tim Peters7f468f22004-10-11 02:40:51 +0000301}
302
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000303void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000304PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000305{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000306 if (gil_created())
307 return;
308 create_gil();
309 take_gil(PyThreadState_GET());
310 main_thread = PyThread_get_thread_ident();
311 if (!pending_lock)
312 pending_lock = PyThread_allocate_lock();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000313}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000314
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000315void
Antoine Pitrou1df15362010-09-13 14:16:46 +0000316_PyEval_FiniThreads(void)
317{
318 if (!gil_created())
319 return;
320 destroy_gil();
321 assert(!gil_created());
322}
323
324void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000325PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000326{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000327 PyThreadState *tstate = PyThreadState_GET();
328 if (tstate == NULL)
329 Py_FatalError("PyEval_AcquireLock: current thread state is NULL");
330 take_gil(tstate);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000331}
332
333void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000334PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000335{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000336 /* This function must succeed when the current thread state is NULL.
337 We therefore avoid PyThreadState_GET() which dumps a fatal error
338 in debug mode.
339 */
340 drop_gil((PyThreadState*)_Py_atomic_load_relaxed(
341 &_PyThreadState_Current));
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342}
343
344void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000345PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000346{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000347 if (tstate == NULL)
348 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
349 /* Check someone has called PyEval_InitThreads() to create the lock */
350 assert(gil_created());
351 take_gil(tstate);
352 if (PyThreadState_Swap(tstate) != NULL)
353 Py_FatalError(
354 "PyEval_AcquireThread: non-NULL old thread state");
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000355}
356
357void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000358PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000359{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000360 if (tstate == NULL)
361 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
362 if (PyThreadState_Swap(NULL) != tstate)
363 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
364 drop_gil(tstate);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000365}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000366
367/* This function is called from PyOS_AfterFork to ensure that newly
368 created child processes don't hold locks referring to threads which
369 are not running in the child process. (This could also be done using
370 pthread_atfork mechanism, at least for the pthreads implementation.) */
371
372void
373PyEval_ReInitThreads(void)
374{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200375 _Py_IDENTIFIER(_after_fork);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000376 PyObject *threading, *result;
377 PyThreadState *tstate = PyThreadState_GET();
Jesse Nollera8513972008-07-17 16:49:17 +0000378
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000379 if (!gil_created())
380 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000381 recreate_gil();
382 pending_lock = PyThread_allocate_lock();
383 take_gil(tstate);
384 main_thread = PyThread_get_thread_ident();
Jesse Nollera8513972008-07-17 16:49:17 +0000385
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000386 /* Update the threading module with the new state.
387 */
388 tstate = PyThreadState_GET();
389 threading = PyMapping_GetItemString(tstate->interp->modules,
390 "threading");
391 if (threading == NULL) {
392 /* threading not imported */
393 PyErr_Clear();
394 return;
395 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200396 result = _PyObject_CallMethodId(threading, &PyId__after_fork, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000397 if (result == NULL)
398 PyErr_WriteUnraisable(threading);
399 else
400 Py_DECREF(result);
401 Py_DECREF(threading);
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 */
745 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
746 WHY_RETURN = 0x0008, /* 'return' statement */
747 WHY_BREAK = 0x0010, /* 'break' statement */
748 WHY_CONTINUE = 0x0020, /* 'continue' statement */
749 WHY_YIELD = 0x0040, /* 'yield' operator */
750 WHY_SILENCED = 0x0080 /* Exception silenced by 'with' */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000751};
Guido van Rossum374a9221991-04-04 10:40:29 +0000752
Benjamin Peterson87880242011-07-03 16:48:31 -0500753static void save_exc_state(PyThreadState *, PyFrameObject *);
754static void swap_exc_state(PyThreadState *, PyFrameObject *);
755static void restore_and_clear_exc_state(PyThreadState *, PyFrameObject *);
Collin Winter828f04a2007-08-31 00:04:24 +0000756static enum why_code do_raise(PyObject *, PyObject *);
Guido van Rossum0368b722007-05-11 16:50:42 +0000757static int unpack_iterable(PyObject *, int, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000758
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +0000759/* Records whether tracing is on for any thread. Counts the number of
760 threads for which tstate->c_tracefunc is non-NULL, so if the value
761 is 0, we know we don't have to check this thread's c_tracefunc.
762 This speeds up the if statement in PyEval_EvalFrameEx() after
763 fast_next_opcode*/
764static int _Py_TracingPossible = 0;
765
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000766
Guido van Rossum374a9221991-04-04 10:40:29 +0000767
Guido van Rossumb209a111997-04-29 18:18:01 +0000768PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000769PyEval_EvalCode(PyObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000770{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000771 return PyEval_EvalCodeEx(co,
772 globals, locals,
773 (PyObject **)NULL, 0,
774 (PyObject **)NULL, 0,
775 (PyObject **)NULL, 0,
776 NULL, NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000777}
778
779
780/* Interpreter main loop */
781
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000782PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000783PyEval_EvalFrame(PyFrameObject *f) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000784 /* This is for backward compatibility with extension modules that
785 used this API; core interpreter code should call
786 PyEval_EvalFrameEx() */
787 return PyEval_EvalFrameEx(f, 0);
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000788}
789
790PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000791PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000792{
Guido van Rossum950361c1997-01-24 13:49:28 +0000793#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000794 int lastopcode = 0;
Guido van Rossum950361c1997-01-24 13:49:28 +0000795#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000796 register PyObject **stack_pointer; /* Next free slot in value stack */
797 register unsigned char *next_instr;
798 register int opcode; /* Current opcode */
799 register int oparg; /* Current opcode argument, if any */
800 register enum why_code why; /* Reason for block stack unwind */
801 register int err; /* Error status -- nonzero if error */
802 register PyObject *x; /* Result object -- NULL if error */
803 register PyObject *v; /* Temporary objects popped off stack */
804 register PyObject *w;
805 register PyObject *u;
806 register PyObject *t;
807 register PyObject **fastlocals, **freevars;
808 PyObject *retval = NULL; /* Return value */
809 PyThreadState *tstate = PyThreadState_GET();
810 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000811
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000812 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000813
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 not (instr_lb <= current_bytecode_offset < instr_ub)
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 is true when the line being executed has changed. The
817 initial values are such as to make this false the first
818 time it is tested. */
819 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000821 unsigned char *first_instr;
822 PyObject *names;
823 PyObject *consts;
Guido van Rossum374a9221991-04-04 10:40:29 +0000824
Brett Cannon368b4b72012-04-02 12:17:59 -0400825#ifdef LLTRACE
Victor Stinner3c1e4812012-03-26 22:10:51 +0200826 _Py_IDENTIFIER(__ltrace__);
Brett Cannon368b4b72012-04-02 12:17:59 -0400827#endif
Victor Stinner3c1e4812012-03-26 22:10:51 +0200828
Antoine Pitroub52ec782009-01-25 16:34:23 +0000829/* Computed GOTOs, or
830 the-optimization-commonly-but-improperly-known-as-"threaded code"
831 using gcc's labels-as-values extension
832 (http://gcc.gnu.org/onlinedocs/gcc/Labels-as-Values.html).
833
834 The traditional bytecode evaluation loop uses a "switch" statement, which
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000835 decent compilers will optimize as a single indirect branch instruction
Antoine Pitroub52ec782009-01-25 16:34:23 +0000836 combined with a lookup table of jump addresses. However, since the
837 indirect jump instruction is shared by all opcodes, the CPU will have a
838 hard time making the right prediction for where to jump next (actually,
839 it will be always wrong except in the uncommon case of a sequence of
840 several identical opcodes).
841
842 "Threaded code" in contrast, uses an explicit jump table and an explicit
843 indirect jump instruction at the end of each opcode. Since the jump
844 instruction is at a different address for each opcode, the CPU will make a
845 separate prediction for each of these instructions, which is equivalent to
846 predicting the second opcode of each opcode pair. These predictions have
847 a much better chance to turn out valid, especially in small bytecode loops.
848
849 A mispredicted branch on a modern CPU flushes the whole pipeline and
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000850 can cost several CPU cycles (depending on the pipeline depth),
Antoine Pitroub52ec782009-01-25 16:34:23 +0000851 and potentially many more instructions (depending on the pipeline width).
852 A correctly predicted branch, however, is nearly free.
853
854 At the time of this writing, the "threaded code" version is up to 15-20%
855 faster than the normal "switch" version, depending on the compiler and the
856 CPU architecture.
857
858 We disable the optimization if DYNAMIC_EXECUTION_PROFILE is defined,
859 because it would render the measurements invalid.
860
861
862 NOTE: care must be taken that the compiler doesn't try to "optimize" the
863 indirect jumps by sharing them between all opcodes. Such optimizations
864 can be disabled on gcc by using the -fno-gcse flag (or possibly
865 -fno-crossjumping).
866*/
867
Antoine Pitrou042b1282010-08-13 21:15:58 +0000868#ifdef DYNAMIC_EXECUTION_PROFILE
Antoine Pitroub52ec782009-01-25 16:34:23 +0000869#undef USE_COMPUTED_GOTOS
Antoine Pitrou042b1282010-08-13 21:15:58 +0000870#define USE_COMPUTED_GOTOS 0
Antoine Pitroub52ec782009-01-25 16:34:23 +0000871#endif
872
Antoine Pitrou042b1282010-08-13 21:15:58 +0000873#ifdef HAVE_COMPUTED_GOTOS
874 #ifndef USE_COMPUTED_GOTOS
875 #define USE_COMPUTED_GOTOS 1
876 #endif
877#else
878 #if defined(USE_COMPUTED_GOTOS) && USE_COMPUTED_GOTOS
879 #error "Computed gotos are not supported on this compiler."
880 #endif
881 #undef USE_COMPUTED_GOTOS
882 #define USE_COMPUTED_GOTOS 0
883#endif
884
885#if USE_COMPUTED_GOTOS
Antoine Pitroub52ec782009-01-25 16:34:23 +0000886/* Import the static jump table */
887#include "opcode_targets.h"
888
889/* This macro is used when several opcodes defer to the same implementation
890 (e.g. SETUP_LOOP, SETUP_FINALLY) */
891#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000892 TARGET_##op: \
893 opcode = op; \
894 if (HAS_ARG(op)) \
895 oparg = NEXTARG(); \
896 case op: \
897 goto impl; \
Antoine Pitroub52ec782009-01-25 16:34:23 +0000898
899#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000900 TARGET_##op: \
901 opcode = op; \
902 if (HAS_ARG(op)) \
903 oparg = NEXTARG(); \
904 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000905
906
907#define DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000908 { \
909 if (!_Py_atomic_load_relaxed(&eval_breaker)) { \
910 FAST_DISPATCH(); \
911 } \
912 continue; \
913 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000914
915#ifdef LLTRACE
916#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000917 { \
918 if (!lltrace && !_Py_TracingPossible) { \
919 f->f_lasti = INSTR_OFFSET(); \
920 goto *opcode_targets[*next_instr++]; \
921 } \
922 goto fast_next_opcode; \
923 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000924#else
925#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000926 { \
927 if (!_Py_TracingPossible) { \
928 f->f_lasti = INSTR_OFFSET(); \
929 goto *opcode_targets[*next_instr++]; \
930 } \
931 goto fast_next_opcode; \
932 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000933#endif
934
935#else
936#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000937 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000938#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000939 /* silence compiler warnings about `impl` unused */ \
940 if (0) goto impl; \
941 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000942#define DISPATCH() continue
943#define FAST_DISPATCH() goto fast_next_opcode
944#endif
945
946
Neal Norwitza81d2202002-07-14 00:27:26 +0000947/* Tuple access macros */
948
949#ifndef Py_DEBUG
950#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
951#else
952#define GETITEM(v, i) PyTuple_GetItem((v), (i))
953#endif
954
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000955#ifdef WITH_TSC
956/* Use Pentium timestamp counter to mark certain events:
957 inst0 -- beginning of switch statement for opcode dispatch
958 inst1 -- end of switch statement (may be skipped)
959 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000960 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000961 (may be skipped)
962 intr1 -- beginning of long interruption
963 intr2 -- end of long interruption
964
965 Many opcodes call out to helper C functions. In some cases, the
966 time in those functions should be counted towards the time for the
967 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
968 calls another Python function; there's no point in charge all the
969 bytecode executed by the called function to the caller.
970
971 It's hard to make a useful judgement statically. In the presence
972 of operator overloading, it's impossible to tell if a call will
973 execute new Python code or not.
974
975 It's a case-by-case judgement. I'll use intr1 for the following
976 cases:
977
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000978 IMPORT_STAR
979 IMPORT_FROM
980 CALL_FUNCTION (and friends)
981
982 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000983 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
984 int ticked = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000985
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000986 READ_TIMESTAMP(inst0);
987 READ_TIMESTAMP(inst1);
988 READ_TIMESTAMP(loop0);
989 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000990
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000991 /* shut up the compiler */
992 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000993#endif
994
Guido van Rossum374a9221991-04-04 10:40:29 +0000995/* Code access macros */
996
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000997#define INSTR_OFFSET() ((int)(next_instr - first_instr))
998#define NEXTOP() (*next_instr++)
999#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
1000#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
1001#define JUMPTO(x) (next_instr = first_instr + (x))
1002#define JUMPBY(x) (next_instr += (x))
Guido van Rossum374a9221991-04-04 10:40:29 +00001003
Raymond Hettingerf606f872003-03-16 03:11:04 +00001004/* OpCode prediction macros
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001005 Some opcodes tend to come in pairs thus making it possible to
1006 predict the second code when the first is run. For example,
1007 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
1008 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001009
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001010 Verifying the prediction costs a single high-speed test of a register
1011 variable against a constant. If the pairing was good, then the
1012 processor's own internal branch predication has a high likelihood of
1013 success, resulting in a nearly zero-overhead transition to the
1014 next opcode. A successful prediction saves a trip through the eval-loop
1015 including its two unpredictable branches, the HAS_ARG test and the
1016 switch-case. Combined with the processor's internal branch prediction,
1017 a successful PREDICT has the effect of making the two opcodes run as if
1018 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001019
Georg Brandl86b2fb92008-07-16 03:43:04 +00001020 If collecting opcode statistics, your choices are to either keep the
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 predictions turned-on and interpret the results as if some opcodes
1022 had been combined or turn-off predictions so that the opcode frequency
1023 counter updates for both opcodes.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001024
1025 Opcode prediction is disabled with threaded code, since the latter allows
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001026 the CPU to record separate branch prediction information for each
1027 opcode.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001028
Raymond Hettingerf606f872003-03-16 03:11:04 +00001029*/
1030
Antoine Pitrou042b1282010-08-13 21:15:58 +00001031#if defined(DYNAMIC_EXECUTION_PROFILE) || USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001032#define PREDICT(op) if (0) goto PRED_##op
1033#define PREDICTED(op) PRED_##op:
1034#define PREDICTED_WITH_ARG(op) PRED_##op:
Raymond Hettingera7216982004-02-08 19:59:27 +00001035#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001036#define PREDICT(op) if (*next_instr == op) goto PRED_##op
1037#define PREDICTED(op) PRED_##op: next_instr++
1038#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Antoine Pitroub52ec782009-01-25 16:34:23 +00001039#endif
1040
Raymond Hettingerf606f872003-03-16 03:11:04 +00001041
Guido van Rossum374a9221991-04-04 10:40:29 +00001042/* Stack manipulation macros */
1043
Martin v. Löwis18e16552006-02-15 17:27:45 +00001044/* The stack can grow at most MAXINT deep, as co_nlocals and
1045 co_stacksize are ints. */
Stefan Krahb7e10102010-06-23 18:42:39 +00001046#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
1047#define EMPTY() (STACK_LEVEL() == 0)
1048#define TOP() (stack_pointer[-1])
1049#define SECOND() (stack_pointer[-2])
1050#define THIRD() (stack_pointer[-3])
1051#define FOURTH() (stack_pointer[-4])
1052#define PEEK(n) (stack_pointer[-(n)])
1053#define SET_TOP(v) (stack_pointer[-1] = (v))
1054#define SET_SECOND(v) (stack_pointer[-2] = (v))
1055#define SET_THIRD(v) (stack_pointer[-3] = (v))
1056#define SET_FOURTH(v) (stack_pointer[-4] = (v))
1057#define SET_VALUE(n, v) (stack_pointer[-(n)] = (v))
1058#define BASIC_STACKADJ(n) (stack_pointer += n)
1059#define BASIC_PUSH(v) (*stack_pointer++ = (v))
1060#define BASIC_POP() (*--stack_pointer)
Guido van Rossum374a9221991-04-04 10:40:29 +00001061
Guido van Rossum96a42c81992-01-12 02:29:51 +00001062#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063#define PUSH(v) { (void)(BASIC_PUSH(v), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001064 lltrace && prtrace(TOP(), "push")); \
1065 assert(STACK_LEVEL() <= co->co_stacksize); }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001067 BASIC_POP())
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001068#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001069 lltrace && prtrace(TOP(), "stackadj")); \
1070 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes0449f632007-12-15 01:27:15 +00001071#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
Stefan Krahb7e10102010-06-23 18:42:39 +00001072 prtrace((STACK_POINTER)[-1], "ext_pop")), \
1073 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001074#else
Stefan Krahb7e10102010-06-23 18:42:39 +00001075#define PUSH(v) BASIC_PUSH(v)
1076#define POP() BASIC_POP()
1077#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001078#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001079#endif
1080
Guido van Rossum681d79a1995-07-18 14:51:37 +00001081/* Local variable macros */
1082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001083#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +00001084
1085/* The SETLOCAL() macro must not DECREF the local variable in-place and
1086 then store the new value; it must copy the old value to a temporary
1087 value, then store the new value, and then DECREF the temporary value.
1088 This is because it is possible that during the DECREF the frame is
1089 accessed by other code (e.g. a __del__ method or gc.collect()) and the
1090 variable would be pointing to already-freed memory. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001091#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
Stefan Krahb7e10102010-06-23 18:42:39 +00001092 GETLOCAL(i) = value; \
1093 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +00001094
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001095
1096#define UNWIND_BLOCK(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001097 while (STACK_LEVEL() > (b)->b_level) { \
1098 PyObject *v = POP(); \
1099 Py_XDECREF(v); \
1100 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001101
1102#define UNWIND_EXCEPT_HANDLER(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001103 { \
1104 PyObject *type, *value, *traceback; \
1105 assert(STACK_LEVEL() >= (b)->b_level + 3); \
1106 while (STACK_LEVEL() > (b)->b_level + 3) { \
1107 value = POP(); \
1108 Py_XDECREF(value); \
1109 } \
1110 type = tstate->exc_type; \
1111 value = tstate->exc_value; \
1112 traceback = tstate->exc_traceback; \
1113 tstate->exc_type = POP(); \
1114 tstate->exc_value = POP(); \
1115 tstate->exc_traceback = POP(); \
1116 Py_XDECREF(type); \
1117 Py_XDECREF(value); \
1118 Py_XDECREF(traceback); \
1119 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001120
Guido van Rossuma027efa1997-05-05 20:56:21 +00001121/* Start of code */
1122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 /* push frame */
1124 if (Py_EnterRecursiveCall(""))
1125 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +00001126
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001127 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +00001128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001129 if (tstate->use_tracing) {
1130 if (tstate->c_tracefunc != NULL) {
1131 /* tstate->c_tracefunc, if defined, is a
1132 function that will be called on *every* entry
1133 to a code block. Its return value, if not
1134 None, is a function that will be called at
1135 the start of each executed line of code.
1136 (Actually, the function must return itself
1137 in order to continue tracing.) The trace
1138 functions are called with three arguments:
1139 a pointer to the current frame, a string
1140 indicating why the function is called, and
1141 an argument which depends on the situation.
1142 The global trace function is also called
1143 whenever an exception is detected. */
1144 if (call_trace_protected(tstate->c_tracefunc,
1145 tstate->c_traceobj,
1146 f, PyTrace_CALL, Py_None)) {
1147 /* Trace function raised an error */
1148 goto exit_eval_frame;
1149 }
1150 }
1151 if (tstate->c_profilefunc != NULL) {
1152 /* Similar for c_profilefunc, except it needn't
1153 return itself and isn't called for "line" events */
1154 if (call_trace_protected(tstate->c_profilefunc,
1155 tstate->c_profileobj,
1156 f, PyTrace_CALL, Py_None)) {
1157 /* Profile function raised an error */
1158 goto exit_eval_frame;
1159 }
1160 }
1161 }
Neil Schemenauer6c0f2002001-09-04 19:03:35 +00001162
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001163 co = f->f_code;
1164 names = co->co_names;
1165 consts = co->co_consts;
1166 fastlocals = f->f_localsplus;
1167 freevars = f->f_localsplus + co->co_nlocals;
1168 first_instr = (unsigned char*) PyBytes_AS_STRING(co->co_code);
1169 /* An explanation is in order for the next line.
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001170
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001171 f->f_lasti now refers to the index of the last instruction
1172 executed. You might think this was obvious from the name, but
1173 this wasn't always true before 2.3! PyFrame_New now sets
1174 f->f_lasti to -1 (i.e. the index *before* the first instruction)
1175 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
1176 does work. Promise.
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001177 YIELD_FROM sets f_lasti to itself, in order to repeated yield
1178 multiple values.
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001180 When the PREDICT() macros are enabled, some opcode pairs follow in
1181 direct succession without updating f->f_lasti. A successful
1182 prediction effectively links the two codes together as if they
1183 were a single new opcode; accordingly,f->f_lasti will point to
1184 the first code in the pair (for instance, GET_ITER followed by
1185 FOR_ITER is effectively a single opcode and f->f_lasti will point
1186 at to the beginning of the combined pair.)
1187 */
1188 next_instr = first_instr + f->f_lasti + 1;
1189 stack_pointer = f->f_stacktop;
1190 assert(stack_pointer != NULL);
1191 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 if (co->co_flags & CO_GENERATOR && !throwflag) {
1194 if (f->f_exc_type != NULL && f->f_exc_type != Py_None) {
1195 /* We were in an except handler when we left,
1196 restore the exception state which was put aside
1197 (see YIELD_VALUE). */
Benjamin Peterson87880242011-07-03 16:48:31 -05001198 swap_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 }
Benjamin Peterson87880242011-07-03 16:48:31 -05001200 else
1201 save_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001203
Tim Peters5ca576e2001-06-18 22:08:13 +00001204#ifdef LLTRACE
Victor Stinner3c1e4812012-03-26 22:10:51 +02001205 lltrace = _PyDict_GetItemId(f->f_globals, &PyId___ltrace__) != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00001206#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00001207
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 why = WHY_NOT;
1209 err = 0;
1210 x = Py_None; /* Not a reference, just anything non-NULL */
1211 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00001212
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 if (throwflag) { /* support for generator.throw() */
1214 why = WHY_EXCEPTION;
1215 goto on_error;
1216 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001217
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001219#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 if (inst1 == 0) {
1221 /* Almost surely, the opcode executed a break
1222 or a continue, preventing inst1 from being set
1223 on the way out of the loop.
1224 */
1225 READ_TIMESTAMP(inst1);
1226 loop1 = inst1;
1227 }
1228 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
1229 intr0, intr1);
1230 ticked = 0;
1231 inst1 = 0;
1232 intr0 = 0;
1233 intr1 = 0;
1234 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001235#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001236 assert(stack_pointer >= f->f_valuestack); /* else underflow */
1237 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001238
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001239 /* Do periodic things. Doing this every time through
1240 the loop would add too much overhead, so we do it
1241 only every Nth instruction. We also do it if
1242 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
1243 event needs attention (e.g. a signal handler or
1244 async I/O handler); see Py_AddPendingCall() and
1245 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +00001246
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001247 if (_Py_atomic_load_relaxed(&eval_breaker)) {
1248 if (*next_instr == SETUP_FINALLY) {
1249 /* Make the last opcode before
Ezio Melotti13925002011-03-16 11:05:33 +02001250 a try: finally: block uninterruptible. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 goto fast_next_opcode;
1252 }
1253 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001254#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 ticked = 1;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001256#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001257 if (_Py_atomic_load_relaxed(&pendingcalls_to_do)) {
1258 if (Py_MakePendingCalls() < 0) {
1259 why = WHY_EXCEPTION;
1260 goto on_error;
1261 }
1262 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001263#ifdef WITH_THREAD
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001264 if (_Py_atomic_load_relaxed(&gil_drop_request)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001265 /* Give another thread a chance */
1266 if (PyThreadState_Swap(NULL) != tstate)
1267 Py_FatalError("ceval: tstate mix-up");
1268 drop_gil(tstate);
1269
1270 /* Other threads may run now */
1271
1272 take_gil(tstate);
1273 if (PyThreadState_Swap(tstate) != NULL)
1274 Py_FatalError("ceval: orphan tstate");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001275 }
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001276#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001277 /* Check for asynchronous exceptions. */
1278 if (tstate->async_exc != NULL) {
1279 x = tstate->async_exc;
1280 tstate->async_exc = NULL;
1281 UNSIGNAL_ASYNC_EXC();
1282 PyErr_SetNone(x);
1283 Py_DECREF(x);
1284 why = WHY_EXCEPTION;
1285 goto on_error;
1286 }
1287 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001288
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001289 fast_next_opcode:
1290 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001291
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001292 /* line-by-line tracing support */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001294 if (_Py_TracingPossible &&
1295 tstate->c_tracefunc != NULL && !tstate->tracing) {
1296 /* see maybe_call_line_trace
1297 for expository comments */
1298 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001299
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001300 err = maybe_call_line_trace(tstate->c_tracefunc,
1301 tstate->c_traceobj,
1302 f, &instr_lb, &instr_ub,
1303 &instr_prev);
1304 /* Reload possibly changed frame fields */
1305 JUMPTO(f->f_lasti);
1306 if (f->f_stacktop != NULL) {
1307 stack_pointer = f->f_stacktop;
1308 f->f_stacktop = NULL;
1309 }
1310 if (err) {
1311 /* trace function raised an exception */
1312 goto on_error;
1313 }
1314 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001315
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001316 /* Extract opcode and argument */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001318 opcode = NEXTOP();
1319 oparg = 0; /* allows oparg to be stored in a register because
1320 it doesn't have to be remembered across a full loop */
1321 if (HAS_ARG(opcode))
1322 oparg = NEXTARG();
Stefan Krahb7e10102010-06-23 18:42:39 +00001323 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001324#ifdef DYNAMIC_EXECUTION_PROFILE
1325#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001326 dxpairs[lastopcode][opcode]++;
1327 lastopcode = opcode;
Guido van Rossum950361c1997-01-24 13:49:28 +00001328#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001329 dxp[opcode]++;
Guido van Rossum950361c1997-01-24 13:49:28 +00001330#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001331
Guido van Rossum96a42c81992-01-12 02:29:51 +00001332#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001333 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001334
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001335 if (lltrace) {
1336 if (HAS_ARG(opcode)) {
1337 printf("%d: %d, %d\n",
1338 f->f_lasti, opcode, oparg);
1339 }
1340 else {
1341 printf("%d: %d\n",
1342 f->f_lasti, opcode);
1343 }
1344 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001345#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001346
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001347 /* Main switch on opcode */
1348 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001349
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001350 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001351
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001352 /* BEWARE!
1353 It is essential that any operation that fails sets either
1354 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1355 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001356
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001357 TARGET(NOP)
1358 FAST_DISPATCH();
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001359
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001360 TARGET(LOAD_FAST)
1361 x = GETLOCAL(oparg);
1362 if (x != NULL) {
1363 Py_INCREF(x);
1364 PUSH(x);
1365 FAST_DISPATCH();
1366 }
1367 format_exc_check_arg(PyExc_UnboundLocalError,
1368 UNBOUNDLOCAL_ERROR_MSG,
1369 PyTuple_GetItem(co->co_varnames, oparg));
1370 break;
Neil Schemenauer63543862002-02-17 19:10:14 +00001371
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 TARGET(LOAD_CONST)
1373 x = GETITEM(consts, oparg);
1374 Py_INCREF(x);
1375 PUSH(x);
1376 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 PREDICTED_WITH_ARG(STORE_FAST);
1379 TARGET(STORE_FAST)
1380 v = POP();
1381 SETLOCAL(oparg, v);
1382 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001383
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001384 TARGET(POP_TOP)
1385 v = POP();
1386 Py_DECREF(v);
1387 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001388
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001389 TARGET(ROT_TWO)
1390 v = TOP();
1391 w = SECOND();
1392 SET_TOP(w);
1393 SET_SECOND(v);
1394 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001395
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001396 TARGET(ROT_THREE)
1397 v = TOP();
1398 w = SECOND();
1399 x = THIRD();
1400 SET_TOP(w);
1401 SET_SECOND(x);
1402 SET_THIRD(v);
1403 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001404
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001405 TARGET(DUP_TOP)
1406 v = TOP();
1407 Py_INCREF(v);
1408 PUSH(v);
1409 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001410
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001411 TARGET(DUP_TOP_TWO)
1412 x = TOP();
1413 Py_INCREF(x);
1414 w = SECOND();
1415 Py_INCREF(w);
1416 STACKADJ(2);
1417 SET_TOP(x);
1418 SET_SECOND(w);
1419 FAST_DISPATCH();
Thomas Wouters434d0822000-08-24 20:11:32 +00001420
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001421 TARGET(UNARY_POSITIVE)
1422 v = TOP();
1423 x = PyNumber_Positive(v);
1424 Py_DECREF(v);
1425 SET_TOP(x);
1426 if (x != NULL) DISPATCH();
1427 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001428
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001429 TARGET(UNARY_NEGATIVE)
1430 v = TOP();
1431 x = PyNumber_Negative(v);
1432 Py_DECREF(v);
1433 SET_TOP(x);
1434 if (x != NULL) DISPATCH();
1435 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001436
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001437 TARGET(UNARY_NOT)
1438 v = TOP();
1439 err = PyObject_IsTrue(v);
1440 Py_DECREF(v);
1441 if (err == 0) {
1442 Py_INCREF(Py_True);
1443 SET_TOP(Py_True);
1444 DISPATCH();
1445 }
1446 else if (err > 0) {
1447 Py_INCREF(Py_False);
1448 SET_TOP(Py_False);
1449 err = 0;
1450 DISPATCH();
1451 }
1452 STACKADJ(-1);
1453 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001454
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001455 TARGET(UNARY_INVERT)
1456 v = TOP();
1457 x = PyNumber_Invert(v);
1458 Py_DECREF(v);
1459 SET_TOP(x);
1460 if (x != NULL) DISPATCH();
1461 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001463 TARGET(BINARY_POWER)
1464 w = POP();
1465 v = TOP();
1466 x = PyNumber_Power(v, w, Py_None);
1467 Py_DECREF(v);
1468 Py_DECREF(w);
1469 SET_TOP(x);
1470 if (x != NULL) DISPATCH();
1471 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001473 TARGET(BINARY_MULTIPLY)
1474 w = POP();
1475 v = TOP();
1476 x = PyNumber_Multiply(v, w);
1477 Py_DECREF(v);
1478 Py_DECREF(w);
1479 SET_TOP(x);
1480 if (x != NULL) DISPATCH();
1481 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001482
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001483 TARGET(BINARY_TRUE_DIVIDE)
1484 w = POP();
1485 v = TOP();
1486 x = PyNumber_TrueDivide(v, w);
1487 Py_DECREF(v);
1488 Py_DECREF(w);
1489 SET_TOP(x);
1490 if (x != NULL) DISPATCH();
1491 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001492
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001493 TARGET(BINARY_FLOOR_DIVIDE)
1494 w = POP();
1495 v = TOP();
1496 x = PyNumber_FloorDivide(v, w);
1497 Py_DECREF(v);
1498 Py_DECREF(w);
1499 SET_TOP(x);
1500 if (x != NULL) DISPATCH();
1501 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001502
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001503 TARGET(BINARY_MODULO)
1504 w = POP();
1505 v = TOP();
1506 if (PyUnicode_CheckExact(v))
1507 x = PyUnicode_Format(v, w);
1508 else
1509 x = PyNumber_Remainder(v, w);
1510 Py_DECREF(v);
1511 Py_DECREF(w);
1512 SET_TOP(x);
1513 if (x != NULL) DISPATCH();
1514 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001515
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001516 TARGET(BINARY_ADD)
1517 w = POP();
1518 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001519 if (PyUnicode_CheckExact(v) &&
1520 PyUnicode_CheckExact(w)) {
1521 x = unicode_concatenate(v, w, f, next_instr);
1522 /* unicode_concatenate consumed the ref to v */
1523 goto skip_decref_vx;
1524 }
1525 else {
1526 x = PyNumber_Add(v, w);
1527 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001528 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001529 skip_decref_vx:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001530 Py_DECREF(w);
1531 SET_TOP(x);
1532 if (x != NULL) DISPATCH();
1533 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001535 TARGET(BINARY_SUBTRACT)
1536 w = POP();
1537 v = TOP();
1538 x = PyNumber_Subtract(v, w);
1539 Py_DECREF(v);
1540 Py_DECREF(w);
1541 SET_TOP(x);
1542 if (x != NULL) DISPATCH();
1543 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 TARGET(BINARY_SUBSCR)
1546 w = POP();
1547 v = TOP();
1548 x = PyObject_GetItem(v, w);
1549 Py_DECREF(v);
1550 Py_DECREF(w);
1551 SET_TOP(x);
1552 if (x != NULL) DISPATCH();
1553 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001554
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001555 TARGET(BINARY_LSHIFT)
1556 w = POP();
1557 v = TOP();
1558 x = PyNumber_Lshift(v, w);
1559 Py_DECREF(v);
1560 Py_DECREF(w);
1561 SET_TOP(x);
1562 if (x != NULL) DISPATCH();
1563 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001564
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001565 TARGET(BINARY_RSHIFT)
1566 w = POP();
1567 v = TOP();
1568 x = PyNumber_Rshift(v, w);
1569 Py_DECREF(v);
1570 Py_DECREF(w);
1571 SET_TOP(x);
1572 if (x != NULL) DISPATCH();
1573 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001574
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001575 TARGET(BINARY_AND)
1576 w = POP();
1577 v = TOP();
1578 x = PyNumber_And(v, w);
1579 Py_DECREF(v);
1580 Py_DECREF(w);
1581 SET_TOP(x);
1582 if (x != NULL) DISPATCH();
1583 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001585 TARGET(BINARY_XOR)
1586 w = POP();
1587 v = TOP();
1588 x = PyNumber_Xor(v, w);
1589 Py_DECREF(v);
1590 Py_DECREF(w);
1591 SET_TOP(x);
1592 if (x != NULL) DISPATCH();
1593 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001595 TARGET(BINARY_OR)
1596 w = POP();
1597 v = TOP();
1598 x = PyNumber_Or(v, w);
1599 Py_DECREF(v);
1600 Py_DECREF(w);
1601 SET_TOP(x);
1602 if (x != NULL) DISPATCH();
1603 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001604
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001605 TARGET(LIST_APPEND)
1606 w = POP();
1607 v = PEEK(oparg);
1608 err = PyList_Append(v, w);
1609 Py_DECREF(w);
1610 if (err == 0) {
1611 PREDICT(JUMP_ABSOLUTE);
1612 DISPATCH();
1613 }
1614 break;
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001615
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001616 TARGET(SET_ADD)
1617 w = POP();
1618 v = stack_pointer[-oparg];
1619 err = PySet_Add(v, w);
1620 Py_DECREF(w);
1621 if (err == 0) {
1622 PREDICT(JUMP_ABSOLUTE);
1623 DISPATCH();
1624 }
1625 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00001626
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001627 TARGET(INPLACE_POWER)
1628 w = POP();
1629 v = TOP();
1630 x = PyNumber_InPlacePower(v, w, Py_None);
1631 Py_DECREF(v);
1632 Py_DECREF(w);
1633 SET_TOP(x);
1634 if (x != NULL) DISPATCH();
1635 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001636
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001637 TARGET(INPLACE_MULTIPLY)
1638 w = POP();
1639 v = TOP();
1640 x = PyNumber_InPlaceMultiply(v, w);
1641 Py_DECREF(v);
1642 Py_DECREF(w);
1643 SET_TOP(x);
1644 if (x != NULL) DISPATCH();
1645 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001647 TARGET(INPLACE_TRUE_DIVIDE)
1648 w = POP();
1649 v = TOP();
1650 x = PyNumber_InPlaceTrueDivide(v, w);
1651 Py_DECREF(v);
1652 Py_DECREF(w);
1653 SET_TOP(x);
1654 if (x != NULL) DISPATCH();
1655 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001656
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001657 TARGET(INPLACE_FLOOR_DIVIDE)
1658 w = POP();
1659 v = TOP();
1660 x = PyNumber_InPlaceFloorDivide(v, w);
1661 Py_DECREF(v);
1662 Py_DECREF(w);
1663 SET_TOP(x);
1664 if (x != NULL) DISPATCH();
1665 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001666
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001667 TARGET(INPLACE_MODULO)
1668 w = POP();
1669 v = TOP();
1670 x = PyNumber_InPlaceRemainder(v, w);
1671 Py_DECREF(v);
1672 Py_DECREF(w);
1673 SET_TOP(x);
1674 if (x != NULL) DISPATCH();
1675 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001677 TARGET(INPLACE_ADD)
1678 w = POP();
1679 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001680 if (PyUnicode_CheckExact(v) &&
1681 PyUnicode_CheckExact(w)) {
1682 x = unicode_concatenate(v, w, f, next_instr);
1683 /* unicode_concatenate consumed the ref to v */
1684 goto skip_decref_v;
1685 }
1686 else {
1687 x = PyNumber_InPlaceAdd(v, w);
1688 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001689 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001690 skip_decref_v:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001691 Py_DECREF(w);
1692 SET_TOP(x);
1693 if (x != NULL) DISPATCH();
1694 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001695
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001696 TARGET(INPLACE_SUBTRACT)
1697 w = POP();
1698 v = TOP();
1699 x = PyNumber_InPlaceSubtract(v, w);
1700 Py_DECREF(v);
1701 Py_DECREF(w);
1702 SET_TOP(x);
1703 if (x != NULL) DISPATCH();
1704 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001705
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001706 TARGET(INPLACE_LSHIFT)
1707 w = POP();
1708 v = TOP();
1709 x = PyNumber_InPlaceLshift(v, w);
1710 Py_DECREF(v);
1711 Py_DECREF(w);
1712 SET_TOP(x);
1713 if (x != NULL) DISPATCH();
1714 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001715
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001716 TARGET(INPLACE_RSHIFT)
1717 w = POP();
1718 v = TOP();
1719 x = PyNumber_InPlaceRshift(v, w);
1720 Py_DECREF(v);
1721 Py_DECREF(w);
1722 SET_TOP(x);
1723 if (x != NULL) DISPATCH();
1724 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001725
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001726 TARGET(INPLACE_AND)
1727 w = POP();
1728 v = TOP();
1729 x = PyNumber_InPlaceAnd(v, w);
1730 Py_DECREF(v);
1731 Py_DECREF(w);
1732 SET_TOP(x);
1733 if (x != NULL) DISPATCH();
1734 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001735
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001736 TARGET(INPLACE_XOR)
1737 w = POP();
1738 v = TOP();
1739 x = PyNumber_InPlaceXor(v, w);
1740 Py_DECREF(v);
1741 Py_DECREF(w);
1742 SET_TOP(x);
1743 if (x != NULL) DISPATCH();
1744 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001745
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001746 TARGET(INPLACE_OR)
1747 w = POP();
1748 v = TOP();
1749 x = PyNumber_InPlaceOr(v, w);
1750 Py_DECREF(v);
1751 Py_DECREF(w);
1752 SET_TOP(x);
1753 if (x != NULL) DISPATCH();
1754 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001756 TARGET(STORE_SUBSCR)
1757 w = TOP();
1758 v = SECOND();
1759 u = THIRD();
1760 STACKADJ(-3);
1761 /* v[w] = u */
1762 err = PyObject_SetItem(v, w, u);
1763 Py_DECREF(u);
1764 Py_DECREF(v);
1765 Py_DECREF(w);
1766 if (err == 0) DISPATCH();
1767 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001768
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001769 TARGET(DELETE_SUBSCR)
1770 w = TOP();
1771 v = SECOND();
1772 STACKADJ(-2);
1773 /* del v[w] */
1774 err = PyObject_DelItem(v, w);
1775 Py_DECREF(v);
1776 Py_DECREF(w);
1777 if (err == 0) DISPATCH();
1778 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001779
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001780 TARGET(PRINT_EXPR)
1781 v = POP();
1782 w = PySys_GetObject("displayhook");
1783 if (w == NULL) {
1784 PyErr_SetString(PyExc_RuntimeError,
1785 "lost sys.displayhook");
1786 err = -1;
1787 x = NULL;
1788 }
1789 if (err == 0) {
1790 x = PyTuple_Pack(1, v);
1791 if (x == NULL)
1792 err = -1;
1793 }
1794 if (err == 0) {
1795 w = PyEval_CallObject(w, x);
1796 Py_XDECREF(w);
1797 if (w == NULL)
1798 err = -1;
1799 }
1800 Py_DECREF(v);
1801 Py_XDECREF(x);
1802 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001803
Thomas Wouters434d0822000-08-24 20:11:32 +00001804#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001805 default: switch (opcode) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001806#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001807 TARGET(RAISE_VARARGS)
1808 v = w = NULL;
1809 switch (oparg) {
1810 case 2:
1811 v = POP(); /* cause */
1812 case 1:
1813 w = POP(); /* exc */
1814 case 0: /* Fallthrough */
1815 why = do_raise(w, v);
1816 break;
1817 default:
1818 PyErr_SetString(PyExc_SystemError,
1819 "bad RAISE_VARARGS oparg");
1820 why = WHY_EXCEPTION;
1821 break;
1822 }
1823 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001824
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001825 TARGET(STORE_LOCALS)
1826 x = POP();
1827 v = f->f_locals;
1828 Py_XDECREF(v);
1829 f->f_locals = x;
1830 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001831
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001832 TARGET(RETURN_VALUE)
1833 retval = POP();
1834 why = WHY_RETURN;
1835 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001836
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001837 TARGET(YIELD_FROM)
1838 u = POP();
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001839 x = TOP();
1840 /* send u to x */
1841 if (PyGen_CheckExact(x)) {
1842 retval = _PyGen_Send((PyGenObject *)x, u);
1843 } else {
Benjamin Peterson302e7902012-03-20 23:17:04 -04001844 _Py_IDENTIFIER(send);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001845 if (u == Py_None)
1846 retval = PyIter_Next(x);
1847 else
Benjamin Peterson302e7902012-03-20 23:17:04 -04001848 retval = _PyObject_CallMethodId(x, &PyId_send, "O", u);
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001849 }
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001850 Py_DECREF(u);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001851 if (!retval) {
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001852 PyObject *val;
1853 x = POP(); /* Remove iter from stack */
1854 Py_DECREF(x);
1855 err = PyGen_FetchStopIterationValue(&val);
1856 if (err < 0) {
1857 x = NULL;
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001858 break;
1859 }
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001860 x = val;
1861 PUSH(x);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001862 continue;
1863 }
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001864 /* x remains on stack, retval is value to be yielded */
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001865 f->f_stacktop = stack_pointer;
1866 why = WHY_YIELD;
Benjamin Peterson2afe6ae2012-03-15 15:37:39 -05001867 /* and repeat... */
1868 f->f_lasti--;
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001869 goto fast_yield;
1870
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001871 TARGET(YIELD_VALUE)
1872 retval = POP();
1873 f->f_stacktop = stack_pointer;
1874 why = WHY_YIELD;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001875 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001876
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 TARGET(POP_EXCEPT)
1878 {
1879 PyTryBlock *b = PyFrame_BlockPop(f);
1880 if (b->b_type != EXCEPT_HANDLER) {
1881 PyErr_SetString(PyExc_SystemError,
1882 "popped block is not an except handler");
1883 why = WHY_EXCEPTION;
1884 break;
1885 }
1886 UNWIND_EXCEPT_HANDLER(b);
1887 }
1888 DISPATCH();
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001889
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 TARGET(POP_BLOCK)
1891 {
1892 PyTryBlock *b = PyFrame_BlockPop(f);
1893 UNWIND_BLOCK(b);
1894 }
1895 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001897 PREDICTED(END_FINALLY);
1898 TARGET(END_FINALLY)
1899 v = POP();
1900 if (PyLong_Check(v)) {
1901 why = (enum why_code) PyLong_AS_LONG(v);
1902 assert(why != WHY_YIELD);
1903 if (why == WHY_RETURN ||
1904 why == WHY_CONTINUE)
1905 retval = POP();
1906 if (why == WHY_SILENCED) {
1907 /* An exception was silenced by 'with', we must
1908 manually unwind the EXCEPT_HANDLER block which was
1909 created when the exception was caught, otherwise
1910 the stack will be in an inconsistent state. */
1911 PyTryBlock *b = PyFrame_BlockPop(f);
1912 assert(b->b_type == EXCEPT_HANDLER);
1913 UNWIND_EXCEPT_HANDLER(b);
1914 why = WHY_NOT;
1915 }
1916 }
1917 else if (PyExceptionClass_Check(v)) {
1918 w = POP();
1919 u = POP();
1920 PyErr_Restore(v, w, u);
1921 why = WHY_RERAISE;
1922 break;
1923 }
1924 else if (v != Py_None) {
1925 PyErr_SetString(PyExc_SystemError,
1926 "'finally' pops bad exception");
1927 why = WHY_EXCEPTION;
1928 }
1929 Py_DECREF(v);
1930 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001931
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001932 TARGET(LOAD_BUILD_CLASS)
Victor Stinner3c1e4812012-03-26 22:10:51 +02001933 {
1934 _Py_IDENTIFIER(__build_class__);
Victor Stinnerb0b22422012-04-19 00:57:45 +02001935
1936 if (PyDict_CheckExact(f->f_builtins)) {
1937 x = _PyDict_GetItemId(f->f_builtins, &PyId___build_class__);
1938 if (x == NULL) {
1939 PyErr_SetString(PyExc_NameError,
1940 "__build_class__ not found");
1941 break;
1942 }
1943 }
1944 else {
1945 PyObject *build_class_str = _PyUnicode_FromId(&PyId___build_class__);
1946 if (build_class_str == NULL)
1947 break;
1948 x = PyObject_GetItem(f->f_builtins, build_class_str);
1949 if (x == NULL) {
1950 if (PyErr_ExceptionMatches(PyExc_KeyError))
1951 PyErr_SetString(PyExc_NameError,
1952 "__build_class__ not found");
1953 break;
1954 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001955 }
1956 Py_INCREF(x);
1957 PUSH(x);
1958 break;
Victor Stinner3c1e4812012-03-26 22:10:51 +02001959 }
Guido van Rossumac7be682001-01-17 15:42:30 +00001960
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001961 TARGET(STORE_NAME)
1962 w = GETITEM(names, oparg);
1963 v = POP();
1964 if ((x = f->f_locals) != NULL) {
1965 if (PyDict_CheckExact(x))
1966 err = PyDict_SetItem(x, w, v);
1967 else
1968 err = PyObject_SetItem(x, w, v);
1969 Py_DECREF(v);
1970 if (err == 0) DISPATCH();
1971 break;
1972 }
1973 PyErr_Format(PyExc_SystemError,
1974 "no locals found when storing %R", w);
1975 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001976
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001977 TARGET(DELETE_NAME)
1978 w = GETITEM(names, oparg);
1979 if ((x = f->f_locals) != NULL) {
1980 if ((err = PyObject_DelItem(x, w)) != 0)
1981 format_exc_check_arg(PyExc_NameError,
1982 NAME_ERROR_MSG,
1983 w);
1984 break;
1985 }
1986 PyErr_Format(PyExc_SystemError,
1987 "no locals when deleting %R", w);
1988 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001989
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001990 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
1991 TARGET(UNPACK_SEQUENCE)
1992 v = POP();
1993 if (PyTuple_CheckExact(v) &&
1994 PyTuple_GET_SIZE(v) == oparg) {
1995 PyObject **items = \
1996 ((PyTupleObject *)v)->ob_item;
1997 while (oparg--) {
1998 w = items[oparg];
1999 Py_INCREF(w);
2000 PUSH(w);
2001 }
2002 Py_DECREF(v);
2003 DISPATCH();
2004 } else if (PyList_CheckExact(v) &&
2005 PyList_GET_SIZE(v) == oparg) {
2006 PyObject **items = \
2007 ((PyListObject *)v)->ob_item;
2008 while (oparg--) {
2009 w = items[oparg];
2010 Py_INCREF(w);
2011 PUSH(w);
2012 }
2013 } else if (unpack_iterable(v, oparg, -1,
2014 stack_pointer + oparg)) {
2015 STACKADJ(oparg);
2016 } else {
2017 /* unpack_iterable() raised an exception */
2018 why = WHY_EXCEPTION;
2019 }
2020 Py_DECREF(v);
2021 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002022
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002023 TARGET(UNPACK_EX)
2024 {
2025 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
2026 v = POP();
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002028 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
2029 stack_pointer + totalargs)) {
2030 stack_pointer += totalargs;
2031 } else {
2032 why = WHY_EXCEPTION;
2033 }
2034 Py_DECREF(v);
2035 break;
2036 }
Guido van Rossum0368b722007-05-11 16:50:42 +00002037
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002038 TARGET(STORE_ATTR)
2039 w = GETITEM(names, oparg);
2040 v = TOP();
2041 u = SECOND();
2042 STACKADJ(-2);
2043 err = PyObject_SetAttr(v, w, u); /* v.w = u */
2044 Py_DECREF(v);
2045 Py_DECREF(u);
2046 if (err == 0) DISPATCH();
2047 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002048
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002049 TARGET(DELETE_ATTR)
2050 w = GETITEM(names, oparg);
2051 v = POP();
2052 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
2053 /* del v.w */
2054 Py_DECREF(v);
2055 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002056
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 TARGET(STORE_GLOBAL)
2058 w = GETITEM(names, oparg);
2059 v = POP();
2060 err = PyDict_SetItem(f->f_globals, w, v);
2061 Py_DECREF(v);
2062 if (err == 0) DISPATCH();
2063 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002064
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002065 TARGET(DELETE_GLOBAL)
2066 w = GETITEM(names, oparg);
2067 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
2068 format_exc_check_arg(
2069 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
2070 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002072 TARGET(LOAD_NAME)
2073 w = GETITEM(names, oparg);
2074 if ((v = f->f_locals) == NULL) {
2075 PyErr_Format(PyExc_SystemError,
2076 "no locals when loading %R", w);
2077 why = WHY_EXCEPTION;
2078 break;
2079 }
2080 if (PyDict_CheckExact(v)) {
2081 x = PyDict_GetItem(v, w);
2082 Py_XINCREF(x);
2083 }
2084 else {
2085 x = PyObject_GetItem(v, w);
2086 if (x == NULL && PyErr_Occurred()) {
2087 if (!PyErr_ExceptionMatches(
2088 PyExc_KeyError))
2089 break;
2090 PyErr_Clear();
2091 }
2092 }
2093 if (x == NULL) {
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002094 x = PyDict_GetItem(f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002095 if (x == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002096 if (PyDict_CheckExact(f->f_builtins)) {
2097 x = PyDict_GetItem(f->f_builtins, w);
2098 if (x == NULL) {
2099 format_exc_check_arg(
2100 PyExc_NameError,
2101 NAME_ERROR_MSG, w);
2102 break;
2103 }
2104 }
2105 else {
2106 x = PyObject_GetItem(f->f_builtins, w);
2107 if (x == NULL) {
2108 if (PyErr_ExceptionMatches(PyExc_KeyError))
2109 format_exc_check_arg(
2110 PyExc_NameError,
2111 NAME_ERROR_MSG, w);
2112 break;
2113 }
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002114 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002115 }
2116 Py_INCREF(x);
2117 }
2118 PUSH(x);
2119 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002120
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002121 TARGET(LOAD_GLOBAL)
2122 w = GETITEM(names, oparg);
Victor Stinnerb0b22422012-04-19 00:57:45 +02002123 if (PyDict_CheckExact(f->f_globals)
2124 && PyDict_CheckExact(f->f_builtins)) {
2125 if (PyUnicode_CheckExact(w)) {
2126 /* Inline the PyDict_GetItem() calls.
2127 WARNING: this is an extreme speed hack.
2128 Do not try this at home. */
2129 Py_hash_t hash = ((PyASCIIObject *)w)->hash;
2130 if (hash != -1) {
2131 PyDictObject *d;
2132 PyDictEntry *e;
2133 d = (PyDictObject *)(f->f_globals);
2134 e = d->ma_lookup(d, w, hash);
2135 if (e == NULL) {
2136 x = NULL;
2137 break;
2138 }
2139 x = e->me_value;
2140 if (x != NULL) {
2141 Py_INCREF(x);
2142 PUSH(x);
2143 DISPATCH();
2144 }
2145 d = (PyDictObject *)(f->f_builtins);
2146 e = d->ma_lookup(d, w, hash);
2147 if (e == NULL) {
2148 x = NULL;
2149 break;
2150 }
2151 x = e->me_value;
2152 if (x != NULL) {
2153 Py_INCREF(x);
2154 PUSH(x);
2155 DISPATCH();
2156 }
2157 goto load_global_error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002158 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002159 }
Victor Stinnerb0b22422012-04-19 00:57:45 +02002160 /* This is the un-inlined version of the code above */
2161 x = PyDict_GetItem(f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002162 if (x == NULL) {
Victor Stinnerb0b22422012-04-19 00:57:45 +02002163 x = PyDict_GetItem(f->f_builtins, w);
2164 if (x == NULL) {
2165 load_global_error:
2166 format_exc_check_arg(
2167 PyExc_NameError,
2168 GLOBAL_NAME_ERROR_MSG, w);
2169 break;
2170 }
2171 }
2172 Py_INCREF(x);
2173 PUSH(x);
2174 DISPATCH();
2175 }
2176
2177 /* Slow-path if globals or builtins is not a dict */
2178 x = PyObject_GetItem(f->f_globals, w);
2179 if (x == NULL) {
2180 x = PyObject_GetItem(f->f_builtins, w);
2181 if (x == NULL) {
2182 if (PyErr_ExceptionMatches(PyExc_KeyError))
2183 format_exc_check_arg(
2184 PyExc_NameError,
2185 GLOBAL_NAME_ERROR_MSG, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002186 break;
2187 }
2188 }
2189 Py_INCREF(x);
2190 PUSH(x);
2191 DISPATCH();
Guido van Rossum681d79a1995-07-18 14:51:37 +00002192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002193 TARGET(DELETE_FAST)
2194 x = GETLOCAL(oparg);
2195 if (x != NULL) {
2196 SETLOCAL(oparg, NULL);
2197 DISPATCH();
2198 }
2199 format_exc_check_arg(
2200 PyExc_UnboundLocalError,
2201 UNBOUNDLOCAL_ERROR_MSG,
2202 PyTuple_GetItem(co->co_varnames, oparg)
2203 );
2204 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002205
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002206 TARGET(DELETE_DEREF)
2207 x = freevars[oparg];
2208 if (PyCell_GET(x) != NULL) {
2209 PyCell_Set(x, NULL);
Benjamin Peterson00ebe2c2010-09-10 22:02:31 +00002210 DISPATCH();
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002211 }
2212 err = -1;
2213 format_exc_unbound(co, oparg);
2214 break;
2215
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002216 TARGET(LOAD_CLOSURE)
2217 x = freevars[oparg];
2218 Py_INCREF(x);
2219 PUSH(x);
2220 if (x != NULL) DISPATCH();
2221 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002222
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002223 TARGET(LOAD_DEREF)
2224 x = freevars[oparg];
2225 w = PyCell_Get(x);
2226 if (w != NULL) {
2227 PUSH(w);
2228 DISPATCH();
2229 }
2230 err = -1;
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002231 format_exc_unbound(co, oparg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002232 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002233
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002234 TARGET(STORE_DEREF)
2235 w = POP();
2236 x = freevars[oparg];
2237 PyCell_Set(x, w);
2238 Py_DECREF(w);
2239 DISPATCH();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002241 TARGET(BUILD_TUPLE)
2242 x = PyTuple_New(oparg);
2243 if (x != NULL) {
2244 for (; --oparg >= 0;) {
2245 w = POP();
2246 PyTuple_SET_ITEM(x, oparg, w);
2247 }
2248 PUSH(x);
2249 DISPATCH();
2250 }
2251 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002252
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002253 TARGET(BUILD_LIST)
2254 x = PyList_New(oparg);
2255 if (x != NULL) {
2256 for (; --oparg >= 0;) {
2257 w = POP();
2258 PyList_SET_ITEM(x, oparg, w);
2259 }
2260 PUSH(x);
2261 DISPATCH();
2262 }
2263 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002264
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002265 TARGET(BUILD_SET)
2266 x = PySet_New(NULL);
2267 if (x != NULL) {
2268 for (; --oparg >= 0;) {
2269 w = POP();
2270 if (err == 0)
2271 err = PySet_Add(x, w);
2272 Py_DECREF(w);
2273 }
2274 if (err != 0) {
2275 Py_DECREF(x);
2276 break;
2277 }
2278 PUSH(x);
2279 DISPATCH();
2280 }
2281 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002283 TARGET(BUILD_MAP)
2284 x = _PyDict_NewPresized((Py_ssize_t)oparg);
2285 PUSH(x);
2286 if (x != NULL) DISPATCH();
2287 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002288
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002289 TARGET(STORE_MAP)
2290 w = TOP(); /* key */
2291 u = SECOND(); /* value */
2292 v = THIRD(); /* dict */
2293 STACKADJ(-2);
2294 assert (PyDict_CheckExact(v));
2295 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2296 Py_DECREF(u);
2297 Py_DECREF(w);
2298 if (err == 0) DISPATCH();
2299 break;
Christian Heimes99170a52007-12-19 02:07:34 +00002300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002301 TARGET(MAP_ADD)
2302 w = TOP(); /* key */
2303 u = SECOND(); /* value */
2304 STACKADJ(-2);
2305 v = stack_pointer[-oparg]; /* dict */
2306 assert (PyDict_CheckExact(v));
2307 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2308 Py_DECREF(u);
2309 Py_DECREF(w);
2310 if (err == 0) {
2311 PREDICT(JUMP_ABSOLUTE);
2312 DISPATCH();
2313 }
2314 break;
Antoine Pitrouf289ae62008-12-18 11:06:25 +00002315
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002316 TARGET(LOAD_ATTR)
2317 w = GETITEM(names, oparg);
2318 v = TOP();
2319 x = PyObject_GetAttr(v, w);
2320 Py_DECREF(v);
2321 SET_TOP(x);
2322 if (x != NULL) DISPATCH();
2323 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002325 TARGET(COMPARE_OP)
2326 w = POP();
2327 v = TOP();
2328 x = cmp_outcome(oparg, v, w);
2329 Py_DECREF(v);
2330 Py_DECREF(w);
2331 SET_TOP(x);
2332 if (x == NULL) break;
2333 PREDICT(POP_JUMP_IF_FALSE);
2334 PREDICT(POP_JUMP_IF_TRUE);
2335 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002336
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002337 TARGET(IMPORT_NAME)
Victor Stinner3c1e4812012-03-26 22:10:51 +02002338 {
2339 _Py_IDENTIFIER(__import__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002340 w = GETITEM(names, oparg);
Victor Stinner3c1e4812012-03-26 22:10:51 +02002341 x = _PyDict_GetItemId(f->f_builtins, &PyId___import__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002342 if (x == NULL) {
2343 PyErr_SetString(PyExc_ImportError,
2344 "__import__ not found");
2345 break;
2346 }
2347 Py_INCREF(x);
2348 v = POP();
2349 u = TOP();
2350 if (PyLong_AsLong(u) != -1 || PyErr_Occurred())
2351 w = PyTuple_Pack(5,
2352 w,
2353 f->f_globals,
2354 f->f_locals == NULL ?
2355 Py_None : f->f_locals,
2356 v,
2357 u);
2358 else
2359 w = PyTuple_Pack(4,
2360 w,
2361 f->f_globals,
2362 f->f_locals == NULL ?
2363 Py_None : f->f_locals,
2364 v);
2365 Py_DECREF(v);
2366 Py_DECREF(u);
2367 if (w == NULL) {
2368 u = POP();
2369 Py_DECREF(x);
2370 x = NULL;
2371 break;
2372 }
2373 READ_TIMESTAMP(intr0);
2374 v = x;
2375 x = PyEval_CallObject(v, w);
2376 Py_DECREF(v);
2377 READ_TIMESTAMP(intr1);
2378 Py_DECREF(w);
2379 SET_TOP(x);
2380 if (x != NULL) DISPATCH();
2381 break;
Victor Stinner3c1e4812012-03-26 22:10:51 +02002382 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002383
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002384 TARGET(IMPORT_STAR)
2385 v = POP();
2386 PyFrame_FastToLocals(f);
2387 if ((x = f->f_locals) == NULL) {
2388 PyErr_SetString(PyExc_SystemError,
2389 "no locals found during 'import *'");
2390 break;
2391 }
2392 READ_TIMESTAMP(intr0);
2393 err = import_all_from(x, v);
2394 READ_TIMESTAMP(intr1);
2395 PyFrame_LocalsToFast(f, 0);
2396 Py_DECREF(v);
2397 if (err == 0) DISPATCH();
2398 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002399
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002400 TARGET(IMPORT_FROM)
2401 w = GETITEM(names, oparg);
2402 v = TOP();
2403 READ_TIMESTAMP(intr0);
2404 x = import_from(v, w);
2405 READ_TIMESTAMP(intr1);
2406 PUSH(x);
2407 if (x != NULL) DISPATCH();
2408 break;
Thomas Wouters52152252000-08-17 22:55:00 +00002409
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002410 TARGET(JUMP_FORWARD)
2411 JUMPBY(oparg);
2412 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002413
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002414 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
2415 TARGET(POP_JUMP_IF_FALSE)
2416 w = POP();
2417 if (w == Py_True) {
2418 Py_DECREF(w);
2419 FAST_DISPATCH();
2420 }
2421 if (w == Py_False) {
2422 Py_DECREF(w);
2423 JUMPTO(oparg);
2424 FAST_DISPATCH();
2425 }
2426 err = PyObject_IsTrue(w);
2427 Py_DECREF(w);
2428 if (err > 0)
2429 err = 0;
2430 else if (err == 0)
2431 JUMPTO(oparg);
2432 else
2433 break;
2434 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002436 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
2437 TARGET(POP_JUMP_IF_TRUE)
2438 w = POP();
2439 if (w == Py_False) {
2440 Py_DECREF(w);
2441 FAST_DISPATCH();
2442 }
2443 if (w == Py_True) {
2444 Py_DECREF(w);
2445 JUMPTO(oparg);
2446 FAST_DISPATCH();
2447 }
2448 err = PyObject_IsTrue(w);
2449 Py_DECREF(w);
2450 if (err > 0) {
2451 err = 0;
2452 JUMPTO(oparg);
2453 }
2454 else if (err == 0)
2455 ;
2456 else
2457 break;
2458 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002459
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002460 TARGET(JUMP_IF_FALSE_OR_POP)
2461 w = TOP();
2462 if (w == Py_True) {
2463 STACKADJ(-1);
2464 Py_DECREF(w);
2465 FAST_DISPATCH();
2466 }
2467 if (w == Py_False) {
2468 JUMPTO(oparg);
2469 FAST_DISPATCH();
2470 }
2471 err = PyObject_IsTrue(w);
2472 if (err > 0) {
2473 STACKADJ(-1);
2474 Py_DECREF(w);
2475 err = 0;
2476 }
2477 else if (err == 0)
2478 JUMPTO(oparg);
2479 else
2480 break;
2481 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002482
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002483 TARGET(JUMP_IF_TRUE_OR_POP)
2484 w = TOP();
2485 if (w == Py_False) {
2486 STACKADJ(-1);
2487 Py_DECREF(w);
2488 FAST_DISPATCH();
2489 }
2490 if (w == Py_True) {
2491 JUMPTO(oparg);
2492 FAST_DISPATCH();
2493 }
2494 err = PyObject_IsTrue(w);
2495 if (err > 0) {
2496 err = 0;
2497 JUMPTO(oparg);
2498 }
2499 else if (err == 0) {
2500 STACKADJ(-1);
2501 Py_DECREF(w);
2502 }
2503 else
2504 break;
2505 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002506
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002507 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
2508 TARGET(JUMP_ABSOLUTE)
2509 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002510#if FAST_LOOPS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002511 /* Enabling this path speeds-up all while and for-loops by bypassing
2512 the per-loop checks for signals. By default, this should be turned-off
2513 because it prevents detection of a control-break in tight loops like
2514 "while 1: pass". Compile with this option turned-on when you need
2515 the speed-up and do not need break checking inside tight loops (ones
2516 that contain only instructions ending with FAST_DISPATCH).
2517 */
2518 FAST_DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002519#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002520 DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002521#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002522
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 TARGET(GET_ITER)
2524 /* before: [obj]; after [getiter(obj)] */
2525 v = TOP();
2526 x = PyObject_GetIter(v);
2527 Py_DECREF(v);
2528 if (x != NULL) {
2529 SET_TOP(x);
2530 PREDICT(FOR_ITER);
2531 DISPATCH();
2532 }
2533 STACKADJ(-1);
2534 break;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002536 PREDICTED_WITH_ARG(FOR_ITER);
2537 TARGET(FOR_ITER)
2538 /* before: [iter]; after: [iter, iter()] *or* [] */
2539 v = TOP();
2540 x = (*v->ob_type->tp_iternext)(v);
2541 if (x != NULL) {
2542 PUSH(x);
2543 PREDICT(STORE_FAST);
2544 PREDICT(UNPACK_SEQUENCE);
2545 DISPATCH();
2546 }
2547 if (PyErr_Occurred()) {
2548 if (!PyErr_ExceptionMatches(
2549 PyExc_StopIteration))
2550 break;
2551 PyErr_Clear();
2552 }
2553 /* iterator ended normally */
2554 x = v = POP();
2555 Py_DECREF(v);
2556 JUMPBY(oparg);
2557 DISPATCH();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002559 TARGET(BREAK_LOOP)
2560 why = WHY_BREAK;
2561 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002562
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002563 TARGET(CONTINUE_LOOP)
2564 retval = PyLong_FromLong(oparg);
2565 if (!retval) {
2566 x = NULL;
2567 break;
2568 }
2569 why = WHY_CONTINUE;
2570 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002571
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002572 TARGET_WITH_IMPL(SETUP_LOOP, _setup_finally)
2573 TARGET_WITH_IMPL(SETUP_EXCEPT, _setup_finally)
2574 TARGET(SETUP_FINALLY)
2575 _setup_finally:
2576 /* NOTE: If you add any new block-setup opcodes that
2577 are not try/except/finally handlers, you may need
2578 to update the PyGen_NeedsFinalizing() function.
2579 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002581 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
2582 STACK_LEVEL());
2583 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002585 TARGET(SETUP_WITH)
2586 {
Benjamin Petersonce798522012-01-22 11:24:29 -05002587 _Py_IDENTIFIER(__exit__);
2588 _Py_IDENTIFIER(__enter__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002589 w = TOP();
Benjamin Petersonce798522012-01-22 11:24:29 -05002590 x = special_lookup(w, &PyId___exit__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002591 if (!x)
2592 break;
2593 SET_TOP(x);
Benjamin Petersonce798522012-01-22 11:24:29 -05002594 u = special_lookup(w, &PyId___enter__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002595 Py_DECREF(w);
2596 if (!u) {
2597 x = NULL;
2598 break;
2599 }
2600 x = PyObject_CallFunctionObjArgs(u, NULL);
2601 Py_DECREF(u);
2602 if (!x)
2603 break;
2604 /* Setup the finally block before pushing the result
2605 of __enter__ on the stack. */
2606 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2607 STACK_LEVEL());
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002608
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002609 PUSH(x);
2610 DISPATCH();
2611 }
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002612
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002613 TARGET(WITH_CLEANUP)
2614 {
2615 /* At the top of the stack are 1-3 values indicating
2616 how/why we entered the finally clause:
2617 - TOP = None
2618 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2619 - TOP = WHY_*; no retval below it
2620 - (TOP, SECOND, THIRD) = exc_info()
2621 (FOURTH, FITH, SIXTH) = previous exception for EXCEPT_HANDLER
2622 Below them is EXIT, the context.__exit__ bound method.
2623 In the last case, we must call
2624 EXIT(TOP, SECOND, THIRD)
2625 otherwise we must call
2626 EXIT(None, None, None)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002627
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002628 In the first two cases, we remove EXIT from the
2629 stack, leaving the rest in the same order. In the
2630 third case, we shift the bottom 3 values of the
2631 stack down, and replace the empty spot with NULL.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002633 In addition, if the stack represents an exception,
2634 *and* the function call returns a 'true' value, we
2635 push WHY_SILENCED onto the stack. END_FINALLY will
2636 then not re-raise the exception. (But non-local
2637 gotos should still be resumed.)
2638 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002639
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002640 PyObject *exit_func;
2641 u = TOP();
2642 if (u == Py_None) {
2643 (void)POP();
2644 exit_func = TOP();
2645 SET_TOP(u);
2646 v = w = Py_None;
2647 }
2648 else if (PyLong_Check(u)) {
2649 (void)POP();
2650 switch(PyLong_AsLong(u)) {
2651 case WHY_RETURN:
2652 case WHY_CONTINUE:
2653 /* Retval in TOP. */
2654 exit_func = SECOND();
2655 SET_SECOND(TOP());
2656 SET_TOP(u);
2657 break;
2658 default:
2659 exit_func = TOP();
2660 SET_TOP(u);
2661 break;
2662 }
2663 u = v = w = Py_None;
2664 }
2665 else {
2666 PyObject *tp, *exc, *tb;
2667 PyTryBlock *block;
2668 v = SECOND();
2669 w = THIRD();
2670 tp = FOURTH();
2671 exc = PEEK(5);
2672 tb = PEEK(6);
2673 exit_func = PEEK(7);
2674 SET_VALUE(7, tb);
2675 SET_VALUE(6, exc);
2676 SET_VALUE(5, tp);
2677 /* UNWIND_EXCEPT_HANDLER will pop this off. */
2678 SET_FOURTH(NULL);
2679 /* We just shifted the stack down, so we have
2680 to tell the except handler block that the
2681 values are lower than it expects. */
2682 block = &f->f_blockstack[f->f_iblock - 1];
2683 assert(block->b_type == EXCEPT_HANDLER);
2684 block->b_level--;
2685 }
2686 /* XXX Not the fastest way to call it... */
2687 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2688 NULL);
2689 Py_DECREF(exit_func);
2690 if (x == NULL)
2691 break; /* Go to error exit */
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002692
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002693 if (u != Py_None)
2694 err = PyObject_IsTrue(x);
2695 else
2696 err = 0;
2697 Py_DECREF(x);
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 if (err < 0)
2700 break; /* Go to error exit */
2701 else if (err > 0) {
2702 err = 0;
2703 /* There was an exception and a True return */
2704 PUSH(PyLong_FromLong((long) WHY_SILENCED));
2705 }
2706 PREDICT(END_FINALLY);
2707 break;
2708 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002710 TARGET(CALL_FUNCTION)
2711 {
2712 PyObject **sp;
2713 PCALL(PCALL_ALL);
2714 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002715#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002716 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002717#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002718 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002719#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002720 stack_pointer = sp;
2721 PUSH(x);
2722 if (x != NULL)
2723 DISPATCH();
2724 break;
2725 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002726
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002727 TARGET_WITH_IMPL(CALL_FUNCTION_VAR, _call_function_var_kw)
2728 TARGET_WITH_IMPL(CALL_FUNCTION_KW, _call_function_var_kw)
2729 TARGET(CALL_FUNCTION_VAR_KW)
2730 _call_function_var_kw:
2731 {
2732 int na = oparg & 0xff;
2733 int nk = (oparg>>8) & 0xff;
2734 int flags = (opcode - CALL_FUNCTION) & 3;
2735 int n = na + 2 * nk;
2736 PyObject **pfunc, *func, **sp;
2737 PCALL(PCALL_ALL);
2738 if (flags & CALL_FLAG_VAR)
2739 n++;
2740 if (flags & CALL_FLAG_KW)
2741 n++;
2742 pfunc = stack_pointer - n - 1;
2743 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002744
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002745 if (PyMethod_Check(func)
Stefan Krahb7e10102010-06-23 18:42:39 +00002746 && PyMethod_GET_SELF(func) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002747 PyObject *self = PyMethod_GET_SELF(func);
2748 Py_INCREF(self);
2749 func = PyMethod_GET_FUNCTION(func);
2750 Py_INCREF(func);
2751 Py_DECREF(*pfunc);
2752 *pfunc = self;
2753 na++;
Brett Cannonb94767f2011-02-22 20:15:44 +00002754 /* n++; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002755 } else
2756 Py_INCREF(func);
2757 sp = stack_pointer;
2758 READ_TIMESTAMP(intr0);
2759 x = ext_do_call(func, &sp, flags, na, nk);
2760 READ_TIMESTAMP(intr1);
2761 stack_pointer = sp;
2762 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002763
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002764 while (stack_pointer > pfunc) {
2765 w = POP();
2766 Py_DECREF(w);
2767 }
2768 PUSH(x);
2769 if (x != NULL)
2770 DISPATCH();
2771 break;
2772 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002774 TARGET_WITH_IMPL(MAKE_CLOSURE, _make_function)
2775 TARGET(MAKE_FUNCTION)
2776 _make_function:
2777 {
2778 int posdefaults = oparg & 0xff;
2779 int kwdefaults = (oparg>>8) & 0xff;
2780 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002781
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002782 w = POP(); /* qualname */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002783 v = POP(); /* code object */
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002784 x = PyFunction_NewWithQualName(v, f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002785 Py_DECREF(v);
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002786 Py_DECREF(w);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002787
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002788 if (x != NULL && opcode == MAKE_CLOSURE) {
2789 v = POP();
2790 if (PyFunction_SetClosure(x, v) != 0) {
2791 /* Can't happen unless bytecode is corrupt. */
2792 why = WHY_EXCEPTION;
2793 }
2794 Py_DECREF(v);
2795 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002796
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002797 if (x != NULL && num_annotations > 0) {
2798 Py_ssize_t name_ix;
2799 u = POP(); /* names of args with annotations */
2800 v = PyDict_New();
2801 if (v == NULL) {
2802 Py_DECREF(x);
2803 x = NULL;
2804 break;
2805 }
2806 name_ix = PyTuple_Size(u);
2807 assert(num_annotations == name_ix+1);
2808 while (name_ix > 0) {
2809 --name_ix;
2810 t = PyTuple_GET_ITEM(u, name_ix);
2811 w = POP();
2812 /* XXX(nnorwitz): check for errors */
2813 PyDict_SetItem(v, t, w);
2814 Py_DECREF(w);
2815 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002817 if (PyFunction_SetAnnotations(x, v) != 0) {
2818 /* Can't happen unless
2819 PyFunction_SetAnnotations changes. */
2820 why = WHY_EXCEPTION;
2821 }
2822 Py_DECREF(v);
2823 Py_DECREF(u);
2824 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002825
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002826 /* XXX Maybe this should be a separate opcode? */
2827 if (x != NULL && posdefaults > 0) {
2828 v = PyTuple_New(posdefaults);
2829 if (v == NULL) {
2830 Py_DECREF(x);
2831 x = NULL;
2832 break;
2833 }
2834 while (--posdefaults >= 0) {
2835 w = POP();
2836 PyTuple_SET_ITEM(v, posdefaults, w);
2837 }
2838 if (PyFunction_SetDefaults(x, v) != 0) {
2839 /* Can't happen unless
2840 PyFunction_SetDefaults changes. */
2841 why = WHY_EXCEPTION;
2842 }
2843 Py_DECREF(v);
2844 }
2845 if (x != NULL && kwdefaults > 0) {
2846 v = PyDict_New();
2847 if (v == NULL) {
2848 Py_DECREF(x);
2849 x = NULL;
2850 break;
2851 }
2852 while (--kwdefaults >= 0) {
2853 w = POP(); /* default value */
2854 u = POP(); /* kw only arg name */
2855 /* XXX(nnorwitz): check for errors */
2856 PyDict_SetItem(v, u, w);
2857 Py_DECREF(w);
2858 Py_DECREF(u);
2859 }
2860 if (PyFunction_SetKwDefaults(x, v) != 0) {
2861 /* Can't happen unless
2862 PyFunction_SetKwDefaults changes. */
2863 why = WHY_EXCEPTION;
2864 }
2865 Py_DECREF(v);
2866 }
2867 PUSH(x);
2868 break;
2869 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002870
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002871 TARGET(BUILD_SLICE)
2872 if (oparg == 3)
2873 w = POP();
2874 else
2875 w = NULL;
2876 v = POP();
2877 u = TOP();
2878 x = PySlice_New(u, v, w);
2879 Py_DECREF(u);
2880 Py_DECREF(v);
2881 Py_XDECREF(w);
2882 SET_TOP(x);
2883 if (x != NULL) DISPATCH();
2884 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002885
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002886 TARGET(EXTENDED_ARG)
2887 opcode = NEXTOP();
2888 oparg = oparg<<16 | NEXTARG();
2889 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002890
Antoine Pitrou042b1282010-08-13 21:15:58 +00002891#if USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002892 _unknown_opcode:
Antoine Pitroub52ec782009-01-25 16:34:23 +00002893#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002894 default:
2895 fprintf(stderr,
2896 "XXX lineno: %d, opcode: %d\n",
2897 PyFrame_GetLineNumber(f),
2898 opcode);
2899 PyErr_SetString(PyExc_SystemError, "unknown opcode");
2900 why = WHY_EXCEPTION;
2901 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002902
2903#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002904 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00002905#endif
2906
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002907 } /* switch */
Guido van Rossum374a9221991-04-04 10:40:29 +00002908
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002909 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002910
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002911 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002912
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002913 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002914
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002915 if (why == WHY_NOT) {
2916 if (err == 0 && x != NULL) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002917#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002918 /* This check is expensive! */
2919 if (PyErr_Occurred())
2920 fprintf(stderr,
2921 "XXX undetected error\n");
2922 else {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002923#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002924 READ_TIMESTAMP(loop1);
2925 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002926#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002927 }
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002928#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002929 }
2930 why = WHY_EXCEPTION;
2931 x = Py_None;
2932 err = 0;
2933 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002934
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002935 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002936
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002937 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
2938 if (!PyErr_Occurred()) {
2939 PyErr_SetString(PyExc_SystemError,
2940 "error return without exception set");
2941 why = WHY_EXCEPTION;
2942 }
2943 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002944#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002945 else {
2946 /* This check is expensive! */
2947 if (PyErr_Occurred()) {
2948 char buf[128];
2949 sprintf(buf, "Stack unwind with exception "
2950 "set and why=%d", why);
2951 Py_FatalError(buf);
2952 }
2953 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002954#endif
2955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002956 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002957
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002958 if (why == WHY_EXCEPTION) {
2959 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002960
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002961 if (tstate->c_tracefunc != NULL)
2962 call_exc_trace(tstate->c_tracefunc,
2963 tstate->c_traceobj, f);
2964 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002966 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002967
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002968 if (why == WHY_RERAISE)
2969 why = WHY_EXCEPTION;
Guido van Rossum374a9221991-04-04 10:40:29 +00002970
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002971 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002972
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002973fast_block_end:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002974 while (why != WHY_NOT && f->f_iblock > 0) {
2975 /* Peek at the current block. */
2976 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002977
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002978 assert(why != WHY_YIELD);
2979 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2980 why = WHY_NOT;
2981 JUMPTO(PyLong_AS_LONG(retval));
2982 Py_DECREF(retval);
2983 break;
2984 }
2985 /* Now we have to pop the block. */
2986 f->f_iblock--;
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002988 if (b->b_type == EXCEPT_HANDLER) {
2989 UNWIND_EXCEPT_HANDLER(b);
2990 continue;
2991 }
2992 UNWIND_BLOCK(b);
2993 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2994 why = WHY_NOT;
2995 JUMPTO(b->b_handler);
2996 break;
2997 }
2998 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
2999 || b->b_type == SETUP_FINALLY)) {
3000 PyObject *exc, *val, *tb;
3001 int handler = b->b_handler;
3002 /* Beware, this invalidates all b->b_* fields */
3003 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
3004 PUSH(tstate->exc_traceback);
3005 PUSH(tstate->exc_value);
3006 if (tstate->exc_type != NULL) {
3007 PUSH(tstate->exc_type);
3008 }
3009 else {
3010 Py_INCREF(Py_None);
3011 PUSH(Py_None);
3012 }
3013 PyErr_Fetch(&exc, &val, &tb);
3014 /* Make the raw exception data
3015 available to the handler,
3016 so a program can emulate the
3017 Python main loop. */
3018 PyErr_NormalizeException(
3019 &exc, &val, &tb);
3020 PyException_SetTraceback(val, tb);
3021 Py_INCREF(exc);
3022 tstate->exc_type = exc;
3023 Py_INCREF(val);
3024 tstate->exc_value = val;
3025 tstate->exc_traceback = tb;
3026 if (tb == NULL)
3027 tb = Py_None;
3028 Py_INCREF(tb);
3029 PUSH(tb);
3030 PUSH(val);
3031 PUSH(exc);
3032 why = WHY_NOT;
3033 JUMPTO(handler);
3034 break;
3035 }
3036 if (b->b_type == SETUP_FINALLY) {
3037 if (why & (WHY_RETURN | WHY_CONTINUE))
3038 PUSH(retval);
3039 PUSH(PyLong_FromLong((long)why));
3040 why = WHY_NOT;
3041 JUMPTO(b->b_handler);
3042 break;
3043 }
3044 } /* unwind stack */
Guido van Rossum374a9221991-04-04 10:40:29 +00003045
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003046 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00003047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003048 if (why != WHY_NOT)
3049 break;
3050 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00003051
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003052 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00003053
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003054 assert(why != WHY_YIELD);
3055 /* Pop remaining stack entries. */
3056 while (!EMPTY()) {
3057 v = POP();
3058 Py_XDECREF(v);
3059 }
Guido van Rossum35974fb2001-12-06 21:28:18 +00003060
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003061 if (why != WHY_RETURN)
3062 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00003063
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003064fast_yield:
Benjamin Petersonac913412011-07-03 16:25:11 -05003065 if (co->co_flags & CO_GENERATOR && (why == WHY_YIELD || why == WHY_RETURN)) {
3066 /* The purpose of this block is to put aside the generator's exception
3067 state and restore that of the calling frame. If the current
3068 exception state is from the caller, we clear the exception values
3069 on the generator frame, so they are not swapped back in latter. The
3070 origin of the current exception state is determined by checking for
3071 except handler blocks, which we must be in iff a new exception
3072 state came into existence in this frame. (An uncaught exception
3073 would have why == WHY_EXCEPTION, and we wouldn't be here). */
3074 int i;
3075 for (i = 0; i < f->f_iblock; i++)
3076 if (f->f_blockstack[i].b_type == EXCEPT_HANDLER)
3077 break;
3078 if (i == f->f_iblock)
3079 /* We did not create this exception. */
Benjamin Peterson87880242011-07-03 16:48:31 -05003080 restore_and_clear_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003081 else
Benjamin Peterson87880242011-07-03 16:48:31 -05003082 swap_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003083 }
Benjamin Peterson83195c32011-07-03 13:44:00 -05003084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003085 if (tstate->use_tracing) {
3086 if (tstate->c_tracefunc) {
3087 if (why == WHY_RETURN || why == WHY_YIELD) {
3088 if (call_trace(tstate->c_tracefunc,
3089 tstate->c_traceobj, f,
3090 PyTrace_RETURN, retval)) {
3091 Py_XDECREF(retval);
3092 retval = NULL;
3093 why = WHY_EXCEPTION;
3094 }
3095 }
3096 else if (why == WHY_EXCEPTION) {
3097 call_trace_protected(tstate->c_tracefunc,
3098 tstate->c_traceobj, f,
3099 PyTrace_RETURN, NULL);
3100 }
3101 }
3102 if (tstate->c_profilefunc) {
3103 if (why == WHY_EXCEPTION)
3104 call_trace_protected(tstate->c_profilefunc,
3105 tstate->c_profileobj, f,
3106 PyTrace_RETURN, NULL);
3107 else if (call_trace(tstate->c_profilefunc,
3108 tstate->c_profileobj, f,
3109 PyTrace_RETURN, retval)) {
3110 Py_XDECREF(retval);
3111 retval = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003112 /* why = WHY_EXCEPTION; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003113 }
3114 }
3115 }
Guido van Rossuma4240131997-01-21 21:18:36 +00003116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003117 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00003118exit_eval_frame:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003119 Py_LeaveRecursiveCall();
3120 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00003121
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003122 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00003123}
3124
Benjamin Petersonb204a422011-06-05 22:04:07 -05003125static void
Benjamin Petersone109c702011-06-24 09:37:26 -05003126format_missing(const char *kind, PyCodeObject *co, PyObject *names)
3127{
3128 int err;
3129 Py_ssize_t len = PyList_GET_SIZE(names);
3130 PyObject *name_str, *comma, *tail, *tmp;
3131
3132 assert(PyList_CheckExact(names));
3133 assert(len >= 1);
3134 /* Deal with the joys of natural language. */
3135 switch (len) {
3136 case 1:
3137 name_str = PyList_GET_ITEM(names, 0);
3138 Py_INCREF(name_str);
3139 break;
3140 case 2:
3141 name_str = PyUnicode_FromFormat("%U and %U",
3142 PyList_GET_ITEM(names, len - 2),
3143 PyList_GET_ITEM(names, len - 1));
3144 break;
3145 default:
3146 tail = PyUnicode_FromFormat(", %U, and %U",
3147 PyList_GET_ITEM(names, len - 2),
3148 PyList_GET_ITEM(names, len - 1));
3149 /* Chop off the last two objects in the list. This shouldn't actually
3150 fail, but we can't be too careful. */
3151 err = PyList_SetSlice(names, len - 2, len, NULL);
3152 if (err == -1) {
3153 Py_DECREF(tail);
3154 return;
3155 }
3156 /* Stitch everything up into a nice comma-separated list. */
3157 comma = PyUnicode_FromString(", ");
3158 if (comma == NULL) {
3159 Py_DECREF(tail);
3160 return;
3161 }
3162 tmp = PyUnicode_Join(comma, names);
3163 Py_DECREF(comma);
3164 if (tmp == NULL) {
3165 Py_DECREF(tail);
3166 return;
3167 }
3168 name_str = PyUnicode_Concat(tmp, tail);
3169 Py_DECREF(tmp);
3170 Py_DECREF(tail);
3171 break;
3172 }
3173 if (name_str == NULL)
3174 return;
3175 PyErr_Format(PyExc_TypeError,
3176 "%U() missing %i required %s argument%s: %U",
3177 co->co_name,
3178 len,
3179 kind,
3180 len == 1 ? "" : "s",
3181 name_str);
3182 Py_DECREF(name_str);
3183}
3184
3185static void
3186missing_arguments(PyCodeObject *co, int missing, int defcount,
3187 PyObject **fastlocals)
3188{
3189 int i, j = 0;
3190 int start, end;
3191 int positional = defcount != -1;
3192 const char *kind = positional ? "positional" : "keyword-only";
3193 PyObject *missing_names;
3194
3195 /* Compute the names of the arguments that are missing. */
3196 missing_names = PyList_New(missing);
3197 if (missing_names == NULL)
3198 return;
3199 if (positional) {
3200 start = 0;
3201 end = co->co_argcount - defcount;
3202 }
3203 else {
3204 start = co->co_argcount;
3205 end = start + co->co_kwonlyargcount;
3206 }
3207 for (i = start; i < end; i++) {
3208 if (GETLOCAL(i) == NULL) {
3209 PyObject *raw = PyTuple_GET_ITEM(co->co_varnames, i);
3210 PyObject *name = PyObject_Repr(raw);
3211 if (name == NULL) {
3212 Py_DECREF(missing_names);
3213 return;
3214 }
3215 PyList_SET_ITEM(missing_names, j++, name);
3216 }
3217 }
3218 assert(j == missing);
3219 format_missing(kind, co, missing_names);
3220 Py_DECREF(missing_names);
3221}
3222
3223static void
3224too_many_positional(PyCodeObject *co, int given, int defcount, PyObject **fastlocals)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003225{
3226 int plural;
3227 int kwonly_given = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003228 int i;
3229 PyObject *sig, *kwonly_sig;
3230
Benjamin Petersone109c702011-06-24 09:37:26 -05003231 assert((co->co_flags & CO_VARARGS) == 0);
3232 /* Count missing keyword-only args. */
Benjamin Petersonb204a422011-06-05 22:04:07 -05003233 for (i = co->co_argcount; i < co->co_argcount + co->co_kwonlyargcount; i++)
Benjamin Petersone109c702011-06-24 09:37:26 -05003234 if (GETLOCAL(i) != NULL)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003235 kwonly_given++;
Benjamin Petersone109c702011-06-24 09:37:26 -05003236 if (defcount) {
3237 int atleast = co->co_argcount - defcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003238 plural = 1;
3239 sig = PyUnicode_FromFormat("from %d to %d", atleast, co->co_argcount);
3240 }
3241 else {
3242 plural = co->co_argcount != 1;
3243 sig = PyUnicode_FromFormat("%d", co->co_argcount);
3244 }
3245 if (sig == NULL)
3246 return;
3247 if (kwonly_given) {
3248 const char *format = " positional argument%s (and %d keyword-only argument%s)";
3249 kwonly_sig = PyUnicode_FromFormat(format, given != 1 ? "s" : "", kwonly_given,
3250 kwonly_given != 1 ? "s" : "");
3251 if (kwonly_sig == NULL) {
3252 Py_DECREF(sig);
3253 return;
3254 }
3255 }
3256 else {
3257 /* This will not fail. */
3258 kwonly_sig = PyUnicode_FromString("");
Benjamin Petersone109c702011-06-24 09:37:26 -05003259 assert(kwonly_sig != NULL);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003260 }
3261 PyErr_Format(PyExc_TypeError,
3262 "%U() takes %U positional argument%s but %d%U %s given",
3263 co->co_name,
3264 sig,
3265 plural ? "s" : "",
3266 given,
3267 kwonly_sig,
3268 given == 1 && !kwonly_given ? "was" : "were");
3269 Py_DECREF(sig);
3270 Py_DECREF(kwonly_sig);
3271}
3272
Guido van Rossumc2e20742006-02-27 22:32:47 +00003273/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00003274 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00003275 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00003276
Tim Peters6d6c1a32001-08-02 04:15:00 +00003277PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003278PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003279 PyObject **args, int argcount, PyObject **kws, int kwcount,
3280 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00003281{
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003282 PyCodeObject* co = (PyCodeObject*)_co;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003283 register PyFrameObject *f;
3284 register PyObject *retval = NULL;
3285 register PyObject **fastlocals, **freevars;
3286 PyThreadState *tstate = PyThreadState_GET();
3287 PyObject *x, *u;
3288 int total_args = co->co_argcount + co->co_kwonlyargcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003289 int i;
3290 int n = argcount;
3291 PyObject *kwdict = NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003292
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003293 if (globals == NULL) {
3294 PyErr_SetString(PyExc_SystemError,
3295 "PyEval_EvalCodeEx: NULL globals");
3296 return NULL;
3297 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003298
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003299 assert(tstate != NULL);
3300 assert(globals != NULL);
3301 f = PyFrame_New(tstate, co, globals, locals);
3302 if (f == NULL)
3303 return NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003304
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003305 fastlocals = f->f_localsplus;
3306 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00003307
Benjamin Petersonb204a422011-06-05 22:04:07 -05003308 /* Parse arguments. */
3309 if (co->co_flags & CO_VARKEYWORDS) {
3310 kwdict = PyDict_New();
3311 if (kwdict == NULL)
3312 goto fail;
3313 i = total_args;
3314 if (co->co_flags & CO_VARARGS)
3315 i++;
3316 SETLOCAL(i, kwdict);
3317 }
3318 if (argcount > co->co_argcount)
3319 n = co->co_argcount;
3320 for (i = 0; i < n; i++) {
3321 x = args[i];
3322 Py_INCREF(x);
3323 SETLOCAL(i, x);
3324 }
3325 if (co->co_flags & CO_VARARGS) {
3326 u = PyTuple_New(argcount - n);
3327 if (u == NULL)
3328 goto fail;
3329 SETLOCAL(total_args, u);
3330 for (i = n; i < argcount; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003331 x = args[i];
3332 Py_INCREF(x);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003333 PyTuple_SET_ITEM(u, i-n, x);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003334 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003335 }
3336 for (i = 0; i < kwcount; i++) {
3337 PyObject **co_varnames;
3338 PyObject *keyword = kws[2*i];
3339 PyObject *value = kws[2*i + 1];
3340 int j;
3341 if (keyword == NULL || !PyUnicode_Check(keyword)) {
3342 PyErr_Format(PyExc_TypeError,
3343 "%U() keywords must be strings",
3344 co->co_name);
3345 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003346 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003347 /* Speed hack: do raw pointer compares. As names are
3348 normally interned this should almost always hit. */
3349 co_varnames = ((PyTupleObject *)(co->co_varnames))->ob_item;
3350 for (j = 0; j < total_args; j++) {
3351 PyObject *nm = co_varnames[j];
3352 if (nm == keyword)
3353 goto kw_found;
3354 }
3355 /* Slow fallback, just in case */
3356 for (j = 0; j < total_args; j++) {
3357 PyObject *nm = co_varnames[j];
3358 int cmp = PyObject_RichCompareBool(
3359 keyword, nm, Py_EQ);
3360 if (cmp > 0)
3361 goto kw_found;
3362 else if (cmp < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003363 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003364 }
3365 if (j >= total_args && kwdict == NULL) {
3366 PyErr_Format(PyExc_TypeError,
3367 "%U() got an unexpected "
3368 "keyword argument '%S'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003369 co->co_name,
3370 keyword);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003371 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003372 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003373 PyDict_SetItem(kwdict, keyword, value);
3374 continue;
3375 kw_found:
3376 if (GETLOCAL(j) != NULL) {
3377 PyErr_Format(PyExc_TypeError,
3378 "%U() got multiple "
3379 "values for argument '%S'",
3380 co->co_name,
3381 keyword);
3382 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003383 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003384 Py_INCREF(value);
3385 SETLOCAL(j, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003386 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003387 if (argcount > co->co_argcount && !(co->co_flags & CO_VARARGS)) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003388 too_many_positional(co, argcount, defcount, fastlocals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003389 goto fail;
3390 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003391 if (argcount < co->co_argcount) {
3392 int m = co->co_argcount - defcount;
Benjamin Petersone109c702011-06-24 09:37:26 -05003393 int missing = 0;
3394 for (i = argcount; i < m; i++)
3395 if (GETLOCAL(i) == NULL)
3396 missing++;
3397 if (missing) {
3398 missing_arguments(co, missing, defcount, fastlocals);
3399 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003400 }
3401 if (n > m)
3402 i = n - m;
3403 else
3404 i = 0;
3405 for (; i < defcount; i++) {
3406 if (GETLOCAL(m+i) == NULL) {
3407 PyObject *def = defs[i];
3408 Py_INCREF(def);
3409 SETLOCAL(m+i, def);
3410 }
3411 }
3412 }
3413 if (co->co_kwonlyargcount > 0) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003414 int missing = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003415 for (i = co->co_argcount; i < total_args; i++) {
3416 PyObject *name;
3417 if (GETLOCAL(i) != NULL)
3418 continue;
3419 name = PyTuple_GET_ITEM(co->co_varnames, i);
3420 if (kwdefs != NULL) {
3421 PyObject *def = PyDict_GetItem(kwdefs, name);
3422 if (def) {
3423 Py_INCREF(def);
3424 SETLOCAL(i, def);
3425 continue;
3426 }
3427 }
Benjamin Petersone109c702011-06-24 09:37:26 -05003428 missing++;
3429 }
3430 if (missing) {
3431 missing_arguments(co, missing, -1, fastlocals);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003432 goto fail;
3433 }
3434 }
3435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003436 /* Allocate and initialize storage for cell vars, and copy free
Benjamin Peterson90037602011-06-25 22:54:45 -05003437 vars into frame. */
3438 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003439 PyObject *c;
Benjamin Peterson90037602011-06-25 22:54:45 -05003440 int arg;
3441 /* Possibly account for the cell variable being an argument. */
3442 if (co->co_cell2arg != NULL &&
3443 (arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG)
3444 c = PyCell_New(GETLOCAL(arg));
3445 else
3446 c = PyCell_New(NULL);
3447 if (c == NULL)
3448 goto fail;
3449 SETLOCAL(co->co_nlocals + i, c);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003450 }
Benjamin Peterson90037602011-06-25 22:54:45 -05003451 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
3452 PyObject *o = PyTuple_GET_ITEM(closure, i);
3453 Py_INCREF(o);
3454 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003455 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003457 if (co->co_flags & CO_GENERATOR) {
3458 /* Don't need to keep the reference to f_back, it will be set
3459 * when the generator is resumed. */
3460 Py_XDECREF(f->f_back);
3461 f->f_back = NULL;
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003463 PCALL(PCALL_GENERATOR);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003464
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003465 /* Create a new generator that owns the ready to run frame
3466 * and return that as the value. */
3467 return PyGen_New(f);
3468 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003470 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003471
Thomas Woutersce272b62007-09-19 21:19:28 +00003472fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003474 /* decref'ing the frame can cause __del__ methods to get invoked,
3475 which can call back into Python. While we're done with the
3476 current Python frame (f), the associated C stack is still in use,
3477 so recursion_depth must be boosted for the duration.
3478 */
3479 assert(tstate != NULL);
3480 ++tstate->recursion_depth;
3481 Py_DECREF(f);
3482 --tstate->recursion_depth;
3483 return retval;
Tim Peters5ca576e2001-06-18 22:08:13 +00003484}
3485
3486
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003487static PyObject *
Benjamin Petersonce798522012-01-22 11:24:29 -05003488special_lookup(PyObject *o, _Py_Identifier *id)
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003489{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003490 PyObject *res;
Benjamin Petersonce798522012-01-22 11:24:29 -05003491 res = _PyObject_LookupSpecial(o, id);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003492 if (res == NULL && !PyErr_Occurred()) {
Benjamin Petersonce798522012-01-22 11:24:29 -05003493 PyErr_SetObject(PyExc_AttributeError, id->object);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003494 return NULL;
3495 }
3496 return res;
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003497}
3498
3499
Benjamin Peterson87880242011-07-03 16:48:31 -05003500/* These 3 functions deal with the exception state of generators. */
3501
3502static void
3503save_exc_state(PyThreadState *tstate, PyFrameObject *f)
3504{
3505 PyObject *type, *value, *traceback;
3506 Py_XINCREF(tstate->exc_type);
3507 Py_XINCREF(tstate->exc_value);
3508 Py_XINCREF(tstate->exc_traceback);
3509 type = f->f_exc_type;
3510 value = f->f_exc_value;
3511 traceback = f->f_exc_traceback;
3512 f->f_exc_type = tstate->exc_type;
3513 f->f_exc_value = tstate->exc_value;
3514 f->f_exc_traceback = tstate->exc_traceback;
3515 Py_XDECREF(type);
3516 Py_XDECREF(value);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02003517 Py_XDECREF(traceback);
Benjamin Peterson87880242011-07-03 16:48:31 -05003518}
3519
3520static void
3521swap_exc_state(PyThreadState *tstate, PyFrameObject *f)
3522{
3523 PyObject *tmp;
3524 tmp = tstate->exc_type;
3525 tstate->exc_type = f->f_exc_type;
3526 f->f_exc_type = tmp;
3527 tmp = tstate->exc_value;
3528 tstate->exc_value = f->f_exc_value;
3529 f->f_exc_value = tmp;
3530 tmp = tstate->exc_traceback;
3531 tstate->exc_traceback = f->f_exc_traceback;
3532 f->f_exc_traceback = tmp;
3533}
3534
3535static void
3536restore_and_clear_exc_state(PyThreadState *tstate, PyFrameObject *f)
3537{
3538 PyObject *type, *value, *tb;
3539 type = tstate->exc_type;
3540 value = tstate->exc_value;
3541 tb = tstate->exc_traceback;
3542 tstate->exc_type = f->f_exc_type;
3543 tstate->exc_value = f->f_exc_value;
3544 tstate->exc_traceback = f->f_exc_traceback;
3545 f->f_exc_type = NULL;
3546 f->f_exc_value = NULL;
3547 f->f_exc_traceback = NULL;
3548 Py_XDECREF(type);
3549 Py_XDECREF(value);
3550 Py_XDECREF(tb);
3551}
3552
3553
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003554/* Logic for the raise statement (too complicated for inlining).
3555 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00003556static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00003557do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003558{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003559 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00003560
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003561 if (exc == NULL) {
3562 /* Reraise */
3563 PyThreadState *tstate = PyThreadState_GET();
3564 PyObject *tb;
3565 type = tstate->exc_type;
3566 value = tstate->exc_value;
3567 tb = tstate->exc_traceback;
3568 if (type == Py_None) {
3569 PyErr_SetString(PyExc_RuntimeError,
3570 "No active exception to reraise");
3571 return WHY_EXCEPTION;
3572 }
3573 Py_XINCREF(type);
3574 Py_XINCREF(value);
3575 Py_XINCREF(tb);
3576 PyErr_Restore(type, value, tb);
3577 return WHY_RERAISE;
3578 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003579
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003580 /* We support the following forms of raise:
3581 raise
Collin Winter828f04a2007-08-31 00:04:24 +00003582 raise <instance>
3583 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 if (PyExceptionClass_Check(exc)) {
3586 type = exc;
3587 value = PyObject_CallObject(exc, NULL);
3588 if (value == NULL)
3589 goto raise_error;
Benjamin Peterson5afa03a2011-07-15 14:09:26 -05003590 if (!PyExceptionInstance_Check(value)) {
3591 PyErr_Format(PyExc_TypeError,
3592 "calling %R should have returned an instance of "
3593 "BaseException, not %R",
3594 type, Py_TYPE(value));
3595 goto raise_error;
3596 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003597 }
3598 else if (PyExceptionInstance_Check(exc)) {
3599 value = exc;
3600 type = PyExceptionInstance_Class(exc);
3601 Py_INCREF(type);
3602 }
3603 else {
3604 /* Not something you can raise. You get an exception
3605 anyway, just not what you specified :-) */
3606 Py_DECREF(exc);
3607 PyErr_SetString(PyExc_TypeError,
3608 "exceptions must derive from BaseException");
3609 goto raise_error;
3610 }
Collin Winter828f04a2007-08-31 00:04:24 +00003611
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003612 if (cause) {
3613 PyObject *fixed_cause;
Nick Coghlanab7bf212012-02-26 17:49:52 +10003614 int result;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003615 if (PyExceptionClass_Check(cause)) {
3616 fixed_cause = PyObject_CallObject(cause, NULL);
3617 if (fixed_cause == NULL)
3618 goto raise_error;
Nick Coghlanab7bf212012-02-26 17:49:52 +10003619 Py_CLEAR(cause);
3620 } else {
3621 /* Let "exc.__cause__ = cause" handle all further checks */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003622 fixed_cause = cause;
Nick Coghlanab7bf212012-02-26 17:49:52 +10003623 cause = NULL; /* Steal the reference */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003624 }
Nick Coghlanab7bf212012-02-26 17:49:52 +10003625 /* We retain ownership of the reference to fixed_cause */
3626 result = _PyException_SetCauseChecked(value, fixed_cause);
3627 Py_DECREF(fixed_cause);
3628 if (result < 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003629 goto raise_error;
3630 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003631 }
Collin Winter828f04a2007-08-31 00:04:24 +00003632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003633 PyErr_SetObject(type, value);
3634 /* PyErr_SetObject incref's its arguments */
3635 Py_XDECREF(value);
3636 Py_XDECREF(type);
3637 return WHY_EXCEPTION;
Collin Winter828f04a2007-08-31 00:04:24 +00003638
3639raise_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003640 Py_XDECREF(value);
3641 Py_XDECREF(type);
3642 Py_XDECREF(cause);
3643 return WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003644}
3645
Tim Petersd6d010b2001-06-21 02:49:55 +00003646/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003647 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00003648
Guido van Rossum0368b722007-05-11 16:50:42 +00003649 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3650 with a variable target.
3651*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003652
Barry Warsawe42b18f1997-08-25 22:13:04 +00003653static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003654unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003655{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003656 int i = 0, j = 0;
3657 Py_ssize_t ll = 0;
3658 PyObject *it; /* iter(v) */
3659 PyObject *w;
3660 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003662 assert(v != NULL);
Tim Petersd6d010b2001-06-21 02:49:55 +00003663
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003664 it = PyObject_GetIter(v);
3665 if (it == NULL)
3666 goto Error;
Tim Petersd6d010b2001-06-21 02:49:55 +00003667
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003668 for (; i < argcnt; i++) {
3669 w = PyIter_Next(it);
3670 if (w == NULL) {
3671 /* Iterator done, via error or exhaustion. */
3672 if (!PyErr_Occurred()) {
3673 PyErr_Format(PyExc_ValueError,
3674 "need more than %d value%s to unpack",
3675 i, i == 1 ? "" : "s");
3676 }
3677 goto Error;
3678 }
3679 *--sp = w;
3680 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003682 if (argcntafter == -1) {
3683 /* We better have exhausted the iterator now. */
3684 w = PyIter_Next(it);
3685 if (w == NULL) {
3686 if (PyErr_Occurred())
3687 goto Error;
3688 Py_DECREF(it);
3689 return 1;
3690 }
3691 Py_DECREF(w);
Georg Brandl0310a832010-07-10 10:32:36 +00003692 PyErr_Format(PyExc_ValueError, "too many values to unpack "
3693 "(expected %d)", argcnt);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003694 goto Error;
3695 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003696
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003697 l = PySequence_List(it);
3698 if (l == NULL)
3699 goto Error;
3700 *--sp = l;
3701 i++;
Guido van Rossum0368b722007-05-11 16:50:42 +00003702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003703 ll = PyList_GET_SIZE(l);
3704 if (ll < argcntafter) {
3705 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3706 argcnt + ll);
3707 goto Error;
3708 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003710 /* Pop the "after-variable" args off the list. */
3711 for (j = argcntafter; j > 0; j--, i++) {
3712 *--sp = PyList_GET_ITEM(l, ll - j);
3713 }
3714 /* Resize the list. */
3715 Py_SIZE(l) = ll - argcntafter;
3716 Py_DECREF(it);
3717 return 1;
Guido van Rossum0368b722007-05-11 16:50:42 +00003718
Tim Petersd6d010b2001-06-21 02:49:55 +00003719Error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003720 for (; i > 0; i--, sp++)
3721 Py_DECREF(*sp);
3722 Py_XDECREF(it);
3723 return 0;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003724}
3725
3726
Guido van Rossum96a42c81992-01-12 02:29:51 +00003727#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003728static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003729prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003730{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003731 printf("%s ", str);
3732 if (PyObject_Print(v, stdout, 0) != 0)
3733 PyErr_Clear(); /* Don't know what else to do */
3734 printf("\n");
3735 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003736}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003737#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003738
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003739static void
Fred Drake5755ce62001-06-27 19:19:46 +00003740call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003741{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003742 PyObject *type, *value, *traceback, *arg;
3743 int err;
3744 PyErr_Fetch(&type, &value, &traceback);
3745 if (value == NULL) {
3746 value = Py_None;
3747 Py_INCREF(value);
3748 }
3749 arg = PyTuple_Pack(3, type, value, traceback);
3750 if (arg == NULL) {
3751 PyErr_Restore(type, value, traceback);
3752 return;
3753 }
3754 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
3755 Py_DECREF(arg);
3756 if (err == 0)
3757 PyErr_Restore(type, value, traceback);
3758 else {
3759 Py_XDECREF(type);
3760 Py_XDECREF(value);
3761 Py_XDECREF(traceback);
3762 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003763}
3764
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003765static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003766call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003767 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003768{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003769 PyObject *type, *value, *traceback;
3770 int err;
3771 PyErr_Fetch(&type, &value, &traceback);
3772 err = call_trace(func, obj, frame, what, arg);
3773 if (err == 0)
3774 {
3775 PyErr_Restore(type, value, traceback);
3776 return 0;
3777 }
3778 else {
3779 Py_XDECREF(type);
3780 Py_XDECREF(value);
3781 Py_XDECREF(traceback);
3782 return -1;
3783 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003784}
3785
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003786static int
Fred Drake5755ce62001-06-27 19:19:46 +00003787call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003788 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003789{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003790 register PyThreadState *tstate = frame->f_tstate;
3791 int result;
3792 if (tstate->tracing)
3793 return 0;
3794 tstate->tracing++;
3795 tstate->use_tracing = 0;
3796 result = func(obj, frame, what, arg);
3797 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3798 || (tstate->c_profilefunc != NULL));
3799 tstate->tracing--;
3800 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003801}
3802
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003803PyObject *
3804_PyEval_CallTracing(PyObject *func, PyObject *args)
3805{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003806 PyFrameObject *frame = PyEval_GetFrame();
3807 PyThreadState *tstate = frame->f_tstate;
3808 int save_tracing = tstate->tracing;
3809 int save_use_tracing = tstate->use_tracing;
3810 PyObject *result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003811
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003812 tstate->tracing = 0;
3813 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3814 || (tstate->c_profilefunc != NULL));
3815 result = PyObject_Call(func, args, NULL);
3816 tstate->tracing = save_tracing;
3817 tstate->use_tracing = save_use_tracing;
3818 return result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003819}
3820
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00003821/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Michael W. Hudson006c7522002-11-08 13:08:46 +00003822static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003823maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003824 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3825 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003826{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003827 int result = 0;
3828 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003830 /* If the last instruction executed isn't in the current
3831 instruction window, reset the window.
3832 */
3833 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
3834 PyAddrPair bounds;
3835 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3836 &bounds);
3837 *instr_lb = bounds.ap_lower;
3838 *instr_ub = bounds.ap_upper;
3839 }
3840 /* If the last instruction falls at the start of a line or if
3841 it represents a jump backwards, update the frame's line
3842 number and call the trace function. */
3843 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3844 frame->f_lineno = line;
3845 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
3846 }
3847 *instr_prev = frame->f_lasti;
3848 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003849}
3850
Fred Drake5755ce62001-06-27 19:19:46 +00003851void
3852PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003854 PyThreadState *tstate = PyThreadState_GET();
3855 PyObject *temp = tstate->c_profileobj;
3856 Py_XINCREF(arg);
3857 tstate->c_profilefunc = NULL;
3858 tstate->c_profileobj = NULL;
3859 /* Must make sure that tracing is not ignored if 'temp' is freed */
3860 tstate->use_tracing = tstate->c_tracefunc != NULL;
3861 Py_XDECREF(temp);
3862 tstate->c_profilefunc = func;
3863 tstate->c_profileobj = arg;
3864 /* Flag that tracing or profiling is turned on */
3865 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003866}
3867
3868void
3869PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3870{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003871 PyThreadState *tstate = PyThreadState_GET();
3872 PyObject *temp = tstate->c_traceobj;
3873 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
3874 Py_XINCREF(arg);
3875 tstate->c_tracefunc = NULL;
3876 tstate->c_traceobj = NULL;
3877 /* Must make sure that profiling is not ignored if 'temp' is freed */
3878 tstate->use_tracing = tstate->c_profilefunc != NULL;
3879 Py_XDECREF(temp);
3880 tstate->c_tracefunc = func;
3881 tstate->c_traceobj = arg;
3882 /* Flag that tracing or profiling is turned on */
3883 tstate->use_tracing = ((func != NULL)
3884 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003885}
3886
Guido van Rossumb209a111997-04-29 18:18:01 +00003887PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003888PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003889{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003890 PyFrameObject *current_frame = PyEval_GetFrame();
3891 if (current_frame == NULL)
3892 return PyThreadState_GET()->interp->builtins;
3893 else
3894 return current_frame->f_builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003895}
3896
Guido van Rossumb209a111997-04-29 18:18:01 +00003897PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003898PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003899{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003900 PyFrameObject *current_frame = PyEval_GetFrame();
3901 if (current_frame == NULL)
3902 return NULL;
3903 PyFrame_FastToLocals(current_frame);
3904 return current_frame->f_locals;
Guido van Rossum5b722181993-03-30 17:46:03 +00003905}
3906
Guido van Rossumb209a111997-04-29 18:18:01 +00003907PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003908PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003909{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003910 PyFrameObject *current_frame = PyEval_GetFrame();
3911 if (current_frame == NULL)
3912 return NULL;
3913 else
3914 return current_frame->f_globals;
Guido van Rossum3f5da241990-12-20 15:06:42 +00003915}
3916
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003917PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003918PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003919{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003920 PyThreadState *tstate = PyThreadState_GET();
3921 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003922}
3923
Guido van Rossum6135a871995-01-09 17:53:26 +00003924int
Tim Peters5ba58662001-07-16 02:29:45 +00003925PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003926{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003927 PyFrameObject *current_frame = PyEval_GetFrame();
3928 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003929
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003930 if (current_frame != NULL) {
3931 const int codeflags = current_frame->f_code->co_flags;
3932 const int compilerflags = codeflags & PyCF_MASK;
3933 if (compilerflags) {
3934 result = 1;
3935 cf->cf_flags |= compilerflags;
3936 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003937#if 0 /* future keyword */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003938 if (codeflags & CO_GENERATOR_ALLOWED) {
3939 result = 1;
3940 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3941 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003942#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003943 }
3944 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003945}
3946
Guido van Rossum3f5da241990-12-20 15:06:42 +00003947
Guido van Rossum681d79a1995-07-18 14:51:37 +00003948/* External interface to call any callable object.
Antoine Pitrou8689a102010-04-01 16:53:15 +00003949 The arg must be a tuple or NULL. The kw must be a dict or NULL. */
Guido van Rossume59214e1994-08-30 08:01:59 +00003950
Guido van Rossumb209a111997-04-29 18:18:01 +00003951PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003952PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003953{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003954 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003956 if (arg == NULL) {
3957 arg = PyTuple_New(0);
3958 if (arg == NULL)
3959 return NULL;
3960 }
3961 else if (!PyTuple_Check(arg)) {
3962 PyErr_SetString(PyExc_TypeError,
3963 "argument list must be a tuple");
3964 return NULL;
3965 }
3966 else
3967 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003968
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003969 if (kw != NULL && !PyDict_Check(kw)) {
3970 PyErr_SetString(PyExc_TypeError,
3971 "keyword list must be a dictionary");
3972 Py_DECREF(arg);
3973 return NULL;
3974 }
Guido van Rossume3e61c11995-08-04 04:14:47 +00003975
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003976 result = PyObject_Call(func, arg, kw);
3977 Py_DECREF(arg);
3978 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00003979}
3980
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003981const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003982PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003983{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003984 if (PyMethod_Check(func))
3985 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
3986 else if (PyFunction_Check(func))
3987 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
3988 else if (PyCFunction_Check(func))
3989 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3990 else
3991 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003992}
3993
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003994const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003995PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003996{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003997 if (PyMethod_Check(func))
3998 return "()";
3999 else if (PyFunction_Check(func))
4000 return "()";
4001 else if (PyCFunction_Check(func))
4002 return "()";
4003 else
4004 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00004005}
4006
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00004007static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00004008err_args(PyObject *func, int flags, int nargs)
4009{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004010 if (flags & METH_NOARGS)
4011 PyErr_Format(PyExc_TypeError,
4012 "%.200s() takes no arguments (%d given)",
4013 ((PyCFunctionObject *)func)->m_ml->ml_name,
4014 nargs);
4015 else
4016 PyErr_Format(PyExc_TypeError,
4017 "%.200s() takes exactly one argument (%d given)",
4018 ((PyCFunctionObject *)func)->m_ml->ml_name,
4019 nargs);
Jeremy Hylton192690e2002-08-16 18:36:11 +00004020}
4021
Armin Rigo1c2d7e52005-09-20 18:34:01 +00004022#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00004023if (tstate->use_tracing && tstate->c_profilefunc) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004024 if (call_trace(tstate->c_profilefunc, \
4025 tstate->c_profileobj, \
4026 tstate->frame, PyTrace_C_CALL, \
4027 func)) { \
4028 x = NULL; \
4029 } \
4030 else { \
4031 x = call; \
4032 if (tstate->c_profilefunc != NULL) { \
4033 if (x == NULL) { \
4034 call_trace_protected(tstate->c_profilefunc, \
4035 tstate->c_profileobj, \
4036 tstate->frame, PyTrace_C_EXCEPTION, \
4037 func); \
4038 /* XXX should pass (type, value, tb) */ \
4039 } else { \
4040 if (call_trace(tstate->c_profilefunc, \
4041 tstate->c_profileobj, \
4042 tstate->frame, PyTrace_C_RETURN, \
4043 func)) { \
4044 Py_DECREF(x); \
4045 x = NULL; \
4046 } \
4047 } \
4048 } \
4049 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00004050} else { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004051 x = call; \
4052 }
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00004053
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004054static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004055call_function(PyObject ***pp_stack, int oparg
4056#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004057 , uint64* pintr0, uint64* pintr1
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004058#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004059 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004060{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004061 int na = oparg & 0xff;
4062 int nk = (oparg>>8) & 0xff;
4063 int n = na + 2 * nk;
4064 PyObject **pfunc = (*pp_stack) - n - 1;
4065 PyObject *func = *pfunc;
4066 PyObject *x, *w;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004067
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004068 /* Always dispatch PyCFunction first, because these are
4069 presumed to be the most frequent callable object.
4070 */
4071 if (PyCFunction_Check(func) && nk == 0) {
4072 int flags = PyCFunction_GET_FLAGS(func);
4073 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00004074
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004075 PCALL(PCALL_CFUNCTION);
4076 if (flags & (METH_NOARGS | METH_O)) {
4077 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
4078 PyObject *self = PyCFunction_GET_SELF(func);
4079 if (flags & METH_NOARGS && na == 0) {
4080 C_TRACE(x, (*meth)(self,NULL));
4081 }
4082 else if (flags & METH_O && na == 1) {
4083 PyObject *arg = EXT_POP(*pp_stack);
4084 C_TRACE(x, (*meth)(self,arg));
4085 Py_DECREF(arg);
4086 }
4087 else {
4088 err_args(func, flags, na);
4089 x = NULL;
4090 }
4091 }
4092 else {
4093 PyObject *callargs;
4094 callargs = load_args(pp_stack, na);
4095 READ_TIMESTAMP(*pintr0);
4096 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
4097 READ_TIMESTAMP(*pintr1);
4098 Py_XDECREF(callargs);
4099 }
4100 } else {
4101 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
4102 /* optimize access to bound methods */
4103 PyObject *self = PyMethod_GET_SELF(func);
4104 PCALL(PCALL_METHOD);
4105 PCALL(PCALL_BOUND_METHOD);
4106 Py_INCREF(self);
4107 func = PyMethod_GET_FUNCTION(func);
4108 Py_INCREF(func);
4109 Py_DECREF(*pfunc);
4110 *pfunc = self;
4111 na++;
4112 n++;
4113 } else
4114 Py_INCREF(func);
4115 READ_TIMESTAMP(*pintr0);
4116 if (PyFunction_Check(func))
4117 x = fast_function(func, pp_stack, n, na, nk);
4118 else
4119 x = do_call(func, pp_stack, na, nk);
4120 READ_TIMESTAMP(*pintr1);
4121 Py_DECREF(func);
4122 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00004123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004124 /* Clear the stack of the function object. Also removes
4125 the arguments in case they weren't consumed already
4126 (fast_function() and err_args() leave them on the stack).
4127 */
4128 while ((*pp_stack) > pfunc) {
4129 w = EXT_POP(*pp_stack);
4130 Py_DECREF(w);
4131 PCALL(PCALL_POP);
4132 }
4133 return x;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004134}
4135
Jeremy Hylton192690e2002-08-16 18:36:11 +00004136/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004137 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004138 For the simplest case -- a function that takes only positional
4139 arguments and is called with only positional arguments -- it
4140 inlines the most primitive frame setup code from
4141 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4142 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004143*/
4144
4145static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004146fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004147{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004148 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
4149 PyObject *globals = PyFunction_GET_GLOBALS(func);
4150 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4151 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
4152 PyObject **d = NULL;
4153 int nd = 0;
Jeremy Hylton52820442001-01-03 23:52:36 +00004154
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004155 PCALL(PCALL_FUNCTION);
4156 PCALL(PCALL_FAST_FUNCTION);
4157 if (argdefs == NULL && co->co_argcount == n &&
4158 co->co_kwonlyargcount == 0 && nk==0 &&
4159 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4160 PyFrameObject *f;
4161 PyObject *retval = NULL;
4162 PyThreadState *tstate = PyThreadState_GET();
4163 PyObject **fastlocals, **stack;
4164 int i;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004165
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004166 PCALL(PCALL_FASTER_FUNCTION);
4167 assert(globals != NULL);
4168 /* XXX Perhaps we should create a specialized
4169 PyFrame_New() that doesn't take locals, but does
4170 take builtins without sanity checking them.
4171 */
4172 assert(tstate != NULL);
4173 f = PyFrame_New(tstate, co, globals, NULL);
4174 if (f == NULL)
4175 return NULL;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004176
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004177 fastlocals = f->f_localsplus;
4178 stack = (*pp_stack) - n;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004180 for (i = 0; i < n; i++) {
4181 Py_INCREF(*stack);
4182 fastlocals[i] = *stack++;
4183 }
4184 retval = PyEval_EvalFrameEx(f,0);
4185 ++tstate->recursion_depth;
4186 Py_DECREF(f);
4187 --tstate->recursion_depth;
4188 return retval;
4189 }
4190 if (argdefs != NULL) {
4191 d = &PyTuple_GET_ITEM(argdefs, 0);
4192 nd = Py_SIZE(argdefs);
4193 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004194 return PyEval_EvalCodeEx((PyObject*)co, globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004195 (PyObject *)NULL, (*pp_stack)-n, na,
4196 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
4197 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004198}
4199
4200static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004201update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4202 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004203{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004204 PyObject *kwdict = NULL;
4205 if (orig_kwdict == NULL)
4206 kwdict = PyDict_New();
4207 else {
4208 kwdict = PyDict_Copy(orig_kwdict);
4209 Py_DECREF(orig_kwdict);
4210 }
4211 if (kwdict == NULL)
4212 return NULL;
4213 while (--nk >= 0) {
4214 int err;
4215 PyObject *value = EXT_POP(*pp_stack);
4216 PyObject *key = EXT_POP(*pp_stack);
4217 if (PyDict_GetItem(kwdict, key) != NULL) {
4218 PyErr_Format(PyExc_TypeError,
4219 "%.200s%s got multiple values "
4220 "for keyword argument '%U'",
4221 PyEval_GetFuncName(func),
4222 PyEval_GetFuncDesc(func),
4223 key);
4224 Py_DECREF(key);
4225 Py_DECREF(value);
4226 Py_DECREF(kwdict);
4227 return NULL;
4228 }
4229 err = PyDict_SetItem(kwdict, key, value);
4230 Py_DECREF(key);
4231 Py_DECREF(value);
4232 if (err) {
4233 Py_DECREF(kwdict);
4234 return NULL;
4235 }
4236 }
4237 return kwdict;
Jeremy Hylton52820442001-01-03 23:52:36 +00004238}
4239
4240static PyObject *
4241update_star_args(int nstack, int nstar, PyObject *stararg,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004242 PyObject ***pp_stack)
Jeremy Hylton52820442001-01-03 23:52:36 +00004243{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004244 PyObject *callargs, *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004245
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004246 callargs = PyTuple_New(nstack + nstar);
4247 if (callargs == NULL) {
4248 return NULL;
4249 }
4250 if (nstar) {
4251 int i;
4252 for (i = 0; i < nstar; i++) {
4253 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4254 Py_INCREF(a);
4255 PyTuple_SET_ITEM(callargs, nstack + i, a);
4256 }
4257 }
4258 while (--nstack >= 0) {
4259 w = EXT_POP(*pp_stack);
4260 PyTuple_SET_ITEM(callargs, nstack, w);
4261 }
4262 return callargs;
Jeremy Hylton52820442001-01-03 23:52:36 +00004263}
4264
4265static PyObject *
4266load_args(PyObject ***pp_stack, int na)
4267{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004268 PyObject *args = PyTuple_New(na);
4269 PyObject *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004270
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004271 if (args == NULL)
4272 return NULL;
4273 while (--na >= 0) {
4274 w = EXT_POP(*pp_stack);
4275 PyTuple_SET_ITEM(args, na, w);
4276 }
4277 return args;
Jeremy Hylton52820442001-01-03 23:52:36 +00004278}
4279
4280static PyObject *
4281do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4282{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004283 PyObject *callargs = NULL;
4284 PyObject *kwdict = NULL;
4285 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004287 if (nk > 0) {
4288 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
4289 if (kwdict == NULL)
4290 goto call_fail;
4291 }
4292 callargs = load_args(pp_stack, na);
4293 if (callargs == NULL)
4294 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004295#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004296 /* At this point, we have to look at the type of func to
4297 update the call stats properly. Do it here so as to avoid
4298 exposing the call stats machinery outside ceval.c
4299 */
4300 if (PyFunction_Check(func))
4301 PCALL(PCALL_FUNCTION);
4302 else if (PyMethod_Check(func))
4303 PCALL(PCALL_METHOD);
4304 else if (PyType_Check(func))
4305 PCALL(PCALL_TYPE);
4306 else if (PyCFunction_Check(func))
4307 PCALL(PCALL_CFUNCTION);
4308 else
4309 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004310#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004311 if (PyCFunction_Check(func)) {
4312 PyThreadState *tstate = PyThreadState_GET();
4313 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4314 }
4315 else
4316 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00004317call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004318 Py_XDECREF(callargs);
4319 Py_XDECREF(kwdict);
4320 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004321}
4322
4323static PyObject *
4324ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4325{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004326 int nstar = 0;
4327 PyObject *callargs = NULL;
4328 PyObject *stararg = NULL;
4329 PyObject *kwdict = NULL;
4330 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004331
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004332 if (flags & CALL_FLAG_KW) {
4333 kwdict = EXT_POP(*pp_stack);
4334 if (!PyDict_Check(kwdict)) {
4335 PyObject *d;
4336 d = PyDict_New();
4337 if (d == NULL)
4338 goto ext_call_fail;
4339 if (PyDict_Update(d, kwdict) != 0) {
4340 Py_DECREF(d);
4341 /* PyDict_Update raises attribute
4342 * error (percolated from an attempt
4343 * to get 'keys' attribute) instead of
4344 * a type error if its second argument
4345 * is not a mapping.
4346 */
4347 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4348 PyErr_Format(PyExc_TypeError,
4349 "%.200s%.200s argument after ** "
4350 "must be a mapping, not %.200s",
4351 PyEval_GetFuncName(func),
4352 PyEval_GetFuncDesc(func),
4353 kwdict->ob_type->tp_name);
4354 }
4355 goto ext_call_fail;
4356 }
4357 Py_DECREF(kwdict);
4358 kwdict = d;
4359 }
4360 }
4361 if (flags & CALL_FLAG_VAR) {
4362 stararg = EXT_POP(*pp_stack);
4363 if (!PyTuple_Check(stararg)) {
4364 PyObject *t = NULL;
4365 t = PySequence_Tuple(stararg);
4366 if (t == NULL) {
4367 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4368 PyErr_Format(PyExc_TypeError,
4369 "%.200s%.200s argument after * "
Victor Stinner0a5f65a2011-03-22 01:09:21 +01004370 "must be a sequence, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004371 PyEval_GetFuncName(func),
4372 PyEval_GetFuncDesc(func),
4373 stararg->ob_type->tp_name);
4374 }
4375 goto ext_call_fail;
4376 }
4377 Py_DECREF(stararg);
4378 stararg = t;
4379 }
4380 nstar = PyTuple_GET_SIZE(stararg);
4381 }
4382 if (nk > 0) {
4383 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
4384 if (kwdict == NULL)
4385 goto ext_call_fail;
4386 }
4387 callargs = update_star_args(na, nstar, stararg, pp_stack);
4388 if (callargs == NULL)
4389 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004390#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004391 /* At this point, we have to look at the type of func to
4392 update the call stats properly. Do it here so as to avoid
4393 exposing the call stats machinery outside ceval.c
4394 */
4395 if (PyFunction_Check(func))
4396 PCALL(PCALL_FUNCTION);
4397 else if (PyMethod_Check(func))
4398 PCALL(PCALL_METHOD);
4399 else if (PyType_Check(func))
4400 PCALL(PCALL_TYPE);
4401 else if (PyCFunction_Check(func))
4402 PCALL(PCALL_CFUNCTION);
4403 else
4404 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004405#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004406 if (PyCFunction_Check(func)) {
4407 PyThreadState *tstate = PyThreadState_GET();
4408 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4409 }
4410 else
4411 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00004412ext_call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 Py_XDECREF(callargs);
4414 Py_XDECREF(kwdict);
4415 Py_XDECREF(stararg);
4416 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004417}
4418
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004419/* Extract a slice index from a PyInt or PyLong or an object with the
4420 nb_index slot defined, and store in *pi.
4421 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4422 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 +00004423 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004424*/
Tim Petersb5196382001-12-16 19:44:20 +00004425/* Note: If v is NULL, return success without storing into *pi. This
4426 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4427 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004428*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004429int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004430_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004431{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004432 if (v != NULL) {
4433 Py_ssize_t x;
4434 if (PyIndex_Check(v)) {
4435 x = PyNumber_AsSsize_t(v, NULL);
4436 if (x == -1 && PyErr_Occurred())
4437 return 0;
4438 }
4439 else {
4440 PyErr_SetString(PyExc_TypeError,
4441 "slice indices must be integers or "
4442 "None or have an __index__ method");
4443 return 0;
4444 }
4445 *pi = x;
4446 }
4447 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004448}
4449
Guido van Rossum486364b2007-06-30 05:01:58 +00004450#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004451 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00004452
Guido van Rossumb209a111997-04-29 18:18:01 +00004453static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004454cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004455{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004456 int res = 0;
4457 switch (op) {
4458 case PyCmp_IS:
4459 res = (v == w);
4460 break;
4461 case PyCmp_IS_NOT:
4462 res = (v != w);
4463 break;
4464 case PyCmp_IN:
4465 res = PySequence_Contains(w, v);
4466 if (res < 0)
4467 return NULL;
4468 break;
4469 case PyCmp_NOT_IN:
4470 res = PySequence_Contains(w, v);
4471 if (res < 0)
4472 return NULL;
4473 res = !res;
4474 break;
4475 case PyCmp_EXC_MATCH:
4476 if (PyTuple_Check(w)) {
4477 Py_ssize_t i, length;
4478 length = PyTuple_Size(w);
4479 for (i = 0; i < length; i += 1) {
4480 PyObject *exc = PyTuple_GET_ITEM(w, i);
4481 if (!PyExceptionClass_Check(exc)) {
4482 PyErr_SetString(PyExc_TypeError,
4483 CANNOT_CATCH_MSG);
4484 return NULL;
4485 }
4486 }
4487 }
4488 else {
4489 if (!PyExceptionClass_Check(w)) {
4490 PyErr_SetString(PyExc_TypeError,
4491 CANNOT_CATCH_MSG);
4492 return NULL;
4493 }
4494 }
4495 res = PyErr_GivenExceptionMatches(v, w);
4496 break;
4497 default:
4498 return PyObject_RichCompare(v, w, op);
4499 }
4500 v = res ? Py_True : Py_False;
4501 Py_INCREF(v);
4502 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004503}
4504
Thomas Wouters52152252000-08-17 22:55:00 +00004505static PyObject *
4506import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004507{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004508 PyObject *x;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004510 x = PyObject_GetAttr(v, name);
4511 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
4512 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
4513 }
4514 return x;
Thomas Wouters52152252000-08-17 22:55:00 +00004515}
Guido van Rossumac7be682001-01-17 15:42:30 +00004516
Thomas Wouters52152252000-08-17 22:55:00 +00004517static int
4518import_all_from(PyObject *locals, PyObject *v)
4519{
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004520 _Py_IDENTIFIER(__all__);
4521 _Py_IDENTIFIER(__dict__);
4522 PyObject *all = _PyObject_GetAttrId(v, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004523 PyObject *dict, *name, *value;
4524 int skip_leading_underscores = 0;
4525 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004527 if (all == NULL) {
4528 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4529 return -1; /* Unexpected error */
4530 PyErr_Clear();
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004531 dict = _PyObject_GetAttrId(v, &PyId___dict__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004532 if (dict == NULL) {
4533 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4534 return -1;
4535 PyErr_SetString(PyExc_ImportError,
4536 "from-import-* object has no __dict__ and no __all__");
4537 return -1;
4538 }
4539 all = PyMapping_Keys(dict);
4540 Py_DECREF(dict);
4541 if (all == NULL)
4542 return -1;
4543 skip_leading_underscores = 1;
4544 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004545
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004546 for (pos = 0, err = 0; ; pos++) {
4547 name = PySequence_GetItem(all, pos);
4548 if (name == NULL) {
4549 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4550 err = -1;
4551 else
4552 PyErr_Clear();
4553 break;
4554 }
4555 if (skip_leading_underscores &&
4556 PyUnicode_Check(name) &&
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004557 PyUnicode_READY(name) != -1 &&
4558 PyUnicode_READ_CHAR(name, 0) == '_')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004559 {
4560 Py_DECREF(name);
4561 continue;
4562 }
4563 value = PyObject_GetAttr(v, name);
4564 if (value == NULL)
4565 err = -1;
4566 else if (PyDict_CheckExact(locals))
4567 err = PyDict_SetItem(locals, name, value);
4568 else
4569 err = PyObject_SetItem(locals, name, value);
4570 Py_DECREF(name);
4571 Py_XDECREF(value);
4572 if (err != 0)
4573 break;
4574 }
4575 Py_DECREF(all);
4576 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004577}
4578
Guido van Rossumac7be682001-01-17 15:42:30 +00004579static void
Neal Norwitzda059e32007-08-26 05:33:45 +00004580format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00004581{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00004583
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004584 if (!obj)
4585 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004587 obj_str = _PyUnicode_AsString(obj);
4588 if (!obj_str)
4589 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004590
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004591 PyErr_Format(exc, format_str, obj_str);
Paul Prescode68140d2000-08-30 20:25:01 +00004592}
Guido van Rossum950361c1997-01-24 13:49:28 +00004593
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00004594static void
4595format_exc_unbound(PyCodeObject *co, int oparg)
4596{
4597 PyObject *name;
4598 /* Don't stomp existing exception */
4599 if (PyErr_Occurred())
4600 return;
4601 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
4602 name = PyTuple_GET_ITEM(co->co_cellvars,
4603 oparg);
4604 format_exc_check_arg(
4605 PyExc_UnboundLocalError,
4606 UNBOUNDLOCAL_ERROR_MSG,
4607 name);
4608 } else {
4609 name = PyTuple_GET_ITEM(co->co_freevars, oparg -
4610 PyTuple_GET_SIZE(co->co_cellvars));
4611 format_exc_check_arg(PyExc_NameError,
4612 UNBOUNDFREE_ERROR_MSG, name);
4613 }
4614}
4615
Victor Stinnerd2a915d2011-10-02 20:34:20 +02004616static PyObject *
4617unicode_concatenate(PyObject *v, PyObject *w,
4618 PyFrameObject *f, unsigned char *next_instr)
4619{
4620 PyObject *res;
4621 if (Py_REFCNT(v) == 2) {
4622 /* In the common case, there are 2 references to the value
4623 * stored in 'variable' when the += is performed: one on the
4624 * value stack (in 'v') and one still stored in the
4625 * 'variable'. We try to delete the variable now to reduce
4626 * the refcnt to 1.
4627 */
4628 switch (*next_instr) {
4629 case STORE_FAST:
4630 {
4631 int oparg = PEEKARG();
4632 PyObject **fastlocals = f->f_localsplus;
4633 if (GETLOCAL(oparg) == v)
4634 SETLOCAL(oparg, NULL);
4635 break;
4636 }
4637 case STORE_DEREF:
4638 {
4639 PyObject **freevars = (f->f_localsplus +
4640 f->f_code->co_nlocals);
4641 PyObject *c = freevars[PEEKARG()];
4642 if (PyCell_GET(c) == v)
4643 PyCell_Set(c, NULL);
4644 break;
4645 }
4646 case STORE_NAME:
4647 {
4648 PyObject *names = f->f_code->co_names;
4649 PyObject *name = GETITEM(names, PEEKARG());
4650 PyObject *locals = f->f_locals;
4651 if (PyDict_CheckExact(locals) &&
4652 PyDict_GetItem(locals, name) == v) {
4653 if (PyDict_DelItem(locals, name) != 0) {
4654 PyErr_Clear();
4655 }
4656 }
4657 break;
4658 }
4659 }
4660 }
4661 res = v;
4662 PyUnicode_Append(&res, w);
4663 return res;
4664}
4665
Guido van Rossum950361c1997-01-24 13:49:28 +00004666#ifdef DYNAMIC_EXECUTION_PROFILE
4667
Skip Montanarof118cb12001-10-15 20:51:38 +00004668static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004669getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004670{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004671 int i;
4672 PyObject *l = PyList_New(256);
4673 if (l == NULL) return NULL;
4674 for (i = 0; i < 256; i++) {
4675 PyObject *x = PyLong_FromLong(a[i]);
4676 if (x == NULL) {
4677 Py_DECREF(l);
4678 return NULL;
4679 }
4680 PyList_SetItem(l, i, x);
4681 }
4682 for (i = 0; i < 256; i++)
4683 a[i] = 0;
4684 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004685}
4686
4687PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004688_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004689{
4690#ifndef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004691 return getarray(dxp);
Guido van Rossum950361c1997-01-24 13:49:28 +00004692#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004693 int i;
4694 PyObject *l = PyList_New(257);
4695 if (l == NULL) return NULL;
4696 for (i = 0; i < 257; i++) {
4697 PyObject *x = getarray(dxpairs[i]);
4698 if (x == NULL) {
4699 Py_DECREF(l);
4700 return NULL;
4701 }
4702 PyList_SetItem(l, i, x);
4703 }
4704 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004705#endif
4706}
4707
4708#endif