blob: 134d1eec4b7a6b3becc9b3d2a416841dd6531dd9 [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 Peterson876b2f22009-06-28 03:18:59 +0000141static PyObject * special_lookup(PyObject *, char *, PyObject **);
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
Antoine Pitroub52ec782009-01-25 16:34:23 +0000825/* Computed GOTOs, or
826 the-optimization-commonly-but-improperly-known-as-"threaded code"
827 using gcc's labels-as-values extension
828 (http://gcc.gnu.org/onlinedocs/gcc/Labels-as-Values.html).
829
830 The traditional bytecode evaluation loop uses a "switch" statement, which
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000831 decent compilers will optimize as a single indirect branch instruction
Antoine Pitroub52ec782009-01-25 16:34:23 +0000832 combined with a lookup table of jump addresses. However, since the
833 indirect jump instruction is shared by all opcodes, the CPU will have a
834 hard time making the right prediction for where to jump next (actually,
835 it will be always wrong except in the uncommon case of a sequence of
836 several identical opcodes).
837
838 "Threaded code" in contrast, uses an explicit jump table and an explicit
839 indirect jump instruction at the end of each opcode. Since the jump
840 instruction is at a different address for each opcode, the CPU will make a
841 separate prediction for each of these instructions, which is equivalent to
842 predicting the second opcode of each opcode pair. These predictions have
843 a much better chance to turn out valid, especially in small bytecode loops.
844
845 A mispredicted branch on a modern CPU flushes the whole pipeline and
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000846 can cost several CPU cycles (depending on the pipeline depth),
Antoine Pitroub52ec782009-01-25 16:34:23 +0000847 and potentially many more instructions (depending on the pipeline width).
848 A correctly predicted branch, however, is nearly free.
849
850 At the time of this writing, the "threaded code" version is up to 15-20%
851 faster than the normal "switch" version, depending on the compiler and the
852 CPU architecture.
853
854 We disable the optimization if DYNAMIC_EXECUTION_PROFILE is defined,
855 because it would render the measurements invalid.
856
857
858 NOTE: care must be taken that the compiler doesn't try to "optimize" the
859 indirect jumps by sharing them between all opcodes. Such optimizations
860 can be disabled on gcc by using the -fno-gcse flag (or possibly
861 -fno-crossjumping).
862*/
863
Antoine Pitrou042b1282010-08-13 21:15:58 +0000864#ifdef DYNAMIC_EXECUTION_PROFILE
Antoine Pitroub52ec782009-01-25 16:34:23 +0000865#undef USE_COMPUTED_GOTOS
Antoine Pitrou042b1282010-08-13 21:15:58 +0000866#define USE_COMPUTED_GOTOS 0
Antoine Pitroub52ec782009-01-25 16:34:23 +0000867#endif
868
Antoine Pitrou042b1282010-08-13 21:15:58 +0000869#ifdef HAVE_COMPUTED_GOTOS
870 #ifndef USE_COMPUTED_GOTOS
871 #define USE_COMPUTED_GOTOS 1
872 #endif
873#else
874 #if defined(USE_COMPUTED_GOTOS) && USE_COMPUTED_GOTOS
875 #error "Computed gotos are not supported on this compiler."
876 #endif
877 #undef USE_COMPUTED_GOTOS
878 #define USE_COMPUTED_GOTOS 0
879#endif
880
881#if USE_COMPUTED_GOTOS
Antoine Pitroub52ec782009-01-25 16:34:23 +0000882/* Import the static jump table */
883#include "opcode_targets.h"
884
885/* This macro is used when several opcodes defer to the same implementation
886 (e.g. SETUP_LOOP, SETUP_FINALLY) */
887#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000888 TARGET_##op: \
889 opcode = op; \
890 if (HAS_ARG(op)) \
891 oparg = NEXTARG(); \
892 case op: \
893 goto impl; \
Antoine Pitroub52ec782009-01-25 16:34:23 +0000894
895#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000896 TARGET_##op: \
897 opcode = op; \
898 if (HAS_ARG(op)) \
899 oparg = NEXTARG(); \
900 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000901
902
903#define DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000904 { \
905 if (!_Py_atomic_load_relaxed(&eval_breaker)) { \
906 FAST_DISPATCH(); \
907 } \
908 continue; \
909 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000910
911#ifdef LLTRACE
912#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 { \
914 if (!lltrace && !_Py_TracingPossible) { \
915 f->f_lasti = INSTR_OFFSET(); \
916 goto *opcode_targets[*next_instr++]; \
917 } \
918 goto fast_next_opcode; \
919 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000920#else
921#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 { \
923 if (!_Py_TracingPossible) { \
924 f->f_lasti = INSTR_OFFSET(); \
925 goto *opcode_targets[*next_instr++]; \
926 } \
927 goto fast_next_opcode; \
928 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000929#endif
930
931#else
932#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000933 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000934#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 /* silence compiler warnings about `impl` unused */ \
936 if (0) goto impl; \
937 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000938#define DISPATCH() continue
939#define FAST_DISPATCH() goto fast_next_opcode
940#endif
941
942
Neal Norwitza81d2202002-07-14 00:27:26 +0000943/* Tuple access macros */
944
945#ifndef Py_DEBUG
946#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
947#else
948#define GETITEM(v, i) PyTuple_GetItem((v), (i))
949#endif
950
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000951#ifdef WITH_TSC
952/* Use Pentium timestamp counter to mark certain events:
953 inst0 -- beginning of switch statement for opcode dispatch
954 inst1 -- end of switch statement (may be skipped)
955 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000956 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000957 (may be skipped)
958 intr1 -- beginning of long interruption
959 intr2 -- end of long interruption
960
961 Many opcodes call out to helper C functions. In some cases, the
962 time in those functions should be counted towards the time for the
963 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
964 calls another Python function; there's no point in charge all the
965 bytecode executed by the called function to the caller.
966
967 It's hard to make a useful judgement statically. In the presence
968 of operator overloading, it's impossible to tell if a call will
969 execute new Python code or not.
970
971 It's a case-by-case judgement. I'll use intr1 for the following
972 cases:
973
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000974 IMPORT_STAR
975 IMPORT_FROM
976 CALL_FUNCTION (and friends)
977
978 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
980 int ticked = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000981
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000982 READ_TIMESTAMP(inst0);
983 READ_TIMESTAMP(inst1);
984 READ_TIMESTAMP(loop0);
985 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000986
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000987 /* shut up the compiler */
988 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000989#endif
990
Guido van Rossum374a9221991-04-04 10:40:29 +0000991/* Code access macros */
992
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000993#define INSTR_OFFSET() ((int)(next_instr - first_instr))
994#define NEXTOP() (*next_instr++)
995#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
996#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
997#define JUMPTO(x) (next_instr = first_instr + (x))
998#define JUMPBY(x) (next_instr += (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000999
Raymond Hettingerf606f872003-03-16 03:11:04 +00001000/* OpCode prediction macros
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001001 Some opcodes tend to come in pairs thus making it possible to
1002 predict the second code when the first is run. For example,
1003 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
1004 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001006 Verifying the prediction costs a single high-speed test of a register
1007 variable against a constant. If the pairing was good, then the
1008 processor's own internal branch predication has a high likelihood of
1009 success, resulting in a nearly zero-overhead transition to the
1010 next opcode. A successful prediction saves a trip through the eval-loop
1011 including its two unpredictable branches, the HAS_ARG test and the
1012 switch-case. Combined with the processor's internal branch prediction,
1013 a successful PREDICT has the effect of making the two opcodes run as if
1014 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001015
Georg Brandl86b2fb92008-07-16 03:43:04 +00001016 If collecting opcode statistics, your choices are to either keep the
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001017 predictions turned-on and interpret the results as if some opcodes
1018 had been combined or turn-off predictions so that the opcode frequency
1019 counter updates for both opcodes.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001020
1021 Opcode prediction is disabled with threaded code, since the latter allows
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 the CPU to record separate branch prediction information for each
1023 opcode.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001024
Raymond Hettingerf606f872003-03-16 03:11:04 +00001025*/
1026
Antoine Pitrou042b1282010-08-13 21:15:58 +00001027#if defined(DYNAMIC_EXECUTION_PROFILE) || USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028#define PREDICT(op) if (0) goto PRED_##op
1029#define PREDICTED(op) PRED_##op:
1030#define PREDICTED_WITH_ARG(op) PRED_##op:
Raymond Hettingera7216982004-02-08 19:59:27 +00001031#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001032#define PREDICT(op) if (*next_instr == op) goto PRED_##op
1033#define PREDICTED(op) PRED_##op: next_instr++
1034#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Antoine Pitroub52ec782009-01-25 16:34:23 +00001035#endif
1036
Raymond Hettingerf606f872003-03-16 03:11:04 +00001037
Guido van Rossum374a9221991-04-04 10:40:29 +00001038/* Stack manipulation macros */
1039
Martin v. Löwis18e16552006-02-15 17:27:45 +00001040/* The stack can grow at most MAXINT deep, as co_nlocals and
1041 co_stacksize are ints. */
Stefan Krahb7e10102010-06-23 18:42:39 +00001042#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
1043#define EMPTY() (STACK_LEVEL() == 0)
1044#define TOP() (stack_pointer[-1])
1045#define SECOND() (stack_pointer[-2])
1046#define THIRD() (stack_pointer[-3])
1047#define FOURTH() (stack_pointer[-4])
1048#define PEEK(n) (stack_pointer[-(n)])
1049#define SET_TOP(v) (stack_pointer[-1] = (v))
1050#define SET_SECOND(v) (stack_pointer[-2] = (v))
1051#define SET_THIRD(v) (stack_pointer[-3] = (v))
1052#define SET_FOURTH(v) (stack_pointer[-4] = (v))
1053#define SET_VALUE(n, v) (stack_pointer[-(n)] = (v))
1054#define BASIC_STACKADJ(n) (stack_pointer += n)
1055#define BASIC_PUSH(v) (*stack_pointer++ = (v))
1056#define BASIC_POP() (*--stack_pointer)
Guido van Rossum374a9221991-04-04 10:40:29 +00001057
Guido van Rossum96a42c81992-01-12 02:29:51 +00001058#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001059#define PUSH(v) { (void)(BASIC_PUSH(v), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001060 lltrace && prtrace(TOP(), "push")); \
1061 assert(STACK_LEVEL() <= co->co_stacksize); }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001062#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001063 BASIC_POP())
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001064#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001065 lltrace && prtrace(TOP(), "stackadj")); \
1066 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes0449f632007-12-15 01:27:15 +00001067#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
Stefan Krahb7e10102010-06-23 18:42:39 +00001068 prtrace((STACK_POINTER)[-1], "ext_pop")), \
1069 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001070#else
Stefan Krahb7e10102010-06-23 18:42:39 +00001071#define PUSH(v) BASIC_PUSH(v)
1072#define POP() BASIC_POP()
1073#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001074#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001075#endif
1076
Guido van Rossum681d79a1995-07-18 14:51:37 +00001077/* Local variable macros */
1078
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001079#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +00001080
1081/* The SETLOCAL() macro must not DECREF the local variable in-place and
1082 then store the new value; it must copy the old value to a temporary
1083 value, then store the new value, and then DECREF the temporary value.
1084 This is because it is possible that during the DECREF the frame is
1085 accessed by other code (e.g. a __del__ method or gc.collect()) and the
1086 variable would be pointing to already-freed memory. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001087#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
Stefan Krahb7e10102010-06-23 18:42:39 +00001088 GETLOCAL(i) = value; \
1089 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +00001090
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001091
1092#define UNWIND_BLOCK(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001093 while (STACK_LEVEL() > (b)->b_level) { \
1094 PyObject *v = POP(); \
1095 Py_XDECREF(v); \
1096 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001097
1098#define UNWIND_EXCEPT_HANDLER(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001099 { \
1100 PyObject *type, *value, *traceback; \
1101 assert(STACK_LEVEL() >= (b)->b_level + 3); \
1102 while (STACK_LEVEL() > (b)->b_level + 3) { \
1103 value = POP(); \
1104 Py_XDECREF(value); \
1105 } \
1106 type = tstate->exc_type; \
1107 value = tstate->exc_value; \
1108 traceback = tstate->exc_traceback; \
1109 tstate->exc_type = POP(); \
1110 tstate->exc_value = POP(); \
1111 tstate->exc_traceback = POP(); \
1112 Py_XDECREF(type); \
1113 Py_XDECREF(value); \
1114 Py_XDECREF(traceback); \
1115 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001116
Guido van Rossuma027efa1997-05-05 20:56:21 +00001117/* Start of code */
1118
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001119 /* push frame */
1120 if (Py_EnterRecursiveCall(""))
1121 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +00001122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +00001124
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001125 if (tstate->use_tracing) {
1126 if (tstate->c_tracefunc != NULL) {
1127 /* tstate->c_tracefunc, if defined, is a
1128 function that will be called on *every* entry
1129 to a code block. Its return value, if not
1130 None, is a function that will be called at
1131 the start of each executed line of code.
1132 (Actually, the function must return itself
1133 in order to continue tracing.) The trace
1134 functions are called with three arguments:
1135 a pointer to the current frame, a string
1136 indicating why the function is called, and
1137 an argument which depends on the situation.
1138 The global trace function is also called
1139 whenever an exception is detected. */
1140 if (call_trace_protected(tstate->c_tracefunc,
1141 tstate->c_traceobj,
1142 f, PyTrace_CALL, Py_None)) {
1143 /* Trace function raised an error */
1144 goto exit_eval_frame;
1145 }
1146 }
1147 if (tstate->c_profilefunc != NULL) {
1148 /* Similar for c_profilefunc, except it needn't
1149 return itself and isn't called for "line" events */
1150 if (call_trace_protected(tstate->c_profilefunc,
1151 tstate->c_profileobj,
1152 f, PyTrace_CALL, Py_None)) {
1153 /* Profile function raised an error */
1154 goto exit_eval_frame;
1155 }
1156 }
1157 }
Neil Schemenauer6c0f2002001-09-04 19:03:35 +00001158
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001159 co = f->f_code;
1160 names = co->co_names;
1161 consts = co->co_consts;
1162 fastlocals = f->f_localsplus;
1163 freevars = f->f_localsplus + co->co_nlocals;
1164 first_instr = (unsigned char*) PyBytes_AS_STRING(co->co_code);
1165 /* An explanation is in order for the next line.
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 f->f_lasti now refers to the index of the last instruction
1168 executed. You might think this was obvious from the name, but
1169 this wasn't always true before 2.3! PyFrame_New now sets
1170 f->f_lasti to -1 (i.e. the index *before* the first instruction)
1171 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
1172 does work. Promise.
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001174 When the PREDICT() macros are enabled, some opcode pairs follow in
1175 direct succession without updating f->f_lasti. A successful
1176 prediction effectively links the two codes together as if they
1177 were a single new opcode; accordingly,f->f_lasti will point to
1178 the first code in the pair (for instance, GET_ITER followed by
1179 FOR_ITER is effectively a single opcode and f->f_lasti will point
1180 at to the beginning of the combined pair.)
1181 */
1182 next_instr = first_instr + f->f_lasti + 1;
1183 stack_pointer = f->f_stacktop;
1184 assert(stack_pointer != NULL);
1185 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001186
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001187 if (co->co_flags & CO_GENERATOR && !throwflag) {
1188 if (f->f_exc_type != NULL && f->f_exc_type != Py_None) {
1189 /* We were in an except handler when we left,
1190 restore the exception state which was put aside
1191 (see YIELD_VALUE). */
Benjamin Peterson87880242011-07-03 16:48:31 -05001192 swap_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 }
Benjamin Peterson87880242011-07-03 16:48:31 -05001194 else
1195 save_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001197
Tim Peters5ca576e2001-06-18 22:08:13 +00001198#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00001200#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00001201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 why = WHY_NOT;
1203 err = 0;
1204 x = Py_None; /* Not a reference, just anything non-NULL */
1205 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00001206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 if (throwflag) { /* support for generator.throw() */
1208 why = WHY_EXCEPTION;
1209 goto on_error;
1210 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001211
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001212 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001213#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 if (inst1 == 0) {
1215 /* Almost surely, the opcode executed a break
1216 or a continue, preventing inst1 from being set
1217 on the way out of the loop.
1218 */
1219 READ_TIMESTAMP(inst1);
1220 loop1 = inst1;
1221 }
1222 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
1223 intr0, intr1);
1224 ticked = 0;
1225 inst1 = 0;
1226 intr0 = 0;
1227 intr1 = 0;
1228 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001229#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001230 assert(stack_pointer >= f->f_valuestack); /* else underflow */
1231 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 /* Do periodic things. Doing this every time through
1234 the loop would add too much overhead, so we do it
1235 only every Nth instruction. We also do it if
1236 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
1237 event needs attention (e.g. a signal handler or
1238 async I/O handler); see Py_AddPendingCall() and
1239 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +00001240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001241 if (_Py_atomic_load_relaxed(&eval_breaker)) {
1242 if (*next_instr == SETUP_FINALLY) {
1243 /* Make the last opcode before
Ezio Melotti13925002011-03-16 11:05:33 +02001244 a try: finally: block uninterruptible. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001245 goto fast_next_opcode;
1246 }
1247 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001248#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 ticked = 1;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001250#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 if (_Py_atomic_load_relaxed(&pendingcalls_to_do)) {
1252 if (Py_MakePendingCalls() < 0) {
1253 why = WHY_EXCEPTION;
1254 goto on_error;
1255 }
1256 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001257#ifdef WITH_THREAD
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001258 if (_Py_atomic_load_relaxed(&gil_drop_request)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 /* Give another thread a chance */
1260 if (PyThreadState_Swap(NULL) != tstate)
1261 Py_FatalError("ceval: tstate mix-up");
1262 drop_gil(tstate);
1263
1264 /* Other threads may run now */
1265
1266 take_gil(tstate);
1267 if (PyThreadState_Swap(tstate) != NULL)
1268 Py_FatalError("ceval: orphan tstate");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001269 }
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001270#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001271 /* Check for asynchronous exceptions. */
1272 if (tstate->async_exc != NULL) {
1273 x = tstate->async_exc;
1274 tstate->async_exc = NULL;
1275 UNSIGNAL_ASYNC_EXC();
1276 PyErr_SetNone(x);
1277 Py_DECREF(x);
1278 why = WHY_EXCEPTION;
1279 goto on_error;
1280 }
1281 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 fast_next_opcode:
1284 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 /* line-by-line tracing support */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001288 if (_Py_TracingPossible &&
1289 tstate->c_tracefunc != NULL && !tstate->tracing) {
1290 /* see maybe_call_line_trace
1291 for expository comments */
1292 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001294 err = maybe_call_line_trace(tstate->c_tracefunc,
1295 tstate->c_traceobj,
1296 f, &instr_lb, &instr_ub,
1297 &instr_prev);
1298 /* Reload possibly changed frame fields */
1299 JUMPTO(f->f_lasti);
1300 if (f->f_stacktop != NULL) {
1301 stack_pointer = f->f_stacktop;
1302 f->f_stacktop = NULL;
1303 }
1304 if (err) {
1305 /* trace function raised an exception */
1306 goto on_error;
1307 }
1308 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001309
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001310 /* Extract opcode and argument */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001311
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001312 opcode = NEXTOP();
1313 oparg = 0; /* allows oparg to be stored in a register because
1314 it doesn't have to be remembered across a full loop */
1315 if (HAS_ARG(opcode))
1316 oparg = NEXTARG();
Stefan Krahb7e10102010-06-23 18:42:39 +00001317 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001318#ifdef DYNAMIC_EXECUTION_PROFILE
1319#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001320 dxpairs[lastopcode][opcode]++;
1321 lastopcode = opcode;
Guido van Rossum950361c1997-01-24 13:49:28 +00001322#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001323 dxp[opcode]++;
Guido van Rossum950361c1997-01-24 13:49:28 +00001324#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001325
Guido van Rossum96a42c81992-01-12 02:29:51 +00001326#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001327 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001329 if (lltrace) {
1330 if (HAS_ARG(opcode)) {
1331 printf("%d: %d, %d\n",
1332 f->f_lasti, opcode, oparg);
1333 }
1334 else {
1335 printf("%d: %d\n",
1336 f->f_lasti, opcode);
1337 }
1338 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001339#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001340
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001341 /* Main switch on opcode */
1342 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001343
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001344 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001345
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001346 /* BEWARE!
1347 It is essential that any operation that fails sets either
1348 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1349 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 TARGET(NOP)
1352 FAST_DISPATCH();
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001354 TARGET(LOAD_FAST)
1355 x = GETLOCAL(oparg);
1356 if (x != NULL) {
1357 Py_INCREF(x);
1358 PUSH(x);
1359 FAST_DISPATCH();
1360 }
1361 format_exc_check_arg(PyExc_UnboundLocalError,
1362 UNBOUNDLOCAL_ERROR_MSG,
1363 PyTuple_GetItem(co->co_varnames, oparg));
1364 break;
Neil Schemenauer63543862002-02-17 19:10:14 +00001365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 TARGET(LOAD_CONST)
1367 x = GETITEM(consts, oparg);
1368 Py_INCREF(x);
1369 PUSH(x);
1370 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001371
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 PREDICTED_WITH_ARG(STORE_FAST);
1373 TARGET(STORE_FAST)
1374 v = POP();
1375 SETLOCAL(oparg, v);
1376 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 TARGET(POP_TOP)
1379 v = POP();
1380 Py_DECREF(v);
1381 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001382
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 TARGET(ROT_TWO)
1384 v = TOP();
1385 w = SECOND();
1386 SET_TOP(w);
1387 SET_SECOND(v);
1388 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001390 TARGET(ROT_THREE)
1391 v = TOP();
1392 w = SECOND();
1393 x = THIRD();
1394 SET_TOP(w);
1395 SET_SECOND(x);
1396 SET_THIRD(v);
1397 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001398
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001399 TARGET(DUP_TOP)
1400 v = TOP();
1401 Py_INCREF(v);
1402 PUSH(v);
1403 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001404
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001405 TARGET(DUP_TOP_TWO)
1406 x = TOP();
1407 Py_INCREF(x);
1408 w = SECOND();
1409 Py_INCREF(w);
1410 STACKADJ(2);
1411 SET_TOP(x);
1412 SET_SECOND(w);
1413 FAST_DISPATCH();
Thomas Wouters434d0822000-08-24 20:11:32 +00001414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001415 TARGET(UNARY_POSITIVE)
1416 v = TOP();
1417 x = PyNumber_Positive(v);
1418 Py_DECREF(v);
1419 SET_TOP(x);
1420 if (x != NULL) DISPATCH();
1421 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001422
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 TARGET(UNARY_NEGATIVE)
1424 v = TOP();
1425 x = PyNumber_Negative(v);
1426 Py_DECREF(v);
1427 SET_TOP(x);
1428 if (x != NULL) DISPATCH();
1429 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001431 TARGET(UNARY_NOT)
1432 v = TOP();
1433 err = PyObject_IsTrue(v);
1434 Py_DECREF(v);
1435 if (err == 0) {
1436 Py_INCREF(Py_True);
1437 SET_TOP(Py_True);
1438 DISPATCH();
1439 }
1440 else if (err > 0) {
1441 Py_INCREF(Py_False);
1442 SET_TOP(Py_False);
1443 err = 0;
1444 DISPATCH();
1445 }
1446 STACKADJ(-1);
1447 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001448
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001449 TARGET(UNARY_INVERT)
1450 v = TOP();
1451 x = PyNumber_Invert(v);
1452 Py_DECREF(v);
1453 SET_TOP(x);
1454 if (x != NULL) DISPATCH();
1455 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001457 TARGET(BINARY_POWER)
1458 w = POP();
1459 v = TOP();
1460 x = PyNumber_Power(v, w, Py_None);
1461 Py_DECREF(v);
1462 Py_DECREF(w);
1463 SET_TOP(x);
1464 if (x != NULL) DISPATCH();
1465 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001467 TARGET(BINARY_MULTIPLY)
1468 w = POP();
1469 v = TOP();
1470 x = PyNumber_Multiply(v, w);
1471 Py_DECREF(v);
1472 Py_DECREF(w);
1473 SET_TOP(x);
1474 if (x != NULL) DISPATCH();
1475 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001477 TARGET(BINARY_TRUE_DIVIDE)
1478 w = POP();
1479 v = TOP();
1480 x = PyNumber_TrueDivide(v, w);
1481 Py_DECREF(v);
1482 Py_DECREF(w);
1483 SET_TOP(x);
1484 if (x != NULL) DISPATCH();
1485 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001486
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001487 TARGET(BINARY_FLOOR_DIVIDE)
1488 w = POP();
1489 v = TOP();
1490 x = PyNumber_FloorDivide(v, w);
1491 Py_DECREF(v);
1492 Py_DECREF(w);
1493 SET_TOP(x);
1494 if (x != NULL) DISPATCH();
1495 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001496
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001497 TARGET(BINARY_MODULO)
1498 w = POP();
1499 v = TOP();
1500 if (PyUnicode_CheckExact(v))
1501 x = PyUnicode_Format(v, w);
1502 else
1503 x = PyNumber_Remainder(v, w);
1504 Py_DECREF(v);
1505 Py_DECREF(w);
1506 SET_TOP(x);
1507 if (x != NULL) DISPATCH();
1508 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001510 TARGET(BINARY_ADD)
1511 w = POP();
1512 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001513 if (PyUnicode_CheckExact(v) &&
1514 PyUnicode_CheckExact(w)) {
1515 x = unicode_concatenate(v, w, f, next_instr);
1516 /* unicode_concatenate consumed the ref to v */
1517 goto skip_decref_vx;
1518 }
1519 else {
1520 x = PyNumber_Add(v, w);
1521 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001522 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001523 skip_decref_vx:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 Py_DECREF(w);
1525 SET_TOP(x);
1526 if (x != NULL) DISPATCH();
1527 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001528
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001529 TARGET(BINARY_SUBTRACT)
1530 w = POP();
1531 v = TOP();
1532 x = PyNumber_Subtract(v, w);
1533 Py_DECREF(v);
1534 Py_DECREF(w);
1535 SET_TOP(x);
1536 if (x != NULL) DISPATCH();
1537 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001539 TARGET(BINARY_SUBSCR)
1540 w = POP();
1541 v = TOP();
1542 x = PyObject_GetItem(v, w);
1543 Py_DECREF(v);
1544 Py_DECREF(w);
1545 SET_TOP(x);
1546 if (x != NULL) DISPATCH();
1547 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001548
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001549 TARGET(BINARY_LSHIFT)
1550 w = POP();
1551 v = TOP();
1552 x = PyNumber_Lshift(v, w);
1553 Py_DECREF(v);
1554 Py_DECREF(w);
1555 SET_TOP(x);
1556 if (x != NULL) DISPATCH();
1557 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001559 TARGET(BINARY_RSHIFT)
1560 w = POP();
1561 v = TOP();
1562 x = PyNumber_Rshift(v, w);
1563 Py_DECREF(v);
1564 Py_DECREF(w);
1565 SET_TOP(x);
1566 if (x != NULL) DISPATCH();
1567 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001569 TARGET(BINARY_AND)
1570 w = POP();
1571 v = TOP();
1572 x = PyNumber_And(v, w);
1573 Py_DECREF(v);
1574 Py_DECREF(w);
1575 SET_TOP(x);
1576 if (x != NULL) DISPATCH();
1577 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001579 TARGET(BINARY_XOR)
1580 w = POP();
1581 v = TOP();
1582 x = PyNumber_Xor(v, w);
1583 Py_DECREF(v);
1584 Py_DECREF(w);
1585 SET_TOP(x);
1586 if (x != NULL) DISPATCH();
1587 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001589 TARGET(BINARY_OR)
1590 w = POP();
1591 v = TOP();
1592 x = PyNumber_Or(v, w);
1593 Py_DECREF(v);
1594 Py_DECREF(w);
1595 SET_TOP(x);
1596 if (x != NULL) DISPATCH();
1597 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001599 TARGET(LIST_APPEND)
1600 w = POP();
1601 v = PEEK(oparg);
1602 err = PyList_Append(v, w);
1603 Py_DECREF(w);
1604 if (err == 0) {
1605 PREDICT(JUMP_ABSOLUTE);
1606 DISPATCH();
1607 }
1608 break;
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001609
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001610 TARGET(SET_ADD)
1611 w = POP();
1612 v = stack_pointer[-oparg];
1613 err = PySet_Add(v, w);
1614 Py_DECREF(w);
1615 if (err == 0) {
1616 PREDICT(JUMP_ABSOLUTE);
1617 DISPATCH();
1618 }
1619 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00001620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001621 TARGET(INPLACE_POWER)
1622 w = POP();
1623 v = TOP();
1624 x = PyNumber_InPlacePower(v, w, Py_None);
1625 Py_DECREF(v);
1626 Py_DECREF(w);
1627 SET_TOP(x);
1628 if (x != NULL) DISPATCH();
1629 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001630
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001631 TARGET(INPLACE_MULTIPLY)
1632 w = POP();
1633 v = TOP();
1634 x = PyNumber_InPlaceMultiply(v, w);
1635 Py_DECREF(v);
1636 Py_DECREF(w);
1637 SET_TOP(x);
1638 if (x != NULL) DISPATCH();
1639 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001640
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001641 TARGET(INPLACE_TRUE_DIVIDE)
1642 w = POP();
1643 v = TOP();
1644 x = PyNumber_InPlaceTrueDivide(v, w);
1645 Py_DECREF(v);
1646 Py_DECREF(w);
1647 SET_TOP(x);
1648 if (x != NULL) DISPATCH();
1649 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 TARGET(INPLACE_FLOOR_DIVIDE)
1652 w = POP();
1653 v = TOP();
1654 x = PyNumber_InPlaceFloorDivide(v, w);
1655 Py_DECREF(v);
1656 Py_DECREF(w);
1657 SET_TOP(x);
1658 if (x != NULL) DISPATCH();
1659 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001661 TARGET(INPLACE_MODULO)
1662 w = POP();
1663 v = TOP();
1664 x = PyNumber_InPlaceRemainder(v, w);
1665 Py_DECREF(v);
1666 Py_DECREF(w);
1667 SET_TOP(x);
1668 if (x != NULL) DISPATCH();
1669 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001670
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001671 TARGET(INPLACE_ADD)
1672 w = POP();
1673 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001674 if (PyUnicode_CheckExact(v) &&
1675 PyUnicode_CheckExact(w)) {
1676 x = unicode_concatenate(v, w, f, next_instr);
1677 /* unicode_concatenate consumed the ref to v */
1678 goto skip_decref_v;
1679 }
1680 else {
1681 x = PyNumber_InPlaceAdd(v, w);
1682 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001683 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001684 skip_decref_v:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001685 Py_DECREF(w);
1686 SET_TOP(x);
1687 if (x != NULL) DISPATCH();
1688 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 TARGET(INPLACE_SUBTRACT)
1691 w = POP();
1692 v = TOP();
1693 x = PyNumber_InPlaceSubtract(v, w);
1694 Py_DECREF(v);
1695 Py_DECREF(w);
1696 SET_TOP(x);
1697 if (x != NULL) DISPATCH();
1698 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001699
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001700 TARGET(INPLACE_LSHIFT)
1701 w = POP();
1702 v = TOP();
1703 x = PyNumber_InPlaceLshift(v, w);
1704 Py_DECREF(v);
1705 Py_DECREF(w);
1706 SET_TOP(x);
1707 if (x != NULL) DISPATCH();
1708 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001710 TARGET(INPLACE_RSHIFT)
1711 w = POP();
1712 v = TOP();
1713 x = PyNumber_InPlaceRshift(v, w);
1714 Py_DECREF(v);
1715 Py_DECREF(w);
1716 SET_TOP(x);
1717 if (x != NULL) DISPATCH();
1718 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001719
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001720 TARGET(INPLACE_AND)
1721 w = POP();
1722 v = TOP();
1723 x = PyNumber_InPlaceAnd(v, w);
1724 Py_DECREF(v);
1725 Py_DECREF(w);
1726 SET_TOP(x);
1727 if (x != NULL) DISPATCH();
1728 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001729
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001730 TARGET(INPLACE_XOR)
1731 w = POP();
1732 v = TOP();
1733 x = PyNumber_InPlaceXor(v, w);
1734 Py_DECREF(v);
1735 Py_DECREF(w);
1736 SET_TOP(x);
1737 if (x != NULL) DISPATCH();
1738 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001739
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001740 TARGET(INPLACE_OR)
1741 w = POP();
1742 v = TOP();
1743 x = PyNumber_InPlaceOr(v, w);
1744 Py_DECREF(v);
1745 Py_DECREF(w);
1746 SET_TOP(x);
1747 if (x != NULL) DISPATCH();
1748 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001749
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001750 TARGET(STORE_SUBSCR)
1751 w = TOP();
1752 v = SECOND();
1753 u = THIRD();
1754 STACKADJ(-3);
1755 /* v[w] = u */
1756 err = PyObject_SetItem(v, w, u);
1757 Py_DECREF(u);
1758 Py_DECREF(v);
1759 Py_DECREF(w);
1760 if (err == 0) DISPATCH();
1761 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001762
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001763 TARGET(DELETE_SUBSCR)
1764 w = TOP();
1765 v = SECOND();
1766 STACKADJ(-2);
1767 /* del v[w] */
1768 err = PyObject_DelItem(v, w);
1769 Py_DECREF(v);
1770 Py_DECREF(w);
1771 if (err == 0) DISPATCH();
1772 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 TARGET(PRINT_EXPR)
1775 v = POP();
1776 w = PySys_GetObject("displayhook");
1777 if (w == NULL) {
1778 PyErr_SetString(PyExc_RuntimeError,
1779 "lost sys.displayhook");
1780 err = -1;
1781 x = NULL;
1782 }
1783 if (err == 0) {
1784 x = PyTuple_Pack(1, v);
1785 if (x == NULL)
1786 err = -1;
1787 }
1788 if (err == 0) {
1789 w = PyEval_CallObject(w, x);
1790 Py_XDECREF(w);
1791 if (w == NULL)
1792 err = -1;
1793 }
1794 Py_DECREF(v);
1795 Py_XDECREF(x);
1796 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001797
Thomas Wouters434d0822000-08-24 20:11:32 +00001798#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001799 default: switch (opcode) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001800#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001801 TARGET(RAISE_VARARGS)
1802 v = w = NULL;
1803 switch (oparg) {
1804 case 2:
1805 v = POP(); /* cause */
1806 case 1:
1807 w = POP(); /* exc */
1808 case 0: /* Fallthrough */
1809 why = do_raise(w, v);
1810 break;
1811 default:
1812 PyErr_SetString(PyExc_SystemError,
1813 "bad RAISE_VARARGS oparg");
1814 why = WHY_EXCEPTION;
1815 break;
1816 }
1817 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001819 TARGET(STORE_LOCALS)
1820 x = POP();
1821 v = f->f_locals;
1822 Py_XDECREF(v);
1823 f->f_locals = x;
1824 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001825
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001826 TARGET(RETURN_VALUE)
1827 retval = POP();
1828 why = WHY_RETURN;
1829 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001830
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001831 TARGET(YIELD_FROM)
1832 u = POP();
1833 x = PyObject_GetIter(u);
1834 Py_DECREF(u);
1835 if (x == NULL)
1836 break;
1837 /* x is now the iterator, make the first next() call */
1838 retval = (*Py_TYPE(x)->tp_iternext)(x);
1839 if (!retval) {
1840 /* iter may be exhausted */
1841 Py_CLEAR(x);
1842 if (!PyErr_ExceptionMatches(PyExc_StopIteration)) {
1843 /* some other exception */
1844 break;
1845 }
1846 /* try to get return value from exception */
1847 PyObject *et, *ev, *tb;
1848 PyErr_Fetch(&et, &ev, &tb);
1849 Py_XDECREF(et);
1850 Py_XDECREF(tb);
1851 /* u is return value */
1852 u = NULL;
1853 if (ev) {
1854 u = PyObject_GetAttrString(ev, "value");
1855 Py_DECREF(ev);
1856 if (u == NULL) {
1857 if (!PyErr_ExceptionMatches(PyExc_AttributeError)) {
1858 /* some other exception */
1859 break;
1860 }
1861 PyErr_Clear();
1862 }
1863 }
1864 if (u == NULL) {
1865 u = Py_None;
1866 Py_INCREF(u);
1867 }
1868 PUSH(u);
1869 continue;
1870 }
1871 /* x is iterator, retval is value to be yielded */
1872 f->f_yieldfrom = x;
1873 f->f_stacktop = stack_pointer;
1874 why = WHY_YIELD;
1875 goto fast_yield;
1876
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 TARGET(YIELD_VALUE)
1878 retval = POP();
1879 f->f_stacktop = stack_pointer;
1880 why = WHY_YIELD;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001881 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001882
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001883 TARGET(POP_EXCEPT)
1884 {
1885 PyTryBlock *b = PyFrame_BlockPop(f);
1886 if (b->b_type != EXCEPT_HANDLER) {
1887 PyErr_SetString(PyExc_SystemError,
1888 "popped block is not an except handler");
1889 why = WHY_EXCEPTION;
1890 break;
1891 }
1892 UNWIND_EXCEPT_HANDLER(b);
1893 }
1894 DISPATCH();
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001895
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001896 TARGET(POP_BLOCK)
1897 {
1898 PyTryBlock *b = PyFrame_BlockPop(f);
1899 UNWIND_BLOCK(b);
1900 }
1901 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001903 PREDICTED(END_FINALLY);
1904 TARGET(END_FINALLY)
1905 v = POP();
1906 if (PyLong_Check(v)) {
1907 why = (enum why_code) PyLong_AS_LONG(v);
1908 assert(why != WHY_YIELD);
1909 if (why == WHY_RETURN ||
1910 why == WHY_CONTINUE)
1911 retval = POP();
1912 if (why == WHY_SILENCED) {
1913 /* An exception was silenced by 'with', we must
1914 manually unwind the EXCEPT_HANDLER block which was
1915 created when the exception was caught, otherwise
1916 the stack will be in an inconsistent state. */
1917 PyTryBlock *b = PyFrame_BlockPop(f);
1918 assert(b->b_type == EXCEPT_HANDLER);
1919 UNWIND_EXCEPT_HANDLER(b);
1920 why = WHY_NOT;
1921 }
1922 }
1923 else if (PyExceptionClass_Check(v)) {
1924 w = POP();
1925 u = POP();
1926 PyErr_Restore(v, w, u);
1927 why = WHY_RERAISE;
1928 break;
1929 }
1930 else if (v != Py_None) {
1931 PyErr_SetString(PyExc_SystemError,
1932 "'finally' pops bad exception");
1933 why = WHY_EXCEPTION;
1934 }
1935 Py_DECREF(v);
1936 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001937
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001938 TARGET(LOAD_BUILD_CLASS)
1939 x = PyDict_GetItemString(f->f_builtins,
1940 "__build_class__");
1941 if (x == NULL) {
1942 PyErr_SetString(PyExc_ImportError,
1943 "__build_class__ not found");
1944 break;
1945 }
1946 Py_INCREF(x);
1947 PUSH(x);
1948 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001949
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001950 TARGET(STORE_NAME)
1951 w = GETITEM(names, oparg);
1952 v = POP();
1953 if ((x = f->f_locals) != NULL) {
1954 if (PyDict_CheckExact(x))
1955 err = PyDict_SetItem(x, w, v);
1956 else
1957 err = PyObject_SetItem(x, w, v);
1958 Py_DECREF(v);
1959 if (err == 0) DISPATCH();
1960 break;
1961 }
1962 PyErr_Format(PyExc_SystemError,
1963 "no locals found when storing %R", w);
1964 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001966 TARGET(DELETE_NAME)
1967 w = GETITEM(names, oparg);
1968 if ((x = f->f_locals) != NULL) {
1969 if ((err = PyObject_DelItem(x, w)) != 0)
1970 format_exc_check_arg(PyExc_NameError,
1971 NAME_ERROR_MSG,
1972 w);
1973 break;
1974 }
1975 PyErr_Format(PyExc_SystemError,
1976 "no locals when deleting %R", w);
1977 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001978
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001979 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
1980 TARGET(UNPACK_SEQUENCE)
1981 v = POP();
1982 if (PyTuple_CheckExact(v) &&
1983 PyTuple_GET_SIZE(v) == oparg) {
1984 PyObject **items = \
1985 ((PyTupleObject *)v)->ob_item;
1986 while (oparg--) {
1987 w = items[oparg];
1988 Py_INCREF(w);
1989 PUSH(w);
1990 }
1991 Py_DECREF(v);
1992 DISPATCH();
1993 } else if (PyList_CheckExact(v) &&
1994 PyList_GET_SIZE(v) == oparg) {
1995 PyObject **items = \
1996 ((PyListObject *)v)->ob_item;
1997 while (oparg--) {
1998 w = items[oparg];
1999 Py_INCREF(w);
2000 PUSH(w);
2001 }
2002 } else if (unpack_iterable(v, oparg, -1,
2003 stack_pointer + oparg)) {
2004 STACKADJ(oparg);
2005 } else {
2006 /* unpack_iterable() raised an exception */
2007 why = WHY_EXCEPTION;
2008 }
2009 Py_DECREF(v);
2010 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002011
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002012 TARGET(UNPACK_EX)
2013 {
2014 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
2015 v = POP();
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002016
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002017 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
2018 stack_pointer + totalargs)) {
2019 stack_pointer += totalargs;
2020 } else {
2021 why = WHY_EXCEPTION;
2022 }
2023 Py_DECREF(v);
2024 break;
2025 }
Guido van Rossum0368b722007-05-11 16:50:42 +00002026
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002027 TARGET(STORE_ATTR)
2028 w = GETITEM(names, oparg);
2029 v = TOP();
2030 u = SECOND();
2031 STACKADJ(-2);
2032 err = PyObject_SetAttr(v, w, u); /* v.w = u */
2033 Py_DECREF(v);
2034 Py_DECREF(u);
2035 if (err == 0) DISPATCH();
2036 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002037
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002038 TARGET(DELETE_ATTR)
2039 w = GETITEM(names, oparg);
2040 v = POP();
2041 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
2042 /* del v.w */
2043 Py_DECREF(v);
2044 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002045
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002046 TARGET(STORE_GLOBAL)
2047 w = GETITEM(names, oparg);
2048 v = POP();
2049 err = PyDict_SetItem(f->f_globals, w, v);
2050 Py_DECREF(v);
2051 if (err == 0) DISPATCH();
2052 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002053
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002054 TARGET(DELETE_GLOBAL)
2055 w = GETITEM(names, oparg);
2056 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
2057 format_exc_check_arg(
2058 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
2059 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002060
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002061 TARGET(LOAD_NAME)
2062 w = GETITEM(names, oparg);
2063 if ((v = f->f_locals) == NULL) {
2064 PyErr_Format(PyExc_SystemError,
2065 "no locals when loading %R", w);
2066 why = WHY_EXCEPTION;
2067 break;
2068 }
2069 if (PyDict_CheckExact(v)) {
2070 x = PyDict_GetItem(v, w);
2071 Py_XINCREF(x);
2072 }
2073 else {
2074 x = PyObject_GetItem(v, w);
2075 if (x == NULL && PyErr_Occurred()) {
2076 if (!PyErr_ExceptionMatches(
2077 PyExc_KeyError))
2078 break;
2079 PyErr_Clear();
2080 }
2081 }
2082 if (x == NULL) {
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002083 x = PyDict_GetItem(f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002084 if (x == NULL) {
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002085 x = PyDict_GetItem(f->f_builtins, w);
2086 if (x == NULL) {
2087 format_exc_check_arg(
2088 PyExc_NameError,
2089 NAME_ERROR_MSG, w);
2090 break;
2091 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002092 }
2093 Py_INCREF(x);
2094 }
2095 PUSH(x);
2096 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002097
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002098 TARGET(LOAD_GLOBAL)
2099 w = GETITEM(names, oparg);
2100 if (PyUnicode_CheckExact(w)) {
2101 /* Inline the PyDict_GetItem() calls.
2102 WARNING: this is an extreme speed hack.
2103 Do not try this at home. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002104 Py_hash_t hash = ((PyASCIIObject *)w)->hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002105 if (hash != -1) {
2106 PyDictObject *d;
2107 PyDictEntry *e;
2108 d = (PyDictObject *)(f->f_globals);
2109 e = d->ma_lookup(d, w, hash);
2110 if (e == NULL) {
2111 x = NULL;
2112 break;
2113 }
2114 x = e->me_value;
2115 if (x != NULL) {
2116 Py_INCREF(x);
2117 PUSH(x);
2118 DISPATCH();
2119 }
2120 d = (PyDictObject *)(f->f_builtins);
2121 e = d->ma_lookup(d, w, hash);
2122 if (e == NULL) {
2123 x = NULL;
2124 break;
2125 }
2126 x = e->me_value;
2127 if (x != NULL) {
2128 Py_INCREF(x);
2129 PUSH(x);
2130 DISPATCH();
2131 }
2132 goto load_global_error;
2133 }
2134 }
2135 /* This is the un-inlined version of the code above */
2136 x = PyDict_GetItem(f->f_globals, w);
2137 if (x == NULL) {
2138 x = PyDict_GetItem(f->f_builtins, w);
2139 if (x == NULL) {
2140 load_global_error:
2141 format_exc_check_arg(
2142 PyExc_NameError,
2143 GLOBAL_NAME_ERROR_MSG, w);
2144 break;
2145 }
2146 }
2147 Py_INCREF(x);
2148 PUSH(x);
2149 DISPATCH();
Guido van Rossum681d79a1995-07-18 14:51:37 +00002150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002151 TARGET(DELETE_FAST)
2152 x = GETLOCAL(oparg);
2153 if (x != NULL) {
2154 SETLOCAL(oparg, NULL);
2155 DISPATCH();
2156 }
2157 format_exc_check_arg(
2158 PyExc_UnboundLocalError,
2159 UNBOUNDLOCAL_ERROR_MSG,
2160 PyTuple_GetItem(co->co_varnames, oparg)
2161 );
2162 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002163
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002164 TARGET(DELETE_DEREF)
2165 x = freevars[oparg];
2166 if (PyCell_GET(x) != NULL) {
2167 PyCell_Set(x, NULL);
Benjamin Peterson00ebe2c2010-09-10 22:02:31 +00002168 DISPATCH();
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002169 }
2170 err = -1;
2171 format_exc_unbound(co, oparg);
2172 break;
2173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002174 TARGET(LOAD_CLOSURE)
2175 x = freevars[oparg];
2176 Py_INCREF(x);
2177 PUSH(x);
2178 if (x != NULL) DISPATCH();
2179 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002180
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002181 TARGET(LOAD_DEREF)
2182 x = freevars[oparg];
2183 w = PyCell_Get(x);
2184 if (w != NULL) {
2185 PUSH(w);
2186 DISPATCH();
2187 }
2188 err = -1;
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002189 format_exc_unbound(co, oparg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002190 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002191
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002192 TARGET(STORE_DEREF)
2193 w = POP();
2194 x = freevars[oparg];
2195 PyCell_Set(x, w);
2196 Py_DECREF(w);
2197 DISPATCH();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002199 TARGET(BUILD_TUPLE)
2200 x = PyTuple_New(oparg);
2201 if (x != NULL) {
2202 for (; --oparg >= 0;) {
2203 w = POP();
2204 PyTuple_SET_ITEM(x, oparg, w);
2205 }
2206 PUSH(x);
2207 DISPATCH();
2208 }
2209 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002211 TARGET(BUILD_LIST)
2212 x = PyList_New(oparg);
2213 if (x != NULL) {
2214 for (; --oparg >= 0;) {
2215 w = POP();
2216 PyList_SET_ITEM(x, oparg, w);
2217 }
2218 PUSH(x);
2219 DISPATCH();
2220 }
2221 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002222
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002223 TARGET(BUILD_SET)
2224 x = PySet_New(NULL);
2225 if (x != NULL) {
2226 for (; --oparg >= 0;) {
2227 w = POP();
2228 if (err == 0)
2229 err = PySet_Add(x, w);
2230 Py_DECREF(w);
2231 }
2232 if (err != 0) {
2233 Py_DECREF(x);
2234 break;
2235 }
2236 PUSH(x);
2237 DISPATCH();
2238 }
2239 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002241 TARGET(BUILD_MAP)
2242 x = _PyDict_NewPresized((Py_ssize_t)oparg);
2243 PUSH(x);
2244 if (x != NULL) DISPATCH();
2245 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002246
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002247 TARGET(STORE_MAP)
2248 w = TOP(); /* key */
2249 u = SECOND(); /* value */
2250 v = THIRD(); /* dict */
2251 STACKADJ(-2);
2252 assert (PyDict_CheckExact(v));
2253 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2254 Py_DECREF(u);
2255 Py_DECREF(w);
2256 if (err == 0) DISPATCH();
2257 break;
Christian Heimes99170a52007-12-19 02:07:34 +00002258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002259 TARGET(MAP_ADD)
2260 w = TOP(); /* key */
2261 u = SECOND(); /* value */
2262 STACKADJ(-2);
2263 v = stack_pointer[-oparg]; /* dict */
2264 assert (PyDict_CheckExact(v));
2265 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2266 Py_DECREF(u);
2267 Py_DECREF(w);
2268 if (err == 0) {
2269 PREDICT(JUMP_ABSOLUTE);
2270 DISPATCH();
2271 }
2272 break;
Antoine Pitrouf289ae62008-12-18 11:06:25 +00002273
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002274 TARGET(LOAD_ATTR)
2275 w = GETITEM(names, oparg);
2276 v = TOP();
2277 x = PyObject_GetAttr(v, w);
2278 Py_DECREF(v);
2279 SET_TOP(x);
2280 if (x != NULL) DISPATCH();
2281 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002283 TARGET(COMPARE_OP)
2284 w = POP();
2285 v = TOP();
2286 x = cmp_outcome(oparg, v, w);
2287 Py_DECREF(v);
2288 Py_DECREF(w);
2289 SET_TOP(x);
2290 if (x == NULL) break;
2291 PREDICT(POP_JUMP_IF_FALSE);
2292 PREDICT(POP_JUMP_IF_TRUE);
2293 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002294
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002295 TARGET(IMPORT_NAME)
2296 w = GETITEM(names, oparg);
2297 x = PyDict_GetItemString(f->f_builtins, "__import__");
2298 if (x == NULL) {
2299 PyErr_SetString(PyExc_ImportError,
2300 "__import__ not found");
2301 break;
2302 }
2303 Py_INCREF(x);
2304 v = POP();
2305 u = TOP();
2306 if (PyLong_AsLong(u) != -1 || PyErr_Occurred())
2307 w = PyTuple_Pack(5,
2308 w,
2309 f->f_globals,
2310 f->f_locals == NULL ?
2311 Py_None : f->f_locals,
2312 v,
2313 u);
2314 else
2315 w = PyTuple_Pack(4,
2316 w,
2317 f->f_globals,
2318 f->f_locals == NULL ?
2319 Py_None : f->f_locals,
2320 v);
2321 Py_DECREF(v);
2322 Py_DECREF(u);
2323 if (w == NULL) {
2324 u = POP();
2325 Py_DECREF(x);
2326 x = NULL;
2327 break;
2328 }
2329 READ_TIMESTAMP(intr0);
2330 v = x;
2331 x = PyEval_CallObject(v, w);
2332 Py_DECREF(v);
2333 READ_TIMESTAMP(intr1);
2334 Py_DECREF(w);
2335 SET_TOP(x);
2336 if (x != NULL) DISPATCH();
2337 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002338
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002339 TARGET(IMPORT_STAR)
2340 v = POP();
2341 PyFrame_FastToLocals(f);
2342 if ((x = f->f_locals) == NULL) {
2343 PyErr_SetString(PyExc_SystemError,
2344 "no locals found during 'import *'");
2345 break;
2346 }
2347 READ_TIMESTAMP(intr0);
2348 err = import_all_from(x, v);
2349 READ_TIMESTAMP(intr1);
2350 PyFrame_LocalsToFast(f, 0);
2351 Py_DECREF(v);
2352 if (err == 0) DISPATCH();
2353 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002354
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002355 TARGET(IMPORT_FROM)
2356 w = GETITEM(names, oparg);
2357 v = TOP();
2358 READ_TIMESTAMP(intr0);
2359 x = import_from(v, w);
2360 READ_TIMESTAMP(intr1);
2361 PUSH(x);
2362 if (x != NULL) DISPATCH();
2363 break;
Thomas Wouters52152252000-08-17 22:55:00 +00002364
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002365 TARGET(JUMP_FORWARD)
2366 JUMPBY(oparg);
2367 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002368
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002369 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
2370 TARGET(POP_JUMP_IF_FALSE)
2371 w = POP();
2372 if (w == Py_True) {
2373 Py_DECREF(w);
2374 FAST_DISPATCH();
2375 }
2376 if (w == Py_False) {
2377 Py_DECREF(w);
2378 JUMPTO(oparg);
2379 FAST_DISPATCH();
2380 }
2381 err = PyObject_IsTrue(w);
2382 Py_DECREF(w);
2383 if (err > 0)
2384 err = 0;
2385 else if (err == 0)
2386 JUMPTO(oparg);
2387 else
2388 break;
2389 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002390
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002391 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
2392 TARGET(POP_JUMP_IF_TRUE)
2393 w = POP();
2394 if (w == Py_False) {
2395 Py_DECREF(w);
2396 FAST_DISPATCH();
2397 }
2398 if (w == Py_True) {
2399 Py_DECREF(w);
2400 JUMPTO(oparg);
2401 FAST_DISPATCH();
2402 }
2403 err = PyObject_IsTrue(w);
2404 Py_DECREF(w);
2405 if (err > 0) {
2406 err = 0;
2407 JUMPTO(oparg);
2408 }
2409 else if (err == 0)
2410 ;
2411 else
2412 break;
2413 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002415 TARGET(JUMP_IF_FALSE_OR_POP)
2416 w = TOP();
2417 if (w == Py_True) {
2418 STACKADJ(-1);
2419 Py_DECREF(w);
2420 FAST_DISPATCH();
2421 }
2422 if (w == Py_False) {
2423 JUMPTO(oparg);
2424 FAST_DISPATCH();
2425 }
2426 err = PyObject_IsTrue(w);
2427 if (err > 0) {
2428 STACKADJ(-1);
2429 Py_DECREF(w);
2430 err = 0;
2431 }
2432 else if (err == 0)
2433 JUMPTO(oparg);
2434 else
2435 break;
2436 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002437
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002438 TARGET(JUMP_IF_TRUE_OR_POP)
2439 w = TOP();
2440 if (w == Py_False) {
2441 STACKADJ(-1);
2442 Py_DECREF(w);
2443 FAST_DISPATCH();
2444 }
2445 if (w == Py_True) {
2446 JUMPTO(oparg);
2447 FAST_DISPATCH();
2448 }
2449 err = PyObject_IsTrue(w);
2450 if (err > 0) {
2451 err = 0;
2452 JUMPTO(oparg);
2453 }
2454 else if (err == 0) {
2455 STACKADJ(-1);
2456 Py_DECREF(w);
2457 }
2458 else
2459 break;
2460 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002461
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002462 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
2463 TARGET(JUMP_ABSOLUTE)
2464 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002465#if FAST_LOOPS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002466 /* Enabling this path speeds-up all while and for-loops by bypassing
2467 the per-loop checks for signals. By default, this should be turned-off
2468 because it prevents detection of a control-break in tight loops like
2469 "while 1: pass". Compile with this option turned-on when you need
2470 the speed-up and do not need break checking inside tight loops (ones
2471 that contain only instructions ending with FAST_DISPATCH).
2472 */
2473 FAST_DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002474#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002475 DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002476#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002478 TARGET(GET_ITER)
2479 /* before: [obj]; after [getiter(obj)] */
2480 v = TOP();
2481 x = PyObject_GetIter(v);
2482 Py_DECREF(v);
2483 if (x != NULL) {
2484 SET_TOP(x);
2485 PREDICT(FOR_ITER);
2486 DISPATCH();
2487 }
2488 STACKADJ(-1);
2489 break;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002490
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002491 PREDICTED_WITH_ARG(FOR_ITER);
2492 TARGET(FOR_ITER)
2493 /* before: [iter]; after: [iter, iter()] *or* [] */
2494 v = TOP();
2495 x = (*v->ob_type->tp_iternext)(v);
2496 if (x != NULL) {
2497 PUSH(x);
2498 PREDICT(STORE_FAST);
2499 PREDICT(UNPACK_SEQUENCE);
2500 DISPATCH();
2501 }
2502 if (PyErr_Occurred()) {
2503 if (!PyErr_ExceptionMatches(
2504 PyExc_StopIteration))
2505 break;
2506 PyErr_Clear();
2507 }
2508 /* iterator ended normally */
2509 x = v = POP();
2510 Py_DECREF(v);
2511 JUMPBY(oparg);
2512 DISPATCH();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002514 TARGET(BREAK_LOOP)
2515 why = WHY_BREAK;
2516 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002518 TARGET(CONTINUE_LOOP)
2519 retval = PyLong_FromLong(oparg);
2520 if (!retval) {
2521 x = NULL;
2522 break;
2523 }
2524 why = WHY_CONTINUE;
2525 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002527 TARGET_WITH_IMPL(SETUP_LOOP, _setup_finally)
2528 TARGET_WITH_IMPL(SETUP_EXCEPT, _setup_finally)
2529 TARGET(SETUP_FINALLY)
2530 _setup_finally:
2531 /* NOTE: If you add any new block-setup opcodes that
2532 are not try/except/finally handlers, you may need
2533 to update the PyGen_NeedsFinalizing() function.
2534 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002536 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
2537 STACK_LEVEL());
2538 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002540 TARGET(SETUP_WITH)
2541 {
2542 static PyObject *exit, *enter;
2543 w = TOP();
2544 x = special_lookup(w, "__exit__", &exit);
2545 if (!x)
2546 break;
2547 SET_TOP(x);
2548 u = special_lookup(w, "__enter__", &enter);
2549 Py_DECREF(w);
2550 if (!u) {
2551 x = NULL;
2552 break;
2553 }
2554 x = PyObject_CallFunctionObjArgs(u, NULL);
2555 Py_DECREF(u);
2556 if (!x)
2557 break;
2558 /* Setup the finally block before pushing the result
2559 of __enter__ on the stack. */
2560 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2561 STACK_LEVEL());
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002562
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002563 PUSH(x);
2564 DISPATCH();
2565 }
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002567 TARGET(WITH_CLEANUP)
2568 {
2569 /* At the top of the stack are 1-3 values indicating
2570 how/why we entered the finally clause:
2571 - TOP = None
2572 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2573 - TOP = WHY_*; no retval below it
2574 - (TOP, SECOND, THIRD) = exc_info()
2575 (FOURTH, FITH, SIXTH) = previous exception for EXCEPT_HANDLER
2576 Below them is EXIT, the context.__exit__ bound method.
2577 In the last case, we must call
2578 EXIT(TOP, SECOND, THIRD)
2579 otherwise we must call
2580 EXIT(None, None, None)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002581
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002582 In the first two cases, we remove EXIT from the
2583 stack, leaving the rest in the same order. In the
2584 third case, we shift the bottom 3 values of the
2585 stack down, and replace the empty spot with NULL.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002587 In addition, if the stack represents an exception,
2588 *and* the function call returns a 'true' value, we
2589 push WHY_SILENCED onto the stack. END_FINALLY will
2590 then not re-raise the exception. (But non-local
2591 gotos should still be resumed.)
2592 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002594 PyObject *exit_func;
2595 u = TOP();
2596 if (u == Py_None) {
2597 (void)POP();
2598 exit_func = TOP();
2599 SET_TOP(u);
2600 v = w = Py_None;
2601 }
2602 else if (PyLong_Check(u)) {
2603 (void)POP();
2604 switch(PyLong_AsLong(u)) {
2605 case WHY_RETURN:
2606 case WHY_CONTINUE:
2607 /* Retval in TOP. */
2608 exit_func = SECOND();
2609 SET_SECOND(TOP());
2610 SET_TOP(u);
2611 break;
2612 default:
2613 exit_func = TOP();
2614 SET_TOP(u);
2615 break;
2616 }
2617 u = v = w = Py_None;
2618 }
2619 else {
2620 PyObject *tp, *exc, *tb;
2621 PyTryBlock *block;
2622 v = SECOND();
2623 w = THIRD();
2624 tp = FOURTH();
2625 exc = PEEK(5);
2626 tb = PEEK(6);
2627 exit_func = PEEK(7);
2628 SET_VALUE(7, tb);
2629 SET_VALUE(6, exc);
2630 SET_VALUE(5, tp);
2631 /* UNWIND_EXCEPT_HANDLER will pop this off. */
2632 SET_FOURTH(NULL);
2633 /* We just shifted the stack down, so we have
2634 to tell the except handler block that the
2635 values are lower than it expects. */
2636 block = &f->f_blockstack[f->f_iblock - 1];
2637 assert(block->b_type == EXCEPT_HANDLER);
2638 block->b_level--;
2639 }
2640 /* XXX Not the fastest way to call it... */
2641 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2642 NULL);
2643 Py_DECREF(exit_func);
2644 if (x == NULL)
2645 break; /* Go to error exit */
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002647 if (u != Py_None)
2648 err = PyObject_IsTrue(x);
2649 else
2650 err = 0;
2651 Py_DECREF(x);
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002652
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002653 if (err < 0)
2654 break; /* Go to error exit */
2655 else if (err > 0) {
2656 err = 0;
2657 /* There was an exception and a True return */
2658 PUSH(PyLong_FromLong((long) WHY_SILENCED));
2659 }
2660 PREDICT(END_FINALLY);
2661 break;
2662 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002663
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002664 TARGET(CALL_FUNCTION)
2665 {
2666 PyObject **sp;
2667 PCALL(PCALL_ALL);
2668 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002669#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002670 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002671#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002672 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002673#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002674 stack_pointer = sp;
2675 PUSH(x);
2676 if (x != NULL)
2677 DISPATCH();
2678 break;
2679 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002680
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002681 TARGET_WITH_IMPL(CALL_FUNCTION_VAR, _call_function_var_kw)
2682 TARGET_WITH_IMPL(CALL_FUNCTION_KW, _call_function_var_kw)
2683 TARGET(CALL_FUNCTION_VAR_KW)
2684 _call_function_var_kw:
2685 {
2686 int na = oparg & 0xff;
2687 int nk = (oparg>>8) & 0xff;
2688 int flags = (opcode - CALL_FUNCTION) & 3;
2689 int n = na + 2 * nk;
2690 PyObject **pfunc, *func, **sp;
2691 PCALL(PCALL_ALL);
2692 if (flags & CALL_FLAG_VAR)
2693 n++;
2694 if (flags & CALL_FLAG_KW)
2695 n++;
2696 pfunc = stack_pointer - n - 1;
2697 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 if (PyMethod_Check(func)
Stefan Krahb7e10102010-06-23 18:42:39 +00002700 && PyMethod_GET_SELF(func) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002701 PyObject *self = PyMethod_GET_SELF(func);
2702 Py_INCREF(self);
2703 func = PyMethod_GET_FUNCTION(func);
2704 Py_INCREF(func);
2705 Py_DECREF(*pfunc);
2706 *pfunc = self;
2707 na++;
Brett Cannonb94767f2011-02-22 20:15:44 +00002708 /* n++; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002709 } else
2710 Py_INCREF(func);
2711 sp = stack_pointer;
2712 READ_TIMESTAMP(intr0);
2713 x = ext_do_call(func, &sp, flags, na, nk);
2714 READ_TIMESTAMP(intr1);
2715 stack_pointer = sp;
2716 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002717
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002718 while (stack_pointer > pfunc) {
2719 w = POP();
2720 Py_DECREF(w);
2721 }
2722 PUSH(x);
2723 if (x != NULL)
2724 DISPATCH();
2725 break;
2726 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002727
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002728 TARGET_WITH_IMPL(MAKE_CLOSURE, _make_function)
2729 TARGET(MAKE_FUNCTION)
2730 _make_function:
2731 {
2732 int posdefaults = oparg & 0xff;
2733 int kwdefaults = (oparg>>8) & 0xff;
2734 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002735
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002736 w = POP(); /* qualname */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002737 v = POP(); /* code object */
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002738 x = PyFunction_NewWithQualName(v, f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002739 Py_DECREF(v);
Antoine Pitrou86a36b52011-11-25 18:56:07 +01002740 Py_DECREF(w);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002742 if (x != NULL && opcode == MAKE_CLOSURE) {
2743 v = POP();
2744 if (PyFunction_SetClosure(x, v) != 0) {
2745 /* Can't happen unless bytecode is corrupt. */
2746 why = WHY_EXCEPTION;
2747 }
2748 Py_DECREF(v);
2749 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002750
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002751 if (x != NULL && num_annotations > 0) {
2752 Py_ssize_t name_ix;
2753 u = POP(); /* names of args with annotations */
2754 v = PyDict_New();
2755 if (v == NULL) {
2756 Py_DECREF(x);
2757 x = NULL;
2758 break;
2759 }
2760 name_ix = PyTuple_Size(u);
2761 assert(num_annotations == name_ix+1);
2762 while (name_ix > 0) {
2763 --name_ix;
2764 t = PyTuple_GET_ITEM(u, name_ix);
2765 w = POP();
2766 /* XXX(nnorwitz): check for errors */
2767 PyDict_SetItem(v, t, w);
2768 Py_DECREF(w);
2769 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002771 if (PyFunction_SetAnnotations(x, v) != 0) {
2772 /* Can't happen unless
2773 PyFunction_SetAnnotations changes. */
2774 why = WHY_EXCEPTION;
2775 }
2776 Py_DECREF(v);
2777 Py_DECREF(u);
2778 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002779
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002780 /* XXX Maybe this should be a separate opcode? */
2781 if (x != NULL && posdefaults > 0) {
2782 v = PyTuple_New(posdefaults);
2783 if (v == NULL) {
2784 Py_DECREF(x);
2785 x = NULL;
2786 break;
2787 }
2788 while (--posdefaults >= 0) {
2789 w = POP();
2790 PyTuple_SET_ITEM(v, posdefaults, w);
2791 }
2792 if (PyFunction_SetDefaults(x, v) != 0) {
2793 /* Can't happen unless
2794 PyFunction_SetDefaults changes. */
2795 why = WHY_EXCEPTION;
2796 }
2797 Py_DECREF(v);
2798 }
2799 if (x != NULL && kwdefaults > 0) {
2800 v = PyDict_New();
2801 if (v == NULL) {
2802 Py_DECREF(x);
2803 x = NULL;
2804 break;
2805 }
2806 while (--kwdefaults >= 0) {
2807 w = POP(); /* default value */
2808 u = POP(); /* kw only arg name */
2809 /* XXX(nnorwitz): check for errors */
2810 PyDict_SetItem(v, u, w);
2811 Py_DECREF(w);
2812 Py_DECREF(u);
2813 }
2814 if (PyFunction_SetKwDefaults(x, v) != 0) {
2815 /* Can't happen unless
2816 PyFunction_SetKwDefaults changes. */
2817 why = WHY_EXCEPTION;
2818 }
2819 Py_DECREF(v);
2820 }
2821 PUSH(x);
2822 break;
2823 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002824
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002825 TARGET(BUILD_SLICE)
2826 if (oparg == 3)
2827 w = POP();
2828 else
2829 w = NULL;
2830 v = POP();
2831 u = TOP();
2832 x = PySlice_New(u, v, w);
2833 Py_DECREF(u);
2834 Py_DECREF(v);
2835 Py_XDECREF(w);
2836 SET_TOP(x);
2837 if (x != NULL) DISPATCH();
2838 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002839
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002840 TARGET(EXTENDED_ARG)
2841 opcode = NEXTOP();
2842 oparg = oparg<<16 | NEXTARG();
2843 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002844
Antoine Pitrou042b1282010-08-13 21:15:58 +00002845#if USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002846 _unknown_opcode:
Antoine Pitroub52ec782009-01-25 16:34:23 +00002847#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002848 default:
2849 fprintf(stderr,
2850 "XXX lineno: %d, opcode: %d\n",
2851 PyFrame_GetLineNumber(f),
2852 opcode);
2853 PyErr_SetString(PyExc_SystemError, "unknown opcode");
2854 why = WHY_EXCEPTION;
2855 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002856
2857#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002858 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00002859#endif
2860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002861 } /* switch */
Guido van Rossum374a9221991-04-04 10:40:29 +00002862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002863 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002864
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002865 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002866
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002867 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002869 if (why == WHY_NOT) {
2870 if (err == 0 && x != NULL) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002871#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002872 /* This check is expensive! */
2873 if (PyErr_Occurred())
2874 fprintf(stderr,
2875 "XXX undetected error\n");
2876 else {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002877#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002878 READ_TIMESTAMP(loop1);
2879 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002880#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002881 }
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002882#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002883 }
2884 why = WHY_EXCEPTION;
2885 x = Py_None;
2886 err = 0;
2887 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002888
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002889 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002890
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002891 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
2892 if (!PyErr_Occurred()) {
2893 PyErr_SetString(PyExc_SystemError,
2894 "error return without exception set");
2895 why = WHY_EXCEPTION;
2896 }
2897 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002898#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002899 else {
2900 /* This check is expensive! */
2901 if (PyErr_Occurred()) {
2902 char buf[128];
2903 sprintf(buf, "Stack unwind with exception "
2904 "set and why=%d", why);
2905 Py_FatalError(buf);
2906 }
2907 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002908#endif
2909
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002910 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002911
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002912 if (why == WHY_EXCEPTION) {
2913 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002914
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002915 if (tstate->c_tracefunc != NULL)
2916 call_exc_trace(tstate->c_tracefunc,
2917 tstate->c_traceobj, f);
2918 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002919
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002920 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002922 if (why == WHY_RERAISE)
2923 why = WHY_EXCEPTION;
Guido van Rossum374a9221991-04-04 10:40:29 +00002924
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002925 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002926
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002927fast_block_end:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002928 while (why != WHY_NOT && f->f_iblock > 0) {
2929 /* Peek at the current block. */
2930 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002931
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002932 assert(why != WHY_YIELD);
2933 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2934 why = WHY_NOT;
2935 JUMPTO(PyLong_AS_LONG(retval));
2936 Py_DECREF(retval);
2937 break;
2938 }
2939 /* Now we have to pop the block. */
2940 f->f_iblock--;
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002941
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002942 if (b->b_type == EXCEPT_HANDLER) {
2943 UNWIND_EXCEPT_HANDLER(b);
2944 continue;
2945 }
2946 UNWIND_BLOCK(b);
2947 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2948 why = WHY_NOT;
2949 JUMPTO(b->b_handler);
2950 break;
2951 }
2952 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
2953 || b->b_type == SETUP_FINALLY)) {
2954 PyObject *exc, *val, *tb;
2955 int handler = b->b_handler;
2956 /* Beware, this invalidates all b->b_* fields */
2957 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
2958 PUSH(tstate->exc_traceback);
2959 PUSH(tstate->exc_value);
2960 if (tstate->exc_type != NULL) {
2961 PUSH(tstate->exc_type);
2962 }
2963 else {
2964 Py_INCREF(Py_None);
2965 PUSH(Py_None);
2966 }
2967 PyErr_Fetch(&exc, &val, &tb);
2968 /* Make the raw exception data
2969 available to the handler,
2970 so a program can emulate the
2971 Python main loop. */
2972 PyErr_NormalizeException(
2973 &exc, &val, &tb);
2974 PyException_SetTraceback(val, tb);
2975 Py_INCREF(exc);
2976 tstate->exc_type = exc;
2977 Py_INCREF(val);
2978 tstate->exc_value = val;
2979 tstate->exc_traceback = tb;
2980 if (tb == NULL)
2981 tb = Py_None;
2982 Py_INCREF(tb);
2983 PUSH(tb);
2984 PUSH(val);
2985 PUSH(exc);
2986 why = WHY_NOT;
2987 JUMPTO(handler);
2988 break;
2989 }
2990 if (b->b_type == SETUP_FINALLY) {
2991 if (why & (WHY_RETURN | WHY_CONTINUE))
2992 PUSH(retval);
2993 PUSH(PyLong_FromLong((long)why));
2994 why = WHY_NOT;
2995 JUMPTO(b->b_handler);
2996 break;
2997 }
2998 } /* unwind stack */
Guido van Rossum374a9221991-04-04 10:40:29 +00002999
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003000 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00003001
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003002 if (why != WHY_NOT)
3003 break;
3004 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00003005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003006 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00003007
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003008 assert(why != WHY_YIELD);
3009 /* Pop remaining stack entries. */
3010 while (!EMPTY()) {
3011 v = POP();
3012 Py_XDECREF(v);
3013 }
Guido van Rossum35974fb2001-12-06 21:28:18 +00003014
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003015 if (why != WHY_RETURN)
3016 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00003017
Raymond Hettinger1dd83092004-02-06 18:32:33 +00003018fast_yield:
Benjamin Petersonac913412011-07-03 16:25:11 -05003019 if (co->co_flags & CO_GENERATOR && (why == WHY_YIELD || why == WHY_RETURN)) {
3020 /* The purpose of this block is to put aside the generator's exception
3021 state and restore that of the calling frame. If the current
3022 exception state is from the caller, we clear the exception values
3023 on the generator frame, so they are not swapped back in latter. The
3024 origin of the current exception state is determined by checking for
3025 except handler blocks, which we must be in iff a new exception
3026 state came into existence in this frame. (An uncaught exception
3027 would have why == WHY_EXCEPTION, and we wouldn't be here). */
3028 int i;
3029 for (i = 0; i < f->f_iblock; i++)
3030 if (f->f_blockstack[i].b_type == EXCEPT_HANDLER)
3031 break;
3032 if (i == f->f_iblock)
3033 /* We did not create this exception. */
Benjamin Peterson87880242011-07-03 16:48:31 -05003034 restore_and_clear_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003035 else
Benjamin Peterson87880242011-07-03 16:48:31 -05003036 swap_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05003037 }
Benjamin Peterson83195c32011-07-03 13:44:00 -05003038
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003039 if (tstate->use_tracing) {
3040 if (tstate->c_tracefunc) {
3041 if (why == WHY_RETURN || why == WHY_YIELD) {
3042 if (call_trace(tstate->c_tracefunc,
3043 tstate->c_traceobj, f,
3044 PyTrace_RETURN, retval)) {
3045 Py_XDECREF(retval);
3046 retval = NULL;
3047 why = WHY_EXCEPTION;
3048 }
3049 }
3050 else if (why == WHY_EXCEPTION) {
3051 call_trace_protected(tstate->c_tracefunc,
3052 tstate->c_traceobj, f,
3053 PyTrace_RETURN, NULL);
3054 }
3055 }
3056 if (tstate->c_profilefunc) {
3057 if (why == WHY_EXCEPTION)
3058 call_trace_protected(tstate->c_profilefunc,
3059 tstate->c_profileobj, f,
3060 PyTrace_RETURN, NULL);
3061 else if (call_trace(tstate->c_profilefunc,
3062 tstate->c_profileobj, f,
3063 PyTrace_RETURN, retval)) {
3064 Py_XDECREF(retval);
3065 retval = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003066 /* why = WHY_EXCEPTION; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003067 }
3068 }
3069 }
Guido van Rossuma4240131997-01-21 21:18:36 +00003070
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003071 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00003072exit_eval_frame:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003073 Py_LeaveRecursiveCall();
3074 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00003075
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003076 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00003077}
3078
Benjamin Petersonb204a422011-06-05 22:04:07 -05003079static void
Benjamin Petersone109c702011-06-24 09:37:26 -05003080format_missing(const char *kind, PyCodeObject *co, PyObject *names)
3081{
3082 int err;
3083 Py_ssize_t len = PyList_GET_SIZE(names);
3084 PyObject *name_str, *comma, *tail, *tmp;
3085
3086 assert(PyList_CheckExact(names));
3087 assert(len >= 1);
3088 /* Deal with the joys of natural language. */
3089 switch (len) {
3090 case 1:
3091 name_str = PyList_GET_ITEM(names, 0);
3092 Py_INCREF(name_str);
3093 break;
3094 case 2:
3095 name_str = PyUnicode_FromFormat("%U and %U",
3096 PyList_GET_ITEM(names, len - 2),
3097 PyList_GET_ITEM(names, len - 1));
3098 break;
3099 default:
3100 tail = PyUnicode_FromFormat(", %U, and %U",
3101 PyList_GET_ITEM(names, len - 2),
3102 PyList_GET_ITEM(names, len - 1));
3103 /* Chop off the last two objects in the list. This shouldn't actually
3104 fail, but we can't be too careful. */
3105 err = PyList_SetSlice(names, len - 2, len, NULL);
3106 if (err == -1) {
3107 Py_DECREF(tail);
3108 return;
3109 }
3110 /* Stitch everything up into a nice comma-separated list. */
3111 comma = PyUnicode_FromString(", ");
3112 if (comma == NULL) {
3113 Py_DECREF(tail);
3114 return;
3115 }
3116 tmp = PyUnicode_Join(comma, names);
3117 Py_DECREF(comma);
3118 if (tmp == NULL) {
3119 Py_DECREF(tail);
3120 return;
3121 }
3122 name_str = PyUnicode_Concat(tmp, tail);
3123 Py_DECREF(tmp);
3124 Py_DECREF(tail);
3125 break;
3126 }
3127 if (name_str == NULL)
3128 return;
3129 PyErr_Format(PyExc_TypeError,
3130 "%U() missing %i required %s argument%s: %U",
3131 co->co_name,
3132 len,
3133 kind,
3134 len == 1 ? "" : "s",
3135 name_str);
3136 Py_DECREF(name_str);
3137}
3138
3139static void
3140missing_arguments(PyCodeObject *co, int missing, int defcount,
3141 PyObject **fastlocals)
3142{
3143 int i, j = 0;
3144 int start, end;
3145 int positional = defcount != -1;
3146 const char *kind = positional ? "positional" : "keyword-only";
3147 PyObject *missing_names;
3148
3149 /* Compute the names of the arguments that are missing. */
3150 missing_names = PyList_New(missing);
3151 if (missing_names == NULL)
3152 return;
3153 if (positional) {
3154 start = 0;
3155 end = co->co_argcount - defcount;
3156 }
3157 else {
3158 start = co->co_argcount;
3159 end = start + co->co_kwonlyargcount;
3160 }
3161 for (i = start; i < end; i++) {
3162 if (GETLOCAL(i) == NULL) {
3163 PyObject *raw = PyTuple_GET_ITEM(co->co_varnames, i);
3164 PyObject *name = PyObject_Repr(raw);
3165 if (name == NULL) {
3166 Py_DECREF(missing_names);
3167 return;
3168 }
3169 PyList_SET_ITEM(missing_names, j++, name);
3170 }
3171 }
3172 assert(j == missing);
3173 format_missing(kind, co, missing_names);
3174 Py_DECREF(missing_names);
3175}
3176
3177static void
3178too_many_positional(PyCodeObject *co, int given, int defcount, PyObject **fastlocals)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003179{
3180 int plural;
3181 int kwonly_given = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003182 int i;
3183 PyObject *sig, *kwonly_sig;
3184
Benjamin Petersone109c702011-06-24 09:37:26 -05003185 assert((co->co_flags & CO_VARARGS) == 0);
3186 /* Count missing keyword-only args. */
Benjamin Petersonb204a422011-06-05 22:04:07 -05003187 for (i = co->co_argcount; i < co->co_argcount + co->co_kwonlyargcount; i++)
Benjamin Petersone109c702011-06-24 09:37:26 -05003188 if (GETLOCAL(i) != NULL)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003189 kwonly_given++;
Benjamin Petersone109c702011-06-24 09:37:26 -05003190 if (defcount) {
3191 int atleast = co->co_argcount - defcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003192 plural = 1;
3193 sig = PyUnicode_FromFormat("from %d to %d", atleast, co->co_argcount);
3194 }
3195 else {
3196 plural = co->co_argcount != 1;
3197 sig = PyUnicode_FromFormat("%d", co->co_argcount);
3198 }
3199 if (sig == NULL)
3200 return;
3201 if (kwonly_given) {
3202 const char *format = " positional argument%s (and %d keyword-only argument%s)";
3203 kwonly_sig = PyUnicode_FromFormat(format, given != 1 ? "s" : "", kwonly_given,
3204 kwonly_given != 1 ? "s" : "");
3205 if (kwonly_sig == NULL) {
3206 Py_DECREF(sig);
3207 return;
3208 }
3209 }
3210 else {
3211 /* This will not fail. */
3212 kwonly_sig = PyUnicode_FromString("");
Benjamin Petersone109c702011-06-24 09:37:26 -05003213 assert(kwonly_sig != NULL);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003214 }
3215 PyErr_Format(PyExc_TypeError,
3216 "%U() takes %U positional argument%s but %d%U %s given",
3217 co->co_name,
3218 sig,
3219 plural ? "s" : "",
3220 given,
3221 kwonly_sig,
3222 given == 1 && !kwonly_given ? "was" : "were");
3223 Py_DECREF(sig);
3224 Py_DECREF(kwonly_sig);
3225}
3226
Guido van Rossumc2e20742006-02-27 22:32:47 +00003227/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00003228 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00003229 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00003230
Tim Peters6d6c1a32001-08-02 04:15:00 +00003231PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003232PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 PyObject **args, int argcount, PyObject **kws, int kwcount,
3234 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00003235{
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003236 PyCodeObject* co = (PyCodeObject*)_co;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003237 register PyFrameObject *f;
3238 register PyObject *retval = NULL;
3239 register PyObject **fastlocals, **freevars;
3240 PyThreadState *tstate = PyThreadState_GET();
3241 PyObject *x, *u;
3242 int total_args = co->co_argcount + co->co_kwonlyargcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003243 int i;
3244 int n = argcount;
3245 PyObject *kwdict = NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003246
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003247 if (globals == NULL) {
3248 PyErr_SetString(PyExc_SystemError,
3249 "PyEval_EvalCodeEx: NULL globals");
3250 return NULL;
3251 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003252
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003253 assert(tstate != NULL);
3254 assert(globals != NULL);
3255 f = PyFrame_New(tstate, co, globals, locals);
3256 if (f == NULL)
3257 return NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003259 fastlocals = f->f_localsplus;
3260 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00003261
Benjamin Petersonb204a422011-06-05 22:04:07 -05003262 /* Parse arguments. */
3263 if (co->co_flags & CO_VARKEYWORDS) {
3264 kwdict = PyDict_New();
3265 if (kwdict == NULL)
3266 goto fail;
3267 i = total_args;
3268 if (co->co_flags & CO_VARARGS)
3269 i++;
3270 SETLOCAL(i, kwdict);
3271 }
3272 if (argcount > co->co_argcount)
3273 n = co->co_argcount;
3274 for (i = 0; i < n; i++) {
3275 x = args[i];
3276 Py_INCREF(x);
3277 SETLOCAL(i, x);
3278 }
3279 if (co->co_flags & CO_VARARGS) {
3280 u = PyTuple_New(argcount - n);
3281 if (u == NULL)
3282 goto fail;
3283 SETLOCAL(total_args, u);
3284 for (i = n; i < argcount; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003285 x = args[i];
3286 Py_INCREF(x);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003287 PyTuple_SET_ITEM(u, i-n, x);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003288 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003289 }
3290 for (i = 0; i < kwcount; i++) {
3291 PyObject **co_varnames;
3292 PyObject *keyword = kws[2*i];
3293 PyObject *value = kws[2*i + 1];
3294 int j;
3295 if (keyword == NULL || !PyUnicode_Check(keyword)) {
3296 PyErr_Format(PyExc_TypeError,
3297 "%U() keywords must be strings",
3298 co->co_name);
3299 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003300 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003301 /* Speed hack: do raw pointer compares. As names are
3302 normally interned this should almost always hit. */
3303 co_varnames = ((PyTupleObject *)(co->co_varnames))->ob_item;
3304 for (j = 0; j < total_args; j++) {
3305 PyObject *nm = co_varnames[j];
3306 if (nm == keyword)
3307 goto kw_found;
3308 }
3309 /* Slow fallback, just in case */
3310 for (j = 0; j < total_args; j++) {
3311 PyObject *nm = co_varnames[j];
3312 int cmp = PyObject_RichCompareBool(
3313 keyword, nm, Py_EQ);
3314 if (cmp > 0)
3315 goto kw_found;
3316 else if (cmp < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003317 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003318 }
3319 if (j >= total_args && kwdict == NULL) {
3320 PyErr_Format(PyExc_TypeError,
3321 "%U() got an unexpected "
3322 "keyword argument '%S'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003323 co->co_name,
3324 keyword);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003325 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003326 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003327 PyDict_SetItem(kwdict, keyword, value);
3328 continue;
3329 kw_found:
3330 if (GETLOCAL(j) != NULL) {
3331 PyErr_Format(PyExc_TypeError,
3332 "%U() got multiple "
3333 "values for argument '%S'",
3334 co->co_name,
3335 keyword);
3336 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003337 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003338 Py_INCREF(value);
3339 SETLOCAL(j, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003340 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003341 if (argcount > co->co_argcount && !(co->co_flags & CO_VARARGS)) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003342 too_many_positional(co, argcount, defcount, fastlocals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003343 goto fail;
3344 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003345 if (argcount < co->co_argcount) {
3346 int m = co->co_argcount - defcount;
Benjamin Petersone109c702011-06-24 09:37:26 -05003347 int missing = 0;
3348 for (i = argcount; i < m; i++)
3349 if (GETLOCAL(i) == NULL)
3350 missing++;
3351 if (missing) {
3352 missing_arguments(co, missing, defcount, fastlocals);
3353 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003354 }
3355 if (n > m)
3356 i = n - m;
3357 else
3358 i = 0;
3359 for (; i < defcount; i++) {
3360 if (GETLOCAL(m+i) == NULL) {
3361 PyObject *def = defs[i];
3362 Py_INCREF(def);
3363 SETLOCAL(m+i, def);
3364 }
3365 }
3366 }
3367 if (co->co_kwonlyargcount > 0) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003368 int missing = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003369 for (i = co->co_argcount; i < total_args; i++) {
3370 PyObject *name;
3371 if (GETLOCAL(i) != NULL)
3372 continue;
3373 name = PyTuple_GET_ITEM(co->co_varnames, i);
3374 if (kwdefs != NULL) {
3375 PyObject *def = PyDict_GetItem(kwdefs, name);
3376 if (def) {
3377 Py_INCREF(def);
3378 SETLOCAL(i, def);
3379 continue;
3380 }
3381 }
Benjamin Petersone109c702011-06-24 09:37:26 -05003382 missing++;
3383 }
3384 if (missing) {
3385 missing_arguments(co, missing, -1, fastlocals);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003386 goto fail;
3387 }
3388 }
3389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003390 /* Allocate and initialize storage for cell vars, and copy free
Benjamin Peterson90037602011-06-25 22:54:45 -05003391 vars into frame. */
3392 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003393 PyObject *c;
Benjamin Peterson90037602011-06-25 22:54:45 -05003394 int arg;
3395 /* Possibly account for the cell variable being an argument. */
3396 if (co->co_cell2arg != NULL &&
3397 (arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG)
3398 c = PyCell_New(GETLOCAL(arg));
3399 else
3400 c = PyCell_New(NULL);
3401 if (c == NULL)
3402 goto fail;
3403 SETLOCAL(co->co_nlocals + i, c);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003404 }
Benjamin Peterson90037602011-06-25 22:54:45 -05003405 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
3406 PyObject *o = PyTuple_GET_ITEM(closure, i);
3407 Py_INCREF(o);
3408 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003409 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003410
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003411 if (co->co_flags & CO_GENERATOR) {
3412 /* Don't need to keep the reference to f_back, it will be set
3413 * when the generator is resumed. */
3414 Py_XDECREF(f->f_back);
3415 f->f_back = NULL;
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003416
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003417 PCALL(PCALL_GENERATOR);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003418
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003419 /* Create a new generator that owns the ready to run frame
3420 * and return that as the value. */
3421 return PyGen_New(f);
3422 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003423
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003424 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003425
Thomas Woutersce272b62007-09-19 21:19:28 +00003426fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003427
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003428 /* decref'ing the frame can cause __del__ methods to get invoked,
3429 which can call back into Python. While we're done with the
3430 current Python frame (f), the associated C stack is still in use,
3431 so recursion_depth must be boosted for the duration.
3432 */
3433 assert(tstate != NULL);
3434 ++tstate->recursion_depth;
3435 Py_DECREF(f);
3436 --tstate->recursion_depth;
3437 return retval;
Tim Peters5ca576e2001-06-18 22:08:13 +00003438}
3439
3440
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003441static PyObject *
3442special_lookup(PyObject *o, char *meth, PyObject **cache)
3443{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003444 PyObject *res;
3445 res = _PyObject_LookupSpecial(o, meth, cache);
3446 if (res == NULL && !PyErr_Occurred()) {
3447 PyErr_SetObject(PyExc_AttributeError, *cache);
3448 return NULL;
3449 }
3450 return res;
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003451}
3452
3453
Benjamin Peterson87880242011-07-03 16:48:31 -05003454/* These 3 functions deal with the exception state of generators. */
3455
3456static void
3457save_exc_state(PyThreadState *tstate, PyFrameObject *f)
3458{
3459 PyObject *type, *value, *traceback;
3460 Py_XINCREF(tstate->exc_type);
3461 Py_XINCREF(tstate->exc_value);
3462 Py_XINCREF(tstate->exc_traceback);
3463 type = f->f_exc_type;
3464 value = f->f_exc_value;
3465 traceback = f->f_exc_traceback;
3466 f->f_exc_type = tstate->exc_type;
3467 f->f_exc_value = tstate->exc_value;
3468 f->f_exc_traceback = tstate->exc_traceback;
3469 Py_XDECREF(type);
3470 Py_XDECREF(value);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02003471 Py_XDECREF(traceback);
Benjamin Peterson87880242011-07-03 16:48:31 -05003472}
3473
3474static void
3475swap_exc_state(PyThreadState *tstate, PyFrameObject *f)
3476{
3477 PyObject *tmp;
3478 tmp = tstate->exc_type;
3479 tstate->exc_type = f->f_exc_type;
3480 f->f_exc_type = tmp;
3481 tmp = tstate->exc_value;
3482 tstate->exc_value = f->f_exc_value;
3483 f->f_exc_value = tmp;
3484 tmp = tstate->exc_traceback;
3485 tstate->exc_traceback = f->f_exc_traceback;
3486 f->f_exc_traceback = tmp;
3487}
3488
3489static void
3490restore_and_clear_exc_state(PyThreadState *tstate, PyFrameObject *f)
3491{
3492 PyObject *type, *value, *tb;
3493 type = tstate->exc_type;
3494 value = tstate->exc_value;
3495 tb = tstate->exc_traceback;
3496 tstate->exc_type = f->f_exc_type;
3497 tstate->exc_value = f->f_exc_value;
3498 tstate->exc_traceback = f->f_exc_traceback;
3499 f->f_exc_type = NULL;
3500 f->f_exc_value = NULL;
3501 f->f_exc_traceback = NULL;
3502 Py_XDECREF(type);
3503 Py_XDECREF(value);
3504 Py_XDECREF(tb);
3505}
3506
3507
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003508/* Logic for the raise statement (too complicated for inlining).
3509 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00003510static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00003511do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003512{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003513 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00003514
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003515 if (exc == NULL) {
3516 /* Reraise */
3517 PyThreadState *tstate = PyThreadState_GET();
3518 PyObject *tb;
3519 type = tstate->exc_type;
3520 value = tstate->exc_value;
3521 tb = tstate->exc_traceback;
3522 if (type == Py_None) {
3523 PyErr_SetString(PyExc_RuntimeError,
3524 "No active exception to reraise");
3525 return WHY_EXCEPTION;
3526 }
3527 Py_XINCREF(type);
3528 Py_XINCREF(value);
3529 Py_XINCREF(tb);
3530 PyErr_Restore(type, value, tb);
3531 return WHY_RERAISE;
3532 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003533
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003534 /* We support the following forms of raise:
3535 raise
Collin Winter828f04a2007-08-31 00:04:24 +00003536 raise <instance>
3537 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003539 if (PyExceptionClass_Check(exc)) {
3540 type = exc;
3541 value = PyObject_CallObject(exc, NULL);
3542 if (value == NULL)
3543 goto raise_error;
Benjamin Peterson5afa03a2011-07-15 14:09:26 -05003544 if (!PyExceptionInstance_Check(value)) {
3545 PyErr_Format(PyExc_TypeError,
3546 "calling %R should have returned an instance of "
3547 "BaseException, not %R",
3548 type, Py_TYPE(value));
3549 goto raise_error;
3550 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003551 }
3552 else if (PyExceptionInstance_Check(exc)) {
3553 value = exc;
3554 type = PyExceptionInstance_Class(exc);
3555 Py_INCREF(type);
3556 }
3557 else {
3558 /* Not something you can raise. You get an exception
3559 anyway, just not what you specified :-) */
3560 Py_DECREF(exc);
3561 PyErr_SetString(PyExc_TypeError,
3562 "exceptions must derive from BaseException");
3563 goto raise_error;
3564 }
Collin Winter828f04a2007-08-31 00:04:24 +00003565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003566 if (cause) {
3567 PyObject *fixed_cause;
3568 if (PyExceptionClass_Check(cause)) {
3569 fixed_cause = PyObject_CallObject(cause, NULL);
3570 if (fixed_cause == NULL)
3571 goto raise_error;
3572 Py_DECREF(cause);
3573 }
3574 else if (PyExceptionInstance_Check(cause)) {
3575 fixed_cause = cause;
3576 }
3577 else {
3578 PyErr_SetString(PyExc_TypeError,
3579 "exception causes must derive from "
3580 "BaseException");
3581 goto raise_error;
3582 }
3583 PyException_SetCause(value, fixed_cause);
3584 }
Collin Winter828f04a2007-08-31 00:04:24 +00003585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003586 PyErr_SetObject(type, value);
3587 /* PyErr_SetObject incref's its arguments */
3588 Py_XDECREF(value);
3589 Py_XDECREF(type);
3590 return WHY_EXCEPTION;
Collin Winter828f04a2007-08-31 00:04:24 +00003591
3592raise_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003593 Py_XDECREF(value);
3594 Py_XDECREF(type);
3595 Py_XDECREF(cause);
3596 return WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003597}
3598
Tim Petersd6d010b2001-06-21 02:49:55 +00003599/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003600 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00003601
Guido van Rossum0368b722007-05-11 16:50:42 +00003602 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3603 with a variable target.
3604*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003605
Barry Warsawe42b18f1997-08-25 22:13:04 +00003606static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003607unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003608{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003609 int i = 0, j = 0;
3610 Py_ssize_t ll = 0;
3611 PyObject *it; /* iter(v) */
3612 PyObject *w;
3613 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003614
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003615 assert(v != NULL);
Tim Petersd6d010b2001-06-21 02:49:55 +00003616
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003617 it = PyObject_GetIter(v);
3618 if (it == NULL)
3619 goto Error;
Tim Petersd6d010b2001-06-21 02:49:55 +00003620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003621 for (; i < argcnt; i++) {
3622 w = PyIter_Next(it);
3623 if (w == NULL) {
3624 /* Iterator done, via error or exhaustion. */
3625 if (!PyErr_Occurred()) {
3626 PyErr_Format(PyExc_ValueError,
3627 "need more than %d value%s to unpack",
3628 i, i == 1 ? "" : "s");
3629 }
3630 goto Error;
3631 }
3632 *--sp = w;
3633 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003634
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003635 if (argcntafter == -1) {
3636 /* We better have exhausted the iterator now. */
3637 w = PyIter_Next(it);
3638 if (w == NULL) {
3639 if (PyErr_Occurred())
3640 goto Error;
3641 Py_DECREF(it);
3642 return 1;
3643 }
3644 Py_DECREF(w);
Georg Brandl0310a832010-07-10 10:32:36 +00003645 PyErr_Format(PyExc_ValueError, "too many values to unpack "
3646 "(expected %d)", argcnt);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003647 goto Error;
3648 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003649
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003650 l = PySequence_List(it);
3651 if (l == NULL)
3652 goto Error;
3653 *--sp = l;
3654 i++;
Guido van Rossum0368b722007-05-11 16:50:42 +00003655
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003656 ll = PyList_GET_SIZE(l);
3657 if (ll < argcntafter) {
3658 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3659 argcnt + ll);
3660 goto Error;
3661 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003663 /* Pop the "after-variable" args off the list. */
3664 for (j = argcntafter; j > 0; j--, i++) {
3665 *--sp = PyList_GET_ITEM(l, ll - j);
3666 }
3667 /* Resize the list. */
3668 Py_SIZE(l) = ll - argcntafter;
3669 Py_DECREF(it);
3670 return 1;
Guido van Rossum0368b722007-05-11 16:50:42 +00003671
Tim Petersd6d010b2001-06-21 02:49:55 +00003672Error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003673 for (; i > 0; i--, sp++)
3674 Py_DECREF(*sp);
3675 Py_XDECREF(it);
3676 return 0;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003677}
3678
3679
Guido van Rossum96a42c81992-01-12 02:29:51 +00003680#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003681static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003682prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003683{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003684 printf("%s ", str);
3685 if (PyObject_Print(v, stdout, 0) != 0)
3686 PyErr_Clear(); /* Don't know what else to do */
3687 printf("\n");
3688 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003689}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003690#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003691
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003692static void
Fred Drake5755ce62001-06-27 19:19:46 +00003693call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003694{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003695 PyObject *type, *value, *traceback, *arg;
3696 int err;
3697 PyErr_Fetch(&type, &value, &traceback);
3698 if (value == NULL) {
3699 value = Py_None;
3700 Py_INCREF(value);
3701 }
3702 arg = PyTuple_Pack(3, type, value, traceback);
3703 if (arg == NULL) {
3704 PyErr_Restore(type, value, traceback);
3705 return;
3706 }
3707 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
3708 Py_DECREF(arg);
3709 if (err == 0)
3710 PyErr_Restore(type, value, traceback);
3711 else {
3712 Py_XDECREF(type);
3713 Py_XDECREF(value);
3714 Py_XDECREF(traceback);
3715 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003716}
3717
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003718static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003719call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003720 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003721{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003722 PyObject *type, *value, *traceback;
3723 int err;
3724 PyErr_Fetch(&type, &value, &traceback);
3725 err = call_trace(func, obj, frame, what, arg);
3726 if (err == 0)
3727 {
3728 PyErr_Restore(type, value, traceback);
3729 return 0;
3730 }
3731 else {
3732 Py_XDECREF(type);
3733 Py_XDECREF(value);
3734 Py_XDECREF(traceback);
3735 return -1;
3736 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003737}
3738
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003739static int
Fred Drake5755ce62001-06-27 19:19:46 +00003740call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003741 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003742{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003743 register PyThreadState *tstate = frame->f_tstate;
3744 int result;
3745 if (tstate->tracing)
3746 return 0;
3747 tstate->tracing++;
3748 tstate->use_tracing = 0;
3749 result = func(obj, frame, what, arg);
3750 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3751 || (tstate->c_profilefunc != NULL));
3752 tstate->tracing--;
3753 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003754}
3755
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003756PyObject *
3757_PyEval_CallTracing(PyObject *func, PyObject *args)
3758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003759 PyFrameObject *frame = PyEval_GetFrame();
3760 PyThreadState *tstate = frame->f_tstate;
3761 int save_tracing = tstate->tracing;
3762 int save_use_tracing = tstate->use_tracing;
3763 PyObject *result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003765 tstate->tracing = 0;
3766 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3767 || (tstate->c_profilefunc != NULL));
3768 result = PyObject_Call(func, args, NULL);
3769 tstate->tracing = save_tracing;
3770 tstate->use_tracing = save_use_tracing;
3771 return result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003772}
3773
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00003774/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Michael W. Hudson006c7522002-11-08 13:08:46 +00003775static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003776maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003777 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3778 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003779{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003780 int result = 0;
3781 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003783 /* If the last instruction executed isn't in the current
3784 instruction window, reset the window.
3785 */
3786 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
3787 PyAddrPair bounds;
3788 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3789 &bounds);
3790 *instr_lb = bounds.ap_lower;
3791 *instr_ub = bounds.ap_upper;
3792 }
3793 /* If the last instruction falls at the start of a line or if
3794 it represents a jump backwards, update the frame's line
3795 number and call the trace function. */
3796 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3797 frame->f_lineno = line;
3798 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
3799 }
3800 *instr_prev = frame->f_lasti;
3801 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003802}
3803
Fred Drake5755ce62001-06-27 19:19:46 +00003804void
3805PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003806{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003807 PyThreadState *tstate = PyThreadState_GET();
3808 PyObject *temp = tstate->c_profileobj;
3809 Py_XINCREF(arg);
3810 tstate->c_profilefunc = NULL;
3811 tstate->c_profileobj = NULL;
3812 /* Must make sure that tracing is not ignored if 'temp' is freed */
3813 tstate->use_tracing = tstate->c_tracefunc != NULL;
3814 Py_XDECREF(temp);
3815 tstate->c_profilefunc = func;
3816 tstate->c_profileobj = arg;
3817 /* Flag that tracing or profiling is turned on */
3818 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003819}
3820
3821void
3822PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3823{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003824 PyThreadState *tstate = PyThreadState_GET();
3825 PyObject *temp = tstate->c_traceobj;
3826 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
3827 Py_XINCREF(arg);
3828 tstate->c_tracefunc = NULL;
3829 tstate->c_traceobj = NULL;
3830 /* Must make sure that profiling is not ignored if 'temp' is freed */
3831 tstate->use_tracing = tstate->c_profilefunc != NULL;
3832 Py_XDECREF(temp);
3833 tstate->c_tracefunc = func;
3834 tstate->c_traceobj = arg;
3835 /* Flag that tracing or profiling is turned on */
3836 tstate->use_tracing = ((func != NULL)
3837 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003838}
3839
Guido van Rossumb209a111997-04-29 18:18:01 +00003840PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003841PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003842{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003843 PyFrameObject *current_frame = PyEval_GetFrame();
3844 if (current_frame == NULL)
3845 return PyThreadState_GET()->interp->builtins;
3846 else
3847 return current_frame->f_builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003848}
3849
Guido van Rossumb209a111997-04-29 18:18:01 +00003850PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003851PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003852{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003853 PyFrameObject *current_frame = PyEval_GetFrame();
3854 if (current_frame == NULL)
3855 return NULL;
3856 PyFrame_FastToLocals(current_frame);
3857 return current_frame->f_locals;
Guido van Rossum5b722181993-03-30 17:46:03 +00003858}
3859
Guido van Rossumb209a111997-04-29 18:18:01 +00003860PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003861PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003862{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003863 PyFrameObject *current_frame = PyEval_GetFrame();
3864 if (current_frame == NULL)
3865 return NULL;
3866 else
3867 return current_frame->f_globals;
Guido van Rossum3f5da241990-12-20 15:06:42 +00003868}
3869
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003870PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003871PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003872{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003873 PyThreadState *tstate = PyThreadState_GET();
3874 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003875}
3876
Guido van Rossum6135a871995-01-09 17:53:26 +00003877int
Tim Peters5ba58662001-07-16 02:29:45 +00003878PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003879{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003880 PyFrameObject *current_frame = PyEval_GetFrame();
3881 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003882
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003883 if (current_frame != NULL) {
3884 const int codeflags = current_frame->f_code->co_flags;
3885 const int compilerflags = codeflags & PyCF_MASK;
3886 if (compilerflags) {
3887 result = 1;
3888 cf->cf_flags |= compilerflags;
3889 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003890#if 0 /* future keyword */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003891 if (codeflags & CO_GENERATOR_ALLOWED) {
3892 result = 1;
3893 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3894 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003895#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003896 }
3897 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003898}
3899
Guido van Rossum3f5da241990-12-20 15:06:42 +00003900
Guido van Rossum681d79a1995-07-18 14:51:37 +00003901/* External interface to call any callable object.
Antoine Pitrou8689a102010-04-01 16:53:15 +00003902 The arg must be a tuple or NULL. The kw must be a dict or NULL. */
Guido van Rossume59214e1994-08-30 08:01:59 +00003903
Guido van Rossumb209a111997-04-29 18:18:01 +00003904PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003905PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003906{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003907 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003908
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003909 if (arg == NULL) {
3910 arg = PyTuple_New(0);
3911 if (arg == NULL)
3912 return NULL;
3913 }
3914 else if (!PyTuple_Check(arg)) {
3915 PyErr_SetString(PyExc_TypeError,
3916 "argument list must be a tuple");
3917 return NULL;
3918 }
3919 else
3920 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003922 if (kw != NULL && !PyDict_Check(kw)) {
3923 PyErr_SetString(PyExc_TypeError,
3924 "keyword list must be a dictionary");
3925 Py_DECREF(arg);
3926 return NULL;
3927 }
Guido van Rossume3e61c11995-08-04 04:14:47 +00003928
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003929 result = PyObject_Call(func, arg, kw);
3930 Py_DECREF(arg);
3931 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00003932}
3933
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003934const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003935PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003936{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003937 if (PyMethod_Check(func))
3938 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
3939 else if (PyFunction_Check(func))
3940 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
3941 else if (PyCFunction_Check(func))
3942 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3943 else
3944 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003945}
3946
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003947const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003948PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003950 if (PyMethod_Check(func))
3951 return "()";
3952 else if (PyFunction_Check(func))
3953 return "()";
3954 else if (PyCFunction_Check(func))
3955 return "()";
3956 else
3957 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00003958}
3959
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00003960static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003961err_args(PyObject *func, int flags, int nargs)
3962{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003963 if (flags & METH_NOARGS)
3964 PyErr_Format(PyExc_TypeError,
3965 "%.200s() takes no arguments (%d given)",
3966 ((PyCFunctionObject *)func)->m_ml->ml_name,
3967 nargs);
3968 else
3969 PyErr_Format(PyExc_TypeError,
3970 "%.200s() takes exactly one argument (%d given)",
3971 ((PyCFunctionObject *)func)->m_ml->ml_name,
3972 nargs);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003973}
3974
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003975#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003976if (tstate->use_tracing && tstate->c_profilefunc) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003977 if (call_trace(tstate->c_profilefunc, \
3978 tstate->c_profileobj, \
3979 tstate->frame, PyTrace_C_CALL, \
3980 func)) { \
3981 x = NULL; \
3982 } \
3983 else { \
3984 x = call; \
3985 if (tstate->c_profilefunc != NULL) { \
3986 if (x == NULL) { \
3987 call_trace_protected(tstate->c_profilefunc, \
3988 tstate->c_profileobj, \
3989 tstate->frame, PyTrace_C_EXCEPTION, \
3990 func); \
3991 /* XXX should pass (type, value, tb) */ \
3992 } else { \
3993 if (call_trace(tstate->c_profilefunc, \
3994 tstate->c_profileobj, \
3995 tstate->frame, PyTrace_C_RETURN, \
3996 func)) { \
3997 Py_DECREF(x); \
3998 x = NULL; \
3999 } \
4000 } \
4001 } \
4002 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00004003} else { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004004 x = call; \
4005 }
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00004006
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004007static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004008call_function(PyObject ***pp_stack, int oparg
4009#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004010 , uint64* pintr0, uint64* pintr1
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00004011#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004012 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004013{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004014 int na = oparg & 0xff;
4015 int nk = (oparg>>8) & 0xff;
4016 int n = na + 2 * nk;
4017 PyObject **pfunc = (*pp_stack) - n - 1;
4018 PyObject *func = *pfunc;
4019 PyObject *x, *w;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004021 /* Always dispatch PyCFunction first, because these are
4022 presumed to be the most frequent callable object.
4023 */
4024 if (PyCFunction_Check(func) && nk == 0) {
4025 int flags = PyCFunction_GET_FLAGS(func);
4026 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00004027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004028 PCALL(PCALL_CFUNCTION);
4029 if (flags & (METH_NOARGS | METH_O)) {
4030 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
4031 PyObject *self = PyCFunction_GET_SELF(func);
4032 if (flags & METH_NOARGS && na == 0) {
4033 C_TRACE(x, (*meth)(self,NULL));
4034 }
4035 else if (flags & METH_O && na == 1) {
4036 PyObject *arg = EXT_POP(*pp_stack);
4037 C_TRACE(x, (*meth)(self,arg));
4038 Py_DECREF(arg);
4039 }
4040 else {
4041 err_args(func, flags, na);
4042 x = NULL;
4043 }
4044 }
4045 else {
4046 PyObject *callargs;
4047 callargs = load_args(pp_stack, na);
4048 READ_TIMESTAMP(*pintr0);
4049 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
4050 READ_TIMESTAMP(*pintr1);
4051 Py_XDECREF(callargs);
4052 }
4053 } else {
4054 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
4055 /* optimize access to bound methods */
4056 PyObject *self = PyMethod_GET_SELF(func);
4057 PCALL(PCALL_METHOD);
4058 PCALL(PCALL_BOUND_METHOD);
4059 Py_INCREF(self);
4060 func = PyMethod_GET_FUNCTION(func);
4061 Py_INCREF(func);
4062 Py_DECREF(*pfunc);
4063 *pfunc = self;
4064 na++;
4065 n++;
4066 } else
4067 Py_INCREF(func);
4068 READ_TIMESTAMP(*pintr0);
4069 if (PyFunction_Check(func))
4070 x = fast_function(func, pp_stack, n, na, nk);
4071 else
4072 x = do_call(func, pp_stack, na, nk);
4073 READ_TIMESTAMP(*pintr1);
4074 Py_DECREF(func);
4075 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00004076
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004077 /* Clear the stack of the function object. Also removes
4078 the arguments in case they weren't consumed already
4079 (fast_function() and err_args() leave them on the stack).
4080 */
4081 while ((*pp_stack) > pfunc) {
4082 w = EXT_POP(*pp_stack);
4083 Py_DECREF(w);
4084 PCALL(PCALL_POP);
4085 }
4086 return x;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004087}
4088
Jeremy Hylton192690e2002-08-16 18:36:11 +00004089/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004090 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004091 For the simplest case -- a function that takes only positional
4092 arguments and is called with only positional arguments -- it
4093 inlines the most primitive frame setup code from
4094 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4095 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004096*/
4097
4098static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004099fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004100{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004101 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
4102 PyObject *globals = PyFunction_GET_GLOBALS(func);
4103 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4104 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
4105 PyObject **d = NULL;
4106 int nd = 0;
Jeremy Hylton52820442001-01-03 23:52:36 +00004107
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004108 PCALL(PCALL_FUNCTION);
4109 PCALL(PCALL_FAST_FUNCTION);
4110 if (argdefs == NULL && co->co_argcount == n &&
4111 co->co_kwonlyargcount == 0 && nk==0 &&
4112 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4113 PyFrameObject *f;
4114 PyObject *retval = NULL;
4115 PyThreadState *tstate = PyThreadState_GET();
4116 PyObject **fastlocals, **stack;
4117 int i;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004118
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004119 PCALL(PCALL_FASTER_FUNCTION);
4120 assert(globals != NULL);
4121 /* XXX Perhaps we should create a specialized
4122 PyFrame_New() that doesn't take locals, but does
4123 take builtins without sanity checking them.
4124 */
4125 assert(tstate != NULL);
4126 f = PyFrame_New(tstate, co, globals, NULL);
4127 if (f == NULL)
4128 return NULL;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004130 fastlocals = f->f_localsplus;
4131 stack = (*pp_stack) - n;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004132
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004133 for (i = 0; i < n; i++) {
4134 Py_INCREF(*stack);
4135 fastlocals[i] = *stack++;
4136 }
4137 retval = PyEval_EvalFrameEx(f,0);
4138 ++tstate->recursion_depth;
4139 Py_DECREF(f);
4140 --tstate->recursion_depth;
4141 return retval;
4142 }
4143 if (argdefs != NULL) {
4144 d = &PyTuple_GET_ITEM(argdefs, 0);
4145 nd = Py_SIZE(argdefs);
4146 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004147 return PyEval_EvalCodeEx((PyObject*)co, globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004148 (PyObject *)NULL, (*pp_stack)-n, na,
4149 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
4150 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004151}
4152
4153static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004154update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4155 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004156{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004157 PyObject *kwdict = NULL;
4158 if (orig_kwdict == NULL)
4159 kwdict = PyDict_New();
4160 else {
4161 kwdict = PyDict_Copy(orig_kwdict);
4162 Py_DECREF(orig_kwdict);
4163 }
4164 if (kwdict == NULL)
4165 return NULL;
4166 while (--nk >= 0) {
4167 int err;
4168 PyObject *value = EXT_POP(*pp_stack);
4169 PyObject *key = EXT_POP(*pp_stack);
4170 if (PyDict_GetItem(kwdict, key) != NULL) {
4171 PyErr_Format(PyExc_TypeError,
4172 "%.200s%s got multiple values "
4173 "for keyword argument '%U'",
4174 PyEval_GetFuncName(func),
4175 PyEval_GetFuncDesc(func),
4176 key);
4177 Py_DECREF(key);
4178 Py_DECREF(value);
4179 Py_DECREF(kwdict);
4180 return NULL;
4181 }
4182 err = PyDict_SetItem(kwdict, key, value);
4183 Py_DECREF(key);
4184 Py_DECREF(value);
4185 if (err) {
4186 Py_DECREF(kwdict);
4187 return NULL;
4188 }
4189 }
4190 return kwdict;
Jeremy Hylton52820442001-01-03 23:52:36 +00004191}
4192
4193static PyObject *
4194update_star_args(int nstack, int nstar, PyObject *stararg,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004195 PyObject ***pp_stack)
Jeremy Hylton52820442001-01-03 23:52:36 +00004196{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004197 PyObject *callargs, *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004199 callargs = PyTuple_New(nstack + nstar);
4200 if (callargs == NULL) {
4201 return NULL;
4202 }
4203 if (nstar) {
4204 int i;
4205 for (i = 0; i < nstar; i++) {
4206 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4207 Py_INCREF(a);
4208 PyTuple_SET_ITEM(callargs, nstack + i, a);
4209 }
4210 }
4211 while (--nstack >= 0) {
4212 w = EXT_POP(*pp_stack);
4213 PyTuple_SET_ITEM(callargs, nstack, w);
4214 }
4215 return callargs;
Jeremy Hylton52820442001-01-03 23:52:36 +00004216}
4217
4218static PyObject *
4219load_args(PyObject ***pp_stack, int na)
4220{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004221 PyObject *args = PyTuple_New(na);
4222 PyObject *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004223
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004224 if (args == NULL)
4225 return NULL;
4226 while (--na >= 0) {
4227 w = EXT_POP(*pp_stack);
4228 PyTuple_SET_ITEM(args, na, w);
4229 }
4230 return args;
Jeremy Hylton52820442001-01-03 23:52:36 +00004231}
4232
4233static PyObject *
4234do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4235{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004236 PyObject *callargs = NULL;
4237 PyObject *kwdict = NULL;
4238 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004239
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004240 if (nk > 0) {
4241 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
4242 if (kwdict == NULL)
4243 goto call_fail;
4244 }
4245 callargs = load_args(pp_stack, na);
4246 if (callargs == NULL)
4247 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004248#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004249 /* At this point, we have to look at the type of func to
4250 update the call stats properly. Do it here so as to avoid
4251 exposing the call stats machinery outside ceval.c
4252 */
4253 if (PyFunction_Check(func))
4254 PCALL(PCALL_FUNCTION);
4255 else if (PyMethod_Check(func))
4256 PCALL(PCALL_METHOD);
4257 else if (PyType_Check(func))
4258 PCALL(PCALL_TYPE);
4259 else if (PyCFunction_Check(func))
4260 PCALL(PCALL_CFUNCTION);
4261 else
4262 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004263#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004264 if (PyCFunction_Check(func)) {
4265 PyThreadState *tstate = PyThreadState_GET();
4266 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4267 }
4268 else
4269 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00004270call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004271 Py_XDECREF(callargs);
4272 Py_XDECREF(kwdict);
4273 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004274}
4275
4276static PyObject *
4277ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4278{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004279 int nstar = 0;
4280 PyObject *callargs = NULL;
4281 PyObject *stararg = NULL;
4282 PyObject *kwdict = NULL;
4283 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004284
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004285 if (flags & CALL_FLAG_KW) {
4286 kwdict = EXT_POP(*pp_stack);
4287 if (!PyDict_Check(kwdict)) {
4288 PyObject *d;
4289 d = PyDict_New();
4290 if (d == NULL)
4291 goto ext_call_fail;
4292 if (PyDict_Update(d, kwdict) != 0) {
4293 Py_DECREF(d);
4294 /* PyDict_Update raises attribute
4295 * error (percolated from an attempt
4296 * to get 'keys' attribute) instead of
4297 * a type error if its second argument
4298 * is not a mapping.
4299 */
4300 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4301 PyErr_Format(PyExc_TypeError,
4302 "%.200s%.200s argument after ** "
4303 "must be a mapping, not %.200s",
4304 PyEval_GetFuncName(func),
4305 PyEval_GetFuncDesc(func),
4306 kwdict->ob_type->tp_name);
4307 }
4308 goto ext_call_fail;
4309 }
4310 Py_DECREF(kwdict);
4311 kwdict = d;
4312 }
4313 }
4314 if (flags & CALL_FLAG_VAR) {
4315 stararg = EXT_POP(*pp_stack);
4316 if (!PyTuple_Check(stararg)) {
4317 PyObject *t = NULL;
4318 t = PySequence_Tuple(stararg);
4319 if (t == NULL) {
4320 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4321 PyErr_Format(PyExc_TypeError,
4322 "%.200s%.200s argument after * "
Victor Stinner0a5f65a2011-03-22 01:09:21 +01004323 "must be a sequence, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004324 PyEval_GetFuncName(func),
4325 PyEval_GetFuncDesc(func),
4326 stararg->ob_type->tp_name);
4327 }
4328 goto ext_call_fail;
4329 }
4330 Py_DECREF(stararg);
4331 stararg = t;
4332 }
4333 nstar = PyTuple_GET_SIZE(stararg);
4334 }
4335 if (nk > 0) {
4336 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
4337 if (kwdict == NULL)
4338 goto ext_call_fail;
4339 }
4340 callargs = update_star_args(na, nstar, stararg, pp_stack);
4341 if (callargs == NULL)
4342 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004343#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004344 /* At this point, we have to look at the type of func to
4345 update the call stats properly. Do it here so as to avoid
4346 exposing the call stats machinery outside ceval.c
4347 */
4348 if (PyFunction_Check(func))
4349 PCALL(PCALL_FUNCTION);
4350 else if (PyMethod_Check(func))
4351 PCALL(PCALL_METHOD);
4352 else if (PyType_Check(func))
4353 PCALL(PCALL_TYPE);
4354 else if (PyCFunction_Check(func))
4355 PCALL(PCALL_CFUNCTION);
4356 else
4357 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004358#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004359 if (PyCFunction_Check(func)) {
4360 PyThreadState *tstate = PyThreadState_GET();
4361 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4362 }
4363 else
4364 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00004365ext_call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 Py_XDECREF(callargs);
4367 Py_XDECREF(kwdict);
4368 Py_XDECREF(stararg);
4369 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004370}
4371
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004372/* Extract a slice index from a PyInt or PyLong or an object with the
4373 nb_index slot defined, and store in *pi.
4374 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4375 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 +00004376 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004377*/
Tim Petersb5196382001-12-16 19:44:20 +00004378/* Note: If v is NULL, return success without storing into *pi. This
4379 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4380 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004381*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004382int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004383_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004384{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004385 if (v != NULL) {
4386 Py_ssize_t x;
4387 if (PyIndex_Check(v)) {
4388 x = PyNumber_AsSsize_t(v, NULL);
4389 if (x == -1 && PyErr_Occurred())
4390 return 0;
4391 }
4392 else {
4393 PyErr_SetString(PyExc_TypeError,
4394 "slice indices must be integers or "
4395 "None or have an __index__ method");
4396 return 0;
4397 }
4398 *pi = x;
4399 }
4400 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004401}
4402
Guido van Rossum486364b2007-06-30 05:01:58 +00004403#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004404 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00004405
Guido van Rossumb209a111997-04-29 18:18:01 +00004406static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004407cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004408{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004409 int res = 0;
4410 switch (op) {
4411 case PyCmp_IS:
4412 res = (v == w);
4413 break;
4414 case PyCmp_IS_NOT:
4415 res = (v != w);
4416 break;
4417 case PyCmp_IN:
4418 res = PySequence_Contains(w, v);
4419 if (res < 0)
4420 return NULL;
4421 break;
4422 case PyCmp_NOT_IN:
4423 res = PySequence_Contains(w, v);
4424 if (res < 0)
4425 return NULL;
4426 res = !res;
4427 break;
4428 case PyCmp_EXC_MATCH:
4429 if (PyTuple_Check(w)) {
4430 Py_ssize_t i, length;
4431 length = PyTuple_Size(w);
4432 for (i = 0; i < length; i += 1) {
4433 PyObject *exc = PyTuple_GET_ITEM(w, i);
4434 if (!PyExceptionClass_Check(exc)) {
4435 PyErr_SetString(PyExc_TypeError,
4436 CANNOT_CATCH_MSG);
4437 return NULL;
4438 }
4439 }
4440 }
4441 else {
4442 if (!PyExceptionClass_Check(w)) {
4443 PyErr_SetString(PyExc_TypeError,
4444 CANNOT_CATCH_MSG);
4445 return NULL;
4446 }
4447 }
4448 res = PyErr_GivenExceptionMatches(v, w);
4449 break;
4450 default:
4451 return PyObject_RichCompare(v, w, op);
4452 }
4453 v = res ? Py_True : Py_False;
4454 Py_INCREF(v);
4455 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004456}
4457
Thomas Wouters52152252000-08-17 22:55:00 +00004458static PyObject *
4459import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004460{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004461 PyObject *x;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004463 x = PyObject_GetAttr(v, name);
4464 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
4465 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
4466 }
4467 return x;
Thomas Wouters52152252000-08-17 22:55:00 +00004468}
Guido van Rossumac7be682001-01-17 15:42:30 +00004469
Thomas Wouters52152252000-08-17 22:55:00 +00004470static int
4471import_all_from(PyObject *locals, PyObject *v)
4472{
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004473 _Py_IDENTIFIER(__all__);
4474 _Py_IDENTIFIER(__dict__);
4475 PyObject *all = _PyObject_GetAttrId(v, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004476 PyObject *dict, *name, *value;
4477 int skip_leading_underscores = 0;
4478 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004479
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004480 if (all == NULL) {
4481 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4482 return -1; /* Unexpected error */
4483 PyErr_Clear();
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004484 dict = _PyObject_GetAttrId(v, &PyId___dict__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004485 if (dict == NULL) {
4486 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4487 return -1;
4488 PyErr_SetString(PyExc_ImportError,
4489 "from-import-* object has no __dict__ and no __all__");
4490 return -1;
4491 }
4492 all = PyMapping_Keys(dict);
4493 Py_DECREF(dict);
4494 if (all == NULL)
4495 return -1;
4496 skip_leading_underscores = 1;
4497 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004498
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004499 for (pos = 0, err = 0; ; pos++) {
4500 name = PySequence_GetItem(all, pos);
4501 if (name == NULL) {
4502 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4503 err = -1;
4504 else
4505 PyErr_Clear();
4506 break;
4507 }
4508 if (skip_leading_underscores &&
4509 PyUnicode_Check(name) &&
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004510 PyUnicode_READY(name) != -1 &&
4511 PyUnicode_READ_CHAR(name, 0) == '_')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004512 {
4513 Py_DECREF(name);
4514 continue;
4515 }
4516 value = PyObject_GetAttr(v, name);
4517 if (value == NULL)
4518 err = -1;
4519 else if (PyDict_CheckExact(locals))
4520 err = PyDict_SetItem(locals, name, value);
4521 else
4522 err = PyObject_SetItem(locals, name, value);
4523 Py_DECREF(name);
4524 Py_XDECREF(value);
4525 if (err != 0)
4526 break;
4527 }
4528 Py_DECREF(all);
4529 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004530}
4531
Guido van Rossumac7be682001-01-17 15:42:30 +00004532static void
Neal Norwitzda059e32007-08-26 05:33:45 +00004533format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00004534{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004535 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00004536
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004537 if (!obj)
4538 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004540 obj_str = _PyUnicode_AsString(obj);
4541 if (!obj_str)
4542 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004543
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004544 PyErr_Format(exc, format_str, obj_str);
Paul Prescode68140d2000-08-30 20:25:01 +00004545}
Guido van Rossum950361c1997-01-24 13:49:28 +00004546
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00004547static void
4548format_exc_unbound(PyCodeObject *co, int oparg)
4549{
4550 PyObject *name;
4551 /* Don't stomp existing exception */
4552 if (PyErr_Occurred())
4553 return;
4554 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
4555 name = PyTuple_GET_ITEM(co->co_cellvars,
4556 oparg);
4557 format_exc_check_arg(
4558 PyExc_UnboundLocalError,
4559 UNBOUNDLOCAL_ERROR_MSG,
4560 name);
4561 } else {
4562 name = PyTuple_GET_ITEM(co->co_freevars, oparg -
4563 PyTuple_GET_SIZE(co->co_cellvars));
4564 format_exc_check_arg(PyExc_NameError,
4565 UNBOUNDFREE_ERROR_MSG, name);
4566 }
4567}
4568
Victor Stinnerd2a915d2011-10-02 20:34:20 +02004569static PyObject *
4570unicode_concatenate(PyObject *v, PyObject *w,
4571 PyFrameObject *f, unsigned char *next_instr)
4572{
4573 PyObject *res;
4574 if (Py_REFCNT(v) == 2) {
4575 /* In the common case, there are 2 references to the value
4576 * stored in 'variable' when the += is performed: one on the
4577 * value stack (in 'v') and one still stored in the
4578 * 'variable'. We try to delete the variable now to reduce
4579 * the refcnt to 1.
4580 */
4581 switch (*next_instr) {
4582 case STORE_FAST:
4583 {
4584 int oparg = PEEKARG();
4585 PyObject **fastlocals = f->f_localsplus;
4586 if (GETLOCAL(oparg) == v)
4587 SETLOCAL(oparg, NULL);
4588 break;
4589 }
4590 case STORE_DEREF:
4591 {
4592 PyObject **freevars = (f->f_localsplus +
4593 f->f_code->co_nlocals);
4594 PyObject *c = freevars[PEEKARG()];
4595 if (PyCell_GET(c) == v)
4596 PyCell_Set(c, NULL);
4597 break;
4598 }
4599 case STORE_NAME:
4600 {
4601 PyObject *names = f->f_code->co_names;
4602 PyObject *name = GETITEM(names, PEEKARG());
4603 PyObject *locals = f->f_locals;
4604 if (PyDict_CheckExact(locals) &&
4605 PyDict_GetItem(locals, name) == v) {
4606 if (PyDict_DelItem(locals, name) != 0) {
4607 PyErr_Clear();
4608 }
4609 }
4610 break;
4611 }
4612 }
4613 }
4614 res = v;
4615 PyUnicode_Append(&res, w);
4616 return res;
4617}
4618
Guido van Rossum950361c1997-01-24 13:49:28 +00004619#ifdef DYNAMIC_EXECUTION_PROFILE
4620
Skip Montanarof118cb12001-10-15 20:51:38 +00004621static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004622getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004623{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004624 int i;
4625 PyObject *l = PyList_New(256);
4626 if (l == NULL) return NULL;
4627 for (i = 0; i < 256; i++) {
4628 PyObject *x = PyLong_FromLong(a[i]);
4629 if (x == NULL) {
4630 Py_DECREF(l);
4631 return NULL;
4632 }
4633 PyList_SetItem(l, i, x);
4634 }
4635 for (i = 0; i < 256; i++)
4636 a[i] = 0;
4637 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004638}
4639
4640PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004641_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004642{
4643#ifndef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004644 return getarray(dxp);
Guido van Rossum950361c1997-01-24 13:49:28 +00004645#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004646 int i;
4647 PyObject *l = PyList_New(257);
4648 if (l == NULL) return NULL;
4649 for (i = 0; i < 257; i++) {
4650 PyObject *x = getarray(dxpairs[i]);
4651 if (x == NULL) {
4652 Py_DECREF(l);
4653 return NULL;
4654 }
4655 PyList_SetItem(l, i, x);
4656 }
4657 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004658#endif
4659}
4660
4661#endif